feat(vscode): add ability to render model

Author: benfdking

vscode/extension/package.json

@@ -87,8 +87,23 @@
         "command": "sqlmesh.signout",
         "title": "Sign out from Tobiko Cloud",
         "description": "SQLMesh"
+      },
+      {
+        "command": "sqlmesh.openTestQuery",
+        "title": "Open Test Query",
+        "description": "Opens a new SQL editor with a test query",
+        "icon": "$(database)"
       }
-    ]
+    ],
+    "menus": {
+      "editor/title": [
+        {
+          "command": "sqlmesh.openTestQuery",
+          "when": "resourceExtname == .sql",
+          "group": "navigation"
+        }
+      ]
+    }
   },
   "scripts": {
     "ci": "pnpm run lint && pnpm run compile",

vscode/extension/src/commands/openTestQuery.ts

@@ -0,0 +1,33 @@
+import * as vscode from 'vscode'
+
+export function openTestQuery() {
+  return async () => {
+    // Get the current active editor
+    const activeEditor = vscode.window.activeTextEditor
+    
+    // Create a new untitled document with SQL language
+    const document = await vscode.workspace.openTextDocument({
+      language: 'sql',
+      content: 'SELECT * FROM test'
+    })
+
+    // Determine the view column for side-by-side display
+    let viewColumn: vscode.ViewColumn
+    if (activeEditor) {
+      // Open beside the current editor
+      viewColumn = activeEditor.viewColumn 
+        ? activeEditor.viewColumn + 1 
+        : vscode.ViewColumn.Two
+    } else {
+      // If no active editor, open in column two
+      viewColumn = vscode.ViewColumn.Two
+    }
+
+    // Open the document in the editor as a preview (preview: true is default)
+    await vscode.window.showTextDocument(document, {
+      viewColumn: viewColumn,
+      preview: true,
+      preserveFocus: false
+    })
+  }
+}

vscode/extension/src/extension.ts

@@ -12,6 +12,7 @@ import { AuthenticationProviderTobikoCloud } from './auth/auth'
 import { signOut } from './commands/signout'
 import { signIn } from './commands/signin'
 import { signInSpecifyFlow } from './commands/signinSpecifyFlow'
+import { openTestQuery } from './commands/openTestQuery'
 import { isErr } from '@bus/result'
 import {
   handleNotSginedInError,
@@ -61,6 +62,9 @@ export async function activate(context: vscode.ExtensionContext) {
   context.subscriptions.push(
     vscode.commands.registerCommand('sqlmesh.format', format(authProvider)),
   )
+  context.subscriptions.push(
+    vscode.commands.registerCommand('sqlmesh.openTestQuery', openTestQuery()),
+  )
 
   lspClient = new LSPClient()