Support emphasis in emmy annotator

Author: taminomara

crates/emmylua_ls/src/handlers/emmy_annotator/build_annotator.rs

@@ -1,14 +1,15 @@
 use std::collections::HashSet;
 
-use emmylua_code_analysis::{DbIndex, LuaDeclId, LuaDocument, SemanticModel};
+use super::{EmmyAnnotator, EmmyAnnotatorType};
+use crate::util::parse_desc;
+use emmylua_code_analysis::{DbIndex, LuaDeclId, LuaDocument, SemanticModel, WorkspaceId};
 use emmylua_parser::{
-    LuaAst, LuaAstNode, LuaAstToken, LuaForRangeStat, LuaForStat, LuaLocalFuncStat, LuaLocalStat,
-    LuaNameExpr, LuaParamList,
+    LuaAst, LuaAstNode, LuaAstToken, LuaDocDescription, LuaForRangeStat, LuaForStat,
+    LuaLocalFuncStat, LuaLocalStat, LuaNameExpr, LuaParamList,
 };
+use emmylua_parser_desc::DescItemKind;
 use rowan::TextRange;
 
-use super::{EmmyAnnotator, EmmyAnnotatorType};
-
 pub fn build_annotators(semantic: &SemanticModel) -> Vec<EmmyAnnotator> {
     let mut result = vec![];
     let document = semantic.get_document();
@@ -59,6 +60,20 @@ pub fn build_annotators(semantic: &SemanticModel) -> Vec<EmmyAnnotator> {
             LuaAst::LuaNameExpr(name_expr) => {
                 build_name_expr_annotator(&document, &mut use_range_set, &mut result, name_expr);
             }
+            LuaAst::LuaDocDescription(description) => {
+                if semantic
+                    .get_emmyrc()
+                    .semantic_tokens
+                    .render_documentation_markup
+                {
+                    build_description_annotator(
+                        &semantic,
+                        &mut use_range_set,
+                        &mut result,
+                        description,
+                    );
+                }
+            }
             _ => {}
         }
     }
@@ -291,3 +306,53 @@ fn build_local_func_stat_annotator(
 
     Some(())
 }
+
+fn build_description_annotator(
+    semantic_model: &SemanticModel,
+    use_range_set: &mut HashSet<TextRange>,
+    result: &mut Vec<EmmyAnnotator>,
+    description: LuaDocDescription,
+) -> Option<()> {
+    let document = semantic_model.get_document();
+    let text = document.get_text();
+    let items = parse_desc(
+        semantic_model
+            .get_module()
+            .map(|m| m.workspace_id)
+            .unwrap_or(WorkspaceId::MAIN),
+        semantic_model.get_emmyrc(),
+        text,
+        description,
+        None,
+    );
+
+    let mut strong = EmmyAnnotator {
+        typ: EmmyAnnotatorType::DocStrong,
+        ranges: vec![],
+    };
+    let mut em = EmmyAnnotator {
+        typ: EmmyAnnotatorType::DocEm,
+        ranges: vec![],
+    };
+
+    for item in items {
+        match item.kind {
+            DescItemKind::Em => {
+                use_range_set.insert(item.range.clone());
+                em.ranges.push(document.to_lsp_range(item.range.clone())?);
+            }
+            DescItemKind::Strong => {
+                use_range_set.insert(item.range.clone());
+                strong
+                    .ranges
+                    .push(document.to_lsp_range(item.range.clone())?);
+            }
+            _ => {}
+        }
+    }
+
+    result.push(em);
+    result.push(strong);
+
+    Some(())
+}

crates/emmylua_ls/src/handlers/emmy_annotator/emmy_annotator_request.rs

@@ -30,6 +30,8 @@ pub enum EmmyAnnotatorType {
     ReadOnlyLocal = 2,
     MutLocal = 3,
     MutParam = 4,
+    DocEm = 5,
+    DocStrong = 6,
 }
 
 impl From<EmmyAnnotatorType> for u8 {
@@ -46,6 +48,8 @@ impl From<u8> for EmmyAnnotatorType {
             2 => EmmyAnnotatorType::ReadOnlyLocal,
             3 => EmmyAnnotatorType::MutLocal,
             4 => EmmyAnnotatorType::MutParam,
+            5 => EmmyAnnotatorType::DocEm,
+            6 => EmmyAnnotatorType::DocStrong,
             _ => EmmyAnnotatorType::ReadOnlyLocal,
         }
     }