import { updateState } from "./state";
import { getState } from "./state";
import { preprocessBrowserEvent } from "./recording/utils";
import { Page } from "playwright";
import _ from "lodash";

export const handleBrowserEvent = (page: Page) => {
  const eventQueue: any[] = [];

  const processEvents = _.debounce(() => {
    if (eventQueue.length === 0) {
      return;
    }

    // Skip events for same element and type
    while (eventQueue.length > 1) {
      const currentEvent = eventQueue[0];
      const nextEvent = eventQueue[1];
      if (currentEvent.type === nextEvent.type && currentEvent.elementUUID === nextEvent.elementUUID) {
        eventQueue.shift();
      } else {
        break;
      }
    }

    const event = eventQueue.shift();
    const state = getState();

    preprocessBrowserEvent(event);

    if (state.messages.length > 0) {
      const lastMessage = state.messages[state.messages.length - 1];
      if (lastMessage.type === 'Interaction') {
        const lastInteraction = JSON.parse(lastMessage.content);
        if (lastInteraction.type === "input" && lastInteraction.elementUUID === event.elementUUID) {
          lastInteraction.typedText = event.typedText;
          state.messages[state.messages.length - 1] = {
            type: 'Interaction',
            content: JSON.stringify(lastInteraction),
            windowUrl: event.windowUrl,
          };
          updateState(page, state);
          return;
        }
      }
    }

    state.messages.push({
      type: 'Interaction',
      content: JSON.stringify(event),
      windowUrl: event.windowUrl,
    });
    updateState(page, state);
  }, 100, { maxWait: 500 });

  return (event: any) => {
    const state = getState();
    if (!state.recordingInteractions || state.pickingType) {
      return;
    }

    eventQueue.push(event);
    processEvents();
  }
}