1
0

write shit

This commit is contained in:
2025-10-05 18:04:20 +08:00
parent f3c7f0905e
commit 03a1796e07
8 changed files with 693 additions and 38 deletions

View File

@ -9,3 +9,4 @@ license = "SPDX:MIT"
[dependencies]
thiserror = { workspace = true }
wfassoc = { path="../wfassoc" }
clap = { version="4.5.48", features=["derive"]}

View File

@ -1,3 +1,94 @@
fn main() {
println!("Hello, world!");
use clap::{Parser, Subcommand};
use std::process;
use wfassoc::{Program, RegisterKind, WfError};
type Result<T> = std::result::Result<T, WfError>;
// 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)
}
fn run_unregister(cli: Cli) -> Result<()> {
let program = Program::new();
program.unregister()
}
fn run_query(cli: Cli) -> Result<()> {
let program = Program::new();
program.query()
}
// 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)
});
}