rs1090 0.5.2

Rust library to decode Mode S and ADS-B signals
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
115
116
117
118
119
120

use std::num::ParseIntError;

use once_cell::sync::Lazy;
use regex::Regex;
use serde::{Deserialize, Serialize};

use super::tail::tail;

#[derive(Debug, Deserialize)]
pub struct Patterns {
    pub registers: Vec<Register>,
    pub properties: Vec<String>,
}

#[derive(Debug, Deserialize)]
pub struct Register {
    pub pattern: Option<String>,
    pub start: Option<String>,
    pub end: Option<String>,
    pub country: String,
    pub flag: String,
    pub comment: Option<String>,
    pub categories: Option<Vec<Category>>,
}

#[derive(Debug, Deserialize)]
pub struct Category {
    pub pattern: String,
    pub category: Option<String>,
    pub country: Option<String>,
    pub flag: Option<String>,
}

#[derive(Debug, Default, Serialize, Deserialize)]
pub struct AircraftInformation {
    pub icao24: String,
    pub registration: Option<String>,
    pub country: Option<String>,
    pub flag: Option<String>,
    pub pattern: Option<String>,
    pub category: Option<String>,
    pub comment: Option<String>,
}

const PATTERNS_JSON: &str = include_str!("../../data/patterns.json");
pub static PATTERNS: Lazy<Patterns> =
    Lazy::new(|| serde_json::from_str(PATTERNS_JSON).unwrap());

pub fn aircraft_information(
    icao24: &str,
    registration: Option<&str>,
) -> Result<AircraftInformation, ParseIntError> {
    let hexid = u32::from_str_radix(icao24, 16)?;

    let mut res = AircraftInformation {
        icao24: icao24.to_ascii_lowercase(),
        ..Default::default()
    };
    res.registration = tail(hexid);
    if let Some(tail) = registration {
        res.registration = Some(tail.to_string());
    }

    if let Some(pattern) = &PATTERNS.registers.iter().find(|elt| {
        if let Some(start) = &elt.start {
            if let Some(end) = &elt.end {
                let start = u32::from_str_radix(&start[2..], 16).unwrap();
                let end = u32::from_str_radix(&end[2..], 16).unwrap();
                return (hexid >= start) & (hexid <= end);
            }
        }
        false
    }) {
        res.country = Some(pattern.country.to_string());
        res.flag = Some(pattern.flag.to_string());
        if let Some(p) = &pattern.pattern {
            res.pattern = Some(p.to_string())
        }
        if let Some(comment) = &pattern.comment {
            res.comment = Some(comment.to_string())
        }

        if let Some(tail) = &res.registration {
            if let Some(categories) = &pattern.categories {
                if let Some(cat) = categories.iter().find(|elt| {
                    let re = Regex::new(&elt.pattern).unwrap();
                    re.is_match(tail)
                }) {
                    res.pattern = Some(cat.pattern.to_string());
                    if let Some(category) = &cat.category {
                        res.category = Some(category.to_string());
                    }
                    if let Some(country) = &cat.country {
                        res.country = Some(country.to_string());
                    }
                    if let Some(flag) = &cat.flag {
                        res.flag = Some(flag.to_string());
                    }
                }
            }
        }
    }
    Ok(res)
}

#[cfg(test)]
mod tests {
    use super::PATTERNS;

    #[test]
    fn test_find_country() {
        for register in &PATTERNS.registers {
            if register.country == "France" {
                assert_eq!(register.start, Some("0x380000".to_string()));
                return;
            }
        }
        unreachable!()
    }
}