105 lines
2.8 KiB
TypeScript
105 lines
2.8 KiB
TypeScript
|
import type { Book } from "../library";
|
||
|
|
import type { Browser, BrowserContext, ImageFile } from "../browser";
|
||
|
|
|
||
|
|
export function DlsiteManiax(browser: Browser) {
|
||
|
|
async function* getAllBooks(ctx: BrowserContext): AsyncGenerator<Book> {
|
||
|
|
const totalCountEndpoint = "https://play.dlsite.com/api/product_count";
|
||
|
|
const endpoint = "https://play.dlsite.com/api/purchases";
|
||
|
|
const pager = {
|
||
|
|
page: 1,
|
||
|
|
perPage: 50,
|
||
|
|
totalCount: Infinity,
|
||
|
|
};
|
||
|
|
|
||
|
|
const res = await ctx.request.get(totalCountEndpoint);
|
||
|
|
const body: {
|
||
|
|
user: number;
|
||
|
|
} = await res.json();
|
||
|
|
|
||
|
|
pager.totalCount = body.user;
|
||
|
|
|
||
|
|
while ((pager.page - 1) * 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: {
|
||
|
|
limit: number;
|
||
|
|
works: Array<{
|
||
|
|
workno: number;
|
||
|
|
name: {
|
||
|
|
ja_JP: string;
|
||
|
|
};
|
||
|
|
maker: {
|
||
|
|
name: {
|
||
|
|
ja_JP: string;
|
||
|
|
};
|
||
|
|
};
|
||
|
|
}>;
|
||
|
|
} = await res.json();
|
||
|
|
|
||
|
|
for (const work of Object.values(body.works).flat()) {
|
||
|
|
yield {
|
||
|
|
id: NaN,
|
||
|
|
platform: "dlsite-maniax",
|
||
|
|
readerUrl: `https://play.dlsite.com/#/work/${work.workno}`,
|
||
|
|
title: work.name.ja_JP || "",
|
||
|
|
authors: [work.maker.name.ja_JP || ""],
|
||
|
|
};
|
||
|
|
|
||
|
|
process.stderr.write(".");
|
||
|
|
}
|
||
|
|
|
||
|
|
pager.page += 1;
|
||
|
|
pager.perPage = body.limit;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
return {
|
||
|
|
async *pull(): AsyncGenerator<Book> {
|
||
|
|
const ctx = await browser.loadBrowserContext("dlsite-maniax");
|
||
|
|
|
||
|
|
yield* getAllBooks(ctx);
|
||
|
|
|
||
|
|
process.stderr.write(`\n`);
|
||
|
|
},
|
||
|
|
|
||
|
|
async getFiles(book: Book): Promise<Array<() => Promise<Blob>>> {
|
||
|
|
const ctx = await browser.loadBrowserContext("dlsite-maniax");
|
||
|
|
const page = await ctx.newPage();
|
||
|
|
|
||
|
|
await page.goto(book.readerUrl);
|
||
|
|
|
||
|
|
const [, workId] =
|
||
|
|
/^https:[/][/]play[.]dlsite[.]com[/]#[/]work[/]([^/]+)/.exec(
|
||
|
|
book.readerUrl,
|
||
|
|
) ?? [];
|
||
|
|
|
||
|
|
if (!workId) {
|
||
|
|
throw new Error(`workId is not included: ${book.readerUrl}`);
|
||
|
|
}
|
||
|
|
|
||
|
|
const url = `https://www.dlsite.com/home/download/=/product_id/${workId}.html`;
|
||
|
|
const imageFile = { url };
|
||
|
|
|
||
|
|
return [
|
||
|
|
async () => {
|
||
|
|
const blob = await browser.drawImage(page, imageFile);
|
||
|
|
|
||
|
|
process.stderr.write(".");
|
||
|
|
|
||
|
|
return blob;
|
||
|
|
},
|
||
|
|
];
|
||
|
|
},
|
||
|
|
loginEndpoints: ["https://www.dlsite.com/home/login"],
|
||
|
|
loginSuccessUrl: (url: URL) => url.origin === "https://www.dlsite.com",
|
||
|
|
logoutEndpoints: ["https://www.dlsite.com/home/logout"],
|
||
|
|
};
|
||
|
|
}
|
||
|
|
|
||
|
|
DlsiteManiax.siteUrl = (url: URL) =>
|
||
|
|
url.href.startsWith("https://play.dlsite.com/#/work/");
|