diff --git a/upload.js b/upload.js index ba55a77..ff07382 100755 --- a/upload.js +++ b/upload.js @@ -4,7 +4,7 @@ const spinner_style = require('./spinner_style') //加载动画样式 const shell = require('shelljs') // 执行shell命令 const node_ssh = require('node-ssh') // ssh连接服务器 const inquirer = require('inquirer') //命令行交互 -const zipFile = require('compressing')// 压缩zip +const zipFile = require('compressing') // 压缩zip const fs = require('fs') // nodejs内置文件模块 const path = require('path') // nodejs内置路径模块 const CONFIG = require('./config') // 配置 @@ -23,9 +23,9 @@ const distZipPath = path.resolve(__dirname, `./smx-bundle.tar.gz`); //打包后 //压缩代码 -const zipDist = async ()=>{ - const distDir = path.resolve(__dirname, '../'+config.OUTPUT_PATH); - const loading = ora( defaultLog('正在连接服务器') ).start(); +const zipDist = async () => { + const distDir = path.resolve(__dirname, '../' + config.OUTPUT_PATH); + const loading = ora(defaultLog('正在连接服务器')).start(); loading.spinner = spinner_style.arrow4; try { await zipFile.tgz.compressDir(distDir, distZipPath) @@ -39,26 +39,25 @@ const zipDist = async ()=>{ } //连接服务器 -const connectSSH = async ()=>{ - const loading = ora( defaultLog('正在连接服务器') ).start(); +const connectSSH = async () => { + const loading = ora(defaultLog('正在连接服务器')).start(); loading.spinner = spinner_style.arrow4; try { - const options = config.privateKey ? - { + const options = config.privateKey ? { host: config.SERVER_PATH, username: config.SSH_USER, privateKey: config.PRIVATE_KEY, //秘钥登录(推荐) 方式一 - tryKeyboard : true - }: { + tryKeyboard: true + } : { host: config.SERVER_PATH, username: config.SSH_USER, password: config.PASSWORD, // 密码登录 方式二 - tryKeyboard : true + tryKeyboard: true } await SSH.connect(options); - successLog('SSH连接成功!'); + successLog('SSH连接成功!'); } catch (error) { errorLog(error); errorLog('SSH连接失败!'); @@ -70,43 +69,47 @@ const connectSSH = async ()=>{ //传送zip文件到服务器 -const uploadZipBySSH = async () =>{ +const uploadZipBySSH = async () => { //连接ssh await connectSSH(); //线上目标文件清空 - if(config.CLEAR_OLDFILES){ - await clearOldFile(); - } - - const loading = ora( defaultLog('准备上传文件') ).start(); + await clearOldFile(); + + + + + const loading = ora(defaultLog('准备上传文件')).start(); loading.spinner = spinner_style.arrow4; try { - await SSH.putFiles([{ local: distZipPath, remote: config.PATH + '/smx-bundle.tar.gz' }]); //local 本地 ; remote 服务器 ; - successLog('上传成功!'); + await SSH.putFiles([{ + local: distZipPath, + remote: config.PATH + '/smx-bundle.tar.gz' + }]); //local 本地 ; remote 服务器 ; + successLog('上传成功!'); loading.text = '正在解压文件'; await runCommand('tar -zxvf ./smx-bundle.tar.gz'); //解压 // 是否备份文件 - if(config.BACKUP){ + if (config.BACKUP) { await runCommand(`mv ${config.PATH}/smx-bundle.tar.gz ${config.PATH}/${config.OLD_NAME}.${new Date().toLocaleDateString()}.tar.gz`); //解压完删除线上压缩包 - }else{ + } else { await runCommand(`rm -rf ${config.PATH}/smx-bundle.tar.gz`); //解压完删除线上压缩包 } - + //将目标目录改名 - if(config.RENAME){ - await runCommand(`mv ${config.PATH}/${config.OLD_NAME} ${config.PATH}/${config.RENAME}`); + if (config.RENAME) { + await runCommand(`mv ${config.PATH}/${config.OLD_NAME} ${config.PATH}/${config.RENAME}`); } // 后续扩展命令 - if(config.EXTENDS){ - for(let v of config.EXTENDS){ - await runCommand(v) + if (config.EXTENDS) { + for (let v of config.EXTENDS) { + await runCommand(v) } } - + SSH.dispose(); //断开连接 } catch (error) { errorLog(error); @@ -118,27 +121,43 @@ const uploadZipBySSH = async () =>{ //清空线上目标目录里的旧文件 -const clearOldFile = async () =>{ - const commands = ['ls', 'rm -rf *']; - await Promise.all(commands.map(async (it)=>{ - return await runCommand(it); - })); +const clearOldFile = async () => { + + try { + await runCommand(`rm -r ${config.PATH}/${config.OLD_NAME}`); + } catch { + // console.log('服务器未发现原始部署文档'); + } + if (config.RENAME) { + try { + await runCommand(`rm -r ${config.PATH}/${config.RENAME}`); + } catch { + // console.log('服务器未发现原始部署文档'); + } + } + + if (config.CLEAR_OLDFILES) { + const commands = ['ls', 'rm -rf *']; + await Promise.all(commands.map(async (it) => { + return await runCommand(it); + })); + } } // 删除本地上传后的打包文件 -const deleteFile = function async() { +const deleteFile = function async () { delPath = distZipPath; try { - /** - * @des 判断文件或文件夹是否存在 - */ - if (fs.existsSync(delPath)) { - fs.unlinkSync(delPath); - } else { - console.log('inexistence path：', delPath); - } + /** + * @des 判断文件或文件夹是否存在 + */ + if (fs.existsSync(delPath)) { + fs.unlinkSync(delPath); + } else { + console.log('inexistence path：', delPath); + } } catch (error) { - console.log('删除本地打包文件失败', error); + console.log('删除本地打包文件失败', error); } } @@ -148,8 +167,10 @@ const deleteFile = function async() { * 线上执行命令 * @param {String} command 命令操作 如 ls */ -const runCommand = async (command)=> { - const result = await SSH.exec(command, [], { cwd: config.PATH}) +const runCommand = async (command) => { + const result = await SSH.exec(command, [], { + cwd: config.PATH + }) // defaultLog(result); } @@ -166,12 +187,12 @@ const runUploadTask = async () => { await zipDist(); //连接服务器上传文件 - await uploadZipBySSH(); + await uploadZipBySSH(); //删除本地打包文件 await deleteFile(); - successLog('大吉大利, 部署成功!'); + successLog('大吉大利, 部署成功!'); process.exit(); } @@ -180,16 +201,16 @@ const runUploadTask = async () => { * * @param {Object} conf 配置对象 */ -const checkConfig = (conf) =>{ +const checkConfig = (conf) => { const checkArr = Object.entries(conf); - checkArr.map(it=>{ + checkArr.map(it => { const key = it[0]; - if(key === 'PATH' && conf[key] === '/') { //上传压缩包前会清空目标目录内所有文件 - errorLog('PATH 不能是服务器根目录!'); + if (key === 'PATH' && conf[key] === '/') { //上传压缩包前会清空目标目录内所有文件 + errorLog('PATH 不能是服务器根目录!'); process.exit(); //退出流程 } - if(!conf[key]) { - errorLog(`配置项 ${key} 不能为空`); + if (!conf[key]) { + errorLog(`配置项 ${key} 不能为空`); process.exit(); //退出流程 } }) @@ -199,31 +220,40 @@ const checkConfig = (conf) =>{ // 执行交互后 启动发布程序 inquirer .prompt([{ + type: 'confirm', + message: '请确认您的上传目录及修改目录(RENAME)在服务器上没有被用于存储其他数据,以下部署将会删除这些目录', + name: 'confirm', + }, { type: 'list', message: '选择您的部署环境', name: 'env', choices: [{ name: '测试环境', value: 'development' - },{ + }, { name: '正式环境', value: 'production' - }] + }], + when: function (answers) { // 当watch为true的时候才会提问当前问题 + return answers.confirm + } }]) .then(answers => { - config = CONFIG[answers.env]; - if(!config.OUTPUT_PATH){ - config.OUTPUT_PATH = 'dist'; - } + if (answers.confirm) { + config = CONFIG[answers.env]; + if (!config.OUTPUT_PATH) { + config.OUTPUT_PATH = 'dist'; + } - const names = config.OUTPUT_PATH.split('/'); - if(names[names.length - 1]){ - config.OLD_NAME = names[names.length - 1]; - }else{ - config.OLD_NAME = names[names.length - 2]; - } - + const names = config.OUTPUT_PATH.split('/'); + if (names[names.length - 1]) { + config.OLD_NAME = names[names.length - 1]; + } else { + config.OLD_NAME = names[names.length - 2]; + } - checkConfig(config); // 检查 - runUploadTask(); // 发布 - }); + + checkConfig(config); // 检查 + runUploadTask(); // 发布 + } + }); \ No newline at end of file