const crossSpawn = require('cross-spawn')
|
const slash = require('slash')
|
const { writeFileSync, rmSync, ensureFileSync } = require('fs-extra')
|
const { globSync } = require('glob')
|
const path = require('path')
|
const os = require('os')
|
const baseBuildFile = './node_modules/.cache/widgets.json'
|
let isSingleBuild = false
|
const isWin = process.platform === 'win32'
|
const argvPath = './script/.argv'
|
const widgetName = process.argv[process.argv.length - 1]
|
const widgetsPath = globSync(`./src/widgets/*/index.ts`)
|
const getWidgetNames = widgetsPath.map((file) => {
|
const parts = isWin
|
? path.resolve(file).split('\\')
|
: path.resolve(file).split('/')
|
return parts[parts.length - 2]
|
})
|
|
if (getWidgetNames.includes(widgetName)) {
|
isSingleBuild = true
|
writeFileSync(argvPath, widgetName)
|
}
|
|
buildWidgets()
|
|
function buildWidgets() {
|
const isWin = process.platform === 'win32'
|
const argv = process.argv || []
|
|
const widgetName = argv[argv.length - 1]
|
|
const widgetsPath = globSync(`./src/widgets/*/index.ts`)
|
const widgetNames = widgetsPath.map((file) => {
|
const parts = isWin
|
? path.resolve(file).split('\\')
|
: path.resolve(file).split('/')
|
return parts[parts.length - 2]
|
}) // 打包一个组件
|
|
if (widgetName && widgetNames.includes(widgetName)) {
|
isSingleBuild = true
|
writeFileSync(argvPath, widgetName)
|
runBuild()
|
} else {
|
// 打包多组件，按CPU默认并行度打包
|
const buildWidgets = divideArray(widgetNames)
|
const slashPath = slash(path.resolve(process.cwd(), baseBuildFile))
|
ensureFileSync(slashPath)
|
writeFileSync(slashPath, JSON.stringify(buildWidgets, null, 2))
|
buildSumCount = 0
|
buildCount = 0
|
for (let index = 0; index < Object.keys(buildWidgets).length; index++) {
|
const widgets = buildWidgets[index]
|
if (widgets.length) {
|
buildSumCount++
|
runBuild(index)
|
}
|
}
|
console.log(buildWidgets)
|
}
|
}
|
/**
|
* 获取等分的组件数据
|
* @param {*} widgets
|
* @param {*} cpus
|
* @returns
|
*/
|
function divideArray(widgets) {
|
// 当打包时，操作电脑可能会卡
|
const cpus = os.availableParallelism() > 1 ? os.availableParallelism() - 1 : 1
|
let result = {}
|
let dataPerKey = Math.floor(widgets.length / cpus)
|
let remainingData = widgets.length
|
for (let i = 0; i < cpus; i++) {
|
let currentDataCount = Math.min(dataPerKey, remainingData)
|
result[i] = widgets.splice(0, currentDataCount)
|
remainingData -= currentDataCount
|
}
|
if (widgets.length) {
|
widgets.forEach((widgetName, index) => {
|
result[index].push(widgetName)
|
})
|
}
|
return result
|
}
|
/**
|
* 运行编译
|
* @param {*} nodeIndex 设置打包组件起点
|
*/
|
function runBuild(nodeIndex) {
|
const cmdParams = ['run', 'build-lib']
|
const run = crossSpawn(
|
process.platform === 'win32' ? 'npm.cmd' : 'npm',
|
cmdParams,
|
{
|
stdio: 'inherit',
|
shell: true,
|
env: {
|
// 编译组件索引映射
|
...process.env,
|
NODE_INDEX: nodeIndex,
|
},
|
}
|
)
|
run.on('close', (code) => {
|
if (code == 0 && isSingleBuild) rmSync(argvPath)
|
})
|
}
|
