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

use std::io;

use clap::{arg, command, Parser, Subcommand};
use puffin_build::vendor;
use regex::Regex;

#[derive(Debug, Parser)]
#[command(about = "helper script to build vendor libraries for tlspuffin", long_about = None)]
struct Cli {
    #[command(subcommand)]
    command: Commands,
}

#[derive(Debug, Subcommand)]
enum Commands {
    /// `mk_vendor make [--force] [--name=<name>] <config>`
    #[command(arg_required_else_help = true)]
    Make {
        /// The configuration to build (e.g. 'openssl:openssl312')
        #[arg(value_name = "CONFIG")]
        #[arg(value_parser = parse_config_arg)]
        config: (String, String),

        /// Override the preset's name
        #[arg(short, long)]
        name: Option<String>,

        /// Force configuration rebuild if it already exists
        #[arg(short, long)]
        #[arg(default_value_t = false)]
        force: bool,
    },
}

pub fn main() -> std::process::ExitCode {
    env_logger::Builder::new()
        .filter_level(log::LevelFilter::Info)
        .parse_env(
            env_logger::Env::default()
                .filter("MK_VENDOR_LOG")
                .write_style("MK_VENDOR_LOG_STYLE"),
        )
        .init();

    let Cli { command } = Cli::parse();

    match command {
        Commands::Make {
            config: (vendor, preset),
            name,
            force,
        } => {
            let Some(config) = vendor::Config::preset(&vendor, &preset) else {
                log::error!("configuration preset '{preset}' not found");
                return std::process::ExitCode::FAILURE;
            };

            let name = name.unwrap_or(preset);

            if let Err(e) = vendor::dir().lock(&name).and_then(|config_dir| {
                if !config_dir.is_empty()? && !force {
                    return Err(io::Error::new(
                        io::ErrorKind::AlreadyExists,
                        format!("a config named '{}' already exists", &name),
                    ));
                }

                config_dir.make(config)
            }) {
                log::error!("Error while building vendor library '{name}': {e}");
                return std::process::ExitCode::FAILURE;
            }

            std::process::ExitCode::SUCCESS
        }
    }
}

fn parse_config_arg(s: &str) -> Result<(String, String), String> {
    let s = s.trim();

    let Some(captures) = Regex::new(r"^(?<vendor>[^:]+):(?<name>[^:]+)$")
        .unwrap()
        .captures(s)
    else {
        return Err("invalid config format (expected '<vendor>:<name>')".to_string());
    };

    Ok((captures["vendor"].to_string(), captures["name"].to_string()))
}