feat: add err & warn num output

Author: Chronostasys

src/ast/compiler.rs

@@ -53,6 +53,8 @@ pub use jit::*;
 #[cfg(feature = "llvm")]
 #[salsa::tracked]
 pub fn compile(db: &dyn Db, docs: MemDocsInput, out: String, op: Options) {
+    use std::process::exit;
+
     IS_JIT.store(op.jit, Ordering::Relaxed);
 
     #[cfg(feature = "jit")]
@@ -68,7 +70,10 @@ pub fn compile(db: &dyn Db, docs: MemDocsInput, out: String, op: Options) {
         op,
         format!("{}[{:2}/{:2}]", LOOKING_GLASS, 1, total_steps),
     );
-    compile_dry(db, docs).unwrap();
+    let _ = compile_dry(db, docs).map_err(|e| {
+        pb.abandon_with_message(format!("{}", e.red()));
+        exit(1);
+    });
 
     pb.finish_with_message("中间代码分析完成");
     let is_present_only = op.printast || op.flow;
@@ -86,11 +91,11 @@ pub fn compile(db: &dyn Db, docs: MemDocsInput, out: String, op: Options) {
 /// compile_dry compiles the source code of pivot-lang into the pivot-lang AST with a wrapper.
 /// the `dry` refers the function ends up parsing at LLVM IR or LSP analysis.
 #[salsa::tracked]
-pub fn compile_dry(db: &dyn Db, docs: MemDocsInput) -> Option<ModWrapper> {
+pub fn compile_dry(db: &dyn Db, docs: MemDocsInput) -> Result<ModWrapper, String> {
     let path = search_config_file(docs.file(db).to_string());
     if path.is_err() {
-        log::error!("lsp error: {}", path.err().unwrap());
-        return None;
+        log::warn!("lsp error: {}", path.err().unwrap());
+        return Err("project config file not found".to_string());
     }
 
     let parser_entry = docs
@@ -102,7 +107,7 @@ pub fn compile_dry(db: &dyn Db, docs: MemDocsInput) -> Option<ModWrapper> {
 
     // calculate find references results for lsp
     if docs.action(db) != ActionType::FindReferences {
-        return re;
+        return re.ok_or("compile failed".to_string());
     }
 
     if let Some(res) = db.get_ref_str() {
@@ -113,7 +118,7 @@ pub fn compile_dry(db: &dyn Db, docs: MemDocsInput) -> Option<ModWrapper> {
         }
         db.set_ref_str(None);
     }
-    re
+    re.ok_or("compile failed".to_string())
 }
 
 /// compile_dry_file parses the file inside parser_entry into AST,

src/ast/compiler/progress.rs

@@ -11,6 +11,8 @@ lazy_static! {
     pub static ref CHECK_PROGRESS: ProgressBar = ProgressBar::hidden();
 }
 
+pub(crate) static CHECK: Emoji<'_, '_> = Emoji("🧐  ", ":-)");
+
 pub(crate) static LOOKING_GLASS: Emoji<'_, '_> = Emoji("🔍  ", ":-)");
 
 pub(crate) static TRUCK: Emoji<'_, '_> = Emoji("🚚  ", ":-)");

src/ast/diag.rs

@@ -323,6 +323,12 @@ impl PLDiag {
             DiagCode::Warn(_) => false,
         }
     }
+    pub fn is_warn(&self) -> bool {
+        match self.raw.code {
+            DiagCode::Err(_) => false,
+            DiagCode::Warn(_) => true,
+        }
+    }
     pub fn get_msg(&self) -> String {
         match self.raw.code {
             DiagCode::Err(code) => ERROR_MSG[&code].to_string(),
@@ -486,12 +492,11 @@ impl<'a> Cache<&str> for PLFileCache<'a> {
     }
 }
 
-mod dot;
-
 /// ensure_no_error validates the results after parsing from db and docs to ensure there is no error.
 /// it panic directly if there is an error.
 pub(crate) fn ensure_no_error(db: &dyn Db, docs: MemDocsInput) {
     let mut errs_num = 0;
+    let mut warn_num = 0;
     let errs = compile_dry::accumulated::<Diagnostics>(db, docs);
     if !errs.is_empty() {
         for e in errs.iter() {
@@ -509,24 +514,23 @@ pub(crate) fn ensure_no_error(db: &dyn Db, docs: MemDocsInput) {
                 );
                 if e.is_err() {
                     errs_num += 1
+                } else if e.is_warn() {
+                    warn_num += 1
                 }
             }
         }
         if errs_num > 0 {
-            if errs_num == 1 {
-                log::error!(
-                    "{}",
-                    format!("check failed: there is {} error", errs_num).bright_red()
-                );
-                println!("{}", dot::ERROR);
-                exit(1);
-            }
-            log::error!(
-                "{}",
-                format!("check failed: there are {} errors", errs_num).bright_red()
+            eprintln!(
+                "check failed: {} error(s), {} warning(s)",
+                errs_num.to_string().bright_red(),
+                warn_num.to_string().yellow()
             );
-            println!("{}", dot::TOOMANYERROR);
             exit(1);
         }
+        eprintln!(
+            "    {} error(s), {} warning(s)",
+            errs_num.to_string().bright_red(),
+            warn_num.to_string().yellow()
+        );
     }
 }