faet(vscode): support multiple workspaces

[ci skip]

Author: benfdking

vscode/extension/tests/lineage.spec.ts

@@ -3,6 +3,7 @@ import path from 'path'
 import fs from 'fs-extra'
 import os from 'os'
 import { startVSCode, SUSHI_SOURCE_PATH } from './utils'
+import { writeFileSync } from 'fs'
 
 /**
  * Helper function to launch VS Code and test lineage with given project path config
@@ -141,3 +142,42 @@ test('Lineage panel renders correctly - absolute path project outside of workspa
     await fs.remove(tempFolder)
   }
 })
+
+test('Lineage panel renders correctly - multiworkspace setup', async () => {
+  const workspaceDir = await fs.mkdtemp(
+    path.join(os.tmpdir(), 'vscode-test-workspace-'),
+  )
+  const projectDir1 = path.join(workspaceDir, 'projects', 'sushi1')
+  const projectDir2 = path.join(workspaceDir, 'projects', 'sushi2')
+  await fs.copy(SUSHI_SOURCE_PATH, projectDir1)
+  await fs.ensureDir(projectDir2)
+
+  // Add a .code-workspace file with multiple projects
+  const workspaceFilePath = path.join(
+    workspaceDir,
+    'multi-workspace.code-workspace',
+  )
+  writeFileSync(
+    workspaceFilePath,
+    JSON.stringify({
+      folders: [
+        {
+          name: 'sushi1',
+          path: 'projects/sushi1',
+        },
+        {
+          name: 'sushi2',
+          path: 'projects/sushi2',
+        },
+      ],
+    }),
+  )
+
+  try {
+    const { window, close } = await startVSCode(workspaceFilePath)
+    await testLineageWithProjectPath(window)
+    await close()
+  } finally {
+    await fs.remove(workspaceDir)
+  }
+})