diff --git a/apps/user/src/sections/auth/email/auth-form.tsx b/apps/user/src/sections/auth/email/auth-form.tsx index 716a531..e796104 100644 --- a/apps/user/src/sections/auth/email/auth-form.tsx +++ b/apps/user/src/sections/auth/email/auth-form.tsx @@ -1,13 +1,13 @@ "use client"; -import { useNavigate } from "@tanstack/react-router"; +import { useNavigate, useSearch } from "@tanstack/react-router"; import { resetPassword, userLogin, userRegister, } from "@workspace/ui/services/common/auth"; import type { ReactNode } from "react"; -import { useState, useTransition } from "react"; +import { useEffect, useState, useTransition } from "react"; import { useTranslation } from "react-i18next"; import { toast } from "sonner"; import { USER_EMAIL, USER_PASSWORD } from "@/config"; @@ -21,16 +21,25 @@ export default function EmailAuthForm() { const { t } = useTranslation("auth"); const navigate = useNavigate(); const { getUserInfo } = useGlobalStore(); + const searchParams = useSearch({ strict: false }) as { invite?: string }; const [type, setType] = useState<"login" | "register" | "reset">("login"); const [loading, startTransition] = useTransition(); const [initialValues, setInitialValues] = useState<{ email?: string; password?: string; + invite?: string; }>({ email: USER_EMAIL, password: USER_PASSWORD, }); + useEffect(() => { + if (searchParams.invite) { + localStorage.setItem("invite", searchParams.invite); + setInitialValues((prev) => ({ ...prev, invite: searchParams.invite })); + } + }, [searchParams.invite]); + const handleFormSubmit = async (params: any) => { const onLogin = async (token?: string) => { if (!token) return; diff --git a/apps/user/src/sections/auth/phone/auth-form.tsx b/apps/user/src/sections/auth/phone/auth-form.tsx index d24bfef..a6679ea 100644 --- a/apps/user/src/sections/auth/phone/auth-form.tsx +++ b/apps/user/src/sections/auth/phone/auth-form.tsx @@ -1,13 +1,13 @@ "use client"; -import { useNavigate } from "@tanstack/react-router"; +import { useNavigate, useSearch } from "@tanstack/react-router"; import { telephoneLogin, telephoneResetPassword, telephoneUserRegister, } from "@workspace/ui/services/common/auth"; import type { ReactNode } from "react"; -import { useState, useTransition } from "react"; +import { useEffect, useState, useTransition } from "react"; import { useTranslation } from "react-i18next"; import { toast } from "sonner"; import { useGlobalStore } from "@/stores/global"; @@ -20,6 +20,7 @@ export default function PhoneAuthForm() { const { t } = useTranslation("auth"); const navigate = useNavigate(); const { getUserInfo } = useGlobalStore(); + const searchParams = useSearch({ strict: false }) as { invite?: string }; const [type, setType] = useState<"login" | "register" | "reset">("login"); const [loading, startTransition] = useTransition(); const [initialValues, setInitialValues] = useState( @@ -32,6 +33,13 @@ export default function PhoneAuthForm() { } ); + useEffect(() => { + if (searchParams.invite) { + localStorage.setItem("invite", searchParams.invite); + setInitialValues((prev) => ({ ...prev, invite: searchParams.invite })); + } + }, [searchParams.invite]); + const handleFormSubmit = async (params: any) => { const onLogin = async (token?: string) => { if (!token) return; diff --git a/apps/user/src/sections/user/affiliate/index.tsx b/apps/user/src/sections/user/affiliate/index.tsx index 23adbcf..1131f96 100644 --- a/apps/user/src/sections/user/affiliate/index.tsx +++ b/apps/user/src/sections/user/affiliate/index.tsx @@ -72,7 +72,7 @@ export default function Affiliate() { toast.success(t("copySuccess", "Copy Success")); } }} - text={`${location?.origin}/auth?invite=${user?.refer_code}`} + text={`${location?.origin}/#/auth?invite=${user?.refer_code}`} >