refactor: share markdown renderer via baoyu-md

skills/baoyu-markdown-to-html/scripts/vendor/baoyu-md/src/images.ts

@@ -0,0 +1,127 @@
+import { createHash } from "node:crypto";
+import fs from "node:fs";
+import http from "node:http";
+import https from "node:https";
+import path from "node:path";
+
+export interface ImagePlaceholder {
+  originalPath: string;
+  placeholder: string;
+}
+
+export interface ResolvedImageInfo extends ImagePlaceholder {
+  localPath: string;
+}
+
+export function replaceMarkdownImagesWithPlaceholders(
+  markdown: string,
+  placeholderPrefix: string,
+): {
+  images: ImagePlaceholder[];
+  markdown: string;
+} {
+  const images: ImagePlaceholder[] = [];
+  let imageCounter = 0;
+
+  const rewritten = markdown.replace(/!\[([^\]]*)\]\(([^)]+)\)/g, (_match, _alt, src) => {
+    const placeholder = `${placeholderPrefix}${++imageCounter}`;
+    images.push({
+      originalPath: src,
+      placeholder,
+    });
+    return placeholder;
+  });
+
+  return { images, markdown: rewritten };
+}
+
+export function getImageExtension(urlOrPath: string): string {
+  const match = urlOrPath.match(/\.(jpg|jpeg|png|gif|webp)(\?|$)/i);
+  return match ? match[1]!.toLowerCase() : "png";
+}
+
+export async function downloadFile(url: string, destPath: string): Promise<void> {
+  return await new Promise((resolve, reject) => {
+    const protocol = url.startsWith("https://") ? https : http;
+    const file = fs.createWriteStream(destPath);
+
+    const request = protocol.get(url, { headers: { "User-Agent": "Mozilla/5.0" } }, (response) => {
+      if (response.statusCode === 301 || response.statusCode === 302) {
+        const redirectUrl = response.headers.location;
+        if (redirectUrl) {
+          file.close();
+          fs.unlinkSync(destPath);
+          void downloadFile(redirectUrl, destPath).then(resolve).catch(reject);
+          return;
+        }
+      }
+
+      if (response.statusCode !== 200) {
+        file.close();
+        fs.unlinkSync(destPath);
+        reject(new Error(`Failed to download: ${response.statusCode}`));
+        return;
+      }
+
+      response.pipe(file);
+      file.on("finish", () => {
+        file.close();
+        resolve();
+      });
+    });
+
+    request.on("error", (error) => {
+      file.close();
+      fs.unlink(destPath, () => {});
+      reject(error);
+    });
+
+    request.setTimeout(30_000, () => {
+      request.destroy();
+      reject(new Error("Download timeout"));
+    });
+  });
+}
+
+export async function resolveImagePath(
+  imagePath: string,
+  baseDir: string,
+  tempDir: string,
+  logLabel = "baoyu-md",
+): Promise<string> {
+  if (imagePath.startsWith("http://") || imagePath.startsWith("https://")) {
+    const hash = createHash("md5").update(imagePath).digest("hex").slice(0, 8);
+    const ext = getImageExtension(imagePath);
+    const localPath = path.join(tempDir, `remote_${hash}.${ext}`);
+
+    if (!fs.existsSync(localPath)) {
+      console.error(`[${logLabel}] Downloading: ${imagePath}`);
+      await downloadFile(imagePath, localPath);
+    }
+    return localPath;
+  }
+
+  if (path.isAbsolute(imagePath)) {
+    return imagePath;
+  }
+
+  return path.resolve(baseDir, imagePath);
+}
+
+export async function resolveContentImages(
+  images: ImagePlaceholder[],
+  baseDir: string,
+  tempDir: string,
+  logLabel = "baoyu-md",
+): Promise<ResolvedImageInfo[]> {
+  const resolved: ResolvedImageInfo[] = [];
+
+  for (const image of images) {
+    resolved.push({
+      ...image,
+      localPath: await resolveImagePath(image.originalPath, baseDir, tempDir, logLabel),
+    });
+  }
+
+  return resolved;
+}