123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283
import { getLock, setLock } from "./constructor-lock.ts";import { Element } from "./element.ts";import { NodeList, nodeListMutatorSym } from "./node-list.ts";import { Comment, Node, NodeType, Text } from "./node.ts";import { DOM as NWAPI } from "./nwsapi-types.ts";
export class DOMImplementation {  constructor() {    if (getLock()) {      throw new TypeError("Illegal constructor.");    }  }
  createDocument() {    throw new Error("Unimplemented"); // TODO  }
  createHTMLDocument(titleStr?: string): HTMLDocument {    titleStr += "";
    // TODO: Figure out a way to make `setLock` invocations less redundant    setLock(false);    const doc = new HTMLDocument();
    setLock(false);    const docType = new DocumentType("html", "", "");    doc.appendChild(docType);
    const html = new Element("html", doc, []);    html._setOwnerDocument(doc);
    const head = new Element("head", html, []);    const body = new Element("body", html, []);
    const title = new Element("title", head, []);    const titleText = new Text(titleStr);    title.appendChild(titleText);
    doc.head = head;    doc.body = body;
    setLock(true);    return doc;  }
  createDocumentType(qualifiedName: string, publicId: string, systemId: string): DocumentType {    setLock(false);    const doctype = new DocumentType(qualifiedName, publicId, systemId);    setLock(true);
    return doctype;  }}
export class DocumentType extends Node {  #qualifiedName = "";  #publicId = "";  #systemId = "";
  constructor(    name: string,    publicId: string,    systemId: string,  ) {    super(      "html",      NodeType.DOCUMENT_TYPE_NODE,      null    );
    this.#qualifiedName = name;    this.#publicId = publicId;    this.#systemId = systemId;  }
  get name() {    return this.#qualifiedName;  }
  get publicId() {    return this.#publicId;  }
  get systemId() {    return this.#systemId;  }}
export interface ElementCreationOptions {  is: string;}
export type VisibilityState = "visible" | "hidden" | "prerender";
export class Document extends Node {  public head: Element = <Element><unknown>null;  public body: Element = <Element><unknown>null;  public implementation: DOMImplementation;
  #lockState = false;  #documentURI = "about:blank"; // TODO  #title = "";  #nwapi = NWAPI(this);
  constructor() {    super(      (setLock(false), "#document"),      NodeType.DOCUMENT_NODE,      null,    );
    setLock(false);    this.implementation = new DOMImplementation();    setLock(true);  }
  get documentURI() {    return this.#documentURI;  }
  get title() {    return this.querySelector("title")?.textContent || "";  }
  get cookie() {    return ""; // TODO  }
  set cookie(newCookie: string) {    // TODO  }
  get visibilityState(): VisibilityState {    return "visible";  }
  get hidden() {    return false;  }
  get compatMode(): string {    return "CSS1Compat";  }
  get documentElement(): Element | null {    for (const node of this.childNodes) {      if (node.nodeType === NodeType.ELEMENT_NODE) {        return <Element>node;      }    }
    return null;  }
  appendChild(child: Node) {    super.appendChild(child);    child._setOwnerDocument(this);  }
  createElement(tagName: string, options?: ElementCreationOptions): Element {    tagName = tagName.toUpperCase();
    setLock(false);    const elm = new Element(tagName, null, []);    elm._setOwnerDocument(this);    setLock(true);    return elm;  }
  createTextNode(data?: string): Text {    return new Text(data);  }
  createComment(data?: string): Comment {    return new Comment(data);  }
  querySelector(selectors: string): Element | null {    return this.#nwapi.first(selectors, this);  }
  querySelectorAll(selectors: string): NodeList {    const nodeList = new NodeList();    const mutator = nodeList[nodeListMutatorSym]();    mutator.push(...this.#nwapi.select(selectors, this))
    return nodeList;  }
  // TODO: DRY!!!  getElementById(id: string): Element | null {    for (const child of this.childNodes) {      if (child.nodeType === NodeType.ELEMENT_NODE) {        if ((<Element>child).id === id) {          return <Element>child;        }
        const search = (<Element>child).getElementById(id);        if (search) {          return search;        }      }    }
    return null;  }
  getElementsByTagName(tagName: string): Element[] {    if (tagName === "*") {      return this.documentElement        ? <Element[]>this._getElementsByTagNameWildcard(this.documentElement, [])        : [];    } else {      return <Element[]>this._getElementsByTagName(tagName.toUpperCase(), []);    }  }
  private _getElementsByTagNameWildcard(node: Node, search: Node[]): Node[] {    for (const child of this.childNodes) {      if (child.nodeType === NodeType.ELEMENT_NODE) {        search.push(child);        (<any>child)._getElementsByTagNameWildcard(search);      }    }
    return search;  }
  private _getElementsByTagName(tagName: string, search: Node[]): Node[] {    for (const child of this.childNodes) {      if (child.nodeType === NodeType.ELEMENT_NODE) {        if ((<Element>child).tagName === tagName) {          search.push(child);        }
        (<any>child)._getElementsByTagName(tagName, search);      }    }
    return search;  }
  getElementsByTagNameNS(_namespace: string, localName: string): Element[] {    return this.getElementsByTagName(localName);  }
  getElementsByClassName(className: string): Element[] {    return <Element[]>this._getElementsByClassName(className, []);  }
  private _getElementsByClassName(className: string, search: Node[]): Node[] {    for (const child of this.childNodes) {      if (child.nodeType === NodeType.ELEMENT_NODE) {        if ((<Element>child).classList.contains(className)) {          search.push(child);        }
        (<any>child)._getElementsByClassName(className, search);      }    }
    return search;  }
  hasFocus(): boolean {    return true;  }}
export class HTMLDocument extends Document {  constructor() {    let lock = getLock();    super();
    if (lock) {      throw new TypeError("Illegal constructor.");    }
    setLock(false);  }}