dns-geolocation-checker 0.2.1

A tool to check the geolocation of a domain based on its DNS records.
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
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114

use std::{env, net::IpAddr, sync::Arc};

use crate::{configs_parser::Config, ip_geo_checker::GeoIpResponse};

use super::{GetGeoIpInfo, IpGeoProviderType, NewProvider};

#[derive(Clone)]
pub struct MMDBClient {
    reader: Arc<maxminddb::Reader<Vec<u8>>>,
}

impl NewProvider for MMDBClient {
    fn new(config: &Config) -> Self {
        let mmdb_path = config
            .mmdb_path
            .clone()
            .unwrap_or(env::var("MMDB_PATH").unwrap_or("./mmdb/GeoLite2-City.mmdb".to_string()));
        let reader = maxminddb::Reader::open_readfile(mmdb_path).unwrap();
        Self {
            reader: Arc::new(reader),
        }
    }

    fn get_provider_type(&self) -> String {
        IpGeoProviderType::MMDB.to_string()
    }
}

impl GetGeoIpInfo for MMDBClient {
    #[allow(refining_impl_trait)]
    async fn get_geoip_info(&self, ip: IpAddr) -> Result<GeoIpResponse, maxminddb::MaxMindDBError> {
        let ip = ip.to_string();
        let record: maxminddb::geoip2::City = self.reader.lookup(ip.parse().unwrap())?;
        Ok(GeoIpResponse {
            query: ip,
            country: record
                .country
                .clone()
                .unwrap()
                .iso_code
                .unwrap()
                .to_string(),
            country_code: record
                .country
                .clone()
                .unwrap()
                .iso_code
                .unwrap()
                .to_string(),
            region: record
                .subdivisions
                .clone()
                .unwrap()
                .get(0)
                .unwrap()
                .iso_code
                .unwrap()
                .to_string(),
            region_name: "".to_string(),
            city: record
                .city
                .unwrap()
                .names
                .unwrap()
                .get("en")
                .unwrap()
                .to_string(),
            lat: record.location.clone().unwrap().latitude.unwrap(),
            lon: record.location.clone().unwrap().longitude.unwrap(),
        })
    }

    #[allow(refining_impl_trait)]
    async fn batch_get_ip_info(
        &self,
        ips: &Vec<IpAddr>,
    ) -> Result<Vec<GeoIpResponse>, maxminddb::MaxMindDBError> {
        let ips = ips.iter().map(|a| a.to_string()).collect::<Vec<String>>();
        let mut results = vec![];
        for ip in ips.iter() {
            let record: maxminddb::geoip2::City = self.reader.lookup(ip.parse().unwrap())?;
            results.push(GeoIpResponse {
                query: ip.to_string(),
                country: record
                    .country
                    .clone()
                    .unwrap()
                    .iso_code
                    .unwrap()
                    .to_string(),
                country_code: record
                    .country
                    .clone()
                    .unwrap()
                    .iso_code
                    .unwrap()
                    .to_string(),
                region: "".to_string(),
                region_name: "".to_string(),
                city: record
                    .city
                    .unwrap()
                    .names
                    .unwrap()
                    .get("en")
                    .unwrap()
                    .to_string(),
                lat: record.location.clone().unwrap().latitude.unwrap(),
                lon: record.location.clone().unwrap().longitude.unwrap(),
            });
        }
        Ok(results)
    }
}