@tomsun28/pizza 0.1.0 → 0.1.1

package/dist/utils/exif-orientation.js

@@ -0,0 +1,157 @@
+function readOrientationFromTiff(bytes, tiffStart) {
+    if (tiffStart + 8 > bytes.length)
+        return 1;
+    const byteOrder = (bytes[tiffStart] << 8) | bytes[tiffStart + 1];
+    const le = byteOrder === 0x4949;
+    const read16 = (pos) => {
+        if (le)
+            return bytes[pos] | (bytes[pos + 1] << 8);
+        return (bytes[pos] << 8) | bytes[pos + 1];
+    };
+    const read32 = (pos) => {
+        if (le)
+            return bytes[pos] | (bytes[pos + 1] << 8) | (bytes[pos + 2] << 16) | (bytes[pos + 3] << 24);
+        return ((bytes[pos] << 24) | (bytes[pos + 1] << 16) | (bytes[pos + 2] << 8) | bytes[pos + 3]) >>> 0;
+    };
+    const ifdOffset = read32(tiffStart + 4);
+    const ifdStart = tiffStart + ifdOffset;
+    if (ifdStart + 2 > bytes.length)
+        return 1;
+    const entryCount = read16(ifdStart);
+    for (let i = 0; i < entryCount; i++) {
+        const entryPos = ifdStart + 2 + i * 12;
+        if (entryPos + 12 > bytes.length)
+            return 1;
+        if (read16(entryPos) === 0x0112) {
+            const value = read16(entryPos + 8);
+            return value >= 1 && value <= 8 ? value : 1;
+        }
+    }
+    return 1;
+}
+function findJpegTiffOffset(bytes) {
+    let offset = 2;
+    while (offset < bytes.length - 1) {
+        if (bytes[offset] !== 0xff)
+            return -1;
+        const marker = bytes[offset + 1];
+        if (marker === 0xff) {
+            offset++;
+            continue;
+        }
+        if (marker === 0xe1) {
+            if (offset + 4 >= bytes.length)
+                return -1;
+            const segmentStart = offset + 4;
+            if (segmentStart + 6 > bytes.length)
+                return -1;
+            if (!hasExifHeader(bytes, segmentStart))
+                return -1;
+            return segmentStart + 6;
+        }
+        if (offset + 4 > bytes.length)
+            return -1;
+        const length = (bytes[offset + 2] << 8) | bytes[offset + 3];
+        offset += 2 + length;
+    }
+    return -1;
+}
+function findWebpTiffOffset(bytes) {
+    let offset = 12;
+    while (offset + 8 <= bytes.length) {
+        const chunkId = String.fromCharCode(bytes[offset], bytes[offset + 1], bytes[offset + 2], bytes[offset + 3]);
+        const chunkSize = bytes[offset + 4] | (bytes[offset + 5] << 8) | (bytes[offset + 6] << 16) | (bytes[offset + 7] << 24);
+        const dataStart = offset + 8;
+        if (chunkId === "EXIF") {
+            if (dataStart + chunkSize > bytes.length)
+                return -1;
+            // Some WebP files have "Exif\0\0" prefix before the TIFF header
+            const tiffStart = chunkSize >= 6 && hasExifHeader(bytes, dataStart) ? dataStart + 6 : dataStart;
+            return tiffStart;
+        }
+        // RIFF chunks are padded to even size
+        offset = dataStart + chunkSize + (chunkSize % 2);
+    }
+    return -1;
+}
+function hasExifHeader(bytes, offset) {
+    return (bytes[offset] === 0x45 &&
+        bytes[offset + 1] === 0x78 &&
+        bytes[offset + 2] === 0x69 &&
+        bytes[offset + 3] === 0x66 &&
+        bytes[offset + 4] === 0x00 &&
+        bytes[offset + 5] === 0x00);
+}
+function getExifOrientation(bytes) {
+    let tiffOffset = -1;
+    // JPEG: starts with FF D8
+    if (bytes.length >= 2 && bytes[0] === 0xff && bytes[1] === 0xd8) {
+        tiffOffset = findJpegTiffOffset(bytes);
+    }
+    // WebP: starts with RIFF....WEBP
+    else if (bytes.length >= 12 &&
+        bytes[0] === 0x52 &&
+        bytes[1] === 0x49 &&
+        bytes[2] === 0x46 &&
+        bytes[3] === 0x46 &&
+        bytes[8] === 0x57 &&
+        bytes[9] === 0x45 &&
+        bytes[10] === 0x42 &&
+        bytes[11] === 0x50) {
+        tiffOffset = findWebpTiffOffset(bytes);
+    }
+    if (tiffOffset === -1)
+        return 1;
+    return readOrientationFromTiff(bytes, tiffOffset);
+}
+function rotate90(photon, image, dstIndex) {
+    const w = image.get_width();
+    const h = image.get_height();
+    const src = image.get_raw_pixels();
+    const dst = new Uint8Array(src.length);
+    for (let y = 0; y < h; y++) {
+        for (let x = 0; x < w; x++) {
+            const srcIdx = (y * w + x) * 4;
+            const dstIdx = dstIndex(x, y, w, h) * 4;
+            dst[dstIdx] = src[srcIdx];
+            dst[dstIdx + 1] = src[srcIdx + 1];
+            dst[dstIdx + 2] = src[srcIdx + 2];
+            dst[dstIdx + 3] = src[srcIdx + 3];
+        }
+    }
+    return new photon.PhotonImage(dst, h, w);
+}
+// Flip orientations mutate in-place. Rotations return a new image (caller must free the old one if different).
+export function applyExifOrientation(photon, image, originalBytes) {
+    const orientation = getExifOrientation(originalBytes);
+    if (orientation === 1)
+        return image;
+    switch (orientation) {
+        case 2:
+            photon.fliph(image);
+            return image;
+        case 3:
+            photon.fliph(image);
+            photon.flipv(image);
+            return image;
+        case 4:
+            photon.flipv(image);
+            return image;
+        case 5: {
+            const rotated = rotate90(photon, image, (x, y, _w, h) => x * h + (h - 1 - y));
+            photon.fliph(rotated);
+            return rotated;
+        }
+        case 6:
+            return rotate90(photon, image, (x, y, _w, h) => x * h + (h - 1 - y));
+        case 7: {
+            const rotated = rotate90(photon, image, (x, y, w, h) => (w - 1 - x) * h + y);
+            photon.fliph(rotated);
+            return rotated;
+        }
+        case 8:
+            return rotate90(photon, image, (x, y, w, h) => (w - 1 - x) * h + y);
+        default:
+            return image;
+    }
+}

package/dist/utils/frontmatter.d.ts

@@ -0,0 +1,7 @@
+type ParsedFrontmatter<T extends Record<string, unknown>> = {
+    frontmatter: T;
+    body: string;
+};
+export declare const parseFrontmatter: <T extends Record<string, unknown> = Record<string, unknown>>(content: string) => ParsedFrontmatter<T>;
+export declare const stripFrontmatter: (content: string) => string;
+export {};

package/dist/utils/frontmatter.js

@@ -0,0 +1,25 @@
+import { parse } from "yaml";
+const normalizeNewlines = (value) => value.replace(/\r\n/g, "\n").replace(/\r/g, "\n");
+const extractFrontmatter = (content) => {
+    const normalized = normalizeNewlines(content);
+    if (!normalized.startsWith("---")) {
+        return { yamlString: null, body: normalized };
+    }
+    const endIndex = normalized.indexOf("\n---", 3);
+    if (endIndex === -1) {
+        return { yamlString: null, body: normalized };
+    }
+    return {
+        yamlString: normalized.slice(4, endIndex),
+        body: normalized.slice(endIndex + 4).trim(),
+    };
+};
+export const parseFrontmatter = (content) => {
+    const { yamlString, body } = extractFrontmatter(content);
+    if (!yamlString) {
+        return { frontmatter: {}, body };
+    }
+    const parsed = parse(yamlString);
+    return { frontmatter: (parsed ?? {}), body };
+};
+export const stripFrontmatter = (content) => parseFrontmatter(content).body;

package/dist/utils/git.d.ts

@@ -0,0 +1,25 @@
+/**
+ * Parsed git URL information.
+ */
+export type GitSource = {
+    /** Always "git" for git sources */
+    type: "git";
+    /** Clone URL (always valid for git clone, without ref suffix) */
+    repo: string;
+    /** Git host domain (e.g., "github.com") */
+    host: string;
+    /** Repository path (e.g., "user/repo") */
+    path: string;
+    /** Git ref (branch, tag, commit) if specified */
+    ref?: string;
+    /** True if ref was specified (package won't be auto-updated) */
+    pinned: boolean;
+};
+/**
+ * Parse git source into a GitSource.
+ *
+ * Rules:
+ * - With git: prefix, accept all historical shorthand forms.
+ * - Without git: prefix, only accept explicit protocol URLs.
+ */
+export declare function parseGitUrl(source: string): GitSource | null;

package/dist/utils/git.js

@@ -0,0 +1,162 @@
+import hostedGitInfo from "hosted-git-info";
+function splitRef(url) {
+    const scpLikeMatch = url.match(/^git@([^:]+):(.+)$/);
+    if (scpLikeMatch) {
+        const pathWithMaybeRef = scpLikeMatch[2] ?? "";
+        const refSeparator = pathWithMaybeRef.indexOf("@");
+        if (refSeparator < 0)
+            return { repo: url };
+        const repoPath = pathWithMaybeRef.slice(0, refSeparator);
+        const ref = pathWithMaybeRef.slice(refSeparator + 1);
+        if (!repoPath || !ref)
+            return { repo: url };
+        return {
+            repo: `git@${scpLikeMatch[1] ?? ""}:${repoPath}`,
+            ref,
+        };
+    }
+    if (url.includes("://")) {
+        try {
+            const parsed = new URL(url);
+            const pathWithMaybeRef = parsed.pathname.replace(/^\/+/, "");
+            const refSeparator = pathWithMaybeRef.indexOf("@");
+            if (refSeparator < 0)
+                return { repo: url };
+            const repoPath = pathWithMaybeRef.slice(0, refSeparator);
+            const ref = pathWithMaybeRef.slice(refSeparator + 1);
+            if (!repoPath || !ref)
+                return { repo: url };
+            parsed.pathname = `/${repoPath}`;
+            return {
+                repo: parsed.toString().replace(/\/$/, ""),
+                ref,
+            };
+        }
+        catch {
+            return { repo: url };
+        }
+    }
+    const slashIndex = url.indexOf("/");
+    if (slashIndex < 0) {
+        return { repo: url };
+    }
+    const host = url.slice(0, slashIndex);
+    const pathWithMaybeRef = url.slice(slashIndex + 1);
+    const refSeparator = pathWithMaybeRef.indexOf("@");
+    if (refSeparator < 0) {
+        return { repo: url };
+    }
+    const repoPath = pathWithMaybeRef.slice(0, refSeparator);
+    const ref = pathWithMaybeRef.slice(refSeparator + 1);
+    if (!repoPath || !ref) {
+        return { repo: url };
+    }
+    return {
+        repo: `${host}/${repoPath}`,
+        ref,
+    };
+}
+function parseGenericGitUrl(url) {
+    const { repo: repoWithoutRef, ref } = splitRef(url);
+    let repo = repoWithoutRef;
+    let host = "";
+    let path = "";
+    const scpLikeMatch = repoWithoutRef.match(/^git@([^:]+):(.+)$/);
+    if (scpLikeMatch) {
+        host = scpLikeMatch[1] ?? "";
+        path = scpLikeMatch[2] ?? "";
+    }
+    else if (repoWithoutRef.startsWith("https://") ||
+        repoWithoutRef.startsWith("http://") ||
+        repoWithoutRef.startsWith("ssh://") ||
+        repoWithoutRef.startsWith("git://")) {
+        try {
+            const parsed = new URL(repoWithoutRef);
+            host = parsed.hostname;
+            path = parsed.pathname.replace(/^\/+/, "");
+        }
+        catch {
+            return null;
+        }
+    }
+    else {
+        const slashIndex = repoWithoutRef.indexOf("/");
+        if (slashIndex < 0) {
+            return null;
+        }
+        host = repoWithoutRef.slice(0, slashIndex);
+        path = repoWithoutRef.slice(slashIndex + 1);
+        if (!host.includes(".") && host !== "localhost") {
+            return null;
+        }
+        repo = `https://${repoWithoutRef}`;
+    }
+    const normalizedPath = path.replace(/\.git$/, "").replace(/^\/+/, "");
+    if (!host || !normalizedPath || normalizedPath.split("/").length < 2) {
+        return null;
+    }
+    return {
+        type: "git",
+        repo,
+        host,
+        path: normalizedPath,
+        ref,
+        pinned: Boolean(ref),
+    };
+}
+/**
+ * Parse git source into a GitSource.
+ *
+ * Rules:
+ * - With git: prefix, accept all historical shorthand forms.
+ * - Without git: prefix, only accept explicit protocol URLs.
+ */
+export function parseGitUrl(source) {
+    const trimmed = source.trim();
+    const hasGitPrefix = trimmed.startsWith("git:");
+    const url = hasGitPrefix ? trimmed.slice(4).trim() : trimmed;
+    if (!hasGitPrefix && !/^(https?|ssh|git):\/\//i.test(url)) {
+        return null;
+    }
+    const split = splitRef(url);
+    const hostedCandidates = [split.ref ? `${split.repo}#${split.ref}` : undefined, url].filter((value) => Boolean(value));
+    for (const candidate of hostedCandidates) {
+        const info = hostedGitInfo.fromUrl(candidate);
+        if (info) {
+            if (split.ref && info.project?.includes("@")) {
+                continue;
+            }
+            const useHttpsPrefix = !split.repo.startsWith("http://") &&
+                !split.repo.startsWith("https://") &&
+                !split.repo.startsWith("ssh://") &&
+                !split.repo.startsWith("git://") &&
+                !split.repo.startsWith("git@");
+            return {
+                type: "git",
+                repo: useHttpsPrefix ? `https://${split.repo}` : split.repo,
+                host: info.domain || "",
+                path: `${info.user}/${info.project}`.replace(/\.git$/, ""),
+                ref: info.committish || split.ref || undefined,
+                pinned: Boolean(info.committish || split.ref),
+            };
+        }
+    }
+    const httpsCandidates = [split.ref ? `https://${split.repo}#${split.ref}` : undefined, `https://${url}`].filter((value) => Boolean(value));
+    for (const candidate of httpsCandidates) {
+        const info = hostedGitInfo.fromUrl(candidate);
+        if (info) {
+            if (split.ref && info.project?.includes("@")) {
+                continue;
+            }
+            return {
+                type: "git",
+                repo: `https://${split.repo}`,
+                host: info.domain || "",
+                path: `${info.user}/${info.project}`.replace(/\.git$/, ""),
+                ref: info.committish || split.ref || undefined,
+                pinned: Boolean(info.committish || split.ref),
+            };
+        }
+    }
+    return parseGenericGitUrl(url);
+}

package/dist/utils/image-convert.d.ts

@@ -0,0 +1,8 @@
+/**
+ * Convert image to PNG format for terminal display.
+ * Kitty graphics protocol requires PNG format (f=100).
+ */
+export declare function convertToPng(base64Data: string, mimeType: string): Promise<{
+    data: string;
+    mimeType: string;
+} | null>;

package/dist/utils/image-convert.js

@@ -0,0 +1,38 @@
+import { applyExifOrientation } from "./exif-orientation.js";
+import { loadPhoton } from "./photon.js";
+/**
+ * Convert image to PNG format for terminal display.
+ * Kitty graphics protocol requires PNG format (f=100).
+ */
+export async function convertToPng(base64Data, mimeType) {
+    // Already PNG, no conversion needed
+    if (mimeType === "image/png") {
+        return { data: base64Data, mimeType };
+    }
+    const photon = await loadPhoton();
+    if (!photon) {
+        // Photon not available, can't convert
+        return null;
+    }
+    try {
+        const bytes = new Uint8Array(Buffer.from(base64Data, "base64"));
+        const rawImage = photon.PhotonImage.new_from_byteslice(bytes);
+        const image = applyExifOrientation(photon, rawImage, bytes);
+        if (image !== rawImage)
+            rawImage.free();
+        try {
+            const pngBuffer = image.get_bytes();
+            return {
+                data: Buffer.from(pngBuffer).toString("base64"),
+                mimeType: "image/png",
+            };
+        }
+        finally {
+            image.free();
+        }
+    }
+    catch {
+        // Conversion failed
+        return null;
+    }
+}

package/dist/utils/image-resize.d.ts

@@ -0,0 +1,35 @@
+import type { ImageContent } from "@earendil-works/pi-ai/compat";
+export interface ImageResizeOptions {
+    maxWidth?: number;
+    maxHeight?: number;
+    maxBytes?: number;
+    jpegQuality?: number;
+}
+export interface ResizedImage {
+    data: string;
+    mimeType: string;
+    originalWidth: number;
+    originalHeight: number;
+    width: number;
+    height: number;
+    wasResized: boolean;
+}
+/**
+ * Resize an image to fit within the specified max dimensions and encoded file size.
+ * Returns null if the image cannot be resized below maxBytes.
+ *
+ * Uses Photon (Rust/WASM) for image processing. If Photon is not available,
+ * returns null.
+ *
+ * Strategy for staying under maxBytes:
+ * 1. First resize to maxWidth/maxHeight
+ * 2. Try both PNG and JPEG formats, pick the smaller one
+ * 3. If still too large, try JPEG with decreasing quality
+ * 4. If still too large, progressively reduce dimensions until 1x1
+ */
+export declare function resizeImage(img: ImageContent, options?: ImageResizeOptions): Promise<ResizedImage | null>;
+/**
+ * Format a dimension note for resized images.
+ * This helps the model understand the coordinate mapping.
+ */
+export declare function formatDimensionNote(result: ResizedImage): string | undefined;

package/dist/utils/image-resize.js

@@ -0,0 +1,136 @@
+import { applyExifOrientation } from "./exif-orientation.js";
+import { loadPhoton } from "./photon.js";
+// 4.5MB of base64 payload. Provides headroom below Anthropic's 5MB limit.
+const DEFAULT_MAX_BYTES = 4.5 * 1024 * 1024;
+const DEFAULT_OPTIONS = {
+    maxWidth: 2000,
+    maxHeight: 2000,
+    maxBytes: DEFAULT_MAX_BYTES,
+    jpegQuality: 80,
+};
+function encodeCandidate(buffer, mimeType) {
+    const data = Buffer.from(buffer).toString("base64");
+    return {
+        data,
+        encodedSize: Buffer.byteLength(data, "utf-8"),
+        mimeType,
+    };
+}
+/**
+ * Resize an image to fit within the specified max dimensions and encoded file size.
+ * Returns null if the image cannot be resized below maxBytes.
+ *
+ * Uses Photon (Rust/WASM) for image processing. If Photon is not available,
+ * returns null.
+ *
+ * Strategy for staying under maxBytes:
+ * 1. First resize to maxWidth/maxHeight
+ * 2. Try both PNG and JPEG formats, pick the smaller one
+ * 3. If still too large, try JPEG with decreasing quality
+ * 4. If still too large, progressively reduce dimensions until 1x1
+ */
+export async function resizeImage(img, options) {
+    const opts = { ...DEFAULT_OPTIONS, ...options };
+    const inputBuffer = Buffer.from(img.data, "base64");
+    const inputBase64Size = Buffer.byteLength(img.data, "utf-8");
+    const photon = await loadPhoton();
+    if (!photon) {
+        return null;
+    }
+    let image;
+    try {
+        const inputBytes = new Uint8Array(inputBuffer);
+        const rawImage = photon.PhotonImage.new_from_byteslice(inputBytes);
+        image = applyExifOrientation(photon, rawImage, inputBytes);
+        if (image !== rawImage)
+            rawImage.free();
+        const originalWidth = image.get_width();
+        const originalHeight = image.get_height();
+        const format = img.mimeType?.split("/")[1] ?? "png";
+        // Check if already within all limits (dimensions AND encoded size)
+        if (originalWidth <= opts.maxWidth && originalHeight <= opts.maxHeight && inputBase64Size < opts.maxBytes) {
+            return {
+                data: img.data,
+                mimeType: img.mimeType ?? `image/${format}`,
+                originalWidth,
+                originalHeight,
+                width: originalWidth,
+                height: originalHeight,
+                wasResized: false,
+            };
+        }
+        // Calculate initial dimensions respecting max limits
+        let targetWidth = originalWidth;
+        let targetHeight = originalHeight;
+        if (targetWidth > opts.maxWidth) {
+            targetHeight = Math.round((targetHeight * opts.maxWidth) / targetWidth);
+            targetWidth = opts.maxWidth;
+        }
+        if (targetHeight > opts.maxHeight) {
+            targetWidth = Math.round((targetWidth * opts.maxHeight) / targetHeight);
+            targetHeight = opts.maxHeight;
+        }
+        function tryEncodings(width, height, jpegQualities) {
+            const resized = photon.resize(image, width, height, photon.SamplingFilter.Lanczos3);
+            try {
+                const candidates = [encodeCandidate(resized.get_bytes(), "image/png")];
+                for (const quality of jpegQualities) {
+                    candidates.push(encodeCandidate(resized.get_bytes_jpeg(quality), "image/jpeg"));
+                }
+                return candidates;
+            }
+            finally {
+                resized.free();
+            }
+        }
+        const qualitySteps = Array.from(new Set([opts.jpegQuality, 85, 70, 55, 40]));
+        let currentWidth = targetWidth;
+        let currentHeight = targetHeight;
+        while (true) {
+            const candidates = tryEncodings(currentWidth, currentHeight, qualitySteps);
+            for (const candidate of candidates) {
+                if (candidate.encodedSize < opts.maxBytes) {
+                    return {
+                        data: candidate.data,
+                        mimeType: candidate.mimeType,
+                        originalWidth,
+                        originalHeight,
+                        width: currentWidth,
+                        height: currentHeight,
+                        wasResized: true,
+                    };
+                }
+            }
+            if (currentWidth === 1 && currentHeight === 1) {
+                break;
+            }
+            const nextWidth = currentWidth === 1 ? 1 : Math.max(1, Math.floor(currentWidth * 0.75));
+            const nextHeight = currentHeight === 1 ? 1 : Math.max(1, Math.floor(currentHeight * 0.75));
+            if (nextWidth === currentWidth && nextHeight === currentHeight) {
+                break;
+            }
+            currentWidth = nextWidth;
+            currentHeight = nextHeight;
+        }
+        return null;
+    }
+    catch {
+        return null;
+    }
+    finally {
+        if (image) {
+            image.free();
+        }
+    }
+}
+/**
+ * Format a dimension note for resized images.
+ * This helps the model understand the coordinate mapping.
+ */
+export function formatDimensionNote(result) {
+    if (!result.wasResized) {
+        return undefined;
+    }
+    const scale = result.originalWidth / result.width;
+    return `[Image: original ${result.originalWidth}x${result.originalHeight}, displayed at ${result.width}x${result.height}. Multiply coordinates by ${scale.toFixed(2)} to map to original image.]`;
+}

package/dist/utils/mime.d.ts

@@ -0,0 +1 @@
+export declare function detectSupportedImageMimeTypeFromFile(filePath: string): Promise<string | null>;

package/dist/utils/mime.js

@@ -0,0 +1,25 @@
+import { open } from "node:fs/promises";
+import { fileTypeFromBuffer } from "file-type";
+const IMAGE_MIME_TYPES = new Set(["image/jpeg", "image/png", "image/gif", "image/webp"]);
+const FILE_TYPE_SNIFF_BYTES = 4100;
+export async function detectSupportedImageMimeTypeFromFile(filePath) {
+    const fileHandle = await open(filePath, "r");
+    try {
+        const buffer = Buffer.alloc(FILE_TYPE_SNIFF_BYTES);
+        const { bytesRead } = await fileHandle.read(buffer, 0, FILE_TYPE_SNIFF_BYTES, 0);
+        if (bytesRead === 0) {
+            return null;
+        }
+        const fileType = await fileTypeFromBuffer(buffer.subarray(0, bytesRead));
+        if (!fileType) {
+            return null;
+        }
+        if (!IMAGE_MIME_TYPES.has(fileType.mime)) {
+            return null;
+        }
+        return fileType.mime;
+    }
+    finally {
+        await fileHandle.close();
+    }
+}

package/dist/utils/path-shims.d.ts

@@ -0,0 +1 @@
+export declare function injectPizzaPathShims(env: NodeJS.ProcessEnv): NodeJS.ProcessEnv;