rewrite

2026-10-11 02:39:58 +00:00 · 2026-06-19 15:27:27 +02:00
parent a8935479e5
commit 099a76c39e
13 changed files with 603 additions and 342 deletions
--- a/src/index.js
+++ b/src/index.js
@@ -0,0 +1,123 @@
+import process from "node:process";
+import path from "node:path";
+import { parseArgs, printUsage, VALID_THEMES } from "./cli.js";
+import { loadPreferences, readCustomCss } from "./config.js";
+import {
+  checkExtension,
+  launchBrowser,
+  discoverExtensionId,
+  getReadabilityPath,
+} from "./browser.js";
+import {
+  extractArticle,
+  storeArticle,
+  setExtensionPreferences,
+  waitForReaderView,
+  extractRenderedContent,
+  getArticleTitle,
+} from "./reader.js";
+import {
+  getOutputFilename,
+  resolveOutputPath,
+  enhanceHtml,
+  generatePdf,
+} from "./pdf.js";
+
+const EXTENSION_BASE = "chrome-extension://";
+
+async function main() {
+  const { url, opts } = parseArgs();
+  if (!url) {
+    printUsage();
+    process.exit(1);
+  }
+
+  checkExtension();
+
+  let preferences = {};
+  if (opts.prefs) {
+    preferences = loadPreferences(path.resolve(opts.prefs));
+  }
+
+  const theme = opts.theme || preferences.mode || "light";
+  if (!VALID_THEMES.includes(theme)) {
+    console.error(
+      `Invalid theme "${theme}". Valid themes: ${VALID_THEMES.join(", ")}`,
+    );
+    process.exit(1);
+  }
+  preferences.mode = theme;
+  console.log(`Theme: ${theme}`);
+
+  let customCss = null;
+  if (opts.css) {
+    customCss = readCustomCss(path.resolve(opts.css));
+    if (customCss !== null) {
+      preferences["user-css"] = customCss;
+      console.log(`Custom CSS loaded: ${path.resolve(opts.css)}`);
+    }
+  }
+
+  const browser = await launchBrowser();
+
+  try {
+    const extId = await discoverExtensionId(browser);
+
+    console.log(`Fetching article: ${url}`);
+    const articlePage = await browser.newPage();
+    await articlePage.goto(url, { waitUntil: "networkidle0", timeout: 30000 });
+    const article = await extractArticle(articlePage, getReadabilityPath());
+    await articlePage.close();
+
+    const extPage = await browser.newPage();
+    await storeArticle(extPage, extId, 1, article);
+
+    const prefsToSet = Object.fromEntries(
+      Object.entries(preferences).filter(
+        ([_, v]) => v !== undefined && v !== null,
+      ),
+    );
+    if (Object.keys(prefsToSet).length > 0) {
+      await setExtensionPreferences(extPage, extId, prefsToSet);
+    }
+    await extPage.close();
+
+    const readerUrl = [
+      `${EXTENSION_BASE}${extId}/data/reader/index.html`,
+      "?id=1",
+      `&url=${encodeURIComponent(url)}`,
+    ].join("");
+    console.log(`Opening Reader View: ${readerUrl}`);
+
+    const readerPage = await browser.newPage();
+    await readerPage.goto(readerUrl, { waitUntil: "load", timeout: 30000 });
+
+    await waitForReaderView(readerPage);
+    await new Promise((r) => setTimeout(r, 1000));
+
+    const title = await getArticleTitle(readerPage);
+    const outputFile = opts.output
+      ? resolveOutputPath(opts.output)
+      : getOutputFilename(title);
+
+    const contentHtml = await extractRenderedContent(
+      readerPage,
+      `${EXTENSION_BASE}${extId}/data/reader/`,
+    );
+    await readerPage.close();
+
+    const enhancedHtml = enhanceHtml(contentHtml, customCss);
+
+    const pdfPage = await browser.newPage();
+    await generatePdf(pdfPage, enhancedHtml, outputFile);
+    await pdfPage.close();
+  } finally {
+    await browser.close();
+    console.log("Browser closed.");
+  }
+}
+
+main().catch((err) => {
+  console.error("Fatal error:", err);
+  process.exit(1);
+});