51 lines
1.3 KiB
TypeScript
51 lines
1.3 KiB
TypeScript
import fs from "node:fs/promises";
|
|
import path from "node:path";
|
|
import type { Book } from "./library";
|
|
import type { Browser } from "./browser";
|
|
import type { Database } from "./database";
|
|
import { DmmBooks } from "./platforms/dmm-books";
|
|
import { GooglePlayBooks } from "./platforms/google-play-books";
|
|
|
|
const platforms = {
|
|
"dmm-books": DmmBooks,
|
|
"google-play-books": GooglePlayBooks,
|
|
};
|
|
|
|
export type TPlatform = keyof typeof platforms;
|
|
|
|
export function site(url: string): TPlatform {
|
|
const { origin } = new URL(url);
|
|
|
|
for (const [platform, { site }] of Object.entries(platforms)) {
|
|
if (site.includes(origin)) return platform as TPlatform;
|
|
}
|
|
|
|
throw new Error(`Unsupported URL: ${url}`);
|
|
}
|
|
|
|
export function createPlatform(opts: {
|
|
platform: TPlatform;
|
|
db: Database;
|
|
browser: Browser;
|
|
}) {
|
|
if (!(opts.platform in platforms)) {
|
|
throw new Error(`Available platform: ${Object.keys(platforms).join(", ")}`);
|
|
}
|
|
|
|
const platform = platforms[opts.platform](opts.browser);
|
|
|
|
return {
|
|
...platform,
|
|
async download(dir: string, book: Book): Promise<void> {
|
|
await fs.mkdir(path.dirname(dir), { recursive: true });
|
|
await fs.mkdir(dir);
|
|
await platform.download(dir, book);
|
|
},
|
|
async logout() {
|
|
await opts.db.run(
|
|
`update platforms set secrets = 'null' where name = ?`,
|
|
opts.platform,
|
|
);
|
|
},
|
|
};
|
|
}
|