yj-deploy 0.0.9 → 0.0.10

package/dist/yj-deploy.mjs

@@ -1,7 +1,7 @@
 var x = Object.defineProperty;
-var L = (f, e, t) => e in f ? x(f, e, { enumerable: !0, configurable: !0, writable: !0, value: t }) : f[e] = t;
-var $ = (f, e, t) => (L(f, typeof e != "symbol" ? e + "" : e, t), t);
-const { stdout: A } = require("single-line-log"), F = require("path"), C = require("fs"), { Client: N } = require("ssh2"), I = require("./rancher");
+var L = (g, e, o) => e in g ? x(g, e, { enumerable: !0, configurable: !0, writable: !0, value: o }) : g[e] = o;
+var $ = (g, e, o) => (L(g, typeof e != "symbol" ? e + "" : e, o), o);
+const { stdout: F } = require("single-line-log"), N = require("path"), S = require("fs"), { Client: U } = require("ssh2"), A = require("axios"), R = require("https");
 module.exports = class {
   constructor(e) {
     $(this, "config", {
@@ -125,9 +125,9 @@ module.exports = class {
       console.log("- 请配置跳板机密码 password");
       return;
     }
-    this.uploading || (this.ssh2Conn = new N(), this.ssh2Conn.on("ready", () => {
-      console.log("- 跳板机连接成功！"), this.ssh2Conn.sftp(async (e, t) => {
-        e && (console.log("- sftp连接失败"), this.breakConnect()), this.config.namespace || (console.log("- 请配置项目命名空间 namespace"), this.breakConnect()), this.uploading = !0, this.getArgv("--reset") !== void 0 && (console.log("- 重置项目"), await this.onShell(`rm -r ${this.config.rootDir}/${this.config.namespace}`)), this.upLoadProject(t, `${this.config.rootDir}/${this.config.namespace}`);
+    this.uploading || (this.ssh2Conn = new U(), this.ssh2Conn.on("ready", () => {
+      console.log("- 跳板机连接成功！"), this.ssh2Conn.sftp(async (e, o) => {
+        e && (console.log("- sftp连接失败"), this.breakConnect()), this.config.namespace || (console.log("- 请配置项目命名空间 namespace"), this.breakConnect()), this.uploading = !0, this.getArgv("--reset") !== void 0 && (console.log("- 重置项目"), await this.onShell(`rm -r ${this.config.rootDir}/${this.config.namespace}`)), this.upLoadProject(o, `${this.config.rootDir}/${this.config.namespace}`);
       });
     }).connect({
       host: this.config.host,
@@ -152,23 +152,23 @@ module.exports = class {
    * @returns Promise
    */
   onShell(e) {
-    const t = this.getArgv("--allLog") !== void 0;
-    return new Promise((o, r) => {
+    const o = this.getArgv("--allLog") !== void 0;
+    return new Promise((t, i) => {
       this.ssh2Conn.shell((l, s) => {
         if (l) {
-          console.log("- 远程命令错误：" + l, e), this.breakConnect(), r(l);
+          console.log("- 远程命令错误：" + l, e), this.breakConnect(), i(l);
           return;
         }
-        let i = [];
+        let n = [];
         s.on("close", () => {
-          o(i.toString());
-        }).on("data", (c) => {
-          let a = c.toString();
-          (t || this.config.allLog || a.startsWith("- ")) && i.push(`
-` + a);
-        }).stderr.on("data", (c) => {
+          t(n.toString());
+        }).on("data", (a) => {
+          let c = a.toString();
+          (o || this.config.allLog || c.startsWith("- ")) && n.push(`
+` + c);
+        }).stderr.on("data", (a) => {
           console.log(`- 远程命令错误：
-` + c), this.breakConnect(), r(c);
+` + a), this.breakConnect(), i(a);
         }), s.end(e + `
 exit
 `);
@@ -180,19 +180,19 @@ exit
    * @param {*} sftp 
    * @param {*} dir 
    */
-  projectInit(e, t) {
-    return new Promise(async (o, r) => {
-      e.mkdir(t, async (l) => {
+  projectInit(e, o) {
+    return new Promise(async (t, i) => {
+      e.mkdir(o, async (l) => {
         if (l)
-          r(l);
+          i(l);
         else {
           console.log("- 正在创建dockerfile");
-          const s = `${t}/${this.config.dockerfile.name}`;
+          const s = `${o}/${this.config.dockerfile.name}`;
           await this.onShell(`touch ${s}`), await this.onShell(`echo '${this.config.dockerfile.content.join(`
 `)}' > ${s}`), console.log("- 正在创建upload.sh");
-          const i = `${t}/${this.config.upload.name}`;
-          await this.onShell(`touch ${i}`), await this.onShell(`echo '${this.config.upload.content.join(`
-`)}' > ${i}`), o();
+          const n = `${o}/${this.config.upload.name}`;
+          await this.onShell(`touch ${n}`), await this.onShell(`echo '${this.config.upload.content.join(`
+`)}' > ${n}`), t();
         }
       });
     });
@@ -203,10 +203,10 @@ exit
    * @param {*} remotePath 
    * @returns 
    */
-  checkRemoteFile(e, t) {
-    return new Promise((o) => {
-      e.stat(t, (r) => {
-        o(!r);
+  checkRemoteFile(e, o) {
+    return new Promise((t) => {
+      e.stat(o, (i) => {
+        t(!i);
       });
     });
   }
@@ -215,98 +215,98 @@ exit
    * @param {*} sftp 
    * @param {*} dir 
    */
-  upLoadProject(e, t) {
-    let o = 0;
-    typeof this.config.rePushNum == "number" && this.config.rePushNum > 0 && (o = this.config.rePushNum), e.readdir(t, async (r) => {
-      if (r) {
-        console.log("- 跳板机不存在项目, 开始创建项目"), this.projectInit(e, t).then(() => {
-          console.log(`- ${this.config.namespace}项目创建成功`), this.upLoadProject(e, t);
-        }).catch((n) => {
-          console.log(`- 创建项目失败: ${n}`), this.breakConnect();
+  upLoadProject(e, o) {
+    let t = 0;
+    typeof this.config.rePushNum == "number" && this.config.rePushNum > 0 && (t = this.config.rePushNum), e.readdir(o, async (i) => {
+      if (i) {
+        console.log("- 跳板机不存在项目, 开始创建项目"), this.projectInit(e, o).then(() => {
+          console.log(`- ${this.config.namespace}项目创建成功`), this.upLoadProject(e, o);
+        }).catch((r) => {
+          console.log(`- 创建项目失败: ${r}`), this.breakConnect();
         });
         return;
       }
-      const l = this.getArgv("--lazyUpload") !== void 0 || this.config.lazyUpload, s = `${t}/${this.config.dist.name}`;
+      const l = this.getArgv("--lazyUpload") !== void 0 || this.config.lazyUpload, s = `${o}/${this.config.dist.name}`;
       if (l || await this.onShell(`rm -rf ${s}`), await this.onShell(`mkdir ${s}`), !this.config.fileDir) {
         console.log("- 请配置待上传文件目录 fileDir"), this.breakConnect();
         return;
       }
-      const { localFileList: i, dirList: c } = await this.getFilesInDirectory(this.config.fileDir);
-      if (i.length === 0) {
+      const { localFileList: n, dirList: a } = await this.getFilesInDirectory(this.config.fileDir);
+      if (n.length === 0) {
         console.log("- 待上传目录没有获取到文件，请检查"), this.breakConnect();
         return;
       }
-      let a = 0, p = i.length;
-      const y = (n) => new Promise(async (u, g) => {
+      let c = 0, m = n.length;
+      const w = (r) => new Promise(async (f, u) => {
         try {
-          if (l && await this.checkRemoteFile(e, `${s}${n.remotePath}`))
-            return u();
-          const h = C.createReadStream(n.localPath), k = e.createWriteStream(`${s}${n.remotePath}`);
-          k.on("close", () => {
-            this.progressBar(++a, p), u();
-          }), k.on("error", (b) => {
-            console.log(`- 文件 ${n.remotePath} 上传失败：${b}`), g(b), this.breakConnect();
-          }), h.pipe(k);
+          if (l && await this.checkRemoteFile(e, `${s}${r.remotePath}`))
+            return f();
+          const h = S.createReadStream(r.localPath), b = e.createWriteStream(`${s}${r.remotePath}`);
+          b.on("close", () => {
+            this.progressBar(++c, m), f();
+          }), b.on("error", (C) => {
+            console.log(`- 文件 ${r.remotePath} 上传失败：${C}`), u(C), this.breakConnect();
+          }), h.pipe(b);
         } catch (h) {
-          g(h);
+          u(h);
         }
-      }), d = (n, u) => {
-        let g = [];
-        for (let h = 0; h < n.length; h += u)
-          g.push(n.slice(h, h + u));
-        return g;
-      }, m = (n) => {
-        if (n.length === 0)
+      }), d = (r, f) => {
+        let u = [];
+        for (let h = 0; h < r.length; h += f)
+          u.push(r.slice(h, h + f));
+        return u;
+      }, p = (r) => {
+        if (r.length === 0)
           return Promise.resolve();
-        const u = n.shift();
-        return Promise.all(u.map((g) => y(g))).then(() => m(n));
-      }, P = (n) => {
-        if (n.length === 0)
+        const f = r.shift();
+        return Promise.all(f.map((u) => w(u))).then(() => p(r));
+      }, k = (r) => {
+        if (r.length === 0)
           return Promise.resolve();
-        let g = n.shift().map((h) => `${s}${h}`).join(" ");
-        return this.onShell(`mkdir -p ${g}`).then(() => P(n));
+        let u = r.shift().map((h) => `${s}${h}`).join(" ");
+        return this.onShell(`mkdir -p ${u}`).then(() => k(r));
       };
-      let w = [...d(c, this.config.parallelDir || 20)];
-      await P(w), console.log("- 创建目录完成"), await new Promise((n) => {
+      let y = [...d(a, this.config.parallelDir || 20)];
+      await k(y), console.log("- 创建目录完成"), await new Promise((r) => {
         setTimeout(() => {
-          n();
+          r();
         }, 500);
       });
-      let D = [...d(i, this.config.parallelFile || 50)];
-      await m(D), l && (this.progressBar(p, p), console.log("\x1B[32m%s\x1B[0m", `- 已开启懒上传，本次共上传 ${a} 个文件`)), console.log("- 文件上传成功"), console.log("- 开始推送镜像");
-      let S = `cd ${t} && sh ${this.config.upload.name}`;
-      if (await this.isNewShell(e, t)) {
-        const n = this.getArgv("--imageStore") || this.config.imageStore;
-        S += ` ${n}`;
+      let I = [...d(n, this.config.parallelFile || 50)];
+      await p(I), l && (this.progressBar(m, m), console.log("\x1B[32m%s\x1B[0m", `- 已开启懒上传，本次共上传 ${c} 个文件`)), console.log("- 文件上传成功"), console.log("- 开始推送镜像");
+      let P = `cd ${o} && sh ${this.config.upload.name}`;
+      if (await this.isNewShell(e, o)) {
+        const r = this.getArgv("--imageStore") || this.config.imageStore;
+        P += ` ${r}`;
       } else
         console.log(`
 - warning 检测到当前脚本为手动创建,不支持镜像仓库配置,imageStore将失效`), console.log("- warning 如需支持imageStore参数,请在命令行后添加 --reset 重新生成脚本");
       const v = this.getArgv("--tmpName") || this.config.tmpName;
-      S += ` ${v}`;
-      const j = Array.from(new Array(o + 1)).map(() => this.onShell(S));
-      await this.pushMirrorImage(j, j.length), this.breakConnect();
+      P += ` ${v}`;
+      const D = Array.from(new Array(t + 1)).map(() => this.onShell(P));
+      await this.pushMirrorImage(D, D.length), this.breakConnect();
     });
   }
   // 推送镜像
-  async pushMirrorImage(e, t) {
+  async pushMirrorImage(e, o) {
     if (e.length === 0)
       return console.log("- 镜像推送失败，请添加 --allLog 查看全部日志"), null;
-    const o = await e[0];
-    return o.includes("- 镜像推送失败") ? (console.log(o), e.length > 1 && console.log(`- 正在尝试第 ${t - e.length + 1} 次推送...`), this.pushMirrorImage(e.slice(1), t)) : (console.log(o), console.log("- 镜像推送结束"), this.getArgv("--restart") !== void 0 && await I(o, this.config.rancherConfig), o);
+    const t = await e[0];
+    return t.includes("- 镜像推送失败") ? (console.log(t), e.length > 1 && console.log(`- 正在尝试第 ${o - e.length + 1} 次推送...`), this.pushMirrorImage(e.slice(1), o)) : (console.log(t), console.log("- 镜像推送结束"), this.getArgv("--restart") !== void 0 && await q(t, this.config.rancherConfig), t);
   }
   /**
    * 获取脚本内容是不是自动创建的脚本
    * 兼容用户手动根据运维文档创建的推送脚本，因参数不同可能报错的问题
    * 有 $2代表是自动创建的版本，否则老版本
   */
-  async isNewShell(e, t) {
-    return new Promise((o, r) => {
-      e.readFile(`${t}/${this.config.upload.name}`, (l, s) => {
+  async isNewShell(e, o) {
+    return new Promise((t, i) => {
+      e.readFile(`${o}/${this.config.upload.name}`, (l, s) => {
         if (l)
-          return r(!1);
-        const i = s.toString();
-        i.includes("- 镜像推送失败") || console.log(`
-- warning 检测到当前脚本可能不是最新版本，推送失败可能不会提示，且没有失败自动重试功能，请在命令后添加 --reset 重新生成脚本`), o(i.includes("$2"));
+          return i(!1);
+        const n = s.toString();
+        n.includes("- 镜像推送失败") || console.log(`
+- warning 检测到当前脚本可能不是最新版本，推送失败可能不会提示，且没有失败自动重试功能，请在命令后添加 --reset 重新生成脚本`), t(n.includes("$2"));
       });
     });
   }
@@ -315,41 +315,41 @@ exit
    * @param description 命令行开头的文字信息
    * @param bar_length 进度条的长度(单位：字符)，默认设为 25
    */
-  progressBar(e, t, o = 25) {
-    let r = (e / t).toFixed(4), l = Math.floor(r * o), s = "";
-    for (let a = 0; a < l; a++)
+  progressBar(e, o, t = 25) {
+    let i = (e / o).toFixed(4), l = Math.floor(i * t), s = "";
+    for (let c = 0; c < l; c++)
       s += "█";
-    let i = "";
-    for (let a = 0; a < o - l; a++)
-      i += "░";
-    let c = `- 文件上传进度: ${s}${i} (${e}/${t}) ${(100 * r).toFixed(2)}%`;
-    A(c), e == t && console.log("");
+    let n = "";
+    for (let c = 0; c < t - l; c++)
+      n += "░";
+    let a = `- 文件上传进度: ${s}${n} (${e}/${o}) ${(100 * i).toFixed(2)}%`;
+    F(a), e == o && console.log("");
   }
   /**
    * 获取目录下所有文件并分类
    * @param {*} dir 
    */
   getFilesInDirectory(e) {
-    const t = [], o = [];
-    return new Promise((r, l) => {
-      const s = (i, c, a) => {
-        const p = C.readdirSync(i);
-        let y = !1;
-        p.forEach((d) => {
-          const m = F.join(i, d), w = C.statSync(m).isDirectory();
-          w ? y = !0 : t.push({
-            localPath: m,
+    const o = [], t = [];
+    return new Promise((i, l) => {
+      const s = (n, a, c) => {
+        const m = S.readdirSync(n);
+        let w = !1;
+        m.forEach((d) => {
+          const p = N.join(n, d), y = S.statSync(p).isDirectory();
+          y ? w = !0 : o.push({
+            localPath: p,
             // 本地路径
-            remotePath: `${c}${d}`,
+            remotePath: `${a}${d}`,
             // 远程路径
-            level: a
+            level: c
             // 文件或目录层级
-          }), w && s(m, `${c}${d}/`, a + 1);
-        }), y || o.push(c);
+          }), y && s(p, `${a}${d}/`, c + 1);
+        }), w || t.push(a);
       };
-      s(e, "/", 0), r({
-        localFileList: t,
-        dirList: o
+      s(e, "/", 0), i({
+        localFileList: o,
+        dirList: t
       });
     });
   }
@@ -358,10 +358,70 @@ exit
    * @param { string } name
    */
   getArgv(e) {
-    const t = process.argv;
-    let o;
-    return t.forEach((r) => {
-      r.indexOf(e) > -1 && (o = r.split("=")[1] || "");
-    }), o;
+    const o = process.argv;
+    let t;
+    return o.forEach((i) => {
+      i.indexOf(e) > -1 && (t = i.split("=")[1] || "");
+    }), t;
   }
 };
+const j = new R.Agent({
+  rejectUnauthorized: !1
+});
+async function q(g, e) {
+  try {
+    const t = {
+      // baseUrl: 'https://paas-test.ymygz.com:10443/v3', // Rancher API 基础地址
+      // token: 'token-2gqbg:pwpqvzqfbj26h9flwzbptcpz5sqn7jwh5c9ht2lfx46j4xz58875fr', // Rancher API 访问令牌
+      // clusterId: 'local:p-r7j28', // Rancher 集群 ID
+      // namespace: 'dev-wutaishan-onetravel', // 服务所在的命名空间
+      // workload: 'web-ymy-img', // 工作负载名称
+      ...{
+        baseUrl: "https://paas-test.ymygz.com:10443/v3",
+        // Rancher API 基础地址
+        clusterId: "local:p-r7j28"
+        // Rancher 集群 ID
+      },
+      ...e
+    };
+    if (!t.baseUrl)
+      return console.error("- Rancher API 基础地址未配置，请检查 rancherConfig.baseUrl 是否正确");
+    if (!t.token)
+      return console.error("- Rancher API 访问令牌未配置，请检查 rancherConfig.token 是否正确");
+    if (!t.clusterId)
+      return console.error("- Rancher 集群 ID 未配置，请检查 rancherConfig.clusterId 是否正确");
+    if (!t.namespace)
+      return console.error("- 服务所在的命名空间未配置，请检查 rancherConfig.namespace 是否正确");
+    if (!t.workload)
+      return console.error("- 工作负载名称未配置，请检查 rancherConfig.workload 是否正确");
+    if (!g)
+      return console.error("- 内部错误");
+    const i = g.match(/- 镜像地址:\s*(.+)/);
+    let l = "";
+    if (i && i[1])
+      l = i[1].trim();
+    else
+      return console.error("- 未提取到镜像地址");
+    console.log("- 开始调用 Rancher API 更新服务镜像地址");
+    const s = `${t.baseUrl}/projects/${t.clusterId}/workloads/deployment:${t.namespace}:${t.workload}`, n = await A.get(s, {
+      headers: { Authorization: `Bearer ${t.token}` },
+      httpsAgent: j
+    });
+    if (!n.data || !n.data.containers || n.data.containers.length === 0)
+      return console.error("- 未找到工作负载的容器信息，请检查 namespace 和 workload 是否正确");
+    const a = n.data;
+    a.containers[0].image = l, await A.put(s, a, {
+      headers: { Authorization: `Bearer ${t.token}` },
+      httpsAgent: j
+    }), console.log("- 镜像地址更新成功，正在重启服务..."), await A.post(
+      `${s}?action=redeploy`,
+      {},
+      {
+        headers: { Authorization: `Bearer ${t.token}` },
+        httpsAgent: j
+      }
+    ), console.log("- 服务重启成功");
+  } catch (o) {
+    console.error("- 调用 Rancher API 失败， 请手动重启服务:", o.message);
+  }
+}

package/dist/yj-deploy.umd.js

@@ -1,9 +1,9 @@
-(function(h){typeof define=="function"&&define.amd?define(h):h()})(function(){"use strict";var F=Object.defineProperty;var N=(h,u,f)=>u in h?F(h,u,{enumerable:!0,configurable:!0,writable:!0,value:f}):h[u]=f;var w=(h,u,f)=>(N(h,typeof u!="symbol"?u+"":u,f),f);const{stdout:h}=require("single-line-log"),u=require("path"),f=require("fs"),{Client:v}=require("ssh2"),x=require("./rancher");module.exports=class{constructor(e){w(this,"config",{host:"",port:"",username:"",password:"",namespace:"",imageStore:"dev-images",tmpName:"dev",delay:0,fileDir:"",rootDir:"/home/yjweb",dockerfile:{name:"Dockerfile",content:["FROM harbor.yunjingtech.cn:30002/yj-base/nginx:latest","RUN rm -rf /usr/share/nginx/html/*","COPY dist /usr/share/nginx/html/"]},upload:{name:"upload.sh",content:["#!/bin/sh","tag=`basename \\`pwd\\``:$2-`date +%Y%m%d`",'echo "- 镜像仓库: $1"','echo "- 镜像环境: $2"',"docker build -t harbor.yunjingtech.cn:30002/$1/$tag .","docker push harbor.yunjingtech.cn:30002/$1/$tag","if [ $? -eq 0 ];then","echo - 推送成功","echo - 镜像地址: harbor.yunjingtech.cn:30002/$1/$tag","else","echo - 镜像推送失败，请重试或联系运维人员，如需查看全部docker日志，请在命令后添加 --allLog查看更多信息","fi"]},dist:{name:"dist"},parallelDir:20,parallelFile:50,allLog:!1,lazyUpload:!1,rePushNum:3,rancherConfig:{token:"",namespace:"",workload:""}});w(this,"ssh2Conn",null);w(this,"uploading",!1);w(this,"trim",null);return this.config=Object.assign(this.config,e),{name:"yj-deploy",isPut:this.isPut.bind(this),upload:this.upload.bind(this),apply:this.apply.bind(this),closeBundle:this.isPut.bind(this)}}apply(e){return e&&e.hooks&&e.hooks.done&&e.hooks.done.tap("yj-deploy",()=>{this.isPut()}),"build"}isPut(){this.getArgv("--deploy")!=null&&(clearTimeout(this.trim),this.trim=setTimeout(()=>{this.upload()},this.config.delay||0))}upload(){if(console.log("-------------- deploy-start --------------"),!this.config.host){console.log("- 请配置跳板机地址 host");return}if(!this.config.port){console.log("- 请配置跳板机端口 port");return}if(!this.config.username){console.log("- 请配置跳板机账号 username");return}if(!this.config.password){console.log("- 请配置跳板机密码 password");return}this.uploading||(this.ssh2Conn=new v,this.ssh2Conn.on("ready",()=>{console.log("- 跳板机连接成功！"),this.ssh2Conn.sftp(async(e,t)=>{e&&(console.log("- sftp连接失败"),this.breakConnect()),this.config.namespace||(console.log("- 请配置项目命名空间 namespace"),this.breakConnect()),this.uploading=!0,this.getArgv("--reset")!==void 0&&(console.log("- 重置项目"),await this.onShell(`rm -r ${this.config.rootDir}/${this.config.namespace}`)),this.upLoadProject(t,`${this.config.rootDir}/${this.config.namespace}`)})}).connect({host:this.config.host,port:this.config.port,username:this.config.username,password:this.config.password}),this.ssh2Conn.on("error",e=>{console.log(`- 连接失败: ${e}`),this.breakConnect()}),this.ssh2Conn.on("end",()=>{this.uploading=!1}))}breakConnect(){this.uploading=!1,console.log("- 已断开连接"),console.log("-------------- deploy-end --------------"),this.ssh2Conn.end()}onShell(e){const t=this.getArgv("--allLog")!==void 0;return new Promise((o,r)=>{this.ssh2Conn.shell((l,i)=>{if(l){console.log("- 远程命令错误："+l,e),this.breakConnect(),r(l);return}let s=[];i.on("close",()=>{o(s.toString())}).on("data",c=>{let a=c.toString();(t||this.config.allLog||a.startsWith("- "))&&s.push(`
-`+a)}).stderr.on("data",c=>{console.log(`- 远程命令错误：
-`+c),this.breakConnect(),r(c)}),i.end(e+`
+(function(h){typeof define=="function"&&define.amd?define(h):h()})(function(){"use strict";var R=Object.defineProperty;var q=(h,u,f)=>u in h?R(h,u,{enumerable:!0,configurable:!0,writable:!0,value:f}):h[u]=f;var y=(h,u,f)=>(q(h,typeof u!="symbol"?u+"":u,f),f);const{stdout:h}=require("single-line-log"),u=require("path"),f=require("fs"),{Client:v}=require("ssh2"),b=require("axios"),L=require("https");module.exports=class{constructor(e){y(this,"config",{host:"",port:"",username:"",password:"",namespace:"",imageStore:"dev-images",tmpName:"dev",delay:0,fileDir:"",rootDir:"/home/yjweb",dockerfile:{name:"Dockerfile",content:["FROM harbor.yunjingtech.cn:30002/yj-base/nginx:latest","RUN rm -rf /usr/share/nginx/html/*","COPY dist /usr/share/nginx/html/"]},upload:{name:"upload.sh",content:["#!/bin/sh","tag=`basename \\`pwd\\``:$2-`date +%Y%m%d`",'echo "- 镜像仓库: $1"','echo "- 镜像环境: $2"',"docker build -t harbor.yunjingtech.cn:30002/$1/$tag .","docker push harbor.yunjingtech.cn:30002/$1/$tag","if [ $? -eq 0 ];then","echo - 推送成功","echo - 镜像地址: harbor.yunjingtech.cn:30002/$1/$tag","else","echo - 镜像推送失败，请重试或联系运维人员，如需查看全部docker日志，请在命令后添加 --allLog查看更多信息","fi"]},dist:{name:"dist"},parallelDir:20,parallelFile:50,allLog:!1,lazyUpload:!1,rePushNum:3,rancherConfig:{token:"",namespace:"",workload:""}});y(this,"ssh2Conn",null);y(this,"uploading",!1);y(this,"trim",null);return this.config=Object.assign(this.config,e),{name:"yj-deploy",isPut:this.isPut.bind(this),upload:this.upload.bind(this),apply:this.apply.bind(this),closeBundle:this.isPut.bind(this)}}apply(e){return e&&e.hooks&&e.hooks.done&&e.hooks.done.tap("yj-deploy",()=>{this.isPut()}),"build"}isPut(){this.getArgv("--deploy")!=null&&(clearTimeout(this.trim),this.trim=setTimeout(()=>{this.upload()},this.config.delay||0))}upload(){if(console.log("-------------- deploy-start --------------"),!this.config.host){console.log("- 请配置跳板机地址 host");return}if(!this.config.port){console.log("- 请配置跳板机端口 port");return}if(!this.config.username){console.log("- 请配置跳板机账号 username");return}if(!this.config.password){console.log("- 请配置跳板机密码 password");return}this.uploading||(this.ssh2Conn=new v,this.ssh2Conn.on("ready",()=>{console.log("- 跳板机连接成功！"),this.ssh2Conn.sftp(async(e,t)=>{e&&(console.log("- sftp连接失败"),this.breakConnect()),this.config.namespace||(console.log("- 请配置项目命名空间 namespace"),this.breakConnect()),this.uploading=!0,this.getArgv("--reset")!==void 0&&(console.log("- 重置项目"),await this.onShell(`rm -r ${this.config.rootDir}/${this.config.namespace}`)),this.upLoadProject(t,`${this.config.rootDir}/${this.config.namespace}`)})}).connect({host:this.config.host,port:this.config.port,username:this.config.username,password:this.config.password}),this.ssh2Conn.on("error",e=>{console.log(`- 连接失败: ${e}`),this.breakConnect()}),this.ssh2Conn.on("end",()=>{this.uploading=!1}))}breakConnect(){this.uploading=!1,console.log("- 已断开连接"),console.log("-------------- deploy-end --------------"),this.ssh2Conn.end()}onShell(e){const t=this.getArgv("--allLog")!==void 0;return new Promise((o,i)=>{this.ssh2Conn.shell((l,s)=>{if(l){console.log("- 远程命令错误："+l,e),this.breakConnect(),i(l);return}let n=[];s.on("close",()=>{o(n.toString())}).on("data",a=>{let c=a.toString();(t||this.config.allLog||c.startsWith("- "))&&n.push(`
+`+c)}).stderr.on("data",a=>{console.log(`- 远程命令错误：
+`+a),this.breakConnect(),i(a)}),s.end(e+`
 exit
-`)})})}projectInit(e,t){return new Promise(async(o,r)=>{e.mkdir(t,async l=>{if(l)r(l);else{console.log("- 正在创建dockerfile");const i=`${t}/${this.config.dockerfile.name}`;await this.onShell(`touch ${i}`),await this.onShell(`echo '${this.config.dockerfile.content.join(`
-`)}' > ${i}`),console.log("- 正在创建upload.sh");const s=`${t}/${this.config.upload.name}`;await this.onShell(`touch ${s}`),await this.onShell(`echo '${this.config.upload.content.join(`
-`)}' > ${s}`),o()}})})}checkRemoteFile(e,t){return new Promise(o=>{e.stat(t,r=>{o(!r)})})}upLoadProject(e,t){let o=0;typeof this.config.rePushNum=="number"&&this.config.rePushNum>0&&(o=this.config.rePushNum),e.readdir(t,async r=>{if(r){console.log("- 跳板机不存在项目, 开始创建项目"),this.projectInit(e,t).then(()=>{console.log(`- ${this.config.namespace}项目创建成功`),this.upLoadProject(e,t)}).catch(n=>{console.log(`- 创建项目失败: ${n}`),this.breakConnect()});return}const l=this.getArgv("--lazyUpload")!==void 0||this.config.lazyUpload,i=`${t}/${this.config.dist.name}`;if(l||await this.onShell(`rm -rf ${i}`),await this.onShell(`mkdir ${i}`),!this.config.fileDir){console.log("- 请配置待上传文件目录 fileDir"),this.breakConnect();return}const{localFileList:s,dirList:c}=await this.getFilesInDirectory(this.config.fileDir);if(s.length===0){console.log("- 待上传目录没有获取到文件，请检查"),this.breakConnect();return}let a=0,y=s.length;const P=n=>new Promise(async(m,d)=>{try{if(l&&await this.checkRemoteFile(e,`${i}${n.remotePath}`))return m();const g=f.createReadStream(n.localPath),C=e.createWriteStream(`${i}${n.remotePath}`);C.on("close",()=>{this.progressBar(++a,y),m()}),C.on("error",j=>{console.log(`- 文件 ${n.remotePath} 上传失败：${j}`),d(j),this.breakConnect()}),g.pipe(C)}catch(g){d(g)}}),p=(n,m)=>{let d=[];for(let g=0;g<n.length;g+=m)d.push(n.slice(g,g+m));return d},$=n=>{if(n.length===0)return Promise.resolve();const m=n.shift();return Promise.all(m.map(d=>P(d))).then(()=>$(n))},k=n=>{if(n.length===0)return Promise.resolve();let d=n.shift().map(g=>`${i}${g}`).join(" ");return this.onShell(`mkdir -p ${d}`).then(()=>k(n))};let S=[...p(c,this.config.parallelDir||20)];await k(S),console.log("- 创建目录完成"),await new Promise(n=>{setTimeout(()=>{n()},500)});let L=[...p(s,this.config.parallelFile||50)];await $(L),l&&(this.progressBar(y,y),console.log("\x1B[32m%s\x1B[0m",`- 已开启懒上传，本次共上传 ${a} 个文件`)),console.log("- 文件上传成功"),console.log("- 开始推送镜像");let b=`cd ${t} && sh ${this.config.upload.name}`;if(await this.isNewShell(e,t)){const n=this.getArgv("--imageStore")||this.config.imageStore;b+=` ${n}`}else console.log(`
-- warning 检测到当前脚本为手动创建,不支持镜像仓库配置,imageStore将失效`),console.log("- warning 如需支持imageStore参数,请在命令行后添加 --reset 重新生成脚本");const A=this.getArgv("--tmpName")||this.config.tmpName;b+=` ${A}`;const D=Array.from(new Array(o+1)).map(()=>this.onShell(b));await this.pushMirrorImage(D,D.length),this.breakConnect()})}async pushMirrorImage(e,t){if(e.length===0)return console.log("- 镜像推送失败，请添加 --allLog 查看全部日志"),null;const o=await e[0];return o.includes("- 镜像推送失败")?(console.log(o),e.length>1&&console.log(`- 正在尝试第 ${t-e.length+1} 次推送...`),this.pushMirrorImage(e.slice(1),t)):(console.log(o),console.log("- 镜像推送结束"),this.getArgv("--restart")!==void 0&&await x(o,this.config.rancherConfig),o)}async isNewShell(e,t){return new Promise((o,r)=>{e.readFile(`${t}/${this.config.upload.name}`,(l,i)=>{if(l)return r(!1);const s=i.toString();s.includes("- 镜像推送失败")||console.log(`
-- warning 检测到当前脚本可能不是最新版本，推送失败可能不会提示，且没有失败自动重试功能，请在命令后添加 --reset 重新生成脚本`),o(s.includes("$2"))})})}progressBar(e,t,o=25){let r=(e/t).toFixed(4),l=Math.floor(r*o),i="";for(let a=0;a<l;a++)i+="█";let s="";for(let a=0;a<o-l;a++)s+="░";let c=`- 文件上传进度: ${i}${s} (${e}/${t}) ${(100*r).toFixed(2)}%`;h(c),e==t&&console.log("")}getFilesInDirectory(e){const t=[],o=[];return new Promise((r,l)=>{const i=(s,c,a)=>{const y=f.readdirSync(s);let P=!1;y.forEach(p=>{const $=u.join(s,p),S=f.statSync($).isDirectory();S?P=!0:t.push({localPath:$,remotePath:`${c}${p}`,level:a}),S&&i($,`${c}${p}/`,a+1)}),P||o.push(c)};i(e,"/",0),r({localFileList:t,dirList:o})})}getArgv(e){const t=process.argv;let o;return t.forEach(r=>{r.indexOf(e)>-1&&(o=r.split("=")[1]||"")}),o}}});
+`)})})}projectInit(e,t){return new Promise(async(o,i)=>{e.mkdir(t,async l=>{if(l)i(l);else{console.log("- 正在创建dockerfile");const s=`${t}/${this.config.dockerfile.name}`;await this.onShell(`touch ${s}`),await this.onShell(`echo '${this.config.dockerfile.content.join(`
+`)}' > ${s}`),console.log("- 正在创建upload.sh");const n=`${t}/${this.config.upload.name}`;await this.onShell(`touch ${n}`),await this.onShell(`echo '${this.config.upload.content.join(`
+`)}' > ${n}`),o()}})})}checkRemoteFile(e,t){return new Promise(o=>{e.stat(t,i=>{o(!i)})})}upLoadProject(e,t){let o=0;typeof this.config.rePushNum=="number"&&this.config.rePushNum>0&&(o=this.config.rePushNum),e.readdir(t,async i=>{if(i){console.log("- 跳板机不存在项目, 开始创建项目"),this.projectInit(e,t).then(()=>{console.log(`- ${this.config.namespace}项目创建成功`),this.upLoadProject(e,t)}).catch(r=>{console.log(`- 创建项目失败: ${r}`),this.breakConnect()});return}const l=this.getArgv("--lazyUpload")!==void 0||this.config.lazyUpload,s=`${t}/${this.config.dist.name}`;if(l||await this.onShell(`rm -rf ${s}`),await this.onShell(`mkdir ${s}`),!this.config.fileDir){console.log("- 请配置待上传文件目录 fileDir"),this.breakConnect();return}const{localFileList:n,dirList:a}=await this.getFilesInDirectory(this.config.fileDir);if(n.length===0){console.log("- 待上传目录没有获取到文件，请检查"),this.breakConnect();return}let c=0,w=n.length;const k=r=>new Promise(async(p,d)=>{try{if(l&&await this.checkRemoteFile(e,`${s}${r.remotePath}`))return p();const g=f.createReadStream(r.localPath),D=e.createWriteStream(`${s}${r.remotePath}`);D.on("close",()=>{this.progressBar(++c,w),p()}),D.on("error",I=>{console.log(`- 文件 ${r.remotePath} 上传失败：${I}`),d(I),this.breakConnect()}),g.pipe(D)}catch(g){d(g)}}),m=(r,p)=>{let d=[];for(let g=0;g<r.length;g+=p)d.push(r.slice(g,g+p));return d},$=r=>{if(r.length===0)return Promise.resolve();const p=r.shift();return Promise.all(p.map(d=>k(d))).then(()=>$(r))},A=r=>{if(r.length===0)return Promise.resolve();let d=r.shift().map(g=>`${s}${g}`).join(" ");return this.onShell(`mkdir -p ${d}`).then(()=>A(r))};let P=[...m(a,this.config.parallelDir||20)];await A(P),console.log("- 创建目录完成"),await new Promise(r=>{setTimeout(()=>{r()},500)});let N=[...m(n,this.config.parallelFile||50)];await $(N),l&&(this.progressBar(w,w),console.log("\x1B[32m%s\x1B[0m",`- 已开启懒上传，本次共上传 ${c} 个文件`)),console.log("- 文件上传成功"),console.log("- 开始推送镜像");let j=`cd ${t} && sh ${this.config.upload.name}`;if(await this.isNewShell(e,t)){const r=this.getArgv("--imageStore")||this.config.imageStore;j+=` ${r}`}else console.log(`
+- warning 检测到当前脚本为手动创建,不支持镜像仓库配置,imageStore将失效`),console.log("- warning 如需支持imageStore参数,请在命令行后添加 --reset 重新生成脚本");const U=this.getArgv("--tmpName")||this.config.tmpName;j+=` ${U}`;const x=Array.from(new Array(o+1)).map(()=>this.onShell(j));await this.pushMirrorImage(x,x.length),this.breakConnect()})}async pushMirrorImage(e,t){if(e.length===0)return console.log("- 镜像推送失败，请添加 --allLog 查看全部日志"),null;const o=await e[0];return o.includes("- 镜像推送失败")?(console.log(o),e.length>1&&console.log(`- 正在尝试第 ${t-e.length+1} 次推送...`),this.pushMirrorImage(e.slice(1),t)):(console.log(o),console.log("- 镜像推送结束"),this.getArgv("--restart")!==void 0&&await F(o,this.config.rancherConfig),o)}async isNewShell(e,t){return new Promise((o,i)=>{e.readFile(`${t}/${this.config.upload.name}`,(l,s)=>{if(l)return i(!1);const n=s.toString();n.includes("- 镜像推送失败")||console.log(`
+- warning 检测到当前脚本可能不是最新版本，推送失败可能不会提示，且没有失败自动重试功能，请在命令后添加 --reset 重新生成脚本`),o(n.includes("$2"))})})}progressBar(e,t,o=25){let i=(e/t).toFixed(4),l=Math.floor(i*o),s="";for(let c=0;c<l;c++)s+="█";let n="";for(let c=0;c<o-l;c++)n+="░";let a=`- 文件上传进度: ${s}${n} (${e}/${t}) ${(100*i).toFixed(2)}%`;h(a),e==t&&console.log("")}getFilesInDirectory(e){const t=[],o=[];return new Promise((i,l)=>{const s=(n,a,c)=>{const w=f.readdirSync(n);let k=!1;w.forEach(m=>{const $=u.join(n,m),P=f.statSync($).isDirectory();P?k=!0:t.push({localPath:$,remotePath:`${a}${m}`,level:c}),P&&s($,`${a}${m}/`,c+1)}),k||o.push(a)};s(e,"/",0),i({localFileList:t,dirList:o})})}getArgv(e){const t=process.argv;let o;return t.forEach(i=>{i.indexOf(e)>-1&&(o=i.split("=")[1]||"")}),o}};const C=new L.Agent({rejectUnauthorized:!1});async function F(S,e){try{const o={...{baseUrl:"https://paas-test.ymygz.com:10443/v3",clusterId:"local:p-r7j28"},...e};if(!o.baseUrl)return console.error("- Rancher API 基础地址未配置，请检查 rancherConfig.baseUrl 是否正确");if(!o.token)return console.error("- Rancher API 访问令牌未配置，请检查 rancherConfig.token 是否正确");if(!o.clusterId)return console.error("- Rancher 集群 ID 未配置，请检查 rancherConfig.clusterId 是否正确");if(!o.namespace)return console.error("- 服务所在的命名空间未配置，请检查 rancherConfig.namespace 是否正确");if(!o.workload)return console.error("- 工作负载名称未配置，请检查 rancherConfig.workload 是否正确");if(!S)return console.error("- 内部错误");const i=S.match(/- 镜像地址:\s*(.+)/);let l="";if(i&&i[1])l=i[1].trim();else return console.error("- 未提取到镜像地址");console.log("- 开始调用 Rancher API 更新服务镜像地址");const s=`${o.baseUrl}/projects/${o.clusterId}/workloads/deployment:${o.namespace}:${o.workload}`,n=await b.get(s,{headers:{Authorization:`Bearer ${o.token}`},httpsAgent:C});if(!n.data||!n.data.containers||n.data.containers.length===0)return console.error("- 未找到工作负载的容器信息，请检查 namespace 和 workload 是否正确");const a=n.data;a.containers[0].image=l,await b.put(s,a,{headers:{Authorization:`Bearer ${o.token}`},httpsAgent:C}),console.log("- 镜像地址更新成功，正在重启服务..."),await b.post(`${s}?action=redeploy`,{},{headers:{Authorization:`Bearer ${o.token}`},httpsAgent:C}),console.log("- 服务重启成功")}catch(t){console.error("- 调用 Rancher API 失败， 请手动重启服务:",t.message)}}});

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "yj-deploy",
-  "version": "0.0.9",
+  "version": "0.0.10",
   "description": "ssh sftp",
   "scripts": {
     "build": "vite build"
@@ -21,11 +21,11 @@
   "author": "",
   "license": "ISC",
   "dependencies": {
+    "axios": "^1.13.5",
     "single-line-log": "^1.1.2",
     "ssh2": "^1.15.0"
   },
   "devDependencies": {
-    "axios": "^1.13.5",
     "vite": "4.5"
   },
   "publishConfig": {