stream feature

image-processing-logic.js

@@ -1,46 +1,36 @@
 const sharp = require('sharp');
+const stream = require('stream');
 
 class ImageProcessor {
-    async processImage(buffer, options = {}) {
-        const { quality = 80, width, height, format = 'jpeg', maintainAspect = true } = options;
-        
+    /**
+     * Processes an image from a readable stream.
+     * @param {stream.Readable} input - The incoming stream of image data.
+     * @param {Object} options - Processing options including quality and format.
+     * @returns {stream.Readable} - A readable stream of the processed image.
+     */
+    processImage(input, options = {}) {
+        const { quality = 80, format = 'jpeg' } = options;
+
         try {
-            let sharpInstance = sharp(buffer);
-            const metadata = await sharpInstance.metadata();
-            
-            let newWidth = width ? parseInt(width) : null;
-            let newHeight = height ? parseInt(height) : null;
-            
-            if (newWidth || newHeight) {
-                if (maintainAspect) {
-                    if (newWidth && !newHeight) {
-                        newHeight = Math.round(newWidth / (metadata.width / metadata.height));
-                    } else if (!newWidth && newHeight) {
-                        newWidth = Math.round(newHeight * (metadata.width / metadata.height));
-                    }
-                }
-                
-                sharpInstance = sharpInstance.resize(newWidth, newHeight, {
-                    fit: 'inside',
-                    withoutEnlargement: true
-                });
-            }
-            
+            // Create a Sharp instance for streaming.
+            let sharpInstance = sharp();
+
+            // Apply format and quality options.
             switch (format.toLowerCase()) {
                 case 'jpeg':
                     sharpInstance = sharpInstance.jpeg({ quality: parseInt(quality), mozjpeg: true });
                     break;
                 case 'png':
-                    sharpInstance = sharpInstance.png({ quality: parseInt(quality) });
+                    sharpInstance = sharpInstance.png({ quality: parseInt(quality), compressionLevel: 1 });
                     break;
                 case 'webp':
-                    sharpInstance = sharpInstance.webp({ quality: parseInt(quality) });
+                    sharpInstance = sharpInstance.webp({ quality: parseInt(quality), speed: 8 });
                     break;
                 case 'avif':
-                    sharpInstance = sharpInstance.avif({ quality: parseInt(quality) });
+                    sharpInstance = sharpInstance.avif({ quality: parseInt(quality), speed: 8 });
                     break;
                 case 'heif':
-                    sharpInstance = sharpInstance.heif({ quality: parseInt(quality) });
+                    sharpInstance = sharpInstance.heif({ quality: parseInt(quality), speed: 8 });
                     break;
                 case 'tiff':
                     sharpInstance = sharpInstance.tiff({ quality: parseInt(quality) });
@@ -48,18 +38,12 @@ class ImageProcessor {
                 default:
                     throw new Error('Unsupported format');
             }
-            
-            const processedBuffer = await sharpInstance.toBuffer();
-            
-            return {
-                buffer: processedBuffer,
-                size: processedBuffer.length,
-                width: newWidth || metadata.width,
-                height: newHeight || metadata.height,
-                format: format
-            };
+
+            // Pipe the input stream to the sharp pipeline and return the output stream.
+            return input.pipe(sharpInstance);
+
         } catch (error) {
-            console.error('Error processing image:', error);
+            console.error('Error setting up image processing stream:', error);
             throw error;
         }
     }