From 1b8414765e76ab8cdb9163554b710e147d90191b Mon Sep 17 00:00:00 2001
From: Sam Becker <sam@sambecker.com>
Date: Tue, 13 Feb 2024 17:19:27 -0600
Subject: [PATCH] Re-introduce PNG support

---
 src/components/ImageInput.tsx | 44 +++++++++++++++++++++++------------
 src/photo/PhotoUpload.tsx     |  3 +--
 src/photo/index.ts            |  1 +
 3 files changed, 31 insertions(+), 17 deletions(-)

diff --git a/src/components/ImageInput.tsx b/src/components/ImageInput.tsx
index b184bdbf..b5b0b6df 100644
--- a/src/components/ImageInput.tsx
+++ b/src/components/ImageInput.tsx
@@ -8,13 +8,15 @@ import { clsx } from 'clsx/lite';
 import Spinner from './Spinner';
 import { ACCEPTED_PHOTO_FILE_TYPES } from '@/photo';
 import { FiUploadCloud } from 'react-icons/fi';
+import { MAX_IMAGE_SIZE } from '@/services/next-image';
 
 const INPUT_ID = 'file';
 
 export default function ImageInput({
   onStart,
   onBlobReady,
-  maxSize,
+  shouldResize,
+  maxSize = MAX_IMAGE_SIZE,
   quality = 0.8,
   loading,
   debug,
@@ -26,6 +28,7 @@ export default function ImageInput({
     hasMultipleUploads?: boolean,
     isLastBlob?: boolean,
   }) => Promise<any>
+  shouldResize?: boolean
   maxSize?: number
   quality?: number
   loading?: boolean
@@ -92,6 +95,8 @@ export default function ImageInput({
                     hasMultipleUploads: files.length > 1,
                     isLastBlob: i === files.length - 1,
                   };
+
+                  const isPng = callbackArgs.extension === 'png';
                   
                   const canvas = ref.current;
 
@@ -100,7 +105,7 @@ export default function ImageInput({
                     '2d', { colorSpace: 'display-p3' }
                   );
 
-                  if (maxSize && canvas && ctx) {
+                  if ((shouldResize || isPng) && canvas && ctx) {
                     // Process images that need resizing
                     const image = await blobToImage(file);
 
@@ -108,18 +113,24 @@ export default function ImageInput({
 
                     ctx.save();
                     
-                    let orientation = await exifr.orientation(file) ?? 1;
-                    // Reverse engineer orientation
-                    // so preserved EXIF data can be copied
-                    switch (orientation) {
-                    case 1: orientation = 1; break;
-                    case 2: orientation = 1; break;
-                    case 3: orientation = 3; break;
-                    case 4: orientation = 1; break;
-                    case 5: orientation = 1; break;
-                    case 6: orientation = 8; break;
-                    case 7: orientation = 1; break;
-                    case 8: orientation = 6; break;
+                    let orientation = await exifr
+                      .orientation(file)
+                      .catch(() => 1) ?? 1;
+
+                    // Preserve EXIF data for PNGs
+                    if (!isPng) {
+                      // Reverse engineer orientation
+                      // so preserved EXIF data can be copied
+                      switch (orientation) {
+                      case 1: orientation = 1; break;
+                      case 2: orientation = 1; break;
+                      case 3: orientation = 3; break;
+                      case 4: orientation = 1; break;
+                      case 5: orientation = 1; break;
+                      case 6: orientation = 8; break;
+                      case 7: orientation = 1; break;
+                      case 8: orientation = 6; break;
+                      }
                     }
 
                     const ratio = image.width / image.height;
@@ -183,7 +194,10 @@ export default function ImageInput({
                     canvas.toBlob(
                       async blob => {
                         if (blob) {
-                          const blobWithExif = await CopyExif(file, blob);
+                          const blobWithExif = await CopyExif(file, blob)
+                            // Fallback to original blob if EXIF data is missing
+                            // or image is in PNG format which cannot be parsed
+                            .catch(() => blob);
                           await onBlobReady?.({
                             ...callbackArgs,
                             blob: blobWithExif,
diff --git a/src/photo/PhotoUpload.tsx b/src/photo/PhotoUpload.tsx
index b50fcdff..2e366c16 100644
--- a/src/photo/PhotoUpload.tsx
+++ b/src/photo/PhotoUpload.tsx
@@ -5,7 +5,6 @@ import { uploadPhotoFromClient } from '@/services/storage';
 import { useRouter } from 'next/navigation';
 import { PATH_ADMIN_UPLOADS, pathForAdminUploadUrl } from '@/site/paths';
 import ImageInput from '../components/ImageInput';
-import { MAX_IMAGE_SIZE } from '@/services/next-image';
 import { clsx } from 'clsx/lite';
 
 export default function PhotoUpload({
@@ -32,8 +31,8 @@ export default function PhotoUpload({
       <div className="flex items-center gap-8">
         <form className="flex items-center min-w-0">
           <ImageInput
-            maxSize={shouldResize ? MAX_IMAGE_SIZE : undefined}
             loading={isUploading}
+            shouldResize={shouldResize}
             onStart={() => {
               setIsUploading(true);
               setUploadError('');
diff --git a/src/photo/index.ts b/src/photo/index.ts
index 3797d0c9..7da71787 100644
--- a/src/photo/index.ts
+++ b/src/photo/index.ts
@@ -17,6 +17,7 @@ export const GRID_THUMBNAILS_TO_SHOW_MAX = 12;
 export const ACCEPTED_PHOTO_FILE_TYPES = [
   'image/jpg',
   'image/jpeg',
+  'image/png',
 ];
 
 export const MAX_PHOTO_UPLOAD_SIZE_IN_BYTES = 50_000_000;