fix: handle NotFound exception correct

mxzinke · mxzinke · commit 50bf74cb14b6 · 2024-11-19T09:47:36.000+01:00
diff --git a/src/index.ts b/src/index.ts
@@ -2,6 +2,7 @@ import {
   DeleteObjectCommand,
   GetObjectCommand,
   HeadObjectCommand,
+  NotFound,
   PutObjectCommand,
   S3Client,
 } from "@aws-sdk/client-s3";
@@ -36,12 +37,21 @@ export const exists = async (
   path: string,
 ): Promise<boolean> => {
   const client = new S3Client(bucket.connection);
-  const res = await client.send(
-    new HeadObjectCommand({
-      Bucket: bucket.name,
-      Key: path,
-    }),
-  );
+  const res = await client
+    .send(
+      new HeadObjectCommand({
+        Bucket: bucket.name,
+        Key: path,
+      }),
+    )
+    .catch((err) => {
+      if (err instanceof NotFound) {
+        return {
+          ContentLength: undefined,
+        };
+      }
+      throw err;
+    });
 
   return res.ContentLength !== undefined;
 };
@@ -51,24 +61,36 @@ export const get = async (
   path: string,
 ): Promise<Buffer | null> => {
   const client = new S3Client(bucket.connection);
-  const res = await client.send(
-    new GetObjectCommand({
-      Bucket: bucket.name,
-      Key: path,
-    }),
-  );
+  const res = await client
+    .send(
+      new GetObjectCommand({
+        Bucket: bucket.name,
+        Key: path,
+      }),
+    )
+    .catch((err) => {
+      if (err instanceof NotFound) {
+        return null;
+      }
+      throw err;
+    });
 
-  return res.Body ? Buffer.from(await res.Body.transformToByteArray()) : null;
+  return res?.Body ? Buffer.from(await res.Body.transformToByteArray()) : null;
 };
 
 export const deleteFile = async (
   bucket: Bucket,
   path: string,
 ): Promise<void> => {
   const client = new S3Client(bucket.connection);
-  await client.send(
-    new DeleteObjectCommand({ Bucket: bucket.name, Key: path }),
-  );
+  await client
+    .send(new DeleteObjectCommand({ Bucket: bucket.name, Key: path }))
+    .catch((err) => {
+      if (err instanceof NotFound) {
+        return;
+      }
+      throw err;
+    });
 };
 
 export const getStreamed = async (
