import type {
  BeforeProviderRequestEvent,
  ExtensionAPI,
  ExtensionContext,
} from "@earendil-works/pi-coding-agent";

export default function (pi: ExtensionAPI) {
  pi.on(
    "before_provider_request",
    (event: BeforeProviderRequestEvent, ctx: ExtensionContext) => {
      const payload = event.payload as Record<string, unknown>;
      if (!payload) return;

      if (!(ctx.model?.provider === "openrouter")) return payload;

      const sessionId = ctx.sessionManager?.getSessionId();
      if (!sessionId) {
        console.warn("missing session id");
        return payload;
      }

      return {
        ...payload,
        session_id: sessionId,
      };
    },
  );
}