build.ts

import fs from 'fs'
import path from 'path'
import debug from 'debug'
import { UserConfig } from 'vite'

import {
  emptyDir,
  EXTNAME_JS_RE,
  isCSSRequest,
  normalizePath,
  hasJsonFile,
  removeExt,
  resolveMainPathOnce,
} from '@dcloudio/uni-cli-shared'
import { GetManualChunk, GetModuleInfo } from 'rollup'
import {
  isUniComponentUrl,
  isUniPageUrl,
  parseVirtualComponentPath,
  parseVirtualPagePath,
} from '../plugins/entry'

const debugChunk = debug('vite:uni:chunk')

export function buildOptions(): UserConfig['build'] {
  const inputDir = process.env.UNI_INPUT_DIR
  const outputDir = process.env.UNI_OUTPUT_DIR
  // 开始编译时，清空输出目录
  if (fs.existsSync(outputDir)) {
    emptyDir(outputDir)
  }
  return {
    // sourcemap: 'inline', // TODO
    target: ['chrome53'],
    emptyOutDir: false, // 不清空输出目录，否则会影响自定义的一些文件输出，比如wxml
    assetsInlineLimit: 40 * 1024, // 40kb
    lib: {
      entry: resolveMainPathOnce(inputDir),
      formats: ['cjs'],
    },
    rollupOptions: {
      output: {
        entryFileNames: 'app.js',
        manualChunks: createMoveToVendorChunkFn(),
        chunkFileNames(chunk) {
          if (chunk.isDynamicEntry && chunk.facadeModuleId) {
            let id = chunk.facadeModuleId
            if (isUniPageUrl(id)) {
              id = path.resolve(
                process.env.UNI_INPUT_DIR,
                parseVirtualPagePath(id)
              )
            } else if (isUniComponentUrl(id)) {
              id = path.resolve(
                process.env.UNI_INPUT_DIR,
                parseVirtualComponentPath(id)
              )
            }
            const filepath = path.relative(inputDir, id)
            return normalizePath(
              filepath.replace(path.extname(filepath), '.js')
            )
          }
          return '[name].js'
        },
        assetFileNames: '[name][extname]',
      },
    },
  }
}

function isVueJs(id: string) {
  return (
    id.includes('plugin-vue:export-helper') ||
    (id.includes('/@vue/') && id.endsWith('.js'))
  )
}

function isDCloudJs(id: string) {
  return id.includes('/@dcloudio/') && id.endsWith('.js')
}

const chunkFileNameBlackList = ['main', 'pages.json', 'manifest.json']

function createMoveToVendorChunkFn(): GetManualChunk {
  const cache = new Map<string, boolean>()
  const inputDir = normalizePath(process.env.UNI_INPUT_DIR)
  return (id, { getModuleInfo }) => {
    id = normalizePath(id)
    if (
      isVueJs(id) ||
      isDCloudJs(id) ||
      (id.includes('node_modules') &&
        !isCSSRequest(id) &&
        staticImportedByEntry(id, getModuleInfo, cache))
    ) {
      debugChunk('common/vendor', id)
      return 'common/vendor'
    }
    const filename = id.split('?')[0]
    // 处理项目内的js,ts文件
    if (EXTNAME_JS_RE.test(filename) && filename.startsWith(inputDir)) {
      const chunkFileName = removeExt(
        normalizePath(path.relative(inputDir, filename))
      )
      if (
        !chunkFileNameBlackList.includes(chunkFileName) &&
        !hasJsonFile(chunkFileName) // 无同名的page,component
      ) {
        debugChunk(chunkFileName, id)
        return chunkFileName
      }
    }
  }
}

function staticImportedByEntry(
  id: string,
  getModuleInfo: GetModuleInfo,
  cache: Map<string, boolean>,
  importStack: string[] = []
): boolean {
  if (cache.has(id)) {
    return cache.get(id) as boolean
  }
  if (importStack.includes(id)) {
    // circular deps!
    cache.set(id, false)
    return false
  }
  const mod = getModuleInfo(id)
  if (!mod) {
    cache.set(id, false)
    return false
  }

  if (mod.isEntry) {
    cache.set(id, true)
    return true
  }
  const someImporterIs = mod.importers.some((importer) =>
    staticImportedByEntry(
      importer,
      getModuleInfo,
      cache,
      importStack.concat(id)
    )
  )
  cache.set(id, someImporterIs)
  return someImporterIs
}