fix: cancel Stripe subscription when account is deleted

Also fix typings for purchase in database

apps/backend/src/database/models/Purchase.ts

@@ -27,7 +27,16 @@ export class Purchase extends Model {
       },
       status: {
         type: ["string"],
-        enum: ["active", "canceled", "trialing", "past_due"],
+        enum: [
+          "active",
+          "canceled",
+          "trialing",
+          "past_due",
+          "incomplete",
+          "unpaid",
+          "incomplete_expired",
+          "paused",
+        ],
       },
       endDate: { type: ["string", "null"] },
       startDate: { type: ["string"] },
@@ -40,12 +49,20 @@ export class Purchase extends Model {
   planId!: string;
   purchaserId!: string | null;
   stripeSubscriptionId!: string | null;
-  source!: string;
+  source!: "github" | "stripe";
   endDate!: string | null;
   startDate!: string;
   trialEndDate!: string | null;
   paymentMethodFilled!: boolean | null;
-  status!: string;
+  status!:
+    | "active"
+    | "canceled"
+    | "trialing"
+    | "past_due"
+    | "incomplete"
+    | "unpaid"
+    | "incomplete_expired"
+    | "paused";
 
   static override get relationMappings(): RelationMappings {
     return {

apps/backend/src/graphql/definitions/Account.ts

@@ -50,6 +50,14 @@ export const typeDefs = gql`
     past_due
     "Post-cancelation date"
     canceled
+    "Incomplete"
+    incomplete
+    "Incomplete expired"
+    incomplete_expired
+    "Unpaid"
+    unpaid
+    "Paused"
+    paused
   }
 
   enum TrialStatus {

apps/backend/src/graphql/services/account.ts

@@ -7,6 +7,7 @@ import {
 } from "@/database/models/index.js";
 
 import { deleteProject } from "./project.js";
+import { cancelStripeSubscription } from "@/stripe/index.js";
 
 export const getWritableAccount = async (args: {
   id: string;
@@ -40,6 +41,12 @@ export const deleteAccount = async (args: {
       });
     }),
   );
+  const subscription = account.$getSubscription();
+  const activePurchase = await subscription.getActivePurchase();
+  // Cancel the Stripe subscription if it exists
+  if (activePurchase?.stripeSubscriptionId) {
+    await cancelStripeSubscription(activePurchase.stripeSubscriptionId);
+  }
   await Purchase.query().where("accountId", account.id).delete();
   await account.$query().delete();
 

apps/backend/src/stripe/index.ts

@@ -187,20 +187,22 @@ export const createPurchaseFromSubscription = async ({
   });
 };
 
+export async function cancelStripeSubscription(subscriptionId: string) {
+  const subscription = await stripe.subscriptions.retrieve(subscriptionId);
+  if (subscription.status === "canceled") {
+    return;
+  }
+  await stripe.subscriptions.cancel(subscriptionId);
+}
+
 const getPurchaseFromStripeSubscriptionId = async (
   stripeSubscriptionId: string,
 ) => {
   const purchase = await Purchase.query().findOne({
     stripeSubscriptionId,
   });
 
-  if (!purchase) {
-    throw new Error(
-      `Purchase not found for subscription ${stripeSubscriptionId}`,
-    );
-  }
-
-  return purchase;
+  return purchase ?? null;
 };
 
 export const getStripePriceFromPlanOrThrow = async (plan: Plan) => {
@@ -322,6 +324,11 @@ export const handleStripeEvent = async ({
         const purchase = await getPurchaseFromStripeSubscriptionId(
           subscription.id,
         );
+        if (!purchase) {
+          throw new Error(
+            `No purchase found for stripe subscription id ${subscription.id}`,
+          );
+        }
         await updatePurchaseFromSubscription(purchase, subscription);
       }
 
@@ -353,16 +360,33 @@ export const handleStripeEvent = async ({
 
       return;
     }
-    case "customer.subscription.updated":
-    case "customer.subscription.deleted": {
+    case "customer.subscription.updated": {
       const subscription = data.object as Stripe.Subscription;
       const purchase = await getPurchaseFromStripeSubscriptionId(
         subscription.id,
       );
+      if (!purchase) {
+        throw new Error(
+          `No purchase found for stripe subscription id ${subscription.id}`,
+        );
+      }
       await updatePurchaseFromSubscription(purchase, subscription);
       return;
     }
 
+    case "customer.subscription.deleted": {
+      const subscription = data.object as Stripe.Subscription;
+      const purchase = await getPurchaseFromStripeSubscriptionId(
+        subscription.id,
+      );
+      // Purchase can be null in case the team as been deleted
+      if (!purchase) {
+        return;
+      }
+      await purchase.$query().patch({ status: "canceled" });
+      return;
+    }
+
     default:
       console.log(`Unhandled event type ${type}`);
   }

apps/backend/src/synchronize/github/updatePurchase.ts

@@ -42,7 +42,7 @@ export const updatePurchase = async (
       Purchase.query(trx)
         .patch({
           endDate: effectiveDate,
-          status: new Date(effectiveDate) > new Date() ? "cancelled" : "active",
+          status: new Date(effectiveDate) > new Date() ? "canceled" : "active",
         })
         .findById(activePurchase.id),
       Purchase.query(trx).insert({

apps/frontend/src/gql/graphql.ts

@@ -630,12 +630,20 @@ export enum PurchaseStatus {
   Active = 'active',
   /** Post-cancelation date */
   Canceled = 'canceled',
+  /** Incomplete */
+  Incomplete = 'incomplete',
+  /** Incomplete expired */
+  IncompleteExpired = 'incomplete_expired',
   /** No paid purchase */
   Missing = 'missing',
   /** Payment due */
   PastDue = 'past_due',
+  /** Paused */
+  Paused = 'paused',
   /** Ongoing trial */
-  Trialing = 'trialing'
+  Trialing = 'trialing',
+  /** Unpaid */
+  Unpaid = 'unpaid'
 }
 
 export type Query = {