From a758157563e8928146d678b188a8b3c8271d2f94 Mon Sep 17 00:00:00 2001
From: Vladimir <sleuths.slews0s@icloud.com>
Date: Mon, 3 Jun 2024 17:19:31 +0200
Subject: [PATCH] fix(browser): print correct transformed module graph (#5833)

---
 packages/ui/client/components/FileDetails.vue             | 2 +-
 .../ui/client/components/ModuleTransformResultView.vue    | 6 +++---
 packages/ui/client/components/views/ViewModuleGraph.vue   | 3 ++-
 packages/vitest/src/api/setup.ts                          | 8 ++++++--
 packages/vitest/src/api/types.ts                          | 2 +-
 5 files changed, 13 insertions(+), 8 deletions(-)

diff --git a/packages/ui/client/components/FileDetails.vue b/packages/ui/client/components/FileDetails.vue
index 41e004ea0816..7cfd5f016067 100644
--- a/packages/ui/client/components/FileDetails.vue
+++ b/packages/ui/client/components/FileDetails.vue
@@ -113,7 +113,7 @@ function relativeToRoot(path?: string) {
 
     <div flex flex-col flex-1 overflow="hidden">
       <div v-if="hasGraphBeenDisplayed" :flex-1="viewMode === 'graph' && ''">
-        <ViewModuleGraph v-show="viewMode === 'graph'" :graph="graph" data-testid="graph" />
+        <ViewModuleGraph v-show="viewMode === 'graph'" :graph="graph" data-testid="graph" :project-name="current.file.projectName || ''" />
       </div>
       <ViewEditor v-if="viewMode === 'editor'" :key="current.filepath" :file="current" data-testid="editor" @draft="onDraft" />
       <ViewConsoleOutput v-else-if="viewMode === 'console'" :file="current" data-testid="console" />
diff --git a/packages/ui/client/components/ModuleTransformResultView.vue b/packages/ui/client/components/ModuleTransformResultView.vue
index 14cbd37f6a65..a1484dc87378 100644
--- a/packages/ui/client/components/ModuleTransformResultView.vue
+++ b/packages/ui/client/components/ModuleTransformResultView.vue
@@ -1,10 +1,10 @@
 <script setup lang="ts">
-import { client } from '~/composables/client'
+import { client, browserState } from '~/composables/client'
 
-const props = defineProps<{ id: string }>()
+const props = defineProps<{ id: string; projectName: string }>()
 const emit = defineEmits<{ (e: 'close'): void }>()
 
-const result = asyncComputed(() => client.rpc.getTransformResult(props.id))
+const result = asyncComputed(() => client.rpc.getTransformResult(props.projectName, props.id, !!browserState))
 const ext = computed(() => props.id?.split(/\./g).pop() || 'js')
 
 const source = computed(() => result.value?.source?.trim() || '')
diff --git a/packages/ui/client/components/views/ViewModuleGraph.vue b/packages/ui/client/components/views/ViewModuleGraph.vue
index a778c5f25dda..f90afb627602 100644
--- a/packages/ui/client/components/views/ViewModuleGraph.vue
+++ b/packages/ui/client/components/views/ViewModuleGraph.vue
@@ -7,6 +7,7 @@ import type { ModuleGraph, ModuleGraphController, ModuleLink, ModuleNode, Module
 
 const props = defineProps<{
   graph: ModuleGraph
+  projectName: string
 }>()
 
 const { graph } = toRefs(props)
@@ -161,7 +162,7 @@ function bindOnClick(selection: Selection<SVGCircleElement, ModuleNode, SVGGElem
     <Modal v-model="modalShow" direction="right">
       <template v-if="selectedModule">
         <Suspense>
-          <ModuleTransformResultView :id="selectedModule" @close="modalShow = false" />
+          <ModuleTransformResultView :id="selectedModule" :project-name="projectName" @close="modalShow = false" />
         </Suspense>
       </template>
     </Modal>
diff --git a/packages/vitest/src/api/setup.ts b/packages/vitest/src/api/setup.ts
index eaf839ec7c40..e3ec7a90357e 100644
--- a/packages/vitest/src/api/setup.ts
+++ b/packages/vitest/src/api/setup.ts
@@ -67,8 +67,12 @@ export function setup(ctx: Vitest, _server?: ViteDevServer) {
         getConfig() {
           return ctx.config
         },
-        async getTransformResult(id) {
-          const result: TransformResultWithSource | null | undefined = await ctx.vitenode.transformRequest(id)
+        async getTransformResult(projectName: string, id, browser = false) {
+          const project = ctx.getProjectByName(projectName)
+          const result: TransformResultWithSource | null | undefined
+            = browser
+              ? await project.browser!.transformRequest(id)
+              : await project.vitenode.transformRequest(id)
           if (result) {
             try {
               result.source = result.source || (await fs.readFile(id, 'utf-8'))
diff --git a/packages/vitest/src/api/types.ts b/packages/vitest/src/api/types.ts
index 8770ea9ade66..d32985f9799d 100644
--- a/packages/vitest/src/api/types.ts
+++ b/packages/vitest/src/api/types.ts
@@ -16,7 +16,7 @@ export interface WebSocketHandlers {
   getPaths: () => string[]
   getConfig: () => ResolvedConfig
   getModuleGraph: (projectName: string, id: string, browser?: boolean) => Promise<ModuleGraphData>
-  getTransformResult: (id: string) => Promise<TransformResultWithSource | undefined>
+  getTransformResult: (projectName: string, id: string, browser?: boolean) => Promise<TransformResultWithSource | undefined>
   readTestFile: (id: string) => Promise<string | null>
   saveTestFile: (id: string, content: string) => Promise<void>
   rerun: (files: string[]) => Promise<void>