From e2ed0399f03c8072deebb28e38805a587b82f3c5 Mon Sep 17 00:00:00 2001
From: Buer <42402987+MartialBE@users.noreply.github.com>
Date: Sat, 20 Jan 2024 12:38:06 +0800
Subject: [PATCH] fix: fix aff not effective (#937)

---
 web/berry/src/hooks/useRegister.js | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/web/berry/src/hooks/useRegister.js b/web/berry/src/hooks/useRegister.js
index d07dc43a..5377e96d 100644
--- a/web/berry/src/hooks/useRegister.js
+++ b/web/berry/src/hooks/useRegister.js
@@ -6,6 +6,10 @@ const useRegister = () => {
   const navigate = useNavigate();
   const register = async (input, turnstile) => {
     try {
+      let affCode = localStorage.getItem('aff');
+      if (affCode) {
+        input = { ...input, aff_code: affCode };
+      }
       const res = await API.post(`/api/user/register?turnstile=${turnstile}`, input);
       const { success, message } = res.data;
       if (success) {