子命令 auto 支持参数指定源和目标路径

Signed-off-by: Jia Chao <jiachao2130@126.com>

src/cli.rs

@@ -74,4 +74,12 @@ pub fn parse() -> Cli {
 
 #[derive(Clone, Debug, Parser)]
 #[command(author, version, about, long_about = None)]
-pub struct AutoCli;
+pub struct AutoCli {
+    /// 指定 CVRF 公告所在的源路径
+    #[arg(long, short)]
+    pub source: Option<String>,
+
+    /// 转换 CUSA 后存储的目标路径
+    #[arg(long, short)]
+    pub target: Option<String>,
+}

src/config.rs

@@ -13,6 +13,11 @@ pub(crate) struct AutoConfig {
 }
 
 impl AutoConfig {
+    // 新建一个配置结构
+    pub fn new(source: String, target: String) -> Self {
+        Self { source, target }
+    }
+
     // 从文件中读取配置
     pub fn from<P: AsRef<Path>>(path: P) -> crate::Result<Self> {
         let data = std::fs::read_to_string(path)?;

src/lib.rs

@@ -8,7 +8,7 @@ use regex::Regex;
 use tracing::{debug, error, info, trace};
 use tracing_subscriber::{fmt, EnvFilter};
 
-use config::RepairConfig;
+use config::{AutoConfig, RepairConfig};
 use cvrf_xmlparser::{
     CVRF,
     // SaInfo 即为 CUSA
@@ -41,7 +41,7 @@ pub fn cumain() -> Result<()> {
     match cli.subcommand {
         cli::CliSub::Convert(cli) => covert(&cli),
         cli::CliSub::Db(cli) => sadb(&cli),
-        cli::CliSub::Auto(_) => auto(),
+        cli::CliSub::Auto(cli) => auto(&cli),
     }
 }
 
@@ -203,20 +203,30 @@ fn walk_dir<P: AsRef<Path>>(path: P, nodir: bool) -> Vec<PathBuf> {
 }
 
 /// 从配置文件中读取 cvrf 和 cusa 数据库的路径，并执行自动转换操作
-pub fn auto() -> Result<()> {
-    // 默认配置为，但也可读取执行命令路径下的配置
-    let config = {
-        let default = Path::new("/etc/cuvars/cvrf2cusa.json");
-        if default.is_file() {
-            default.to_str().unwrap()
-        } else {
-            "cvrf2cusa.json"
-        }
-    };
-    info!("Config file is: {}", config);
+pub fn auto(cli: &cli::AutoCli) -> Result<()> {
+    let auto = if cli.source.is_some() && cli.target.is_some() {
+        let auto = AutoConfig::new(
+            cli.source.as_ref().unwrap().to_owned(),
+            cli.target.as_ref().unwrap().to_owned(),
+        );
+        info!("从参数中读取 AutoConfig: {:#?}", auto);
+        auto
+    } else {
+        // 默认配置为，但也可读取执行命令路径下的配置
+        let config = {
+            let default = Path::new("/etc/cuvars/cvrf2cusa.json");
+            if default.is_file() {
+                default.to_str().unwrap()
+            } else {
+                "cvrf2cusa.json"
+            }
+        };
+        info!("Config file is: {}", config);
 
-    let auto = config::AutoConfig::from(config)?;
-    info!("Load AutoConfig: {:?}", auto);
+        let auto = config::AutoConfig::from(config)?;
+        info!("Load AutoConfig: {:#?}", auto);
+        auto
+    };
 
     let files = walk_dir(auto.source(), true);