p2sh 0.4.3

The p2sh Programming language interpreter
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

use clap::Parser;

#[derive(Parser, Debug)]
#[command(author, version, about, long_about)]
struct Args {
    /// Program via command line argument
    #[arg(short, long)]
    command: Option<String>,
    /// Script to run
    script: Option<String>,
    /// Script arguments
    args: Vec<String>,
    /// Skip pcap output in filter mode; defaults to false
    #[arg(short, long, default_value_t = false)]
    skip_pcap: bool,
}

pub struct CliArgs {
    args: Vec<String>,
    cmd: Option<String>,
    skip_pcap: bool,
}

impl CliArgs {
    pub fn new() -> Self {
        let cliargs = Args::parse();
        // Create args with script as first element
        let mut args = Vec::new();
        if let Some(script) = cliargs.script.clone() {
            args.push(script);
        }
        // Extend args with script arguments
        args.extend_from_slice(cliargs.args.as_slice());

        Self {
            args,
            cmd: cliargs.command,
            skip_pcap: cliargs.skip_pcap,
        }
    }
    pub fn get_args(&self) -> &[String] {
        self.args.as_slice()
    }
    pub fn get_cmd(&self) -> Option<String> {
        self.cmd.clone()
    }
    pub fn skip_pcap(&self) -> bool {
        self.skip_pcap
    }
}