Vim-Vixen/e2e/lib/Console.ts

import { WebDriver, By } from 'selenium-webdriver';

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

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

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

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

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

      let objs = [];
      for (let 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') {
          let 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;
    });
  }
}