Don't show referral banner if codebuff account > 1 week old

Author: jahooma

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

@@ -26,6 +26,7 @@ export async function GET() {
         auto_topup_enabled: true,
         auto_topup_threshold: true,
         auto_topup_amount: true,
+        created_at: true,
       },
     })
 
@@ -43,6 +44,7 @@ export async function GET() {
       auto_topup_threshold: user.auto_topup_threshold ?? 500,
       auto_topup_amount: user.auto_topup_amount ?? 2000,
       auto_topup_blocked_reason,
+      created_at: user.created_at,
     }
 
     return NextResponse.json(response)

web/src/components/ui/banner.tsx

@@ -10,14 +10,26 @@ import posthog from 'posthog-js'
 import { Suspense, useState } from 'react'
 
 import { Button } from './button'
+import { useUserProfile } from '@/hooks/use-user-profile'
 
 function BannerContent() {
   const [isVisible, setIsVisible] = useState(true)
   const searchParams = useSearchParams()
   const referrer = searchParams.get('referrer')
   const { data: session } = useSession()
 
-  if (!isVisible || !session?.user) return null
+  const { data: userProfile } = useUserProfile()
+
+  if (!isVisible || !session?.user || !userProfile) return null
+
+  // Check if account is less than a week old
+  const isNewAccount = userProfile.created_at
+    ? new Date().getTime() - new Date(userProfile.created_at).getTime() <
+      7 * 24 * 60 * 60 * 1000
+    : false
+
+  // Only show banner for new accounts (less than a week old)
+  if (!isNewAccount) return null
 
   const isPersonalReferral = !!referrer
 

web/src/hooks/use-user-profile.ts

@@ -0,0 +1,93 @@
+import { useQuery } from '@tanstack/react-query'
+import { useSession } from 'next-auth/react'
+import { useEffect } from 'react'
+
+import type { UserProfile } from '@/types/user'
+
+const USER_PROFILE_STORAGE_KEY = 'codebuff-user-profile'
+
+// Helper functions for local storage
+const getUserProfileFromStorage = (): UserProfile | null => {
+  if (typeof window === 'undefined') return null
+  
+  try {
+    const stored = localStorage.getItem(USER_PROFILE_STORAGE_KEY)
+    if (!stored) return null
+    
+    const parsed = JSON.parse(stored)
+    // Convert created_at string back to Date if it exists
+    if (parsed.created_at) {
+      parsed.created_at = new Date(parsed.created_at)
+    }
+    return parsed
+  } catch {
+    return null
+  }
+}
+
+const setUserProfileToStorage = (profile: UserProfile) => {
+  if (typeof window === 'undefined') return
+  
+  try {
+    localStorage.setItem(USER_PROFILE_STORAGE_KEY, JSON.stringify(profile))
+  } catch {
+    // Silently fail if localStorage is not available
+  }
+}
+
+const clearUserProfileFromStorage = () => {
+  if (typeof window === 'undefined') return
+  
+  try {
+    localStorage.removeItem(USER_PROFILE_STORAGE_KEY)
+  } catch {
+    // Silently fail if localStorage is not available
+  }
+}
+
+export const useUserProfile = () => {
+  const { data: session } = useSession()
+
+  const query = useQuery<UserProfile>({
+    queryKey: ['user-profile'],
+    queryFn: async () => {
+      const response = await fetch('/api/user/profile')
+      if (!response.ok) {
+        throw new Error('Failed to fetch user profile')
+      }
+      const data = await response.json()
+      
+      // Convert created_at string to Date if it exists
+      if (data.created_at) {
+        data.created_at = new Date(data.created_at)
+      }
+      
+      return data
+    },
+    enabled: !!session?.user,
+    staleTime: 5 * 60 * 1000, // 5 minutes
+    initialData: () => {
+      // Return undefined if no data, which is compatible with useQuery
+      return getUserProfileFromStorage() ?? undefined
+    },
+  })
+
+  // Persist to localStorage whenever data changes
+  useEffect(() => {
+    if (query.data) {
+      setUserProfileToStorage(query.data)
+    }
+  }, [query.data])
+
+  // Clear localStorage when user logs out
+  useEffect(() => {
+    if (!session?.user) {
+      clearUserProfileFromStorage()
+    }
+  }, [session?.user])
+
+  return {
+    ...query,
+    clearCache: clearUserProfileFromStorage,
+  }
+}

web/src/types/user.ts

@@ -11,4 +11,5 @@ export interface UserProfile {
   auto_topup_threshold: number | null
   auto_topup_amount: number | null
   auto_topup_blocked_reason: string | null
+  created_at: Date | null
 }