feat(go): add GoLangPlaygroundSketchProvider

Introduce GoLangPlaygroundSketchProvider to support Go playground sketches. Includes modifications to sketch classes and extension registration.

Author: phodal

core/src/main/kotlin/cc/unitmesh/devti/gui/toolbar/NewChatAction.kt

@@ -3,7 +3,6 @@ package cc.unitmesh.devti.gui.toolbar
 import cc.unitmesh.devti.gui.AutoDevToolWindowFactory
 import cc.unitmesh.devti.gui.chat.ChatCodingPanel
 import cc.unitmesh.devti.settings.LanguageChangedCallback.componentStateChanged
-import cc.unitmesh.devti.sketch.SketchToolWindow
 import com.intellij.openapi.actionSystem.*
 import com.intellij.openapi.actionSystem.ex.CustomComponentAction
 import com.intellij.openapi.diagnostic.logger

core/src/main/kotlin/cc/unitmesh/devti/sketch/ui/preview/FileEditorPreviewSketch.kt

@@ -24,9 +24,8 @@ abstract class FileEditorPreviewSketch(
     val project: Project,
     val virtualFile: VirtualFile,
     val withPreviewEditorId: String
-) :
-    ExtensionLangSketch {
-    val editor: FileEditor = getEditorProvider().createEditor(project, virtualFile)
+) : ExtensionLangSketch {
+    open val editor: FileEditor = getEditorProvider().createEditor(project, virtualFile)
 
     open val mainPanel = editor.component
 
@@ -37,7 +36,7 @@ abstract class FileEditorPreviewSketch(
 
     override fun getComponent(): JComponent = mainPanel
 
-    override fun getViewText(): String = virtualFile.readText()
+    override fun getViewText(): String = editor.file.readText()
 
     override fun updateViewText(text: String, complete: Boolean) {}
 

goland/src/main/kotlin/cc/unitmesh/go/provider/GoLangPlaygroundSketchProvider.kt

@@ -0,0 +1,79 @@
+package cc.unitmesh.go.provider
+
+import cc.unitmesh.devti.sketch.ui.ExtensionLangSketch
+import cc.unitmesh.devti.sketch.ui.LanguageSketchProvider
+import cc.unitmesh.devti.sketch.ui.code.CodeHighlightSketch
+import cc.unitmesh.devti.sketch.ui.preview.FileEditorPreviewSketch
+import com.goide.GoLanguage
+import com.goide.playground.ui.GoPlaygroundFileEditorWithPreview
+import com.intellij.openapi.Disposable
+import com.intellij.openapi.editor.EditorFactory
+import com.intellij.openapi.editor.EditorKind
+import com.intellij.openapi.fileEditor.FileEditor
+import com.intellij.openapi.fileEditor.TextEditor
+import com.intellij.openapi.fileEditor.impl.text.TextEditorProvider
+import com.intellij.openapi.project.Project
+import com.intellij.openapi.util.Disposer
+import com.intellij.openapi.vfs.VirtualFile
+import com.intellij.testFramework.LightVirtualFile
+import javax.swing.JComponent
+
+class GoLangPlaygroundSketchProvider : LanguageSketchProvider {
+    override fun isSupported(lang: String): Boolean = lang == "go"
+
+    override fun create(project: Project, content: String): ExtensionLangSketch {
+        if (content.contains("package main")) {
+            val file = LightVirtualFile("main.go", content)
+            return GoLangPlaygroundSketch(project, content, file)
+        }
+
+        return object : CodeHighlightSketch(project, content, GoLanguage.INSTANCE), ExtensionLangSketch {
+            override fun getExtensionName(): String = "GoHighlight"
+        }
+    }
+}
+
+class GoLangPlaygroundSketch(val myProject: Project, val content: String, val file: LightVirtualFile) :
+    FileEditorPreviewSketch(myProject, file, "GoPlaygroundFileEditorWithPreview") {
+    override fun getExtensionName(): String = "GoMain"
+
+    val editorWithPreview = createEditorWithPreview(myProject, file) as? GoPlaygroundFileEditorWithPreview
+
+    override val mainPanel: JComponent
+        get() {
+            return editorWithPreview?.component ?: editor.component
+        }
+
+    override fun getComponent(): JComponent {
+        return editorWithPreview?.component ?: editor.component
+    }
+
+    override fun getViewText(): String {
+        return editorWithPreview?.editor?.document?.text ?: editor.file.readText()
+    }
+
+    companion object {
+        fun createEditorWithPreview(project: Project, file: VirtualFile): FileEditor {
+            val textEditorProvider = TextEditorProvider.getInstance()
+            val editorFactory = EditorFactory.getInstance()
+            val editor = textEditorProvider.createEditor(project, file)
+
+            return if (editor is TextEditor) {
+                val viewer = editorFactory.createViewer(
+                    editorFactory.createDocument("" as CharSequence),
+                    project,
+                    EditorKind.PREVIEW
+                )
+                val previewEditor = textEditorProvider.getTextEditor(viewer)
+
+                Disposer.register(editor as Disposable, Disposable {
+                    EditorFactory.getInstance().releaseEditor(viewer)
+                })
+
+                GoPlaygroundFileEditorWithPreview(editor, previewEditor)
+            } else {
+                editor
+            }
+        }
+    }
+}

goland/src/main/resources/cc.unitmesh.go.xml

@@ -17,5 +17,7 @@
         <chatContextProvider implementation="cc.unitmesh.go.provider.GoVersionChatContextProvider"/>
 
         <livingDocumentation language="go" implementationClass="cc.unitmesh.go.provider.GoLivingDocumentationProvider"/>
+
+        <langSketchProvider implementation="cc.unitmesh.go.provider.GoLangPlaygroundSketchProvider"/>
     </extensions>
 </idea-plugin>