temp [ci skip]

Author: benfdking

vscode/extension/src/extension.ts

@@ -71,7 +71,7 @@ export async function activate(context: vscode.ExtensionContext) {
   // Register the webview
   const lineagePanel = new LineagePanel(
     context.extensionUri,
-    () => webServer?.url ?? ""
+    webServer,
   );
   context.subscriptions.push(
     vscode.window.registerWebviewViewProvider(

vscode/extension/src/webviews/lineagePanel.ts

@@ -12,22 +12,25 @@ import {
 } from "vscode";
 import { isProduction } from "../utilities/isDev";
 import { getWorkspaceFolders } from "../utilities/common/vscodeapi";
-import { queryServer } from "../web-server/server";
+import { queryServer, WebServer } from "../web-server/server";
 import { isErr } from "../utilities/functional/result";
 import { traceError } from "../utilities/common/log";
 
 export class LineagePanel implements WebviewViewProvider, Disposable {
   public static readonly viewType = "sqlmesh.lineage";
 
   private panel: WebviewView | undefined;
-  private getServerUrl: () => string;
+  private webServer: WebServer | undefined;
   private _extensionUri: Uri;
 
   private _disposables: Disposable[] = [];
 
-  public constructor(extensionUri: Uri, getServerUrl: () => string) {
+  public constructor(
+    extensionUri: Uri,
+    webServer: WebServer | undefined,
+  ) {
     this._extensionUri = extensionUri;
-    this.getServerUrl = getServerUrl;
+    this.webServer = webServer;
 
     if (this.panel) {
       const { externalUrl, externalAuthority } = this.externalUrlAndAutority();
@@ -106,9 +109,10 @@ export class LineagePanel implements WebviewViewProvider, Disposable {
 
   externalUrlAndAutority(): { externalUrl: string; externalAuthority: string } {
     const isProd = isProduction();
-    const externalUrl = !isProd
-      ? "http://localhost:5173/lineage"
-      : this.getServerUrl() + "/lineage";
+    if (!isProd) {
+      throw new Error("external url and authority is not supported in dev mode")
+    }
+    const externalUrl = "http://localhost:5173/lineage"
     const externalAuthority = new URL(externalUrl).origin;
     return { externalUrl, externalAuthority };
   }
@@ -181,8 +185,12 @@ export class LineagePanel implements WebviewViewProvider, Disposable {
             console.log('got query request in extension', message.payload, "with requestId", requestId)
 
             console.log("querying server")
+            const url = this.webServer?.get_url() 
+            if (!url) {
+              throw new Error("web server not found")
+            }
             const responseData = await queryServer(
-              this.getServerUrl(),
+              url,
               message.payload.queryKey,
               message.payload.body,
               message.payload.method