phishtank 0.1.0

Phishtank API
Documentation 
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

use crate::{PhishtankError, PhishtankResult};
use reqwest::{
    blocking::{Client, Response},
    StatusCode,
};
use serde::de::DeserializeOwned;

#[inline]
fn process_resp<T>(resp: Response) -> PhishtankResult<T>
where
    T: DeserializeOwned,
{
    let status = resp.status();

    match status {
        StatusCode::OK => {
            let resp_text = resp.text().unwrap();
            // let resp_json = resp.json();
            if resp_text.contains("You must supply a URL to use this function") {
                Err(PhishtankError::InvalidUrl)
            } else {
                Ok(serde_json::from_str(&resp_text)?)
            }
        } // 200
        _ => Err(status.into()),
    }
}

#[inline]
fn process_resp_bz<T>(resp: Response) -> PhishtankResult<T>
where
    T: DeserializeOwned,
{
    let status = resp.status();

    match status {
        StatusCode::OK => {
            let read = bzip2::read::BzDecoder::new(resp);
            Ok(serde_json::from_reader(read)?) // 200
        }
        _ => Err(status.into()),
    }
}

// POST to a URL
pub(crate) fn http_post<T>(
    user_agent: &str,
    url: &str,
    form_data: &[(&str, &str)],
) -> PhishtankResult<T>
where
    T: DeserializeOwned,
{
    let client = Client::builder().user_agent(user_agent).build()?;
    let resp = client
        .post(url)
        .header("Accept", "application/json")
        .form(form_data)
        .send()?;
    process_resp(resp)
}

/// GET from a URL with bzip response
pub(crate) fn http_get_bz<T>(user_agent: &str, url: &str) -> PhishtankResult<T>
where
    T: DeserializeOwned,
{
    let client = Client::builder().user_agent(user_agent).build()?;
    let resp = client.get(url).send()?;
    process_resp_bz(resp)
}