74 lines
1.8 KiB
Rust
74 lines
1.8 KiB
Rust
use std::fs;
|
||
use std::io::Write;
|
||
use std::path::Path;
|
||
|
||
use cvrf_xmlparser::{
|
||
CVRF,
|
||
// SaInfo 即为 CUSA
|
||
// SaInfo,
|
||
};
|
||
|
||
pub mod cli;
|
||
|
||
/// 定义 crate::Error
|
||
/// 大部分函数返回的错误
|
||
pub type Error = Box<dyn std::error::Error + Send + Sync>;
|
||
|
||
/// 定义 crate::Result
|
||
pub type Result<T> = std::result::Result<T, Error>;
|
||
|
||
pub fn cumain() -> Result<()> {
|
||
let cli = cli::parse();
|
||
|
||
match cli.subcommand {
|
||
cli::CliSub::Convert(cli) => covert(&cli),
|
||
cli::CliSub::Db(cli) => sadb(&cli),
|
||
}
|
||
}
|
||
|
||
fn covert(cli: &cli::ConvertCli) -> Result<()> {
|
||
// 检查 input，此为必须项
|
||
let input = Path::new(&cli.input);
|
||
if !input.is_file() {
|
||
return Err("输入的文件不是有效的文件路径！".into());
|
||
}
|
||
|
||
// 从 input 读取 cvrf，并转换为 cusa
|
||
let mut cvrf = CVRF::new();
|
||
cvrf.load_xml(&cli.input)?;
|
||
let cusa = cvrf.sainfo();
|
||
|
||
let data = serde_json::to_string_pretty(&cusa)?;
|
||
|
||
// 是否将 cusa 输出至指定文件
|
||
if let Some(output) = &cli.output {
|
||
let output = Path::new(output);
|
||
|
||
// output 所在父路径须提前创建
|
||
let parent = output.parent().unwrap();
|
||
if !parent.exists() {
|
||
return Err("无法访问输出文件所在路径，请确认！".into());
|
||
}
|
||
|
||
// 写入文件
|
||
let mut json_file = fs::OpenOptions::new().read(true).write(true).create(true).open(output)?;
|
||
json_file.write(data.as_bytes())?;
|
||
json_file.flush()?;
|
||
|
||
// 在输出到文件模式下，默认不再标准输出打印 CUSA
|
||
if !cli.print {
|
||
return Ok(())
|
||
}
|
||
}
|
||
|
||
// 在标准输出打印 CUSA
|
||
println!("{}", data);
|
||
Ok(())
|
||
}
|
||
|
||
fn sadb(cli: &cli::SaDbCli) -> Result<()> {
|
||
let _ = cli;
|
||
|
||
Ok(())
|
||
}
|