add image generation

Author: willwade

src/processors/gridsetProcessor.ts

@@ -572,6 +572,32 @@ class GridsetProcessor extends BaseProcessor {
       );
     }
 
+    // Pre-load all image data for conversion to other formats (e.g., Snap)
+    const imageDataCache = new Map<string, Buffer>();
+    const imageEntries = entries.filter((e) => {
+      const name = e.entryName.toLowerCase();
+      return (
+        name.endsWith('.png') ||
+        name.endsWith('.jpg') ||
+        name.endsWith('.jpeg') ||
+        name.endsWith('.gif') ||
+        name.endsWith('.svg')
+      );
+    });
+
+    for (const imageEntry of imageEntries) {
+      try {
+        const raw = await imageEntry.getData();
+        const data = isEncryptedArchive
+          ? decryptGridsetEntry(Buffer.from(raw), encryptedContentPassword)
+          : Buffer.from(raw);
+        const normalizedEntry = imageEntry.entryName.replace(/\\/g, '/');
+        imageDataCache.set(normalizedEntry, data);
+      } catch (err) {
+        // Silently fail - individual image loading failures shouldn't break the entire load
+      }
+    }
+
     // First pass: collect all grid names and IDs for navigation resolution
     const gridNameToIdMap = new Map<string, string>();
     const gridIdToNameMap = new Map<string, string>();
@@ -1095,6 +1121,11 @@ class GridsetProcessor extends BaseProcessor {
               );
             }
 
+            // Load binary image data from cache for conversion to other formats (e.g., Snap)
+            const imageData = resolvedImageEntry
+              ? imageDataCache.get(resolvedImageEntry)
+              : undefined;
+
             // Check if image is a symbol library reference
             let symbolLibraryRef: SymbolReference | null = null;
             if (declaredImageName && isSymbolLibraryReference(declaredImageName)) {
@@ -1673,6 +1704,8 @@ class GridsetProcessor extends BaseProcessor {
                   !isMoreButton
                     ? wordListCellIndex - 1
                     : undefined,
+                // Store binary image data for conversion to other formats
+                ...(imageData ? { imageData, image_id: resolvedImageEntry } : {}),
               },
             });
 

src/processors/snapProcessor.ts

@@ -1106,6 +1106,51 @@ class SnapProcessor extends BaseProcessor {
             useMessageRecording = 1;
           }
 
+          // Handle image data from button.parameters.imageData or button.image (data URL)
+          let pageSetImageId: number | null = null;
+          if (button.parameters?.imageData && Buffer.isBuffer(button.parameters.imageData)) {
+            // Use existing image data buffer
+            const imageIdentifier: string =
+              (button.parameters.image_id as string) || `IMG_${buttonIdCounter}`;
+            let imageId = pageSetDataIdentifierMap.get(imageIdentifier);
+            if (!imageId) {
+              imageId = pageSetDataIdCounter++;
+              insertPageSetData.run(imageId, imageIdentifier, button.parameters.imageData, 1);
+              pageSetDataIdentifierMap.set(imageIdentifier, imageId);
+            } else {
+              incrementRefCount.run(imageId);
+            }
+            pageSetImageId = imageId;
+          } else if (
+            button.image &&
+            typeof button.image === 'string' &&
+            button.image.startsWith('data:image')
+          ) {
+            // Convert data URL to buffer
+            try {
+              const matches = button.image.match(/^data:image\/(\w+);base64,(.+)$/);
+              if (matches && matches[2]) {
+                const imageData = Buffer.from(matches[2], 'base64');
+                const imageIdentifier: string =
+                  (button.parameters?.image_id as string) || `IMG_${buttonIdCounter}`;
+                let imageId = pageSetDataIdentifierMap.get(imageIdentifier);
+                if (!imageId) {
+                  imageId = pageSetDataIdCounter++;
+                  insertPageSetData.run(imageId, imageIdentifier, imageData, 1);
+                  pageSetDataIdentifierMap.set(imageIdentifier, imageId);
+                } else {
+                  incrementRefCount.run(imageId);
+                }
+                pageSetImageId = imageId;
+              }
+            } catch (err) {
+              console.warn(
+                `[SnapProcessor] Failed to convert data URL to Buffer for button ${button.id}:`,
+                err
+              );
+            }
+          }
+
           // Retry logic for SQLite operations
           let retries = 3;
           while (retries > 0) {
@@ -1116,8 +1161,8 @@ class SnapProcessor extends BaseProcessor {
                 button.message || button.label || '',
                 navigatePageId,
                 elementRefId,
-                null,
-                null,
+                null, // LibrarySymbolId - not used for embedded images
+                pageSetImageId, // PageSetImageId - references embedded image in PageSetData
                 messageRecordingId,
                 serializedMetadata,
                 useMessageRecording,