visible/visible

import {
  BaseSession,
  format,
  HTMLRootNode,
  Session,
  Settings,
  Source,
  SourceType,
} from '@visi/core';
import { parseDOM } from 'htmlparser2';
import { JSDOM } from 'jsdom';

const HTML_ID = 'main';

export class SessionJsdomImpl extends BaseSession implements Session {
  readonly sources = new Map<string, Source>();
  private htmlIdsMap = new Map<string, string>();
  private domWindow!: JSDOM;
  private url!: string;

  constructor(private readonly settings: Settings) {
    super();
  }

  async render(_html: string): Promise<void> {
    this.url = 'https://localhost';
    this.domWindow = new JSDOM(_html, {
      resources: 'usable',
      runScripts: 'dangerously',
    });
    this.storeHTML();
  }

  async goto(url: string): Promise<void> {
    this.url = url;
    this.domWindow = await JSDOM.fromURL(url, {
      resources: 'usable',
      runScripts: 'dangerously',
    });
    this.storeHTML();
  }

  getActiveHTML(): Source {
    const id = this.htmlIdsMap.get(HTML_ID);
    if (id == null) throw new Error('No html stored with id main');
    const source = this.sources.get(id);
    if (source == null) throw new Error(`No source stored with id ${id}`);
    return source;
  }

  async getTitle(): Promise<string> {
    return (
      this.domWindow.window.document.querySelector('title')?.textContent ??
      'Untitled'
    );
  }

  async getURL(): Promise<string> {
    return this.url;
  }

  async close(): Promise<void> {
    return this.domWindow.window.close();
  }

  async eval<T>(script: string): Promise<T> {
    return this.domWindow.window.eval(script) as T;
  }

  async takeScreenshotForPage(): Promise<string> {
    // eslint-disable-next-line
    console.warn('unimplemented');
    return '/';
  }

  async takeScreenshotForXPath(): Promise<string> {
    // eslint-disable-next-line
    console.warn('unimplemented');
    return '/';
  }

  async findCSS(_xpath: string, _propertyName: string): Promise<undefined> {
    throw new Error('stub');
  }

  private storeHTML() {
    const html = this.settings.format
      ? format(SourceType.HTML, this.domWindow.serialize())
      : this.domWindow.serialize();

    const node = parseDOM(html, {
      withEndIndices: true,
      withStartIndices: true,
    });
    const source = new Source({
      type: SourceType.HTML,
      url: this.url,
      node: new HTMLRootNode(node),
    });
    this.htmlIdsMap.set(HTML_ID, source.id);
    this.sources.set(source.id, source);
  }
}