82 lines
1.8 KiB
Rust
82 lines
1.8 KiB
Rust
use crate::config::Context;
|
|
use crate::ip::{IpData, IpEvent};
|
|
use crate::utils::sleep_s;
|
|
|
|
use reqwest::Client;
|
|
use reqwest::Error as ReqError;
|
|
|
|
pub async fn send_to_ipbl_api(client: &Client, server: &str, ip: &IpEvent) {
|
|
let mut i = 1;
|
|
loop {
|
|
match push_ip(&client, &server, &ip.ipdata).await {
|
|
Ok(_) => {
|
|
break;
|
|
}
|
|
Err(err) => {
|
|
println!("{err}");
|
|
sleep_s(1).await;
|
|
if i == 10 {
|
|
break;
|
|
}
|
|
i += 1;
|
|
}
|
|
};
|
|
}
|
|
}
|
|
|
|
async fn push_ip(client: &Client, server: &str, ip: &IpData) -> Result<(), ReqError> {
|
|
let mut data: Vec<IpData> = vec![];
|
|
|
|
data.push(IpData {
|
|
ip: ip.ip.to_string(),
|
|
src: ip.src.to_string(),
|
|
date: ip.date.to_string(),
|
|
hostname: ip.hostname.to_string(),
|
|
});
|
|
|
|
client
|
|
.post(format!("{server}/ips"))
|
|
.json(&data)
|
|
.send()
|
|
.await?;
|
|
|
|
Ok(())
|
|
}
|
|
|
|
async fn _push_ip_bulk(
|
|
ctx: &Context,
|
|
ips: &Vec<IpData>,
|
|
ret: &mut Vec<String>,
|
|
) -> Result<(), ReqError> {
|
|
let result: String;
|
|
let mut data: Vec<IpData> = vec![];
|
|
|
|
for ip in ips {
|
|
data.push(IpData {
|
|
ip: ip.ip.to_string(),
|
|
src: ip.src.to_string(),
|
|
date: ip.date.to_string(),
|
|
hostname: ip.hostname.to_string(),
|
|
})
|
|
}
|
|
|
|
let resp = ctx
|
|
.client
|
|
.post(format!("{server}/ips", server = ctx.flags.server))
|
|
.json(&data)
|
|
.send()
|
|
.await?;
|
|
ret.push(format!("status: {status}", status = resp.status()));
|
|
|
|
let res = resp.text().await.unwrap();
|
|
|
|
if res.trim().len() > 0 {
|
|
result = res.trim().to_string();
|
|
} else {
|
|
result = "".to_string();
|
|
}
|
|
|
|
ret.push(format!("response: {result}"));
|
|
Ok(())
|
|
}
|