fix: subscription not found in generate referral fn

README.md

@@ -36,4 +36,4 @@
 6. **Documentation and Deployment**
    - [x] Document API endpoints using Swagger/OpenAPI.
    - [x] Set up deployment using Docker.
-   - [ ] Ensure proper security and performance settings in production.
+   - [x] Ensure proper security and performance settings in production.

src/app/errors/subscription-not-found-error.ts

@@ -0,0 +1,5 @@
+export class SubscriptionNotFound extends Error {
+  constructor() {
+    super('Subscription not found.')
+  }
+}

src/app/functions/generate-referral.ts

@@ -4,6 +4,7 @@ import { schema } from '@/db/schema'
 import { and, eq } from 'drizzle-orm'
 import { ReferralLinkAlreadyExists } from '../errors/referral-link-already-exists'
 import { randomBytes } from 'node:crypto'
+import { SubscriptionNotFound } from '../errors/subscription-not-found-error'
 
 type GenerateReferralInput = {
   email: string
@@ -14,47 +15,47 @@ export async function generateReferral({
   email,
   eventId,
 }: GenerateReferralInput) {
-  try {
-    const [existingReferralLink] = await db
-      .select()
-      .from(schema.referral)
-      .where(
-        and(
-          eq(schema.referral.email, email),
-          eq(schema.referral.eventId, eventId)
-        )
+  const [existingReferralLink] = await db
+    .select()
+    .from(schema.referral)
+    .where(
+      and(
+        eq(schema.referral.email, email),
+        eq(schema.referral.eventId, eventId)
       )
+    )
 
-    if (existingReferralLink) {
-      return makeLeft(new ReferralLinkAlreadyExists())
-    }
-
-    const token = randomBytes(16).toString('hex')
-    const url = `${process.env.BASE_URL}/referral?token=${token}`
-
-    const [subscription] = await db
-      .select()
-      .from(schema.subscriptions)
-      .where(
-        and(
-          eq(schema.subscriptions.email, email),
-          eq(schema.subscriptions.eventId, eventId)
-        )
+  if (existingReferralLink) {
+    return makeLeft(new ReferralLinkAlreadyExists())
+  }
+
+  const token = randomBytes(16).toString('hex')
+  const url = `${process.env.BASE_URL}/referral?token=${token}`
+
+  const [subscription] = await db
+    .select()
+    .from(schema.subscriptions)
+    .where(
+      and(
+        eq(schema.subscriptions.email, email),
+        eq(schema.subscriptions.eventId, eventId)
       )
+    )
 
-    const [referral] = await db
-      .insert(schema.referral)
-      .values({
-        email,
-        eventId,
-        link: url,
-        token: token,
-        parentId: subscription.referralId || null,
-      })
-      .returning()
-
-    return makeRight({ referral })
-  } catch (error) {
-    throw error
+  if (!subscription) {
+    return makeLeft(new SubscriptionNotFound())
   }
+
+  const [referral] = await db
+    .insert(schema.referral)
+    .values({
+      email,
+      eventId,
+      link: url,
+      token: token,
+      parentId: subscription.referralId || null,
+    })
+    .returning()
+
+  return makeRight({ referral })
 }