import type { Book } from "../library";
import type { Browser, BrowserContext, ImageFile } from "../browser";
import { getImageFiles } from "./dmm-books";

export function FanzaDoujin(browser: Browser) {
  async function* getAllBooks(ctx: BrowserContext): AsyncGenerator<Book> {
    const endpoint =
      "https://www.dmm.co.jp/dc/doujin/api/mylibraries/?limit=20";
    const pager = {
      page: 1,
      perPage: 20,
      totalCount: Infinity,
    };

    while (pager.page * pager.perPage <= pager.totalCount) {
      const res = await ctx.request.get(`${endpoint}&page=${pager.page}`);

      if (!res.ok()) {
        throw new Error(`${res.status()} ${res.statusText()}`);
      }

      const body: {
        data: {
          items: Record<
            string,
            Array<{
              productId: string;
              title: string;
              makerName: string;
            }>
          >;
          total: number;
        };
      } = await res.json();

      for (const item of Object.values(body.data.items).flat()) {
        yield {
          id: NaN,
          platform: "fanza-doujin",
          readerUrl: `https://www.dmm.co.jp/dc/-/mylibrary/detail/=/product_id=${item.productId}/`,
          title: item.title || "",
          authors: [item.makerName],
        };

        process.stderr.write(".");
      }

      pager.page += 1;
      pager.totalCount = body.data.total;
    }
  }

  return {
    async login() {
      const ctx = await browser.newContext();
      const page = await ctx.newPage();
      await page.goto("https://accounts.dmm.co.jp/service/login/password");
      await page.waitForURL("https://www.dmm.co.jp/top/", { timeout: 0 });
      await browser.saveBrowserContext("fanza-doujin", ctx);
    },

    async *pull(): AsyncGenerator<Book> {
      const ctx = await browser.loadBrowserContext("fanza-doujin");

      yield* getAllBooks(ctx);

      process.stderr.write(`\n`);
    },

    async getFiles(book: Book): Promise<Array<() => Promise<string>>> {
      const ctx = await browser.loadBrowserContext("fanza-doujin");
      const page = await ctx.newPage();

      await page.goto(book.readerUrl);
      await page.waitForSelector(`li[class^="fileTreeItem"]`);
      await page.click(`li[class^="fileTreeItem"]>a`);
      await page.waitForURL((url) =>
        url.href.startsWith("https://www.dmm.co.jp/dc/-/viewer/=/product_id="),
      );

      const imageFiles: Array<ImageFile> = await page.evaluate(getImageFiles);

      return imageFiles.map((imageFile) => async () => {
        const dataUrl = await browser.drawImage(page, imageFile);

        process.stderr.write(".");

        return dataUrl;
      });
    },
  };
}

FanzaDoujin.siteUrl = (url: URL) =>
  url.href.startsWith(
    "https://www.dmm.co.jp/dc/-/mylibrary/detail/=/product_id=",
  );