affiliate: Make affiliateCode available only for active affiliates

podkrepi-bg · Oct 19, 2023 · ed56cfc · ed56cfc
1 parent 39b7f7a
commit ed56cfc
Showing 1 changed file with 11 additions and 4 deletions.
diff --git a/apps/api/src/affiliate/affiliate.controller.ts b/apps/api/src/affiliate/affiliate.controller.ts
@@ -51,16 +51,23 @@ export class AffiliateController {
 
     if (!affiliate) throw new NotFoundException('Affiliate not found')
 
+    let affiliateCode: string | null = affiliate.affiliateCode
+
     if (affiliate.status === newStatus) {
       throw new ConflictException('Status is the same')
     }
 
-    if (affiliate.status === 'pending' && newStatus === 'active') {
-      const affiliateCode = affiliateCodeGenerator(affiliate.id)
-      return await this.affiliateService.updateStatus(affiliateId, newStatus, affiliateCode)
+    if (
+      (affiliate.status === 'pending' || affiliate.status === 'cancelled') &&
+      newStatus === 'active'
+    ) {
+      affiliateCode = affiliateCodeGenerator(affiliate.id)
     }
 
-    return await this.affiliateService.updateStatus(affiliateId, newStatus, affiliate.affiliateCode)
+    if (affiliate.status === 'active' && newStatus !== 'active') {
+      affiliateCode = null
+    }
+    return await this.affiliateService.updateStatus(affiliateId, newStatus, affiliateCode)
   }
 
   @Get(':affiliateCode')