ao3-deno/src/classes/Chapter.ts

import { DOMParser, Element, HTMLDocument, ID } from "../types.d.ts";

export default class Chapter {
    #session: {
        get: (path: string) => Promise<Response>;
    };
    #document!: HTMLDocument;
    #DOMParser: DOMParser;
    id: ID;
    workID: ID;
    name!: string;
    html!: string;
    text!: string;
    summary!: string;
    startNote!: string;
    endNote!: string;

    constructor(
        workId: ID,
        id: ID,
        session: {
            get: (path: string) => Promise<Response>;
        },
        DOMParser: DOMParser,
        // deno-lint-ignore no-explicit-any
        extraInfo: Record<string, any>,
    ) {
        this.#session = session;
        this.workID = workId;
        this.id = id;
        this.#DOMParser = DOMParser;
        this.name = extraInfo.name;
    }

    async init() {
        const res = await this.#session.get(
            `/works/${this.workID}/chapters/${this.id}?view_adult=true`,
        );
        this.#document = this.#DOMParser.parseFromString(
            await res.text(),
            "text/html",
        ) as HTMLDocument;
        this.populateMetadata();
        this.populateSummary();
        this.populateNotes();
        await this.populateText();
    }

    // deletes our #document if we want to clean up memory and not use a shitton
    freeDocument() {
        //completely sane behaviour
        this.#document = undefined as unknown as HTMLDocument;
    }

    populateMetadata() {
        this.name = this.#document.querySelector("h3.title")?.innerText
            .replace(
                /Chapter \d+: /,
                "",
            ).trim() as string;
    }

    populateSummary() {
        this.summary = this.#document.querySelector("#summary > .userstuff")
            ?.innerText.trim() as string;
    }

    populateNotes() {
        const notesList = Array.from(
            this.#document.querySelectorAll(".notes > .userstuff"),
        ).map((n) => (n as Element).innerHTML);
        this.startNote = notesList[0]?.trim()?.replace(/<\/{0,1}p>/g, "\n")
            ?.trim();
        this.endNote = notesList[1]?.trim()?.replace(/<\/{0,1}p>/g, "\n")
            ?.trim();
    }

    async populateText() {
        this.text = "";

        const elements = this.#document.querySelectorAll(
            "div.userstuff[role='article'] > p",
        );

        for (let i = 0; i < elements.length; i++) {
            const element = elements[i] as Element;

            this.text += element.innerText + "\n";
        }
        try {
            this.text = this.text.trim();

            this.html = (this.#document.querySelector(
                "div.userstuff[role='article']",
            ) as Element).innerHTML;
        } catch {
            //assume single chapter work
            const res = await this.#session.get(
                `/works/${this.workID}?view_adult=true`,
            );
            this.#document = this.#DOMParser.parseFromString(
                await res.text(),
                "text/html",
            ) as HTMLDocument;

            this.html = (this.#document.querySelector(
                "[role='article'] > div.userstuff",
            ) as Element).innerHTML;

            const elements = this.#document.querySelectorAll(
                "[role='article'] > div.userstuff > p",
            );

            for (let i = 0; i < elements.length; i++) {
                const element = elements[i] as Element;

                this.text += element.innerText + "\n";
                this.html += element.innerHTML;
            }
        }
    }
}