parser support donot parse emmylua doc

Author: CppCXY

crates/emmylua_code_analysis/src/config/mod.rs

@@ -88,6 +88,7 @@ impl Emmyrc {
             Some(node_cache),
             special_like,
             non_std_symbols,
+            true,
         )
     }
 

crates/emmylua_parser/src/parser/lua_parser.rs

@@ -44,7 +44,6 @@ impl MarkerEventContainer for LuaParser<'_> {
 }
 
 impl<'a> LuaParser<'a> {
-    #[allow(unused)]
     pub fn parse(text: &'a str, config: ParserConfig) -> LuaSyntaxTree {
         let mut errors: Vec<LuaParseError> = Vec::new();
         let tokens = {
@@ -279,6 +278,16 @@ impl<'a> LuaParser<'a> {
     }
 
     fn parse_comments(&mut self, comment_tokens: &[LuaTokenData]) {
+        if !self.parse_config.support_emmylua_doc() {
+            for token in comment_tokens {
+                self.events.push(MarkEvent::EatToken {
+                    kind: token.kind,
+                    range: token.range,
+                });
+            }
+            return;
+        }
+
         let mut trivia_token_start = comment_tokens.len();
         // Reverse iterate over comment_tokens, removing whitespace and end-of-line tokens
         for i in (0..comment_tokens.len()).rev() {

crates/emmylua_parser/src/parser/parser_config.rs

@@ -9,6 +9,7 @@ pub struct ParserConfig<'cache> {
     lexer_config: LexerConfig,
     node_cache: Option<&'cache mut NodeCache>,
     special_like: HashMap<String, SpecialFunction>,
+    pub enable_emmylua_doc: bool,
 }
 
 impl<'cache> ParserConfig<'cache> {
@@ -17,6 +18,7 @@ impl<'cache> ParserConfig<'cache> {
         node_cache: Option<&'cache mut NodeCache>,
         special_like: HashMap<String, SpecialFunction>,
         non_std_symbols: LuaNonStdSymbolSet,
+        enable_emmylua_doc: bool,
     ) -> Self {
         Self {
             level,
@@ -26,6 +28,7 @@ impl<'cache> ParserConfig<'cache> {
             },
             node_cache,
             special_like,
+            enable_emmylua_doc,
         }
     }
 
@@ -37,6 +40,10 @@ impl<'cache> ParserConfig<'cache> {
         self.level >= LuaLanguageLevel::Lua54
     }
 
+    pub fn support_emmylua_doc(&self) -> bool {
+        self.enable_emmylua_doc
+    }
+
     pub fn node_cache(&mut self) -> Option<&mut NodeCache> {
         self.node_cache.as_deref_mut()
     }
@@ -64,6 +71,7 @@ impl<'cache> ParserConfig<'cache> {
             },
             node_cache: None,
             special_like: HashMap::new(),
+            enable_emmylua_doc: true,
         }
     }
 }
@@ -78,6 +86,7 @@ impl Default for ParserConfig<'_> {
             },
             node_cache: None,
             special_like: HashMap::new(),
+            enable_emmylua_doc: true,
         }
     }
 }

crates/emmylua_parser/src/syntax/tree/test.rs

@@ -43,6 +43,7 @@ end
             None,
             HashMap::new(),
             LuaNonStdSymbolSet::new(),
+            false,
         );
         let tree = LuaParser::parse(code, parse_config);
         assert_eq!(tree.get_errors().len(), 0);
@@ -86,4 +87,23 @@ local t
 
         let _ = LuaParser::parse(code, ParserConfig::default());
     }
+
+    #[test]
+    fn test_without_emmylua() {
+        let code = r#"
+        ---@param key string
+        ---@return boolean
+        local t
+        "#;
+
+        let c = ParserConfig::new(
+            LuaLanguageLevel::Lua54,
+            None,
+            HashMap::new(),
+            LuaNonStdSymbolSet::new(),
+            false,
+        );
+        let t = LuaParser::parse(code, c);
+        println!("{:#?}", t.get_red_root());
+    }
 }