flippy 0.4.3

Automates upgrades and pulls remote databases, files, and firmware for the Flipper Zero
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

use std::path::PathBuf;

use anyhow::Context;
use tracing::{debug, instrument, trace};

use crate::types::{flip::Flip, mapping::MappingEntry};

#[instrument]
pub async fn run(
    mut flip: Flip,
    db_type: String,
    repo: String,
    path: PathBuf,
    excludes: bool,
) -> anyhow::Result<()> {
    debug!("Checking if repo {repo} exists");

    let repo = flip
        .repositories
        .get_mut(&repo)
        .context(format!("repository {} not found", repo))?;

    trace!("Pushing database mapping to list");

    let mapping = match db_type.as_str() {
        "subghz" => &mut repo.mappings.subghz,
        "rfid" => &mut repo.mappings.rfid,
        "nfc" => &mut repo.mappings.nfc,
        "ir" => &mut repo.mappings.ir,
        "ibutton" => &mut repo.mappings.ibutton,
        "badusb" => &mut repo.mappings.badusb,
        _ => unreachable!("matched by clap"),
    };

    if excludes {
        add_exclude(mapping, path);
    } else {
        add_include(mapping, path);
    }

    flip.write().await?;

    Ok(())
}

#[instrument]
fn add_include(mapping: &mut Option<MappingEntry>, path: PathBuf) {
    let path = path.to_string_lossy().to_string();

    match mapping {
        Some(existing) => {
            existing.include.push(path);
        }
        None => {
            *mapping = Some(MappingEntry {
                include: vec![path],
                exclude: vec![],
            })
        }
    }
}

#[instrument]
fn add_exclude(mapping: &mut Option<MappingEntry>, path: PathBuf) {
    let path = format!(":(exclude){}", path.to_string_lossy());

    match mapping {
        Some(existing) => {
            existing.exclude.push(path);
        }
        None => {
            *mapping = Some(MappingEntry {
                include: vec![],
                exclude: vec![path],
            })
        }
    }
}