uni_modules.js

'use strict'
var __importDefault = (this && this.__importDefault) || function (mod) {
  return (mod && mod.__esModule) ? mod : { default: mod }
}
Object.defineProperty(exports, '__esModule', { value: true })
exports.parseDefines = exports.parseExports = exports.genUniModulesExports = void 0
const path_1 = __importDefault(require('path'))
const fs_extra_1 = __importDefault(require('fs-extra'))
const merge_1 = require('merge')
function genUniModulesExports () {
  const uniModulesDir = path_1.default.resolve(process.env.UNI_INPUT_DIR, 'uni_modules')
  if (!fs_extra_1.default.existsSync(uniModulesDir)) {
    return ''
  }
  const importCodes = []
  const assignCodes = []
  fs_extra_1.default.readdirSync(uniModulesDir).forEach((uniModuleDir) => {
    var _a, _b
    const pkgPath = path_1.default.resolve(uniModulesDir, uniModuleDir, 'package.json')
    if (!fs_extra_1.default.existsSync(pkgPath)) {
      return
    }
    const exports = (_b = (_a = JSON.parse(fs_extra_1.default.readFileSync(pkgPath, 'utf8'))) === null || _a === void 0 ? void 0 : _a.uni_modules) === null || _b === void 0 ? void 0 : _b.exports
    if (exports) {
      const [exportsImportCodes, exportsAssignCodes] = parseExports(process.env.UNI_PLATFORM === 'h5' ? 'web' : process.env.UNI_PLATFORM, `@/uni_modules/${uniModuleDir}`, exports)
      importCodes.push(...exportsImportCodes)
      assignCodes.push(...exportsAssignCodes)
    }
  })
  if (!importCodes.length) {
    return ''
  }
  return `${importCodes.join('\n')}
${assignCodes.join('\n')}`
}
exports.genUniModulesExports = genUniModulesExports
function parseExports (platform, source, exports = {}) {
  const rootDefines = {}
  Object.keys(exports).forEach((name) => {
    if (name.startsWith('uni')) {
      rootDefines[name] = exports[name]
    }
  })
  const platformDefines = exports[platform]
  // 该平台不支持
  if (platformDefines === false) {
    return [[], []]
  }
  return parseDefines(source, (0, merge_1.recursive)(true, rootDefines, platformDefines))
}
exports.parseExports = parseExports
function parseDefines (source, defines = {}) {
  const importCodes = []
  const assignCodes = []
  Object.keys(defines).forEach((name) => {
    const [defineImportCodes, defineAssignCodes] = parseDefine(source, name, defines[name])
    importCodes.push(...defineImportCodes)
    assignCodes.push(...defineAssignCodes)
  })
  return [importCodes, assignCodes]
}
exports.parseDefines = parseDefines
/**
 *  uni:'getBatteryInfo'
 * import getBatteryInfo from '..'
 *
 * uni:['getBatteryInfo']
 * import { getBatteryInfo } from '..'
 *
 * uni:['openLocation','chooseLocation']
 * import { openLocation, chooseLocation } from '..'
 *
 * uni:{
 *  onUserCaptureScreen: "onCaptureScreen"
 *  offUserCaptureScreen: "offCaptureScreen"
 * }
 *
 * uni.getBatteryInfo = getBatteryInfo
 * @param source
 * @param globalObject
 * @param define
 * @returns
 */
function parseDefine (source, globalObject, define) {
  const importCodes = []
  const assignCodes = []
  if (typeof define === 'string') {
    importCodes.push(`import ${define} from '${source}'`)
    assignCodes.push(`${globalObject}.${define} = ${define}`)
  } else if (Array.isArray(define)) {
    importCodes.push(`import { ${define.join(', ')} } from '${source}'`)
    define.forEach((d) => {
      assignCodes.push(`${globalObject}.${d} = ${d}`)
    })
  } else {
    const keys = Object.keys(define)
    const specifiers = []
    keys.forEach((d) => {
      if (d !== define[d]) {
        specifiers.push(`${define[d]} as ${d}`)
      } else {
        specifiers.push(d)
      }
      assignCodes.push(`${globalObject}.${d} = ${d}`)
    })
    importCodes.push(`import { ${specifiers.join(', ')} } from '${source}'`)
  }
  return [importCodes, assignCodes]
}