@oracle-agent/oracle 0.9.8 → 0.11.0

package/src/tui/standalone-client.mjs

@@ -0,0 +1,666 @@
+import fs from "node:fs";
+import path from "node:path";
+import { randomUUID } from "node:crypto";
+import { EventEmitter } from "node:events";
+import { spawnSync } from "node:child_process";
+
+import { PACKAGE_ROOT } from "../cli/paths.mjs";
+
+const ALLOWED_COMMANDS = new Set(["chain", "data", "scan", "route", "prepare"]);
+const ALLOWED_CHAIN = new Set(["list", "show"]);
+const ALLOWED_DATA = new Set(["call", "catalog", "health"]);
+const ALLOWED_ROUTE = new Set(["swap", "bridge", "prepare", "prepare-bridge"]);
+const OAUTH_BASE_URLS = Object.freeze({
+  "anthropic-oauth": "https://api.anthropic.com/v1",
+  "openai-codex": "https://chatgpt.com/backend-api/codex",
+  "xai-oauth": "https://api.x.ai/v1",
+});
+const MAX_TOOL_OUTPUT = 20_000;
+const MAX_AGENT_STEPS = 6;
+
+const ORACLE_TOOL = Object.freeze({
+  type: "function",
+  function: {
+    name: "oracle_cli",
+    description: "Run a bounded read or prepare-only Oracle CLI command. It cannot sign or broadcast transactions.",
+    parameters: {
+      type: "object",
+      properties: {
+        argv: {
+          type: "array",
+          items: { type: "string" },
+          minItems: 1,
+          maxItems: 24,
+          description: "Arguments after the oracle executable, for example [\"scan\",\"chains\"] or [\"data\",\"health\"].",
+        },
+      },
+      required: ["argv"],
+      additionalProperties: false,
+    },
+  },
+});
+
+function trimOutput(value) {
+  const text = String(value || "");
+  if (text.length <= MAX_TOOL_OUTPUT) return text;
+  return `${text.slice(0, MAX_TOOL_OUTPUT)}\n[truncated ${text.length - MAX_TOOL_OUTPUT} chars]`;
+}
+
+function validateCommand(argv) {
+  if (!Array.isArray(argv) || argv.length === 0 || argv.length > 24) {
+    return "argv must contain 1 to 24 arguments";
+  }
+  if (argv.some((arg) => typeof arg !== "string" || arg.length > 1_000)) {
+    return "every argument must be a string no longer than 1000 characters";
+  }
+  const command = argv[0];
+  if (!ALLOWED_COMMANDS.has(command)) return `command not allowed: ${command}`;
+  if (command === "chain" && !ALLOWED_CHAIN.has(argv[1] || "list")) {
+    return `command not allowed: chain${argv[1] ? ` ${argv[1]}` : ""}`;
+  }
+  if (command === "data" && !ALLOWED_DATA.has(argv[1] || "")) {
+    return `command not allowed: data${argv[1] ? ` ${argv[1]}` : ""}`;
+  }
+  if (command === "route" && !ALLOWED_ROUTE.has(argv[1] || "")) {
+    return `command not allowed: route${argv[1] ? ` ${argv[1]}` : ""}`;
+  }
+  return null;
+}
+
+export async function runOracleCommand(argv, options = {}) {
+  const invalid = validateCommand(argv);
+  if (invalid) return { ok: false, error: invalid };
+  const spawnFn = options.spawnFn || spawnSync;
+  const root = options.packageRoot || PACKAGE_ROOT;
+  const result = spawnFn(
+    process.execPath,
+    [path.join(root, "bin", "oracle.mjs"), ...argv],
+    {
+      encoding: "utf8",
+      timeout: options.timeoutMs || 45_000,
+      maxBuffer: 1024 * 1024,
+      env: { ...process.env, ORACLE_CHAT_TOOL: "1" },
+    },
+  );
+  if (result.error) return { ok: false, error: result.error.message };
+  return {
+    ok: result.status === 0,
+    status: typeof result.status === "number" ? result.status : 1,
+    stdout: trimOutput(result.stdout),
+    stderr: trimOutput(result.stderr),
+  };
+}
+
+function defaultPersona() {
+  const file = path.join(PACKAGE_ROOT, "profiles", "oracle", "SOUL.md");
+  const soul = fs.readFileSync(file, "utf8");
+  return `${soul.trim()}\n\nYou have one bounded tool named oracle_cli. Use it for live chain facts, scans, route comparison, and unsigned preparation. Never claim a tool action succeeded unless its returned output proves it.`;
+}
+
+function eventFrame(type, sessionId, payload = {}) {
+  return {
+    method: "event",
+    params: {
+      type,
+      session_id: sessionId,
+      payload,
+    },
+  };
+}
+
+async function responseError(response) {
+  return new Error(`model request failed (HTTP ${response?.status || "unknown"})`);
+}
+
+async function* parseSse(response) {
+  if (!response?.ok) throw await responseError(response);
+  if (!response.body) throw new Error("model response had no body");
+  const reader = response.body.getReader();
+  const decoder = new TextDecoder();
+  let buffer = "";
+  while (true) {
+    const { value, done } = await reader.read();
+    buffer += decoder.decode(value || new Uint8Array(), { stream: !done });
+    const blocks = buffer.split(/\r?\n\r?\n/);
+    buffer = blocks.pop() || "";
+    for (const block of blocks) {
+      const data = block
+        .split(/\r?\n/)
+        .filter((line) => line.startsWith("data:"))
+        .map((line) => line.slice(5).trimStart())
+        .join("\n");
+      if (!data || data === "[DONE]") continue;
+      yield JSON.parse(data);
+    }
+    if (done) break;
+  }
+  const tail = buffer.trim();
+  if (tail.startsWith("data:")) {
+    const data = tail.slice(5).trim();
+    if (data && data !== "[DONE]") yield JSON.parse(data);
+  }
+}
+
+function mergeToolDelta(map, delta) {
+  const index = Number(delta.index || 0);
+  const current = map.get(index) || {
+    id: "",
+    type: "function",
+    function: { name: "", arguments: "" },
+  };
+  if (delta.id) current.id += delta.id;
+  if (delta.type) current.type = delta.type;
+  if (delta.function?.name) current.function.name += delta.function.name;
+  if (delta.function?.arguments) current.function.arguments += delta.function.arguments;
+  map.set(index, current);
+}
+
+function usagePayload(usage, contextLength) {
+  const input = Number(usage?.prompt_tokens || usage?.input_tokens || 0);
+  const output = Number(usage?.completion_tokens || usage?.output_tokens || 0);
+  return {
+    context_used: input,
+    context_max: contextLength,
+    context_percent: contextLength > 0 ? Math.round((input / contextLength) * 100) : 0,
+    input,
+    output,
+  };
+}
+
+function responseTool() {
+  return {
+    type: "function",
+    name: ORACLE_TOOL.function.name,
+    description: ORACLE_TOOL.function.description,
+    parameters: ORACLE_TOOL.function.parameters,
+    strict: false,
+  };
+}
+
+function anthropicTool() {
+  return {
+    name: ORACLE_TOOL.function.name,
+    description: ORACLE_TOOL.function.description,
+    input_schema: ORACLE_TOOL.function.parameters,
+  };
+}
+
+function appendAnthropicMessage(output, role, blocks) {
+  const normalized = Array.isArray(blocks) ? blocks : [{ type: "text", text: String(blocks || "") }];
+  const previous = output.at(-1);
+  if (previous?.role === role) previous.content.push(...normalized);
+  else output.push({ role, content: normalized });
+}
+
+function toAnthropicMessages(messages) {
+  const output = [];
+  for (const message of messages) {
+    if (message.role === "system") continue;
+    if (message.role === "user") {
+      appendAnthropicMessage(output, "user", [{ type: "text", text: String(message.content || "") }]);
+      continue;
+    }
+    if (message.role === "assistant") {
+      const blocks = [];
+      if (message.content) blocks.push({ type: "text", text: String(message.content) });
+      for (const call of message.tool_calls || []) {
+        let input = {};
+        try { input = JSON.parse(call.function?.arguments || "{}"); } catch {}
+        blocks.push({
+          type: "tool_use",
+          id: call.id,
+          name: call.function?.name,
+          input,
+        });
+      }
+      if (blocks.length) appendAnthropicMessage(output, "assistant", blocks);
+      continue;
+    }
+    if (message.role === "tool") {
+      appendAnthropicMessage(output, "user", [{
+        type: "tool_result",
+        tool_use_id: message.tool_call_id,
+        content: String(message.content || ""),
+      }]);
+    }
+  }
+  return output;
+}
+
+function toResponsesInput(messages) {
+  const output = [];
+  for (const message of messages) {
+    if (message.role === "system") continue;
+    if (message.role === "user") {
+      output.push({ role: "user", content: [{ type: "input_text", text: String(message.content || "") }] });
+      continue;
+    }
+    if (message.role === "assistant") {
+      if (message.content) {
+        output.push({
+          role: "assistant",
+          status: "completed",
+          content: [{ type: "output_text", text: String(message.content) }],
+        });
+      }
+      for (const call of message.tool_calls || []) {
+        output.push({
+          type: "function_call",
+          id: call.response_item_id || undefined,
+          call_id: call.id,
+          name: call.function?.name,
+          arguments: call.function?.arguments || "{}",
+        });
+      }
+      continue;
+    }
+    if (message.role === "tool") {
+      output.push({
+        type: "function_call_output",
+        call_id: message.tool_call_id,
+        output: String(message.content || ""),
+      });
+    }
+  }
+  return output.map((item) => Object.fromEntries(Object.entries(item).filter(([, value]) => value !== undefined)));
+}
+
+function accountIdFromJwt(token) {
+  try {
+    const payload = JSON.parse(Buffer.from(String(token).split(".")[1], "base64url").toString("utf8"));
+    return String(payload?.["https://api.openai.com/auth"]?.chatgpt_account_id || "");
+  } catch {
+    return "";
+  }
+}
+
+async function chatCompletionStep({ config, messages, fetchFn, signal, onText }) {
+  const response = await fetchFn(`${config.baseUrl}/chat/completions`, {
+    method: "POST",
+    headers: {
+      "content-type": "application/json",
+      ...(config.apiKey ? { Authorization: `Bearer ${config.apiKey}` } : {}),
+      ...(config.provider === "openrouter" ? {
+        "HTTP-Referer": "https://oracle-agent.dev",
+        "X-Title": "Oracle CLI",
+      } : {}),
+    },
+    body: JSON.stringify({
+      model: config.model,
+      messages,
+      tools: [ORACLE_TOOL],
+      tool_choice: "auto",
+      stream: true,
+      stream_options: { include_usage: true },
+    }),
+    signal,
+  });
+  const toolDeltas = new Map();
+  let text = "";
+  let usage = null;
+  for await (const frame of parseSse(response)) {
+    if (frame.usage) usage = frame.usage;
+    const delta = frame.choices?.[0]?.delta || {};
+    if (typeof delta.content === "string" && delta.content) {
+      text += delta.content;
+      onText(delta.content);
+    }
+    for (const toolDelta of delta.tool_calls || []) mergeToolDelta(toolDeltas, toolDelta);
+  }
+  return {
+    text,
+    usage,
+    toolCalls: [...toolDeltas.entries()].sort(([a], [b]) => a - b).map(([, value]) => value),
+  };
+}
+
+async function anthropicStep({ config, messages, persona, credential, fetchFn, signal, onText }) {
+  const response = await fetchFn(`${config.baseUrl}/messages`, {
+    method: "POST",
+    headers: {
+      "content-type": "application/json",
+      accept: "text/event-stream",
+      Authorization: `Bearer ${credential.accessToken}`,
+      "anthropic-version": "2023-06-01",
+      "anthropic-beta": "interleaved-thinking-2025-05-14,fine-grained-tool-streaming-2025-05-14,claude-code-20250219,oauth-2025-04-20",
+      "user-agent": `claude-code/${process.env.ORACLE_CLAUDE_CODE_VERSION || "2.1.220"} (external, cli)`,
+      "x-app": "cli",
+    },
+    body: JSON.stringify({
+      model: config.model,
+      max_tokens: Number(config.maxOutputTokens || 8192),
+      system: [
+        { type: "text", text: "You are Claude Code, Anthropic's official CLI for Claude." },
+        { type: "text", text: persona },
+      ],
+      messages: toAnthropicMessages(messages),
+      tools: [anthropicTool()],
+      tool_choice: { type: "auto" },
+      stream: true,
+    }),
+    signal,
+  });
+  const toolCalls = new Map();
+  let text = "";
+  let usage = null;
+  for await (const frame of parseSse(response)) {
+    if (frame.type === "message_start") usage = { ...(usage || {}), ...(frame.message?.usage || {}) };
+    if (frame.type === "message_delta") usage = { ...(usage || {}), ...(frame.usage || {}) };
+    if (frame.type === "content_block_start" && frame.content_block?.type === "tool_use") {
+      const block = frame.content_block;
+      toolCalls.set(Number(frame.index || 0), {
+        id: block.id,
+        type: "function",
+        function: {
+          name: block.name,
+          arguments: block.input && Object.keys(block.input).length ? JSON.stringify(block.input) : "",
+        },
+      });
+    }
+    if (frame.type === "content_block_delta" && frame.delta?.type === "text_delta") {
+      text += frame.delta.text || "";
+      if (frame.delta.text) onText(frame.delta.text);
+    }
+    if (frame.type === "content_block_delta" && frame.delta?.type === "input_json_delta") {
+      const index = Number(frame.index || 0);
+      const call = toolCalls.get(index);
+      if (call) call.function.arguments += frame.delta.partial_json || "";
+    }
+    if (frame.type === "error") throw new Error(frame.error?.message || "Claude stream failed");
+  }
+  return {
+    text,
+    usage,
+    toolCalls: [...toolCalls.entries()].sort(([a], [b]) => a - b).map(([, value]) => {
+      if (!value.function.arguments) value.function.arguments = "{}";
+      return value;
+    }),
+  };
+}
+
+async function responsesStep({ config, messages, persona, sessionId, credential, fetchFn, signal, onText }) {
+  const isCodex = config.provider === "openai-codex";
+  const accountId = isCodex ? accountIdFromJwt(credential.accessToken) : "";
+  const headers = {
+    "content-type": "application/json",
+    accept: "text/event-stream",
+    Authorization: `Bearer ${credential.accessToken}`,
+    ...(isCodex ? {
+      "user-agent": "codex_cli_rs/0.0.0 (Oracle CLI)",
+      originator: "codex_cli_rs",
+      ...(accountId ? { "ChatGPT-Account-ID": accountId } : {}),
+      session_id: sessionId,
+      "x-client-request-id": sessionId,
+    } : {
+      "x-grok-conv-id": sessionId,
+    }),
+  };
+  const body = {
+    model: config.model,
+    instructions: persona,
+    input: toResponsesInput(messages),
+    tools: [responseTool()],
+    tool_choice: "auto",
+    parallel_tool_calls: true,
+    store: false,
+    stream: true,
+    ...(isCodex ? { reasoning: { effort: config.reasoningEffort || "high", summary: "auto" } } : {}),
+  };
+  const response = await fetchFn(`${config.baseUrl}/responses`, {
+    method: "POST",
+    headers,
+    body: JSON.stringify(body),
+    signal,
+  });
+  const toolCalls = new Map();
+  let text = "";
+  let usage = null;
+  for await (const frame of parseSse(response)) {
+    if (frame.type === "response.output_text.delta" && typeof frame.delta === "string") {
+      text += frame.delta;
+      onText(frame.delta);
+    }
+    const item = frame.item;
+    if ((frame.type === "response.output_item.added" || frame.type === "response.output_item.done") && item?.type === "function_call") {
+      const key = String(item.call_id || item.id || frame.output_index || toolCalls.size);
+      toolCalls.set(key, {
+        id: item.call_id || item.id,
+        response_item_id: item.id,
+        type: "function",
+        function: { name: item.name, arguments: item.arguments || "{}" },
+      });
+    }
+    if (frame.type === "response.function_call_arguments.delta") {
+      const key = String(frame.call_id || frame.item_id || frame.output_index || "");
+      const call = toolCalls.get(key);
+      if (call) call.function.arguments += frame.delta || "";
+    }
+    if (frame.type === "response.completed") usage = frame.response?.usage || usage;
+    if (frame.type === "response.failed" || frame.type === "error") {
+      throw new Error(frame.response?.error?.message || frame.error?.message || "Responses stream failed");
+    }
+  }
+  return { text, usage, toolCalls: [...toolCalls.values()] };
+}
+
+export function createStandaloneClient(options = {}) {
+  const config = options.config;
+  if (!config?.baseUrl || !config?.model) throw new Error("standalone model configuration is incomplete");
+  if (config.authType === "oauth") {
+    const expected = OAUTH_BASE_URLS[config.provider];
+    const actual = String(config.baseUrl).replace(/\/+$/, "");
+    if (!expected || actual !== expected) throw new Error(`${config.provider || "unknown"} OAuth endpoint is not trusted`);
+  }
+  const emitter = new EventEmitter();
+  const fetchFn = options.fetchFn || globalThis.fetch;
+  const toolRunner = options.toolRunner || ((argv) => runOracleCommand(argv, options));
+  const credentialProvider = options.credentialProvider || (async (provider) => {
+    const { resolveOAuthCredentials } = await import("../auth/oauth.mjs");
+    return resolveOAuthCredentials(provider);
+  });
+  const persona = options.persona || defaultPersona();
+  const sessionId = options.sessionId || `oracle-${randomUUID()}`;
+  let history = [];
+  let controller = null;
+  let stopped = false;
+  let active = false;
+
+  function emitEvent(type, payload = {}) {
+    emitter.emit("event", eventFrame(type, sessionId, payload));
+  }
+
+  function emitError(error) {
+    emitEvent("error", { message: error?.message || String(error) });
+    if (emitter.listenerCount("error") > 0) emitter.emit("error", error);
+  }
+
+  async function performTurn(text) {
+    active = true;
+    controller = new AbortController();
+    const messages = [
+      { role: "system", content: persona },
+      ...history,
+      { role: "user", content: text },
+    ];
+    let finalText = "";
+    let usage = null;
+    let messageStarted = false;
+
+    try {
+      for (let step = 0; step < MAX_AGENT_STEPS; step += 1) {
+        const credential = config.authType === "oauth"
+          ? await credentialProvider(config.provider)
+          : null;
+        const onText = (delta) => {
+          if (!messageStarted) {
+            emitEvent("message.start", {});
+            messageStarted = true;
+          }
+          finalText += delta;
+          emitEvent("message.delta", { text: delta });
+        };
+        const request = {
+          config,
+          messages,
+          persona,
+          sessionId,
+          credential,
+          fetchFn,
+          signal: controller.signal,
+          onText,
+        };
+        const result = config.protocol === "anthropic-messages"
+          ? await anthropicStep(request)
+          : config.protocol === "responses"
+            ? await responsesStep(request)
+            : await chatCompletionStep(request);
+        const stepText = result.text;
+        const toolCalls = result.toolCalls;
+        usage = result.usage || usage;
+
+        if (toolCalls.length === 0) {
+          messages.push({ role: "assistant", content: stepText });
+          history = messages.slice(1);
+          if (!messageStarted) emitEvent("message.start", {});
+          emitEvent("message.complete", {
+            text: finalText,
+            usage: usagePayload(usage, config.contextLength),
+          });
+          return;
+        }
+
+        messages.push({ role: "assistant", content: stepText || null, tool_calls: toolCalls });
+        for (const call of toolCalls) {
+          let result;
+          try {
+            const parsed = JSON.parse(call.function.arguments || "{}");
+            result = call.function.name === "oracle_cli"
+              ? await toolRunner(parsed.argv)
+              : { ok: false, error: `unknown tool: ${call.function.name}` };
+          } catch (error) {
+            result = { ok: false, error: `invalid tool arguments: ${error.message}` };
+          }
+          messages.push({
+            role: "tool",
+            tool_call_id: call.id || `call_${randomUUID()}`,
+            content: JSON.stringify(result),
+          });
+        }
+      }
+      throw new Error(`agent exceeded ${MAX_AGENT_STEPS} tool steps`);
+    } catch (error) {
+      if (error?.name !== "AbortError") emitError(error);
+    } finally {
+      active = false;
+      controller = null;
+    }
+  }
+
+  async function start() {
+    if (stopped) throw new Error("standalone client is stopped");
+  }
+
+  async function request(method, params = {}) {
+    if (stopped) throw new Error("standalone client is stopped");
+    if (method === "session.create") {
+      emitEvent("session.info", {
+        model: params.model || config.model,
+        reasoning_effort: config.reasoningEffort,
+        usage: { context_max: config.contextLength },
+        backend: "standalone",
+      });
+      return {
+        session_id: sessionId,
+        info: {
+          model: params.model || config.model,
+          reasoning_effort: config.reasoningEffort,
+          usage: { context_max: config.contextLength },
+          backend: "standalone",
+        },
+      };
+    }
+    if (method === "prompt.submit") {
+      if (active) throw new Error("a turn is already active");
+      const text = String(params.text || "").trim();
+      if (!text) throw new Error("prompt text is required");
+      queueMicrotask(() => { performTurn(text).catch(emitError); });
+      return { accepted: true };
+    }
+    if (method === "session.interrupt") {
+      const interrupted = Boolean(controller && active);
+      controller?.abort();
+      return { interrupted };
+    }
+    if (method === "session.usage") {
+      return { calls: 0, input: 0, output: 0, total: 0 };
+    }
+    throw new Error(`unsupported standalone method: ${method}`);
+  }
+
+  async function stop() {
+    if (stopped) return;
+    stopped = true;
+    controller?.abort();
+  }
+
+  return Object.freeze({
+    start,
+    request,
+    stop,
+    on: emitter.on.bind(emitter),
+    off: emitter.off.bind(emitter),
+  });
+}
+
+export async function runStandaloneQuery({
+  client,
+  text,
+  stdout = process.stdout,
+  stderr = process.stderr,
+} = {}) {
+  if (!client) throw new Error("standalone client is required");
+  await client.start();
+  const session = await client.request("session.create", { source: "cli" });
+  let written = "";
+  let settled = false;
+
+  try {
+    return await new Promise((resolve, reject) => {
+      const handle = (frame) => {
+        const type = frame?.params?.type;
+        const payload = frame?.params?.payload || {};
+        if (type === "message.delta" && typeof payload.text === "string") {
+          written += payload.text;
+          stdout.write(payload.text);
+          return;
+        }
+        if (type === "message.complete") {
+          if (!written && payload.text) stdout.write(String(payload.text));
+          stdout.write("\n");
+          settled = true;
+          resolve(0);
+          return;
+        }
+        if (type === "error") {
+          const message = payload.message || payload.text || "standalone chat failed";
+          stderr.write(`oracle chat: ${message}\n`);
+          settled = true;
+          reject(new Error(message));
+        }
+      };
+      client.on("event", handle);
+      client.request("prompt.submit", {
+        session_id: session.session_id,
+        text,
+      }).catch((error) => {
+        if (settled) return;
+        settled = true;
+        reject(error);
+      });
+    });
+  } finally {
+    await client.stop();
+  }
+}
+
+export default { createStandaloneClient, runOracleCommand, runStandaloneQuery };