feat(commands): add rust-analyzer.cancelFlycheck

fannheyward · Aug 22, 2022 · 73248ba · 73248ba
1 parent 47e279d
commit 73248ba
Show file tree

Hide file tree

Showing 5 changed files with 29 additions and 12 deletions.
diff --git a/README.md b/README.md
@@ -160,6 +160,7 @@ You can use these commands by `:CocCommand XYZ`.
 | rust-analyzer.viewCrateGraph | View Crate Graph |
 | rust-analyzer.viewFullCrateGraph | View Crate Graph (Full) |
 | rust-analyzer.shuffleCrateGraph | Shuffle Crate Graph |
+| rust-analyzer.cancelFlycheck | Cancel running flychecks |
 
 ## Highlight Group
 

diff --git a/package.json b/package.json
@@ -1029,6 +1029,11 @@
         "title": "Move item down",
         "category": "Rust Analyzer"
       },
+      {
+        "command": "rust-analyzer.cancelFlycheck",
+        "title": "Cancel running flychecks",
+        "category": "rust-analyzer"
+      },
       {
         "command": "rust-analyzer.explainError",
         "title": "Explain the currently hovered diagnostic",

diff --git a/src/commands.ts b/src/commands.ts
@@ -244,7 +244,7 @@ export function testCurrent(ctx: Ctx): Cmd {
       return;
     }
 
-    const testRunnable = runnables.find(run => run.label.startsWith('cargo test'));
+    const testRunnable = runnables.find((run) => run.label.startsWith('cargo test'));
     if (!testRunnable) return;
 
     await runSingle(ctx)(testRunnable);
@@ -537,6 +537,12 @@ export function applySnippetWorkspaceEditCommand(): Cmd {
   };
 }
 
+export function cancelFlycheck(ctx: Ctx): Cmd {
+  return async () => {
+    await ctx.client.sendRequest(ra.cancelFlycheck);
+  };
+}
+
 export function resolveCodeAction(ctx: Ctx): Cmd {
   return async (params: CodeAction) => {
     params.command = undefined;

diff --git a/src/index.ts b/src/index.ts
@@ -67,6 +67,7 @@ export async function activate(context: ExtensionContext): Promise<void> {
   ctx.registerCommand('matchingBrace', cmds.matchingBrace);
   ctx.registerCommand('openCargoToml', cmds.openCargoToml);
   ctx.registerCommand('serverVersion', cmds.serverVersion);
+  ctx.registerCommand('cancelFlycheck', cmds.cancelFlycheck);
   ctx.registerCommand('analyzerStatus', cmds.analyzerStatus);
   ctx.registerCommand('viewCrateGraph', cmds.viewCrateGraph);
   ctx.registerCommand('shuffleCrateGraph', cmds.shuffleCrateGraph);

diff --git a/src/lsp_ext.ts b/src/lsp_ext.ts
@@ -60,6 +60,21 @@ export interface ExpandedMacro {
 }
 export const expandMacro = new lc.RequestType<ExpandMacroParams, ExpandedMacro | null, void>('rust-analyzer/expandMacro');
 
+export const relatedTests = new lc.RequestType<lc.TextDocumentPositionParams, TestInfo[], void>('rust-analyzer/relatedTests');
+
+export const cancelFlycheck = new lc.RequestType0<void, void>('rust-analyzer/cancelFlycheck');
+
+// Experimental extensions
+
+export interface SsrParams {
+  query: string;
+  parseOnly: boolean;
+  textDocument: lc.TextDocumentIdentifier;
+  position: lc.Position;
+  selections: readonly lc.Range[];
+}
+export const ssr = new lc.RequestType<SsrParams, lc.WorkspaceEdit, void>('experimental/ssr');
+
 export interface MatchingBraceParams {
   textDocument: lc.TextDocumentIdentifier;
   positions: lc.Position[];
@@ -100,17 +115,6 @@ export interface TestInfo {
   runnable: Runnable;
 }
 
-export const relatedTests = new lc.RequestType<lc.TextDocumentPositionParams, TestInfo[], void>('rust-analyzer/relatedTests');
-
-export interface SsrParams {
-  query: string;
-  parseOnly: boolean;
-  textDocument: lc.TextDocumentIdentifier;
-  position: lc.Position;
-  selections: readonly lc.Range[];
-}
-export const ssr = new lc.RequestType<SsrParams, lc.WorkspaceEdit, void>('experimental/ssr');
-
 export interface CommandLink extends lc.Command {
   /**
    * A tooltip for the command, when represented in the UI.