Use generated updated_at for subscription table

Author: jahooma

packages/billing/src/subscription-webhooks.ts

@@ -146,7 +146,6 @@ export async function handleSubscriptionInvoicePaid(params: {
         ),
         billing_period_end: new Date(stripeSub.current_period_end * 1000),
         cancel_at_period_end: stripeSub.cancel_at_period_end,
-        updated_at: new Date(),
       },
     })
 
@@ -197,7 +196,6 @@ export async function handleSubscriptionInvoicePaymentFailed(params: {
     .update(schema.subscription)
     .set({
       status: 'past_due',
-      updated_at: new Date(),
     })
     .where(eq(schema.subscription.stripe_subscription_id, subscriptionId))
 
@@ -310,7 +308,6 @@ export async function handleSubscriptionUpdated(params: {
         billing_period_end: new Date(
           stripeSubscription.current_period_end * 1000,
         ),
-        updated_at: new Date(),
       },
     })
 
@@ -356,7 +353,6 @@ export async function handleSubscriptionDeleted(params: {
       status: 'canceled',
       scheduled_tier: null,
       canceled_at: new Date(),
-      updated_at: new Date(),
     })
     .where(eq(schema.subscription.stripe_subscription_id, subscriptionId))
 
@@ -456,7 +452,6 @@ export async function handleSubscriptionScheduleCreatedOrUpdated(params: {
     .update(schema.subscription)
     .set({
       scheduled_tier: scheduledTier,
-      updated_at: new Date(),
     })
     .where(eq(schema.subscription.stripe_subscription_id, subscriptionId))
     .returning({ tier: schema.subscription.tier })
@@ -526,7 +521,6 @@ export async function handleSubscriptionScheduleReleasedOrCanceled(params: {
     .update(schema.subscription)
     .set({
       scheduled_tier: null,
-      updated_at: new Date(),
     })
     .where(eq(schema.subscription.stripe_subscription_id, subscriptionId))
     .returning({ tier: schema.subscription.tier })

packages/internal/src/db/migrations/0039_automatic_updated_at.sql

@@ -0,0 +1,24 @@
+-- Create a reusable function that sets updated_at to NOW()
+CREATE OR REPLACE FUNCTION set_updated_at()
+RETURNS TRIGGER AS $$
+BEGIN
+  NEW.updated_at = NOW();
+  RETURN NEW;
+END;
+$$ LANGUAGE plpgsql;
+
+--> statement-breakpoint
+
+-- Add trigger to subscription table
+CREATE TRIGGER trigger_subscription_updated_at
+  BEFORE UPDATE ON "subscription"
+  FOR EACH ROW
+  EXECUTE FUNCTION set_updated_at();
+
+--> statement-breakpoint
+
+-- Add trigger to limit_override table
+CREATE TRIGGER trigger_limit_override_updated_at
+  BEFORE UPDATE ON "limit_override"
+  FOR EACH ROW
+  EXECUTE FUNCTION set_updated_at();

web/src/app/api/stripe/cancel-subscription/route.ts

@@ -44,7 +44,7 @@ export async function POST() {
   try {
     await db
       .update(schema.subscription)
-      .set({ cancel_at_period_end: true, scheduled_tier: null, updated_at: new Date() })
+      .set({ cancel_at_period_end: true, scheduled_tier: null })
       .where(
         eq(
           schema.subscription.stripe_subscription_id,

web/src/app/api/stripe/change-subscription-tier/route.ts

@@ -122,7 +122,7 @@ export async function POST(req: NextRequest) {
       if (isCancelDowngrade) {
         await db
           .update(schema.subscription)
-          .set({ scheduled_tier: null, updated_at: new Date() })
+          .set({ scheduled_tier: null })
           .where(
             eq(
               schema.subscription.stripe_subscription_id,
@@ -137,7 +137,6 @@ export async function POST(req: NextRequest) {
               tier,
               stripe_price_id: newPriceId,
               scheduled_tier: null,
-              updated_at: new Date(),
             })
             .where(
               eq(
@@ -158,7 +157,6 @@ export async function POST(req: NextRequest) {
           .update(schema.subscription)
           .set({
             scheduled_tier: tier,
-            updated_at: new Date(),
           })
           .where(
             eq(

web/src/app/api/user/subscription/route.ts

@@ -10,6 +10,11 @@ import { getServerSession } from 'next-auth'
 import { authOptions } from '@/app/api/auth/[...nextauth]/auth-options'
 import { logger } from '@/util/logger'
 
+import type {
+  NoSubscriptionResponse,
+  ActiveSubscriptionResponse,
+} from '@codebuff/common/types/subscription'
+
 export async function GET() {
   const session = await getServerSession(authOptions)
   if (!session?.user?.id) {
@@ -19,16 +24,17 @@ export async function GET() {
   const userId = session.user.id
   const subscription = await getActiveSubscription({ userId, logger })
 
-  if (!subscription) {
-    return NextResponse.json({ hasSubscription: false })
+  if (!subscription || !subscription.tier) {
+    const response: NoSubscriptionResponse = { hasSubscription: false }
+    return NextResponse.json(response)
   }
 
   const [rateLimit, limits] = await Promise.all([
     checkRateLimit({ userId, subscription, logger }),
     getSubscriptionLimits({ userId, logger, tier: subscription.tier }),
   ])
 
-  return NextResponse.json({
+  const response: ActiveSubscriptionResponse = {
     hasSubscription: true,
     displayName: SUBSCRIPTION_DISPLAY_NAME,
     subscription: {
@@ -52,5 +58,6 @@ export async function GET() {
       weeklyPercentUsed: rateLimit.weeklyPercentUsed,
     },
     limits,
-  })
+  }
+  return NextResponse.json(response)
 }