import { afterEach, describe, expect, it } from "vitest";

import { isTerminalFocused } from "./terminalFocus";

class MockHTMLElement {
  isConnected = false;
  className = "";

  readonly classList = {
    contains: (value: string) => this.className.split(/\s+/).includes(value),
  };

  closest(selector: string): MockHTMLElement | null {
    if (!this.isConnected) {
      return null;
    }
    if (selector === ".thread-terminal-drawer .xterm" || selector === ".thread-terminal-drawer") {
      return this;
    }
    return null;
  }
}

const originalDocument = globalThis.document;
const originalHTMLElement = globalThis.HTMLElement;

afterEach(() => {
  if (originalDocument === undefined) {
    delete (globalThis as { document?: Document }).document;
  } else {
    globalThis.document = originalDocument;
  }

  if (originalHTMLElement === undefined) {
    delete (globalThis as { HTMLElement?: typeof HTMLElement }).HTMLElement;
  } else {
    globalThis.HTMLElement = originalHTMLElement;
  }
});

describe("isTerminalFocused", () => {
  it("returns false for detached xterm helper textareas", () => {
    const detached = new MockHTMLElement();
    detached.className = "xterm-helper-textarea";

    globalThis.HTMLElement = MockHTMLElement as unknown as typeof HTMLElement;
    globalThis.document = { activeElement: detached } as Document;

    expect(isTerminalFocused()).toBe(false);
  });

  it("returns true for connected xterm helper textareas", () => {
    const attached = new MockHTMLElement();
    attached.className = "xterm-helper-textarea";
    attached.isConnected = true;

    globalThis.HTMLElement = MockHTMLElement as unknown as typeof HTMLElement;
    globalThis.document = { activeElement: attached } as Document;

    expect(isTerminalFocused()).toBe(true);
  });

  it("returns true for focus inside the terminal drawer (e.g. sidebar)", () => {
    const sidebarButton = new MockHTMLElement();
    sidebarButton.className = "terminal-sidebar-button";
    sidebarButton.isConnected = true;

    globalThis.HTMLElement = MockHTMLElement as unknown as typeof HTMLElement;
    globalThis.document = { activeElement: sidebarButton } as Document;

    expect(isTerminalFocused()).toBe(true);
  });
});