wfassoc/wfassoc_exec/src/main.rs

use clap::{Parser, Subcommand};
use std::process;
use wfassoc::{Error as WfError};
use thiserror::Error as TeError;

// region: Basic Types

/// All errors occurs in this executable.
#[derive(TeError, Debug)]
enum Error {
    /// Error from wfassoc core.
    #[error("{0}")]
    Core(#[from] WfError)
}

/// Result type used in this executable.
type Result<T> = std::result::Result<T, Error>;

// endregion

// region: Command Line Parser

/// Simple program to manage Windows file associations
#[derive(Parser)]
#[command(name = "Windows File Association Operator", version, about)]
struct Cli {
    /// The toml file introducing the complete program
    #[arg(
        short = 'c',
        long = "config",
        value_name = "PROG_CONFIG",
        required = true
    )]
    config_file: String,

    /// The scope where wfassoc operate
    #[arg(short = 'f', long = "for", value_name = "TARGET", value_enum, default_value_t = ForTarget::User)]
    for_which: ForTarget,

    #[command(subcommand)]
    command: Commands,
}

#[derive(clap::ValueEnum, Clone)]
enum ForTarget {
    #[value(name = "user")]
    User,
    #[value(name = "system")]
    System,
}

// impl From<ForTarget> for RegisterKind {
//     fn from(target: ForTarget) -> Self {
//         match target {
//             ForTarget::User => RegisterKind::User,
//             ForTarget::System => RegisterKind::System,
//         }
//     }
// }

#[derive(Subcommand)]
enum Commands {
    /// Register the program
    #[command(name = "register")]
    Register,
    /// Unregister the program
    #[command(name = "unregister")]
    Unregister,
    /// Query file associations
    #[command(name = "query")]
    Query,
}

// endregion

// region: Correponding Runner

fn run_register(cli: Cli) -> Result<()> {
    // let program = Program::new();
    // let kind: RegisterKind = cli.for_which.into();
    // program.register(kind)?;
    Ok(())
}

fn run_unregister(cli: Cli) -> Result<()> {
    // let program = Program::new();
    // program.unregister()?;
    Ok(())
}

fn run_query(cli: Cli) -> Result<()> {
    // let program = Program::new();
    // program.query()?;
    println!("Has privilege: {}", wfassoc::has_privilege());
    Ok(())
}

// endregion

fn main() {
    let cli = Cli::parse();

    let rv = match &cli.command {
        Commands::Register => run_register(cli),
        Commands::Unregister => run_unregister(cli),
        Commands::Query => run_query(cli),
    };
    rv.unwrap_or_else(|e| {
        eprintln!("Runtime error: {}.", e);
        process::exit(1)
    });
}