Merge pull request #561 from myzhan/support_multiple_input

emmylua_doc_cli: 支持多个input

Author: CppCXY

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,34 @@ 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() {
+            match current_path.join(path).to_str() {
+                Some(p) => {
+                    files.push(p.to_string());
+                }
+                None => {
+                    eprintln!("Error: {} is not a valid path.", path.to_str().unwrap());
+                    exit(1);
+                }
+            }
+        } else {
+            match path.to_str() {
+                Some(p) => {
+                    files.push(p.to_string());
+                }
+                None => {
+                    eprintln!("Error: {} is not a valid path.", path.to_str().unwrap());
+                    exit(1);
+                }
+            }
+        }
     }
 
-    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(