ksmm 0.4.3

KernelSU Module Manager CLI
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

use clap::{Parser, Subcommand, builder::Styles, CommandFactory};
use std::env;

mod commands;

#[derive(Parser)]
#[command(
    name = "ksmm",
    about = "KernelSU Module Manager",
    styles = get_styles()
)]
struct Cli {
    #[command(subcommand)]
    command: Option<Commands>,

    /// Show version information
    #[arg(short = 'V', long)]
    version: bool,
}

fn get_styles() -> Styles {
    Styles::styled()
        .header(clap::builder::styling::AnsiColor::Green.on_default() | clap::builder::styling::Effects::BOLD)
        .usage(clap::builder::styling::AnsiColor::Cyan.on_default())
        .literal(clap::builder::styling::AnsiColor::Yellow.on_default())
        .placeholder(clap::builder::styling::AnsiColor::Magenta.on_default())
        .error(clap::builder::styling::AnsiColor::Red.on_default() | clap::builder::styling::Effects::BOLD)
}

#[derive(Subcommand)]
enum Commands {
    /// 初始化模块
    Init,
    /// 构建模块
    Build,
    /// 签名文件
    Sign {
        /// 要签名的文件
        file: String,
    },
    /// 密钥管理
    Key {
        #[command(subcommand)]
        key_command: commands::sign::KeyCommands,
    },
    /// 显示版本信息
    Version,
}

fn main() {
    unsafe {
        env::set_var("FORCE_COLOR", "1");
    }

    let cli = Cli::parse();

    // Handle version flag
    if cli.version {
        commands::version::execute();
        return;
    }

    // Handle commands
    match cli.command {
        Some(Commands::Build) => commands::build::execute(),
        Some(Commands::Init) => commands::init::execute(),
        Some(Commands::Sign { file }) => commands::sign::execute_sign_file(file),
        Some(Commands::Key { key_command }) => commands::sign::execute_key_command(key_command),
        Some(Commands::Version) => commands::version::execute(),
        None => {
            // No command provided, show help
            let _ = Cli::command().print_help();
        }
    }
}