From 25570c57c96d2f5489f1a1cf76a870dd9c461b4a Mon Sep 17 00:00:00 2001
From: aniket-engg <aniketengg.225@gmail.com>
Date: Wed, 26 Jul 2023 19:42:19 +0530
Subject: [PATCH] add quickfix for visibility

---
 .../src/lib/providers/codeActionProvider.ts   | 69 ++++++++++++-------
 .../editor/src/lib/providers/quickfixes.ts    |  4 ++
 2 files changed, 50 insertions(+), 23 deletions(-)

diff --git a/libs/remix-ui/editor/src/lib/providers/codeActionProvider.ts b/libs/remix-ui/editor/src/lib/providers/codeActionProvider.ts
index 4f37162731..12175c8646 100644
--- a/libs/remix-ui/editor/src/lib/providers/codeActionProvider.ts
+++ b/libs/remix-ui/editor/src/lib/providers/codeActionProvider.ts
@@ -2,6 +2,7 @@ import { Monaco } from "@monaco-editor/react"
 import monaco from "../../types/monaco"
 import { EditorUIProps } from "../remix-ui-editor"
 import { default as fixes } from "./quickfixes"
+import { monacoTypes } from "@remix-ui/editor"
 
 export class RemixCodeActionProvider implements monaco.languages.CodeActionProvider {
   props: EditorUIProps
@@ -12,34 +13,56 @@ export class RemixCodeActionProvider implements monaco.languages.CodeActionProvi
   }
 
   async provideCodeActions (
-    model /**ITextModel*/,
-    range /**Range*/,
-    context /**CodeActionContext*/,
-    token /**CancellationToken*/
+    model: monaco.editor.ITextModel /**ITextModel*/,
+    range: monaco.Range /**Range*/,
+    context: monaco.languages.CodeActionContext /**CodeActionContext*/,
+    token: monaco.CancellationToken /**CancellationToken*/
   ) {
-
-    const actions = context.markers.map(error => {
+    const actions = []
+    for (const error of context.markers) {
+      console.log('error----->', error)
       const errStrings = Object.keys(fixes)
       const errStr = errStrings.find(es => error.message.includes(es))
-      const fix = fixes[errStr]
-      return {
-        title: fix.title,
-        diagnostics: [error],
-        kind: "quickfix",
-        edit: {
-          edits: [
-            {
-              resource: model.uri,
-              edit: {
-                range: fix.range || error,
-                text: fix.message
-              }
+      if (errStr) {
+        let fix = fixes[errStr]
+        const cursorPosition = this.props.editorAPI.getHoverPosition({lineNumber: error.startLineNumber, column: error.startColumn})
+        const nodeAtPosition = await this.props.plugin.call('codeParser', 'definitionAtPosition', cursorPosition)
+        if (nodeAtPosition && nodeAtPosition.nodeType === "FunctionDefinition") {
+          console.log('nodeAtPosition---->', nodeAtPosition)
+          if (nodeAtPosition.parameters && nodeAtPosition.parameters.length > 0) {
+            const paramNodes = nodeAtPosition.parameters
+            const lastParamNode = paramNodes[paramNodes.length - 1]
+            console.log('lastParamNode---->', lastParamNode)
+            const lastParamEndLoc = lastParamNode.loc.end
+            console.log('lastParamEndLoc---->', lastParamEndLoc)
+            fix.range = {
+              startLineNumber: lastParamEndLoc.line,
+              endLineNumber: lastParamEndLoc.line,
+              startColumn: lastParamEndLoc.column + 11,
+              endColumn: lastParamEndLoc.column + 19
             }
-          ]
-        },
-        isPreferred: true
+          }
+        }
+        actions.push({
+          title: fix.title,
+          diagnostics: [error],
+          kind: "quickfix",
+          edit: {
+            edits: [
+              {
+                resource: model.uri,
+                edit: {
+                  range: fix.range || error,
+                  text: fix.message
+                }
+              }
+            ]
+          },
+          isPreferred: true
+        })
       }
-    })
+    }
+
     return {
       actions: actions,
       dispose: () => {}
diff --git a/libs/remix-ui/editor/src/lib/providers/quickfixes.ts b/libs/remix-ui/editor/src/lib/providers/quickfixes.ts
index 49cc0d13d6..f2c698ec70 100644
--- a/libs/remix-ui/editor/src/lib/providers/quickfixes.ts
+++ b/libs/remix-ui/editor/src/lib/providers/quickfixes.ts
@@ -12,5 +12,9 @@ export default {
       startColumn: 1,
       endColumn: 1
     }
+  },
+  "SyntaxError: No visibility specified. Did you intend to add \"public\"": {
+    "title": "Add public visibility",
+    "message": "public",
   }
 }
\ No newline at end of file