@fudrouter/fsrouter 0.6.58 → 0.6.60

package/open-sse/utils/sessionManager.js

@@ -1,240 +0,0 @@
-/**
- * Session Manager for Antigravity Cloud Code
- *
- * Handles session ID generation and caching for prompt caching continuity.
- * Mimics the Antigravity binary behavior: generates a session ID at startup
- * and keeps it for the process lifetime, scoped per account/connection.
- *
- * Reference: antigravity-claude-proxy/src/cloudcode/session-manager.js
- */
-
-import crypto from "crypto";
-import { MEMORY_CONFIG } from "../config/runtimeConfig.js";
-
-// Runtime storage: Key = connectionId, Value = { sessionId, lastUsed }
-const runtimeSessionStore = new Map();
-
-// Periodically evict entries that haven't been used within TTL
-const cleanupInterval = setInterval(() => {
-    const now = Date.now();
-    for (const [key, entry] of runtimeSessionStore) {
-        if (now - entry.lastUsed > MEMORY_CONFIG.sessionTtlMs) {
-            runtimeSessionStore.delete(key);
-        }
-    }
-}, MEMORY_CONFIG.sessionCleanupIntervalMs);
-
-// Allow Node.js to exit even if interval is still active
-if (cleanupInterval.unref) cleanupInterval.unref();
-
-/**
- * Get or create a session ID for the given connection.
- *
- * The binary generates a session ID once at startup: `rs() + Date.now()`.
- * Since 9router is long-running, we simulate this "per-launch" behavior by
- * storing a generated ID in memory for each connection.
- *
- * - If 9router restarts, the ID changes (matching binary restart behavior).
- * - Within a running instance, the ID is stable for that connection.
- * - This enables prompt caching while using the EXACT random logic of the binary.
- *
- * @param {string} connectionId - The connection identifier (email or unique ID)
- * @returns {string} A stable session ID string matching binary format
- */
-export function deriveSessionId(connectionId) {
-    if (!connectionId) {
-        return generateBinaryStyleId();
-    }
-
-    const existing = runtimeSessionStore.get(connectionId);
-    if (existing) {
-        existing.lastUsed = Date.now();
-        return existing.sessionId;
-    }
-
-    // Evict oldest entry if store exceeds max size (safety cap between cleanup cycles)
-    const MAX_SESSIONS = 1000;
-    if (runtimeSessionStore.size >= MAX_SESSIONS) {
-      const oldest = runtimeSessionStore.keys().next().value;
-      runtimeSessionStore.delete(oldest);
-    }
-
-    const sessionId = generateBinaryStyleId();
-    runtimeSessionStore.set(connectionId, { sessionId, lastUsed: Date.now() });
-    return sessionId;
-}
-
-/**
- * Generate a Session ID using the binary's exact logic.
- * Format: `rs() + Date.now()` where `rs()` is randomUUID
- *
- * @returns {string} A session ID in binary format
- */
-export function generateBinaryStyleId() {
-    return crypto.randomUUID() + Date.now().toString();
-}
-
-/**
- * Clears all session IDs (e.g. useful for testing or explicit reset)
- */
-export function clearSessionStore() {
-    runtimeSessionStore.clear();
-    assistantSessionStore.clear();
-}
-
-// Conversation-stable session store: Key = hash(scope+assistant text), Value = { sessionId, lastUsed }
-const assistantSessionStore = new Map();
-const ASSISTANT_MIN_LEN = 50;
-const ASSISTANT_CAP_LEN = 50;
-const MAX_ASSISTANT_SESSIONS = 5000;
-
-// Client headers/body fields that carry an upstream session id (priority order)
-const SESSION_HEADER_KEYS = ["x-session-id", "session-id", "session_id", "x-amp-thread-id", "x-client-request-id"];
-const CLAUDE_CODE_SESSION_RE = /_session_([a-f0-9-]+)$/;
-
-function sha16(text) {
-    return crypto.createHash("sha256").update(text).digest("hex").slice(0, 16);
-}
-
-// Normalize a session id candidate (trim, length cap)
-function normalizeSessionId(value) {
-    if (typeof value !== "string") return null;
-    const v = value.trim();
-    if (!v || v.length > 256) return null;
-    return v;
-}
-
-// Extract Claude Code session id from metadata.user_id (_session_{uuid} | JSON {session_id})
-function extractClaudeCodeSession(userId) {
-    if (typeof userId !== "string" || !userId) return null;
-    const m = userId.match(CLAUDE_CODE_SESSION_RE);
-    if (m) return m[1];
-    if (userId[0] === "{") {
-        try { return normalizeSessionId(JSON.parse(userId)?.session_id); } catch { /* noop */ }
-    }
-    return null;
-}
-
-// Lowercase-key lookup for raw client headers
-function headerValue(headers, key) {
-    if (!headers || typeof headers !== "object") return null;
-    return normalizeSessionId(headers[key] ?? headers[key.toLowerCase()]);
-}
-
-// Read client-provided session id from headers/body (no generation)
-// Antigravity envelope carries session in request.sessionId; requestId embeds conversation uuid
-const ANTIGRAVITY_CONV_RE = /^[a-z]+\/([0-9a-f-]{36})\//i;
-function extractAntigravitySession(body) {
-    const sid = body?.request?.sessionId;
-    if (sid != null && sid !== "") return normalizeSessionId(String(sid));
-    const m = typeof body?.requestId === "string" ? body.requestId.match(ANTIGRAVITY_CONV_RE) : null;
-    return m ? normalizeSessionId(m[1]) : null;
-}
-
-function extractClientSessionId(headers, body) {
-    const claude = extractClaudeCodeSession(body?.metadata?.user_id);
-    if (claude) return `claude:${claude}`;
-    const antigravity = extractAntigravitySession(body);
-    if (antigravity) return `antigravity:${antigravity}`;
-    for (const key of SESSION_HEADER_KEYS) {
-        const v = headerValue(headers, key);
-        if (v) return v;
-    }
-    const fromBody =
-        normalizeSessionId(body?.prompt_cache_key) ||
-        normalizeSessionId(body?.session_id) ||
-        normalizeSessionId(body?.conversation_id) ||
-        normalizeSessionId(body?.metadata?.user_id);
-    return fromBody || null;
-}
-
-// Accumulate assistant text from OpenAI/Responses-style input/messages (cap-limited)
-function accumulateAssistantText(body) {
-    const items = Array.isArray(body?.input) ? body.input
-        : Array.isArray(body?.messages) ? body.messages : null;
-    if (!items) return "";
-    let text = "";
-    for (const item of items) {
-        if (item?.role !== "assistant") continue;
-        if (typeof item.content === "string") text += item.content;
-        else if (Array.isArray(item.content)) {
-            for (const c of item.content) text += c?.text || c?.output || "";
-        }
-        if (text.length >= ASSISTANT_CAP_LEN) break;
-    }
-    return text;
-}
-
-// Stable session id keyed on accumulated assistant text (avoids collision on identical first user prompt)
-function assistantTextSessionId(scope, body) {
-    const text = accumulateAssistantText(body);
-    if (text.length < ASSISTANT_MIN_LEN) return null;
-    const hash = sha16(`${scope}:${text.slice(0, ASSISTANT_CAP_LEN)}`);
-    const existing = assistantSessionStore.get(hash);
-    if (existing) {
-        existing.lastUsed = Date.now();
-        return existing.sessionId;
-    }
-    if (assistantSessionStore.size >= MAX_ASSISTANT_SESSIONS) {
-        assistantSessionStore.delete(assistantSessionStore.keys().next().value);
-    }
-    const sessionId = generateBinaryStyleId();
-    assistantSessionStore.set(hash, { sessionId, lastUsed: Date.now() });
-    return sessionId;
-}
-
-/**
- * Resolve a conversation-stable session id (generalizes Codex resolveCacheSessionId).
- * Priority: client session → accumulated-assistant-text hash → workspaceId → per-connection.
- *
- * @param {object} opts
- * @param {object} [opts.headers] - Raw client request headers (lowercase keys)
- * @param {object} [opts.body] - Parsed request body
- * @param {string} [opts.connectionId] - Connection identifier (fallback scope)
- * @param {string} [opts.workspaceId] - Provider workspace id (account-wide fallback)
- * @param {string} [opts.scope] - Provider scope to isolate cache keys across providers
- * @returns {string} A stable session id
- */
-export function resolveSessionId({ headers, body, connectionId, workspaceId, scope = "" } = {}) {
-    const client = extractClientSessionId(headers, body);
-    if (client) return client;
-    const fromAssistant = assistantTextSessionId(`${scope}:${connectionId || ""}`, body);
-    if (fromAssistant) return fromAssistant;
-    const ws = normalizeSessionId(workspaceId);
-    if (ws) return ws;
-    return deriveSessionId(connectionId);
-}
-
-// Capture session id from request body + credentials (envelope still intact here)
-export function captureSessionId(body, credentials, connectionId, scope = "") {
-    return resolveSessionId({ headers: credentials?.rawHeaders, body, connectionId, scope });
-}
-
-// Convert any session id to Antigravity numeric format "-<int64>" (matches real AG / CLIProxyAPI).
-// Already-numeric ids (native AG sessionId) pass through unchanged.
-export function toNumericSessionId(sessionId) {
-    const v = normalizeSessionId(sessionId);
-    if (!v) return null;
-    if (/^-?\d+$/.test(v)) return v;
-    const h = crypto.createHash("sha256").update(v).digest();
-    const n = h.readBigUInt64BE(0) & 0x7fffffffffffffffn;
-    return `-${n.toString()}`;
-}
-
-// Cleanup expired assistant-session entries
-const assistantCleanup = setInterval(() => {
-    const now = Date.now();
-    for (const [key, entry] of assistantSessionStore) {
-        if (now - entry.lastUsed > MEMORY_CONFIG.sessionTtlMs) assistantSessionStore.delete(key);
-    }
-}, MEMORY_CONFIG.sessionCleanupIntervalMs);
-if (assistantCleanup.unref) assistantCleanup.unref();
-
-/**
- * Stub: get session info by session ID.
- * Returns session object if found, null otherwise.
- * TODO: Wire to actual session store when available.
- */
-export function getSessionInfo(sessionId) {
-  return null;
-}

package/open-sse/utils/sleep.js

@@ -1,6 +0,0 @@
-function sleep(ms) {
-  return new Promise((resolve) => setTimeout(resolve, ms));
-}
-export {
-  sleep
-};

package/open-sse/utils/sse.js

@@ -1,14 +0,0 @@
-export function sseChunk(data) {
-  return `data: ${JSON.stringify(data)}\n\n`;
-}
-
-// Build OpenAI chat.completion.chunk SSE frame. Key order: id, object, created, model, choices.
-export function chatChunkSse({ id, created, model, delta, finishReason = null }) {
-  return sseChunk({
-    id,
-    object: "chat.completion.chunk",
-    created,
-    model,
-    choices: [{ index: 0, delta, finish_reason: finishReason }],
-  });
-}

package/open-sse/utils/sseConstants.js

@@ -1,23 +0,0 @@
-// Shared SSE primitives (no imports → safe for executors + stream.js)
-export const SSE_DONE = "data: [DONE]\n\n";
-
-export const SSE_HEADERS = {
-  "Content-Type": "text/event-stream",
-  "Cache-Control": "no-cache",
-  "Connection": "keep-alive"
-};
-
-// Variant for web-cookie executors behind nginx (disable proxy buffering)
-export const SSE_HEADERS_NO_BUFFER = {
-  "Content-Type": "text/event-stream",
-  "Cache-Control": "no-cache",
-  "X-Accel-Buffering": "no"
-};
-
-// Variant for client-facing SSE responses (adds permissive CORS)
-export const SSE_HEADERS_CORS = {
-  "Content-Type": "text/event-stream",
-  "Cache-Control": "no-cache",
-  "Connection": "keep-alive",
-  "Access-Control-Allow-Origin": "*"
-};

package/open-sse/utils/sseHeartbeat.js

@@ -1,104 +0,0 @@
-export const DEFAULT_SSE_HEARTBEAT_INTERVAL_MS = 15_000;
-
-export const HEARTBEAT_SHAPES = {
-  COMMENT: "comment",
-  ANTHROPIC_PING: "anthropic-ping",
-  OPENAI_CHUNK: "openai-chunk",
-  OPENAI_RESPONSES_IN_PROGRESS: "openai-responses-in-progress",
-};
-
-export const DEFAULT_SSE_HEARTBEAT_SHAPE = HEARTBEAT_SHAPES.COMMENT;
-
-export function shapeForClientFormat(clientResponseFormat) {
-  switch (clientResponseFormat) {
-    case "claude":
-      return HEARTBEAT_SHAPES.ANTHROPIC_PING;
-    case "openai":
-      return HEARTBEAT_SHAPES.OPENAI_CHUNK;
-    case "openai-responses":
-      return HEARTBEAT_SHAPES.OPENAI_RESPONSES_IN_PROGRESS;
-    default:
-      return HEARTBEAT_SHAPES.COMMENT;
-  }
-}
-
-function buildHeartbeatPayload(shape, opts = {}) {
-  switch (shape) {
-    case HEARTBEAT_SHAPES.ANTHROPIC_PING:
-      return "event: ping\ndata: {}\n\n";
-    case HEARTBEAT_SHAPES.OPENAI_RESPONSES_IN_PROGRESS:
-      return 'data: {"type":"response.in_progress"}\n\n';
-    case HEARTBEAT_SHAPES.OPENAI_CHUNK: {
-      const payload = {
-        id: opts.chunkId ?? "omniroute-keepalive",
-        object: "chat.completion.chunk",
-        created: Math.floor(Date.now() / 1000),
-        model: opts.chunkModel ?? "omniroute",
-        choices: [{ index: 0, delta: {}, finish_reason: null }],
-      };
-      return `data: ${JSON.stringify(payload)}\n\n`;
-    }
-    case HEARTBEAT_SHAPES.COMMENT:
-    default:
-      return `: keepalive ${new Date().toISOString()}\n\n`;
-  }
-}
-
-const HEARTBEAT_ENCODER = new TextEncoder();
-
-export function createSseHeartbeatTransform({
-  intervalMs = DEFAULT_SSE_HEARTBEAT_INTERVAL_MS,
-  signal,
-  shape = DEFAULT_SSE_HEARTBEAT_SHAPE,
-  chunkId,
-  chunkModel,
-} = {}) {
-  if (!Number.isFinite(intervalMs) || intervalMs <= 0) {
-    return new TransformStream();
-  }
-
-  let intervalId;
-
-  const stop = () => {
-    if (!intervalId) return;
-    globalThis.clearInterval(intervalId);
-    intervalId = undefined;
-  };
-
-  return new TransformStream({
-    start(controller) {
-      intervalId = globalThis.setInterval(() => {
-        if (signal?.aborted) {
-          stop();
-          return;
-        }
-
-        try {
-          controller.enqueue(
-            HEARTBEAT_ENCODER.encode(buildHeartbeatPayload(shape, { chunkId, chunkModel }))
-          );
-        } catch {
-          stop();
-        }
-      }, intervalMs);
-
-      if (intervalId && typeof intervalId === "object" && "unref" in intervalId) {
-        intervalId.unref?.();
-      }
-
-      signal?.addEventListener("abort", stop, { once: true });
-    },
-
-    transform(chunk, controller) {
-      controller.enqueue(chunk);
-    },
-
-    flush() {
-      stop();
-    },
-
-    cancel() {
-      stop();
-    },
-  });
-}