@8-/gemini-web-api 1.0.0 → 1.0.1

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@8-/gemini-web-api",
-  "version": "1.0.0",
+  "version": "1.0.1",
   "description": "OpenAI-compatible Gemini Web API reverse proxy running seamlessly on both Bun and Node.js with zero external dependencies.",
   "type": "module",
   "main": "index.js",
@@ -9,7 +9,9 @@
   },
   "scripts": {
     "start": "bun index.js",
-    "start:node": "node index.js"
+    "start:node": "node index.js",
+    "test": "node --test",
+    "test:bun": "bun test"
   },
   "repository": {
     "type": "git",

package/src/constant.js

@@ -0,0 +1,89 @@
+import { homedir } from "node:os";
+import { join } from "node:path";
+
+export const HOST = process.env.HOST ?? "0.0.0.0",
+  PORT = parseInt(process.env.PORT ?? "7860", 10),
+  API_KEY = process.env.API_KEY ?? "",
+  ENABLE_THINKING = process.env.ENABLE_THINKING === "true",
+  INIT_URL = "https://gemini.google.com/app",
+  GEN_URL =
+    "https://gemini.google.com/_/BardChatUi/data/assistant.lamda.BardFrontendService/StreamGenerate",
+  BATCH_URL = "https://gemini.google.com/_/BardChatUi/data/batchexecute",
+  USER_AGENT =
+    "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/134.0.0.0 Safari/537.36",
+  SALT = new TextEncoder().encode("saltysalt"),
+  IV = new Uint8Array(16).fill(32),
+  CORS_HEADERS = {
+    "Access-Control-Allow-Origin": "*",
+    "Access-Control-Allow-Methods": "GET, POST, OPTIONS",
+    "Access-Control-Allow-Headers": "*",
+  },
+  STATUS_OK = 200,
+  STATUS_NO_CONTENT = 204,
+  STATUS_BAD_REQUEST = 400,
+  STATUS_UNAUTHORIZED = 401,
+  STATUS_NOT_FOUND = 404,
+  STATUS_SERVER_ERR = 500,
+  ERR_MISSING_URL = 4001,
+  ERR_UNAUTHORIZED = 4011,
+  ERR_NOT_FOUND = 4041,
+  THINKING_LEVEL_DISABLED = 1,
+  THINKING_LEVEL_ENABLED = 2,
+  ROLE_ASSISTANT = "assistant",
+  ROLE_USER = "user",
+  ROLE_SYSTEM = "system",
+  ROLE_TOOL = "tool",
+  STOP_REASON_STOP = "stop",
+  STOP_REASON_TOOL_CALLS = "tool_calls",
+  TOOL_FENCE_OPEN = "```tool_call",
+  TOOL_FENCE_CLOSE = "```",
+  PAYLOAD_INNER_REQ_IDX = 0,
+  PAYLOAD_MODEL_HEADER_IDX = 1,
+  PAYLOAD_UUID_IDX = 2,
+  BROWSER_PATH_LI = [
+    {
+      db_path: join(homedir(), "Library/Application Support/Google/Chrome/Default/Cookies"),
+      account: "Chrome",
+      service: "Chrome Safe Storage",
+    },
+    {
+      db_path: join(homedir(), "Library/Application Support/Google/Chrome/Profile 1/Cookies"),
+      account: "Chrome",
+      service: "Chrome Safe Storage",
+    },
+    {
+      db_path: join(
+        homedir(),
+        "Library/Application Support/BraveSoftware/Brave-Browser/Default/Cookies",
+      ),
+      account: "Brave",
+      service: "Brave Safe Storage",
+    },
+    {
+      db_path: join(homedir(), "Library/Application Support/Microsoft Edge/Default/Cookies"),
+      account: "Microsoft Edge",
+      service: "Microsoft Edge Safe Storage",
+    },
+  ],
+  DEFAULT_MODEL_LI = [
+    {
+      id: "gemini-flash",
+      model_id: "56fdd199312815e2",
+      disp: "Flash",
+      cat: "Flash",
+      desc: "Default Flash",
+      num: 1,
+      capacity: 1,
+      alias_li: ["gemini-flash", "flash", "default"],
+    },
+    {
+      id: "gemini-pro",
+      model_id: "797f3d0293f288ad",
+      disp: "Pro",
+      cat: "Pro",
+      desc: "Default Pro",
+      num: 3,
+      capacity: 1,
+      alias_li: ["gemini-pro", "pro"],
+    },
+  ];

package/src/cookieRead.js

@@ -0,0 +1,65 @@
+import { spawnSync } from "node:child_process";
+import { existsSync } from "node:fs";
+import { DatabaseSync } from "node:sqlite";
+import { BROWSER_PATH_LI, IV, SALT } from "./constant.js";
+
+export const keychainPwdRead = (account, service) => {
+  const proc = spawnSync(
+    "/usr/bin/security",
+    ["-q", "find-generic-password", "-w", "-a", account, "-s", service],
+    { encoding: "utf8" },
+  );
+  return (proc.stdout ?? "").trim();
+},
+
+valDecrypt = async (enc_data, aes_key, has_integrity) => {
+  const data = enc_data.slice(3),
+    dec = await crypto.subtle.decrypt({ name: "AES-CBC", iv: IV }, aes_key, data);
+  let byte_li = new Uint8Array(dec);
+  if (has_integrity) byte_li = byte_li.slice(32);
+  return new TextDecoder().decode(byte_li);
+},
+
+cookieRead = async () => {
+  if (process.env.SECURE_1PSID && process.env.SECURE_1PSIDTS) {
+    return (
+      "__Secure-1PSID=" +
+      process.env.SECURE_1PSID +
+      "; __Secure-1PSIDTS=" +
+      process.env.SECURE_1PSIDTS
+    );
+  }
+  const target = BROWSER_PATH_LI.find((browser_cfg) => existsSync(browser_cfg.db_path));
+  if (!target) return "";
+  const pwd = keychainPwdRead(target.account, target.service),
+    key_mat = await crypto.subtle.importKey("raw", new TextEncoder().encode(pwd), "PBKDF2", false, [
+      "deriveKey",
+    ]),
+    aes_key = await crypto.subtle.deriveKey(
+      { name: "PBKDF2", salt: SALT, iterations: 1003, hash: "SHA-1" },
+      key_mat,
+      { name: "AES-CBC", length: 128 },
+      false,
+      ["decrypt"],
+    ),
+    db = new DatabaseSync(target.db_path, { readOnly: true }),
+    version_row = db.prepare("SELECT value FROM meta WHERE key = 'version'").get(),
+    has_integrity = parseInt(version_row?.value ?? "0", 10) >= 24,
+    row_li = db
+      .prepare(
+        "SELECT name, encrypted_value FROM cookies WHERE host_key LIKE '%google.com%' AND name IN ('__Secure-1PSID', '__Secure-1PSIDTS')",
+      )
+      .all(),
+    cookie_map = {};
+
+  for (const { name, encrypted_value } of row_li) {
+    if (encrypted_value && encrypted_value.length > 3) {
+      const val = await valDecrypt(encrypted_value, aes_key, has_integrity);
+      cookie_map[name] = val;
+    }
+  }
+  db.close();
+  return Object.entries(cookie_map)
+    .map(([cookie_name, cookie_val]) => cookie_name + "=" + cookie_val)
+    .join("; ");
+};

package/src/modelDiscover.js

@@ -0,0 +1,278 @@
+import { BATCH_URL, DEFAULT_MODEL_LI, USER_AGENT } from "./constant.js";
+import { session_state } from "./sessionState.js";
+
+export const versionCompare = (a_ver, b_ver) => {
+  const a_match = (a_ver ?? "").match(/(\d+)(?:\.(\d+))?(?:\.(\d+))?/),
+    b_match = (b_ver ?? "").match(/(\d+)(?:\.(\d+))?(?:\.(\d+))?/),
+    a_maj = a_match ? parseInt(a_match[1], 10) : 0,
+    a_min = a_match && a_match[2] ? parseInt(a_match[2], 10) : 0,
+    a_pat = a_match && a_match[3] ? parseInt(a_match[3], 10) : 0,
+    b_maj = b_match ? parseInt(b_match[1], 10) : 0,
+    b_min = b_match && b_match[2] ? parseInt(b_match[2], 10) : 0,
+    b_pat = b_match && b_match[3] ? parseInt(b_match[3], 10) : 0;
+  if (a_maj !== b_maj) return a_maj - b_maj;
+  if (a_min !== b_min) return a_min - b_min;
+  return a_pat - b_pat;
+},
+
+stringVisualWidth = (str) => {
+  let width = 0;
+  for (const ch of str ?? "") {
+    const code = ch.codePointAt(0);
+    if (
+      (code >= 0x1100 && code <= 0x115f) ||
+      (code >= 0x2e80 && code <= 0xa4cf) ||
+      (code >= 0xac00 && code <= 0xd7a3) ||
+      (code >= 0xf900 && code <= 0xfaff) ||
+      (code >= 0xfe10 && code <= 0xfe19) ||
+      (code >= 0xfe30 && code <= 0xfe6f) ||
+      (code >= 0xff00 && code <= 0xff60) ||
+      (code >= 0xffe0 && code <= 0xffe6)
+    ) {
+      width += 2;
+    } else {
+      width += 1;
+    }
+  }
+  return width;
+},
+
+stringPadEnd = (str, target_width) => {
+  const current_width = stringVisualWidth(str);
+  return str + " ".repeat(Math.max(0, target_width - current_width));
+},
+
+modelListFormat = (model_li = null, default_model = null) => {
+  const target_model_li =
+      model_li && model_li.length > 0
+        ? model_li
+        : session_state.model_li.length > 0
+          ? session_state.model_li
+          : DEFAULT_MODEL_LI,
+    target_default = default_model ?? session_state.default_model ?? target_model_li[0],
+    sorted_model_li = [...target_model_li].sort((a, b) => {
+      if (a.id === target_default?.id) return -1;
+      if (b.id === target_default?.id) return 1;
+      return 0;
+    }),
+    is_tty = Boolean(process.stdout?.isTTY),
+    color = {
+      reset: is_tty ? "\x1b[0m" : "",
+      bold: is_tty ? "\x1b[1m" : "",
+      dim: is_tty ? "\x1b[2m" : "",
+      cyan: is_tty ? "\x1b[36m" : "",
+      green: is_tty ? "\x1b[32m" : "",
+      yellow: is_tty ? "\x1b[33m" : "",
+      gray: is_tty ? "\x1b[90m" : "",
+    },
+    headers = ["模型 ID (Model ID)", "显示名称 (Name)", "别名 (Aliases)"],
+    raw_rows = sorted_model_li.map((item) => {
+      const is_def = item.id === target_default?.id,
+        clean_aliases = (item.alias_li ?? [])
+          .filter((a) => a && a !== item.id && !/^[0-9a-f]{16}$/i.test(a))
+          .filter((a, idx, arr) => arr.indexOf(a) === idx)
+          .slice(0, 5)
+          .join(", ");
+      return [
+        (is_def ? "★ " : "  ") + item.id + (is_def ? " (默认)" : ""),
+        item.disp ?? item.id,
+        clean_aliases || "-",
+      ];
+    }),
+    col_widths = headers.map((h, i) =>
+      Math.max(stringVisualWidth(h), ...raw_rows.map((r) => stringVisualWidth(r[i]))),
+    ),
+    border_color = color.gray,
+    top_border =
+      border_color +
+      "┌─" +
+      col_widths.map((w) => "─".repeat(w)).join("─┬─") +
+      "─┐" +
+      color.reset,
+    mid_border =
+      border_color +
+      "├─" +
+      col_widths.map((w) => "─".repeat(w)).join("─┼─") +
+      "─┤" +
+      color.reset,
+    bot_border =
+      border_color +
+      "└─" +
+      col_widths.map((w) => "─".repeat(w)).join("─┴─") +
+      "─┘" +
+      color.reset,
+    header_line =
+      border_color +
+      "│ " +
+      headers
+        .map((h, i) => color.bold + color.cyan + stringPadEnd(h, col_widths[i]) + color.reset)
+        .join(border_color + " │ ") +
+      border_color +
+      " │" +
+      color.reset,
+    row_lines = raw_rows.map((r, idx) => {
+      const item = sorted_model_li[idx],
+        is_def = item.id === target_default?.id,
+        c0_pad = stringPadEnd(r[0], col_widths[0]),
+        c1_pad = stringPadEnd(r[1], col_widths[1]),
+        c2_pad = stringPadEnd(r[2], col_widths[2]),
+        c0 = is_def
+          ? color.green + color.bold + c0_pad + color.reset
+          : color.yellow + c0_pad + color.reset,
+        c1 = color.bold + c1_pad + color.reset,
+        c2 = color.dim + c2_pad + color.reset;
+      return (
+        border_color +
+        "│ " +
+        c0 +
+        border_color +
+        " │ " +
+        c1 +
+        border_color +
+        " │ " +
+        c2 +
+        border_color +
+        " │" +
+        color.reset
+      );
+    }),
+    output_parts = [
+      color.bold + color.cyan + "支持的模型列表 (Supported Models):" + color.reset,
+      top_border,
+      header_line,
+      mid_border,
+      ...row_lines,
+      bot_border,
+      color.dim + "  ★ = 默认模型 | 支持在请求中传入模型 ID 或任意别名" + color.reset,
+    ];
+  return output_parts.join("\n");
+},
+
+modelListPrint = (model_li = null, default_model = null) => {
+  console.log(modelListFormat(model_li, default_model));
+},
+
+modelFetch = async (at, bl, fsid, cookie_header) => {
+  ++session_state.req_id;
+  const session_id = crypto.randomUUID().toUpperCase(),
+    batch_headers = {
+      "Content-Type": "application/x-www-form-urlencoded;charset=utf-8",
+      Origin: "https://gemini.google.com",
+      Referer: "https://gemini.google.com/",
+      "X-Same-Domain": "1",
+      "User-Agent": USER_AGENT,
+      Cookie: cookie_header,
+      "x-goog-ext-525001261-jspb": JSON.stringify([
+        1,
+        null,
+        null,
+        null,
+        null,
+        null,
+        null,
+        null,
+        [4, 5, 6, 8],
+        null,
+        null,
+        null,
+        null,
+        null,
+        null,
+        null,
+        session_id,
+      ]),
+      "x-goog-ext-73010989-jspb": "[0]",
+    },
+    payload = [[["otAQ7b", "[]", null, "generic"]]],
+    form_data = new URLSearchParams(),
+    batch_url = new URL(BATCH_URL),
+    discovered_li = [];
+
+  form_data.set("at", at);
+  form_data.set("f.req", JSON.stringify(payload));
+
+  batch_url.searchParams.set("rpcids", "otAQ7b");
+  batch_url.searchParams.set("hl", "en");
+  batch_url.searchParams.set("_reqid", String(session_state.req_id));
+  batch_url.searchParams.set("rt", "c");
+  batch_url.searchParams.set("source-path", "/app");
+  if (bl) batch_url.searchParams.set("bl", bl);
+  if (fsid) batch_url.searchParams.set("f.sid", fsid);
+
+  const res = await fetch(batch_url.toString(), {
+    method: "POST",
+    headers: batch_headers,
+    body: form_data.toString(),
+  });
+
+  const res_text = await res.text();
+  let clean_text = res_text;
+  if (clean_text.startsWith(")]}\x27")) clean_text = clean_text.slice(4).trimStart();
+  const match_res = clean_text.match(/^(\d+)\n/);
+  if (match_res) {
+    const chunk_len = parseInt(match_res[1], 10),
+      start_idx = match_res[1].length,
+      chunk_str = clean_text.slice(start_idx, start_idx + chunk_len).trim();
+    if (chunk_str.startsWith("[")) {
+      const part_li = JSON.parse(chunk_str);
+      part_li.forEach((part) => {
+        if (part[1] === "otAQ7b" && part[2]) {
+          const part_body = JSON.parse(part[2]),
+            model_li = part_body[15],
+            tier_flag_li = part_body[16] ?? [],
+            capability_flag_li = part_body[17] ?? [];
+          let cap = 1;
+          if (capability_flag_li.includes(115)) cap = 4;
+          else if (tier_flag_li.includes(16) || capability_flag_li.includes(106)) cap = 3;
+          else if (tier_flag_li.includes(8) || capability_flag_li.includes(19)) cap = 2;
+
+          (model_li ?? []).forEach((item) => {
+            const id = item[0],
+              cat = String(item[1] ?? item[10] ?? ""),
+              disp = String(item[11] ?? item[19] ?? item[1] ?? ""),
+              desc = String(item[12] ?? item[2] ?? ""),
+              num =
+                typeof item[17] === "number" ? item[17] : typeof item[9] === "number" ? item[9] : 1,
+              disp_slug = disp.toLowerCase().replaceAll(/\s+/g, "-"),
+              cat_slug = cat.toLowerCase().replaceAll(/\s+/g, "-"),
+              primary_id = "gemini-" + disp_slug,
+              alias_li = [id.toLowerCase(), primary_id, "gemini-" + cat_slug, cat_slug, disp_slug];
+            discovered_li.push({
+              id: primary_id,
+              model_id: id,
+              disp,
+              cat,
+              desc,
+              num,
+              capacity: cap,
+              alias_li,
+            });
+          });
+        }
+      });
+    }
+  }
+
+  const non_lite_li = discovered_li.filter(
+    (model_item) =>
+      !model_item.disp.toLowerCase().includes("lite") &&
+      !model_item.cat.toLowerCase().includes("lite") &&
+      !model_item.id.toLowerCase().includes("lite"),
+  );
+  non_lite_li.sort((a, b) => versionCompare(b.disp, a.disp));
+  session_state.default_model = non_lite_li[0] ?? discovered_li[0] ?? DEFAULT_MODEL_LI[0];
+
+  return discovered_li;
+},
+
+modelMap = (req_model) => {
+  const norm = (req_model ?? "").toLowerCase(),
+    available_li = session_state.model_li.length > 0 ? session_state.model_li : DEFAULT_MODEL_LI,
+    found = available_li.find(
+      (model_item) =>
+        model_item.id === norm ||
+        model_item.model_id === norm ||
+        model_item.alias_li.some((alias_item) => norm.includes(alias_item) || alias_item === norm),
+    );
+  return found ?? session_state.default_model ?? available_li[0];
+};

package/src/payloadBuild.js

@@ -0,0 +1,115 @@
+import {
+  ENABLE_THINKING,
+  THINKING_LEVEL_DISABLED,
+  THINKING_LEVEL_ENABLED,
+} from "./constant.js";
+import {
+  toolPromptBuild,
+  toolResultFormat,
+  toolTailAnchorBuild,
+} from "./toolHandle.js";
+
+export const conversationFormat = (msg_li, tool_li = null, tool_choice = null) => {
+  const parts_li = [],
+    tools_prompt = toolPromptBuild(tool_li, tool_choice);
+
+  if (tools_prompt) {
+    parts_li.push("[System instruction]: " + tools_prompt);
+  }
+
+  (msg_li ?? []).forEach((msg) => {
+    const { role, content, name, toolName, tool_name, toolCallId, tool_call_id } = msg,
+      raw_calls = msg.tool_calls ?? msg.toolCalls,
+      tool_calls = Array.isArray(raw_calls) ? raw_calls : null,
+      text_content = Array.isArray(content)
+        ? content
+            .filter((item) => item.type === "text" || item.type === "input_text")
+            .map((item) => item.text)
+            .join("")
+        : content ?? "";
+
+    if (role === "system") {
+      parts_li.push("System: " + text_content);
+    } else if (role === "user") {
+      parts_li.push("Human: " + text_content);
+    } else if (role === "assistant") {
+      if (tool_calls && tool_calls.length > 0) {
+        const tc_str_li = tool_calls.map((tc) => {
+          const fn = tc.function ?? tc,
+            fn_name = fn.name ?? tc.name ?? "",
+            raw_args = fn.arguments ?? tc.arguments ?? tc.input ?? {};
+          let parsed_args = {};
+          try {
+            parsed_args = typeof raw_args === "string" ? JSON.parse(raw_args) : raw_args;
+          } catch {
+            parsed_args = {};
+          }
+          return (
+            "```tool_call\n" +
+            JSON.stringify({ name: fn_name, arguments: parsed_args }) +
+            "\n```"
+          );
+        });
+        const prefix = text_content ? "Assistant: " + text_content + "\n" : "Assistant:\n";
+        parts_li.push(prefix + tc_str_li.join("\n"));
+      } else {
+        parts_li.push("Assistant: " + text_content);
+      }
+    } else if (role === "tool") {
+      const tool_label = name ?? toolName ?? tool_name ?? toolCallId ?? tool_call_id ?? "";
+      parts_li.push(toolResultFormat(tool_label, text_content));
+    }
+  });
+
+  if (tools_prompt) {
+    parts_li.push(toolTailAnchorBuild());
+  }
+
+  return parts_li.join("\n\n").trim();
+},
+
+payloadBuild = (prompt, model_info) => {
+  const session_id = crypto.randomUUID().toUpperCase(),
+    uuid_val = crypto.randomUUID().toUpperCase(),
+    thinking_level = ENABLE_THINKING ? THINKING_LEVEL_ENABLED : THINKING_LEVEL_DISABLED,
+    model_header = [
+      1,
+      null,
+      null,
+      null,
+      model_info.model_id,
+      null,
+      null,
+      0,
+      [4, 5, 6, 8],
+      null,
+      null,
+      model_info.capacity ?? 1,
+      null,
+      null,
+      model_info.num,
+      thinking_level,
+      session_id,
+    ],
+    inner_req = Array.from({ length: 81 }, () => null);
+
+  inner_req[0] = [prompt, 0, null, null, null, null, 0];
+  inner_req[1] = ["en"];
+  inner_req[2] = ["", "", "", null, null, null, null, null, null, ""];
+  inner_req[6] = [1];
+  inner_req[7] = 1;
+  inner_req[10] = 1;
+  inner_req[11] = 0;
+  inner_req[17] = [[0]];
+  inner_req[18] = 0;
+  inner_req[27] = 1;
+  inner_req[30] = [4];
+  inner_req[41] = [1];
+  inner_req[53] = 0;
+  inner_req[59] = uuid_val;
+  inner_req[61] = [];
+  inner_req[68] = 1;
+  inner_req[79] = model_info.num;
+  inner_req[80] = thinking_level;
+  return [inner_req, model_header, uuid_val];
+};