temp [ci skip]

Author: benfdking

examples/sushi/models/active_customers.sql

@@ -16,4 +16,4 @@ MODEL (
 
 SELECT customer_id, zip
 FROM sushi.customers
-WHERE status = 'active'
+WHERE status = 'active'

vscode/extension/src/extension.ts

@@ -5,11 +5,7 @@ import {
   onDidChangeConfiguration,
   registerCommand,
 } from './utilities/common/vscodeapi'
-import {
-  registerLogger,
-  traceInfo,
-  traceVerbose,
-} from './utilities/common/log'
+import { registerLogger, traceInfo, traceVerbose } from './utilities/common/log'
 import { onDidChangePythonInterpreter } from './utilities/common/python'
 import { LSPClient } from './lsp/lsp'
 import { AuthenticationProviderTobikoCloud } from './auth/auth'
@@ -18,12 +14,11 @@ import { signIn } from './commands/signin'
 import { signInSpecifyFlow } from './commands/signinSpecifyFlow'
 import { isErr } from './utilities/functional/result'
 import {
+  handleNotSginedInError,
   handleSqlmeshLspNotFoundError,
   handleSqlmeshLspDependenciesMissingError,
-  handleNotSginedInError,
 } from './utilities/errors'
 import { selector, completionProvider } from './completion/completion'
-import { WebServer } from './web-server/server'
 import { LineagePanel } from './webviews/lineagePanel'
 
 let lspClient: LSPClient | undefined
@@ -38,12 +33,6 @@ export async function activate(context: vscode.ExtensionContext) {
   )
   traceInfo('Activating SQLMesh extension')
 
-  const webServerOutputChannel = createOutputChannel('sqlmesh-webserver')
-  context.subscriptions.push(
-    webServerOutputChannel,
-    registerLogger(webServerOutputChannel),
-  )
-
   traceInfo('Registering authentication provider')
   const authProvider = new AuthenticationProviderTobikoCloud()
   context.subscriptions.push(
@@ -95,6 +84,13 @@ export async function activate(context: vscode.ExtensionContext) {
     context.subscriptions.push(lspClient)
   }
 
+  context.subscriptions.push(
+    vscode.languages.registerCompletionItemProvider(
+      selector,
+      completionProvider(lspClient),
+    ),
+  )
+
   // Register the webview
   const lineagePanel = new LineagePanel(context.extensionUri, lspClient)
   context.subscriptions.push(
@@ -109,8 +105,24 @@ export async function activate(context: vscode.ExtensionContext) {
       traceVerbose('Restarting LSP client')
       const restartResult = await lspClient.restart()
       if (isErr(restartResult)) {
-        handleNotSginedInError(authProvider)
+        switch (restartResult.error.type) {
+          case 'not_signed_in':
+            await handleNotSginedInError(authProvider)
+            return
+          case 'sqlmesh_lsp_not_found':
+            await handleSqlmeshLspNotFoundError()
+            return
+          case 'sqlmesh_lsp_dependencies_missing':
+            await handleSqlmeshLspDependenciesMissingError(restartResult.error)
+            return
+          case 'generic':
+            await vscode.window.showErrorMessage(
+              `Failed to restart LSP: ${restartResult.error.message}`,
+            )
+            return
+        }
       }
+      context.subscriptions.push(lspClient)
     }
   }