fix(vscode): improve handling of uris (#4440)

Author: benfdking

pnpm-lock.yaml

@@ -31,7 +31,7 @@ export class LineagePanel implements WebviewViewProvider, Disposable {
         key: 'vscode_send',
         payload: {
           key: 'savedFile',
-          payload: { fileUri: document.uri.fsPath },
+          payload: { fileUri: document.uri.toString() },
         },
       })
     })
@@ -42,7 +42,7 @@ export class LineagePanel implements WebviewViewProvider, Disposable {
           key: 'vscode_send',
           payload: {
             key: 'changeFocusOnFile',
-            payload: { path: editor.document.uri.fsPath },
+            payload: { path: editor.document.uri.toString() },
           },
         })
       }

vscode/extension/src/webviews/lineagePanel.ts

@@ -31,7 +31,8 @@
     "react-dom": "^18.2.0",
     "react-router": "^7.0.0",
     "reactflow": "^11.8.3",
-    "tailwindcss": "^4.1.3"
+    "tailwindcss": "^4.1.3",
+    "vscode-uri": "^3.1.0"
   },
   "devDependencies": {
     "@testing-library/dom": "^10.4.0",

vscode/react/package.json

@@ -13,6 +13,7 @@ import React, { useState } from 'react'
 import { ModelSQLMeshModel } from '@/domain/sqlmesh-model'
 import { useEventBus } from '@/hooks/eventBus'
 import type { VSCodeEvent } from '@bus/callbacks'
+import { URI } from 'vscode-uri'
 
 export function LineagePage() {
   const { emit } = useEventBus()
@@ -97,10 +98,7 @@ function Lineage() {
   )
   const selectedModel = selectedModelSet
   on('changeFocusedFile', fileUri => {
-    console.log('on changeFocusedFile', fileUri)
-    const full_path = fileUri.fileUri.startsWith('file://')
-      ? fileUri.fileUri.substring(7)
-      : fileUri.fileUri
+    const full_path = URI.parse(fileUri.fileUri).fsPath
     const model = Object.values(modelsRecord).find(
       (m: Model) => m.full_path === full_path,
     )
@@ -135,7 +133,7 @@ export function LineageComponentFromWeb({
     if (!model) {
       throw new Error('Model not found')
     }
-    vscode('openFile', { uri: 'file://' + model.full_path })
+    vscode('openFile', { uri: URI.file(model.full_path).toString() })
   }
 
   function handleError(error: any): void {