Restore referral landing pages and wire freebuff creator attribution

Brings back the codebuff referral landing routes (/[sponsee], /referrals/[code],
/api/referrals/[code]) without free-credit mentions, and ensures freebuff persists
?referrer= through OAuth by saving to localStorage in all sign-in entrypoints and
mounting ReferrerTracker in the root layout.

Co-Authored-By: Claude Opus 4.7 <noreply@anthropic.com>

Author: jahooma

freebuff/web/src/app/layout.tsx

@@ -3,6 +3,7 @@ import '@/styles/globals.css'
 import type { Metadata } from 'next'
 
 import { Footer } from '@/components/footer'
+import { ReferrerTracker } from '@/components/referrer-tracker'
 import { ThemeProvider } from '@/components/theme-provider'
 import { siteConfig } from '@/lib/constant'
 import { fonts } from '@/lib/fonts'
@@ -55,6 +56,7 @@ export default function RootLayout({
         <ThemeProvider attribute="class">
           <SessionProvider>
             <PostHogProvider>
+              <ReferrerTracker />
               <div className="flex-grow">{children}</div>
               <Footer />
             </PostHogProvider>

freebuff/web/src/app/onboard/page.tsx

@@ -13,7 +13,6 @@ import {
 import { isAuthCodeExpired, parseAuthCode, validateAuthCode } from './_helpers'
 import { authOptions } from '../api/auth/[...nextauth]/auth-options'
 
-import { ReferrerTracker } from '@/components/referrer-tracker'
 import {
   Card,
   CardHeader,
@@ -47,7 +46,6 @@ function StatusCard({
 }) {
   return (
     <main className="container mx-auto flex flex-col items-center py-20">
-      <ReferrerTracker />
       <div className="w-full sm:w-1/2 md:w-2/3">
         <Card>
           <CardHeader>

freebuff/web/src/components/login/login-card.tsx

@@ -19,7 +19,16 @@ export function LoginCard({ authCode }: { authCode?: string | null }) {
   const { data: session } = useSession()
   const searchParams = useSearchParams() ?? new URLSearchParams()
 
+  const persistReferrer = () => {
+    const referrer = searchParams.get('referrer')
+    if (referrer) {
+      localStorage.setItem('freebuff_referrer', referrer)
+    }
+  }
+
   const handleContinueAsUser = () => {
+    persistReferrer()
+
     let callbackUrl = '/'
 
     if (authCode) {
@@ -30,6 +39,8 @@ export function LoginCard({ authCode }: { authCode?: string | null }) {
   }
 
   const handleUseAnotherAccount = () => {
+    persistReferrer()
+
     const searchParamsString = searchParams.toString()
 
     let callbackUrl = '/login'

freebuff/web/src/components/sign-in/sign-in-button.tsx

@@ -26,6 +26,11 @@ export function SignInButton({
       let callbackUrl =
         pathname + (searchParamsString ? `?${searchParamsString}` : '')
 
+      const referrer = searchParams.get('referrer')
+      if (referrer) {
+        localStorage.setItem('freebuff_referrer', referrer)
+      }
+
       if (pathname === '/login') {
         const authCode = searchParams.get('auth_code')
 

web/src/app/[sponsee]/page.tsx

@@ -0,0 +1,87 @@
+'use server'
+
+import { env } from '@codebuff/common/env'
+import db from '@codebuff/internal/db'
+import * as schema from '@codebuff/internal/db/schema'
+import { eq } from 'drizzle-orm'
+import Link from 'next/link'
+import { redirect } from 'next/navigation'
+
+import type { Metadata } from 'next'
+
+import CardWithBeams from '@/components/card-with-beams'
+
+export const generateMetadata = async ({
+  params,
+}: {
+  params: Promise<{ sponsee: string }>
+}): Promise<Metadata> => {
+  const { sponsee } = await params
+  return {
+    title: `${sponsee}'s Referral | Codebuff`,
+  }
+}
+
+export default async function SponseePage({
+  params,
+  searchParams,
+}: {
+  params: Promise<{ sponsee: string }>
+  searchParams: Promise<Record<string, string | string[] | undefined>>
+}) {
+  const { sponsee } = await params
+  const resolvedSearchParams = await searchParams
+  const sponseeName = sponsee.toLowerCase()
+
+  const referralCode = await db
+    .select({
+      referralCode: schema.user.referral_code,
+    })
+    .from(schema.user)
+    .where(eq(schema.user.handle, sponseeName))
+    .limit(1)
+    .then((result) => result[0]?.referralCode ?? null)
+
+  if (!referralCode) {
+    return (
+      <CardWithBeams
+        title="Hmm, that link doesn't look right."
+        description={`We don't have a referral code for "${sponsee}".`}
+        content={
+          <>
+            <p className="text-center">
+              Please double-check the link you used or try contacting the person
+              who shared it.
+            </p>
+            <p className="text-center text-sm text-muted-foreground">
+              You can also reach out to our support team at{' '}
+              <Link
+                href={`mailto:${env.NEXT_PUBLIC_SUPPORT_EMAIL}`}
+                className="underline"
+              >
+                {env.NEXT_PUBLIC_SUPPORT_EMAIL}
+              </Link>
+              .
+            </p>
+          </>
+        }
+      />
+    )
+  }
+
+  const queryParams = new URLSearchParams()
+  for (const [key, value] of Object.entries(resolvedSearchParams)) {
+    if (value !== undefined) {
+      if (Array.isArray(value)) {
+        for (const v of value) {
+          queryParams.append(key, v)
+        }
+      } else {
+        queryParams.set(key, value)
+      }
+    }
+  }
+  queryParams.set('referrer', sponseeName)
+
+  redirect(`/referrals/${referralCode}?${queryParams.toString()}`)
+}

web/src/app/api/referrals/[code]/route.ts

@@ -0,0 +1,37 @@
+import db from '@codebuff/internal/db'
+import * as schema from '@codebuff/internal/db/schema'
+import { eq } from 'drizzle-orm'
+import { NextResponse } from 'next/server'
+
+export type ReferralCodeResponse = {
+  referrerName: string | null
+}
+
+export async function GET(
+  _req: Request,
+  { params }: { params: Promise<{ code: string }> },
+): Promise<NextResponse<ReferralCodeResponse | { error: string }>> {
+  const { code } = await params
+
+  try {
+    const user = await db.query.user.findFirst({
+      where: eq(schema.user.referral_code, code),
+      columns: { name: true },
+    })
+
+    if (!user) {
+      return NextResponse.json(
+        { error: 'Invalid referral code' },
+        { status: 400 },
+      )
+    }
+
+    return NextResponse.json({ referrerName: user.name })
+  } catch (error) {
+    console.error(error)
+    return NextResponse.json(
+      { error: 'Internal Server Error' },
+      { status: 500 },
+    )
+  }
+}

web/src/app/referrals/[code]/page.tsx

@@ -0,0 +1,87 @@
+import { env } from '@codebuff/common/env'
+import { headers } from 'next/headers'
+import Link from 'next/link'
+
+import type { ReferralCodeResponse } from '../../api/referrals/[code]/route'
+import type { Metadata } from 'next'
+
+import CardWithBeams from '@/components/card-with-beams'
+import { Button } from '@/components/ui/button'
+import { InstallInstructions } from '@/components/ui/install-instructions'
+
+export const generateMetadata = async ({
+  searchParams,
+}: {
+  params: Promise<{ code: string }>
+  searchParams: Promise<{ referrer?: string }>
+}): Promise<Metadata> => {
+  const resolvedSearchParams = await searchParams
+  const referrerName = resolvedSearchParams.referrer
+  const title = referrerName
+    ? `${referrerName} invited you to Codebuff!`
+    : 'You were invited to Codebuff!'
+
+  return {
+    title,
+    description: 'Install Codebuff and start building with AI in your terminal.',
+  }
+}
+
+export default async function ReferralPage({
+  params,
+  searchParams,
+}: {
+  params: Promise<{ code: string }>
+  searchParams: Promise<{ referrer?: string }>
+}) {
+  const { code } = await params
+  const resolvedSearchParams = await searchParams
+  const referrerParam = resolvedSearchParams.referrer
+
+  let referrerName: string | null = null
+  try {
+    const baseUrl = env.NEXT_PUBLIC_CODEBUFF_APP_URL || 'http://localhost:3000'
+    const headerList = await headers()
+    const cookie = headerList.get('Cookie') ?? ''
+    const response = await fetch(`${baseUrl}/api/referrals/${code}`, {
+      headers: { Cookie: cookie },
+    })
+
+    if (!response.ok) {
+      throw new Error('Failed to fetch referral data')
+    }
+
+    const referralData: ReferralCodeResponse = await response.json()
+    referrerName = referralData.referrerName
+  } catch {
+    return (
+      <CardWithBeams
+        title="Invalid Referral Link"
+        description="This referral link is not valid or has expired."
+        content={
+          <>
+            <p className="text-center text-muted-foreground">
+              Please double-check the link you used or contact the person who
+              shared it.
+            </p>
+            <div className="flex justify-center mt-4">
+              <Button asChild>
+                <Link href="/">Go to Homepage</Link>
+              </Button>
+            </div>
+          </>
+        }
+      />
+    )
+  }
+
+  const displayName = referrerName || referrerParam || 'Someone'
+
+  return (
+    <CardWithBeams
+      title={`${displayName} invited you to Codebuff!`}
+      description="Install Codebuff and start building with AI in your terminal."
+      content={<InstallInstructions />}
+    />
+  )
+}