emmylua_doc_cli: 支持多个input

Author: myzhan

crates/emmylua_doc_cli/src/cmd_args.rs

@@ -4,8 +4,8 @@ use std::path::PathBuf;
 #[derive(Debug, Parser)]
 pub struct CmdArgs {
     /// The path of the lua project
-    #[arg(long, short)]
-    pub input: PathBuf,
+    #[arg(long, short, num_args = 1..)]
+    pub input: Vec<PathBuf>,
 
     /// Format of the output, default is Markdown
     #[arg(long, short, default_value = "markdown")]

crates/emmylua_doc_cli/src/init.rs

@@ -5,7 +5,7 @@ use emmylua_code_analysis::{
 };
 
 #[allow(unused)]
-pub fn load_workspace(workspace_folders: Vec<&str>) -> Option<EmmyLuaAnalysis> {
+pub fn load_workspace(workspace_folders: Vec<String>) -> Option<EmmyLuaAnalysis> {
     let mut analysis = EmmyLuaAnalysis::new();
     analysis.init_std_lib(None);
 

crates/emmylua_doc_cli/src/main.rs

@@ -11,12 +11,19 @@ mod markdown_generator;
 
 fn main() {
     let args = CmdArgs::parse();
-    let mut input = args.input;
-    if input.is_relative() {
-        input = std::env::current_dir().ok().unwrap().join(&input);
+    let current_path = std::env::current_dir().ok().unwrap();
+    let input = args.input;
+    let mut files: Vec<String> = Vec::new();
+    for path in &input {
+        if path.is_relative() {
+            let abs_path = current_path.join(path).to_str().unwrap().to_string();
+            files.push(abs_path);
+        } else {
+            files.push(path.to_str().unwrap().to_string());
+        }
     }
 
-    let analysis = init::load_workspace(vec![input.to_str().unwrap()]);
+    let analysis = init::load_workspace(files);
     if let Some(mut analysis) = analysis {
         let res = match args.format {
             Format::Markdown => markdown_generator::generate_markdown(