Add set auth on authenticated

2 files changed, 12 insertions, 2 deletions

diff --git a/src-migrate/modules/account-activation/components/FormOTP.tsx b/src-migrate/modules/account-activation/components/FormOTP.tsx
index e38fa8fe..a4775d61 100644
--- a/src-migrate/modules/account-activation/components/FormOTP.tsx
+++ b/src-migrate/modules/account-activation/components/FormOTP.tsx
@@ -5,6 +5,7 @@ import { useEffect, useState } from "react"
 import { useMutation } from "react-query"
 import { useCountdown } from "usehooks-ts"
 import Modal from '~/common/components/elements/Modal'
+import { setAuth } from "~/common/libs/auth"
 import { ActivationOtpProps, ActivationReqProps } from "~/common/types/auth"
 import { activationReq, activationUserOTP } from "~/services/auth"
 
@@ -50,7 +51,11 @@ const FormOTP = () => {
     //eslint-disable-next-line
   }, [otp])
 
-  // TODO: Save user to local storage
+  useEffect(() => {
+    if (mutationActivation.data?.user) {
+      setAuth(mutationActivation.data.user)
+    }
+  }, [mutationActivation.data])
 
   return (
     <Modal active={active} className="w-10/12 md:w-fit px-10" mode="desktop">
diff --git a/src-migrate/modules/account-activation/components/FormToken.tsx b/src-migrate/modules/account-activation/components/FormToken.tsx
index 6af24413..a1525fe6 100644
--- a/src-migrate/modules/account-activation/components/FormToken.tsx
+++ b/src-migrate/modules/account-activation/components/FormToken.tsx
@@ -7,6 +7,7 @@ import { useMutation } from "react-query"
 import Modal from "~/common/components/elements/Modal"
 import { ActivationTokenProps } from "~/common/types/auth"
 import { activationUserToken } from "~/services/auth"
+import { setAuth } from "~/common/libs/auth"
 
 const FormToken = () => {
   const { query } = useRouter()
@@ -24,7 +25,11 @@ const FormToken = () => {
     //eslint-disable-next-line
   }, [query.activation, query.token])
 
-  // TODO: Save user to local storage
+  useEffect(() => {
+    if (mutation.data?.user) {
+      setAuth(mutation.data.user)
+    }
+  }, [mutation.data])
 
   return (
     <Modal active={active} mode="desktop">