support emmylua_check as lib

Author: CppCXY

crates/emmylua_check/Cargo.toml

@@ -23,7 +23,7 @@ log.workspace = true
 fern.workspace = true
 rowan.workspace = true
 walkdir.workspace = true
-tokio.workspace = true
 tokio-util.workspace = true
 clap.workspace = true
 ansi_term.workspace = true
+tokio.workspace = true

crates/emmylua_check/src/init.rs

@@ -1,7 +1,7 @@
 use std::{path::PathBuf, str::FromStr, sync::Arc};
 
 use emmylua_code_analysis::{
-    load_configs, load_workspace_files, update_code_style, EmmyLuaAnalysis, Emmyrc, LuaFileInfo,
+    load_configs, load_workspace_files, update_code_style, DbIndex, EmmyLuaAnalysis, Emmyrc, FileId, LuaFileInfo
 };
 
 fn root_from_configs(config_paths: &Vec<PathBuf>, fallback: &PathBuf) -> PathBuf {
@@ -174,3 +174,15 @@ pub fn calculate_include_and_exclude(
 
     (include, exclude, exclude_dirs)
 }
+
+pub fn get_need_check_ids(db: &DbIndex, files: Vec<FileId>, workspace: &PathBuf) -> Vec<FileId> {
+    let mut need_check_files = Vec::new();
+    for file_id in files {
+        let file_path = db.get_vfs().get_file_path(&file_id).unwrap();
+        if file_path.starts_with(workspace) {
+            need_check_files.push(file_id);
+        }
+    }
+
+    need_check_files
+}

crates/emmylua_check/src/lib.rs

@@ -0,0 +1,94 @@
+mod cmd_args;
+mod init;
+mod output;
+mod terminal_display;
+
+use cmd_args::CmdArgs;
+use fern::Dispatch;
+use log::LevelFilter;
+use output::output_result;
+use std::{error::Error, sync::Arc};
+use tokio_util::sync::CancellationToken;
+
+use crate::init::get_need_check_ids;
+
+#[allow(unused)]
+async fn run_check(cmd_args: CmdArgs) -> Result<(), Box<dyn Error + Sync + Send>> {
+    let mut workspace = cmd_args.workspace;
+    if !workspace.is_absolute() {
+        workspace = std::env::current_dir()?.join(workspace);
+    }
+
+    let verbose = cmd_args.verbose;
+    let logger = Dispatch::new()
+        .format(move |out, message, record| {
+            let (color, reset) = match record.level() {
+                log::Level::Error => ("\x1b[31m", "\x1b[0m"), // Red
+                log::Level::Warn => ("\x1b[33m", "\x1b[0m"),  // Yellow
+                log::Level::Info | log::Level::Debug | log::Level::Trace => ("", ""),
+            };
+            out.finish(format_args!(
+                "{}{}: {}{}",
+                color,
+                record.level(),
+                if verbose {
+                    format!("({}) {}", record.target(), message)
+                } else {
+                    message.to_string()
+                },
+                reset
+            ))
+        })
+        .level(if verbose {
+            LevelFilter::Info
+        } else {
+            LevelFilter::Warn
+        })
+        .chain(std::io::stderr());
+
+    if let Err(e) = logger.apply() {
+        eprintln!("Failed to apply logger: {:?}", e);
+    }
+
+    let analysis = match init::load_workspace(workspace.clone(), cmd_args.config, cmd_args.ignore) {
+        Some(analysis) => analysis,
+        None => {
+            return Err("Failed to load workspace".into());
+        }
+    };
+
+    let files = analysis.compilation.get_db().get_vfs().get_all_file_ids();
+    let db = analysis.compilation.get_db();
+    let need_check_files = get_need_check_ids(db, files, &workspace);
+
+    let (sender, receiver) = tokio::sync::mpsc::channel(100);
+    let analysis = Arc::new(analysis);
+    let db = analysis.compilation.get_db();
+    for file_id in need_check_files.clone() {
+        let sender = sender.clone();
+        let analysis = analysis.clone();
+        tokio::spawn(async move {
+            let cancel_token = CancellationToken::new();
+            let diagnostics = analysis.diagnose_file(file_id, cancel_token);
+            sender.send((file_id, diagnostics)).await.unwrap();
+        });
+    }
+
+    let exit_code = output_result(
+        need_check_files.len(),
+        db,
+        workspace,
+        receiver,
+        cmd_args.output_format,
+        cmd_args.output,
+        cmd_args.warnings_as_errors,
+    )
+    .await;
+
+    if exit_code != 0 {
+        return Err(format!("exit code: {}", exit_code).into());
+    }
+
+    eprintln!("Check finished");
+    Ok(())
+}

crates/emmylua_check/src/main.rs

@@ -5,13 +5,14 @@ mod terminal_display;
 
 use clap::Parser;
 use cmd_args::CmdArgs;
-use emmylua_code_analysis::{DbIndex, FileId};
 use fern::Dispatch;
 use log::LevelFilter;
 use output::output_result;
-use std::{error::Error, path::PathBuf, sync::Arc};
+use std::{error::Error, sync::Arc};
 use tokio_util::sync::CancellationToken;
 
+use crate::init::get_need_check_ids;
+
 #[tokio::main]
 async fn main() -> Result<(), Box<dyn Error + Sync + Send>> {
     let cmd_args = CmdArgs::parse();
@@ -95,14 +96,3 @@ async fn main() -> Result<(), Box<dyn Error + Sync + Send>> {
     Ok(())
 }
 
-fn get_need_check_ids(db: &DbIndex, files: Vec<FileId>, workspace: &PathBuf) -> Vec<FileId> {
-    let mut need_check_files = Vec::new();
-    for file_id in files {
-        let file_path = db.get_vfs().get_file_path(&file_id).unwrap();
-        if file_path.starts_with(workspace) {
-            need_check_files.push(file_id);
-        }
-    }
-
-    need_check_files
-}