template.ts

import path from 'path'
import debug from 'debug'
import { EmittedFile, GetModuleInfo } from 'rollup'
import { ResolvedConfig } from 'vite'
import {
  addMiniProgramTemplateFile,
  removeExt,
  normalizeMiniProgramFilename,
  MiniProgramFilterOptions,
  findMiniProgramTemplateFiles,
  addMiniProgramTemplateFilter,
  clearMiniProgramTemplateFiles,
} from '@dcloudio/uni-cli-shared'
import { getFiltersCache } from '../plugins/renderjs'
import { UniMiniProgramPluginOptions } from '.'

const debugTemplate = debug('vite:uni:mp-template')

export function getFilterFiles(
  resolvedConfig: ResolvedConfig,
  getModuleInfo: GetModuleInfo
) {
  const filters: Record<string, MiniProgramFilterOptions> = Object.create(null)
  const filtersCache = getFiltersCache(resolvedConfig)
  if (!filtersCache.length) {
    return filters
  }
  const inputDir = process.env.UNI_INPUT_DIR
  function addFilter(id: string, filter: MiniProgramFilterOptions) {
    const templateFilename = removeExt(
      normalizeMiniProgramFilename(id, inputDir)
    )
    addMiniProgramTemplateFilter(templateFilename, filter)
    const filterFilename = removeExt(
      normalizeMiniProgramFilename(filter.id, inputDir)
    )
    if (templateFilename !== filterFilename) {
      // 外链
      filter.src = filterFilename
      filters[filterFilename] = filter
    }
  }
  filtersCache.forEach((filter) => {
    const moduleInfo = getModuleInfo(filter.id)
    if (!moduleInfo) {
      return
    }
    const { importers } = moduleInfo
    if (!importers.length) {
      return
    }
    importers.forEach((importer) => addFilter(importer, filter))
  })
  return filters
}

export function getTemplateFiles(
  template: UniMiniProgramPluginOptions['template']
) {
  const files = findMiniProgramTemplateFiles(template.filter?.generate)
  clearMiniProgramTemplateFiles()
  return files
}

export const emitFile: (emittedFile: EmittedFile) => string = (emittedFile) => {
  if (emittedFile.type === 'asset') {
    const filename = emittedFile.fileName!
    addMiniProgramTemplateFile(
      removeExt(
        normalizeMiniProgramFilename(
          path.relative(process.env.UNI_INPUT_DIR, filename)
        )
      ),
      emittedFile.source!.toString()
    )
    debugTemplate(filename)
    return filename
  }
  return ''
}