@ghini/kit/src/lib/http/router.js

js practical tools to assist efficient development

export { router_find_resolve, addr, _404 };
import { rf } from "../index.js";
import { hd_default } from "./routes.js";
function addr(...argv) {
  let path,
    method,
    ct,
    fn_end,
    fn_data,
    config = {};
  if (typeof argv[0] === "string") {
    const arr = argv[0].split(" ");
    if (arr.length > 1) {
      arr.forEach((item) => {
        if (item.startsWith("/")) {
          path = item;
        } else if (item.includes("/")) {
          ct = item;
        } else {
          method = item.toUpperCase();
        }
      });
    }
  }
  argv.forEach((item) => {
    if (typeof item === "string") {
      if (!path && item.startsWith("/")) path = item;
      else if (!ct && !item.match(" ") && item.includes("/")) ct = item;
      else if (!method) method = item.toUpperCase();
    } else if (item instanceof RegExp) path = item;
    else if (typeof item === "function") {
      if (!fn_end) fn_end = item;
      else fn_data = item;
    } else if (typeof item === "object") config = item;
  });
  if (!path) {
    console.error("path is required,以'/'开头的精确路径string 或 regexp");
    return;
  }
  if (!method) method = "*";
  if (!ct) ct = "*";
  if (!fn_end) fn_end = hd_default;
  const index = this.routes.findIndex((row) => {
    return (
      row[0].toString() === path.toString() &&
      row[1] === method &&
      row[2] === ct
    );
  });
  if (index > -1) {
    this.routes[index][3] = fn_end;
    this.routes[index][4] = fn_data;
    this.routes[index][5] = config;
    return;
  }
  this.routes.push([path, method, ct, fn_end, fn_data, config]);
}
function router_find_resolve(server, stream, gold) {
  server.router_begin?.(server, gold);
  if (server.open === 1) {
    const privateIPs = [
      /^10\./, 
      /^172\.(1[6-9]|2[0-9]|3[0-1])\./, 
      /^192\.168\./, 
      /^fc00::/, 
      /^fd/, 
      /^127\./, 
      /^::1$/, 
    ];
    const isPrivateIP = privateIPs.some((pattern) =>
      pattern.test(gold.direct_ip)
    );
    if (!isPrivateIP) {
      server._404?.(gold);
      return;
    }
  } else if (server.open === 2) {
  } else {
    if (
      gold.direct_ip !== "127.0.0.1" &&
      gold.direct_ip !== "::1" &&
      gold.direct_ip !== "::ffff:127.0.0.1"
    ) {
      server._404?.(gold);
      return;
    }
  }
  let arr,
    arr0 = [],
    arr1 = [];
  server.routes.forEach((row) => {
    if (gold.path === row[0]) {
      arr0.push(row);
    } else if (
      arr0.length === 0 &&
      row[0] instanceof RegExp &&
      row[0].test(gold.path)
    ) {
      arr1.push(row);
    }
  });
  if (arr0.length > 0) {
    arr = arr0;
  } else if (arr1.length > 0) {
    arr = arr1;
  } else {
    if (gold.path === "/favicon.ico") {
      gold.respond({
        ":status": 200,
        "content-type": "image/x-icon",
      });
      const data = rf("../../store/favicon.png", null);
      return gold.end(data);
    }
    return server._404?.(gold);
  }
  arr0 = [];
  arr1 = [];
  arr.forEach((row) => {
    if (row[1] === gold.method) {
      arr0.push(row);
    } else if (arr0.length === 0 && row[1] === "*") {
      arr1.push(row);
    }
  });
  if (arr0.length > 0) {
    arr = arr0;
  } else if (arr1.length > 0) {
    arr = arr1;
  } else {
    server._404?.(gold);
    return;
  }
  arr0 = undefined;
  arr1 = [];
  for (const row of arr) {
    if (gold.ct?.startsWith(row[2])) {
      arr0 = row;
      break;
    } else if (row[2] === "*") {
      arr1.push(row);
    }
  }
  let router_target;
  if (arr0) {
    router_target = arr0;
  } else if (arr1) {
    router_target = arr1[0];
  } else {
    server._404?.(gold);
    return;
  }
  gold.config = { ...gold.config, ...router_target.at(-1) };
  let chunks = [],
    length = 0,
    notresponded = true,
    maxbody = gold.config.MAX_BODY || 4 * 1024 * 1024;
  stream.on("data", async (chunk) => {
    try {
      length += chunk.length;
      if (notresponded && length > maxbody) {
        notresponded = false;
        gold.jerr(
          { msg: "Payload Too Large", maxBody: `${maxbody / 1048576}MB` },
          413
        );
      }
      if (typeof router_target[4] === "function") {
        await router_target[4](gold, chunk, chunks);
      } else {
        chunks.push(chunk);
      }
    } catch (err) {
      console.error(err);
      gold.err();
    }
  });
  stream.on("end", async () => {
    try {
      gold.body = Buffer.concat(chunks).toString();
      gold.data = body2data(gold) || {};
      await router_target[3](gold);
    } catch (err) {
      console.error(err, err.stack);
      gold.jerr();
    }
  });
}
function body2data(gold) {
  let data;
  try {
    data = JSON.parse(gold.body);
  } catch {
    data = {};
    if (gold.ct === "application/x-www-form-urlencoded") {
      const params = new URLSearchParams(gold.body);
      for (const [key, value] of params) {
        data[key] = value;
      }
    } else if (gold.ct?.startsWith("multipart/form-data")) {
      const boundaryMatch = gold.ct.match(/boundary=(.+)$/);
      if (!boundaryMatch) {
        throw new Error("Boundary not found in Content-Type");
      }
      const boundary = boundaryMatch[1];
      const parts = gold.body.split(`--${boundary}`);
      for (let part of parts) {
        part = part.trim();
        if (!part || part === "--") continue; 
        const [rawHeaders, ...rest] = part.split("\r\n\r\n");
        const content = rest.join("\r\n\r\n").replace(/\r\n$/, "");
        const headers = rawHeaders.split("\r\n");
        let name = null;
        let filename = null;
        let contentType = null;
        headers.forEach((header) => {
          const nameMatch = header.match(/name="([^"]+)"/);
          if (nameMatch) {
            name = nameMatch[1];
          }
          const filenameMatch = header.match(/filename="([^"]+)"/);
          if (filenameMatch) {
            filename = filenameMatch[1];
          }
          const ctMatch = header.match(/Content-Type:\s*(.+)/i);
          if (ctMatch) {
            contentType = ctMatch[1];
          }
        });
        if (!name) continue; 
        if (filename) {
          const fileObj = {
            filename: filename,
            content: content,
            contentType: contentType || "application/octet-stream", 
          };
          if (data[name]) {
            if (Array.isArray(data[name])) {
              data[name].push(fileObj);
            } else {
              data[name] = [data[name], fileObj];
            }
          } else {
            data[name] = fileObj;
          }
        } else {
          if (data[name] !== undefined) {
            if (Array.isArray(data[name])) {
              data[name].push(content);
            } else {
              data[name] = [data[name], content];
            }
          } else {
            data[name] = content;
          }
        }
      }
      for (const key in data) {
        if (Array.isArray(data[key]) && data[key].length === 1) {
          data[key] = data[key][0];
        }
      }
    }
  }
  return data;
}
function _404(gold) {
  gold.err();
}