openhuman/scripts/mock-api/admin.mjs

import { json } from "./http.mjs";
import { resetConversationFixturesState } from "./routes/conversations.mjs";
import { resetCronFixturesState } from "./routes/cron.mjs";
import {
  clearSocketEventLog,
  clearRequestLog,
  getSocketEventLog,
  listSocketSessions,
  getMockBehavior,
  getMockConversations,
  getMockCronJobs,
  getMockMessages,
  listMockLlmThreads,
  getMockWebhookTriggers,
  getMockTelegramSent,
  getRequestLog,
  pushMockTelegramUpdate,
  resetMockBehavior,
  resetMockConversations,
  resetMockCronJobs,
  resetMockMessages,
  resetMockLlmThreads,
  resetMockTelegram,
  resetSocketSessions,
  resetMockTunnels,
  resetMockWebhookTriggers,
  setMockBehavior,
  setMockBehaviors,
} from "./state.mjs";
import { disconnectMockSockets, emitMockSocketEvent } from "./socket.mjs";

export function handleAdmin(ctx) {
  const { method, url, parsedBody, res, getPort } = ctx;

  if (method === "GET" && /^\/__admin\/health\/?$/.test(url)) {
    json(res, 200, { ok: true, port: getPort() });
    return true;
  }
  if (method === "GET" && /^\/__admin\/requests\/?$/.test(url)) {
    json(res, 200, { success: true, data: getRequestLog() });
    return true;
  }
  if (method === "GET" && /^\/__admin\/behavior\/?$/.test(url)) {
    json(res, 200, { success: true, data: getMockBehavior() });
    return true;
  }
  if (method === "GET" && /^\/__admin\/state\/?$/.test(url)) {
    json(res, 200, {
      success: true,
      data: {
        requestCount: getRequestLog().length,
        conversations: getMockConversations(),
        messages: getMockMessages(),
        cronJobs: getMockCronJobs(),
        webhookTriggers: getMockWebhookTriggers(),
        llmThreads: listMockLlmThreads(),
        socketSessions: listSocketSessions(),
        socketEventCount: getSocketEventLog().length,
      },
    });
    return true;
  }
  if (method === "GET" && /^\/__admin\/socket\/sessions\/?$/.test(url)) {
    json(res, 200, { success: true, data: listSocketSessions() });
    return true;
  }
  if (method === "GET" && /^\/__admin\/socket\/events\/?$/.test(url)) {
    json(res, 200, { success: true, data: getSocketEventLog() });
    return true;
  }
  if (method === "POST" && /^\/__admin\/reset\/?$/.test(url)) {
    const keepBehavior = parsedBody?.keepBehavior === true;
    const keepRequests = parsedBody?.keepRequests === true;
    if (!keepBehavior) resetMockBehavior();
    if (!keepRequests) clearRequestLog();
    resetMockTunnels();
    resetMockConversations();
    resetMockMessages();
    resetMockCronJobs();
    resetMockWebhookTriggers();
    resetMockLlmThreads();
    resetConversationFixturesState();
    resetCronFixturesState();
    resetSocketSessions();
    resetMockTelegram();
    json(res, 200, {
      success: true,
      data: {
        behavior: getMockBehavior(),
        requestCount: getRequestLog().length,
      },
    });
    return true;
  }
  if (method === "POST" && /^\/__admin\/behavior\/?$/.test(url)) {
    if (parsedBody?.behavior && typeof parsedBody.behavior === "object") {
      setMockBehaviors(parsedBody.behavior, parsedBody.mode);
    } else if (parsedBody?.key) {
      setMockBehavior(parsedBody.key, parsedBody.value ?? "");
    }
    json(res, 200, { success: true, data: getMockBehavior() });
    return true;
  }
  if (method === "POST" && /^\/__admin\/socket\/emit\/?$/.test(url)) {
    const delivered = emitMockSocketEvent({
      event: parsedBody?.event,
      data: parsedBody?.data,
      targetSid: parsedBody?.targetSid,
      targetUserId: parsedBody?.targetUserId,
      excludeSid: parsedBody?.excludeSid,
      delayMs: parsedBody?.delayMs,
    });
    json(res, 200, { success: true, data: { delivered } });
    return true;
  }
  if (method === "POST" && /^\/__admin\/socket\/disconnect\/?$/.test(url)) {
    const disconnected = disconnectMockSockets({
      targetSid: parsedBody?.targetSid,
      targetUserId: parsedBody?.targetUserId,
    });
    json(res, 200, { success: true, data: { disconnected } });
    return true;
  }
  if (method === "POST" && /^\/__admin\/socket\/clear-events\/?$/.test(url)) {
    clearSocketEventLog();
    json(res, 200, { success: true, data: [] });
    return true;
  }

  // ── Telegram admin endpoints ───────────────────────────────────────────
  if (method === "POST" && /^\/__admin\/telegram\/inject-update\/?$/.test(url)) {
    const updates = Array.isArray(parsedBody?.updates)
      ? parsedBody.updates
      : parsedBody && typeof parsedBody === "object" && !Array.isArray(parsedBody)
        ? [parsedBody]
        : [];
    for (const update of updates) {
      pushMockTelegramUpdate(update);
    }
    console.log(`[telegram-mock] inject-update: queued ${updates.length} update(s)`);
    json(res, 200, { ok: true, queued: updates.length });
    return true;
  }
  if (method === "GET" && /^\/__admin\/telegram\/sent\/?$/.test(url)) {
    json(res, 200, { ok: true, messages: getMockTelegramSent() });
    return true;
  }
  if (method === "POST" && /^\/__admin\/telegram\/reset\/?$/.test(url)) {
    resetMockTelegram();
    console.log("[telegram-mock] admin reset: telegram state cleared");
    json(res, 200, { ok: true });
    return true;
  }

  return false;
}