use reqwest::blocking::{multipart::Form, multipart::Part};
use std::{fs::File, io::Read};
mod response;
use response::{Attributes, Root, ScanRoot};
mod model;
pub use model::VtFiles;
use crate::{
utils::{http_get, http_multipart_post, http_post},
VtClient, VtResult,
};
impl VtClient {
pub fn file_info(&self, id: &str) -> VtResult<Root> {
let url = format!("{}/files/{}", &self.endpoint, id);
http_get(&self.api_key, &self.user_agent, &url)
}
pub fn file_scan(&self, file: &str) -> VtResult<ScanRoot> {
let mut f = File::open(file)?;
let mut buffer = Vec::new();
{
f.read_to_end(&mut buffer)?;
}
let form_data = Form::new().part("file", Part::bytes(buffer).file_name(file.to_owned()));
let url = format!("{}/files", &self.endpoint);
http_multipart_post(&self.api_key, &self.user_agent, &url, form_data)
}
pub fn file_rescan(&self, id: &str) -> VtResult<ScanRoot> {
let url = format!("{}/files/{}/analyse", &self.endpoint, id);
let form_data = &[("id", id)];
http_post(&self.api_key, &self.user_agent, &url, form_data)
}
}