yyc12345/wfassoc
1
0
Fork 0
Code Activity

write shit

Browse Source
This commit is contained in:
yyc12345
2025-10-17 14:19:26 +08:00
parent dab91f1581
commit bdb6b4ceee
6 changed files with 251 additions and 115 deletions
Download Patch File Download Diff File Expand all files Collapse all files

86
Cargo.lock generated
View File

@ -377,6 +377,45 @@ version = "1.2.0"
source = "registry+https://github.com/rust-lang/crates.io-index" source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "94143f37725109f92c262ed2cf5e59bce7498c01bcc1502d7b9afe439a4e9f49" checksum = "94143f37725109f92c262ed2cf5e59bce7498c01bcc1502d7b9afe439a4e9f49"
[[package]]
name = "serde"
version = "1.0.228"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "9a8e94ea7f378bd32cbbd37198a4a91436180c5bb472411e48b5ec2e2124ae9e"
dependencies = [
"serde_core",
"serde_derive",
]
[[package]]
name = "serde_core"
version = "1.0.228"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "41d385c7d4ca58e59fc732af25c3983b67ac852c1a25000afe1175de458b67ad"
dependencies = [
"serde_derive",
]
[[package]]
name = "serde_derive"
version = "1.0.228"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "d540f220d3187173da220f885ab66608367b6574e925011a9353e4badda91d79"
dependencies = [
"proc-macro2",
"quote",
"syn",
]
[[package]]
name = "serde_spanned"
version = "1.0.3"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "e24345aa0fe688594e73770a5f6d1b216508b4f93484c0026d521acd30134392"
dependencies = [
"serde_core",
]
[[package]] [[package]]
name = "smallvec" name = "smallvec"
version = "1.15.1" version = "1.15.1"
@ -420,6 +459,45 @@ dependencies = [
"syn", "syn",
] ]
[[package]]
name = "toml"
version = "0.9.8"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "f0dc8b1fb61449e27716ec0e1bdf0f6b8f3e8f6b05391e8497b8b6d7804ea6d8"
dependencies = [
"indexmap",
"serde_core",
"serde_spanned",
"toml_datetime",
"toml_parser",
"toml_writer",
"winnow",
]
[[package]]
name = "toml_datetime"
version = "0.7.3"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "f2cdb639ebbc97961c51720f858597f7f24c4fc295327923af55b74c3c724533"
dependencies = [
"serde_core",
]
[[package]]
name = "toml_parser"
version = "1.0.4"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "c0cbe268d35bdb4bb5a56a2de88d0ad0eb70af5384a99d648cd4b3d04039800e"
dependencies = [
"winnow",
]
[[package]]
name = "toml_writer"
version = "1.0.4"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "df8b2b54733674ad286d16267dcfc7a71ed5c776e4ac7aa3c3e2561f7c637bf2"
[[package]] [[package]]
name = "unicode-ident" name = "unicode-ident"
version = "1.0.19" version = "1.0.19"
@ -539,7 +617,9 @@ version = "0.1.0"
dependencies = [ dependencies = [
"clap", "clap",
"comfy-table", "comfy-table",
"serde",
"thiserror", "thiserror",
"toml",
"wfassoc", "wfassoc",
] ]
@ -718,6 +798,12 @@ version = "0.53.0"
source = "registry+https://github.com/rust-lang/crates.io-index" source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "271414315aff87387382ec3d271b52d7ae78726f5d44ac98b4f4030c91880486" checksum = "271414315aff87387382ec3d271b52d7ae78726f5d44ac98b4f4030c91880486"
[[package]]
name = "winnow"
version = "0.7.13"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "21a0236b59786fed61e2a80582dd500fe61f18b5dca67a4a067d0bc9039339cf"
[[package]] [[package]]
name = "winreg" name = "winreg"
version = "0.55.0" version = "0.55.0"

42
wfassoc/src/lib.rs
View File

@ -4,8 +4,8 @@
#[cfg(not(target_os = "windows"))] #[cfg(not(target_os = "windows"))]
compile_error!("Crate wfassoc is only supported on Windows."); compile_error!("Crate wfassoc is only supported on Windows.");
pub(crate) mod assoc; pub mod assoc;
pub(crate) mod utilities; pub mod utilities;
use indexmap::{IndexMap, IndexSet}; use indexmap::{IndexMap, IndexSet};
use std::ffi::OsStr; use std::ffi::OsStr;
@ -26,6 +26,8 @@ pub enum Error {
BadRegOper(#[from] std::io::Error), BadRegOper(#[from] std::io::Error),
#[error("{0}")] #[error("{0}")]
CastOsStr(#[from] utilities::CastOsStrError), CastOsStr(#[from] utilities::CastOsStrError),
#[error("{0}")]
ParseExt(#[from] assoc::ParseExtError),
#[error("no administrative privilege")] #[error("no administrative privilege")]
NoPrivilege, NoPrivilege,
@ -85,7 +87,7 @@ impl TryFrom<View> for Scope {
impl Scope { impl Scope {
/// Check whether we have enough privilege when operating in current scope. /// Check whether we have enough privilege when operating in current scope.
/// If we have, return true, otherwise false. /// If we have, return true, otherwise false.
fn has_privilege(&self) -> bool { pub fn has_privilege(&self) -> bool {
// If we operate on System, and we do not has privilege, // If we operate on System, and we do not has privilege,
// we think we do not have privilege, otherwise, // we think we do not have privilege, otherwise,
// there is no privilege required. // there is no privilege required.
@ -127,7 +129,7 @@ pub struct Manner {
} }
impl Manner { impl Manner {
pub fn new(argv: &str) -> Self { fn new(argv: &str) -> Self {
Self { Self {
argv: argv.to_string(), argv: argv.to_string(),
} }
@ -167,23 +169,25 @@ impl Program {
/// ///
/// - `My App` /// - `My App`
/// - `3DViewer` /// - `3DViewer`
/// - `我的Qt最时尚` /// - `我的Qt程序` (means "My Qt App" in English)
/// ///
/// More preciously, `identifier` will be used as the vendor part of ProgId. /// More preciously, `identifier` will be used as the vendor part of ProgId.
/// ///
/// `full_path` is the fully qualified path to the application. /// `full_path` is the fully qualified path to the application.
pub fn new(identifier: &str, full_path: &Path) -> Self { pub fn new(identifier: &str, full_path: &Path) -> Result<Self> {
// TODO: Add checker for identifier // TODO: Add checker for identifier
Self { Ok(Self {
identifier: identifier.to_string(), identifier: identifier.to_string(),
full_path: full_path.to_path_buf(), full_path: full_path.to_path_buf(),
manners: IndexSet::new(), manners: IndexSet::new(),
exts: IndexMap::new(), exts: IndexMap::new(),
} })
} }
/// Add manner provided by this program. /// Add manner provided by this program.
pub fn add_manner(&mut self, manner: Manner) -> Result<Token> { pub fn add_manner(&mut self, manner: &str) -> Result<Token> {
// Create manner from string
let manner = Manner::new(manner);
// Backup a stringfied manner for error output. // Backup a stringfied manner for error output.
let manner_str = manner.to_string(); let manner_str = manner.to_string();
// Insert manner. // Insert manner.
@ -201,11 +205,14 @@ impl Program {
} }
/// Add file extension supported by this program and its associated manner. /// Add file extension supported by this program and its associated manner.
pub fn add_ext(&mut self, ext: assoc::Ext, token: Token) -> Result<Token> { pub fn add_ext(&mut self, ext: &str, token: Token) -> Result<Token> {
// Check manner token // Check manner token
if let None = self.get_manner(token) { if let None = self.get_manner(token) {
return Err(Error::InvalidAssocManner); return Err(Error::InvalidAssocManner);
} }
// Create extension from string
let ext = assoc::Ext::new(ext)?;
// Backup a stringfied extension for error output. // Backup a stringfied extension for error output.
let ext_str = ext.to_string(); let ext_str = ext.to_string();
// Insert file extension // Insert file extension
@ -353,4 +360,19 @@ impl Program {
} }
} }
impl Program {
/// Set the default "open with" of given extension to this program.
pub fn link_ext(&self, ext: Token) -> Result<()> {
todo!()
}
/// Remove this program from the default "open with" of given extension.
///
/// If the default "open with" of given extension is not our program,
/// this function do nothing.
pub fn unlink_ext(&self, ext: Token) -> Result<()> {
todo!()
}
}
// endregion // endregion

2
wfassoc_exec/Cargo.toml
View File

@ -10,4 +10,6 @@ license = "SPDX:MIT"
thiserror = { workspace = true } thiserror = { workspace = true }
wfassoc = { path="../wfassoc" } wfassoc = { path="../wfassoc" }
clap = { version="4.5.48", features=["derive"]} clap = { version="4.5.48", features=["derive"]}
serde = { version = "1.0.228", features=["derive"]}
toml = "0.9.8"
comfy-table = "7.2.1" comfy-table = "7.2.1"

52
wfassoc_exec/src/cli.rs Normal file
View File

@ -0,0 +1,52 @@
use clap::{Parser, Subcommand};
/// Simple program to manage Windows file associations
#[derive(Parser)]
#[command(name = "Windows File Association Operator", version, about)]
pub struct Cli {
/// The toml file introducing the complete program
#[arg(
short = 'c',
long = "config",
value_name = "PROG_CONFIG",
required = true
)]
pub(crate) config_file: String,
/// The scope where wfassoc operate
#[arg(short = 'f', long = "for", value_name = "TARGET", value_enum, default_value_t = ForTarget::User)]
pub(crate) for_which: ForTarget,
#[command(subcommand)]
pub(crate) command: Commands,
}
#[derive(clap::ValueEnum, Clone)]
pub 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)]
pub enum Commands {
/// Register the program
#[command(name = "register")]
Register,
/// Unregister the program
#[command(name = "unregister")]
Unregister,
/// Query file associations
#[command(name = "query")]
Query,
}

152
wfassoc_exec/src/main.rs
View File

@ -1,8 +1,12 @@
use clap::{Parser, Subcommand}; pub(crate) mod cli;
use comfy_table::Table; pub(crate) mod manifest;
use std::process;
use clap::Parser;
use std::{collections::HashMap, path::Path, process};
use thiserror::Error as TeError; use thiserror::Error as TeError;
use wfassoc::{Error as WfError, Ext, Scope, View}; use wfassoc::{Program, Token};
use cli::{Cli, Commands};
use manifest::Manifest;
// region: Basic Types // region: Basic Types
@ -11,7 +15,17 @@ use wfassoc::{Error as WfError, Ext, Scope, View};
enum Error { enum Error {
/// Error from wfassoc core. /// Error from wfassoc core.
#[error("{0}")] #[error("{0}")]
Core(#[from] WfError), Core(#[from] wfassoc::Error),
/// Error when parsing manifest TOML file.
#[error("invalid manifest file: {0}")]
Manifest(#[from] manifest::Error),
/// Error when specifying invalid manner name for extension.
#[error("extension {ext} associated manner {manner} is invalid in manifest file")]
InvalidMannerName {
manner: String,
ext: String
},
} }
/// Result type used in this executable. /// Result type used in this executable.
@ -19,116 +33,46 @@ type Result<T> = std::result::Result<T, Error>;
// endregion // 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 // region: Correponding Runner
fn run_register(cli: Cli) -> Result<()> { fn build_program(cli: &Cli) -> Result<Program> {
// Open file and read manifest TOML file
let mf = Manifest::from_file(&cli.config_file)?;
// Create instance
let rv = Program::new(&mf.identifier, &Path::from(mf.path.as_str()))?;
// Setup manner
let manners: HashMap<&str, Token> = HashMap::new();
for (k, v) in mf.manners.iter() {
let token = rv.add_manner(v.as_str())?;
manners.insert(k.as_str(), token);
}
// Setup extension
for (k, v) in mf.exts.iter() {
let token = match manners.get(v.as_str()) {
Some(v) => v,
None => return Err(Error::InvalidMannerName { manner: v.to_string(), ext: k.clone() }),
};
rv.add_ext(k.as_str(), *token)?;
}
// Okey
Ok(rv)
}
fn run_register(cli: &Cli) -> Result<()> {
// let program = Program::new(); // let program = Program::new();
// let kind: RegisterKind = cli.for_which.into(); // let kind: RegisterKind = cli.for_which.into();
// program.register(kind)?; // program.register(kind)?;
Ok(()) Ok(())
} }
fn run_unregister(cli: Cli) -> Result<()> { fn run_unregister(cli: &Cli) -> Result<()> {
// let program = Program::new(); // let program = Program::new();
// program.unregister()?; // program.unregister()?;
Ok(()) Ok(())
} }
fn run_query(cli: Cli) -> Result<()> { fn run_query(cli: &Cli) -> Result<()> {
let exts = [
".jpg", ".jfif", ".gif", ".bmp", ".png", ".ico", ".jpeg", ".tif", ".tiff", ".webp", ".svg",
".kra", ".xcf", ".avif", ".qoi", ".apng", ".exr",
];
for ext in exts.iter().map(|e| Ext::new(e).unwrap()) {
if let Some(ext_assoc) = ext.query(View::Hybrid) {
println!("{:?}", ext_assoc)
}
}
// let mut table = Table::new();
// table.set_header(["Extension", "Default Open", "Open With"]);
// for ext in exts.iter().map(|e| FileExt::new(e).unwrap()) {
// if let Some(ext_assoc) = ext.query(View::Hybrid) {
// if ext_assoc.len_open_with_progid() == 0 {
// table.add_row([ext.to_string().as_str(), ext_assoc.get_default(), ""]);
// } else {
// for (i, open_with_entry) in ext_assoc.iter_open_with_progids().enumerate() {
// if i == 0 {
// table.add_row([
// ext.to_string().as_str(),
// ext_assoc.get_default(),
// open_with_entry,
// ]);
// } else {
// table.add_row(["", "", open_with_entry]);
// }
// }
// }
// } else {
// table.add_row([ext.to_string().as_str(), "", ""]);
// }
// }
// println!("{table}");
// let program = Program::new();
// program.query()?;
//println!("Has privilege: {}", wfassoc::has_privilege());
Ok(()) Ok(())
} }
@ -138,9 +82,9 @@ fn main() {
let cli = Cli::parse(); let cli = Cli::parse();
let rv = match &cli.command { let rv = match &cli.command {
Commands::Register => run_register(cli), Commands::Register => run_register(&cli),
Commands::Unregister => run_unregister(cli), Commands::Unregister => run_unregister(&cli),
Commands::Query => run_query(cli), Commands::Query => run_query(&cli),
}; };
rv.unwrap_or_else(|e| { rv.unwrap_or_else(|e| {
eprintln!("Runtime error: {}.", e); eprintln!("Runtime error: {}.", e);

30
wfassoc_exec/src/manifest.rs Normal file
View File

@ -0,0 +1,30 @@
use serde::Deserialize;
use std::collections::HashMap;
use thiserror::Error as TeError;
use toml;
#[derive(Debug, TeError)]
#[error("{0}")]
pub enum Error {
Io(#[from] std::io::Error),
Toml(#[from] toml::de::Error),
}
pub type Result<T> = std::result::Result<T, Error>;
#[derive(Debug, Deserialize)]
pub struct Manifest {
pub(crate) identifier: String,
pub(crate) path: String,
pub(crate) clsid: String,
pub(crate) manners: HashMap<String, String>,
pub(crate) exts: HashMap<String, String>,
}
impl Manifest {
pub fn from_file(path: &str) -> Result<Manifest> {
let contents = std::fs::read_to_string(path)?;
let config: Manifest = toml::from_str(&contents)?;
Ok(config)
}
}