From 6017872b7f19ef5c3fcad404fca9ffd5b8ba5d87 Mon Sep 17 00:00:00 2001
From: Aumy <46275902+AumyF@users.noreply.github.com>
Date: Fri, 10 Dec 2021 11:26:04 +0100
Subject: [PATCH] feat: provide 'Execute Query' for /* GraphQL */ templates

---
 .changeset/silver-radios-invent.md      |  5 +++++
 src/client/graphql-codelens-provider.ts | 15 +++++++++++----
 2 files changed, 16 insertions(+), 4 deletions(-)
 create mode 100644 .changeset/silver-radios-invent.md

diff --git a/.changeset/silver-radios-invent.md b/.changeset/silver-radios-invent.md
new file mode 100644
index 00000000..9c181a96
--- /dev/null
+++ 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
index eff60000..5c31aa77 100644
--- 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<[]>
   }
 }