perf(vite): 跳过 Generic Web 生产候选状态 (#1140)

* perf(vite): 跳过 Generic Web 生产候选状态

Refs #1107

* chore(ci): keep vite runtime within line limit

Refs #1107

Author: sonofmagic

.changeset/quick-web-candidates-skip.md

@@ -0,0 +1,5 @@
+---
+"weapp-tailwindcss": patch
+---
+
+优化 Generic Web 非 watch 生产构建：跳过不需要的小程序 source-candidates 状态维护，减少重复扫描与构建生命周期开销。

packages/weapp-tailwindcss/src/bundlers/vite/shared/create-framework-plugins-runtime.ts

@@ -63,7 +63,7 @@ import { collectConfiguredCssEntries, inferPlatformFromOutDir, isInternalUserDef
 import { createFrameworkSourceCandidatesPlugin } from './framework-source-candidates-plugin'
 import { createFrameworkSourceScanSession, syncFrameworkSourceCandidatesForHotUpdate } from './framework-source-scan-session'
 import { createFrameworkTailwindRootCss } from './framework-tailwind-root-css'
-import { createGenericWebProductionBundleHooks, createGenericWebProductionSourceCandidatesApply } from './generic-web-production-fast-path'
+import { createGenericWebProductionBundleHooks, createGenericWebProductionSourceCandidatesApply, shouldSkipGenericWebProductionSourceCandidates } from './generic-web-production-fast-path'
 
 const debug = createDebug()
 const weappTailwindcssPackageDir = resolvePackageDir('weapp-tailwindcss'); const weappTailwindcssDirPosix = slash(weappTailwindcssPackageDir); const generatorPlaceholderCssFile = path.join(weappTailwindcssPackageDir, 'generator-placeholder.css'); const ENV_PLATFORM_KEYS = ['UNI_PLATFORM', 'UNI_UTS_PLATFORM', 'TARO_ENV', 'MPX_CURRENT_TARGET_MODE', 'MPX_CLI_MODE']
@@ -231,6 +231,7 @@ function createViteFrameworkPlugins(options = {}, frameworkBranch): any {
   const getSourceCandidateSourcesForEntries = (entries, options2) => sourceCandidateCollector.sourcesForEntries(entries, options2)
   const isWatchBuild = () => resolvedConfig?.command === 'build' && resolvedConfig.build.watch != null
   const isWatchLikeBuild = () => isWatchBuild() || resolvedConfig?.command === 'serve' || process.env['WEAPP_TW_WATCH_REGRESSION'] === '1' || process.env['WEAPP_TW_HMR_TIMING'] === '1'
+  const shouldSkipSourceCandidateState = () => shouldSkipGenericWebProductionSourceCandidates({ command: resolvedConfig?.command, frameworkName: frameworkBranch.frameworkName, isWebGeneratorTarget: resolveCurrentGeneratorBranch().isWeb, requiresSourceCandidateState: isCssSourceTraceEnabled(opts), watch: resolvedConfig?.build?.watch })
   const isCurrentWebLikeStylePlatform = () => { const platform = resolveViteStylePlatform(); return platform ? isWebOrNativeAppPlatform(platform) : resolveCurrentGeneratorBranch().isWeb }
   const normalizeGeneratedCssCacheFile = file => normalizeVitePersistentCacheKey(cleanUrl(file))
   const hmrCandidateState = createViteHmrCandidateState({
@@ -451,6 +452,7 @@ ${tracedCss}`
     resolveViteStylePlatform, runtimeState, shouldOwnTailwindGeneration, sourceCandidateCollector, sourceScanSession,
     tailwindRootCssModuleIds, transformEarlyMiniProgramCss, viteProcessedCssSourceFiles: processedCssRegistry.sourceFiles,
     collectSourceCandidates: capability.sourceCandidates,
+    shouldSkipSourceCandidateState,
   }, createGenericWebProductionSourceCandidatesApply({ frameworkName: frameworkBranch.frameworkName, getIsWebGeneratorTarget: () => resolveCurrentGeneratorBranch().isWeb, requiresSourceCandidateState: isCssSourceTraceEnabled(opts) }))
   /* eslint-enable antfu/consistent-list-newline */
   const postPlugin = createFrameworkPostPlugin({

packages/weapp-tailwindcss/src/bundlers/vite/shared/framework-source-candidates-plugin.ts

@@ -12,6 +12,7 @@ import { isSourceCandidateRequest } from '../source-candidates'
 import { cleanUrl, isCSSRequest } from '../utils'
 
 export function createFrameworkSourceCandidatesPlugin(options: any, apply?: Plugin['apply']): Plugin {
+  const shouldSkipSourceCandidateState = () => options.shouldSkipSourceCandidateState?.() === true
   const hasDifferentHotModules = (left: ModuleNode[], right: ModuleNode[]) => left.length !== right.length
     || left.some((mod, index) => mod !== right[index])
   const hasTemplateHotSourceModule = (modules: Array<{ id?: string | null, url?: string | null }>) => modules.some((mod) => {
@@ -51,6 +52,7 @@ export function createFrameworkSourceCandidatesPlugin(options: any, apply?: Plug
     async load(id) {
       if (
         !options.shouldOwnTailwindGeneration
+        || shouldSkipSourceCandidateState()
         || options.isWebOrNativeAppPlatform(options.resolveViteStylePlatform())
         || !isCSSRequest(id)
         || !shouldCollectTransformedSourceCandidates(id)
@@ -73,6 +75,9 @@ export function createFrameworkSourceCandidatesPlugin(options: any, apply?: Plug
     transform: {
       order: 'pre',
       async handler(code, id) {
+        if (shouldSkipSourceCandidateState()) {
+          return
+        }
         if (options.hasUserCssLayerBlocks(code)) {
           options.rememberOriginalCssLayerSource(id, code)
         }
@@ -87,7 +92,7 @@ export function createFrameworkSourceCandidatesPlugin(options: any, apply?: Plug
             options.rememberTailwindRootCssModule(id)
           }
         }
-        if (!options.shouldOwnTailwindGeneration || options.collectSourceCandidates === false || !isSourceCandidateRequest(id) || !shouldCollectTransformedSourceCandidates(id)) {
+        if (!options.shouldOwnTailwindGeneration || shouldSkipSourceCandidateState() || options.collectSourceCandidates === false || !isSourceCandidateRequest(id) || !shouldCollectTransformedSourceCandidates(id)) {
           return shouldReturnTransformedCode ? { code: transformedCode, map: null } : undefined
         }
         return options.hmrTimingRecorder.measure('sourceCandidates.transform', async () => {
@@ -105,6 +110,9 @@ export function createFrameworkSourceCandidatesPlugin(options: any, apply?: Plug
       },
     },
     async watchChange(id, change) {
+      if (shouldSkipSourceCandidateState()) {
+        return
+      }
       recordCompilationDependencyChanges(options.runtimeState, createCompilationDependencyChanges([path.resolve(cleanUrl(id))]))
       await options.hmrTimingRecorder.measure('sourceCandidates.watchChange', async () => {
         if (options.shouldOwnTailwindGeneration && options.collectSourceCandidates !== false && isSourceCandidateRequest(id)) {
@@ -147,6 +155,9 @@ export function createFrameworkSourceCandidatesPlugin(options: any, apply?: Plug
     handleHotUpdate: {
       order: 'post',
       async handler(ctx) {
+        if (shouldSkipSourceCandidateState()) {
+          return
+        }
         recordCompilationDependencyChanges(options.runtimeState, createCompilationDependencyChanges([path.resolve(cleanUrl(ctx.file))]))
         return options.hmrTimingRecorder.measure('sourceCandidates.handleHotUpdate', async () => {
           const isSourceCandidateHotUpdate = options.shouldOwnTailwindGeneration && options.collectSourceCandidates !== false && isSourceCandidateRequest(ctx.file)
@@ -307,8 +318,16 @@ export function createFrameworkSourceCandidatesPlugin(options: any, apply?: Plug
       options.hmrCssModuleVersions?.clear()
     },
     async buildStart() {
+      if (shouldSkipSourceCandidateState()) {
+        return
+      }
       await options.hmrTimingRecorder.measure('sourceCandidates.buildStart', options.prepareTailwindGeneration, { emit: false })
     },
-    generateBundle: options.preGenerateBundleHook,
+    async generateBundle(...args: any[]) {
+      if (shouldSkipSourceCandidateState()) {
+        return
+      }
+      return options.preGenerateBundleHook?.apply(this, args)
+    },
   }
 }

packages/weapp-tailwindcss/test/bundlers/vite-dispatcher.unit.test.ts

@@ -15,6 +15,9 @@ describe('vite 单入口 dispatcher', () => {
     setCurrentContext(context)
     const { WeappTailwindcss } = await import('@/bundlers/vite')
     const plugins = WeappTailwindcss()
+    const sourceCandidates = findPlugin(plugins, ':source-candidates')!
+    expect(sourceCandidates.apply).toBeTypeOf('function')
+    expect(typeof sourceCandidates.apply === 'function' && sourceCandidates.apply({ build: {} }, { command: 'build', mode: 'production' })).toBe(true)
     const post = findPlugin(plugins, ':post')!
     await post.configResolved?.call(post, {
       command: 'build',
@@ -24,6 +27,7 @@ describe('vite 单入口 dispatcher', () => {
     } as ResolvedConfig)
 
     expect(context.generator).toMatchObject({ target: 'web' })
+    expect(typeof sourceCandidates.apply === 'function' && sourceCandidates.apply({ build: {} }, { command: 'build', mode: 'production' })).toBe(false)
     const js = findPlugin(plugins, ':js:serve')
     const jsResult = await js?.transform?.call(js, 'const cls = "text-red-500"', '/project/main.ts')
     expect(jsResult).toBeUndefined()

packages/weapp-tailwindcss/test/bundlers/vite-source-candidates-hmr.unit.test.ts

@@ -13,6 +13,26 @@ function getTransformHandler(plugin: ReturnType<typeof createFrameworkSourceCand
 }
 
 describe('Vite source candidate HMR transactions', () => {
+  it('skips source candidate lifecycle work after Generic Web production profile resolves', async () => {
+    const prepareTailwindGeneration = vi.fn()
+    const preGenerateBundleHook = vi.fn()
+    const plugin = createFrameworkSourceCandidatesPlugin({
+      hmrTimingRecorder: { measure: (_name: string, task: () => unknown) => task() },
+      preGenerateBundleHook,
+      prepareTailwindGeneration,
+      shouldSkipSourceCandidateState: () => true,
+    })
+
+    await plugin.buildStart?.call(plugin)
+    await plugin.generateBundle?.call(plugin, {}, {})
+    await expect(getTransformHandler(plugin)?.call(plugin, 'const cls = "text-red-500"', '/project/src/page.ts')).resolves.toBeUndefined()
+    await plugin.watchChange?.('/project/src/page.ts', { event: 'update' } as any)
+    await getHandleHotUpdateHandler(plugin)?.call(plugin, { file: '/project/src/page.ts' } as any)
+
+    expect(prepareTailwindGeneration).not.toHaveBeenCalled()
+    expect(preGenerateBundleHook).not.toHaveBeenCalled()
+  })
+
   it('keeps weapp-vite sidecar requests out of source candidate memory', async () => {
     const rememberKnownSfcSource = vi.fn()
     const rememberTailwindRootCssModule = vi.fn()