agent-reader 1.0.0 → 1.1.0

package/src/core/exporter.js

@@ -1,4 +1,4 @@
-import { promises as fs } from 'node:fs';
+import { accessSync, constants as fsConstants, promises as fs } from 'node:fs';
 import os from 'node:os';
 import path from 'node:path';
 import { createRequire } from 'node:module';
@@ -7,6 +7,7 @@ import { execa } from 'execa';
 import MarkdownIt from 'markdown-it';
 import {
   Document,
+  ExternalHyperlink,
   HeadingLevel,
   Packer,
   Paragraph,
@@ -25,6 +26,109 @@ const LUA_TABLE_FILTER = path.join(__dirname, 'templates', 'docx-table.lua');
 const POSTPROCESS_SCRIPT = path.join(__dirname, '..', '..', 'scripts', 'postprocess-docx.py');
 
 const markdownParser = new MarkdownIt({ html: false, linkify: true, typographer: true });
+const SANDBOX_DISABLED_ARGS = ['--no-sandbox', '--disable-setuid-sandbox', '--disable-dev-shm-usage'];
+const SANDBOX_MODES = new Set(['auto', 'on', 'off']);
+
+function isSandboxMode(mode) {
+  return SANDBOX_MODES.has(mode);
+}
+
+function normalizeSandboxMode(mode) {
+  if (typeof mode !== 'string') {
+    return null;
+  }
+  const normalized = mode.trim().toLowerCase();
+  return isSandboxMode(normalized) ? normalized : null;
+}
+
+function fileExistsSync(filePath) {
+  try {
+    accessSync(filePath, fsConstants.F_OK);
+    return true;
+  } catch {
+    return false;
+  }
+}
+
+export function detectContainerEnv(runtime = {}) {
+  const env = runtime.env ?? process.env;
+  const fileExists = runtime.fileExists ?? fileExistsSync;
+  const getuid = Object.prototype.hasOwnProperty.call(runtime, 'getuid')
+    ? runtime.getuid
+    : process.getuid;
+
+  if (fileExists('/.dockerenv')) {
+    return true;
+  }
+  if (fileExists('/run/.containerenv')) {
+    return true;
+  }
+  if (env?.CI === 'true') {
+    return true;
+  }
+
+  try {
+    if (typeof getuid === 'function' && getuid() === 0) {
+      return true;
+    }
+  } catch {
+    return false;
+  }
+
+  return false;
+}
+
+export function resolveSandboxMode(requestedMode, env = process.env) {
+  const fromExplicit = normalizeSandboxMode(requestedMode);
+  if (fromExplicit) {
+    return fromExplicit;
+  }
+  const fromEnv = normalizeSandboxMode(env?.AGENT_READER_SANDBOX);
+  if (fromEnv) {
+    return fromEnv;
+  }
+  return 'auto';
+}
+
+export function getSandboxArgs(sandboxMode = 'auto', runtime = {}) {
+  const mode = normalizeSandboxMode(sandboxMode) || 'auto';
+  if (mode === 'on') {
+    return [];
+  }
+  if (mode === 'off') {
+    return [...SANDBOX_DISABLED_ARGS];
+  }
+  return detectContainerEnv(runtime) ? [...SANDBOX_DISABLED_ARGS] : [];
+}
+
+function runHasProperty(run, propKey) {
+  return run.properties?.root?.some((item) => item.rootKey === propKey);
+}
+
+function cloneRunWithStyle(run, { bold, italics }) {
+  const baseTextNode = run.root.find((node) => node.rootKey === 'w:t');
+  const text = typeof baseTextNode?.root?.[baseTextNode.root.length - 1] === 'string'
+    ? baseTextNode.root[baseTextNode.root.length - 1]
+    : '';
+  const hasCodeStyle = runHasProperty(run, 'w:rFonts');
+  const hasShading = runHasProperty(run, 'w:shd');
+
+  return new TextRun({
+    text,
+    bold: bold || undefined,
+    italics: italics || undefined,
+    ...(hasCodeStyle ? { font: 'Consolas' } : {}),
+    ...(hasShading
+      ? {
+        shading: {
+          type: ShadingType.CLEAR,
+          color: 'auto',
+          fill: 'F4F4F5',
+        },
+      }
+      : {}),
+  });
+}
 
 function paragraphFromText(text) {
   return new Paragraph({
@@ -65,6 +169,139 @@ function inlineToPlainText(token) {
     .join('');
 }
 
+function createInlineTextRun(text, { bold, italics, code = false }) {
+  return new TextRun({
+    text,
+    bold: bold || undefined,
+    italics: italics || undefined,
+    ...(code
+      ? {
+        font: 'Consolas',
+        shading: {
+          type: ShadingType.CLEAR,
+          color: 'auto',
+          fill: 'F4F4F5',
+        },
+      }
+      : {}),
+  });
+}
+
+function pushInlineNode(target, linkStack, node) {
+  if (linkStack.length > 0) {
+    linkStack[linkStack.length - 1].children.push(node);
+    return;
+  }
+  target.push(node);
+}
+
+function closeCurrentLink(target, linkStack, style) {
+  const current = linkStack.pop();
+  if (!current) {
+    return;
+  }
+
+  const href = current.href || '';
+  if (!href) {
+    const fallbackRuns = current.children.length > 0
+      ? current.children
+      : [createInlineTextRun('', style)];
+    for (const run of fallbackRuns) {
+      pushInlineNode(target, linkStack, run);
+    }
+    return;
+  }
+
+  const hyperlinkChildren = current.children.length > 0
+    ? current.children
+    : [createInlineTextRun(href, style)];
+  const hyperlink = new ExternalHyperlink({
+    link: href,
+    children: hyperlinkChildren.map((node) => (
+      node instanceof TextRun ? cloneRunWithStyle(node, style) : node
+    )),
+  });
+  pushInlineNode(target, linkStack, hyperlink);
+}
+
+export function inlineToDocxRuns(token) {
+  if (!token?.children?.length) {
+    const fallbackText = token?.content || '';
+    return fallbackText ? [new TextRun(fallbackText)] : [];
+  }
+
+  const output = [];
+  const linkStack = [];
+  let boldDepth = 0;
+  let italicsDepth = 0;
+
+  for (const child of token.children) {
+    const style = {
+      bold: boldDepth > 0,
+      italics: italicsDepth > 0,
+    };
+
+    if (child.type === 'strong_open') {
+      boldDepth += 1;
+      continue;
+    }
+    if (child.type === 'strong_close') {
+      boldDepth = Math.max(0, boldDepth - 1);
+      continue;
+    }
+    if (child.type === 'em_open') {
+      italicsDepth += 1;
+      continue;
+    }
+    if (child.type === 'em_close') {
+      italicsDepth = Math.max(0, italicsDepth - 1);
+      continue;
+    }
+    if (child.type === 'link_open') {
+      linkStack.push({
+        href: child.attrGet('href') || '',
+        children: [],
+      });
+      continue;
+    }
+    if (child.type === 'link_close') {
+      closeCurrentLink(output, linkStack, style);
+      continue;
+    }
+    if (child.type === 'text') {
+      if (child.content) {
+        pushInlineNode(output, linkStack, createInlineTextRun(child.content, style));
+      }
+      continue;
+    }
+    if (child.type === 'code_inline') {
+      pushInlineNode(output, linkStack, createInlineTextRun(child.content || '', { ...style, code: true }));
+      continue;
+    }
+    if (child.type === 'softbreak' || child.type === 'hardbreak') {
+      pushInlineNode(output, linkStack, createInlineTextRun('\n', style));
+      continue;
+    }
+    if (child.type === 'image') {
+      const alt = child.content || child.attrGet('alt') || 'image';
+      pushInlineNode(output, linkStack, createInlineTextRun(`[${alt}]`, style));
+      continue;
+    }
+    if (child.content) {
+      pushInlineNode(output, linkStack, createInlineTextRun(child.content, style));
+    }
+  }
+
+  while (linkStack.length > 0) {
+    closeCurrentLink(output, linkStack, {
+      bold: boldDepth > 0,
+      italics: italicsDepth > 0,
+    });
+  }
+
+  return output;
+}
+
 function collectTocEntries(tokens) {
   const entries = [];
   for (let i = 0; i < tokens.length; i += 1) {
@@ -123,9 +360,11 @@ function parseTable(tokens, startIndex) {
           rows: rows.map(
             (items) =>
               new TableRow({
-                children: items.map((item) =>
+                children: items.map((itemRuns) =>
                   new TableCell({
-                    children: [paragraphFromText(item || '')],
+                    children: [new Paragraph({
+                      children: itemRuns.length > 0 ? itemRuns : [new TextRun('')],
+                    })],
                   }),
                 ),
               }),
@@ -146,17 +385,17 @@ function parseTable(tokens, startIndex) {
     }
 
     if (token.type === 'th_open' || token.type === 'td_open') {
-      currentCell = '';
+      currentCell = [];
       continue;
     }
 
     if (token.type === 'inline' && currentCell !== null) {
-      currentCell += inlineToPlainText(token);
+      currentCell.push(...inlineToDocxRuns(token));
       continue;
     }
 
     if (token.type === 'th_close' || token.type === 'td_close') {
-      row.push(currentCell || '');
+      row.push(currentCell || []);
       currentCell = null;
     }
   }
@@ -167,10 +406,10 @@ function parseTable(tokens, startIndex) {
   };
 }
 
-function markdownToDocx(markdown) {
+export function markdownToDocx(markdown) {
   const tokens = markdownParser.parse(markdown, {});
   const tocEntries = collectTocEntries(tokens);
-  const children = [];
+  const children = [...buildDocxTocParagraphs(tocEntries)];
   const listStack = [];
   let pendingBlock = null;
 
@@ -237,15 +476,15 @@ function markdownToDocx(markdown) {
       continue;
     }
 
-    const text = inlineToPlainText(token).trim();
-    if (!text) {
+    const runs = inlineToDocxRuns(token);
+    if (runs.length === 0) {
       continue;
     }
 
     if (pendingBlock?.type === 'heading') {
       children.push(
         new Paragraph({
-          text,
+          children: runs,
           heading: headingLevelFromNumber(pendingBlock.level),
         }),
       );
@@ -256,11 +495,13 @@ function markdownToDocx(markdown) {
     if (listStack.length > 0) {
       const top = listStack[listStack.length - 1];
       const prefix = top.type === 'ordered' ? `${top.count}. ` : '• ';
-      children.push(paragraphFromText(`${prefix}${text}`));
+      children.push(new Paragraph({
+        children: [new TextRun(prefix), ...runs],
+      }));
       continue;
     }
 
-    children.push(paragraphFromText(text));
+    children.push(new Paragraph({ children: runs }));
     pendingBlock = null;
   }
 
@@ -431,6 +672,24 @@ export async function checkPuppeteer() {
   }
 }
 
+export function getPandocInstallHint(platform = process.platform) {
+  if (platform === 'darwin') {
+    return 'brew install pandoc';
+  }
+  if (platform === 'win32') {
+    return 'winget install pandoc (or choco install pandoc / scoop install pandoc)';
+  }
+  return 'apt-get install pandoc (or yum install pandoc)';
+}
+
+export function createPandocFallbackWarnings(platform = process.platform) {
+  const installHint = getPandocInstallHint(platform);
+  return [
+    'pandoc_not_found',
+    `⚡ 当前使用基础排版模式。想要更精美的代码高亮和智能表格？一行命令解锁：${installHint}`,
+  ];
+}
+
 export async function exportPDF(html, options = {}) {
   const {
     pageSize = 'A4',
@@ -438,6 +697,7 @@ export async function exportPDF(html, options = {}) {
     outDir = os.tmpdir(),
     fileName = 'output.pdf',
     htmlPath,
+    sandbox,
   } = options;
 
   let puppeteer;
@@ -450,10 +710,13 @@ export async function exportPDF(html, options = {}) {
 
   const warnings = [];
   const pdfPath = path.join(path.resolve(outDir), fileName);
+  const resolvedSandboxMode = resolveSandboxMode(sandbox);
+  const launchArgs = landscape ? ['--allow-file-access-from-files'] : [];
+  launchArgs.push(...getSandboxArgs(resolvedSandboxMode));
 
   const browser = await puppeteer.launch({
     headless: true,
-    args: landscape ? ['--allow-file-access-from-files'] : [],
+    args: launchArgs,
   });
   try {
     const page = await browser.newPage();
@@ -587,8 +850,7 @@ export async function exportDOCX(markdownString, options = {}) {
       await fs.rm(tempInput, { force: true }).catch(() => {});
     }
   } else {
-    const warning = 'Pandoc not found, using docx fallback. Install Pandoc for better results.';
-    warnings.push(warning);
+    warnings.push(...createPandocFallbackWarnings());
 
     const document = markdownToDocx(markdownString);
     const buffer = await Packer.toBuffer(document);
@@ -677,7 +939,7 @@ export async function exportDOCXFromHTML(htmlString, options = {}) {
       await fs.rm(tempInput, { force: true }).catch(() => {});
     }
   } else {
-    warnings.push('Pandoc not found, using text-only DOCX fallback.');
+    warnings.push(...createPandocFallbackWarnings());
 
     const plainText = String(htmlString || '')
       .replace(/<style[\s\S]*?<\/style>/gi, ' ')

package/src/core/opener.js

@@ -3,7 +3,7 @@ import path from 'node:path';
 import { createOutputDir } from '../utils/output.js';
 import { normalizeOpenMode } from '../utils/preferences.js';
 import { renderMarkdown } from './renderer.js';
-import { exportDOCX, exportPDF } from './exporter.js';
+import { exportDOCX, exportPDF, resolveSandboxMode } from './exporter.js';
 import { createSlideshow } from './slideshow.js';
 
 const MARKDOWN_EXTENSIONS = new Set(['.md', '.markdown', '.mdown', '.mdx']);
@@ -94,7 +94,9 @@ export async function openTarget(targetPath, options = {}) {
     fetchRemote = true,
     returnContent = false,
     maxContentBytes = 50 * 1024 * 1024,
+    sandbox,
   } = options;
+  const resolvedSandbox = resolveSandboxMode(sandbox, process.env);
 
   const requestedMode = normalizeOpenMode(mode, 'auto');
   const preferredMode = requestedMode === 'auto'
@@ -258,6 +260,7 @@ export async function openTarget(targetPath, options = {}) {
       outDir: outputDir,
       fileName: `${name}.pdf`,
       htmlPath,
+      sandbox: resolvedSandbox,
     });
     warnings.push(...pdf.warnings);
 

package/src/core/renderer.js

@@ -13,7 +13,8 @@ const THEME_DIR = path.join(__dirname, 'themes');
 const TEMPLATE_PATH = path.resolve(__dirname, '../templates/document.html');
 
 const GITHUB_MARKDOWN_CSS_PATH = require.resolve('github-markdown-css/github-markdown.css');
-const HIGHLIGHT_CSS_PATH = require.resolve('highlight.js/styles/github.css');
+const HIGHLIGHT_CSS_LIGHT = require.resolve('highlight.js/styles/github.css');
+const HIGHLIGHT_CSS_DARK = require.resolve('highlight.js/styles/github-dark.css');
 
 const TOC_BASE_CSS = `
 .markdown-body :is(h1, h2, h3, h4, h5, h6) {
@@ -165,18 +166,18 @@ async function loadStyles(theme) {
   const builtInThemePath =
     themeName === 'dark'
       ? path.join(THEME_DIR, 'dark.css')
-      : themeName === 'print'
-        ? path.join(THEME_DIR, 'print.css')
-        : path.join(THEME_DIR, 'light.css');
+      : path.join(THEME_DIR, 'light.css');
   const customThemePath =
-    themeName === 'light' || themeName === 'dark' || themeName === 'print'
+    themeName === 'light' || themeName === 'dark'
       ? null
       : path.resolve(themeName);
 
+  const highlightCssPath = themeName === 'dark' ? HIGHLIGHT_CSS_DARK : HIGHLIGHT_CSS_LIGHT;
+
   const [builtInTheme, githubCss, highlightCss, customTheme = ''] = await Promise.all([
     fs.readFile(builtInThemePath, 'utf8'),
     fs.readFile(GITHUB_MARKDOWN_CSS_PATH, 'utf8'),
-    fs.readFile(HIGHLIGHT_CSS_PATH, 'utf8'),
+    fs.readFile(highlightCssPath, 'utf8'),
     customThemePath ? fs.readFile(customThemePath, 'utf8') : Promise.resolve(''),
   ]);