feat: support custom s3 service

app/utils/s3_file_storage.ts

@@ -0,0 +1,72 @@
+import {
+  S3Client,
+  ListBucketsCommand,
+  ListObjectsV2Command,
+  GetObjectCommand,
+  PutObjectCommand,
+} from "@aws-sdk/client-s3";
+import { getSignedUrl } from "@aws-sdk/s3-request-presigner";
+
+const R2_ACCOUNT_ID = process.env.R2_ACCOUNT_ID;
+const R2_ACCESS_KEY_ID = process.env.R2_ACCESS_KEY_ID;
+const R2_SECRET_ACCESS_KEY = process.env.R2_SECRET_ACCESS_KEY;
+const R2_BUCKET = process.env.R2_BUCKET;
+
+const S3_ENDPOINT = process.env.S3_ENDPOINT;
+const S3_ACCESS_KEY_ID = process.env.S3_ACCESS_KEY_ID;
+const S3_SECRET_ACCESS_KEY = process.env.S3_SECRET_ACCESS_KEY;
+const S3_BUCKET = process.env.S3_BUCKET;
+
+const getR2Client = () => {
+  return new S3Client({
+    region: "auto",
+    endpoint:
+      S3_ENDPOINT ?? `https://${R2_ACCOUNT_ID}.r2.cloudflarestorage.com`,
+    credentials: {
+      accessKeyId: S3_ACCESS_KEY_ID ?? R2_ACCESS_KEY_ID!,
+      secretAccessKey: S3_SECRET_ACCESS_KEY ?? R2_SECRET_ACCESS_KEY!,
+    },
+  });
+};
+
+export default class S3FileStorage {
+  static async get(fileName: string) {
+    const file = await getR2Client().send(
+      new GetObjectCommand({
+        Bucket: S3_BUCKET ?? R2_BUCKET,
+        Key: fileName,
+      }),
+    );
+
+    if (!file) {
+      throw new Error("not found.");
+    }
+
+    return file.Body;
+  }
+
+  static async put(fileName: string, data: Buffer) {
+    const signedUrl = await getSignedUrl(
+      getR2Client(),
+      new PutObjectCommand({
+        Bucket: S3_BUCKET ?? R2_BUCKET,
+        Key: fileName,
+      }),
+      { expiresIn: 60 },
+    );
+
+    console.log(signedUrl);
+
+    try {
+      await fetch(signedUrl, {
+        method: "PUT",
+        body: data,
+      });
+
+      return `/api/file/${fileName}`;
+    } catch (e) {
+      console.error("[S3]", e);
+      throw e;
+    }
+  }
+}