hotfix: move setHasLoginStateFlag (#202)

* hotfix: move setHasLoginStateFlag

* use async await

src/features/sign-in/components/SgidCallback.tsx

@@ -1,12 +1,15 @@
 import { useRouter } from 'next/router'
 import { FullscreenSpinner } from '~/components/FullscreenSpinner'
+import { useLoginState } from '~/features/auth'
 import { trpc } from '~/utils/trpc'
 
 /**
  * This component is responsible for handling the callback from the SGID login.
  */
 export const SgidCallback = () => {
+  const utils = trpc.useContext()
   const router = useRouter()
+  const { setHasLoginStateFlag } = useLoginState()
   const {
     query: { code, state },
   } = router
@@ -18,12 +21,13 @@ export const SgidCallback = () => {
     },
     {
       onSuccess: async ({ redirectUrl }) => {
+        setHasLoginStateFlag()
+        await utils.me.get.invalidate()
         await router.replace(redirectUrl)
       },
       onError: async (error) => {
-        // Server should return redirectUrl even on error, this function is a fallback.
         console.error(error)
-        await router.replace('/sign-in')
+        await router.replace(`/sign-in?error=${error.message}`)
       },
     }
   )

src/features/sign-in/components/SgidLoginButton.tsx

@@ -3,17 +3,12 @@ import { useRouter } from 'next/router'
 import { useFeatures } from '~/components/AppProviders'
 import { trpc } from '~/utils/trpc'
 import { CALLBACK_URL_KEY } from '~/constants/params'
-import { useLoginState } from '~/features/auth'
 
 export const SgidLoginButton = (): JSX.Element | null => {
-  const { setHasLoginStateFlag } = useLoginState()
-  const utils = trpc.useContext()
   const router = useRouter()
   const { sgid } = useFeatures()
   const sgidLoginMutation = trpc.auth.sgid.login.useMutation({
     onSuccess: async ({ redirectUrl }) => {
-      await utils.me.get.invalidate()
-      setHasLoginStateFlag()
       await router.push(redirectUrl)
     },
   })

src/server/modules/auth/sgid/sgid.router.ts

@@ -79,20 +79,23 @@ export const sgidRouter = router({
     )
     .query(async ({ ctx, input: { state, code } }) => {
       if (!env.NEXT_PUBLIC_ENABLE_SGID) {
-        return {
-          redirectUrl: `/sign-in?error=${'SGID is not enabled'}`,
-        }
+        throw new TRPCError({
+          code: 'BAD_REQUEST',
+          message: 'SGID is not enabled',
+        })
       }
       if (!ctx.session?.sgidSessionState) {
-        return {
-          redirectUrl: `/sign-in?error=${'Invalid login flow'}`,
-        }
+        throw new TRPCError({
+          code: 'BAD_REQUEST',
+          message: 'Invalid login flow',
+        })
       }
       const parsedState = sgidCallbackStateSchema.safeParse(state)
       if (!parsedState.success) {
-        return {
-          redirectUrl: `/sign-in?error=${'Invalid SGID state'}`,
-        }
+        throw new TRPCError({
+          code: 'BAD_REQUEST',
+          message: 'Invalid SGID callback state',
+        })
       }
 
       const { codeVerifier, nonce } = ctx.session.sgidSessionState