feat: implement post-authentication redirect via redirect_after_login parameter

src/app/pages/Router.tsx

@@ -87,10 +87,15 @@ export const createRouter = (clientConfig: ClientConfig, screenSize: ScreenSize)
         }}
       />
       <Route
-        loader={() => {
+        loader={({ request }) => {
           if (getFallbackSession()) {
             return redirect(getHomePath());
           }
+          const url = new URL(request.url);
+          const redirectAfterLogin = url.searchParams.get('redirect_after_login');
+          if (redirectAfterLogin) {
+            setAfterLoginRedirectPath(redirectAfterLogin);
+          }
 
           return null;
         }}

src/app/pages/auth/login/Login.tsx

@@ -13,6 +13,7 @@ import { getLoginPath, getRegisterPath, withSearchParam } from '../../pathUtils'
 import { usePathWithOrigin } from '../../../hooks/usePathWithOrigin';
 import { LoginPathSearchParams } from '../../paths';
 import { useClientConfig } from '../../../hooks/useClientConfig';
+import { setAfterLoginRedirectPath } from '../../afterLoginRedirectPath';
 
 const getLoginTokenSearchParam = () => {
   // when using hasRouter query params in existing route
@@ -30,6 +31,7 @@ const useLoginSearchParams = (searchParams: URLSearchParams): LoginPathSearchPar
       username: searchParams.get('username') ?? undefined,
       email: searchParams.get('email') ?? undefined,
       loginToken: searchParams.get('loginToken') ?? undefined,
+      redirect_after_login: searchParams.get('redirect_after_login') ?? undefined,
     }),
     [searchParams]
   );
@@ -54,6 +56,18 @@ export function Login() {
 
   const parsedFlows = useParsedLoginFlows(loginFlows.flows);
 
+  React.useEffect(() => {
+    if (loginSearchParams.redirect_after_login) {
+      setAfterLoginRedirectPath(loginSearchParams.redirect_after_login);
+    }
+  }, [loginSearchParams.redirect_after_login]);
+
+  const registerPath = loginSearchParams.redirect_after_login
+    ? withSearchParam(getRegisterPath(server), {
+        redirect_after_login: loginSearchParams.redirect_after_login,
+      })
+    : getRegisterPath(server);
+
   return (
     <Box direction="Column" gap="500">
       <Text size="H2" priority="400">
@@ -92,7 +106,7 @@ export function Login() {
         </>
       )}
       <Text align="Center">
-        Do not have an account? <Link to={getRegisterPath(server)}>Register</Link>
+        Do not have an account? <Link to={registerPath}>Register</Link>
       </Text>
     </Box>
   );

src/app/pages/auth/register/Register.tsx

@@ -12,13 +12,16 @@ import { SupportedUIAFlowsLoader } from '../../../components/SupportedUIAFlowsLo
 import { getLoginPath } from '../../pathUtils';
 import { usePathWithOrigin } from '../../../hooks/usePathWithOrigin';
 import { RegisterPathSearchParams } from '../../paths';
+import { setAfterLoginRedirectPath } from '../../afterLoginRedirectPath';
+import { withSearchParam } from '../../pathUtils';
 
 const useRegisterSearchParams = (searchParams: URLSearchParams): RegisterPathSearchParams =>
   useMemo(
     () => ({
       username: searchParams.get('username') ?? undefined,
       email: searchParams.get('email') ?? undefined,
       token: searchParams.get('token') ?? undefined,
+      redirect_after_login: searchParams.get('redirect_after_login') ?? undefined,
     }),
     [searchParams]
   );
@@ -33,6 +36,18 @@ export function Register() {
   // redirect to /login because only that path handle m.login.token
   const ssoRedirectUrl = usePathWithOrigin(getLoginPath(server));
 
+  React.useEffect(() => {
+    if (registerSearchParams.redirect_after_login) {
+      setAfterLoginRedirectPath(registerSearchParams.redirect_after_login);
+    }
+  }, [registerSearchParams.redirect_after_login]);
+
+  const loginPath = registerSearchParams.redirect_after_login
+    ? withSearchParam(getLoginPath(server), {
+        redirect_after_login: registerSearchParams.redirect_after_login,
+      })
+    : getLoginPath(server);
+
   return (
     <Box direction="Column" gap="500">
       <Text size="H2" priority="400">
@@ -91,7 +106,7 @@ export function Register() {
         </>
       )}
       <Text align="Center">
-        Already have an account? <Link to={getLoginPath(server)}>Login</Link>
+        Already have an account? <Link to={loginPath}>Login</Link>
       </Text>
     </Box>
   );

src/app/pages/paths.ts

@@ -4,13 +4,15 @@ export type LoginPathSearchParams = {
   username?: string;
   email?: string;
   loginToken?: string;
+  redirect_after_login?: string;
 };
 export const LOGIN_PATH = '/login/:server?/';
 
 export type RegisterPathSearchParams = {
   username?: string;
   email?: string;
   token?: string;
+  redirect_after_login?: string;
 };
 export const REGISTER_PATH = '/register/:server?/';