Fix missing survey

Author: jp-agenta

web/oss/src/hooks/usePostAuthRedirect.ts

@@ -16,6 +16,7 @@ import {userAtom} from "@/oss/state/profile/selectors/user"
 import {useProjectData} from "@/oss/state/project"
 import {authFlowAtom} from "@/oss/state/session"
 import {buildPostLoginPath, waitForWorkspaceContext} from "@/oss/state/url/postLoginRedirect"
+import {writePostSignupPending} from "@/oss/state/url/auth"
 
 interface AuthUserLike {
     createdNewRecipeUser?: boolean
@@ -37,6 +38,7 @@ const usePostAuthRedirect = () => {
     const [invite] = useLocalStorage<Record<string, unknown>>("invite", {})
     const authUpgradeOrgKey = "authUpgradeOrgId"
     const lastSsoOrgSlugKey = "lastSsoOrgSlug"
+    const postSignupPendingKey = "postSignupPending"
 
     const hasInviteFromQuery = useMemo(() => {
         const token = router.query?.token
@@ -79,6 +81,7 @@ const usePostAuthRedirect = () => {
                     await router.push("/workspaces/accept?survey=true")
                 } else {
                     console.log("[post-auth] redirect new user -> /post-signup")
+                    writePostSignupPending()
                     await resetAuthState()
                     await router.push("/post-signup")
                 }

web/oss/src/state/url/auth.ts

@@ -27,6 +27,8 @@ export interface InvitePayload {
 }
 
 const INVITE_STORAGE_KEY = "invite"
+const POST_SIGNUP_PENDING_KEY = "postSignupPending"
+const POST_SIGNUP_TTL_MS = 2 * 60 * 1000
 let authOrgFetchInFlight = false
 
 export const protectedRouteReadyAtom = atom(false)
@@ -99,6 +101,41 @@ export const isCurrentAcceptRouteForInvite = (appState: any, invite: InvitePaylo
     return currentToken === invite.token
 }
 
+export const writePostSignupPending = () => {
+    if (!isBrowser) return
+    try {
+        window.sessionStorage.setItem(
+            POST_SIGNUP_PENDING_KEY,
+            JSON.stringify({ts: Date.now()}),
+        )
+    } catch {
+        // ignore storage failures
+    }
+}
+
+const readPostSignupPending = (): boolean => {
+    if (!isBrowser) return false
+    try {
+        const raw = window.sessionStorage.getItem(POST_SIGNUP_PENDING_KEY)
+        if (!raw) return false
+        const parsed = JSON.parse(raw) as {ts?: number}
+        if (!parsed?.ts) return false
+        if (Date.now() - parsed.ts > POST_SIGNUP_TTL_MS) return false
+        return true
+    } catch {
+        return false
+    }
+}
+
+const clearPostSignupPending = () => {
+    if (!isBrowser) return
+    try {
+        window.sessionStorage.removeItem(POST_SIGNUP_PENDING_KEY)
+    } catch {
+        // ignore storage failures
+    }
+}
+
 export const syncAuthStateFromUrl = (nextUrl?: string) => {
     if (!isBrowser) return
 
@@ -148,11 +185,22 @@ export const syncAuthStateFromUrl = (nextUrl?: string) => {
                     asPath,
                     baseAppURL,
                 })
+                clearPostSignupPending()
             }
             if (isAuthCallbackRoute) {
                 store.set(protectedRouteReadyAtom, false)
                 return
             }
+            if (isAuthRoute) {
+                if (readPostSignupPending()) {
+                    clearPostSignupPending()
+                    void Router.replace("/post-signup").catch((error) => {
+                        console.error("Failed to redirect to post-signup:", error)
+                    })
+                    store.set(protectedRouteReadyAtom, false)
+                    return
+                }
+            }
             if (typeof window !== "undefined") {
                 const upgradeOrgId = window.localStorage.getItem("authUpgradeOrgId")
                 const identifiers = store.get(appIdentifiersAtom)