123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329
import { CTOR_KEY } from "../constructor-lock.ts";import { Node, NodeType, Text, Comment } from "./node.ts";import { NodeList, nodeListMutatorSym } from "./node-list.ts";import { Element } from "./element.ts";import { DOM as NWAPI } from "./nwsapi-types.ts";
export class DOMImplementation {  constructor(key: typeof CTOR_KEY) {    if (key !== CTOR_KEY) {      throw new TypeError("Illegal constructor.");    }  }
  createDocument() {    throw new Error("Unimplemented"); // TODO  }
  createHTMLDocument(titleStr?: string): HTMLDocument {    titleStr += "";
    const doc = new HTMLDocument(CTOR_KEY);
    const docType = new DocumentType("html", "", "", CTOR_KEY);    doc.appendChild(docType);
    const html = new Element("html", doc, [], CTOR_KEY);    html._setOwnerDocument(doc);
    const head = new Element("head", html, [], CTOR_KEY);    const body = new Element("body", html, [], CTOR_KEY);
    const title = new Element("title", head, [], CTOR_KEY);    const titleText = new Text(titleStr);    title.appendChild(titleText);
    doc.head = head;    doc.body = body;
    return doc;  }
  createDocumentType(qualifiedName: string, publicId: string, systemId: string): DocumentType {    const doctype = new DocumentType(qualifiedName, publicId, systemId, CTOR_KEY);
    return doctype;  }}
export class DocumentType extends Node {  #qualifiedName = "";  #publicId = "";  #systemId = "";
  constructor(    name: string,    publicId: string,    systemId: string,    key: typeof CTOR_KEY,  ) {    super(      "html",       NodeType.DOCUMENT_TYPE_NODE,       null,      key    );
    this.#qualifiedName = name;    this.#publicId = publicId;    this.#systemId = systemId;  }
  get name() {    return this.#qualifiedName;  }
  get publicId() {    return this.#publicId;  }
  get systemId() {    return this.#systemId;  }
  _shallowClone(): Node {    return new DocumentType(this.#qualifiedName, this.#publicId, this.#systemId, CTOR_KEY);  }}
export interface ElementCreationOptions {  is: string;}
export type VisibilityState = "visible" | "hidden" | "prerender";export type NamespaceURI = "http://www.w3.org/1999/xhtml" | "http://www.w3.org/2000/svg" | "http://www.w3.org/1998/Math/MathML";
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(      "#document",      NodeType.DOCUMENT_NODE,      null,      CTOR_KEY,    );
    this.implementation = new DOMImplementation(CTOR_KEY);  }
  _shallowClone(): Node {    return new Document();  }
  // Expose the document's NWAPI for Element's access to  // querySelector/querySelectorAll  get _nwapi() {    return this.#nwapi;  }
  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;  }
  get doctype(): DocumentType | null {    for (const node of this.childNodes) {      if (node.nodeType === NodeType.DOCUMENT_TYPE_NODE) {        return <DocumentType> node;      }    }
    return null;  }
  appendChild(child: Node): Node {    super.appendChild(child);    child._setOwnerDocument(this);    return child;  }
  createElement(tagName: string, options?: ElementCreationOptions): Element {    tagName = tagName.toUpperCase();
    const elm = new Element(tagName, null, [], CTOR_KEY);    elm._setOwnerDocument(this);    return elm;  }
  createElementNS(    namespace: NamespaceURI,    qualifiedName: string,    options?: ElementCreationOptions,  ): Element {    if (namespace === "http://www.w3.org/1999/xhtml") {      return this.createElement(qualifiedName, options);    } else {      throw new Error(`createElementNS: "${ namespace }" namespace unimplemented`); // TODO    }  }
  createTextNode(data?: string): Text {    return new Text(data);  }
  createComment(data?: string): Comment {    return new Comment(data);  }
  importNode(node: Node, deep: boolean = false) {    const copy = node.cloneNode(deep);
    copy._setOwnerDocument(this);
    return copy;  }
  adoptNode(node: Node) {    node._setParent(null);    node._setOwnerDocument(this);
    return node;  }
  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(key: typeof CTOR_KEY) {    if (key !== CTOR_KEY) {      throw new TypeError("Illegal constructor.");    }    super();  }
  _shallowClone(): Node {    return new HTMLDocument(CTOR_KEY);  }}