upload image option (#64)

apps/web/src/server/api/routers/campaign.ts

@@ -1,17 +1,24 @@
 import { Prisma } from "@prisma/client";
 import { TRPCError } from "@trpc/server";
 import { z } from "zod";
+import { env } from "~/env";
 import {
   teamProcedure,
   createTRPCRouter,
   campaignProcedure,
   publicProcedure,
 } from "~/server/api/trpc";
+import { nanoid } from "~/server/nanoid";
 import {
   sendCampaign,
   subscribeContact,
 } from "~/server/service/campaign-service";
 import { validateDomainFromEmail } from "~/server/service/domain-service";
+import {
+  DEFAULT_BUCKET,
+  getDocumentUploadUrl,
+  isStorageConfigured,
+} from "~/server/service/storage-service";
 
 export const campaignRouter = createTRPCRouter({
   getCampaigns: teamProcedure
@@ -137,13 +144,19 @@ export const campaignRouter = createTRPCRouter({
       });
     }
 
+    const imageUploadSupported = isStorageConfigured();
+
     if (campaign?.contactBookId) {
       const contactBook = await db.contactBook.findUnique({
         where: { id: campaign.contactBookId },
       });
-      return { ...campaign, contactBook };
+      return { ...campaign, contactBook, imageUploadSupported };
     }
-    return { ...campaign, contactBook: null };
+    return {
+      ...campaign,
+      contactBook: null,
+      imageUploadSupported,
+    };
   }),
 
   sendCampaign: campaignProcedure.mutation(
@@ -180,4 +193,25 @@ export const campaignRouter = createTRPCRouter({
       return newCampaign;
     }
   ),
+
+  generateImagePresignedUrl: campaignProcedure
+    .input(
+      z.object({
+        name: z.string(),
+        type: z.string(),
+      })
+    )
+    .mutation(async ({ ctx: { team }, input }) => {
+      const extension = input.name.split(".").pop();
+      const randomName = `${nanoid()}.${extension}`;
+
+      const url = await getDocumentUploadUrl(
+        `${team.id}/${randomName}`,
+        input.type
+      );
+
+      const imageUrl = `${env.S3_COMPATIBLE_PUBLIC_URL}/${DEFAULT_BUCKET}/${team.id}/${randomName}`;
+
+      return { uploadUrl: url, imageUrl };
+    }),
 });

apps/web/src/server/nanoid.ts

@@ -4,3 +4,8 @@ export const smallNanoid = customAlphabet(
   "1234567890abcdefghijklmnopqrstuvwxyz",
   10
 );
+
+export const nanoid = customAlphabet(
+  "1234567890abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ",
+  21
+);

apps/web/src/server/service/storage-service.ts

@@ -0,0 +1,63 @@
+import { S3Client, PutObjectCommand } from "@aws-sdk/client-s3";
+import { getSignedUrl } from "@aws-sdk/s3-request-presigner";
+import { env } from "~/env";
+
+let S3: S3Client | null = null;
+export const DEFAULT_BUCKET = "unsend";
+
+export const isStorageConfigured = () =>
+  !!(
+    env.S3_COMPATIBLE_ACCESS_KEY &&
+    env.S3_COMPATIBLE_API_URL &&
+    env.S3_COMPATIBLE_PUBLIC_URL &&
+    env.S3_COMPATIBLE_SECRET_KEY
+  );
+
+const getClient = () => {
+  if (
+    !S3 &&
+    env.S3_COMPATIBLE_ACCESS_KEY &&
+    env.S3_COMPATIBLE_API_URL &&
+    env.S3_COMPATIBLE_PUBLIC_URL &&
+    env.S3_COMPATIBLE_SECRET_KEY
+  ) {
+    S3 = new S3Client({
+      region: "auto",
+      endpoint: env.S3_COMPATIBLE_API_URL,
+      credentials: {
+        accessKeyId: env.S3_COMPATIBLE_ACCESS_KEY,
+        secretAccessKey: env.S3_COMPATIBLE_SECRET_KEY,
+      },
+      forcePathStyle: true, // needed for minio
+    });
+  }
+
+  return S3;
+};
+
+export const getDocumentUploadUrl = async (
+  key: string,
+  fileType: string,
+  bucket: string = DEFAULT_BUCKET
+) => {
+  const s3Client = getClient();
+
+  if (!s3Client) {
+    throw new Error("R2 is not configured");
+  }
+
+  const url = await getSignedUrl(
+    s3Client,
+    new PutObjectCommand({
+      Bucket: bucket,
+      Key: key,
+      ContentType: fileType,
+    }),
+    {
+      expiresIn: 3600,
+      signableHeaders: new Set(["content-type"]),
+    }
+  );
+
+  return url;
+};