From b338760d283ddd475536c887b07c2883289a04b0 Mon Sep 17 00:00:00 2001
From: Jordan Eldredge <jordan@jordaneldredge.com>
Date: Tue, 26 Mar 2024 08:40:27 -0700
Subject: [PATCH] Surface fixes in the playground

---
 .../components/PlaygroundFeatures/linter.ts   | 34 +++++++++++++++++--
 1 file changed, 32 insertions(+), 2 deletions(-)

diff --git a/website/src/components/PlaygroundFeatures/linter.ts b/website/src/components/PlaygroundFeatures/linter.ts
index 4e079151..0663f8f5 100644
--- a/website/src/components/PlaygroundFeatures/linter.ts
+++ b/website/src/components/PlaygroundFeatures/linter.ts
@@ -2,6 +2,7 @@ import { createSystem, createVirtualCompilerHost } from "@typescript/vfs";
 import * as ts from "typescript";
 import { buildSchemaAndDocResultWithHost } from "grats/src/lib";
 import { codegen } from "grats/src/codegen";
+import { ReportableDiagnostics } from "grats/src/utils/DiagnosticError";
 import { printSDLWithoutMetadata } from "grats/src/printSchema";
 import { linter } from "@codemirror/lint";
 import { DocumentNode, GraphQLSchema, print } from "graphql";
@@ -56,7 +57,10 @@ function buildSchemaResultWithFsMap(fsMap, text: string, config) {
     const message = `Grats playground bug encountered. Please report this error:\n\n ${e.stack}`;
     return {
       kind: "ERROR",
-      err: { formatDiagnosticsWithContext: () => message, _diagnostics: [] },
+      err: {
+        formatDiagnosticsWithContext: () => message,
+        _diagnostics: [] as ReportableDiagnostics[],
+      },
     };
   }
 }
@@ -81,12 +85,17 @@ export function createLinter(fsMap, view, config) {
       });
 
       return result.err._diagnostics.map((diagnostic) => {
+        const actions = [];
+        if (diagnostic.fix) {
+          const action = gratsFixToCodeMirrorAction(diagnostic.fix);
+          actions.push(action);
+        }
         return {
           from: diagnostic.start,
           to: diagnostic.start + diagnostic.length,
           severity: "error",
           message: diagnostic.messageText,
-          actions: [],
+          actions,
         };
       });
     }
@@ -124,3 +133,24 @@ function commentLines(text: string): string {
     .map((line) => `# ${line}`)
     .join("\n");
 }
+
+function gratsFixToCodeMirrorAction(fix) {
+  const change = fix.changes[0]?.textChanges[0];
+  if (change == null) {
+    return null;
+  }
+  return {
+    name: fix.description,
+    apply: (view) => {
+      view.dispatch({
+        changes: [
+          {
+            from: change.span.start,
+            to: change.span.start + change.span.length,
+            insert: change.newText,
+          },
+        ],
+      });
+    },
+  };
+}