mirror of
https://github.com/dtzp555-max/ocp.git
synced 2026-07-21 21:15:09 +00:00
taodengandClaude Opus 4.6
b87992fa3b
Replaces the old approach (sending a real messages API request just to read rate-limit headers) with the dedicated usage endpoint that Claude Code CLI uses. Fixes intermittent "unknown" plan usage. - GET /api/oauth/usage with Bearer auth + anthropic-beta header - Auto-refresh expired OAuth tokens via refresh_token grant - Try both keychain label formats (claude-code-credentials, Claude Code-credentials) - Zero API quota consumption for usage checks - Parse new JSON response format (five_hour/seven_day structure) Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
1376 lines
53 KiB
JavaScript
1376 lines
53 KiB
JavaScript
#!/usr/bin/env node
|
|
/**
|
|
* openclaw-claude-proxy — OpenAI-compatible proxy for Claude CLI
|
|
*
|
|
* Translates OpenAI chat/completions requests into `claude -p` CLI calls,
|
|
* letting you use your Claude Pro/Max subscription as an OpenClaw model provider.
|
|
*
|
|
* Timeout design: single CLAUDE_TIMEOUT (default 600s / 10 min).
|
|
* No separate first-byte or idle timeout — Claude tool-use causes long pauses
|
|
* in the token stream (30s-5min) that make fine-grained timeouts unreliable.
|
|
* This matches LiteLLM, OpenAI SDK, and other major LLM proxies.
|
|
*
|
|
* Env vars:
|
|
* CLAUDE_PROXY_PORT — listen port (default: 3456)
|
|
* CLAUDE_BIN — path to claude binary (default: auto-detect)
|
|
* CLAUDE_TIMEOUT — per-request timeout in ms (default: 600000)
|
|
* CLAUDE_ALLOWED_TOOLS — comma-separated tools to allow (default: expanded set)
|
|
* CLAUDE_SKIP_PERMISSIONS — "true" to bypass all permission checks (default: false)
|
|
* CLAUDE_SYSTEM_PROMPT — system prompt appended to all requests
|
|
* CLAUDE_MCP_CONFIG — path to MCP server config JSON file
|
|
* CLAUDE_SESSION_TTL — session TTL in ms (default: 3600000 = 1h)
|
|
* CLAUDE_MAX_CONCURRENT — max concurrent claude processes (default: 8)
|
|
* CLAUDE_BREAKER_THRESHOLD — failures in window before circuit opens (default: 6)
|
|
* CLAUDE_BREAKER_COOLDOWN — base ms to wait before retrying after circuit opens (default: 120000)
|
|
* CLAUDE_BREAKER_WINDOW — sliding window duration in ms (default: 300000 = 5min)
|
|
* CLAUDE_BREAKER_HALF_OPEN_MAX — max concurrent probes in half-open state (default: 2)
|
|
* PROXY_API_KEY — Bearer token for API auth (optional)
|
|
*/
|
|
import { createServer } from "node:http";
|
|
import { spawn, execFileSync } from "node:child_process";
|
|
import { randomUUID, timingSafeEqual } from "node:crypto";
|
|
import { readFileSync, accessSync, constants } from "node:fs";
|
|
import { fileURLToPath } from "node:url";
|
|
import { dirname, join } from "node:path";
|
|
import { homedir } from "node:os";
|
|
import { validateKey, recordUsage, getUsageByKey, getUsageTimeline, getRecentUsage, createKey, listKeys, revokeKey, closeDb } from "./keys.mjs";
|
|
|
|
const __dirname = dirname(fileURLToPath(import.meta.url));
|
|
const _pkg = JSON.parse(readFileSync(join(__dirname, "package.json"), "utf8"));
|
|
|
|
// ── Resolve claude binary ───────────────────────────────────────────────
|
|
// Priority: CLAUDE_BIN env > well-known paths > which lookup
|
|
// Fail-fast if not found — never start with an unresolvable binary.
|
|
function resolveClaude() {
|
|
if (process.env.CLAUDE_BIN) {
|
|
try {
|
|
accessSync(process.env.CLAUDE_BIN, constants.X_OK);
|
|
return process.env.CLAUDE_BIN;
|
|
} catch {
|
|
console.error(`FATAL: CLAUDE_BIN="${process.env.CLAUDE_BIN}" is set but not executable.`);
|
|
process.exit(1);
|
|
}
|
|
}
|
|
|
|
const candidates = [
|
|
"/opt/homebrew/bin/claude",
|
|
"/usr/local/bin/claude",
|
|
"/usr/bin/claude",
|
|
join(process.env.HOME || "", ".local/bin/claude"),
|
|
];
|
|
for (const p of candidates) {
|
|
try { accessSync(p, constants.X_OK); console.warn(`[init] CLAUDE_BIN not set, resolved to ${p}`); return p; } catch {}
|
|
}
|
|
|
|
try {
|
|
const resolved = execFileSync("which", ["claude"], { encoding: "utf8", timeout: 5000 }).trim();
|
|
if (resolved) { console.warn(`[init] CLAUDE_BIN not set, resolved via which: ${resolved}`); return resolved; }
|
|
} catch {}
|
|
|
|
console.error(
|
|
"FATAL: claude binary not found.\n" +
|
|
" Set CLAUDE_BIN=/path/to/claude or ensure claude is in PATH.\n" +
|
|
" Checked: " + candidates.join(", ")
|
|
);
|
|
process.exit(1);
|
|
}
|
|
|
|
// ── Configuration ───────────────────────────────────────────────────────
|
|
// Settings marked with `let` can be changed at runtime via PATCH /settings.
|
|
const PORT = parseInt(process.env.CLAUDE_PROXY_PORT || "3456", 10);
|
|
const CLAUDE = resolveClaude();
|
|
let TIMEOUT = parseInt(process.env.CLAUDE_TIMEOUT || "600000", 10);
|
|
const PROXY_API_KEY = process.env.PROXY_API_KEY || "";
|
|
const SKIP_PERMISSIONS = process.env.CLAUDE_SKIP_PERMISSIONS === "true";
|
|
const ALLOWED_TOOLS = (process.env.CLAUDE_ALLOWED_TOOLS ||
|
|
"Bash,Read,Write,Edit,Glob,Grep,WebSearch,WebFetch,Agent"
|
|
).split(",").map(s => s.trim()).filter(Boolean);
|
|
const SYSTEM_PROMPT = process.env.CLAUDE_SYSTEM_PROMPT || "";
|
|
const MCP_CONFIG = process.env.CLAUDE_MCP_CONFIG || "";
|
|
let SESSION_TTL = parseInt(process.env.CLAUDE_SESSION_TTL || "3600000", 10);
|
|
let MAX_CONCURRENT = parseInt(process.env.CLAUDE_MAX_CONCURRENT || "8", 10);
|
|
const BREAKER_THRESHOLD = parseInt(process.env.CLAUDE_BREAKER_THRESHOLD || "6", 10);
|
|
const BREAKER_COOLDOWN = parseInt(process.env.CLAUDE_BREAKER_COOLDOWN || "120000", 10);
|
|
const BREAKER_WINDOW = parseInt(process.env.CLAUDE_BREAKER_WINDOW || "300000", 10);
|
|
const BREAKER_HALF_OPEN_MAX = parseInt(process.env.CLAUDE_BREAKER_HALF_OPEN_MAX || "2", 10);
|
|
const BIND_ADDRESS = process.env.CLAUDE_BIND || "127.0.0.1";
|
|
const AUTH_MODE = process.env.CLAUDE_AUTH_MODE || (PROXY_API_KEY ? "shared" : "none");
|
|
const ADMIN_KEY = process.env.OCP_ADMIN_KEY || "";
|
|
|
|
if (AUTH_MODE === "shared" && !PROXY_API_KEY) {
|
|
console.warn("WARNING: AUTH_MODE=shared but PROXY_API_KEY is not set — all requests will pass unauthenticated");
|
|
}
|
|
|
|
const VERSION = _pkg.version;
|
|
const START_TIME = Date.now();
|
|
|
|
// ── Structured logging helper ───────────────────────────────────────────
|
|
function logEvent(level, event, data = {}) {
|
|
const entry = { ts: new Date().toISOString(), level, event, ...data };
|
|
if (level === "error" || level === "warn") {
|
|
console.error(JSON.stringify(entry));
|
|
} else {
|
|
console.log(JSON.stringify(entry));
|
|
}
|
|
}
|
|
|
|
// ── Circuit breaker (DISABLED) ──────────────────────────────────────────
|
|
// Disabled: CLI proxy has its own retry logic, and the breaker was causing
|
|
// cascading failures — once API got briefly slow, ALL agents lost connectivity
|
|
// for 120s+ due to the breaker rejecting every request.
|
|
// The timeout/failure tracking stubs below are kept as no-ops so callers
|
|
// don't need to be changed.
|
|
function breakerRecordSuccess(_cliModel) {}
|
|
function breakerRecordTimeout(_cliModel) {}
|
|
function getBreakerState(_cliModel) { return { state: "closed" }; }
|
|
function getBreakerSnapshot() { return { _note: "circuit breaker disabled" }; }
|
|
|
|
// Legacy constants kept for /health display
|
|
const _BREAKER_DISABLED_NOTE = "disabled";
|
|
/* Original breaker code removed — see git history for v2.5.0 implementation.
|
|
Re-enable by reverting this block if needed in the future.
|
|
Reason for disabling: CLI-proxy architecture means each request spawns a
|
|
fresh claude process. The breaker was designed for persistent API connections
|
|
where a degraded backend benefits from back-off. With CLI spawning, timeouts
|
|
are usually transient (API load, large prompts) and the breaker's 120s+
|
|
cooldown with graduated backoff made things worse, not better.
|
|
*/
|
|
|
|
|
|
// ── Model mapping ───────────────────────────────────────────────────────
|
|
// Maps request model IDs and aliases to canonical claude CLI model IDs.
|
|
const MODEL_MAP = {
|
|
"claude-opus-4-6": "claude-opus-4-6",
|
|
"claude-sonnet-4-6": "claude-sonnet-4-6",
|
|
"claude-haiku-4-5-20251001": "claude-haiku-4-5-20251001",
|
|
"claude-opus-4": "claude-opus-4-6",
|
|
"claude-haiku-4": "claude-haiku-4-5-20251001",
|
|
"claude-haiku-4-5": "claude-haiku-4-5-20251001",
|
|
"opus": "claude-opus-4-6",
|
|
"sonnet": "claude-sonnet-4-6",
|
|
"haiku": "claude-haiku-4-5-20251001",
|
|
};
|
|
|
|
const MODELS = [
|
|
{ id: "claude-opus-4-6", name: "Claude Opus 4.6" },
|
|
{ id: "claude-sonnet-4-6", name: "Claude Sonnet 4.6" },
|
|
{ id: "claude-haiku-4-5-20251001", name: "Claude Haiku 4.5" },
|
|
];
|
|
|
|
// ── Session management ──────────────────────────────────────────────────
|
|
// Maps conversation IDs (from caller) to Claude CLI session UUIDs.
|
|
// Enables --resume for multi-turn conversations, reducing token waste.
|
|
const sessions = new Map(); // conversationId → { uuid, messageCount, lastUsed, model }
|
|
|
|
const sessionCleanupInterval = setInterval(() => {
|
|
const now = Date.now();
|
|
for (const [id, s] of sessions) {
|
|
if (now - s.lastUsed > SESSION_TTL) {
|
|
sessions.delete(id);
|
|
console.log(`[session] expired ${id.slice(0, 12)}... (idle ${Math.round((now - s.lastUsed) / 60000)}m)`);
|
|
}
|
|
}
|
|
}, 60000);
|
|
|
|
// ── Active child process tracking ────────────────────────────────────────
|
|
const activeProcesses = new Set();
|
|
|
|
// ── Stats & diagnostics ─────────────────────────────────────────────────
|
|
const stats = {
|
|
totalRequests: 0,
|
|
activeRequests: 0,
|
|
errors: 0,
|
|
timeouts: 0,
|
|
sessionHits: 0,
|
|
sessionMisses: 0,
|
|
oneOffRequests: 0,
|
|
};
|
|
const recentErrors = []; // last 20 errors
|
|
|
|
// Per-model request stats
|
|
const modelStats = new Map(); // cliModel → { requests, errors, timeouts, totalElapsed, maxElapsed, totalPromptChars, maxPromptChars }
|
|
|
|
function getModelStats(cliModel) {
|
|
if (!modelStats.has(cliModel)) {
|
|
modelStats.set(cliModel, {
|
|
requests: 0, successes: 0, errors: 0, timeouts: 0,
|
|
totalElapsed: 0, maxElapsed: 0,
|
|
totalPromptChars: 0, maxPromptChars: 0,
|
|
});
|
|
}
|
|
return modelStats.get(cliModel);
|
|
}
|
|
|
|
function recordModelRequest(cliModel, promptChars) {
|
|
const m = getModelStats(cliModel);
|
|
m.requests++;
|
|
m.totalPromptChars += promptChars;
|
|
if (promptChars > m.maxPromptChars) m.maxPromptChars = promptChars;
|
|
}
|
|
|
|
function recordModelSuccess(cliModel, elapsedMs) {
|
|
const m = getModelStats(cliModel);
|
|
m.successes++;
|
|
m.totalElapsed += elapsedMs;
|
|
if (elapsedMs > m.maxElapsed) m.maxElapsed = elapsedMs;
|
|
}
|
|
|
|
function recordModelError(cliModel, isTimeout) {
|
|
const m = getModelStats(cliModel);
|
|
m.errors++;
|
|
if (isTimeout) m.timeouts++;
|
|
}
|
|
|
|
function getModelStatsSnapshot() {
|
|
const result = {};
|
|
for (const [model, m] of modelStats) {
|
|
result[model] = {
|
|
requests: m.requests,
|
|
successes: m.successes,
|
|
errors: m.errors,
|
|
timeouts: m.timeouts,
|
|
avgElapsed: m.successes > 0 ? Math.round(m.totalElapsed / m.successes) : 0,
|
|
maxElapsed: m.maxElapsed,
|
|
avgPromptChars: m.requests > 0 ? Math.round(m.totalPromptChars / m.requests) : 0,
|
|
maxPromptChars: m.maxPromptChars,
|
|
};
|
|
}
|
|
return result;
|
|
}
|
|
|
|
function trackError(msg) {
|
|
stats.errors++;
|
|
recentErrors.push({ time: new Date().toISOString(), message: String(msg).slice(0, 200) });
|
|
if (recentErrors.length > 20) recentErrors.shift();
|
|
}
|
|
|
|
// ── Auth health check ───────────────────────────────────────────────────
|
|
let authStatus = { ok: null, lastCheck: 0, message: "" };
|
|
|
|
async function checkAuth() {
|
|
try {
|
|
const env = { ...process.env };
|
|
delete env.CLAUDECODE;
|
|
delete env.ANTHROPIC_API_KEY;
|
|
delete env.ANTHROPIC_BASE_URL;
|
|
delete env.ANTHROPIC_AUTH_TOKEN;
|
|
execFileSync(CLAUDE, ["auth", "status"], { encoding: "utf8", timeout: 10000, env });
|
|
authStatus = { ok: true, lastCheck: Date.now(), message: "authenticated" };
|
|
} catch (e) {
|
|
const msg = (e.stderr || e.message || "").slice(0, 200);
|
|
authStatus = { ok: false, lastCheck: Date.now(), message: msg };
|
|
console.error(`[auth] check failed: ${msg}`);
|
|
}
|
|
}
|
|
|
|
// Check auth on start and every 10 minutes
|
|
checkAuth();
|
|
const authCheckInterval = setInterval(checkAuth, 600000);
|
|
|
|
// ── Build CLI arguments ─────────────────────────────────────────────────
|
|
function buildCliArgs(cliModel, sessionInfo) {
|
|
const args = ["-p", "--model", cliModel, "--output-format", "text"];
|
|
|
|
// Session handling
|
|
if (sessionInfo?.resume) {
|
|
args.push("--resume", sessionInfo.uuid);
|
|
} else if (sessionInfo?.uuid) {
|
|
args.push("--session-id", sessionInfo.uuid);
|
|
} else {
|
|
args.push("--no-session-persistence");
|
|
}
|
|
|
|
// Permissions
|
|
if (SKIP_PERMISSIONS) {
|
|
args.push("--dangerously-skip-permissions");
|
|
} else if (ALLOWED_TOOLS.length > 0) {
|
|
args.push("--allowedTools", ...ALLOWED_TOOLS);
|
|
}
|
|
|
|
// System prompt
|
|
if (SYSTEM_PROMPT) {
|
|
args.push("--append-system-prompt", SYSTEM_PROMPT);
|
|
}
|
|
|
|
// MCP config
|
|
if (MCP_CONFIG) {
|
|
args.push("--mcp-config", MCP_CONFIG);
|
|
}
|
|
|
|
return args;
|
|
}
|
|
|
|
// ── Format messages to prompt text ──────────────────────────────────────
|
|
// Truncation guard: if total chars exceed MAX_PROMPT_CHARS, keep the system
|
|
// message(s) + first user message + last N messages, dropping the middle.
|
|
// This prevents runaway context from gateway-side conversation accumulation.
|
|
let MAX_PROMPT_CHARS = parseInt(process.env.CLAUDE_MAX_PROMPT_CHARS || "150000", 10);
|
|
|
|
function messagesToPrompt(messages) {
|
|
const full = messages.map((m) => {
|
|
const text = typeof m.content === "string" ? m.content : JSON.stringify(m.content);
|
|
if (m.role === "system") return `[System] ${text}`;
|
|
if (m.role === "assistant") return `[Assistant] ${text}`;
|
|
return text;
|
|
});
|
|
|
|
const joined = full.join("\n\n");
|
|
if (joined.length <= MAX_PROMPT_CHARS) return joined;
|
|
|
|
// Truncation: keep system messages, first user msg, and trim from the tail
|
|
logEvent("warn", "prompt_truncated", {
|
|
originalChars: joined.length,
|
|
maxChars: MAX_PROMPT_CHARS,
|
|
originalMessages: messages.length,
|
|
});
|
|
|
|
const system = [];
|
|
const rest = [];
|
|
for (let i = 0; i < full.length; i++) {
|
|
if (messages[i].role === "system") system.push(full[i]);
|
|
else rest.push(full[i]);
|
|
}
|
|
|
|
// Keep system + as many recent messages as fit
|
|
const systemText = system.join("\n\n");
|
|
const budget = MAX_PROMPT_CHARS - systemText.length - 200; // 200 for separator
|
|
const kept = [];
|
|
let used = 0;
|
|
for (let i = rest.length - 1; i >= 0; i--) {
|
|
if (used + rest[i].length + 2 > budget) break;
|
|
kept.unshift(rest[i]);
|
|
used += rest[i].length + 2;
|
|
}
|
|
|
|
const truncNote = `[System] Note: ${rest.length - kept.length} older messages were truncated to fit context limit.`;
|
|
const result = [systemText, truncNote, ...kept].filter(Boolean).join("\n\n");
|
|
|
|
logEvent("info", "prompt_after_truncation", {
|
|
chars: result.length,
|
|
keptMessages: kept.length,
|
|
droppedMessages: rest.length - kept.length,
|
|
});
|
|
|
|
return result;
|
|
}
|
|
|
|
// Model tier — used for logging only (no timeout logic).
|
|
function getModelTier(cliModel) {
|
|
if (cliModel.includes("opus")) return "opus";
|
|
if (cliModel.includes("haiku")) return "haiku";
|
|
return "sonnet";
|
|
}
|
|
|
|
// ── Spawn claude CLI (shared setup) ─────────────────────────────────────
|
|
// Resolves session logic, builds CLI args, spawns the process, and sets up
|
|
// timeouts. Returns context object or throws synchronously.
|
|
function spawnClaudeProcess(model, messages, conversationId) {
|
|
if (stats.activeRequests >= MAX_CONCURRENT) {
|
|
throw new Error(`concurrency limit reached (${stats.activeRequests}/${MAX_CONCURRENT})`);
|
|
}
|
|
|
|
const cliModel = MODEL_MAP[model] || model;
|
|
|
|
// Circuit breaker: disabled (see comment at top of breaker section)
|
|
|
|
stats.activeRequests++;
|
|
stats.totalRequests++;
|
|
|
|
let sessionInfo = null;
|
|
let prompt;
|
|
|
|
// ── Session logic ──
|
|
if (conversationId && sessions.has(conversationId)) {
|
|
const session = sessions.get(conversationId);
|
|
session.lastUsed = Date.now();
|
|
sessionInfo = { uuid: session.uuid, resume: true };
|
|
stats.sessionHits++;
|
|
|
|
const lastUserMsg = [...messages].reverse().find((m) => m.role === "user");
|
|
prompt = lastUserMsg
|
|
? (typeof lastUserMsg.content === "string" ? lastUserMsg.content : JSON.stringify(lastUserMsg.content))
|
|
: "";
|
|
session.messageCount = messages.length;
|
|
|
|
console.log(`[session] resume conv=${conversationId.slice(0, 12)}... uuid=${session.uuid.slice(0, 8)}... msgs=${messages.length} prompt_chars=${prompt.length}`);
|
|
|
|
} else if (conversationId) {
|
|
const uuid = randomUUID();
|
|
sessions.set(conversationId, { uuid, messageCount: messages.length, lastUsed: Date.now(), model: cliModel });
|
|
sessionInfo = { uuid, resume: false };
|
|
stats.sessionMisses++;
|
|
prompt = messagesToPrompt(messages);
|
|
|
|
console.log(`[session] new conv=${conversationId.slice(0, 12)}... uuid=${uuid.slice(0, 8)}... msgs=${messages.length}`);
|
|
|
|
} else {
|
|
stats.oneOffRequests++;
|
|
prompt = messagesToPrompt(messages);
|
|
}
|
|
|
|
const cliArgs = buildCliArgs(cliModel, sessionInfo);
|
|
|
|
const env = { ...process.env };
|
|
delete env.CLAUDECODE;
|
|
delete env.ANTHROPIC_API_KEY;
|
|
delete env.ANTHROPIC_BASE_URL;
|
|
delete env.ANTHROPIC_AUTH_TOKEN;
|
|
|
|
const proc = spawn(CLAUDE, cliArgs, { env, stdio: ["pipe", "pipe", "pipe"] });
|
|
activeProcesses.add(proc);
|
|
|
|
const t0 = Date.now();
|
|
let gotFirstByte = false;
|
|
let cleaned = false;
|
|
|
|
function cleanup() {
|
|
if (cleaned) return;
|
|
cleaned = true;
|
|
clearTimeout(overallTimer);
|
|
stats.activeRequests--;
|
|
}
|
|
|
|
function handleSessionFailure() {
|
|
if (sessionInfo?.resume && conversationId) {
|
|
console.warn(`[session] resume failed for ${conversationId.slice(0, 12)}..., removing stale session`);
|
|
sessions.delete(conversationId);
|
|
}
|
|
}
|
|
|
|
function markFirstByte() {
|
|
if (!gotFirstByte) {
|
|
gotFirstByte = true;
|
|
console.log(`[claude] first-byte model=${cliModel} elapsed=${Date.now() - t0}ms`);
|
|
}
|
|
}
|
|
|
|
// Write prompt to stdin immediately
|
|
proc.stdin.write(prompt);
|
|
proc.stdin.end();
|
|
|
|
recordModelRequest(cliModel, prompt.length);
|
|
logEvent("info", "claude_spawned", { model: cliModel, promptChars: prompt.length, timeout: TIMEOUT, tier: getModelTier(cliModel), session: conversationId ? conversationId.slice(0, 12) + "..." : "none" });
|
|
|
|
// Single request timeout — no separate first-byte timer.
|
|
// Claude tool-use causes long pauses in the token stream (30s-5min),
|
|
// making first-byte/idle timeouts unreliable. One generous timeout is simpler and correct.
|
|
const overallTimer = setTimeout(() => {
|
|
if (!cleaned) {
|
|
stats.timeouts++;
|
|
recordModelError(cliModel, true);
|
|
breakerRecordTimeout(cliModel);
|
|
logEvent("error", "request_timeout", { model: cliModel, timeoutMs: TIMEOUT, elapsed: Date.now() - t0 });
|
|
try { proc.kill("SIGTERM"); } catch {}
|
|
setTimeout(() => { try { proc.kill("SIGKILL"); } catch {} }, 5000);
|
|
}
|
|
}, TIMEOUT);
|
|
|
|
return { proc, cliModel, conversationId, t0, cleanup, handleSessionFailure, markFirstByte };
|
|
}
|
|
|
|
// ── Call claude CLI (non-streaming) ─────────────────────────────────────
|
|
// On-demand spawning: each request spawns a fresh `claude -p` process.
|
|
// No pool = no crash loops, no stale workers, no degraded states.
|
|
// Stdin is written immediately so there's no 3s stdin timeout issue.
|
|
function callClaude(model, messages, conversationId) {
|
|
return new Promise((resolve, reject) => {
|
|
let ctx;
|
|
try {
|
|
ctx = spawnClaudeProcess(model, messages, conversationId);
|
|
} catch (err) {
|
|
return reject(err);
|
|
}
|
|
|
|
const { proc, cliModel, conversationId: convId, t0, cleanup, handleSessionFailure, markFirstByte } = ctx;
|
|
let stdout = "";
|
|
let stderr = "";
|
|
|
|
proc.stdout.on("data", (d) => {
|
|
markFirstByte();
|
|
stdout += d;
|
|
});
|
|
proc.stderr.on("data", (d) => (stderr += d));
|
|
|
|
proc.on("close", (code, signal) => {
|
|
activeProcesses.delete(proc);
|
|
const elapsed = Date.now() - t0;
|
|
cleanup();
|
|
if (code !== 0) {
|
|
recordModelError(cliModel, false);
|
|
logEvent("error", "claude_exit", { model: cliModel, code, signal: signal || "none", elapsed, stderr: stderr.slice(0, 300) });
|
|
trackError(stderr.slice(0, 300) || stdout.slice(0, 300) || `claude exit ${code}`);
|
|
handleSessionFailure();
|
|
reject(new Error(stderr.slice(0, 300) || stdout.slice(0, 300) || `claude exit ${code}`));
|
|
} else {
|
|
recordModelSuccess(cliModel, elapsed);
|
|
breakerRecordSuccess(cliModel);
|
|
logEvent("info", "claude_ok", { model: cliModel, chars: stdout.length, elapsed, session: convId ? convId.slice(0, 12) + "..." : "none" });
|
|
resolve(stdout.trim());
|
|
}
|
|
});
|
|
|
|
proc.on("error", (err) => {
|
|
console.error(`[claude] spawn error: ${err.message}`);
|
|
cleanup();
|
|
trackError(err.message);
|
|
handleSessionFailure();
|
|
reject(err);
|
|
});
|
|
});
|
|
}
|
|
|
|
// ── Call claude CLI (real streaming) ─────────────────────────────────────
|
|
// Pipes stdout from the claude process directly to SSE chunks as they arrive.
|
|
// Each data chunk becomes a proper SSE event with delta content in real time.
|
|
function callClaudeStreaming(model, messages, conversationId, res, authInfo = {}) {
|
|
const id = `chatcmpl-${randomUUID()}`;
|
|
const created = Math.floor(Date.now() / 1000);
|
|
|
|
let ctx;
|
|
try {
|
|
ctx = spawnClaudeProcess(model, messages, conversationId);
|
|
} catch (err) {
|
|
return jsonResponse(res, 500, { error: { message: err.message, type: "proxy_error" } });
|
|
}
|
|
|
|
const { proc, cliModel, conversationId: convId, t0, cleanup, handleSessionFailure, markFirstByte } = ctx;
|
|
let stderr = "";
|
|
let headersSent = false;
|
|
let totalChars = 0;
|
|
|
|
function ensureHeaders() {
|
|
if (headersSent || res.writableEnded || res.destroyed) return false;
|
|
headersSent = true;
|
|
res.writeHead(200, {
|
|
"Content-Type": "text/event-stream",
|
|
"Cache-Control": "no-cache",
|
|
"Connection": "keep-alive",
|
|
});
|
|
// Send initial role chunk
|
|
sendSSE(res, {
|
|
id, object: "chat.completion.chunk", created, model,
|
|
choices: [{ index: 0, delta: { role: "assistant" }, finish_reason: null }],
|
|
});
|
|
return true;
|
|
}
|
|
|
|
proc.stdout.on("data", (d) => {
|
|
markFirstByte();
|
|
const text = d.toString();
|
|
totalChars += text.length;
|
|
|
|
if (!ensureHeaders()) return;
|
|
|
|
// Stream each chunk as it arrives from the CLI process
|
|
sendSSE(res, {
|
|
id, object: "chat.completion.chunk", created, model,
|
|
choices: [{ index: 0, delta: { content: text }, finish_reason: null }],
|
|
});
|
|
});
|
|
|
|
proc.stderr.on("data", (d) => (stderr += d));
|
|
|
|
proc.on("close", (code, signal) => {
|
|
activeProcesses.delete(proc);
|
|
cleanup();
|
|
const elapsed = Date.now() - t0;
|
|
|
|
if (code !== 0) {
|
|
recordModelError(cliModel, false);
|
|
try { recordUsage({ keyId: authInfo.keyId, keyName: authInfo.keyName, model, promptChars: messages.reduce((a, m) => a + (typeof m.content === "string" ? m.content.length : JSON.stringify(m.content).length), 0), responseChars: 0, elapsedMs: elapsed, success: false }); } catch (e) { logEvent("error", "usage_record_failed", { error: e.message }); }
|
|
logEvent("error", "claude_exit", { model: cliModel, code, signal: signal || "none", elapsed, stderr: stderr.slice(0, 300) });
|
|
trackError(stderr.slice(0, 300) || `claude exit ${code}`);
|
|
handleSessionFailure();
|
|
|
|
if (!headersSent && !res.writableEnded && !res.destroyed) {
|
|
jsonResponse(res, 500, { error: { message: stderr.slice(0, 300) || `claude exit ${code}`, type: "proxy_error" } });
|
|
} else if (!res.writableEnded && !res.destroyed) {
|
|
sendSSE(res, {
|
|
id, object: "chat.completion.chunk", created, model,
|
|
choices: [{ index: 0, delta: {}, finish_reason: "stop" }],
|
|
});
|
|
res.write("data: [DONE]\n\n");
|
|
res.end();
|
|
}
|
|
} else {
|
|
recordModelSuccess(cliModel, elapsed);
|
|
breakerRecordSuccess(cliModel);
|
|
try { recordUsage({ keyId: authInfo.keyId, keyName: authInfo.keyName, model, promptChars: messages.reduce((a, m) => a + (typeof m.content === "string" ? m.content.length : JSON.stringify(m.content).length), 0), responseChars: totalChars, elapsedMs: elapsed, success: true }); } catch (e) { logEvent("error", "usage_record_failed", { error: e.message }); }
|
|
logEvent("info", "claude_ok", { model: cliModel, chars: totalChars, elapsed, session: convId ? convId.slice(0, 12) + "..." : "none" });
|
|
|
|
if (!headersSent) ensureHeaders();
|
|
if (!res.writableEnded && !res.destroyed) {
|
|
sendSSE(res, {
|
|
id, object: "chat.completion.chunk", created, model,
|
|
choices: [{ index: 0, delta: {}, finish_reason: "stop" }],
|
|
});
|
|
res.write("data: [DONE]\n\n");
|
|
res.end();
|
|
}
|
|
}
|
|
});
|
|
|
|
proc.on("error", (err) => {
|
|
console.error(`[claude] spawn error: ${err.message}`);
|
|
cleanup();
|
|
trackError(err.message);
|
|
handleSessionFailure();
|
|
if (!headersSent && !res.writableEnded && !res.destroyed) {
|
|
jsonResponse(res, 500, { error: { message: err.message, type: "proxy_error" } });
|
|
} else if (!res.writableEnded && !res.destroyed) {
|
|
res.end();
|
|
}
|
|
});
|
|
|
|
// If client disconnects, kill the process to free resources
|
|
res.on("close", () => {
|
|
if (!proc.killed) {
|
|
try { proc.kill("SIGTERM"); } catch {}
|
|
}
|
|
});
|
|
}
|
|
|
|
// ── Response helpers ────────────────────────────────────────────────────
|
|
function jsonResponse(res, status, data) {
|
|
if (res.headersSent || res.writableEnded || res.destroyed) return;
|
|
res.writeHead(status, { "Content-Type": "application/json" });
|
|
res.end(JSON.stringify(data));
|
|
}
|
|
|
|
function sendSSE(res, data) {
|
|
res.write(`data: ${JSON.stringify(data)}\n\n`);
|
|
}
|
|
|
|
function completionResponse(res, id, model, content) {
|
|
jsonResponse(res, 200, {
|
|
id, object: "chat.completion",
|
|
created: Math.floor(Date.now() / 1000),
|
|
model,
|
|
choices: [{ index: 0, message: { role: "assistant", content }, finish_reason: "stop" }],
|
|
usage: { prompt_tokens: 0, completion_tokens: 0, total_tokens: 0 },
|
|
});
|
|
}
|
|
|
|
// ── Plan usage probe ────────────────────────────────────────────────────
|
|
// Uses the dedicated /api/oauth/usage endpoint (same as Claude Code CLI)
|
|
// with Bearer auth + anthropic-beta header. Auto-refreshes expired tokens.
|
|
// Caches the result for 5 minutes to avoid excessive API calls.
|
|
|
|
let usageCache = { data: null, fetchedAt: 0 };
|
|
const USAGE_CACHE_TTL = 300000; // 5 min
|
|
const OAUTH_CLIENT_ID = "9d1c250a-e61b-44d9-88ed-5944d1962f5e";
|
|
const OAUTH_TOKEN_URL = "https://platform.claude.com/v1/oauth/token";
|
|
const OAUTH_BETA_HEADER = "oauth-2025-04-20";
|
|
|
|
function getOAuthCredentials() {
|
|
// Try Linux file-based credentials first
|
|
try {
|
|
const credPath = join(homedir(), ".claude", ".credentials.json");
|
|
const creds = JSON.parse(readFileSync(credPath, "utf8"));
|
|
if (creds?.claudeAiOauth?.accessToken) return creds.claudeAiOauth;
|
|
} catch { /* fall through to macOS keychain */ }
|
|
|
|
// Try macOS keychain (both label formats)
|
|
for (const label of ["claude-code-credentials", "Claude Code-credentials"]) {
|
|
try {
|
|
const raw = execFileSync("security", [
|
|
"find-generic-password", "-s", label, "-w"
|
|
], { encoding: "utf8", timeout: 5000 }).trim();
|
|
const creds = JSON.parse(raw);
|
|
if (creds?.claudeAiOauth?.accessToken) return creds.claudeAiOauth;
|
|
} catch { /* try next */ }
|
|
}
|
|
return null;
|
|
}
|
|
|
|
async function refreshOAuthToken(refreshToken) {
|
|
try {
|
|
const resp = await fetch(OAUTH_TOKEN_URL, {
|
|
method: "POST",
|
|
headers: { "Content-Type": "application/json" },
|
|
body: JSON.stringify({
|
|
grant_type: "refresh_token",
|
|
refresh_token: refreshToken,
|
|
client_id: OAUTH_CLIENT_ID,
|
|
scope: "user:inference user:profile",
|
|
}),
|
|
});
|
|
if (!resp.ok) {
|
|
const body = await resp.text();
|
|
logEvent("warn", "oauth_refresh_failed", { status: resp.status, body: body.slice(0, 200) });
|
|
return null;
|
|
}
|
|
const data = await resp.json();
|
|
return data.access_token || null;
|
|
} catch (err) {
|
|
logEvent("warn", "oauth_refresh_error", { error: err.message });
|
|
return null;
|
|
}
|
|
}
|
|
|
|
async function fetchUsageFromApi() {
|
|
const creds = getOAuthCredentials();
|
|
if (!creds?.accessToken) {
|
|
return { error: "No OAuth token found in keychain" };
|
|
}
|
|
|
|
let token = creds.accessToken;
|
|
|
|
// Check if token looks expired (5 min buffer, same as Claude Code)
|
|
if (creds.expiresAt && Date.now() + 300000 >= creds.expiresAt) {
|
|
if (creds.refreshToken) {
|
|
logEvent("info", "oauth_token_expired_refreshing");
|
|
const newToken = await refreshOAuthToken(creds.refreshToken);
|
|
if (newToken) token = newToken;
|
|
}
|
|
}
|
|
|
|
const controller = new AbortController();
|
|
const timeout = setTimeout(() => controller.abort(), 10000);
|
|
|
|
try {
|
|
const resp = await fetch("https://api.anthropic.com/api/oauth/usage", {
|
|
method: "GET",
|
|
headers: {
|
|
"Authorization": `Bearer ${token}`,
|
|
"anthropic-beta": OAUTH_BETA_HEADER,
|
|
"Content-Type": "application/json",
|
|
},
|
|
signal: controller.signal,
|
|
});
|
|
clearTimeout(timeout);
|
|
|
|
if (!resp.ok) {
|
|
// If 401, try refreshing token once
|
|
if (resp.status === 401 && creds.refreshToken) {
|
|
logEvent("info", "oauth_usage_401_refreshing");
|
|
const newToken = await refreshOAuthToken(creds.refreshToken);
|
|
if (newToken) {
|
|
const retryResp = await fetch("https://api.anthropic.com/api/oauth/usage", {
|
|
method: "GET",
|
|
headers: {
|
|
"Authorization": `Bearer ${newToken}`,
|
|
"anthropic-beta": OAUTH_BETA_HEADER,
|
|
"Content-Type": "application/json",
|
|
},
|
|
});
|
|
if (retryResp.ok) {
|
|
const retryData = await retryResp.json();
|
|
return parseUsageResponse(retryData);
|
|
}
|
|
}
|
|
return { error: `Usage API auth failed after refresh (${resp.status})` };
|
|
}
|
|
return { error: `Usage API returned ${resp.status}` };
|
|
}
|
|
|
|
const data = await resp.json();
|
|
return parseUsageResponse(data);
|
|
} catch (err) {
|
|
clearTimeout(timeout);
|
|
return { error: `Failed to fetch usage: ${err.message}` };
|
|
}
|
|
}
|
|
|
|
function parseUsageResponse(data) {
|
|
const now = Date.now();
|
|
|
|
function formatReset(isoStr) {
|
|
if (!isoStr) return "unknown";
|
|
const diff = new Date(isoStr).getTime() - now;
|
|
if (diff <= 0) return "now";
|
|
const h = Math.floor(diff / 3600000);
|
|
const m = Math.floor((diff % 3600000) / 60000);
|
|
if (h > 24) {
|
|
const d = Math.floor(h / 24);
|
|
return `${d}d ${h % 24}h`;
|
|
}
|
|
return h > 0 ? `${h}h ${m}m` : `${m}m`;
|
|
}
|
|
|
|
function resetDay(isoStr) {
|
|
if (!isoStr) return "";
|
|
const d = new Date(isoStr);
|
|
return d.toLocaleDateString("en-US", { weekday: "short", month: "short", day: "numeric", hour: "numeric", minute: "2-digit" });
|
|
}
|
|
|
|
const fiveHour = data.five_hour || {};
|
|
const sevenDay = data.seven_day || {};
|
|
const extraUsage = data.extra_usage || {};
|
|
|
|
return {
|
|
status: "active",
|
|
fetchedAt: new Date(now).toISOString(),
|
|
plan: {
|
|
currentSession: {
|
|
utilization: (fiveHour.utilization || 0) / 100,
|
|
percent: `${Math.round(fiveHour.utilization || 0)}%`,
|
|
resetsIn: formatReset(fiveHour.resets_at),
|
|
resetsAt: fiveHour.resets_at || null,
|
|
resetsAtHuman: resetDay(fiveHour.resets_at),
|
|
},
|
|
weeklyLimits: {
|
|
allModels: {
|
|
utilization: (sevenDay.utilization || 0) / 100,
|
|
percent: `${Math.round(sevenDay.utilization || 0)}%`,
|
|
resetsIn: formatReset(sevenDay.resets_at),
|
|
resetsAt: sevenDay.resets_at || null,
|
|
resetsAtHuman: resetDay(sevenDay.resets_at),
|
|
},
|
|
},
|
|
extraUsage: {
|
|
status: extraUsage.is_enabled ? "enabled" : "disabled",
|
|
monthlyLimit: extraUsage.monthly_limit,
|
|
usedCredits: extraUsage.used_credits,
|
|
utilization: extraUsage.utilization,
|
|
},
|
|
},
|
|
proxy: {
|
|
totalRequests: stats.totalRequests,
|
|
activeRequests: stats.activeRequests,
|
|
errors: stats.errors,
|
|
timeouts: stats.timeouts,
|
|
uptime: `${Math.floor((now - START_TIME) / 3600000)}h ${Math.floor(((now - START_TIME) % 3600000) / 60000)}m`,
|
|
},
|
|
models: getModelStatsSnapshot(),
|
|
_raw: data,
|
|
};
|
|
}
|
|
|
|
async function handleUsage(_req, res) {
|
|
const now = Date.now();
|
|
let data;
|
|
if (usageCache.data && (now - usageCache.fetchedAt) < USAGE_CACHE_TTL) {
|
|
data = usageCache.data;
|
|
} else {
|
|
data = await fetchUsageFromApi();
|
|
if (!data.error) {
|
|
usageCache = { data, fetchedAt: now };
|
|
}
|
|
}
|
|
// Always attach live model stats and proxy stats (not cached)
|
|
const uptimeMs = now - START_TIME;
|
|
const response = {
|
|
...data,
|
|
proxy: {
|
|
totalRequests: stats.totalRequests,
|
|
activeRequests: stats.activeRequests,
|
|
errors: stats.errors,
|
|
timeouts: stats.timeouts,
|
|
uptime: `${Math.floor(uptimeMs / 3600000)}h ${Math.floor((uptimeMs % 3600000) / 60000)}m`,
|
|
},
|
|
models: getModelStatsSnapshot(),
|
|
};
|
|
jsonResponse(res, data.error ? 502 : 200, response);
|
|
}
|
|
|
|
// ── Logs endpoint ──────────────────────────────────────────────────────
|
|
// Returns recent structured log entries from the proxy log file.
|
|
// GET /logs?n=20&level=error (default: n=30, level=all)
|
|
function handleLogs(req, res) {
|
|
const url = new URL(req.url, `http://127.0.0.1:${PORT}`);
|
|
const n = Math.min(parseInt(url.searchParams.get("n") || "30", 10), 200);
|
|
const level = url.searchParams.get("level") || "all"; // all | error | warn | info
|
|
|
|
const LOG_PATH = join(process.env.HOME || "/tmp", ".openclaw/logs/proxy.log");
|
|
let lines;
|
|
try {
|
|
const raw = readFileSync(LOG_PATH, "utf8");
|
|
lines = raw.split("\n").filter(Boolean);
|
|
} catch (err) {
|
|
return jsonResponse(res, 500, { error: `Cannot read log: ${err.message}` });
|
|
}
|
|
|
|
// Parse JSON lines, fall back to raw text
|
|
let entries = lines.slice(-n * 3).map(line => {
|
|
try { return JSON.parse(line); } catch { return { raw: line }; }
|
|
});
|
|
|
|
// Filter by level
|
|
if (level !== "all") {
|
|
entries = entries.filter(e => {
|
|
if (e.level) return e.level === level;
|
|
if (level === "error") return e.raw?.includes("error") || e.raw?.includes("Error");
|
|
return true;
|
|
});
|
|
}
|
|
|
|
entries = entries.slice(-n);
|
|
|
|
return jsonResponse(res, 200, {
|
|
count: entries.length,
|
|
level,
|
|
entries,
|
|
});
|
|
}
|
|
|
|
// ── Status endpoint (combined summary) ─────────────────────────────────
|
|
async function handleStatus(_req, res) {
|
|
const now = Date.now();
|
|
const uptimeMs = now - START_TIME;
|
|
|
|
// Get usage (from cache if fresh)
|
|
let usage = null;
|
|
if (usageCache.data && (now - usageCache.fetchedAt) < USAGE_CACHE_TTL) {
|
|
usage = usageCache.data;
|
|
} else {
|
|
usage = await fetchUsageFromApi();
|
|
if (!usage.error) usageCache = { data: usage, fetchedAt: now };
|
|
}
|
|
|
|
// Auth
|
|
let binaryOk = false;
|
|
try { accessSync(CLAUDE, constants.X_OK); binaryOk = true; } catch {}
|
|
|
|
return jsonResponse(res, 200, {
|
|
proxy: {
|
|
status: binaryOk && authStatus.ok !== false ? "ok" : "degraded",
|
|
version: VERSION,
|
|
uptime: `${Math.floor(uptimeMs / 3600000)}h ${Math.floor((uptimeMs % 3600000) / 60000)}m`,
|
|
auth: authStatus.ok ? "ok" : authStatus.message,
|
|
activeSessions: sessions.size,
|
|
},
|
|
requests: {
|
|
total: stats.totalRequests,
|
|
active: stats.activeRequests,
|
|
errors: stats.errors,
|
|
timeouts: stats.timeouts,
|
|
},
|
|
plan: usage?.plan || usage?.error || null,
|
|
recentErrors: recentErrors.slice(-3),
|
|
});
|
|
}
|
|
|
|
// ── Settings endpoint ───────────────────────────────────────────────────
|
|
// GET /settings → view current tunable parameters
|
|
// PATCH /settings → update one or more parameters at runtime
|
|
//
|
|
// Tunable keys and their types/ranges:
|
|
const SETTINGS_SCHEMA = {
|
|
timeout: { type: "number", min: 30000, max: 1800000, unit: "ms", desc: "Request timeout (default: 600s)" },
|
|
maxConcurrent: { type: "number", min: 1, max: 32, unit: "", desc: "Max concurrent claude processes" },
|
|
sessionTTL: { type: "number", min: 60000, max: 86400000, unit: "ms", desc: "Session idle expiry" },
|
|
maxPromptChars: { type: "number", min: 10000, max: 1000000, unit: "chars", desc: "Prompt truncation limit" },
|
|
};
|
|
|
|
function getSettings() {
|
|
return {
|
|
timeout: { value: TIMEOUT, ...SETTINGS_SCHEMA.timeout },
|
|
maxConcurrent: { value: MAX_CONCURRENT, ...SETTINGS_SCHEMA.maxConcurrent },
|
|
sessionTTL: { value: SESSION_TTL, ...SETTINGS_SCHEMA.sessionTTL },
|
|
maxPromptChars: { value: MAX_PROMPT_CHARS, ...SETTINGS_SCHEMA.maxPromptChars },
|
|
};
|
|
}
|
|
|
|
function applySettingUpdate(key, value) {
|
|
const schema = SETTINGS_SCHEMA[key];
|
|
if (!schema) return `unknown setting: ${key}`;
|
|
if (typeof value !== schema.type) return `${key}: expected ${schema.type}, got ${typeof value}`;
|
|
if (value < schema.min || value > schema.max) return `${key}: value ${value} out of range [${schema.min}, ${schema.max}]`;
|
|
|
|
switch (key) {
|
|
case "timeout": TIMEOUT = value; break;
|
|
case "maxConcurrent": MAX_CONCURRENT = value; break;
|
|
case "sessionTTL": SESSION_TTL = value; break;
|
|
case "maxPromptChars": MAX_PROMPT_CHARS = value; break;
|
|
default: return `${key}: not implemented`;
|
|
}
|
|
logEvent("info", "setting_changed", { key, value });
|
|
return null; // success
|
|
}
|
|
|
|
async function handleSettings(req, res) {
|
|
if (req.method === "GET") {
|
|
return jsonResponse(res, 200, getSettings());
|
|
}
|
|
|
|
// PATCH
|
|
let body = "";
|
|
for await (const chunk of req) {
|
|
body += chunk;
|
|
if (body.length > 10000) return jsonResponse(res, 413, { error: "Body too large" });
|
|
}
|
|
let updates;
|
|
try { updates = JSON.parse(body); } catch { return jsonResponse(res, 400, { error: "Invalid JSON" }); }
|
|
|
|
if (typeof updates !== "object" || Array.isArray(updates)) {
|
|
return jsonResponse(res, 400, { error: "Expected JSON object with key-value pairs" });
|
|
}
|
|
|
|
const results = {};
|
|
const errors = [];
|
|
for (const [key, value] of Object.entries(updates)) {
|
|
const err = applySettingUpdate(key, value);
|
|
if (err) {
|
|
errors.push(err);
|
|
results[key] = { error: err };
|
|
} else {
|
|
results[key] = { ok: true, value };
|
|
}
|
|
}
|
|
|
|
const status = errors.length === 0 ? 200 : (Object.keys(results).length > errors.length ? 207 : 400);
|
|
return jsonResponse(res, status, {
|
|
results,
|
|
...(errors.length ? { errors } : {}),
|
|
current: getSettings(),
|
|
});
|
|
}
|
|
|
|
// ── Handle chat completions ─────────────────────────────────────────────
|
|
const MAX_BODY_SIZE = 5 * 1024 * 1024; // 5 MB
|
|
|
|
// Set of all valid model identifiers (canonical IDs + aliases)
|
|
const VALID_MODELS = new Set(Object.keys(MODEL_MAP));
|
|
|
|
async function handleChatCompletions(req, res) {
|
|
let body = "";
|
|
for await (const chunk of req) {
|
|
body += chunk;
|
|
if (body.length > MAX_BODY_SIZE) {
|
|
return jsonResponse(res, 413, { error: { message: "Request body too large (max 5MB)", type: "invalid_request_error" } });
|
|
}
|
|
}
|
|
|
|
let parsed;
|
|
try { parsed = JSON.parse(body); } catch { return jsonResponse(res, 400, { error: "Invalid JSON" }); }
|
|
|
|
const messages = parsed.messages || parsed.input || [{ role: "user", content: parsed.prompt || "" }];
|
|
const model = parsed.model || "claude-sonnet-4-6";
|
|
const stream = parsed.stream;
|
|
|
|
// Validate model against known models
|
|
if (!VALID_MODELS.has(model)) {
|
|
return jsonResponse(res, 400, { error: { message: `Unknown model: ${model}. Valid models: ${[...VALID_MODELS].join(", ")}`, type: "invalid_request_error" } });
|
|
}
|
|
|
|
// Session ID: from request body, header, or null (one-off)
|
|
const conversationId = parsed.session_id || parsed.conversation_id || req.headers["x-session-id"] || req.headers["x-conversation-id"] || null;
|
|
|
|
if (!messages?.length) return jsonResponse(res, 400, { error: "messages required" });
|
|
|
|
if (stream) {
|
|
// Real streaming: pipe stdout from claude process directly as SSE chunks
|
|
return callClaudeStreaming(model, messages, conversationId, res, { keyId: req._authKeyId, keyName: req._authKeyName });
|
|
}
|
|
|
|
const t0Usage = Date.now();
|
|
const promptChars = messages.reduce((a, m) => a + (typeof m.content === "string" ? m.content.length : JSON.stringify(m.content).length), 0);
|
|
try {
|
|
const content = await callClaude(model, messages, conversationId);
|
|
const id = `chatcmpl-${randomUUID()}`;
|
|
completionResponse(res, id, model, content);
|
|
try { recordUsage({ keyId: req._authKeyId, keyName: req._authKeyName, model, promptChars, responseChars: content.length, elapsedMs: Date.now() - t0Usage, success: true }); } catch (e) { logEvent("error", "usage_record_failed", { error: e.message }); }
|
|
} catch (err) {
|
|
try { recordUsage({ keyId: req._authKeyId, keyName: req._authKeyName, model, promptChars, responseChars: 0, elapsedMs: Date.now() - t0Usage, success: false }); } catch (e) { logEvent("error", "usage_record_failed", { error: e.message }); }
|
|
console.error(`[proxy] error: ${err.message}`);
|
|
if (res.headersSent || res.writableEnded || res.destroyed) {
|
|
try { res.end(); } catch {}
|
|
return;
|
|
}
|
|
// Sanitize error: strip internal file paths before sending to client
|
|
const safeMessage = (err.message || "Internal error").replace(/\/[\w/.\-]+/g, "[path]");
|
|
jsonResponse(res, 500, { error: { message: safeMessage, type: "proxy_error" } });
|
|
}
|
|
}
|
|
|
|
// ── HTTP server ─────────────────────────────────────────────────────────
|
|
const server = createServer(async (req, res) => {
|
|
// Dynamic CORS: allow localhost and LAN origins
|
|
const origin = req.headers["origin"] || "";
|
|
const isAllowedOrigin = /^https?:\/\/(127\.0\.0\.1|localhost|192\.168\.\d+\.\d+|172\.(1[6-9]|2\d|3[01])\.\d+\.\d+|10\.\d+\.\d+\.\d+)(:\d+)?$/.test(origin);
|
|
res.setHeader("Access-Control-Allow-Origin", isAllowedOrigin ? origin : `http://127.0.0.1:${PORT}`);
|
|
res.setHeader("Access-Control-Allow-Methods", "GET, POST, DELETE, OPTIONS, PATCH");
|
|
res.setHeader("Access-Control-Allow-Headers", "Content-Type, Authorization, X-Session-Id, X-Conversation-Id");
|
|
if (req.method === "OPTIONS") { res.writeHead(204); res.end(); return; }
|
|
|
|
// 3-mode auth: none | shared | multi
|
|
const pathname = req.url.split("?")[0];
|
|
const isPublicEndpoint = pathname === "/health" || pathname === "/dashboard";
|
|
const remoteAddr = req.socket.remoteAddress || "";
|
|
const isLocalhost = remoteAddr === "127.0.0.1" || remoteAddr === "::1" || remoteAddr === "::ffff:127.0.0.1";
|
|
let authKeyName = isLocalhost ? "local" : "remote";
|
|
let authKeyId = null;
|
|
|
|
if (!isPublicEndpoint) {
|
|
const auth = req.headers["authorization"] || "";
|
|
const token = auth.startsWith("Bearer ") ? auth.slice(7) : "";
|
|
|
|
if (isLocalhost) {
|
|
// Localhost always allowed — try to identify key if provided, but never reject
|
|
if (token) {
|
|
if (ADMIN_KEY) {
|
|
const adminBuf = Buffer.from(ADMIN_KEY);
|
|
const tokenBuf = Buffer.from(token);
|
|
if (adminBuf.length === tokenBuf.length && timingSafeEqual(adminBuf, tokenBuf)) {
|
|
authKeyName = "admin";
|
|
}
|
|
}
|
|
if (authKeyName !== "admin") {
|
|
const keyInfo = validateKey(token);
|
|
if (keyInfo) { authKeyName = keyInfo.name; authKeyId = keyInfo.id; }
|
|
}
|
|
}
|
|
} else if (AUTH_MODE === "shared") {
|
|
if (PROXY_API_KEY) {
|
|
const tokenBuf = Buffer.from(token);
|
|
const keyBuf = Buffer.from(PROXY_API_KEY);
|
|
if (tokenBuf.length !== keyBuf.length || !timingSafeEqual(tokenBuf, keyBuf)) {
|
|
return jsonResponse(res, 401, { error: { message: "Unauthorized: invalid or missing Bearer token", type: "auth_error" } });
|
|
}
|
|
authKeyName = "shared";
|
|
}
|
|
} else if (AUTH_MODE === "multi") {
|
|
// If a token is provided, validate it; if not, allow as anonymous
|
|
if (token) {
|
|
let isAdminToken = false;
|
|
if (ADMIN_KEY) {
|
|
const adminBuf = Buffer.from(ADMIN_KEY);
|
|
const tokenBuf2 = Buffer.from(token);
|
|
if (adminBuf.length === tokenBuf2.length && timingSafeEqual(adminBuf, tokenBuf2)) {
|
|
authKeyName = "admin";
|
|
isAdminToken = true;
|
|
}
|
|
}
|
|
if (!isAdminToken) {
|
|
const keyInfo = validateKey(token);
|
|
if (!keyInfo) {
|
|
return jsonResponse(res, 401, { error: { message: "Unauthorized: invalid or revoked API key", type: "auth_error" } });
|
|
}
|
|
authKeyName = keyInfo.name;
|
|
authKeyId = keyInfo.id;
|
|
}
|
|
} else {
|
|
authKeyName = "anonymous";
|
|
}
|
|
}
|
|
}
|
|
|
|
req._authKeyName = authKeyName;
|
|
req._authKeyId = authKeyId;
|
|
|
|
// GET /v1/models
|
|
if (req.url === "/v1/models" && req.method === "GET") {
|
|
return jsonResponse(res, 200, {
|
|
object: "list",
|
|
data: MODELS.map((m) => ({
|
|
id: m.id, object: "model", owned_by: "anthropic",
|
|
created: Math.floor(Date.now() / 1000),
|
|
})),
|
|
});
|
|
}
|
|
|
|
// POST /v1/chat/completions
|
|
if (req.url === "/v1/chat/completions" && req.method === "POST") {
|
|
return handleChatCompletions(req, res);
|
|
}
|
|
|
|
// GET /health — comprehensive diagnostics
|
|
if (req.url === "/health") {
|
|
let binaryOk = false;
|
|
try { accessSync(CLAUDE, constants.X_OK); binaryOk = true; } catch {}
|
|
|
|
const uptimeMs = Date.now() - START_TIME;
|
|
const sessionList = [];
|
|
for (const [id, s] of sessions) {
|
|
sessionList.push({
|
|
id: id.slice(0, 12) + "...",
|
|
model: s.model,
|
|
messages: s.messageCount,
|
|
idleMs: Date.now() - s.lastUsed,
|
|
});
|
|
}
|
|
|
|
return jsonResponse(res, 200, {
|
|
status: binaryOk && authStatus.ok !== false ? "ok" : "degraded",
|
|
version: VERSION,
|
|
architecture: "on-demand (v2)",
|
|
uptime: uptimeMs,
|
|
uptimeHuman: `${Math.floor(uptimeMs / 3600000)}h ${Math.floor((uptimeMs % 3600000) / 60000)}m`,
|
|
claudeBinary: CLAUDE,
|
|
claudeBinaryOk: binaryOk,
|
|
authMode: AUTH_MODE,
|
|
auth: authStatus,
|
|
config: {
|
|
timeout: TIMEOUT,
|
|
maxConcurrent: MAX_CONCURRENT,
|
|
sessionTTL: SESSION_TTL,
|
|
circuitBreaker: "disabled",
|
|
allowedTools: SKIP_PERMISSIONS ? "all (skip-permissions)" : ALLOWED_TOOLS,
|
|
systemPrompt: SYSTEM_PROMPT ? `${SYSTEM_PROMPT.slice(0, 50)}...` : "(none)",
|
|
mcpConfig: MCP_CONFIG || "(none)",
|
|
},
|
|
stats,
|
|
circuitBreaker: "disabled",
|
|
sessions: sessionList,
|
|
recentErrors: recentErrors.slice(-5),
|
|
});
|
|
}
|
|
|
|
// DELETE /sessions — clear all sessions
|
|
if (req.url === "/sessions" && req.method === "DELETE") {
|
|
const count = sessions.size;
|
|
sessions.clear();
|
|
return jsonResponse(res, 200, { cleared: count });
|
|
}
|
|
|
|
// GET /sessions — list active sessions
|
|
if (req.url === "/sessions" && req.method === "GET") {
|
|
const list = [];
|
|
for (const [id, s] of sessions) {
|
|
list.push({ id, uuid: s.uuid, model: s.model, messages: s.messageCount, lastUsed: new Date(s.lastUsed).toISOString() });
|
|
}
|
|
return jsonResponse(res, 200, { sessions: list });
|
|
}
|
|
|
|
// GET /usage — fetch plan usage limits from Anthropic API
|
|
if (req.url === "/usage" && req.method === "GET") {
|
|
return handleUsage(req, res);
|
|
}
|
|
|
|
// GET /logs — recent proxy log entries (errors and key events)
|
|
if (req.url?.startsWith("/logs") && req.method === "GET") {
|
|
return handleLogs(req, res);
|
|
}
|
|
|
|
// GET /status — combined usage + health summary
|
|
if (req.url === "/status" && req.method === "GET") {
|
|
return handleStatus(req, res);
|
|
}
|
|
|
|
// GET /settings — view current tunable settings
|
|
// PATCH /settings — update settings at runtime (JSON body)
|
|
if (req.url === "/settings" && (req.method === "GET" || req.method === "PATCH")) {
|
|
return handleSettings(req, res);
|
|
}
|
|
|
|
// ── Key management API ──
|
|
const isAdmin = AUTH_MODE !== "multi" || authKeyName === "admin" || isLocalhost;
|
|
|
|
if (req.url === "/api/keys" && req.method === "POST") {
|
|
if (!isAdmin) return jsonResponse(res, 403, { error: "Admin access required" });
|
|
let body = "";
|
|
for await (const chunk of req) body += chunk;
|
|
let parsed;
|
|
try { parsed = JSON.parse(body); } catch { return jsonResponse(res, 400, { error: "Invalid JSON" }); }
|
|
const name = parsed.name || `key-${Date.now()}`;
|
|
const newKey = createKey(name);
|
|
return jsonResponse(res, 201, newKey);
|
|
}
|
|
|
|
if (req.url === "/api/keys" && req.method === "GET") {
|
|
if (!isAdmin) return jsonResponse(res, 403, { error: "Admin access required" });
|
|
return jsonResponse(res, 200, { keys: listKeys() });
|
|
}
|
|
|
|
if (req.url?.startsWith("/api/keys/") && req.method === "DELETE") {
|
|
if (!isAdmin) return jsonResponse(res, 403, { error: "Admin access required" });
|
|
const idOrName = decodeURIComponent(req.url.split("/api/keys/")[1]);
|
|
const revoked = revokeKey(idOrName);
|
|
return jsonResponse(res, 200, { revoked, idOrName });
|
|
}
|
|
|
|
if (req.url?.startsWith("/api/usage") && req.method === "GET") {
|
|
if (!isAdmin) return jsonResponse(res, 403, { error: "Admin access required" });
|
|
const url = new URL(req.url, `http://${BIND_ADDRESS}:${PORT}`);
|
|
const since = url.searchParams.get("since");
|
|
const until = url.searchParams.get("until");
|
|
return jsonResponse(res, 200, {
|
|
byKey: getUsageByKey({ since, until }),
|
|
timeline: getUsageTimeline({ hours: Math.min(parseInt(url.searchParams.get("hours") || "24", 10), 720) }),
|
|
recent: getRecentUsage(Math.min(parseInt(url.searchParams.get("limit") || "50", 10), 500)),
|
|
});
|
|
}
|
|
|
|
// GET /dashboard — web dashboard
|
|
if (pathname === "/dashboard" && req.method === "GET") {
|
|
try {
|
|
const html = readFileSync(join(__dirname, "dashboard.html"), "utf8");
|
|
res.writeHead(200, { "Content-Type": "text/html; charset=utf-8" });
|
|
res.end(html);
|
|
} catch (err) {
|
|
return jsonResponse(res, 500, { error: "Dashboard file not found" });
|
|
}
|
|
return;
|
|
}
|
|
|
|
jsonResponse(res, 404, { error: "Not found. Endpoints: GET /v1/models, POST /v1/chat/completions, GET /health, GET /usage, GET /status, GET /logs, GET|PATCH /settings, GET|DELETE /sessions, GET /dashboard, GET|POST|DELETE /api/keys, GET /api/usage" });
|
|
});
|
|
|
|
|
|
// ── Graceful shutdown ────────────────────────────────────────────────────
|
|
let shuttingDown = false;
|
|
|
|
function gracefulShutdown(signal) {
|
|
if (shuttingDown) return;
|
|
shuttingDown = true;
|
|
logEvent("info", "shutdown_start", { signal });
|
|
|
|
// 1. Stop accepting new connections
|
|
server.close(() => {
|
|
logEvent("info", "shutdown_server_closed", {});
|
|
});
|
|
|
|
// 2. Clear intervals/timers
|
|
clearInterval(sessionCleanupInterval);
|
|
clearInterval(authCheckInterval);
|
|
closeDb();
|
|
|
|
// 3. Kill all active child processes
|
|
for (const proc of activeProcesses) {
|
|
try { proc.kill("SIGTERM"); } catch {}
|
|
}
|
|
|
|
// Force-kill any remaining processes after 5s, then exit
|
|
const forceExitTimer = setTimeout(() => {
|
|
for (const proc of activeProcesses) {
|
|
try { proc.kill("SIGKILL"); } catch {}
|
|
}
|
|
logEvent("warn", "shutdown_forced", { remainingProcesses: activeProcesses.size });
|
|
process.exit(1);
|
|
}, 5000);
|
|
forceExitTimer.unref();
|
|
|
|
// If no active processes, exit immediately
|
|
if (activeProcesses.size === 0) {
|
|
logEvent("info", "shutdown_complete", {});
|
|
process.exit(0);
|
|
}
|
|
|
|
// Wait for active processes to finish
|
|
const checkDone = setInterval(() => {
|
|
if (activeProcesses.size === 0) {
|
|
clearInterval(checkDone);
|
|
logEvent("info", "shutdown_complete", {});
|
|
process.exit(0);
|
|
}
|
|
}, 200);
|
|
checkDone.unref();
|
|
}
|
|
|
|
process.on("SIGTERM", () => gracefulShutdown("SIGTERM"));
|
|
process.on("SIGINT", () => gracefulShutdown("SIGINT"));
|
|
|
|
// ── Start ───────────────────────────────────────────────────────────────
|
|
server.listen(PORT, BIND_ADDRESS, () => {
|
|
const bindMsg = BIND_ADDRESS === "0.0.0.0" ? `http://0.0.0.0:${PORT} (LAN mode)` : `http://127.0.0.1:${PORT}`;
|
|
console.log(`openclaw-claude-proxy v${VERSION} listening on ${bindMsg}`);
|
|
console.log(`Architecture: on-demand spawning (no pool)`);
|
|
console.log(`Models: ${MODELS.map((m) => m.id).join(", ")}`);
|
|
console.log(`Claude binary: ${CLAUDE}`);
|
|
console.log(`Timeout: ${TIMEOUT / 1000}s | Max concurrent: ${MAX_CONCURRENT}`);
|
|
console.log(`Circuit breaker: disabled`);
|
|
console.log(`Tools: ${SKIP_PERMISSIONS ? "all (skip-permissions)" : ALLOWED_TOOLS.join(", ")}`);
|
|
console.log(`Sessions: TTL=${SESSION_TTL / 1000}s`);
|
|
if (SYSTEM_PROMPT) console.log(`System prompt: "${SYSTEM_PROMPT.slice(0, 80)}..."`);
|
|
if (MCP_CONFIG) console.log(`MCP config: ${MCP_CONFIG}`);
|
|
console.log(`Auth: ${PROXY_API_KEY ? "enabled (PROXY_API_KEY set)" : "disabled (no PROXY_API_KEY)"}`);
|
|
console.log(`Auth mode: ${AUTH_MODE}${AUTH_MODE === "shared" ? " (PROXY_API_KEY)" : AUTH_MODE === "multi" ? " (per-user keys)" : " (open)"}`);
|
|
console.log(`Bind: ${BIND_ADDRESS}${BIND_ADDRESS === "0.0.0.0" ? " ⚠ LAN-accessible" : ""}`);
|
|
console.log(`---`);
|
|
console.log(`Coexistence: This proxy does NOT conflict with Claude Code interactive mode.`);
|
|
console.log(` OCP uses: localhost:${PORT} (HTTP) → claude -p (per-request process)`);
|
|
console.log(` CC uses: MCP protocol (in-process) → persistent session`);
|
|
console.log(` Both can run simultaneously on the same machine.`);
|
|
});
|