优化打包方法

src/rest/index.js

@@ -1,16 +1,16 @@
 const {findFile} = require("../utils/fsUtils");
+const chalk = require("chalk");
+const log = console.log
 module.exports = {
   init: ins => {
     return findFile('./src/rest', file => {
       if (!/index\.js/.test(file)) {
         let rest = require(file.replace(/src[\\\/]rest/, '.'))
-        console.log(`初始化接口:${rest.action}`)
+        log(`${chalk.bgBlue.black(" REST ")} ${rest.action}`)
         if (rest.method == "post") {
           ins.post(rest.action, (req, res) => rest.execute(req, res))
         }
       }
-    }).then(() => {
-      console.log("接口初始化完毕")
     })
   }
 }

src/tools/childProcess.js

@@ -0,0 +1,28 @@
+const {exec} = require("child_process");
+
+class Process {
+  constructor(cmd) {
+    this.cmd = cmd
+    this.pid = exec(cmd, {windowsHide: true, encoding: "utf8"});
+  }
+
+  static new(cmd) {
+    return new Promise((resolve, reject) => exec(cmd, {windowsHide: true}, (err) => {
+      if (!err) {
+        resolve()
+      } else reject(err)
+    }))
+  }
+
+  ins() {
+    return new Promise((resolve, reject) => {
+      this.pid.on('close', (code, err) => {
+        if (code == 0 && !err) {
+          resolve()
+        } else reject(err)
+      })
+    })
+  }
+}
+
+module.exports = Process

src/utils/dbUitls.js

@@ -2,14 +2,14 @@ const mysql = require("mysql");
 const dbConfig = require("../config/db");
 const {v4: uuid} = require("uuid");
 const dayjs = require("dayjs");
+const chalk = require("chalk");
 const query = sql => new Promise((resolve, reject) => {
   this.pool?.getConnection((err, conn) => {
     if (err) {
-      console.log(err)
+      reject(err)
     } else {
       conn.query(sql, (err, result) => {
         if (err) {
-          console.log(err)
           reject(err)
         } else {
           conn.release()
@@ -23,6 +23,7 @@ module.exports = {
   pool: null,
   init: () => {
     this.pool = mysql.createPool(dbConfig)
+    console.log(`${chalk.bgBlue.black(" DATABASE ")} 数据库已连接`)
   },
   query,
   list: ({table, search, con = ''}) => {

src/websocket/custom/getZip.js

@@ -0,0 +1,60 @@
+const db = require("../../utils/dbUitls");
+const execute = require("../../tools/exec")
+const process = require("../../tools/childProcess")
+const dayjs = require("dayjs")
+const fse = require("fs-extra");
+
+const handleZip = (id, ws) => {
+  const uniCon = `id='${id}'`
+  const sendMessage = data => ws.send(JSON.stringify(data))
+  db.detail({table: "node_custom_config", id}).then(info => {
+    if (info?.id) {
+      sendMessage({code: 0, progress: 1, data: `正在处理 ${info.name} 的打包工作...`})
+      const buildPath = {
+        web: 'base-web',
+        wxwork: 'base-wxcp',
+        mp: 'dvcp_v2_wxmp'
+      }[info.type] || {}
+      let path = `../${buildPath}`, {dist} = info, pid
+      dist = dist || `../zips/${info.name}v${info.version || "1.0.0"}`
+      Promise.all([
+        db.query(`update node_custom_config set download=null,error=null,zipTime='${dayjs().format("YYYY-MM-DD HH:mm:ss")}' where ${uniCon}`),
+        process.new(`./shell/update.sh ${info.name} ${path}`)])
+      .then(() => sendMessage({code: 0, progress: 30}))
+      .then(() => pid = new process(`cd ${path}&&npm run apps&&node bin/pages.js ${id}&&npm run build`)).then(() => pid.ins())
+      .then(() => fse.emptyDir(dist))
+      .then(() => sendMessage({code: 0, progress: 70}))
+      .then(() => pid = new process(`./shell/move.sh ${info.type} ${path} ${dist}`)).then(() => pid.ins())
+      .then(() => sendMessage({code: 0, progress: 90}))
+      .then(() => db.query(`update node_custom_config set download='${dayjs().format("YYYY-MM-DD HH:mm:ss")}',error='' where ${uniCon}`))
+      .then(() => db.detail({table: "node_custom_config", id}))
+      .then(row => sendMessage({code: 0, progress: 100, row}))
+      .catch(err => {
+        console.log(err)
+        const msg = `执行失败:${err.cmd}`
+        return db.query(`update node_custom_config set error='${msg}',zipTime=null where ${uniCon}`)
+        .then(() => db.detail({table: "node_custom_config", id}))
+        .then(row => sendMessage({code: 1, row}))
+        .catch(() => 0)
+      })
+    }
+  }).catch(err => {
+    console.log(err)
+  })
+}
+
+module.exports = {
+  action: "/custom/getZip",
+  execute: (ws, request) => {
+    const {id} = request.query
+    handleZip(id, ws)
+    ws.onmessage = res => {
+      if (res?.data) {
+        const data = JSON.parse(res.data)
+        if (data.cid == id) {
+
+        } else if (id == data.id) handleZip(id, ws)
+      }
+    }
+  }
+}

src/websocket/index.js

@@ -0,0 +1,14 @@
+const {findFile} = require("../utils/fsUtils");
+const chalk = require("chalk");
+const {log} = console
+module.exports = {
+  init: ins => {
+    return findFile('./src/websocket', file => {
+      if (!/index\.js/.test(file)) {
+        const ws = require(file.replace(/src[\\\/]websocket/, '.'))
+        log(`${chalk.bgBlue.black(" WEBSOCKET ")} ${ws.action}`)
+        ins.ws(ws.action, ws.execute)
+      }
+    })
+  }
+}