use-vibes 0.5.7 → 0.11.0-dev-preview1

package/{dist/hooks → hooks}/image-gen/image-generator.js

@@ -1,33 +1,19 @@
 import { imageGen as originalImageGen, } from 'call-ai';
-import { MODULE_STATE, getRelevantOptions } from './utils';
-/**
- * Wrapper for imageGen that prevents duplicate calls
- * This function maintains a module-level cache to prevent duplicate API calls
- */
+import { MODULE_STATE, getRelevantOptions } from './utils.js';
 export async function imageGen(prompt, options) {
-    // Get the relevant options to form a stable key
     const relevantOptions = getRelevantOptions(options);
-    // Track regeneration requests when an ID is provided
-    // This was previously used for logging, which has been removed
-    // Create a stable key for the request cache
-    // Include regeneration ID if present to ensure unique keys for regeneration requests
     const stableKey = options?._regenerationId
         ? `${prompt}-${JSON.stringify(relevantOptions)}-regen-${options._regenerationId}`
         : `${prompt}-${JSON.stringify(relevantOptions)}`;
-    // Create a unique ID for this specific request instance (for logging)
     const requestId = ++MODULE_STATE.requestCounter;
-    // Check if this prompt+options combination is already being processed
     if (MODULE_STATE.pendingPrompts.has(stableKey)) {
-        // Return the existing promise for this prompt+options combination
         if (MODULE_STATE.pendingImageGenCalls.has(stableKey)) {
             return MODULE_STATE.pendingImageGenCalls.get(stableKey);
         }
     }
-    // Mark this prompt+options as being processed
     MODULE_STATE.pendingPrompts.add(stableKey);
     MODULE_STATE.processingRequests.add(stableKey);
     MODULE_STATE.requestTimestamps.set(stableKey, Date.now());
-    // Log complete request details if debug is true
     if (options?.debug) {
         console.log(`[ImgGen Debug] Generating image with prompt: "${prompt}"`, {
             requestId,
@@ -38,23 +24,18 @@ export async function imageGen(prompt, options) {
     }
     let promise;
     try {
-        // Check if we need to extract images from document
         const enhancedOptions = { ...options };
-        // Extract input files from document._files if present
         if (!enhancedOptions.images && enhancedOptions.document && enhancedOptions.document._files) {
             const imageFiles = [];
-            // Look for files with keys starting with 'in' (input files) or the 'original' file
             for (const key of Object.keys(enhancedOptions.document._files)) {
                 if (key.startsWith('in') || key === 'original') {
                     const file = enhancedOptions.document._files[key];
                     if (file) {
-                        // Handle both direct File objects and Fireproof's DocFileMeta
                         if (file instanceof File) {
                             imageFiles.push(file);
                         }
                         else if (typeof file.file === 'function') {
                             try {
-                                // For DocFileMeta objects, we need to call the file() method
                                 const fileObj = await file.file();
                                 imageFiles.push(fileObj);
                             }
@@ -65,7 +46,6 @@ export async function imageGen(prompt, options) {
                     }
                 }
             }
-            // If we found image files, add them to the images option
             if (imageFiles.length > 0) {
                 enhancedOptions.images = imageFiles;
                 if (enhancedOptions.debug) {
@@ -76,59 +56,42 @@ export async function imageGen(prompt, options) {
                 }
             }
         }
-        // Direct import from call-ai - this works consistently with test mocks
         promise = originalImageGen(prompt, enhancedOptions);
     }
     catch (e) {
         console.error(`[ImgGen Debug] Error with imageGen for request #${requestId}:`, e);
         promise = Promise.reject(e);
     }
-    // Store the promise so other requests for the same prompt+options can use it
     MODULE_STATE.pendingImageGenCalls.set(stableKey, promise);
-    // Clean up after the promise resolves or rejects
     promise
         .then((response) => {
-        // Remove from processing set but KEEP in pendingPrompts to ensure deduplication persists
-        // until page reload
         MODULE_STATE.processingRequests.delete(stableKey);
         return response;
     })
         .catch((error) => {
         console.error(`[ImgGen Debug] Request #${requestId} failed [key:${stableKey.slice(0, 12)}...]: ${error}`);
-        // Even on failure, we'll keep the key in pendingPrompts to prevent repeated failures
-        // but remove it from processing to allow potential retries after page reload
         MODULE_STATE.processingRequests.delete(stableKey);
         return Promise.reject(error);
     });
     return promise;
 }
-/**
- * Create a wrapper function for generating images with logging and tracking
- */
 export function createImageGenerator(requestHash) {
     return async (promptText, genOptions) => {
-        // Options key no longer used for logging
-        JSON.stringify(getRelevantOptions(genOptions)); // Still generate to maintain behavior
-        // Check if the document has input files that should be included
+        JSON.stringify(getRelevantOptions(genOptions));
         const debug = genOptions?.debug;
-        // Look for input files in the document
         if (genOptions && 'document' in genOptions && genOptions.document) {
             const document = genOptions.document;
             const imageFiles = [];
-            // Extract input files from document._files
             if (document._files) {
-                // Find files with keys starting with 'in' (input files)
                 for (const key of Object.keys(document._files)) {
                     if (key.startsWith('in')) {
                         const file = document._files[key];
                         if (file) {
-                            // Handle both direct File objects and Fireproof's DocFileMeta
                             if (file instanceof File) {
                                 imageFiles.push(file);
                             }
                             else if (typeof file.file === 'function') {
                                 try {
-                                    // For DocFileMeta objects, we need to call the file() method
                                     const fileObj = await file.file();
                                     imageFiles.push(fileObj);
                                 }
@@ -140,7 +103,6 @@ export function createImageGenerator(requestHash) {
                     }
                 }
             }
-            // If we found image files, add them to options
             if (imageFiles.length > 0) {
                 if (debug) {
                     console.log('[imageGenerator] Found images to include:', {
@@ -148,7 +110,6 @@ export function createImageGenerator(requestHash) {
                         fileTypes: imageFiles.map((f) => f.type),
                     });
                 }
-                // Update options with images array
                 genOptions = {
                     ...genOptions,
                     images: imageFiles,
@@ -156,7 +117,6 @@ export function createImageGenerator(requestHash) {
             }
         }
         try {
-            // Log a debug message if debug is enabled
             if (debug) {
                 console.log(`[imageGenerator] Generating image with prompt: "${promptText.slice(0, 30)}...", options:`, {
                     ...genOptions,

package/hooks/image-gen/image-generator.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"image-generator.js","sourceRoot":"","sources":["../../../jsr/hooks/image-gen/image-generator.ts"],"names":[],"mappings":"AAAA,OAAO,EAGL,QAAQ,IAAI,gBAAgB,GAC7B,MAAM,SAAS,CAAC;AACjB,OAAO,EAAE,YAAY,EAAE,kBAAkB,EAAE,MAAM,YAAY,CAAC;AAiB9D,MAAM,CAAC,KAAK,UAAU,QAAQ,CAC5B,MAAc,EACd,OAAkC,EACV;IAExB,MAAM,eAAe,GAAG,kBAAkB,CAAC,OAAO,CAAC,CAAC;IAOpD,MAAM,SAAS,GAAG,OAAO,EAAE,eAAe;QACxC,CAAC,CAAC,GAAG,MAAM,IAAI,IAAI,CAAC,SAAS,CAAC,eAAe,CAAC,UAAU,OAAO,CAAC,eAAe,EAAE;QACjF,CAAC,CAAC,GAAG,MAAM,IAAI,IAAI,CAAC,SAAS,CAAC,eAAe,CAAC,EAAE,CAAC;IAGnD,MAAM,SAAS,GAAG,EAAE,YAAY,CAAC,cAAc,CAAC;IAGhD,IAAI,YAAY,CAAC,cAAc,CAAC,GAAG,CAAC,SAAS,CAAC,EAAE,CAAC;QAE/C,IAAI,YAAY,CAAC,oBAAoB,CAAC,GAAG,CAAC,SAAS,CAAC,EAAE,CAAC;YACrD,OAAO,YAAY,CAAC,oBAAoB,CAAC,GAAG,CAAC,SAAS,CAA2B,CAAC;QACpF,CAAC;IACH,CAAC;IAGD,YAAY,CAAC,cAAc,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;IAC3C,YAAY,CAAC,kBAAkB,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;IAC/C,YAAY,CAAC,iBAAiB,CAAC,GAAG,CAAC,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE,CAAC,CAAC;IAG1D,IAAI,OAAO,EAAE,KAAK,EAAE,CAAC;QACnB,OAAO,CAAC,GAAG,CAAC,iDAAiD,MAAM,GAAG,EAAE;YACtE,SAAS;YACT,OAAO;YACP,SAAS,EAAE,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,OAAO,CAAC,MAAM,CAAC,MAAM,GAAG,QAAQ,CAAC,CAAC,CAAC,IAAI;YACnE,SAAS;SACV,CAAC,CAAC;IACL,CAAC;IAED,IAAI,OAA+B,CAAC;IAEpC,IAAI,CAAC;QAEH,MAAM,eAAe,GAAG,EAAE,GAAG,OAAO,EAAE,CAAC;QAGvC,IAAI,CAAC,eAAe,CAAC,MAAM,IAAI,eAAe,CAAC,QAAQ,IAAI,eAAe,CAAC,QAAQ,CAAC,MAAM,EAAE,CAAC;YAC3F,MAAM,UAAU,GAAW,EAAE,CAAC;YAG9B,KAAK,MAAM,GAAG,IAAI,MAAM,CAAC,IAAI,CAAC,eAAe,CAAC,QAAQ,CAAC,MAAM,CAAC,EAAE,CAAC;gBAC/D,IAAI,GAAG,CAAC,UAAU,CAAC,IAAI,CAAC,IAAI,GAAG,KAAK,UAAU,EAAE,CAAC;oBAC/C,MAAM,IAAI,GAAG,eAAe,CAAC,QAAQ,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;oBAClD,IAAI,IAAI,EAAE,CAAC;wBAET,IAAI,IAAI,YAAY,IAAI,EAAE,CAAC;4BACzB,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;wBACxB,CAAC;6BAAM,IAAI,OAAO,IAAI,CAAC,IAAI,KAAK,UAAU,EAAE,CAAC;4BAC3C,IAAI,CAAC;gCAEH,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,IAAI,EAAE,CAAC;gCAClC,UAAU,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;4BAC3B,CAAC;4BAAC,OAAO,CAAC,EAAE,CAAC;gCACX,OAAO,CAAC,KAAK,CAAC,+CAA+C,GAAG,GAAG,EAAE,CAAC,CAAC,CAAC;4BAC1E,CAAC;wBACH,CAAC;oBACH,CAAC;gBACH,CAAC;YACH,CAAC;YAGD,IAAI,UAAU,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBAC1B,eAAe,CAAC,MAAM,GAAG,UAAU,CAAC;gBAEpC,IAAI,eAAe,CAAC,KAAK,EAAE,CAAC;oBAC1B,OAAO,CAAC,GAAG,CAAC,gDAAgD,EAAE;wBAC5D,KAAK,EAAE,UAAU,CAAC,MAAM;wBACxB,IAAI,EAAE,MAAM,CAAC,IAAI,CAAC,eAAe,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,MAAM,CACvD,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,UAAU,CAC9C;qBACF,CAAC,CAAC;gBACL,CAAC;YACH,CAAC;QACH,CAAC;QAGD,OAAO,GAAG,gBAAgB,CAAC,MAAM,EAAE,eAAe,CAAC,CAAC;IACtD,CAAC;IAAC,OAAO,CAAC,EAAE,CAAC;QACX,OAAO,CAAC,KAAK,CAAC,mDAAmD,SAAS,GAAG,EAAE,CAAC,CAAC,CAAC;QAClF,OAAO,GAAG,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC;IAC9B,CAAC;IAGD,YAAY,CAAC,oBAAoB,CAAC,GAAG,CAAC,SAAS,EAAE,OAAO,CAAC,CAAC;IAG1D,OAAO;SACJ,IAAI,CAAC,CAAC,QAAQ,EAAE,EAAE,CAAC;QAGlB,YAAY,CAAC,kBAAkB,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;QAClD,OAAO,QAAQ,CAAC;IAAA,CACjB,CAAC;SACD,KAAK,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC;QAChB,OAAO,CAAC,KAAK,CACX,2BAA2B,SAAS,gBAAgB,SAAS,CAAC,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,SAAS,KAAK,EAAE,CAC3F,CAAC;QAGF,YAAY,CAAC,kBAAkB,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;QAClD,OAAO,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;IAAA,CAC9B,CAAC,CAAC;IAEL,OAAO,OAAO,CAAC;AAAA,CAChB;AAKD,MAAM,UAAU,oBAAoB,CAAC,WAAmB,EAAE;IACxD,OAAO,KAAK,EACV,UAAkB,EAClB,UAAqC,EACb,EAAE,CAAC;QAE3B,IAAI,CAAC,SAAS,CAAC,kBAAkB,CAAC,UAAU,CAAC,CAAC,CAAC;QAG/C,MAAM,KAAK,GAAG,UAAU,EAAE,KAAK,CAAC;QAGhC,IAAI,UAAU,IAAI,UAAU,IAAI,UAAU,IAAI,UAAU,CAAC,QAAQ,EAAE,CAAC;YAClE,MAAM,QAAQ,GAAG,UAAU,CAAC,QAAQ,CAAC;YACrC,MAAM,UAAU,GAAW,EAAE,CAAC;YAG9B,IAAI,QAAQ,CAAC,MAAM,EAAE,CAAC;gBAEpB,KAAK,MAAM,GAAG,IAAI,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,EAAE,CAAC;oBAC/C,IAAI,GAAG,CAAC,UAAU,CAAC,IAAI,CAAC,EAAE,CAAC;wBACzB,MAAM,IAAI,GAAG,QAAQ,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;wBAClC,IAAI,IAAI,EAAE,CAAC;4BAET,IAAI,IAAI,YAAY,IAAI,EAAE,CAAC;gCACzB,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;4BACxB,CAAC;iCAAM,IAAI,OAAO,IAAI,CAAC,IAAI,KAAK,UAAU,EAAE,CAAC;gCAC3C,IAAI,CAAC;oCAEH,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,IAAI,EAAE,CAAC;oCAClC,UAAU,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;gCAC3B,CAAC;gCAAC,OAAO,CAAC,EAAE,CAAC;oCACX,OAAO,CAAC,KAAK,CAAC,+CAA+C,GAAG,GAAG,EAAE,CAAC,CAAC,CAAC;gCAC1E,CAAC;4BACH,CAAC;wBACH,CAAC;oBACH,CAAC;gBACH,CAAC;YACH,CAAC;YAGD,IAAI,UAAU,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBAC1B,IAAI,KAAK,EAAE,CAAC;oBACV,OAAO,CAAC,GAAG,CAAC,2CAA2C,EAAE;wBACvD,KAAK,EAAE,UAAU,CAAC,MAAM;wBACxB,SAAS,EAAE,UAAU,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC;qBACzC,CAAC,CAAC;gBACL,CAAC;gBAGD,UAAU,GAAG;oBACX,GAAG,UAAU;oBACb,MAAM,EAAE,UAAU;iBACnB,CAAC;YACJ,CAAC;QACH,CAAC;QAED,IAAI,CAAC;YAEH,IAAI,KAAK,EAAE,CAAC;gBACV,OAAO,CAAC,GAAG,CACT,mDAAmD,UAAU,CAAC,KAAK,CACjE,CAAC,EACD,EAAE,CACH,gBAAgB,EACjB;oBACE,GAAG,UAAU;oBACb,SAAS,EAAE,UAAU,EAAE,MAAM,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI;oBAC5C,UAAU,EAAE,UAAU,EAAE,MAAM,CAAC,CAAC,CAAC,UAAU,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC;iBAC9D,CACF,CAAC;YACJ,CAAC;YAED,MAAM,QAAQ,GAAG,MAAM,QAAQ,CAAC,UAAU,EAAE,UAAU,CAAC,CAAC;YACxD,OAAO,QAAQ,CAAC;QAClB,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,KAAK,CAAC,qCAAqC,WAAW,MAAM,KAAK,EAAE,CAAC,CAAC;YAC7E,MAAM,KAAK,CAAC;QACd,CAAC;IAAA,CACF,CAAC;AAAA,CACH"}

package/hooks/image-gen/index.d.ts

@@ -0,0 +1,7 @@
+import { useImageGen } from './use-image-gen.js';
+import { imageGen } from './image-generator.js';
+import { base64ToFile, hashInput, MODULE_STATE } from './utils.js';
+import type { ImageDocument, UseImageGenOptions, UseImageGenResult } from './types.js';
+export { MODULE_STATE };
+export { useImageGen, imageGen, base64ToFile, hashInput };
+export type { ImageDocument, UseImageGenOptions, UseImageGenResult };

package/hooks/image-gen/index.js

@@ -0,0 +1,6 @@
+import { useImageGen } from './use-image-gen.js';
+import { imageGen } from './image-generator.js';
+import { base64ToFile, hashInput, MODULE_STATE } from './utils.js';
+export { MODULE_STATE };
+export { useImageGen, imageGen, base64ToFile, hashInput };
+//# sourceMappingURL=index.js.map

package/hooks/image-gen/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../jsr/hooks/image-gen/index.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,WAAW,EAAE,MAAM,oBAAoB,CAAC;AACjD,OAAO,EAAE,QAAQ,EAAE,MAAM,sBAAsB,CAAC;AAChD,OAAO,EAAE,YAAY,EAAE,SAAS,EAAE,YAAY,EAAE,MAAM,YAAY,CAAC;AAGnE,OAAO,EAAE,YAAY,EAAE,CAAC;AAExB,OAAO,EAAE,WAAW,EAAE,QAAQ,EAAE,YAAY,EAAE,SAAS,EAAE,CAAC"}

package/hooks/image-gen/types.d.ts

@@ -0,0 +1,66 @@
+import type { Database, DocWithId } from 'use-fireproof';
+import { ImageGenOptions, ImageResponse } from 'call-ai';
+export interface PromptEntry {
+    readonly text: string;
+    readonly created: number;
+}
+export interface ImageDocumentPlain {
+    readonly _rev?: string;
+    readonly type: 'image';
+    readonly prompt?: string;
+    readonly prompts?: Record<string, PromptEntry>;
+    readonly created: number;
+    readonly currentVersion: number;
+    readonly versions: VersionInfo[];
+    readonly currentPromptKey: string;
+}
+export type ImageDocument = DocWithId<ImageDocumentPlain>;
+export type PartialImageDocument = DocWithId<Partial<ImageDocumentPlain>>;
+export interface VersionInfo {
+    readonly id: string;
+    readonly created: number;
+    readonly promptKey?: string;
+}
+export type GenerationPhase = 'idle' | 'generating' | 'complete' | 'error';
+export interface UseImageGenOptions {
+    readonly prompt: string;
+    readonly _id: string;
+    readonly _rev?: string;
+    readonly database: string | Database;
+    readonly options: Partial<ImageGenOptions>;
+    readonly generationId: string;
+    readonly skip: boolean;
+    readonly type?: string;
+    readonly currentVersion?: number;
+    readonly versions?: {
+        readonly id: string;
+        readonly created: number;
+        readonly promptKey: string;
+    }[];
+    readonly _files?: Record<string, File>;
+    readonly prompts?: Record<string, PromptEntry>;
+    readonly editedPrompt: string;
+}
+export interface UseImageGenResult {
+    readonly imageData?: string | null;
+    readonly loading: boolean;
+    readonly progress: number;
+    readonly error?: Error | null;
+    readonly size?: {
+        readonly width: number;
+        readonly height: number;
+    };
+    readonly document?: PartialImageDocument | null;
+}
+export interface ModuleState {
+    readonly pendingImageGenCalls: Map<string, Promise<ImageResponse>>;
+    readonly pendingPrompts: Set<string>;
+    readonly processingRequests: Set<string>;
+    readonly requestTimestamps: Map<string, number>;
+    requestCounter: number;
+    readonly createdDocuments: Map<string, string>;
+    readonly pendingDocumentCreations: Map<string, Promise<{
+        id: string;
+        doc: ImageDocument;
+    }>>;
+}

package/{dist/hooks → hooks}/image-gen/types.js.map

@@ -1 +1 @@
-{"version":3,"file":"types.js","sourceRoot":"","sources":["../../../src/hooks/image-gen/types.ts"],"names":[],"mappings":""}
+{"version":3,"file":"types.js","sourceRoot":"","sources":["../../../jsr/hooks/image-gen/types.ts"],"names":[],"mappings":""}

package/hooks/image-gen/use-image-gen.d.ts

@@ -0,0 +1,4 @@
+import { UseImageGenOptions, UseImageGenResult } from './types.js';
+export declare function useImageGen({ prompt, _id, _rev, options, database, skip, // Skip processing flag
+generationId, // Unique ID that changes for each new generation request
+editedPrompt }: Partial<UseImageGenOptions>): UseImageGenResult;