iocaine 3.4.0

The deadliest poison known to AI
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

// SPDX-FileCopyrightText: Gergely Nagy
// SPDX-FileContributor: Gergely Nagy
//
// SPDX-License-Identifier: MIT

use anyhow::Result;
use clap::Args;
use glob::Pattern;

use iocaine_powder::bullshit::SquashFS;

pub fn run(args: ShowEmbedArgs) -> Result<()> {
    let patterns = if args.patterns.is_empty() {
        Vec::from([Pattern::new("*")?])
    } else {
        args.patterns
    };

    let files: Vec<_> = SquashFS::iter()
        .filter(|name| patterns.iter().any(|pattern| pattern.matches(name)))
        .collect();

    for name in &files {
        if args.contents {
            if files.len() > 1 {
                println!("==> {name} <==");
            }
            // Allow unwrap() here, because the named file is guaranteed to
            // exist at this point.
            #[allow(clippy::unwrap_used)]
            let contents = SquashFS::get(name).unwrap();
            let contents = &String::from_utf8_lossy(&contents);
            println!("{contents}");
        } else {
            println!("{name}");
        }
    }

    if !files.is_empty() {
        return Ok(());
    }
    anyhow::bail!("No matching embedded files found");
}

#[derive(Clone, Debug, Args)]
pub struct ShowEmbedArgs {
    /// Show the contents of matching files, instead of their names.
    #[arg(short, long, default_value_t = false)]
    contents: bool,

    /// Filter the list to filenames matching any of the patterns.
    patterns: Vec<Pattern>,
}