@bike4mind/cli 0.2.6-fix-cli-mcp-type-mismatch.17119 → 0.2.7-feat-cli-rewind-command.17128

package/dist/ImageStore-RQZ7OF7P.js

@@ -0,0 +1,225 @@
+#!/usr/bin/env node
+import "./chunk-PDX44BCA.js";
+
+// src/storage/ImageStore.ts
+import { createHash } from "crypto";
+import { mkdirSync, writeFileSync, readFileSync, existsSync, unlinkSync } from "fs";
+import { join, extname, resolve } from "path";
+import { homedir } from "os";
+import Database from "better-sqlite3";
+import sharp from "sharp";
+var ImageStore = class _ImageStore {
+  static {
+    this.BASE_DIR = join(homedir(), ".bike4mind", "cli");
+  }
+  static {
+    this.IMAGES_DIR = join(_ImageStore.BASE_DIR, "images");
+  }
+  static {
+    this.DB_PATH = join(_ImageStore.BASE_DIR, "images.db");
+  }
+  static {
+    this.RETENTION_DAYS = 7;
+  }
+  static {
+    // Target max size before base64 encoding. Base64 adds ~33% overhead.
+    // To stay under 1MB after base64, we need images < 750KB
+    this.MAX_RAW_SIZE = 750 * 1024;
+  }
+  static {
+    // 750KB
+    // Maximum size before compression to prevent OOM with very large images
+    this.MAX_PRECOMPRESS_SIZE = 50 * 1024 * 1024;
+  }
+  constructor() {
+    this.ensureDirectories();
+    this.db = new Database(_ImageStore.DB_PATH);
+    this.initDatabase();
+    try {
+      this.cleanupOldImages();
+    } catch (err) {
+      console.warn("Failed to cleanup old images:", err);
+    }
+  }
+  ensureDirectories() {
+    mkdirSync(_ImageStore.BASE_DIR, { recursive: true });
+    mkdirSync(_ImageStore.IMAGES_DIR, { recursive: true });
+  }
+  initDatabase() {
+    this.db.exec(`
+			CREATE TABLE IF NOT EXISTS images (
+				hash TEXT PRIMARY KEY,
+				format TEXT NOT NULL,
+				size INTEGER NOT NULL,
+				timestamp INTEGER NOT NULL,
+				uploaded INTEGER NOT NULL DEFAULT 0,
+				s3_url TEXT,
+				original_filename TEXT
+			)
+		`);
+  }
+  /**
+   * Store an image locally with content-based hashing for deduplication
+   */
+  async store(imageData, originalFilename) {
+    const { buffer: processedData, format: processedFormat } = await this.processImage(imageData, originalFilename);
+    const hash = this.generateHash(processedData);
+    const format = processedFormat;
+    const ext = format.toLowerCase();
+    const localPath = join(_ImageStore.IMAGES_DIR, `${hash}.${ext}`);
+    const existing = this.getMetadata(hash);
+    if (existing) {
+      return {
+        hash,
+        placeholder: "",
+        // Will be set by caller
+        localPath,
+        metadata: existing
+      };
+    }
+    writeFileSync(localPath, processedData);
+    const metadata = {
+      hash,
+      format,
+      size: processedData.length,
+      timestamp: Date.now(),
+      uploaded: false,
+      originalFilename
+    };
+    this.db.prepare(
+      `INSERT INTO images (hash, format, size, timestamp, uploaded, original_filename)
+				 VALUES (?, ?, ?, ?, ?, ?)`
+    ).run(hash, format, metadata.size, metadata.timestamp, 0, originalFilename || null);
+    return {
+      hash,
+      placeholder: "",
+      localPath,
+      metadata
+    };
+  }
+  /**
+   * Get image metadata by hash
+   */
+  getMetadata(hash) {
+    const row = this.db.prepare("SELECT * FROM images WHERE hash = ?").get(hash);
+    if (!row) return null;
+    return {
+      hash: row.hash,
+      format: row.format,
+      size: row.size,
+      timestamp: row.timestamp,
+      uploaded: Boolean(row.uploaded),
+      s3Url: row.s3_url,
+      originalFilename: row.original_filename
+    };
+  }
+  /**
+   * Read image data from local cache
+   */
+  readImage(hash) {
+    const metadata = this.getMetadata(hash);
+    if (!metadata) return null;
+    const localPath = join(_ImageStore.IMAGES_DIR, `${hash}.${metadata.format.toLowerCase()}`);
+    const normalizedPath = resolve(localPath);
+    if (!normalizedPath.startsWith(resolve(_ImageStore.IMAGES_DIR))) {
+      throw new Error("Invalid image path");
+    }
+    if (!existsSync(localPath)) return null;
+    return readFileSync(localPath);
+  }
+  /**
+   * Generate content hash for deduplication
+   */
+  generateHash(data) {
+    return createHash("sha256").update(data).digest("hex").substring(0, 16);
+  }
+  /**
+   * Process image: compress if needed to stay under size limit
+   */
+  async processImage(data, originalFilename) {
+    if (data.length > _ImageStore.MAX_PRECOMPRESS_SIZE) {
+      throw new Error(
+        `Image too large to process (${Math.round(data.length / 1024 / 1024)}MB). Maximum size is ${_ImageStore.MAX_PRECOMPRESS_SIZE / 1024 / 1024}MB.`
+      );
+    }
+    const originalFormat = this.detectFormat(data, originalFilename);
+    if (data.length <= _ImageStore.MAX_RAW_SIZE) {
+      return { buffer: data, format: originalFormat };
+    }
+    let image = sharp(data);
+    const metadata = await image.metadata();
+    if (!metadata.width || !metadata.height) {
+      throw new Error("Unable to read image dimensions");
+    }
+    const maxDimension = 2048;
+    let width = metadata.width;
+    let height = metadata.height;
+    if (width > maxDimension || height > maxDimension) {
+      if (width > height) {
+        height = Math.round(height * maxDimension / width);
+        width = maxDimension;
+      } else {
+        width = Math.round(width * maxDimension / height);
+        height = maxDimension;
+      }
+    }
+    image = image.resize(width, height);
+    let quality = 85;
+    let buffer = await image.jpeg({ quality }).toBuffer();
+    while (buffer.length > _ImageStore.MAX_RAW_SIZE && quality > 30) {
+      quality -= 10;
+      buffer = await image.jpeg({ quality }).toBuffer();
+    }
+    if (buffer.length > _ImageStore.MAX_RAW_SIZE) {
+      throw new Error(`Unable to compress image below ${_ImageStore.MAX_RAW_SIZE / 1024}KB limit`);
+    }
+    return { buffer, format: "jpg" };
+  }
+  /**
+   * Detect image format from buffer or filename
+   */
+  detectFormat(data, filename) {
+    if (data[0] === 137 && data[1] === 80 && data[2] === 78 && data[3] === 71) {
+      return "png";
+    }
+    if (data[0] === 255 && data[1] === 216 && data[2] === 255) {
+      return "jpg";
+    }
+    if (data[0] === 71 && data[1] === 73 && data[2] === 70) {
+      return "gif";
+    }
+    if (data[0] === 82 && data[1] === 73 && data[2] === 70 && data[3] === 70) {
+      return "webp";
+    }
+    if (filename) {
+      const ext = extname(filename).substring(1).toLowerCase();
+      if (["png", "jpg", "jpeg", "gif", "webp"].includes(ext)) {
+        return ext === "jpeg" ? "jpg" : ext;
+      }
+    }
+    return "png";
+  }
+  /**
+   * Clean up images older than retention period
+   */
+  cleanupOldImages() {
+    const cutoffTime = Date.now() - _ImageStore.RETENTION_DAYS * 24 * 60 * 60 * 1e3;
+    const oldImages = this.db.prepare("SELECT hash, format FROM images WHERE timestamp < ?").all(cutoffTime);
+    for (const { hash, format } of oldImages) {
+      const localPath = join(_ImageStore.IMAGES_DIR, `${hash}.${format.toLowerCase()}`);
+      if (existsSync(localPath)) {
+        unlinkSync(localPath);
+      }
+    }
+    this.db.prepare("DELETE FROM images WHERE timestamp < ?").run(cutoffTime);
+  }
+  /**
+   * Close database connection
+   */
+  close() {
+    this.db.close();
+  }
+};
+export {
+  ImageStore
+};