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

class ComradeMaoPlugin implements Plugin.PluginBase {
    id = "comrademao";
    name = "Comrade Mao";
    site = "https://comrademao.com/";
    version = "1.0.0";
    icon = "src/en/comrademao/icon.png";

    parseNovels(loadedCheerio: CheerioAPI) {
        const novels: Plugin.NovelItem[] = [];

        loadedCheerio(".listupd")
            .find("div.bs")
            .each(function () {
                const novelName = loadedCheerio(this).find(".tt").text().trim();
                const novelCover = loadedCheerio(this).find("img").attr("src");
                const novelUrl = loadedCheerio(this).find("a").attr("href");

                if (!novelUrl) return;

                const novel = {
                    name: novelName,
                    cover: novelCover,
                    url: novelUrl,
                };
                novels.push(novel);
            });
        return novels;
    }

    async popularNovels(
        pageNo: number,
        { filters }: Plugin.PopularNovelsOptions<typeof this.filters>,
    ): Promise<Plugin.NovelItem[]> {
        let link = this.site;

        link += filters.category.value;

        link += "page/" + pageNo + "/?post_type=novel";

        const body = await fetchApi(link).then((result) => result.text());

        const loadedCheerio = parseHTML(body);
        return this.parseNovels(loadedCheerio);
    }

    async parseNovelAndChapters(novelUrl: string): Promise<Plugin.SourceNovel> {
        const body = await fetchApi(novelUrl).then((result) => result.text());

        let loadedCheerio = parseHTML(body);

        const novel: Plugin.SourceNovel = {
            url: novelUrl,
            chapters: [],
        };

        novel.name = loadedCheerio(".entry-title").text().trim();

        novel.cover = loadedCheerio("div.thumbook > div > img").attr("src");

        loadedCheerio(".infox .wd-full").each((i, el) => {
            const detailName = loadedCheerio(el).find("b").text();
            const detail = loadedCheerio(el).find("span").text();

            switch (detailName) {
                case "Publisher: ":
                    novel.author = detail.trim();
                    break;
                case "Status: ":
                    novel.status = detail.trim();
                    break;
                case "Genres:":
                    novel.genres = detail.trim().replace(/ , /g, ",");
                    break;
                case "Description: ":
                    novel.summary = detail.trim();
                    break;
            }
        });

        const chapter: Plugin.ChapterItem[] = [];

        loadedCheerio("#chapterlist")
            .find("li")
            .each(function () {
                const chapterUrl = loadedCheerio(this).find("a").attr("href");
                if (!chapterUrl) return;

                const chapterName = loadedCheerio(this)
                    .find(".chapternum")
                    .text();
                let releaseTime = loadedCheerio(this)
                    .find(".chapterdate")
                    .text();

                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");
                    }
                }
                chapter.push({
                    name: chapterName,
                    url: chapterUrl,
                    releaseTime,
                });
            });

        novel.chapters = chapter.reverse();

        return novel;
    }

    async parseChapter(chapterUrl: string): Promise<string> {
        const body = await fetchApi(chapterUrl).then((result) => result.text());
        const loadedCheerio = parseHTML(body);

        const chapterText = loadedCheerio("#chaptercontent").html()?.trim() || "";

        return chapterText;
    }

    async searchNovels(
        searchTerm: string,
        pageNo: number,
    ): Promise<Plugin.NovelItem[]> {
        const url = `${this.site}page/${pageNo}/?s=${searchTerm}&post_type=novel`;

        const body = await fetchApi(url).then((result) => result.text());

        const loadedCheerio = parseHTML(body);
        return this.parseNovels(loadedCheerio);
    }

    async fetchImage(url: string): Promise<string | undefined> {
        return await fetchFile(url);
    }

    filters = {
        category: {
            value: "",
            label: "Type",
            options: [
                { label: "None", value: "" },
                { label: "Chinese", value: "/mtype/chinese/" },
                { label: "Japanese", value: "/mtype/japanese/" },
                { label: "Korean", value: "/mtype/korean/" },
                { label: "Manhua", value: "/mtype/manhua/" },
            ],
            type: FilterTypes.Picker,
        },
    } satisfies Filters;
}

export default new ComradeMaoPlugin();