asai-nodejs-host 0.1.0

package/AsaiHost.ts

@@ -0,0 +1,522 @@
+module.exports = ($asai: any, opt: any) => {
+  const { sysWork, apiWork, wsWork } = opt;
+  const fs: any = require("fs");
+  const path: any = require("path");
+  const url: any = require("url");
+
+  // 读取配置文件
+  if (!$asai.hostconfig?.ip) {
+    $asai.hostconfig.ip = getIp();
+  }
+  // 通过环境变量控制日志输出
+  // if (process.env.NODE_ENV === 'production') {
+  //     console.log = () => {}; // 关闭调试日志
+  // }
+  console.log(`※※※※※※※※※※※※※※※※※※※※※※※※※※※※※※※※※※※※※※※※※※※※※※※※`);
+  // 获取http或https的服务
+  // const http: any = require('http2');
+  const http: any = require("http");
+  const https: any = require("https");
+  // 获取当前时间
+  function getNowTime() {
+    const datanow: any = new Date();
+    return (
+      datanow.getMonth() +
+      1 +
+      "-" +
+      datanow.getDate() +
+      " " +
+      datanow.getHours() +
+      ":" +
+      datanow.getMinutes() +
+      ":" +
+      datanow.getSeconds() +
+      "." +
+      datanow.getMilliseconds()
+    );
+  }
+  // 获取本机IP地址
+  function getIp() {
+    const ifaces: any = require("os").networkInterfaces();
+    for (const el in ifaces) {
+      const iface: any = ifaces[el];
+      if (iface && iface.length) {
+        for (let i: number = 0; i < iface.length; i++) {
+          const elc: any = iface[i];
+          if (
+            elc.family === "IPv4" &&
+            elc.address !== "127.0.0.1" &&
+            !elc.internal
+          ) {
+            return elc.address;
+          }
+        }
+      }
+    }
+    return "";
+  }
+
+  function getHTTPServer() {
+    try {
+      if ($asai.hostconfig.httptype.includes("https")) {
+        https.AsCreateServer = (callback: any) => {
+          return https.createServer($asai.hostconfig.httpscert, callback);
+        };
+        return https;
+      } else {
+        http.AsCreateServer = http.createServer;
+        return http;
+      }
+    } catch (err: any) {
+      console.log(666.907, err);
+      http.AsCreateServer = http.createServer;
+      return http;
+    }
+  }
+  function getWsUserinfo(req: any) {
+    try {
+      const headers = Object.fromEntries(
+        Array.from({ length: req.rawHeaders.length / 2 }, (_, i) => [
+          req.rawHeaders[i * 2].toLowerCase(),
+          req.rawHeaders[i * 2 + 1],
+        ])
+      );
+      const userinfo = headers["sec-websocket-protocol"]?.split("##");
+      return deUserInfo(userinfo);
+    } catch (err) {}
+    return null;
+  }
+  // Ws请求头校验
+  function checkWsHeader(req: any) {
+    req.Userinfo = getWsUserinfo(req);
+    console.log(666.2001, "WS", req.Userinfo[0]);
+    return true;
+  }
+  function getHttpUserinfo(req: any) {
+    try {
+      const headers = Object.fromEntries(
+        Array.from({ length: req.rawHeaders.length / 2 }, (_, i) => [
+          req.rawHeaders[i * 2].toLowerCase(),
+          req.rawHeaders[i * 2 + 1],
+        ])
+      );
+      const userinfo = headers["userinfo"]?.split("##");
+      return deUserInfo(userinfo);
+    } catch (err) {}
+    return null;
+  }
+  // tcp请求头校验
+  function checkHttpHeader(req: any) {
+    req.Userinfo = getHttpUserinfo(req);
+    return true;
+  }
+  function deUserInfo(userinfo: any) {
+    if (userinfo[2]) {
+      userinfo[2] = $asai.$lib.AsCode.asdecode(
+        userinfo[2],
+        $asai.hostconfig.asaisn
+      );
+      userinfo[0] = $asai.$lib.AsCode.asdecode(userinfo[0], userinfo[2]);
+      userinfo[1] = $asai.$lib.AsCode.asdecode(userinfo[1], userinfo[2]);
+    }
+    return userinfo;
+  }
+  // 代理服务器
+  if ($asai.hostconfig?.proxyhosts) {
+    // 循环启动配置中的服务
+    Object.keys($asai.hostconfig.proxyhosts).forEach((el) => {
+      startProxyServer(el);
+    });
+    // 启动代理服务器
+    function startProxyServer(proxyhost: string) {
+      const proxycfg = $asai.hostconfig.proxyhosts[proxyhost];
+      if (!proxycfg.url && proxycfg.port) {
+        proxycfg.url =
+          $asai.hostconfig.httptype + $asai.hostconfig.ip + ":" + proxycfg.port;
+      }
+      if (proxycfg.url) {
+        const http = getHTTPServer();
+        const proxyServer = http.AsCreateServer((req: any, res: any) => {
+          // 启用Keep-Alive复用连接
+          const agent = new http.Agent({ keepAlive: true });
+          const forwardRequest = http.request(
+            {
+              agent: agent, // 复用连接池
+              method: req.method,
+              ...url.parse(proxycfg.url + req.url),
+            },
+            (forwardRes: any) => {
+              // 设置响应头（可以选择性地转发原始响应头）
+              Object.keys(forwardRes.headers).forEach((key) => {
+                res.setHeader(key, forwardRes.headers[key]);
+              });
+              res.statusCode = forwardRes.statusCode;
+              console.log(666.001, getHttpUserinfo(req));
+              res.setHeader("Userinfo", getHttpUserinfo(req));
+              // 转发响应数据
+              req.pipe(forwardRequest); // 自动处理背压
+              forwardRes.pipe(res);
+              // forwardRes.pipe(res, { end: true }); // 手动数据传递
+            }
+          );
+
+          // 处理转发请求的错误
+          forwardRequest.on("error", (err: any) => {
+            console.error("Error during forward request:", err);
+            res.writeHead(502); // Bad Gateway
+            res.end("Bad Gateway");
+          });
+          // 转发请求数据
+          req.pipe(forwardRequest, { end: true });
+
+          // 请求超时
+          forwardRequest.setTimeout(50000, () => {
+            forwardRequest.destroy();
+            res.writeHead(504);
+            res.end("Gateway Timeout");
+          });
+        });
+        if (proxycfg.proxyip) {
+          proxyServer.listen(
+            proxycfg.proxyport || $asai.hostconfig.proxyport,
+            proxycfg.proxyip,
+            () => {
+              console.log(`[${proxyhost}] Start At ${getNowTime()}.`);
+              console.log(
+                `\x1B[45;28mProxy Server ${$asai.hostconfig.httptype}${
+                  proxycfg.proxyip
+                }:${proxycfg.proxyport || $asai.hostconfig.proxyport}\x1B[0m`
+              );
+              console.log(`================================================`);
+            }
+          );
+        } else {
+          proxyServer.listen(
+            proxycfg.proxyport || $asai.hostconfig.proxyport,
+            () => {
+              console.log(`[${proxyhost}] Start At ${getNowTime()}.`);
+              console.log(
+                `\x1B[45;28mProxy server ${
+                  $asai.hostconfig.httptype
+                }localhost:${
+                  proxycfg.proxyport || $asai.hostconfig.proxyport
+                }\x1B[0m`
+              );
+              console.log(`================================================`);
+            }
+          );
+        }
+      }
+    }
+  }
+
+  // 网页与文件服务器
+  if ($asai.hostconfig?.hosts) {
+    $asai.startHTTPServer = startHTTPServer;
+    // 循环启动配置中的服务
+    Object.keys($asai.hostconfig.hosts).forEach((el) => {
+      startHTTPServer(el);
+    });
+
+    // 创建WS服务
+    function createWSHost(hostdir: string | number) {
+      try {
+        if (!$asai.serversws[hostdir]) {
+          const { WebSocket, WebSocketServer } = require("ws");
+          const wsopt = {};
+          // ws的配置项
+          $asai.serversws[hostdir] = new WebSocketServer({
+            server: $asai.servershttp[hostdir],
+            // 附加配置信息
+            perMessageDeflate: false, // 关闭压缩减少CPU开销
+            // perMessageDeflate: {
+            //     zlibDeflateOptions: {
+            //         // See zlib defaults.
+            //         chunkSize: 1024,
+            //         memLevel: 7,
+            //         level: 3,
+            //     },
+            //     zlibInflateOptions: {
+            //         chunkSize: 10 * 1024,
+            //     },
+            //     // Other options settable:
+            //     clientNoContextTakeover: true, // Defaults to negotiated value.
+            //     serverNoContextTakeover: true, // Defaults to negotiated value.
+            //     serverMaxWindowBits: 10, // Defaults to negotiated value.
+            //     // Below options specified as default values.
+            //     concurrencyLimit: 10, // Limits zlib concurrency for perf.
+            //     threshold: 1024, // Size (in bytes) below which messages
+            //     // should not be compressed if context takeover is disabled.
+            // },
+          });
+          // 服务器ws服务关闭操作，清空定时器
+          function wsServerClose() {
+            if (
+              $asai.tasksws[hostdir] &&
+              typeof $asai.tasksws[hostdir] === "object"
+            ) {
+              Object.values($asai.tasksws[hostdir]).forEach((el: any) => {
+                el && clearInterval(el);
+              });
+              $asai.tasksws[hostdir] = {};
+            }
+          }
+
+          // 服务端异常错误关闭
+          $asai.serversws[hostdir].on("error", () => {
+            wsServerClose();
+            $asai.hostconfig.log && console.log(666.303, "WS    Server error!");
+          });
+          // 服务端异常关闭
+          $asai.serversws[hostdir].on("close", () => {
+            wsServerClose();
+            $asai.hostconfig.log && console.log(666.301, "WS    Server close!");
+          });
+
+          if ($asai.hostconfig.status.connectionsws) {
+            // 初始化记录ws服务连接
+            if (!$asai.connectionsws[hostdir]) {
+              $asai.connectionsws[hostdir] = new WeakMap();
+            }
+          }
+          // 服务被连接后...
+          $asai.serversws[hostdir].on("connection", (ws: any, req: any) => {
+            if (!$asai.tasksws[hostdir]) {
+              $asai.tasksws[hostdir] = {};
+            }
+            if (!ws?.tasksws) {
+              ws.tasksws = {};
+            }
+            if (checkWsHeader(req)) {
+              $asai.hostconfig.log &&
+                console.log(666.2002, req.Userinfo, "wsClient connection!");
+              function closeWs() {
+                ws.close();
+              }
+              if (
+                $asai.hostconfig.status.connectionsws &&
+                (!req.Userinfo[0] ||
+                  !req.Userinfo[2] ||
+                  $asai.connectionsws[hostdir][req.Userinfo[0]])
+              ) {
+                // 连接用户信息不符合条件，已存在同账户的登录
+                closeWs();
+              } else {
+                $asai.hostconfig.log && console.log(666.2006, "ok!");
+                // 正常登录
+                const [us, lv, tk] = req.Userinfo || [];
+                ws.Userinfo = { us, lv, tk }; // 赋值给ws
+                if ($asai.hostconfig.status.connectionsws) {
+                  $asai.connectionsws[hostdir][req.Userinfo[0]] = ws;
+                }
+                // 发送消息
+                $asai.serversws[hostdir].sendws = (
+                  msg: any,
+                  type: number,
+                  curws: any
+                ) => {
+                  if (type) {
+                    // 广播消息
+                    // wsopt.ulen = 0;
+                    $asai.serversws[hostdir].clients.forEach((client: any) => {
+                      // wsopt.ulen++;
+                      if (
+                        (type == 2 || client !== curws) &&
+                        client.readyState == WebSocket.OPEN
+                      ) {
+                        client.send(msg);
+                      }
+                      console.info(666.801, client.Userinfo);
+                    });
+                    // $asai.hostconfig.log &&
+                    //   console.log(666.808, "wsClient Public!", wsopt.ulen);
+                  } else {
+                    // api方式回复消息
+                    curws.send(msg);
+                  }
+                };
+
+                //监听客户端发来的消息
+                ws.on("message", (msg: any) => {
+                  wsWork(msg, ws, hostdir, req);
+                });
+                // 客户端关闭
+                ws.on("close", (err: any) => {
+                  $asai.hostconfig.log &&
+                    console.log(
+                      666.103,
+                      req.Userinfo[0] + " wsClient close!",
+                      err
+                    );
+                  if ($asai.hostconfig.status.connectionsws) {
+                    // 关闭点对点客户端的任务
+                    if (ws.tasksws && typeof ws.tasksws === "object") {
+                      Object.values(ws.tasksws).forEach((el: any) => {
+                        el && clearInterval(el);
+                      });
+                      ws.tasksws = null;
+                    }
+                    if (
+                      req.Userinfo[0] &&
+                      $asai.connectionsws[hostdir][req.Userinfo[0]]
+                    ) {
+                      delete $asai.connectionsws[hostdir][req.Userinfo[0]];
+                    }
+                  }
+                });
+              }
+            }
+          });
+        }
+      } catch (err) {}
+      return $asai.serversws[hostdir];
+    }
+
+    // 请求的文件相对路径
+    function getFilePath(hostdir: string, requrl: string) {
+      let filePath = "" + requrl;
+      if (filePath == "/" || filePath == "") {
+        filePath = "/index.html";
+      }
+      if (hostdir == "home") {
+        filePath = "." + filePath;
+      } else {
+        filePath = "./" + hostdir + filePath;
+      }
+      return filePath;
+    }
+    // 读取服务器上文件
+    function fsReadFile(hostdirs: any, requrl: any) {
+      return new Promise((resolve, reject) => {
+        // 尝试顺序读取多个路径
+        const tryRead = async (index = 0) => {
+          if (index >= hostdirs.length) return reject(404);
+          const filePath = getFilePath(hostdirs[index], requrl);
+          fs.readFile(filePath, (err: any, data: any) => {
+            if (err) tryRead(index + 1);
+            else resolve({ resdata: data, filepath: filePath });
+          });
+        };
+        tryRead();
+      });
+    }
+    // 启动HTTP服务
+    function startHTTPServer(hostdir: string) {
+      // 初始化服务配置
+      const hostcfg = $asai.hostconfig.hosts[hostdir];
+      try {
+        if (!$asai.servershttp[hostdir]) {
+          if (hostcfg.mimetypes) {
+            Object.assign(hostcfg.mimetypes, $asai.hostconfig.mimetypes);
+          } else {
+            hostcfg.mimetypes = $asai.hostconfig.mimetypes;
+          }
+          // 创建服务
+          const http = getHTTPServer();
+          $asai.servershttp[hostdir] = http.AsCreateServer(
+            (req: any, res: any) => {
+              if (checkHttpHeader(req)) {
+                if (req.url.startsWith("/sys")) {
+                  sysWork(req, res, hostdir);
+                } else if (req.url.startsWith("/api")) {
+                  apiWork(req, res, hostdir);
+                } else if (hostcfg.hostdirs) {
+                  fsReadFile(hostcfg.hostdirs, req.url)
+                    .then((resfile: any) => {
+                      res.writeHead(200, {
+                        "Content-Encoding": "gzip",
+                        "Content-Type":
+                          hostcfg.mimetypes[path.extname(resfile.filepath)] ||
+                          "application/octet-stream",
+                        "Cache-Control": "public, max-age=3600", // 客户端缓存1小时
+                      });
+                      // 提前处理压缩数据为.gz
+                      fs.createReadStream("file.gz").pipe(res);
+                      res.end(resfile.resdata);
+                    })
+                    .catch((err) => {
+                      if (err === 404) {
+                        res.writeHead(404, {
+                          "Content-Type": "text/plain; charset=utf-8",
+                        });
+                        res.end(`File not found!`);
+                      } else {
+                        res.writeHead(500, {
+                          "Content-Type": "text/plain; charset=utf-8",
+                        });
+                        res.end(`${err} error!`);
+                      }
+                    });
+                } else {
+                  res.writeHead(404, {
+                    "Content-Type": "text/plain; charset=utf-8",
+                  });
+                  res.end(`404`);
+                }
+              } else {
+                res.writeHead(502, {
+                  "Content-Type": "text/plain; charset=utf-8",
+                });
+                res.end(`502`);
+              }
+            }
+          );
+          hostcfg.ws && createWSHost(hostdir); // 创建ws服务
+          // 启动服务
+          if (hostcfg.ip || $asai.hostconfig.ip) {
+            $asai.servershttp[hostdir].listen(
+              hostcfg.port,
+              hostcfg.ip || $asai.hostconfig.ip,
+              () => {
+                console.log(`[${hostdir}] Start At ${getNowTime()}.`);
+                hostcfg.ws &&
+                  console.log(
+                    `\x1B[44;28mWS    Server ${
+                      $asai.hostconfig.wstype +
+                      (hostcfg.ip || $asai.hostconfig.ip)
+                    }:${hostcfg.port}\x1B[0m`
+                  );
+                console.log(
+                  `\x1B[42;28mHTTP  Server ${
+                    $asai.hostconfig.httptype +
+                    (hostcfg.ip || $asai.hostconfig.ip)
+                  }:${hostcfg.port}\x1B[0m`
+                );
+                console.log(`================================================`);
+              }
+            );
+          } else {
+            $asai.servershttp[hostdir].listen(hostcfg.port, () => {
+              console.log(`[${hostdir}] Start At ${getNowTime()}.`);
+              hostcfg.ws &&
+                console.log(
+                  `\x1B[41;28mWS    Server${$asai.hostconfig.wstype}localhost:${hostcfg.port}\x1B[0m`
+                );
+              console.log(
+                `\x1B[42;28mHTTP  Server ${$asai.hostconfig.httptype}localhost:${hostcfg.port}\x1B[0m`
+              );
+              console.log(`================================================`);
+            });
+          }
+        }
+      } catch (err) {
+        console.log(666.303, err);
+      }
+
+      if ($asai.hostconfig.status.connectionshttp) {
+        // 记录http服务连接（先删除连接用户，再操作node服务，成功几率更大，但记录用户会造成变量变大）
+        $asai.servershttp[hostdir].on("connection", (connection: any) => {
+          if (!$asai.connectionshttp[hostdir]) {
+            $asai.connectionshttp[hostdir] = new Set();
+          }
+          $asai.connectionshttp[hostdir].add(connection);
+          // 处理关闭服务
+          connection.on("close", () => {
+            $asai.connectionshttp[hostdir].delete(connection);
+          });
+        });
+      }
+      return $asai.servershttp[hostdir];
+    }
+  }
+};