@playcraft/cli 0.0.39 → 0.0.41

package/dist/commands/tools-utils.js

@@ -0,0 +1,164 @@
+import { writeFileSync, mkdirSync, readFileSync } from 'fs';
+import { join, parse } from 'path';
+import { tmpdir } from 'os';
+export const TMP_DIR = join(tmpdir(), 'playcraft');
+export const MAX_REFERENCE_IMAGES = 8;
+export function ensureTmpDir() {
+    mkdirSync(TMP_DIR, { recursive: true });
+}
+export function tmpPath(prefix) {
+    return join(TMP_DIR, `${prefix}-${Date.now()}.json`);
+}
+export function writeResult(prefix, data) {
+    ensureTmpDir();
+    const path = tmpPath(prefix);
+    writeFileSync(path, JSON.stringify(data, null, 2));
+    return path;
+}
+export function handleError(err) {
+    const msg = err instanceof Error ? err.message : String(err);
+    console.error(`Error: ${msg}`);
+    process.exit(1);
+}
+/** Fallback extension from API mime when magic-byte sniff fails. */
+export function extensionForImageMime(mimeType) {
+    const base = mimeType.toLowerCase().split(';')[0]?.trim() ?? '';
+    switch (base) {
+        case 'image/png':
+            return '.png';
+        case 'image/jpeg':
+        case 'image/jpg':
+            return '.jpg';
+        case 'image/webp':
+            return '.webp';
+        case 'image/gif':
+            return '.gif';
+        case 'image/bmp':
+            return '.bmp';
+        default:
+            return '.png';
+    }
+}
+/**
+ * Detect container from decoded bytes (magic). Prefer this over MIME alone.
+ */
+export function sniffImageExtension(buf) {
+    if (buf.length >= 8 &&
+        buf[0] === 0x89 &&
+        buf[1] === 0x50 &&
+        buf[2] === 0x4e &&
+        buf[3] === 0x47) {
+        return '.png';
+    }
+    if (buf.length >= 3 && buf[0] === 0xff && buf[1] === 0xd8 && buf[2] === 0xff) {
+        return '.jpg';
+    }
+    if (buf.length >= 6 &&
+        buf[0] === 0x47 &&
+        buf[1] === 0x49 &&
+        buf[2] === 0x46 &&
+        buf[3] === 0x38 &&
+        (buf[4] === 0x37 || buf[4] === 0x39) &&
+        buf[5] === 0x61) {
+        return '.gif';
+    }
+    if (buf.length >= 12 &&
+        buf[0] === 0x52 &&
+        buf[1] === 0x49 &&
+        buf[2] === 0x46 &&
+        buf[3] === 0x46 &&
+        buf[8] === 0x57 &&
+        buf[9] === 0x45 &&
+        buf[10] === 0x42 &&
+        buf[11] === 0x50) {
+        return '.webp';
+    }
+    if (buf.length >= 2 && buf[0] === 0x42 && buf[1] === 0x4d) {
+        return '.bmp';
+    }
+    return null;
+}
+/** Replace output extension when it does not match the chosen format extension. */
+export function resolveImageOutputPath(outputPath, wantExt) {
+    try {
+        const { dir, name, ext } = parse(outputPath);
+        if (ext.toLowerCase() === wantExt) {
+            return outputPath;
+        }
+        return join(dir, `${name}${wantExt}`);
+    }
+    catch (e) {
+        const detail = e instanceof Error ? e.message : String(e);
+        console.warn(`Could not adjust output path for extension ${wantExt} (${detail}); using path as given: ${outputPath}`);
+        return outputPath;
+    }
+}
+export function collectReferenceImagePaths(value, previous) {
+    return previous.concat([value]);
+}
+export function mimeTypeForImagePath(filePath) {
+    const ext = filePath.split('.').pop()?.toLowerCase();
+    if (ext === 'png')
+        return 'image/png';
+    if (ext === 'webp')
+        return 'image/webp';
+    if (ext === 'jpg' || ext === 'jpeg')
+        return 'image/jpeg';
+    if (ext === undefined || ext === '') {
+        throw new Error(`Reference image path has no file extension: ${filePath}. Use .png, .webp, .jpg, or .jpeg.`);
+    }
+    throw new Error(`Unsupported reference image extension ".${ext}" in ${filePath}. Only png, webp, jpg, and jpeg are allowed.`);
+}
+async function downloadImageFromUrl(url) {
+    let res;
+    try {
+        res = await fetch(url, { signal: AbortSignal.timeout(30_000) });
+    }
+    catch (e) {
+        const detail = e instanceof Error ? e.message : String(e);
+        throw new Error(`下载参考图失败（网络错误）：${url}\n  ${detail}`);
+    }
+    if (!res.ok) {
+        throw new Error(`下载参考图失败（HTTP ${res.status}）：${url}\n  服务器返回了非 2xx 状态码，请检查 URL 是否有效或可公开访问。`);
+    }
+    const contentType = res.headers.get('content-type') ?? '';
+    const mimeType = contentType.split(';')[0].trim() || 'image/png';
+    const allowed = ['image/png', 'image/jpeg', 'image/webp', 'image/gif'];
+    if (!allowed.some((t) => mimeType.startsWith(t.split('/')[0]) && mimeType.includes('/')) || !mimeType.startsWith('image/')) {
+        throw new Error(`参考图 URL 返回的 Content-Type 不是图片（${mimeType}）：${url}\n  只支持 PNG / JPG / WEBP / GIF。`);
+    }
+    const buf = await res.arrayBuffer();
+    if (buf.byteLength === 0) {
+        throw new Error(`参考图 URL 返回了空响应体：${url}`);
+    }
+    return { base64: Buffer.from(buf).toString('base64'), mimeType };
+}
+/** Load reference images for generate-image; skip unreadable entries after console.warn. */
+export async function collectReferenceImagePayloads(paths) {
+    if (paths.length === 0)
+        return undefined;
+    const out = [];
+    for (const p of paths) {
+        const isUrl = p.startsWith('http://') || p.startsWith('https://');
+        try {
+            if (isUrl) {
+                const payload = await downloadImageFromUrl(p);
+                out.push(payload);
+            }
+            else {
+                const mimeType = mimeTypeForImagePath(p);
+                out.push({ base64: readFileSync(p).toString('base64'), mimeType });
+            }
+        }
+        catch (e) {
+            const detail = e instanceof Error ? e.message : String(e);
+            const code = !isUrl && e && typeof e === 'object' && 'code' in e ? String(e.code) : '';
+            const hint = code ? ` (${code})` : '';
+            console.warn(`Skipping reference image (load failed): ${p}${hint}\n  ${detail}`);
+        }
+    }
+    if (paths.length > 0 && out.length === 0) {
+        console.warn('All reference images failed to load; continuing with text-only generation.');
+    }
+    return out.length > 0 ? out : undefined;
+}