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

package/index.js

@@ -1,11 +1,5 @@
 #!/usr/bin/env -S bun
-import { spawnSync } from "node:child_process";
-import { existsSync } from "node:fs";
-import { createServer } from "node:http";
-import { homedir } from "node:os";
 import { join } from "node:path";
-import { DatabaseSync } from "node:sqlite";
-import { Readable } from "node:stream";
 
 try {
   process.loadEnvFile(join(import.meta.dirname, ".env"));
@@ -13,876 +7,68 @@ try {
   if (err.code !== "ENOENT") throw err;
 }
 
-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",
-  STOP_REASON_STOP = "stop",
-  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"],
-    },
-  ];
-
-export let session_state = {
-  cookie_header: "",
-  at: "",
-  bl: "",
-  fsid: "",
-  req_id: Math.floor(Math.random() * 90000) + 10000,
-  model_li: [],
-  default_model: null,
-};
-
-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("; ");
-},
-
-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;
-},
-
-modelFetch = async (at, bl, fsid, cookie_header) => {
-  session_state.req_id += 1;
-  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);
-      for (const part of part_li) {
-        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;
-
-          for (const item of model_li ?? []) {
-            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];
-  console.log(
-    "默认模型: " + session_state.default_model.disp + " (" + session_state.default_model.id + ")",
-  );
-
-  return discovered_li;
-},
-
-sessionInit = async () => {
-  try {
-    if (!session_state.cookie_header) {
-      session_state.cookie_header = await cookieRead();
-    }
-    const res = await fetch(INIT_URL, {
-      headers: {
-        "User-Agent": USER_AGENT,
-        Cookie: session_state.cookie_header,
-      },
-      redirect: "manual",
-    });
-
-    let html = "";
-    if (res.status === STATUS_OK) {
-      html = await res.text();
-    } else if (res.status >= 300 && res.status < 400) {
-      const location = res.headers.get("location") ?? "";
-      if (location.includes("/sorry/")) {
-        console.warn("Gemini 会话初始化提示: 触发了 Google 验证码重定向 (sorry/index)");
-      } else if (location) {
-        const redirect_res = await fetch(location, {
-          headers: {
-            "User-Agent": USER_AGENT,
-            Cookie: session_state.cookie_header,
-          },
-        });
-        if (redirect_res.ok) html = await redirect_res.text();
-      }
-    }
-
-    const at_match = html.match(/"SNlM0e":\s*"([^"]+)"/),
-      bl_match = html.match(/"cfb2h":\s*"([^"]+)"/),
-      fsid_match = html.match(/"FdrFJe":\s*"([^"]+)"/);
-    session_state.at = at_match ? at_match[1] : "";
-    session_state.bl = bl_match ? bl_match[1] : "";
-    session_state.fsid = fsid_match ? fsid_match[1] : "";
-    if (session_state.at) {
-      session_state.model_li = await modelFetch(
-        session_state.at,
-        session_state.bl,
-        session_state.fsid,
-        session_state.cookie_header,
-      );
-    }
-  } catch (err) {
-    console.warn("会话初始化异常:", err.message);
-  }
-  return session_state;
-},
-
-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];
-},
-
-conversationFormat = (msg_li) => {
-  let conversation = "";
-  for (const { role, content } of msg_li ?? []) {
-    const text_content = Array.isArray(content)
-      ? content
-          .filter((content_item) => content_item.type === "text")
-          .map((content_item) => content_item.text)
-          .join("")
-      : content ?? "";
-    if (role === "system") {
-      conversation += "System: " + text_content + "\n\n";
-    } else if (role === "user") {
-      conversation += "Human: " + text_content + "\n\n";
-    } else if (role === "assistant") {
-      conversation += "Assistant: " + text_content + "\n\n";
-    }
-  }
-  return conversation.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];
-},
-
-streamChunkExtract = (buf) => {
-  let cur_buf = buf;
-  if (cur_buf.startsWith(")]}\x27")) {
-    cur_buf = cur_buf.slice(4).trimStart();
-  }
-  const extracted_li = [];
-  while (true) {
-    cur_buf = cur_buf.trimStart();
-    const match_res = cur_buf.match(/^(\d+)\n/);
-    if (!match_res) break;
-    const len_str = match_res[1],
-      chunk_len = parseInt(len_str, 10),
-      start_idx = len_str.length;
-    if (cur_buf.length < start_idx + chunk_len) break;
-    const chunk_str = cur_buf.slice(start_idx, start_idx + chunk_len).trim();
-    cur_buf = cur_buf.slice(start_idx + chunk_len);
-
-    if (chunk_str.startsWith("[")) {
-      const item_li = JSON.parse(chunk_str);
-      for (const item of item_li) {
-        if (Array.isArray(item) && item[2]) {
-          const inner = JSON.parse(item[2]),
-            cand_li = inner[4];
-          if (Array.isArray(cand_li) && cand_li.length > 0) {
-            const cand = cand_li[0],
-              thoughts = cand[37]?.[0]?.[0] ?? "",
-              raw_text = cand[1]?.[0] ?? "",
-              card_text = raw_text.startsWith("http://googleusercontent.com/card_content/")
-                ? (cand[22]?.[0] ?? raw_text)
-                : raw_text,
-              text = card_text.replaceAll(
-                /https?:\/\/googleusercontent\.com\/(?:\w+\/)+\d+\n*/g,
-                "",
-              );
-            extracted_li.push({ thoughts, text });
-          }
-        }
-      }
-    }
-  }
-  return [extracted_li, cur_buf];
-},
-
-sseStreamCreate = (body, completion_id, model, created_time) => {
-  const enc = new TextEncoder();
-  return new ReadableStream({
-    async start(controller) {
-      const chunkSend = (delta, finish_reason = null) => {
-        const payload = JSON.stringify({
-          id: completion_id,
-          object: "chat.completion.chunk",
-          created: created_time,
-          model,
-          choices: [
-            {
-              index: 0,
-              delta,
-              finish_reason,
-            },
-          ],
-        });
-        console.log("<-- [SSE 块]:", JSON.stringify(delta));
-        controller.enqueue(enc.encode("data: " + payload + "\n\n"));
-      };
-
-      chunkSend({ role: ROLE_ASSISTANT });
-
-      const reader = body.getReader(),
-        decoder = new TextDecoder();
-      let buf = "",
-        last_text = "",
-        last_thought = "",
-        thinking_started = false;
-
-      while (true) {
-        const read_res = await reader.read();
-        if (read_res.done) break;
-        buf += decoder.decode(read_res.value, { stream: true });
-
-        const [extracted_li, next_buf] = streamChunkExtract(buf);
-        buf = next_buf;
-
-        for (const { thoughts, text } of extracted_li) {
-          if (ENABLE_THINKING && thoughts) {
-            if (thoughts.startsWith(last_thought)) {
-              const thought_delta = thoughts.slice(last_thought.length);
-              if (thought_delta) {
-                if (!thinking_started) {
-                  chunkSend({ content: "<think>\n" });
-                  thinking_started = true;
-                }
-                chunkSend({
-                  content: thought_delta,
-                  reasoning_content: thought_delta,
-                });
-                last_thought = thoughts;
-              }
-            } else {
-              chunkSend({
-                content: thoughts,
-                reasoning_content: thoughts,
-              });
-              last_thought = thoughts;
-            }
-          }
-
-          if (text.startsWith(last_text)) {
-            const text_delta = text.slice(last_text.length);
-            if (text_delta) {
-              if (ENABLE_THINKING && thinking_started && !last_text) {
-                chunkSend({ content: "</think>\n\n" });
-              }
-              chunkSend({ content: text_delta });
-              last_text = text;
-            }
-          } else if (text) {
-            chunkSend({ content: text });
-            last_text = text;
-          }
-        }
-      }
-
-      if (ENABLE_THINKING && thinking_started && !last_text) {
-        chunkSend({ content: "</think>\n\n" });
-      }
-
-      chunkSend({}, STOP_REASON_STOP);
-      controller.enqueue(enc.encode("data: [DONE]\n\n"));
-      controller.close();
-      console.log(
-        "<-- [SSE 完成] 完整输出:\n" +
-          (last_thought ? "<think>\n" + last_thought + "\n</think>\n\n" : "") +
-          last_text,
-      );
-    },
-  });
-},
-
-fullResponseCollect = async (body) => {
-  const reader = body.getReader(),
-    decoder = new TextDecoder();
-  let buf = "",
-    final_text = "",
-    final_thought = "";
-
-  while (true) {
-    const read_res = await reader.read();
-    if (read_res.done) break;
-    buf += decoder.decode(read_res.value, { stream: true });
-
-    const [extracted_li, next_buf] = streamChunkExtract(buf);
-    buf = next_buf;
-
-    for (const { thoughts, text } of extracted_li) {
-      if (text) final_text = text;
-      if (thoughts) final_thought = thoughts;
-    }
-  }
-
-  let content = final_text;
-  if (ENABLE_THINKING && final_thought) {
-    content = "<think>\n" + final_thought + "\n</think>\n\n" + final_text;
-  }
-  return content;
-},
-
-authVerify = (req) => {
-  if (!API_KEY) return true;
-  const auth_header = req.headers.get("authorization") ?? "",
-    key = auth_header.replace(/^Bearer\s+/i, "");
-  return key === API_KEY;
-},
-
-healthHandle = () => {
-  const res_body = JSON.stringify({
-    status: "healthy",
-    service: "Gemini API 代理 (Node/Bun)",
-    version: "1.0.0",
-    endpoints: ["/v1/models", "/v1/chat/completions"],
-  });
-  console.log("<-- 响应: [200 OK]", res_body);
-  return new Response(res_body, {
-    status: STATUS_OK,
-    headers: { ...CORS_HEADERS, "Content-Type": "application/json" },
-  });
-},
-
-modelListHandle = (now) => {
-  const available_model_li =
-      session_state.model_li.length > 0 ? session_state.model_li : DEFAULT_MODEL_LI,
-    model_data_li = available_model_li.map((model_item) => ({
-      id: model_item.id,
-      object: "model",
-      created: now,
-      owned_by: "google-gemini-web",
-    })),
-    res_body = JSON.stringify({ object: "list", data: model_data_li });
-  console.log("<-- 响应: [200 OK] 模型列表:", res_body);
-  return new Response(res_body, {
-    status: STATUS_OK,
-    headers: { ...CORS_HEADERS, "Content-Type": "application/json" },
-  });
-},
-
-modelDetailHandle = (req_model_id, now) => {
-  const target_model = modelMap(req_model_id),
-    res_body = JSON.stringify({
-      id: target_model.id,
-      object: "model",
-      created: now,
-      owned_by: "google-gemini-web",
-    });
-  console.log("<-- 响应: [200 OK] 模型详情:", res_body);
-  return new Response(res_body, {
-    status: STATUS_OK,
-    headers: { ...CORS_HEADERS, "Content-Type": "application/json" },
-  });
-},
-
-imageProxyHandle = async (req_url) => {
-  const target_url = req_url.searchParams.get("url");
-  if (!target_url) {
-    return new Response(
-      JSON.stringify({ code: ERR_MISSING_URL, error: "缺少 url 参数" }),
-      {
-        status: STATUS_BAD_REQUEST,
-        headers: { ...CORS_HEADERS, "Content-Type": "application/json" },
-      },
-    );
-  }
-  const img_res = await fetch(target_url, {
-      headers: {
-        "User-Agent": USER_AGENT,
-        Referer: "https://gemini.google.com/",
-      },
-    }),
-    img_bytes = new Uint8Array(await img_res.arrayBuffer()),
-    headers = {
-      ...CORS_HEADERS,
-      "Content-Type": img_res.headers.get("content-type") ?? "image/png",
-    };
-  return new Response(img_bytes, { status: STATUS_OK, headers });
-},
-
-chatCompletionsHandle = async (req, body) => {
-  if (!authVerify(req)) {
-    const res_body = JSON.stringify({ code: ERR_UNAUTHORIZED, error: "未授权" });
-    console.log("<-- 响应: [401 Unauthorized]", res_body);
-    return new Response(res_body, {
-      status: STATUS_UNAUTHORIZED,
-      headers: { ...CORS_HEADERS, "Content-Type": "application/json" },
-    });
-  }
-
-  if (!body.messages && body.prompt) {
-    body.messages = [{ role: "user", content: body.prompt }];
-  }
-
-  const model_info = modelMap(body.model),
-    conversation = conversationFormat(body.messages);
-
-  if (!session_state.at) {
-    await sessionInit();
-  }
-
-  session_state.req_id += 1;
-  const [inner_req, model_header, uuid_val] = payloadBuild(conversation, model_info),
-    form_data = new URLSearchParams();
-  form_data.set("at", session_state.at);
-  form_data.set("f.req", JSON.stringify([null, JSON.stringify(inner_req)]));
-
-  const gen_url = new URL(GEN_URL);
-  gen_url.searchParams.set("hl", "en");
-  gen_url.searchParams.set("_reqid", String(session_state.req_id));
-  gen_url.searchParams.set("rt", "c");
-  if (session_state.bl) gen_url.searchParams.set("bl", session_state.bl);
-  if (session_state.fsid) gen_url.searchParams.set("f.sid", session_state.fsid);
-
-  const gen_res = await fetch(gen_url.toString(), {
-      method: "POST",
-      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: session_state.cookie_header,
-        "x-goog-ext-525001261-jspb": JSON.stringify(model_header),
-        "x-goog-ext-73010989-jspb": "[0]",
-        "x-goog-ext-73010990-jspb": "[0,0,0]",
-        "x-goog-ext-525005358-jspb": JSON.stringify([uuid_val, 1]),
-      },
-      body: form_data.toString(),
-    }),
-    completion_id = "chatcmpl-" + crypto.randomUUID(),
-    created_time = Math.floor(Date.now() / 1000);
-
-  if (body.stream) {
-    console.log("<-- 响应: [200 OK] 开始流式传输 (SSE)...");
-    const sse_stream = sseStreamCreate(gen_res.body, completion_id, model_info.id, created_time);
-    return new Response(sse_stream, {
-      status: STATUS_OK,
-      headers: {
-        ...CORS_HEADERS,
-        "Content-Type": "text/event-stream",
-        "Cache-Control": "no-cache",
-        Connection: "keep-alive",
-      },
-    });
-  }
-
-  const full_content = await fullResponseCollect(gen_res.body),
-    prompt_tokens = conversation.split(/\s+/).length,
-    completion_tokens = full_content.split(/\s+/).length,
-    res_json = {
-      id: completion_id,
-      object: "chat.completion",
-      created: created_time,
-      model: model_info.id,
-      choices: [
-        {
-          index: 0,
-          message: {
-            role: ROLE_ASSISTANT,
-            content: full_content,
-          },
-          finish_reason: STOP_REASON_STOP,
-        },
-      ],
-      usage: {
-        prompt_tokens,
-        completion_tokens,
-        total_tokens: prompt_tokens + completion_tokens,
-      },
-    },
-    res_body = JSON.stringify(res_json);
-  console.log("<-- 响应: [200 OK]", res_body);
-  return new Response(res_body, {
-    status: STATUS_OK,
-    headers: { ...CORS_HEADERS, "Content-Type": "application/json" },
-  });
-},
-
-reqHandle = async (req) => {
-  const req_url = new URL(req.url),
-    pathname = req_url.pathname,
-    norm_path = pathname.replace(/\/+$/, "") || "/",
-    is_body_method = req.method === "POST" || req.method === "PUT" || req.method === "PATCH";
-  let body = {},
-    req_text = "";
-
-  if (is_body_method) {
-    req_text = await req.text();
-    const trimmed = req_text.trim();
-    if (trimmed && (trimmed.startsWith("{") || trimmed.startsWith("["))) {
-      body = JSON.parse(trimmed);
-    }
-  }
-
-  console.log("\n--> [" + req.method + "] " + pathname);
-  console.log("--> 请求头:", JSON.stringify(Object.fromEntries(req.headers.entries())));
-  if (req_text.trim()) {
-    console.log("--> 请求体:", req_text);
-  } else {
-    console.log("--> 请求体: (无)");
-  }
-
-  if (req.method === "OPTIONS") {
-    console.log("<-- 响应: [204 No Content]");
-    return new Response(null, { status: STATUS_NO_CONTENT, headers: CORS_HEADERS });
-  }
-
-  if (norm_path === "/" || norm_path === "/v1") {
-    return healthHandle();
-  }
-
-  const now = Math.floor(Date.now() / 1000);
-  if ((norm_path === "/models" || norm_path.endsWith("/models")) && req.method === "GET") {
-    return modelListHandle(now);
-  }
-
-  if (norm_path.includes("/models/") && req.method === "GET") {
-    const req_model_id = norm_path.split("/").pop();
-    return modelDetailHandle(req_model_id, now);
-  }
-
-  if (norm_path.endsWith("/gemini-proxy/image") && req.method === "GET") {
-    console.log("<-- 响应: [200 OK] 代理图片");
-    return imageProxyHandle(req_url);
-  }
-
-  if (
-    (norm_path === "/chat/completions" ||
-      norm_path.endsWith("/chat/completions") ||
-      norm_path === "/completions" ||
-      norm_path.endsWith("/completions")) &&
-    req.method === "POST"
-  ) {
-    return chatCompletionsHandle(req, body);
-  }
-
-  const res_body = JSON.stringify({
-    code: ERR_NOT_FOUND,
-    error: "未找到接口",
-    path: pathname,
-    method: req.method,
-  });
-  console.log("<-- 响应: [404 Not Found] 未找到接口: " + req.method + " " + pathname);
-  return new Response(res_body, {
-    status: STATUS_NOT_FOUND,
-    headers: { ...CORS_HEADERS, "Content-Type": "application/json" },
-  });
-},
-
-serverStart = () => {
-  if (typeof Bun !== "undefined" && Bun.serve) {
-    const bun_server = Bun.serve({
-      port: PORT,
-      hostname: HOST,
-      fetch: reqHandle,
-    });
-    console.log("服务运行在 (Bun) http://" + HOST + ":" + PORT);
-    return bun_server;
-  }
-
-  const node_server = createServer(async (node_req, node_res) => {
-    try {
-      const url = "http://" + (node_req.headers.host ?? (HOST + ":" + PORT)) + node_req.url,
-        has_body = node_req.method !== "GET" && node_req.method !== "HEAD",
-        web_req = new Request(url, {
-          method: node_req.method,
-          headers: node_req.headers,
-          body: has_body ? Readable.toWeb(node_req) : null,
-          duplex: "half",
-        }),
-        web_res = await reqHandle(web_req);
-
-      node_res.statusCode = web_res.status;
-      for (const [header_key, header_val] of web_res.headers.entries()) {
-        node_res.setHeader(header_key, header_val);
-      }
-      if (web_res.body) {
-        Readable.fromWeb(web_res.body).pipe(node_res);
-      } else {
-        node_res.end();
-      }
-    } catch (err) {
-      if (!node_res.headersSent) {
-        node_res.statusCode = STATUS_SERVER_ERR;
-        node_res.end(JSON.stringify({ code: STATUS_SERVER_ERR, error: "Internal Server Error" }));
-      }
-      console.error("服务器异常:", err);
-    }
-  });
-
-  node_server.listen(PORT, HOST, () => {
-    console.log("服务运行在 (Node) http://" + HOST + ":" + PORT);
-  });
-  return node_server;
-};
+import { sessionInit } from "./src/sessionState.js";
+import { serverStart } from "./src/serverStart.js";
+
+export {
+  API_KEY,
+  BROWSER_PATH_LI,
+  CORS_HEADERS,
+  DEFAULT_MODEL_LI,
+  ENABLE_THINKING,
+  ERR_MISSING_URL,
+  ERR_NOT_FOUND,
+  ERR_UNAUTHORIZED,
+  HOST,
+  INIT_URL,
+  PORT,
+  ROLE_ASSISTANT,
+  ROLE_SYSTEM,
+  ROLE_TOOL,
+  ROLE_USER,
+  STATUS_BAD_REQUEST,
+  STATUS_NO_CONTENT,
+  STATUS_NOT_FOUND,
+  STATUS_OK,
+  STATUS_SERVER_ERR,
+  STOP_REASON_STOP,
+  STOP_REASON_TOOL_CALLS,
+  TOOL_FENCE_CLOSE,
+  TOOL_FENCE_OPEN,
+  USER_AGENT,
+} from "./src/constant.js";
+
+export { cookieRead, keychainPwdRead, valDecrypt } from "./src/cookieRead.js";
+export { sessionInit, session_state } from "./src/sessionState.js";
+export {
+  modelFetch,
+  modelListFormat,
+  modelListPrint,
+  modelMap,
+  stringPadEnd,
+  stringVisualWidth,
+  versionCompare,
+} from "./src/modelDiscover.js";
+export {
+  jsonFormat,
+  toolCallExtract,
+  toolFenceGateCreate,
+  toolPromptBuild,
+  toolResultFormat,
+} from "./src/toolHandle.js";
+export { conversationFormat, payloadBuild } from "./src/payloadBuild.js";
+export { fullResponseCollect, streamChunkExtract } from "./src/streamExtract.js";
+export { sseStreamCreate } from "./src/sseResponse.js";
+export {
+  authVerify,
+  chatCompletionsHandle,
+  healthHandle,
+  imageProxyHandle,
+  modelDetailHandle,
+  modelListHandle,
+  reqHandle,
+} from "./src/router.js";
+export { serverStart } from "./src/serverStart.js";
 
 await sessionInit();