add custom request for view syntax tree

Author: CppCXY

crates/emmylua_ls/src/handlers/emmy_syntax_tree/emmy_syntax_tree_request.rs

@@ -0,0 +1,21 @@
+use lsp_types::request::Request;
+use serde::{Deserialize, Serialize};
+
+#[derive(Debug)]
+pub enum EmmySyntaxTreeRequest {}
+
+impl Request for EmmySyntaxTreeRequest {
+    type Params = EmmySyntaxTreeParams;
+    type Result = Option<SyntaxTreeResponse>;
+    const METHOD: &'static str = "emmy/syntaxTree";
+}
+
+#[derive(Debug, Eq, PartialEq, Clone, Deserialize, Serialize)]
+pub struct EmmySyntaxTreeParams {
+    pub uri: String,
+}
+
+#[derive(Debug, Eq, PartialEq, Clone, Deserialize, Serialize)]
+pub struct SyntaxTreeResponse {
+    pub content: String,
+}

crates/emmylua_ls/src/handlers/emmy_syntax_tree/mod.rs

@@ -0,0 +1,30 @@
+mod emmy_syntax_tree_request;
+
+use std::str::FromStr;
+
+use emmylua_parser::LuaAstNode;
+use lsp_types::Uri;
+use tokio_util::sync::CancellationToken;
+
+use crate::{
+    context::ServerContextSnapshot,
+    handlers::emmy_syntax_tree::emmy_syntax_tree_request::{
+        EmmySyntaxTreeParams, SyntaxTreeResponse,
+    },
+};
+pub use emmy_syntax_tree_request::*;
+
+pub async fn on_emmy_syntax_tree_handler(
+    context: ServerContextSnapshot,
+    params: EmmySyntaxTreeParams,
+    _: CancellationToken,
+) -> Option<SyntaxTreeResponse> {
+    let uri = Uri::from_str(&params.uri).ok()?;
+    let analysis = context.analysis().read().await;
+    let file_id = analysis.get_file_id(&uri)?;
+    let semantic_model = analysis.compilation.get_semantic_model(file_id)?;
+
+    let root = semantic_model.get_root();
+    let content = format!("{:#?}", root.syntax());
+    Some(SyntaxTreeResponse { content })
+}

crates/emmylua_ls/src/handlers/mod.rs

@@ -16,6 +16,7 @@ mod document_symbol;
 mod document_type_format;
 mod emmy_annotator;
 mod emmy_gutter;
+mod emmy_syntax_tree;
 mod fold_range;
 mod hover;
 mod implementation;

crates/emmylua_ls/src/handlers/request_handler.rs

@@ -20,6 +20,7 @@ use crate::{
         diagnostic::{on_pull_document_diagnostic, on_pull_workspace_diagnostic},
         document_type_format::on_type_formatting_handler,
         emmy_gutter::{EmmyGutterRequest, on_emmy_gutter_handler},
+        emmy_syntax_tree::{EmmySyntaxTreeRequest, on_emmy_syntax_tree_handler},
     },
 };
 
@@ -96,6 +97,7 @@ pub async fn on_request_handler(
         DocumentLinkResolve => on_document_link_resolve_handler,
         EmmyAnnotatorRequest => on_emmy_annotator_handler,
         EmmyGutterRequest => on_emmy_gutter_handler,
+        EmmySyntaxTreeRequest => on_emmy_syntax_tree_handler,
         SelectionRangeRequest => on_document_selection_range_handle,
         Completion => on_completion_handler,
         ResolveCompletionItem => on_completion_resolve_handler,