huhuih

hzgjuigik

client/src/pages/Register.tsx

@@ -3,6 +3,7 @@ import { Link, useNavigate, useSearchParams } from 'react-router-dom'
 import { useAuth } from '@/context/AuthContext'
 import { analytics } from '@/hooks/useAnalytics'
 import { captureUTMParams } from '@/lib/analytics'
+import { api } from '@/lib/api'
 import { Button } from '@/components/ui/button'
 import { Input } from '@/components/ui/input'
 import { Label } from '@/components/ui/label'
@@ -12,6 +13,7 @@ import { Mail, Lock, User, ArrowRight, AlertCircle, Check, Sparkles } from 'luci
 export function Register() {
   const [searchParams] = useSearchParams()
   const selectedPlan = searchParams.get('plan') || 'pro'
+  const referralCode = searchParams.get('ref') || null
   
   const [name, setName] = useState('')
   const [email, setEmail] = useState('')
@@ -20,7 +22,7 @@ export function Register() {
   const [error, setError] = useState('')
   const [loading, setLoading] = useState(false)
   
-  const { register } = useAuth()
+  const { register, user } = useAuth()
   const navigate = useNavigate()
 
   // Capture UTM parameters on mount
@@ -28,6 +30,22 @@ export function Register() {
     captureUTMParams()
   }, [])
 
+  // Track referral and signup after user is registered
+  useEffect(() => {
+    if (user?.$id && referralCode) {
+      // Track referral if code exists
+      api.trackReferral(user.$id, referralCode).catch((err) => {
+        console.error('Failed to track referral:', err)
+      })
+    }
+    
+    if (user?.$id) {
+      // Track signup conversion with UTM parameters
+      analytics.trackSignup(user.$id, email)
+      analytics.setUserId(user.$id)
+    }
+  }, [user, referralCode, email])
+
   const handleSubmit = async (e: React.FormEvent) => {
     e.preventDefault()
     setError('')
@@ -45,14 +63,7 @@ export function Register() {
     setLoading(true)
 
     try {
-      const user = await register(email, password, name)
-      
-      // Track signup conversion with UTM parameters
-      if (user?.$id) {
-        analytics.trackSignup(user.$id, email)
-        analytics.setUserId(user.$id)
-      }
-      
+      await register(email, password, name)
       navigate('/setup')
     } catch (err: any) {
       setError(err.message || 'Registration failed. Please try again.')
@@ -111,7 +122,7 @@ export function Register() {
               <Mail className="w-5 h-5 text-white" />
             </div>
             <span className="text-xl font-bold text-slate-900">
-              Email<span className="text-primary-600">Sorter</span>
+              E-Mail-<span className="text-primary-600">Sorter</span>
             </span>
           </Link>