const fs = require('fs-extra');
|
const { execSync } = require('child_process');
|
const os = require('os');
|
const path = require('path');
|
const chalk = require('chalk');
|
|
// 获取当前脚本所在的目录
|
const scriptDir = __dirname;
|
const rootDir = path.resolve(scriptDir, '..');
|
const argv2 = process.argv[2];
|
const arg2 = argv2?.split(' ');
|
|
const customerList = arg2?.[0] ? (arg2[0] === 'pro' ? ['ch:pro'] : ['ch']) : ['ch'];
|
const publicDir = path.join(rootDir, 'public');
|
const distDir = path.join(rootDir, 'dist');
|
const customerListDir = path.join(rootDir, 'customer_list');
|
const customerProjectListDir = path.join(rootDir, 'src', 'views', 'project');
|
const homeDir = os.homedir();
|
// const deployEnv = process.argv[3];
|
|
const logColor = (text, color) => {
|
console.log(chalk[color](text));
|
};
|
|
const logError = (text) => {
|
logColor(text, 'red');
|
};
|
|
const logSuccess = (text) => {
|
logColor(text, 'green');
|
};
|
|
/**
|
* 检查文件是否存在
|
* @param {*} file
|
* @param {*} fileText
|
*/
|
const checkFileExist = (file, fileText) => {
|
// 验证源文件夹是否存在
|
if (!fs.existsSync(file)) {
|
console.error(chalk.red(`${fileText} "${file}" 不存在！`));
|
process.exit(1); // 退出脚本
|
}
|
};
|
|
const checkCustomerDirExist = (customer) => {
|
const customerDir = path.join(customerListDir, customer);
|
checkFileExist(customerDir, '客户文件夹');
|
};
|
|
/**
|
* 检查命令，是否加上用户名。以及用户文件夹是否存在
|
* @param {*} command
|
* @param {*} customer
|
*/
|
const checkCustomer = (command, customer) => {
|
if (!customer) {
|
console.error(chalk.red(`请正确使用命令 “${command} [customer]” `));
|
|
process.exit(1); // 退出脚本
|
}
|
checkCustomerDirExist(customer);
|
};
|
|
let tmpBakDir = '';
|
const moveProjectsToBakDir = (command) => {
|
const customer = customerList[0]?.split(':')[0];
|
checkCustomer(command, customer);
|
//#region ====================== 创建备份文件夹 ======================
|
let tmpBakRootDir = path.join(homeDir, 'Desktop');
|
if (!fs.existsSync(tmpBakRootDir)) {
|
tmpBakRootDir = homeDir;
|
}
|
tmpBakDir = path.join(tmpBakRootDir, 'bak_project_list');
|
|
fs.ensureDirSync(tmpBakDir);
|
fs.emptyDirSync(tmpBakDir);
|
//#endregion
|
|
const contents = fs.readdirSync(customerProjectListDir);
|
for (let index = 0; index < contents.length; index++) {
|
const item = contents[index];
|
if (item === customer) {
|
continue;
|
}
|
const itemPath = path.join(customerProjectListDir, item);
|
const stats = fs.statSync(itemPath);
|
if (stats.isDirectory()) {
|
const destBakDir = path.join(tmpBakDir, item);
|
fs.moveSync(itemPath, destBakDir, { overwrite: true });
|
}
|
}
|
};
|
|
const restoreProjectDir = () => {
|
const contents = fs.readdirSync(tmpBakDir);
|
for (let index = 0; index < contents.length; index++) {
|
const item = contents[index];
|
const itemPath = path.join(tmpBakDir, item);
|
const destProjectDir = path.join(customerProjectListDir, item);
|
fs.moveSync(itemPath, destProjectDir, { overwrite: true });
|
}
|
};
|
|
/**
|
* 复制文件到 public 文件夹中
|
* (只能复制一个客户文件配置)
|
* @param {*} command
|
* @param {*} customer
|
*/
|
function copyFile(command) {
|
const customer = customerList[0]?.split(':')[0];
|
checkCustomer(command, customer);
|
// 确保 public 文件，不存在则创建
|
fs.ensureDirSync(publicDir);
|
// 清空 public 文件夹
|
fs.emptyDirSync(publicDir);
|
|
// 使用 filter 过滤器，排除特定文件
|
const filter = (src, dest) => {
|
const fileName = path.basename(src);
|
// 排除特定文件
|
if (fileName === 'deploy.json') {
|
return false;
|
}
|
// 其他文件保留
|
return true;
|
};
|
|
const customerDir = path.join(customerListDir, customer);
|
|
// 复制源文件夹中的所有文件到 public 文件夹
|
fs.copySync(customerDir, publicDir, { filter });
|
}
|
|
/**
|
* 上传文件到服务器
|
* @param {*} command
|
*/
|
const uploadFiles = async () => {
|
let customerConfig = {};
|
for (let index = 0; index < customerList.length; index++) {
|
const item = customerList[index];
|
const customerSplit = item.split(':');
|
const customerName = customerSplit[0];
|
checkCustomerDirExist(customerName);
|
const deployJSON = path.join(customerListDir, customerName, 'deploy.json');
|
// 验证部署配置文件是否存在
|
checkFileExist(deployJSON, '配置文件');
|
// 读取 JSON 文件
|
const config = await fs.readJson(deployJSON).catch((err) => {
|
console.error(`读取配置文件“${deployJSON}”出错:`, err);
|
process.exit(1);
|
});
|
|
let deployConfig = null;
|
const deployEnv = customerSplit[1];
|
if (deployEnv === 'pro') {
|
deployConfig = config?.product;
|
} else {
|
deployConfig = config?.test;
|
}
|
if (!deployConfig || Object.values(deployConfig).some((item) => !item)) {
|
console.error(chalk.red(`${customerName} ${deployEnv} 配置不完整！`));
|
process.exit(1); // 退出脚本
|
}
|
// 缓存部署配置
|
customerConfig[item] = deployConfig;
|
}
|
for (const customerDeployName in customerConfig) {
|
if (Object.hasOwnProperty.call(customerConfig, customerDeployName)) {
|
const customerName = customerDeployName.split(':')[0];
|
const deployConfig = customerConfig[customerDeployName];
|
const remoteFolderPath = deployConfig.path;
|
// 获取用户临时目录
|
const userTempDir = os.tmpdir();
|
const timeStamp = new Date().getTime();
|
const uploadScriptFile = path.join(userTempDir, `psftpBatchFile${timeStamp}${customerDeployName}`);
|
// 构建 sftp 命令
|
const sftpCommand = 'psftp';
|
// 上传 dist 中的 assets 和 index.html
|
let uploadCommand = `put -r ${path.join(distDir, 'assets')} ${remoteFolderPath + '/assets'}
|
put ${path.join(distDir, 'index.html')} ${remoteFolderPath + '/index.html'}\n`;
|
|
const customerFolder = path.join(customerListDir, customerName);
|
// 客户文件夹中除 deploy.json 全部上传
|
try {
|
const contents = fs.readdirSync(customerFolder);
|
for (let index = 0; index < contents.length; index++) {
|
const item = contents[index];
|
if (item === 'deploy.json') {
|
continue;
|
}
|
const itemPath = path.join(customerFolder, item);
|
const remotePath = remoteFolderPath + `/${item}`;
|
const stats = fs.statSync(itemPath);
|
|
// 文件夹需要加 -r
|
if (stats.isDirectory()) {
|
uploadCommand += `put -r ${itemPath} ${remotePath}\n`;
|
// 文件不需要加
|
} else if (stats.isFile()) {
|
uploadCommand += `put ${itemPath} ${remotePath}\n`;
|
}
|
}
|
} catch (error) {
|
console.error(`读取“${customerFolder}”失败`, error);
|
}
|
|
try {
|
fs.writeFileSync(uploadScriptFile, uploadCommand);
|
const sftpArgs = [
|
`${deployConfig.username}@${deployConfig.host} -P ${deployConfig.port} -pw ${deployConfig.password} -b ${uploadScriptFile}`,
|
];
|
// 完整的执行命令
|
const fullCommand = `${sftpCommand} ${sftpArgs.join(' ')}`;
|
logSuccess(`正在上传【${customerName}】项目至服务器【${deployConfig.host}:${remoteFolderPath}】...`);
|
execSync(fullCommand, { stdio: ['pipe', 'inherit', 'inherit'] });
|
} catch (error) {
|
console.error(chalk.red(error));
|
} finally {
|
if (fs.existsSync(uploadScriptFile)) {
|
fs.unlinkSync(uploadScriptFile);
|
}
|
}
|
}
|
}
|
logSuccess(`🎉🎉🎉【${customerList.join(',')}】项目已成功部署！🎉🎉🎉`);
|
};
|
|
module.exports = {
|
customerList,
|
//#region ====================== 文件路径 ======================
|
rootDir,
|
scriptDir,
|
publicDir,
|
distDir,
|
customerListDir,
|
//#endregion
|
|
//#region ====================== 工具函数 ======================
|
checkCustomerDirExist,
|
//#endregion
|
checkCustomer,
|
copyFile,
|
uploadFiles,
|
|
//#region ====================== 打印 ======================
|
logError,
|
logSuccess,
|
//#endregion
|
|
moveProjectsToBakDir,
|
restoreProjectDir,
|
};
|
