cleanup referral attributes

Author: waleedlatif1

apps/sim/app/api/attribution/route.ts

@@ -1,17 +1,17 @@
 /**
  * POST /api/attribution
  *
- * Automatic UTM-based referral attribution for new signups.
+ * Automatic UTM-based referral attribution.
  *
- * Reads the `sim_utm` cookie (set by proxy on auth pages), verifies the user
- * account was created after the cookie was set, matches a campaign by UTM
- * specificity, and atomically inserts an attribution record + applies bonus credits.
+ * Reads the `sim_utm` cookie (set by proxy on auth pages), matches a campaign
+ * by UTM specificity, and atomically inserts an attribution record + applies
+ * bonus credits.
  *
  * Idempotent — the unique constraint on `userId` prevents double-attribution.
  */
 
 import { db } from '@sim/db'
-import { referralAttribution, referralCampaigns, user, userStats } from '@sim/db/schema'
+import { referralAttribution, referralCampaigns, userStats } from '@sim/db/schema'
 import { createLogger } from '@sim/logger'
 import { eq } from 'drizzle-orm'
 import { nanoid } from 'nanoid'
@@ -24,7 +24,6 @@ import { applyBonusCredits } from '@/lib/billing/credits/bonus'
 const logger = createLogger('AttributionAPI')
 
 const COOKIE_NAME = 'sim_utm'
-const CLOCK_DRIFT_TOLERANCE_MS = 60 * 1000
 
 const UtmCookieSchema = z.object({
   utm_source: z.string().optional(),
@@ -33,7 +32,7 @@ const UtmCookieSchema = z.object({
   utm_content: z.string().optional(),
   referrer_url: z.string().optional(),
   landing_page: z.string().optional(),
-  created_at: z.string().min(1),
+  created_at: z.string().optional(),
 })
 
 /**
@@ -114,34 +113,6 @@ export async function POST() {
       return NextResponse.json({ attributed: false, reason: 'invalid_cookie' })
     }
 
-    const cookieCreatedAt = Number(utmData.created_at)
-    if (!Number.isFinite(cookieCreatedAt)) {
-      logger.warn('UTM cookie has invalid created_at timestamp', { userId: session.user.id })
-      cookieStore.delete(COOKIE_NAME)
-      return NextResponse.json({ attributed: false, reason: 'invalid_cookie' })
-    }
-
-    const userRows = await db
-      .select({ createdAt: user.createdAt })
-      .from(user)
-      .where(eq(user.id, session.user.id))
-      .limit(1)
-
-    if (userRows.length === 0) {
-      return NextResponse.json({ error: 'User not found' }, { status: 404 })
-    }
-
-    const userCreatedAt = userRows[0].createdAt.getTime()
-    if (userCreatedAt < cookieCreatedAt - CLOCK_DRIFT_TOLERANCE_MS) {
-      logger.info('User account predates UTM cookie, skipping attribution', {
-        userId: session.user.id,
-        userCreatedAt: new Date(userCreatedAt).toISOString(),
-        cookieCreatedAt: new Date(cookieCreatedAt).toISOString(),
-      })
-      cookieStore.delete(COOKIE_NAME)
-      return NextResponse.json({ attributed: false, reason: 'account_predates_cookie' })
-    }
-
     const matchedCampaign = await findMatchingCampaign(utmData)
     if (!matchedCampaign) {
       cookieStore.delete(COOKIE_NAME)

apps/sim/app/workspace/[workspaceId]/w/components/sidebar/components/settings-modal/components/subscription/components/referral-code/referral-code.tsx

@@ -67,10 +67,10 @@ export function ReferralCode({ onRedeemComplete }: ReferralCodeProps) {
   }
 
   return (
-    <div className='flex items-center justify-between gap-[12px]'>
-      <Label className='shrink-0'>Referral Code</Label>
-      <div className='flex items-center gap-[8px]'>
-        <div className='flex flex-col'>
+    <div className='flex flex-col'>
+      <div className='flex items-center justify-between gap-[12px]'>
+        <Label className='shrink-0'>Referral Code</Label>
+        <div className='flex items-center gap-[8px]'>
           <Input
             type='text'
             value={code}
@@ -85,16 +85,18 @@ export function ReferralCode({ onRedeemComplete }: ReferralCodeProps) {
             className='h-[32px] w-[140px] text-[12px]'
             disabled={isRedeeming}
           />
-          {error && <span className='mt-[4px] text-[11px] text-[var(--text-error)]'>{error}</span>}
+          <Button
+            variant='active'
+            className='h-[32px] shrink-0 rounded-[6px] text-[12px]'
+            onClick={handleRedeem}
+            disabled={isRedeeming || !code.trim()}
+          >
+            {isRedeeming ? 'Redeeming...' : 'Redeem'}
+          </Button>
         </div>
-        <Button
-          variant='active'
-          className='h-[32px] shrink-0 rounded-[6px] text-[12px]'
-          onClick={handleRedeem}
-          disabled={isRedeeming || !code.trim()}
-        >
-          {isRedeeming ? 'Redeeming...' : 'Redeem'}
-        </Button>
+      </div>
+      <div className='mt-[4px] min-h-[18px] text-right'>
+        {error && <span className='text-[11px] text-[var(--text-error)]'>{error}</span>}
       </div>
     </div>
   )

apps/sim/hooks/use-referral-attribution.ts

@@ -8,7 +8,6 @@ const logger = createLogger('ReferralAttribution')
 const COOKIE_NAME = 'sim_utm'
 
 const TERMINAL_REASONS = new Set([
-  'account_predates_cookie',
   'invalid_cookie',
   'no_utm_cookie',
   'no_matching_campaign',

apps/sim/proxy.ts

@@ -143,8 +143,7 @@ const UTM_COOKIE_MAX_AGE = 3600
 
 /**
  * Sets a `sim_utm` cookie when UTM params are present on auth pages.
- * Captures UTM values, the HTTP Referer, landing page, and a timestamp
- * used by the attribution API to verify the user signed up after visiting the link.
+ * Captures UTM values, the HTTP Referer, landing page, and a timestamp.
  */
 function setUtmCookie(request: NextRequest, response: NextResponse): void {
   const { searchParams, pathname } = request.nextUrl
@@ -179,7 +178,9 @@ export async function proxy(request: NextRequest) {
 
   if (url.pathname === '/login' || url.pathname === '/signup') {
     if (hasActiveSession) {
-      return NextResponse.redirect(new URL('/workspace', request.url))
+      const redirect = NextResponse.redirect(new URL('/workspace', request.url))
+      setUtmCookie(request, redirect)
+      return redirect
     }
     const response = NextResponse.next()
     response.headers.set('Content-Security-Policy', generateRuntimeCSP())