ReMe/website/app/api.ts

import type {
  AppConfig,
  FileStat,
  GraphSnapshot,
  ReMeResponse,
  StreamChunk,
} from "./types";
import { decodeSseEvent } from "./chat-stream";
import { translate, useLanguageStore, type TranslationKey } from "./i18n";

export const REME_API_URL = (
  process.env.NEXT_PUBLIC_REME_API_URL || "http://127.0.0.1:2333"
).replace(/\/$/, "");
const message = (key: TranslationKey, status: number) =>
  translate(useLanguageStore.getState().language, key, {
    status: String(status),
  });

async function parseResponse<T>(response: Response): Promise<ReMeResponse<T>> {
  let payload: ReMeResponse<T>;
  try {
    payload = (await response.json()) as ReMeResponse<T>;
  } catch {
    throw new Error(message("invalidResponse", response.status));
  }
  if (!response.ok || !payload.success) {
    const detail =
      typeof payload.answer === "string"
        ? payload.answer
        : message("requestFailed", response.status);
    throw new Error(detail);
  }
  return payload;
}

export async function callReMe<T>(
  action: string,
  body: Record<string, unknown> = {},
): Promise<ReMeResponse<T>> {
  const response = await fetch(`${REME_API_URL}/${action}`, {
    method: "POST",
    headers: { "Content-Type": "application/json" },
    body: JSON.stringify(body),
  });
  return parseResponse<T>(response);
}

export async function getAppConfig(): Promise<AppConfig> {
  return (await callReMe<AppConfig>("app_config")).answer;
}

export async function getReMeVersion(): Promise<string> {
  return String((await callReMe<string>("version")).answer);
}

export async function getReMeStatus(): Promise<ReMeResponse<string>> {
  return callReMe<string>("status");
}

export async function rebuildReMeIndex(): Promise<ReMeResponse<unknown>> {
  return callReMe<unknown>("reindex");
}

export async function getGraphSnapshot(): Promise<GraphSnapshot> {
  return (await callReMe<GraphSnapshot>("graph_snapshot")).answer;
}

export async function listWorkspaceFiles(
  extensions: string[],
): Promise<string[]> {
  const response = await callReMe<string>("list", {
    path: "",
    recursive: true,
    limit: 5000,
    sort_by: "mtime",
    extensions,
  });
  return Array.isArray(response.metadata.items)
    ? (response.metadata.items as string[])
    : [];
}

export async function readWorkspaceFile(
  path: string,
): Promise<{ content: string; stat: FileStat }> {
  const response = await callReMe<string>("load", { path });
  return {
    content: String(response.answer ?? ""),
    stat: response.metadata as unknown as FileStat,
  };
}

export async function saveWorkspaceFile(
  path: string,
  content: string,
  expectedMtime?: string,
): Promise<FileStat> {
  const response = await callReMe<string>("save", {
    path,
    content,
    expected_mtime: expectedMtime || null,
  });
  return response.metadata as unknown as FileStat;
}

export async function streamChat(
  query: string,
  sessionId: string | undefined,
  signal: AbortSignal,
  onChunk: (chunk: StreamChunk) => void,
): Promise<void> {
  const response = await fetch(`${REME_API_URL}/chat`, {
    method: "POST",
    headers: {
      "Content-Type": "application/json",
      Accept: "text/event-stream",
    },
    body: JSON.stringify({ query, session_id: sessionId || null }),
    signal,
  });
  if (!response.ok || !response.body)
    throw new Error(message("agentUnavailable", response.status));

  const reader = response.body.getReader();
  const decoder = new TextDecoder();
  let buffer = "";
  while (true) {
    const { value, done } = await reader.read();
    buffer += decoder.decode(value, { stream: !done });
    const events = buffer.split(/\r?\n\r?\n/);
    buffer = events.pop() || "";
    for (const event of events) {
      const chunk = decodeSseEvent(event);
      if (chunk) onChunk(chunk);
    }
    if (done) break;
  }
  const finalChunk = decodeSseEvent(buffer);
  if (finalChunk) onChunk(finalChunk);
}