@ghini/kit/dev/lib/http/hs.js

js practical tools to assist efficient development

export { h2s, hs, hss };

import { gcatch, rf, xpath, style, metaroot } from "../index.js";
import http2 from "http2";
import https from "https";
import http from "http";
import EventEmitter from "events";
import { hd_stream } from "./gold.js";
import { addr, _404 } from "./router.js";
import { fn_static } from "./static.js";
/*
 * hs定位:业务核心服务器,及h2测试服务器,生产环境中主要反代使用
 * 安防交给nginx cf等网关
 */
/**
 * @typedef {Object} ServerExtension
 * @property {string} ip
 * @property {number} open
 * @property {any[]} routes
 * @property {Function} addr
 * @property {Function} static
 * @property {Function} _404
 * @property {Function} router_begin
 * @property {number} cnn
 */

/**
 * @param {...any} argv
 * @returns {Promise<import('http').Server & ServerExtension>}
 */
async function hs(...argv) {
  return new Promise((resolve, reject) => {
    let { port, config } = argv_port_config(argv),
      server,
      scheme,
      open = 0,
      protocol = "http/1.1",
      currentConnections = 0; //记录当前连接数
    if (config?.key) {
      if (config.hasOwnProperty("allowHTTP1")) {
        server = http2.createSecureServer(config);
        if (config.allowHTTP1) protocol = "h2,http/1.1";
        else protocol = "h2";
      } else server = https.createServer(config);
      scheme = "https";
      open = 2;
    } else {
      server = http.createServer({ insecureHTTPParser: false });
      scheme = "http";
    }
    server.listen(port, () => {
      console.info.bind({ xinfo: 2 })(
        `${style.reset}${style.bold}${style.brightGreen} ✓ ${
          style.brightWhite
        }Running on ${
          style.underline
        }${scheme}://${"127.0.0.1"}:${port}${style.reset}  open:${open}`
      );
      gcatch();
      server.port = port;
      if (config?.key) {
        server.on("stream", (stream, headers) => {
          stream.protocol = "h2";
          hd_stream(server, stream, headers);
        });
      }
      return resolve(server);
    });
    // 监听tcp connection✅  session(❌不够底层,防不了tcp级别ddos)
    // 先不管理连接数,在此作用不大,徒增开销
    // server.on("connection", (socket) => {
    //   currentConnections++; // 增加连接数
    //   // console.dev(`当前连接数：${currentConnections}`);
    //   socket.on("close", () => {
    //     currentConnections--; // 减少连接数
    //   });
    // });
    server.on("request", (req, res) => {
      if (req.headers[":path"]) return;
      req.scheme = scheme;
      let { stream, headers } = simulateHttp2Stream(req, res);
      hd_stream(server, stream, headers);
    });
    server.on("error", (err) => {
      if (err.code === "EADDRINUSE" && port < 65535) {
        console.warn.bind({ xinfo: 2 })(
          `${style.bold}${style.yellow} ⚠ ${style.dim}${
            style.brightMagenta
          }Port ${port} is in use, trying ${port + 1} instead...${style.reset}`
        );
        port++;
        server.listen(port);
      } else {
        console.error(`Server error: ${err.message}`);
        return reject(err);
      }
    });
    console.info.bind({ xinfo: 2 })(
      `🚀 Start [${protocol}] ${scheme} server...`
    );
    // 虽然不赋值server也进行了修改,但ide跟踪不到,所以这里赋值一下
    server = Object.assign(server, {
      open, //开放级别 0本地 1局域网 2公网
      routes: [],
      addr,
      static: fn_static,
      _404,
      router_begin: (server, gold) => {},
      cnn: 0,
    });
    Object.defineProperties(server, {
      routes: { writable: false, configurable: false },
      addr: { writable: false, configurable: false },
      cnn: {
        get: () => currentConnections,
        enumerable: true,
      },
      _404: {
        get: () => server.routes.__404 || _404,
        set: (v) => {
          server.routes.__404 = typeof v === "function" ? v : () => {};
        },
        enumerable: true,
      },
    });
  });
}
/**
 * h2s 默认创建 h2tls 兼容 http1.1 tls 的服务器,也可通过配置仅创建h2server
 * @param  {...any} argv
 * @returns
 */
async function h2s(...argv) {
  let { port, config } = argv_port_config(argv);
  config = {
    ...{
      key: rf(xpath("store/cert/selfsigned.key", metaroot)),
      cert: rf(xpath("store/cert/selfsigned.cert", metaroot)),
      allowHTTP1: true,
      // settings: {
      //   maxConcurrentStreams: 100, // 最大并发流数
      //   maxHeaderListSize: 32 * 1024, // 最大请求头大小 (32KB)
      // },
    },
    ...config,
  };
  return hs(port, config);
}
/**
 * hss 创建http1.1 tls server
 * @param  {...any} argv
 * @returns
 */
async function hss(...argv) {
  // 启动一个 HTTPS 服务器，使用指定的证书和密钥文件
  let { port, config } = argv_port_config(argv);
  config = {
    ...{
      key: rf(xpath("store/cert/selfsigned.key", metaroot)),
      cert: rf(xpath("store/cert/selfsigned.cert", metaroot)),
    },
    config,
  };
  return hs(port, config);
}
/**
 * argv_port_config 动态分析argv返回端口和配置
 * @param {*} argv
 * @returns
 */
function argv_port_config(argv) {
  let port, config;
  argv.forEach((item) => {
    // 简单判断,写错了走报错去
    if (typeof item === "object") {
      config = item;
    } else {
      port = item;
    }
  });
  port = port || 3000;
  return { port, config };
}
// 模拟 HTTP/2 的 `stream` 对象
function simulateHttp2Stream(req, res) {
  const headers = { ...req.headers };
  headers[":method"] = req.method;
  headers[":path"] = req.url;
  headers[":scheme"] = req.scheme;
  headers[":authority"] = req.headers.host || "";
  const stream = new EventEmitter(); // 添加事件发射器功能
  stream.protocol = "HTTP/" + req.httpVersion;
  stream.ip = req.socket.remoteAddress;
  stream.respond = (responseHeaders) => {
    const status = responseHeaders[":status"] || 200; // 默认状态码 200
    const filteredHeaders = Object.fromEntries(
      Object.entries(responseHeaders).filter(([key]) => !key.startsWith(":"))
    );
    res.writeHead(status, filteredHeaders);
  };
  stream.write = res.write.bind(res);
  stream.end = res.end.bind(res);
  // 将 req 的数据事件转发到 stream 上
  req.on("data", (chunk) => stream.emit("data", chunk));
  req.on("end", () => stream.emit("end"));
  req.on("error", (err) => stream.emit("error", err));
  return { stream, headers };
}