[PB-1435]: feat/update users tier on subscription change.

src/constants.ts

@@ -1 +1,2 @@
 export const FREE_PLAN_BYTES_SPACE = 2 * 1024 * 1024 * 1024;
+export const FREE_INDIVIDUAL_TIER = 'free_individual_tier';

src/services/StorageService.ts

@@ -43,3 +43,16 @@ export async function createOrUpdateUser(maxSpaceBytes: string, email: string, c
     },
   );
 }
+
+export async function updateUserTier(uuid: string, planId: string, config: AppConfig) {
+  return axios.put(
+    `${config.DRIVE_GATEWAY_URL}/api/gateway/user/update/tier`,
+    { planId, uuid },
+    {
+      headers: {
+        'Content-Type': 'application/json',
+      },
+      auth: { username: config.DRIVE_GATEWAY_USER, password: config.DRIVE_GATEWAY_PASSWORD },
+    },
+  );
+}

src/webhooks/handleCheckoutSessionCompleted.ts

@@ -4,7 +4,7 @@ import Stripe from 'stripe';
 import { type AppConfig } from '../config';
 import CacheService from '../services/CacheService';
 import { PaymentService, PriceMetadata } from '../services/PaymentService';
-import { createOrUpdateUser } from '../services/StorageService';
+import { createOrUpdateUser, updateUserTier } from '../services/StorageService';
 import { UsersService } from '../services/UsersService';
 
 export default async function handleCheckoutSessionCompleted(
@@ -59,6 +59,17 @@ export default async function handleCheckoutSessionCompleted(
     throw err;
   }
 
+  try {
+    await updateUserTier(user.uuid, price.product as string, config);
+  } catch (err) {
+    log.error(
+      `Error while updating user tier: email: ${session.customer_email}, planId: ${price.product} `,
+    );
+    log.error(err);
+
+    throw err;
+  }
+
   try {
     const { customerId } = await usersService.findUserByUuid(user.uuid);
     if ((price.metadata as PriceMetadata).planType === 'one_time') {

src/webhooks/handleLifetimeRefunded.ts

@@ -1,15 +1,17 @@
 import { FastifyLoggerInstance } from 'fastify';
-import { FREE_PLAN_BYTES_SPACE } from '../constants';
+import { FREE_INDIVIDUAL_TIER, FREE_PLAN_BYTES_SPACE } from '../constants';
 import CacheService from '../services/CacheService';
-import { StorageService } from '../services/StorageService';
+import { StorageService, updateUserTier } from '../services/StorageService';
 import { UsersService } from '../services/UsersService';
+import { AppConfig } from '../config';
 
 export default async function handleLifetimeRefunded(
   storageService: StorageService,
   usersService: UsersService,
   customerId: string,
   cacheService: CacheService,
   log: FastifyLoggerInstance,
+  config: AppConfig,
 ): Promise<void> {
   const { uuid } = await usersService.findUserByCustomerID(customerId);
 
@@ -20,5 +22,15 @@ export default async function handleLifetimeRefunded(
   } catch (err) {
     log.error(`Error in handleLifetimeRefunded after trying to clear ${customerId} subscription`);
   }
+  try {
+    await updateUserTier(uuid, FREE_INDIVIDUAL_TIER, config);
+  } catch (err) {
+    log.error(
+      `Error while updating user tier: uuid: ${uuid} `,
+    );
+    log.error(err);
+    throw err;
+  }
+
   return storageService.changeStorage(uuid, FREE_PLAN_BYTES_SPACE);
 }

src/webhooks/handleSubscriptionCanceled.ts

@@ -1,21 +1,33 @@
 import { FastifyLoggerInstance } from 'fastify';
-import { FREE_PLAN_BYTES_SPACE } from '../constants';
+import { FREE_INDIVIDUAL_TIER, FREE_PLAN_BYTES_SPACE } from '../constants';
 import CacheService from '../services/CacheService';
-import { StorageService } from '../services/StorageService';
+import { StorageService, updateUserTier } from '../services/StorageService';
 import { UsersService } from '../services/UsersService';
+import { AppConfig } from '../config';
 
 export default async function handleSubscriptionCanceled(
   storageService: StorageService,
   usersService: UsersService,
   customerId: string,
   cacheService: CacheService,
   log: FastifyLoggerInstance,
+  config: AppConfig,
 ): Promise<void> {
   const { uuid } = await usersService.findUserByCustomerID(customerId);
   try {
     await cacheService.clearSubscription(customerId);
   } catch (err) {
     log.error(`Error in handleSubscriptionCanceled after trying to clear ${customerId} subscription`);
   }
+
+  try {
+    await updateUserTier(uuid, FREE_INDIVIDUAL_TIER, config);
+  } catch (err) {
+    log.error(
+      `[TIER/SUB_CANCELED] Error while updating user tier: uuid: ${uuid} `,
+    );
+    log.error(err);
+  }
+
   return storageService.changeStorage(uuid, FREE_PLAN_BYTES_SPACE);
 }

src/webhooks/handleSubscriptionUpdated.ts

@@ -1,34 +1,50 @@
 import { FastifyLoggerInstance } from 'fastify';
 import Stripe from 'stripe';
-import { FREE_PLAN_BYTES_SPACE } from '../constants';
+import { FREE_INDIVIDUAL_TIER, FREE_PLAN_BYTES_SPACE } from '../constants';
 import CacheService from '../services/CacheService';
 import { PriceMetadata } from '../services/PaymentService';
-import { StorageService } from '../services/StorageService';
+import { StorageService, updateUserTier } from '../services/StorageService';
 import { UsersService } from '../services/UsersService';
+import { AppConfig } from '../config';
 
 export default async function handleSubscriptionUpdated(
   storageService: StorageService,
   usersService: UsersService,
   subscription: Stripe.Subscription,
   cacheService: CacheService,
   log: FastifyLoggerInstance,
+  config: AppConfig,
 ): Promise<void> {
   const customerId = subscription.customer as string;
   const { uuid, lifetime } = await usersService.findUserByCustomerID(customerId);
   if (lifetime) {
     return;
   }
+  const isSubscriptionCanceled = subscription.status === 'canceled';
 
   const bytesSpace =
-    subscription.status === 'canceled'
+    isSubscriptionCanceled
       ? FREE_PLAN_BYTES_SPACE
       : parseInt((subscription.items.data[0].price.metadata as unknown as PriceMetadata).maxSpaceBytes);
 
+  const planId = isSubscriptionCanceled
+    ? FREE_INDIVIDUAL_TIER : subscription.items.data[0].price.product as string;
+
   try {
     await cacheService.clearSubscription(customerId);
   } catch (err) {
     log.error(`Error in handleSubscriptionUpdated after trying to clear ${customerId} subscription`);
   }
 
+  try {
+    await updateUserTier(uuid, planId, config);
+  } catch (err) {
+    log.error(
+      `Error while updating user tier: uuid: ${uuid} `,
+    );
+    log.error(err);
+    throw err;
+  }
+
   return storageService.changeStorage(uuid, bytesSpace);
 }

src/webhooks/index.ts

@@ -48,6 +48,7 @@ export default function (
             (event.data.object as Stripe.Subscription).customer as string,
             cacheService,
             fastify.log,
+            config,
           );
           break;
         case 'customer.subscription.updated':
@@ -57,6 +58,7 @@ export default function (
             event.data.object as Stripe.Subscription,
             cacheService,
             fastify.log,
+            config,
           );
           break;
         case 'payment_method.attached':
@@ -94,6 +96,7 @@ export default function (
             (event.data.object as Stripe.Charge).customer as string,
             cacheService,
             fastify.log,
+            config,
           );
           break;
         default: