feat: generate module relationship for build (#149)

Author: sapphi-red

src/node/build.ts

@@ -1,4 +1,4 @@
-import type { Environment } from 'vite'
+import type { Rollup } from 'vite'
 import type { ModuleTransformInfo } from '../types'
 import type { InspectContext } from './context'
 import fs from 'node:fs/promises'
@@ -70,7 +70,8 @@ export function createBuildGenerator(ctx: InspectContext) {
         ),
       ])
     },
-    async generateForEnv(env: Environment) {
+    async generateForEnv(pluginCtx: Rollup.PluginContext) {
+      const env = pluginCtx.environment
       await Promise.all([...ctx._idToInstances.values()]
         .filter(v => v.environments.has(env.name))
         .map((v) => {
@@ -84,7 +85,7 @@ export function createBuildGenerator(ctx: InspectContext) {
           return await Promise.all([
             writeJSON(
               join(reportsDir, key, 'modules.json'),
-              env.getModulesList(),
+              env.getModulesList(pluginCtx),
             ),
             writeJSON(
               join(reportsDir, key, 'metric-plugins.json'),

src/node/context.ts

@@ -1,4 +1,4 @@
-import type { Environment, ResolvedConfig } from 'vite'
+import type { Environment, ResolvedConfig, Rollup } from 'vite'
 import type { Metadata, ModuleInfo, PluginMetricInfo, QueryEnv, ResolveIdInfo, ServerMetrics, TransformInfo } from '../types'
 import type { ViteInspectOptions } from './options'
 import { Buffer } from 'node:buffer'
@@ -158,16 +158,24 @@ export class InspectContextViteEnv {
     return id
   }
 
-  getModulesList() {
+  getModulesList(pluginCtx: Rollup.PluginContext) {
     const moduleGraph = this.env.mode === 'dev' ? this.env.moduleGraph : undefined
 
-    const getDeps = (id: string) => Array.from(moduleGraph?.getModuleById(id)?.importedModules || [])
-      .map(i => i.id || '')
-      .filter(Boolean)
-
-    const getImporters = (id: string) => Array.from(moduleGraph?.getModuleById(id)?.importers || [])
-      .map(i => i.id || '')
-      .filter(Boolean)
+    const getDeps = moduleGraph
+      ? (id: string) => Array.from(moduleGraph.getModuleById(id)?.importedModules || [])
+          .map(i => i.id || '')
+          .filter(Boolean)
+      : pluginCtx
+        ? (id: string) => pluginCtx.getModuleInfo(id)?.importedIds || []
+        : () => []
+
+    const getImporters = moduleGraph
+      ? (id: string) => Array.from(moduleGraph?.getModuleById(id)?.importers || [])
+          .map(i => i.id || '')
+          .filter(Boolean)
+      : pluginCtx
+        ? (id: string) => pluginCtx.getModuleInfo(id)?.importers || []
+        : () => []
 
     function isVirtual(pluginName: string, transformName: string) {
       return pluginName !== DUMMY_LOAD_PLUGIN_NAME && transformName !== 'vite:load-fallback' && transformName !== 'vite:build-load-fallback'

src/node/index.ts

@@ -220,7 +220,7 @@ export default function PluginInspect(options: ViteInspectOptions = {}): Plugin
             await buildGenerator.setupOutputDir()
           },
           async onEach(pluginCtx) {
-            await buildGenerator.generateForEnv(pluginCtx.environment)
+            await buildGenerator.generateForEnv(pluginCtx)
           },
           async onLast(pluginCtx) {
             const dir = buildGenerator.getOutputDir()