/**
 * HTTP session history revocation tests.
 */
import { EventEmitter } from "node:events";
import type { IncomingMessage, ServerResponse } from "node:http";
import { afterEach, describe, expect, it, vi } from "vitest";

let transcriptUpdateHandler:
  | ((update: { sessionFile?: string; message?: unknown; messageId?: string }) => void)
  | undefined;
let authRevoked = false;
let gatewayConfig: {
  trustedProxies?: string[];
  allowRealIpFallback?: boolean;
} = {
  trustedProxies: ["10.0.0.1"],
  allowRealIpFallback: false,
};
let authCheckCalls = 0;

vi.mock("../config/config.js", () => ({
  getRuntimeConfig: () => ({
    gateway: gatewayConfig,
  }),
}));

vi.mock("../sessions/transcript-events.js", () => ({
  onInternalSessionTranscriptUpdate: (cb: typeof transcriptUpdateHandler) => {
    transcriptUpdateHandler = cb;
    return () => {
      if (transcriptUpdateHandler === cb) {
        transcriptUpdateHandler = undefined;
      }
    };
  },
}));

vi.mock("./http-utils.js", () => ({
  getHeader: (req: IncomingMessage, name: string) => {
    const value = req.headers[name.toLowerCase()];
    return Array.isArray(value) ? value[0] : value;
  },
  resolveSharedSecretHttpOperatorScopes: () => ["operator.read"],
  authorizeScopedGatewayHttpRequestOrReply: async () => ({
    cfg: { gateway: {} },
    requestAuth: { trustDeclaredOperatorScopes: true },
    operatorScopes: ["operator.read"],
  }),
  checkGatewayHttpRequestAuth: async (params: {
    trustedProxies?: string[];
    allowRealIpFallback?: boolean;
  }) => {
    authCheckCalls += 1;
    if (authRevoked) {
      return {
        ok: false as const,
        authResult: { ok: false, reason: "trusted_proxy_user_not_allowed" },
      };
    }
    if (
      gatewayConfig.trustedProxies === undefined &&
      gatewayConfig.allowRealIpFallback === undefined
    ) {
      return params.trustedProxies === undefined && params.allowRealIpFallback === undefined
        ? {
            ok: false as const,
            authResult: { ok: false, reason: "trusted_proxy_no_proxies_configured" },
          }
        : {
            ok: true as const,
            requestAuth: { trustDeclaredOperatorScopes: true },
          };
    }
    return {
      ok: true as const,
      requestAuth: { trustDeclaredOperatorScopes: true },
    };
  },
}));

vi.mock("./session-utils.js", () => ({
  resolveGatewaySessionStoreTargetWithStore: () => ({
    storePath: "/tmp",
    storeKeys: ["agent:main"],
    canonicalKey: "agent:main",
    agentId: "main",
    store: {},
  }),
  resolveFreshestSessionEntryFromStoreKeys: () => ({
    sessionId: "session-1",
    sessionFile: "/tmp/session-1.jsonl",
  }),
  resolveSessionTranscriptCandidates: () => ["/tmp/session-1.jsonl"],
}));

vi.mock("./session-transcript-readers.js", () => ({
  readRecentSessionMessagesWithStatsAsync: async () => ({ messages: [], totalMessages: 0 }),
  readSessionMessagesAsync: async () => [],
  readSessionMessagesWithSourceAsync: async () => ({ messages: [] }),
}));

vi.mock("./session-history-state.js", () => ({
  buildSessionHistorySnapshot: () => ({
    history: { items: [], nextCursor: null, messages: [] },
  }),
  SessionHistorySseState: {
    fromRawSnapshot: (_params: unknown) => ({
      snapshot: () => ({ items: [], nextCursor: null, messages: [] }),
      appendInlineMessage: ({ message, messageId }: { message: unknown; messageId?: string }) => ({
        message,
        messageSeq: 1,
        messageId,
      }),
      shouldRefreshForTranscriptPath: () => false,
      refreshAsync: async () => ({ items: [], nextCursor: null, messages: [] }),
    }),
  },
}));

import { handleSessionHistoryHttpRequest } from "./sessions-history-http.js";

const SESSION_HISTORY_URL = "/sessions/agent%3Amain/history";
const SESSION_FILE = "/tmp/session-1.jsonl";
const TRUSTED_PROXY_STARTUP_OPTIONS = {
  auth: { mode: "trusted-proxy" } as never,
  trustedProxies: ["10.0.0.1"],
  allowRealIpFallback: false,
} satisfies Parameters<typeof handleSessionHistoryHttpRequest>[2];

class MockReq extends EventEmitter {
  url: string;
  method: string;
  headers: Record<string, string>;
  socket = new EventEmitter();

  constructor(url: string) {
    super();
    this.url = url;
    this.method = "GET";
    this.headers = {
      host: "localhost",
      accept: "text/event-stream",
      authorization: "Bearer token",
      "x-openclaw-scopes": "operator.read",
    };
  }
}

class MockRes extends EventEmitter {
  statusCode = 0;
  headers = new Map<string, string>();
  writes: string[] = [];
  writableEnded = false;
  socket = new EventEmitter();

  setHeader(name: string, value: string) {
    this.headers.set(name.toLowerCase(), value);
  }

  write(chunk: string) {
    this.writes.push(chunk);
    return true;
  }

  end(chunk?: string) {
    if (chunk !== undefined) {
      this.writes.push(chunk);
    }
    this.writableEnded = true;
    this.emit("finish");
    this.emit("close");
    return this;
  }

  flushHeaders() {}
}

async function openSessionHistoryStream(
  options: Parameters<typeof handleSessionHistoryHttpRequest>[2],
) {
  const req = new MockReq(SESSION_HISTORY_URL);
  const res = new MockRes();

  const handled = await handleSessionHistoryHttpRequest(
    req as unknown as IncomingMessage,
    res as unknown as ServerResponse,
    options,
  );

  expect(handled).toBe(true);
  expect(transcriptUpdateHandler).toBeTypeOf("function");

  return res;
}

function emitTranscriptTextUpdate({
  sessionFile = SESSION_FILE,
  text,
  messageId,
}: {
  sessionFile?: string;
  text: string;
  messageId: string;
}) {
  transcriptUpdateHandler?.({
    sessionFile,
    message: { role: "assistant", content: [{ type: "text", text }] },
    messageId,
  });
}

async function expectStreamClosedWithoutMessage(res: MockRes, text: string) {
  await vi.waitFor(() => {
    expect(res.writableEnded).toBe(true);
  });

  const joined = res.writes.join("");
  expect(joined).not.toContain("event: message");
  expect(joined).not.toContain(text);
  expect(res.writableEnded).toBe(true);
}

afterEach(() => {
  transcriptUpdateHandler = undefined;
  authRevoked = false;
  authCheckCalls = 0;
  gatewayConfig = {
    trustedProxies: ["10.0.0.1"],
    allowRealIpFallback: false,
  };
});

describe("session history SSE auth revocation", () => {
  it("closes the stream before delivering transcript updates after auth is revoked", async () => {
    const res = await openSessionHistoryStream({ auth: { mode: "trusted-proxy" } as never });

    expect(res.headers.get("content-type")).toContain("text/event-stream");

    authRevoked = true;

    emitTranscriptTextUpdate({
      text: "post-revocation secret",
      messageId: "m-1",
    });

    await expectStreamClosedWithoutMessage(res, "post-revocation secret");
  });

  it("rechecks SSE auth against live proxy config instead of startup fallbacks", async () => {
    const res = await openSessionHistoryStream(TRUSTED_PROXY_STARTUP_OPTIONS);

    gatewayConfig = {};

    emitTranscriptTextUpdate({
      text: "stale-proxy event",
      messageId: "m-2",
    });

    await expectStreamClosedWithoutMessage(res, "stale-proxy event");
  });

  it("skips SSE reauth for transcript updates outside this stream", async () => {
    const res = await openSessionHistoryStream(TRUSTED_PROXY_STARTUP_OPTIONS);

    authCheckCalls = 0;
    gatewayConfig = {};

    emitTranscriptTextUpdate({
      sessionFile: "/tmp/other-session.jsonl",
      text: "other session",
      messageId: "m-3",
    });

    const joined = res.writes.join("");
    expect(authCheckCalls).toBe(0);
    expect(joined).not.toContain("other session");
    expect(res.writableEnded).toBe(false);
  });
});
