lnreader-plugins/scripts/multisrc/readwn/template.ts

import { fetchFile, fetchApi } from "@libs/fetch";
import { Filters, FilterTypes } from "@libs/filterInputs";
import { Plugin } from "@typings/plugin";
import { NovelStatus } from "@libs/novelStatus";
import { load as parseHTML } from "cheerio";
import dayjs from "dayjs";
import qs from "qs";

export interface ReadwnMetadata {
  id: string;
  sourceSite: string;
  sourceName: string;
  filters?: Filters;
}

class ReadwnPlugin implements Plugin.PluginBase {
  id: string;
  name: string;
  icon: string;
  site: string;
  version: string;
  filters?: Filters;

  constructor(metadata: ReadwnMetadata) {
    this.id = metadata.id;
    this.name = metadata.sourceName;
    this.icon = `multisrc/readwn/icons/${metadata.id}.png`;
    this.site = metadata.sourceSite;
    this.version = "1.0.0";
    this.filters = metadata.filters;
  }

  async popularNovels(
    pageNo: number,
    { filters, showLatestNovels }: Plugin.PopularNovelsOptions,
  ): Promise<Plugin.NovelItem[]> {
    let url = this.site + "/list/";
    url += (filters?.genres?.value || "all") + "/";
    url += (filters?.status?.value || "all") + "-";
    url += showLatestNovels ? "lastdotime" : filters?.sort?.value || "newstime";
    url += "-" + (pageNo - 1) + ".html";

    if (filters?.tags?.value) {
      //only 1 page
      url = this.site + "/tags/" + filters.tags.value + "-0.html";
    }

    const body = await fetchApi(url).then((res) => res.text());
    const loadedCheerio = parseHTML(body);

    const novels: Plugin.NovelItem[] = loadedCheerio("li.novel-item")
      .map((index, element) => ({
        name: loadedCheerio(element).find("h4").text() || "",
        cover:
          this.site +
          loadedCheerio(element).find(".novel-cover > img").attr("data-src"),
        path: loadedCheerio(element).find("a").attr("href") || "",
      }))
      .get()
      .filter((novel) => novel.name && novel.path);

    return novels;
  }

  async parseNovel(novelPath: string): Promise<Plugin.SourceNovel> {
    const body = await fetchApi(this.site + novelPath).then((res) => res.text());
    const loadedCheerio = parseHTML(body);

    const novel: Plugin.SourceNovel = {
      path: novelPath,
      name: loadedCheerio("h1.novel-title").text() || "",
    };

    novel.author = loadedCheerio("span[itemprop=author]").text();
    novel.cover =
      this.site + loadedCheerio("figure.cover > img").attr("data-src");

    novel.summary = loadedCheerio(".summary")
      .text()
      .replace("Summary", "")
      .trim();

    novel.genres = loadedCheerio("div.categories > ul > li")
      .map((index, element) => loadedCheerio(element).text()?.trim())
      .get()
      .join(",");

    loadedCheerio("div.header-stats > span").each(function () {
      if (loadedCheerio(this).find("small").text() === "Status") {
        novel.status =
          loadedCheerio(this).find("strong").text() === "Ongoing"
            ? NovelStatus.Ongoing
            : NovelStatus.Completed;
      }
    });

    const latestChapterNo = parseInt(
      loadedCheerio(".header-stats")
        .find("span > strong")
        .first()
        .text()
        .trim(),
    );

    const chapters: Plugin.ChapterItem[] = loadedCheerio(".chapter-list li")
      .map((chapterIndex, element) => {
        const name = loadedCheerio(element)
          .find("a .chapter-title")
          .text()
          .trim();
        const path = loadedCheerio(element).find("a").attr("href")?.trim();
        if (!name || !path) return null;

        let releaseTime = loadedCheerio(element)
          .find("a .chapter-update")
          .text()
          .trim();
        if (releaseTime?.includes?.("ago")) {
          const timeAgo = releaseTime.match(/\d+/)?.[0] || "0";
          const timeAgoInt = parseInt(timeAgo, 10);

          if (timeAgoInt) {
            const dayJSDate = dayjs(); // today
            if (
              releaseTime.includes("hours ago") ||
              releaseTime.includes("hour ago")
            ) {
              dayJSDate.subtract(timeAgoInt, "hours"); // go back N hours
            }

            if (
              releaseTime.includes("days ago") ||
              releaseTime.includes("day ago")
            ) {
              dayJSDate.subtract(timeAgoInt, "days"); // go back N days
            }

            if (
              releaseTime.includes("months ago") ||
              releaseTime.includes("month ago")
            ) {
              dayJSDate.subtract(timeAgoInt, "months"); // go back N months
            }

            releaseTime = dayJSDate.format("LL");
          }
        }

        return {
          name,
          path,
          releaseTime,
          chapterNumber: chapterIndex + 1,
        };
      })
      .get()
      .filter((chapter) => chapter);

    if (latestChapterNo > chapters.length) {
      const lastChapterNo = parseInt(
        chapters[chapters.length - 1].path.match(/_(\d+)\.html/)?.[1] || "",
        10,
      );

      for (let i = (lastChapterNo || chapters.length) + 1; i <= latestChapterNo; i++) {
        chapters.push({
          name: "Chapter " + i,
          path: novelPath.replace(".html", "_" + i + ".html"),
          releaseTime: null,
          chapterNumber: i,
        });
      }
    }

    novel.chapters = chapters;
    return novel;
  }

  async parseChapter(chapterPath: string): Promise<string> {
    const body = await fetchApi(this.site + chapterPath).then((res) => res.text());
    const loadedCheerio = parseHTML(body);

    const chapterText = loadedCheerio(".chapter-content").html() || "";
    return chapterText;
  }

  async searchNovels(searchTerm: string): Promise<Plugin.NovelItem[]> {
    const result = await fetchApi(this.site + "/e/search/index.php", {
      headers: {
        "Content-Type": "application/x-www-form-urlencoded",
        Referer: this.site + "/search.html",
        Origin: this.site,
      },
      method: "POST",
      body: qs.stringify({
        show: "title",
        tempid: 1,
        tbname: "news",
        keyboard: searchTerm,
      }),
    }).then((res) => res.text());
    const loadedCheerio = parseHTML(result);

    const novels: Plugin.NovelItem[] = loadedCheerio("li.novel-item")
      .map((index, element) => ({
        name: loadedCheerio(element).find("h4").text() || "",
        cover: this.site + loadedCheerio(element).find("img").attr("data-src"),
        path: loadedCheerio(element).find("a").attr("href") || "",
      }))
      .get()
      .filter((novel) => novel.name && novel.path);
    return novels;
  }

  fetchImage = fetchFile;
}