update @export

Author: xuhuanzy

crates/emmylua_ls/src/handlers/completion/providers/doc_name_token_provider.rs

@@ -38,6 +38,9 @@ pub fn add_completion(builder: &mut CompletionBuilder) -> Option<()> {
         DocCompletionExpected::Using => {
             add_tag_using_completion(builder);
         }
+        DocCompletionExpected::Export => {
+            add_tag_export_completion(builder);
+        }
     }
 
     builder.stop_here();
@@ -74,6 +77,7 @@ fn get_doc_completion_expected(trigger_token: &LuaSyntaxToken) -> Option<DocComp
                 }
                 LuaTokenKind::TkTagNamespace => Some(DocCompletionExpected::Namespace),
                 LuaTokenKind::TkTagUsing => Some(DocCompletionExpected::Using),
+                LuaTokenKind::TkTagExport => Some(DocCompletionExpected::Export),
                 LuaTokenKind::TkComma => {
                     let parent = left_token.parent()?;
                     match parent.kind().into() {
@@ -122,6 +126,7 @@ enum DocCompletionExpected {
     ClassAttr,
     Namespace,
     Using,
+    Export,
 }
 
 fn add_tag_param_name_completion(builder: &mut CompletionBuilder) -> Option<()> {
@@ -306,3 +311,16 @@ fn add_tag_using_completion(builder: &mut CompletionBuilder) {
         builder.add_completion_item(completion_item);
     }
 }
+
+fn add_tag_export_completion(builder: &mut CompletionBuilder) {
+    let key = vec!["namespace", "global"];
+    for (sorted_index, key) in key.iter().enumerate() {
+        let completion_item = CompletionItem {
+            label: key.to_string(),
+            kind: Some(lsp_types::CompletionItemKind::ENUM_MEMBER),
+            sort_text: Some(format!("{:03}", sorted_index)),
+            ..Default::default()
+        };
+        builder.add_completion_item(completion_item);
+    }
+}

crates/emmylua_ls/src/handlers/completion/providers/env_provider.rs

@@ -59,9 +59,13 @@ fn check_can_add_completion(builder: &CompletionBuilder) -> Option<()> {
     } else if builder.trigger_kind == CompletionTriggerKind::INVOKED {
         let parent = builder.trigger_token.parent()?;
         let prev_token = builder.trigger_token.prev_token()?;
-        if prev_token.kind() == LuaTokenKind::TkTagUsing.into() {
-            return None;
+        match prev_token.kind().into() {
+            LuaTokenKind::TkTagUsing | LuaTokenKind::TkTagExport | LuaTokenKind::TkTagNamespace => {
+                return None;
+            }
+            _ => {}
         }
+
         // 即时是主动触发, 也不允许在函数定义的参数列表中添加
         if trigger_text == "(" {
             if LuaParamList::can_cast(parent.kind().into()) {

crates/emmylua_ls/src/handlers/semantic_token/build_semantic_tokens.rs

@@ -312,6 +312,10 @@ fn build_node_semantic_token(
             let name = doc_using.get_name_token()?;
             builder.push(name.syntax(), SemanticTokenType::NAMESPACE);
         }
+        LuaAst::LuaDocTagExport(doc_export) => {
+            let name = doc_export.get_name_token()?;
+            builder.push(name.syntax(), SemanticTokenType::NAMESPACE);
+        }
         LuaAst::LuaParamName(param_name) => {
             let name = param_name.get_name_token()?;
             if name.get_name_text() == "self" {