feat: provide 'Execute Query' for /* GraphQL */ templates

graphql · Dec 10, 2021 · 6017872 · 6017872
1 parent f390366
commit 6017872
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 4 deletions.
diff --git a/.changeset/silver-radios-invent.md b/.changeset/silver-radios-invent.md
@@ -0,0 +1,5 @@
+---
+"vscode-graphql": patch
+---
+
+provide 'Execute Query' for `/* GraphQL */` templates
diff --git a/src/client/graphql-codelens-provider.ts b/src/client/graphql-codelens-provider.ts
@@ -6,6 +6,7 @@ import {
   CodeLens,
   Range,
   Position,
+  ProviderResult,
 } from "vscode"
 
 import { SourceHelper, ExtractedTemplateLiteral } from "./source-helper"
@@ -23,10 +24,14 @@ export class GraphQLCodeLensProvider implements CodeLensProvider {
   public provideCodeLenses(
     document: TextDocument,
     _token: CancellationToken,
-  ): CodeLens[] {
-    const literals: ExtractedTemplateLiteral[] =
-      this.sourceHelper.extractAllTemplateLiterals(document, ["gql", "graphql"])
-    return literals.map(literal => {
+    // for some reason, ProviderResult<CodeLens[]> doesn't work here
+    // anymore after upgrading types
+  ): ProviderResult<[]> {
+    const literals: ExtractedTemplateLiteral[] = this.sourceHelper.extractAllTemplateLiterals(
+      document,
+      ["gql", "graphql", "/\\* GraphQL \\*/"],
+    )
+    const results = literals.map(literal => {
       return new CodeLens(
         new Range(
           new Position(literal.position.line, 0),
@@ -39,5 +44,7 @@ export class GraphQLCodeLensProvider implements CodeLensProvider {
         },
       )
     })
+
+    return results as ProviderResult<[]>
   }
 }