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
pub use clap;
use clap::Parser;
use prettytable::format::consts::FORMAT_CLEAN;
use prettytable::{row, Attr, Cell, Row, Table};

mod parse_ip;
mod parse_mcu;
mod paths;

#[derive(Debug, Parser)]
pub struct Args {
    pub chip: String,
    pub filter: Option<String>,
}

pub fn run(args: &Args) -> anyhow::Result<()> {
    let filter = args.filter.clone().unwrap_or_default().to_ascii_lowercase();
    let db = paths::obtain_db(&args.chip)?;
    let db = std::fs::read_to_string(db)?;

    let db = std::thread::spawn(move || parse_mcu::parse_xml(&db));

    let gpios = std::thread::spawn(move || parse_ip::parse_ip());

    let mut db = db.join().unwrap()?;
    let gpios = gpios.join().unwrap()?;

    // convert gpios into HashMap name: Vec<PinSignal>
    let gpios = gpios
        .gpio_pins
        .into_iter()
        .map(|pin| (pin.name.to_string(), pin.pin_signals))
        .collect::<std::collections::HashMap<_, _>>();

    db.pins.sort_by_key(|pin| pin.name.to_ascii_lowercase());

    let mut table = Table::new();
    table.set_format(*FORMAT_CLEAN);
    table.add_row(Row::new(vec![
        Cell::new("Pin").with_style(Attr::Bold),
        Cell::new("Use").with_style(Attr::Bold),
        Cell::new("Mode").with_style(Attr::Bold),
    ]));

    for pin in db.pins {
        for signal in pin.signals {
            let signal_name = signal.name.to_ascii_lowercase();
            if signal_name.contains(&filter) {
                let gpio = gpios.get(&pin.name).unwrap();

                let signal = gpio
                    .iter()
                    .find(|signal| signal.name.to_ascii_lowercase() == signal_name)
                    .unwrap();

                let signal_value = signal
                    .specific_parameters
                    .iter()
                    .flat_map(|p| &p.possible_values)
                    .next()
                    .unwrap();

                let (_, signal_value, _) = signal_name_components(signal_value).unwrap();

                table.add_row(row![pin.name, signal.name, signal_value]);
            }
        }
    }

    table.printstd();

    Ok(())
}

/// name: GPIO_{X}_{Y}
fn signal_name_components(name: &str) -> Option<(&str, &str, &str)> {
    let mut parts = name.split('_');
    let gpio = parts.next()?;
    let x = parts.next()?;
    let y = parts.next()?;
    Some((gpio, x, y))
}