feat: autofix unknown variable: a and file not found (searched at a.typ) by code action (#1743)

* feat: check context

* feat: implement it

* fix: warnings

* test: update snapshot

Author: Myriad-Dreamin

crates/tinymist-query/src/analysis/code_action.rs

@@ -377,11 +377,11 @@ type Cursor<'a> = CompletionCursor<'a>;
 
 /// A node selected by [`CompletionCursor`].
 enum SelectedNode<'a> {
-    /// Selects an identifier, e.g. `foo|` or `fo|o`.
+    /// Selects an identifier, e.g. `foobar|` or `foo|bar`.
     Ident(LinkedNode<'a>),
-    /// Selects a label, e.g. `<foo|>` or `<fo|o>`.
+    /// Selects a label, e.g. `<foobar|>` or `<foo|bar>`.
     Label(LinkedNode<'a>),
-    /// Selects a reference, e.g. `@foo|` or `@fo|o`.
+    /// Selects a reference, e.g. `@foobar|` or `@foo|bar`.
     Ref(LinkedNode<'a>),
 }
 

crates/tinymist-query/src/analysis/completion.rs

@@ -65,6 +65,14 @@ pub struct Analysis {
     pub allow_multiline_token: bool,
     /// Whether to remove html from markup content in responses.
     pub remove_html: bool,
+    /// Whether to utilize the extended `tinymist.resolveCodeAction` at client
+    /// side.
+    ///
+    /// The extended feature by `tinymist.resolveCodeAction`:
+    /// - supports Snippet edit.
+    ///
+    /// The example implementation can be found in the VS Code extension.
+    pub extended_code_action: bool,
     /// Tinymist's completion features.
     pub completion_feat: CompletionFeat,
     /// The editor's color theme.

crates/tinymist-query/src/analysis/global.rs

@@ -60,8 +60,10 @@ impl LinkTarget {
             LinkTarget::Url(url) => Some(url.as_ref().clone()),
             LinkTarget::Path(id, path) => {
                 // Avoid creating new ids here.
-                let root = ctx.path_for_id(id.join("")).ok()?;
-                crate::path_res_to_url(root.join(path).ok()?).ok()
+                let root = ctx.path_for_id(id.join("/")).ok()?;
+                let path_in_workspace = id.vpath().join(Path::new(path.as_str()));
+                let path = root.resolve_to(&path_in_workspace)?;
+                crate::path_res_to_url(path).ok()
             }
         }
     }

crates/tinymist-query/src/analysis/link_expr.rs

@@ -1,5 +1,9 @@
+use lsp_types::CodeActionContext;
+
 use crate::{analysis::CodeActionWorker, prelude::*, SemanticRequest};
 
+pub(crate) mod proto;
+
 /// The [`textDocument/codeAction`] request is sent from the client to the
 /// server to compute commands for a given text document and range. These
 /// commands are typically code fixes to either fix problems or to
@@ -64,18 +68,23 @@ pub struct CodeActionRequest {
     pub path: PathBuf,
     /// The range of the document to get code actions for.
     pub range: LspRange,
+    /// The context of the code action request.
+    pub context: CodeActionContext,
 }
 
 impl SemanticRequest for CodeActionRequest {
-    type Response = Vec<CodeActionOrCommand>;
+    type Response = Vec<CodeAction>;
 
     fn request(self, ctx: &mut LocalContext) -> Option<Self::Response> {
+        log::info!("requested code action: {self:?}");
+
         let source = ctx.source_by_path(&self.path).ok()?;
         let range = ctx.to_typst_range(self.range, &source)?;
 
         let root = LinkedNode::new(source.root());
         let mut worker = CodeActionWorker::new(ctx, source.clone());
-        worker.work(root, range);
+        worker.autofix(&root, &range, &self.context);
+        worker.scoped(&root, &range);
 
         (!worker.actions.is_empty()).then_some(worker.actions)
     }
@@ -94,12 +103,13 @@ mod tests {
             let request = CodeActionRequest {
                 path: path.clone(),
                 range: find_test_range(&source),
+                context: CodeActionContext::default(),
             };
 
             let result = request.request(ctx);
 
             with_settings!({
-                description => format!("Code Action on {}", make_range_annoation(&source)),
+                description => format!("Code Action on {}", make_range_annotation(&source)),
             }, {
                 assert_snapshot!(JsonRepr::new_redacted(result, &REDACT_LOC));
             })