Deactivate subscription affiliations on cancellation

Author: aecsocket

apps/labrinth/migrations/20251024182919_subscription_affiliations.sql

@@ -1,9 +1,9 @@
 CREATE TABLE users_subscriptions_affiliations (
-	id              BIGSERIAL   NOT NULL PRIMARY KEY,
 	subscription_id BIGINT      NOT NULL REFERENCES users_subscriptions(id),
 	affiliate_code  BIGINT      NOT NULL REFERENCES affiliate_codes(id),
 	created_at      TIMESTAMPTZ NOT NULL DEFAULT CURRENT_TIMESTAMP,
-	deactivated_at  TIMESTAMPTZ
+	deactivated_at  TIMESTAMPTZ,
+	UNIQUE (subscription_id)
 );
 
 CREATE TABLE users_subscriptions_affiliations_payouts(

apps/labrinth/src/database/models/users_subscriptions_affiliations.rs

@@ -7,47 +7,43 @@ use crate::database::models::{
 
 #[derive(Debug, Clone, Serialize, Deserialize)]
 pub struct DBUsersSubscriptionsAffiliations {
-    pub id: i64,
     pub subscription_id: DBUserSubscriptionId,
     pub affiliate_code: DBAffiliateCodeId,
     pub deactivated_at: Option<DateTime<Utc>>,
 }
 
 impl DBUsersSubscriptionsAffiliations {
-    pub async fn insert<'a, E>(&mut self, exec: E) -> sqlx::Result<()>
+    pub async fn insert<'a, E>(&self, exec: E) -> sqlx::Result<()>
     where
         E: sqlx::PgExecutor<'a>,
     {
-        let id = sqlx::query_scalar!(
+        sqlx::query_scalar!(
             "
             INSERT INTO users_subscriptions_affiliations
                 (subscription_id, affiliate_code, deactivated_at)
             VALUES ($1, $2, $3)
-            RETURNING id
             ",
             self.subscription_id.0,
             self.affiliate_code.0,
             self.deactivated_at,
         )
         .fetch_one(exec)
         .await?;
-
-        self.id = id;
         Ok(())
     }
 
-    pub async fn update<'a, E>(&mut self, exec: E) -> sqlx::Result<()>
+    pub async fn deactivate<'a, E>(
+        subscription_id: DBUserSubscriptionId,
+        exec: E,
+    ) -> sqlx::Result<()>
     where
         E: sqlx::PgExecutor<'a>,
     {
         sqlx::query!(
             "UPDATE users_subscriptions_affiliations
-            SET subscription_id = $1, affiliate_code = $2, deactivated_at = $3
-            WHERE id = $4",
-            self.subscription_id.0,
-            self.affiliate_code.0,
-            self.deactivated_at,
-            self.id
+            SET deactivated_at = NOW()
+            WHERE subscription_id = $1",
+            subscription_id.0,
         )
         .execute(exec)
         .await?;

apps/labrinth/src/queue/payouts/affiliate.rs

@@ -48,10 +48,10 @@ pub async fn process_affiliate_payouts(postgres: &PgPool) -> Result<()> {
         INNER JOIN users_subscriptions_affiliations usa
             ON c.subscription_id = usa.subscription_id
             AND c.subscription_id IS NOT NULL
+            AND usa.deactivated_at IS NULL
         -- ...which have an affiliate code...
         INNER JOIN affiliate_codes ac
             ON usa.affiliate_code = ac.id
-            AND usa.deactivated_at IS NULL
         -- ...and where no payout to an affiliate has been made for this charge yet
         LEFT JOIN users_subscriptions_affiliations_payouts usap
             ON c.id = usap.charge_id

apps/labrinth/src/routes/internal/billing.rs

@@ -616,6 +616,11 @@ pub async fn edit_subscription(
             ..
         } if open_charge.status == ChargeStatus::Failed => {
             if cancelled {
+                DBUsersSubscriptionsAffiliations::deactivate(
+                    subscription.id,
+                    &mut *transaction,
+                )
+                .await?;
                 open_charge.status = ChargeStatus::Cancelled;
             } else {
                 // Forces another resubscription attempt
@@ -635,6 +640,11 @@ pub async fn edit_subscription(
         ) =>
         {
             open_charge.status = if cancelled {
+                DBUsersSubscriptionsAffiliations::deactivate(
+                    subscription.id,
+                    &mut *transaction,
+                )
+                .await?;
                 ChargeStatus::Cancelled
             } else {
                 ChargeStatus::Open
@@ -2073,7 +2083,6 @@ pub async fn stripe_webhook(
                                 .and_then(|m| m.affiliate_code)
                             {
                                 DBUsersSubscriptionsAffiliations {
-                                    id: 0,
                                     subscription_id: subscription.id,
                                     affiliate_code: DBAffiliateCodeId::from(
                                         affiliate_code,
@@ -2083,8 +2092,6 @@ pub async fn stripe_webhook(
                                 .insert(&mut *transaction)
                                 .await?;
                             }
-
-                            // TODO affiliate code
                         };
 
                         subscription.status = SubscriptionStatus::Provisioned;