1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
use reqwest::Client;
use serde_json::from_str;
use super::*;

impl <'a>VtClient<'a> {
    /// Upload and scan a file
    ///
    /// # Example
    /// 
    /// ```
    /// use virustotal::*;
    ///
    /// let vt = VtClient::new("Your API Key");
    /// vt.scan_file("eicar.txt")
    /// ```
    pub fn scan_file(self, filename: &str) -> FileScanResponse {
    
        let form = reqwest::multipart::Form::new()
            .text("apikey".to_string(), self.api_key.to_string())
            .file("file", filename)
            .expect("Not found");

        let url = &[self.endpoint, "/file/scan"].join("");
        let mut resp = Client::new()
            .post(url)
            .multipart(form)
            .send()
            .unwrap();

        let text: &str = &resp.text().unwrap();
        from_str(&text).unwrap()
    }

    /// Retrieve file scan reports
    ///
    /// # Example
    /// 
    /// ```
    /// use virustotal::*;
    ///
    /// let vt = VtClient::new("Your API Key");
    /// vt.report_file("eicar.txt")
    /// ```
    pub fn report_file(self, resource: &str) -> FileReportResponse {
        
        let params: &str = &format!(
            "?apikey={}&resource={}", self.api_key, &resource
        );
        let url = &[self.endpoint, "/file/report", params].join("");
        let mut resp = Client::new()
            .get(url)
            .send()
            .unwrap();
        
        let text: &str = &resp.text().unwrap();
        from_str(&text).unwrap()
     
    }
     
    /// Rescanning already submitted files
    ///
    /// # Example
    /// 
    /// ```
    /// use virustotal::*;
    ///
    /// let vt = VtClient::new("Your API Key");
    /// vt.rescan(resource);
    /// ```
    pub fn rescan_file(self, resource: &str) -> FileRescanResponse {

        let url = &[self.endpoint, "/file/rescan"].join("");
        let mut resp = Client::new()
            .post(url)
            .form(&[("apikey", self.api_key), ("resource", &resource)])
            .send()
            .unwrap();
        
        let text: &str = &resp.text().unwrap();
        from_str(&text).unwrap()
     
    }
}