import type { ExtensionAPI } from "@earendil-works/pi-coding-agent";
import { createDoublePasteHandler } from "./double-paste.ts";

export default function piExpandDoublePaste(pi: ExtensionAPI): void {
  let unsubscribe: (() => void) | undefined;

  pi.on("session_start", (_event, ctx) => {
    if (ctx.mode !== "tui") return;

    unsubscribe?.();
    const handleDoublePaste = createDoublePasteHandler(ctx.ui, {
      warn: (message) => ctx.ui.notify(message, "warning"),
    });
    unsubscribe = ctx.ui.onTerminalInput((data) => {
      const result = handleDoublePaste(data);
      if (result?.consume) {
        // TODO: Remove this notification after Pi repaints setEditorText()
        // calls from consumed terminal input: https://github.com/earendil-works/pi/issues/5620#issuecomment-4963678354
        try {
          ctx.ui.notify("Paste expanded.", "info");
        } catch {
          // Notification delivery must never interfere with input.
        }
      }
      return result;
    });
  });

  pi.on("session_shutdown", () => {
    unsubscribe?.();
    unsubscribe = undefined;
  });
}

export { createDoublePasteHandler } from "./double-paste.ts";
export {
  BRACKETED_PASTE_END,
  BRACKETED_PASTE_START,
  extractBracketedPaste,
  fingerprint,
  isLongPaste,
  normalizePaste,
} from "./paste.ts";