|
|
@@ -0,0 +1,103 @@
|
|
|
+import Client from 'ssh2-sftp-client';
|
|
|
+import ssh2 from 'ssh2';
|
|
|
+import chalk from 'chalk';
|
|
|
+import ora from 'ora';
|
|
|
+import shell from 'shelljs';
|
|
|
+import path from 'path';
|
|
|
+import { fileURLToPath } from 'url';
|
|
|
+import { config } from './config.js';
|
|
|
+
|
|
|
+const rawArgv = process.argv.slice(2);
|
|
|
+const filterStage = rawArgv.includes("--prod") ? "prod" : "test";
|
|
|
+const __dirname = path.dirname(fileURLToPath(import.meta.url));
|
|
|
+
|
|
|
+// 构建
|
|
|
+const compileDist = async () => {
|
|
|
+ console.log(chalk.blue("项目开始构建"));
|
|
|
+ if (shell.exec(`npm run build:${filterStage}`).code === 0) {
|
|
|
+ console.log(chalk.blue("项目构建成功"));
|
|
|
+ }
|
|
|
+}
|
|
|
+
|
|
|
+// 获取配置
|
|
|
+const getConfig = () => {
|
|
|
+ const [result] = config.filter(({ nodeEnv }) => nodeEnv === filterStage);
|
|
|
+ return result;
|
|
|
+}
|
|
|
+
|
|
|
+const execRemoteCommand = async () => {
|
|
|
+ const config = getConfig();
|
|
|
+ const conn = new ssh2.Client();
|
|
|
+ conn.on("ready", function () {
|
|
|
+ console.log(chalk.blue("ssh2 链接成功 - 执行 pm2 restart bigModelAdmin"));
|
|
|
+ conn.exec("pm2 restart bigModelAdmin", function (err, stream) {
|
|
|
+ if (err) throw err;
|
|
|
+ console.log(
|
|
|
+ chalk.blue(``) + chalk.green(`---执行成功---`),
|
|
|
+ );
|
|
|
+ console.log(
|
|
|
+ chalk.blue(`✅`) + chalk.green(`---部署完成---`),
|
|
|
+ );
|
|
|
+ conn.end();
|
|
|
+ });
|
|
|
+ }).connect({ ...config });
|
|
|
+}
|
|
|
+
|
|
|
+
|
|
|
+// 部署
|
|
|
+const connectShell = async () => {
|
|
|
+ const sftp = new Client();
|
|
|
+ const item = getConfig();
|
|
|
+ let spinner = null;
|
|
|
+
|
|
|
+ // 打印信息
|
|
|
+ const printMsg = ({ color, text }) => {
|
|
|
+ console.log(chalk.red(`${item.host} --> `) + chalk[color](text));
|
|
|
+ }
|
|
|
+
|
|
|
+ printMsg({ color: 'green', text: '服务器连接中' });
|
|
|
+
|
|
|
+ sftp.connect({
|
|
|
+ host: item.host,
|
|
|
+ port: item.port,
|
|
|
+ username: item.username,
|
|
|
+ password: item.password,
|
|
|
+ })
|
|
|
+ .then(() => {
|
|
|
+ printMsg({ color: 'green', text: '服务器连接成功' });
|
|
|
+ printMsg({ color: 'yellow', text: '执行删除文件中' });
|
|
|
+ return sftp.rmdir(item.path, true);
|
|
|
+ })
|
|
|
+ .then(() => {
|
|
|
+ printMsg({ color: 'green', text: '执行删除文件成功' });
|
|
|
+ printMsg({ color: 'green', text: '即将开始上传' });
|
|
|
+ spinner = ora().start();
|
|
|
+ spinner.text = '文件上传中,请等待'
|
|
|
+ return sftp.uploadDir(path.resolve(__dirname, "../dist"), item.path);
|
|
|
+ })
|
|
|
+ .then(() => {
|
|
|
+ spinner.info('文件上传结束')
|
|
|
+ spinner.stop();
|
|
|
+
|
|
|
+ if (filterStage === 'prod') {
|
|
|
+ execRemoteCommand()
|
|
|
+ }
|
|
|
+
|
|
|
+ printMsg({ color: 'green', text: '上传完成,部署成功' });
|
|
|
+ sftp.end();
|
|
|
+ })
|
|
|
+ .catch((err) => {
|
|
|
+ console.error(
|
|
|
+ err,
|
|
|
+ chalk.red(`${item.host} -->`) + chalk.red(`上传失败`)
|
|
|
+ );
|
|
|
+ sftp.end();
|
|
|
+ });
|
|
|
+}
|
|
|
+
|
|
|
+async function runStart() {
|
|
|
+ await compileDist();
|
|
|
+ await connectShell();
|
|
|
+}
|
|
|
+
|
|
|
+runStart();
|
sunxiao