role fit index referrals

Author: kanekoshoyu

src/components/interactive/ReferralHandler.tsx

@@ -6,56 +6,106 @@ interface ReferralHandlerProps {
   children?: React.ReactNode;
 }
 
-export default function ReferralHandler({ children }: ReferralHandlerProps) {
-  useEffect(() => {
-    async function handleReferralCode() {
-      try {
-        // Check for ref parameter in URL
-        const urlParams = new URLSearchParams(window.location.search);
-        const refCode = urlParams.get('ref');
+async function storeReferralCodeFromUrl(): Promise<void> {
+  const urlParams = new URLSearchParams(window.location.search);
+  const refCode = urlParams.get('ref');
+  
+  if (refCode) {
+    localStorage.setItem('ref_code', refCode);
+  }
+}
 
-        if (refCode) {
-          // Store referral code in localStorage
-          localStorage.setItem('ref_code', refCode);
-        }
+async function checkIfUserAlreadyClaimed(refCode: string, userId: string): Promise<boolean> {
+  const checkUrl = `${EXTERNAL.directus_url}/items/referral_code?filter[code][_eq]=${encodeURIComponent(refCode)}&filter[user_claimed][directus_users_id][_eq]=${userId}&limit=1`;
+  
+  const response = await fetch(checkUrl, {
+    credentials: 'include',
+    headers: { 'Accept': 'application/json' }
+  });
 
+  if (!response.ok) {
+    throw new Error('Failed to check existing claims');
+  }
 
-        // Check if user is logged in and handle claimed referral codes
-        const user = await getUserProfile(EXTERNAL.directus_url);
+  const result = await response.json();
+  return result.data && result.data.length > 0;
+}
+
+async function getReferralCodeDetails(refCode: string): Promise<any> {
+  const url = `${EXTERNAL.directus_url}/items/referral_code?filter[code][_eq]=${encodeURIComponent(refCode)}&limit=1`;
+  
+  const response = await fetch(url, {
+    credentials: 'include',
+    headers: { 'Accept': 'application/json' }
+  });
+
+  if (!response.ok) {
+    throw new Error('Failed to get referral code details');
+  }
 
-        if (user) {
-          const storedRefCode = localStorage.getItem('ref_code');
+  const result = await response.json();
+  return result.data && result.data.length > 0 ? result.data[0] : null;
+}
 
-          if (storedRefCode) {
-            console.log('Referral code handling initiated.');
+async function claimReferralCode(referralCodeId: string, userId: string): Promise<void> {
+  const url = `${EXTERNAL.directus_url}/items/referral_code_directus_users`;
+  
+  const response = await fetch(url, {
+    method: 'POST',
+    credentials: 'include',
+    headers: {
+      'Content-Type': 'application/json',
+      'Accept': 'application/json'
+    },
+    body: JSON.stringify({
+      referral_code_id: referralCodeId,
+      directus_users_id: userId
+    })
+  });
 
-            // get data if currently passed ref code
-            const checkUrl = `${EXTERNAL.directus_url}/items/referral_code?filter[code][_eq]=${encodeURIComponent(storedRefCode)}&fields=*,user_claimed&limit=1`;
+  if (!response.ok) {
+    const errorData = await response.json().catch(() => null);
+    throw new Error(`Failed to claim referral code: ${response.status} ${response.statusText} ${JSON.stringify(errorData)}`);
+  }
+}
 
-            const response = await fetch(checkUrl, {
-              credentials: 'include',
-              headers: { 'Accept': 'application/json' }
-            });
+async function processReferralCode(refCode: string, userId: string): Promise<void> {
+  const alreadyClaimed = await checkIfUserAlreadyClaimed(refCode, userId);
+  
+  if (alreadyClaimed) {
+    console.log('user already claimed ref code before');
+    localStorage.removeItem('ref_code');
+    return;
+  }
 
-            if (response.ok) {
-              const result = await response.json();
-              console.log(result);
+  const referralCode = await getReferralCodeDetails(refCode);
+  
+  if (!referralCode) {
+    console.log('referral code not found');
+    localStorage.removeItem('ref_code');
+    return;
+  }
 
-              // Check if the current user is in the user_claimed many-to-many relationship
-              if (result.data && result.data.length > 0) {
-                const referralCode = result.data[0];
-                const isAlreadyClaimed = referralCode.user_claimed?.some((userId: string) => userId === user.id);
+  await claimReferralCode(referralCode.id, userId);
+  console.log('User successfully added to referral code claims');
+  localStorage.removeItem('ref_code');
+}
+
+export default function ReferralHandler({ children }: ReferralHandlerProps) {
+  useEffect(() => {
+    async function handleReferralCode() {
+      try {
+        await storeReferralCodeFromUrl();
+        
+        const user = await getUserProfile(EXTERNAL.directus_url);
+        if (!user) return;
 
-                if (isAlreadyClaimed) {
-                  console.log('user already claimed ref code before');
+        const storedRefCode = localStorage.getItem('ref_code');
+        if (!storedRefCode) return;
 
-                  localStorage.removeItem('ref_code');
-                }
-                // TODO
-              }
-            }
-          }
-        }
+        console.log('Referral code handling initiated.');
+        await processReferralCode(storedRefCode, user.id);
+        
       } catch (error) {
         console.error('Error handling referral code:', error);
       }
@@ -64,6 +114,5 @@ export default function ReferralHandler({ children }: ReferralHandlerProps) {
     handleReferralCode();
   }, []);
 
-  // Just render children if provided, otherwise render nothing
   return children ? <>{children}</> : null;
 }