64 lines
2.0 KiB
JavaScript
64 lines
2.0 KiB
JavaScript
const { Arch } = require('electron-builder')
|
|
const fs = require('fs')
|
|
const path = require('path')
|
|
|
|
exports.default = async function (context) {
|
|
const platform = context.packager.platform.name
|
|
const arch = context.arch
|
|
|
|
if (platform === 'mac') {
|
|
const node_modules_path = path.join(
|
|
context.appOutDir,
|
|
'Cherry Studio.app',
|
|
'Contents',
|
|
'Resources',
|
|
'app.asar.unpacked',
|
|
'node_modules'
|
|
)
|
|
|
|
keepPackageNodeFiles(node_modules_path, '@libsql', arch === Arch.arm64 ? ['darwin-arm64'] : ['darwin-x64'])
|
|
}
|
|
|
|
if (platform === 'linux') {
|
|
const node_modules_path = path.join(context.appOutDir, 'resources', 'app.asar.unpacked', 'node_modules')
|
|
const _arch = arch === Arch.arm64 ? ['linux-arm64-gnu', 'linux-arm64-musl'] : ['linux-x64-gnu', 'linux-x64-musl']
|
|
keepPackageNodeFiles(node_modules_path, '@libsql', _arch)
|
|
}
|
|
|
|
if (platform === 'windows') {
|
|
const node_modules_path = path.join(context.appOutDir, 'resources', 'app.asar.unpacked', 'node_modules')
|
|
if (arch === Arch.arm64) {
|
|
keepPackageNodeFiles(node_modules_path, '@strongtz', ['win32-arm64-msvc'])
|
|
keepPackageNodeFiles(node_modules_path, '@libsql', ['win32-arm64-msvc'])
|
|
}
|
|
if (arch === Arch.x64) {
|
|
keepPackageNodeFiles(node_modules_path, '@strongtz', ['win32-x64-msvc'])
|
|
keepPackageNodeFiles(node_modules_path, '@libsql', ['win32-x64-msvc'])
|
|
}
|
|
}
|
|
}
|
|
|
|
/**
|
|
* 使用指定架构的 node_modules 文件
|
|
* @param {*} nodeModulesPath
|
|
* @param {*} packageName
|
|
* @param {*} arch
|
|
* @returns
|
|
*/
|
|
function keepPackageNodeFiles(nodeModulesPath, packageName, arch) {
|
|
const modulePath = path.join(nodeModulesPath, packageName)
|
|
|
|
if (!fs.existsSync(modulePath)) {
|
|
console.log(`[After Pack] Directory does not exist: ${modulePath}`)
|
|
return
|
|
}
|
|
|
|
const dirs = fs.readdirSync(modulePath)
|
|
dirs
|
|
.filter((dir) => !arch.includes(dir))
|
|
.forEach((dir) => {
|
|
fs.rmSync(path.join(modulePath, dir), { recursive: true, force: true })
|
|
console.log(`[After Pack] Removed dir: ${dir}`, arch)
|
|
})
|
|
}
|