import { WebDriver, By, Key } from "selenium-webdriver";

export type CompletionItem = {
  type: string;
  text: string;
  highlight: boolean;
};

export class Console {
  constructor(private webdriver: WebDriver) {}

  async sendKeys(...keys: string[]) {
    const input = await this.webdriver.findElement(By.css("input"));
    input.sendKeys(...keys);
  }

  async currentValue() {
    return await this.webdriver.executeScript(() => {
      const input = document.querySelector("input");
      if (input === null) {
        throw new Error("could not find input element");
      }
      return input.value;
    });
  }

  async execCommand(command: string): Promise<void> {
    const input = await this.webdriver.findElement(
      By.css("input.vimvixen-console-command-input")
    );
    await input.sendKeys(command, Key.ENTER);
  }

  async getErrorMessage(): Promise<string> {
    const p = await this.webdriver.findElement(
      By.css(".vimvixen-console-error")
    );
    return p.getText();
  }

  async getInformationMessage(): Promise<string> {
    const p = await this.webdriver.findElement(
      By.css(".vimvixen-console-info")
    );
    return p.getText();
  }

  async inputKeys(...keys: string[]) {
    const input = await this.webdriver.findElement(By.css("input"));
    await input.sendKeys(...keys);
  }

  getCompletions(): Promise<CompletionItem[]> {
    return this.webdriver.executeScript(() => {
      const items = document.querySelectorAll(
        ".vimvixen-console-completion > li"
      );
      if (items.length === 0) {
        throw new Error("completion items not found");
      }

      const objs = [];
      for (const li of Array.from(items)) {
        if (li.classList.contains("vimvixen-console-completion-title")) {
          objs.push({ type: "title", text: li.textContent!!.trim() });
        } else if ("vimvixen-console-completion-item") {
          const highlight = li.classList.contains(
            "vimvixen-completion-selected"
          );
          objs.push({ type: "item", text: li.textContent!!.trim(), highlight });
        } else {
          throw new Error(`unexpected class: ${li.className}`);
        }
      }
      return objs;
    });
  }

  async getTheme(): Promise<string> {
    const wrapper = await this.webdriver.findElement(By.css("div[data-theme]"));
    const theme = await wrapper.getAttribute("data-theme");
    return theme;
  }

  async close(): Promise<void> {
    const input = await this.webdriver.findElement(By.css("input"));
    await input.sendKeys(Key.ESCAPE);
    // TODO remove sleep
    await new Promise((resolve) => setTimeout(resolve, 100));
    await (this.webdriver.switchTo() as any).parentFrame();
  }
}