fix(billing): keep the upgrade intent through workspace creation

A first-time visitor to /upgrade has no workspace to resolve, so /workspace
creates one — and then hardcoded a redirect to home, silently dropping the
upgrade intent. Route both exits through one destination helper so the created
workspace lands on the plan picker with its reason intact.

Author: TheodoreSpeaks

apps/sim/app/workspace/page.tsx

@@ -85,11 +85,21 @@ export default function WorkspacePage() {
     const urlParams = new URLSearchParams(window.location.search)
     const redirectWorkflowId = urlParams.get('redirect_workflow')
     const redirectTarget = urlParams.get('redirect')
+    const rawReason = urlParams.get(UPGRADE_REASON_PARAM)
+
+    // `?redirect=upgrade` is how a caller that cannot know a workspace id — a
+    // self-hosted deployment, an email — reaches the plan picker. It has to
+    // survive workspace creation too: a first-time visitor has no workspace to
+    // resolve, and dropping the intent lands them on home with no explanation.
+    const destinationFor = (id: string) =>
+      redirectTarget === 'upgrade'
+        ? buildUpgradeHref(id, isUpgradeReason(rawReason) ? rawReason : undefined)
+        : `/workspace/${id}/home`
 
     const { workspaces, lastActiveWorkspaceId, creationPolicy } = data
 
     if (workspaces.length === 0) {
-      handleNoWorkspaces(router, creationPolicy)
+      handleNoWorkspaces(router, creationPolicy, destinationFor)
       return
     }
 
@@ -105,21 +115,8 @@ export default function WorkspacePage() {
       return
     }
 
-    // `?redirect=upgrade` is how a caller that cannot know a workspace id — a
-    // self-hosted deployment, an email — reaches the plan picker.
-    if (redirectTarget === 'upgrade') {
-      const rawReason = urlParams.get(UPGRADE_REASON_PARAM)
-      const href = buildUpgradeHref(
-        targetWorkspace.id,
-        isUpgradeReason(rawReason) ? rawReason : undefined
-      )
-      logger.info(`Redirecting to upgrade: ${targetWorkspace.id}`)
-      router.replace(href)
-      return
-    }
-
     logger.info(`Redirecting to workspace: ${targetWorkspace.id}`)
-    router.replace(`/workspace/${targetWorkspace.id}/home`)
+    router.replace(destinationFor(targetWorkspace.id))
   }, [session, isSessionPending, sessionError, isWorkspacesLoading, workspacesError, data, router])
 
   const failedToLoad =
@@ -193,7 +190,8 @@ async function handleWorkflowRedirect(
 
 async function handleNoWorkspaces(
   router: ReturnType<typeof useRouter>,
-  creationPolicy: WorkspaceCreationPolicy | null
+  creationPolicy: WorkspaceCreationPolicy | null,
+  destinationFor: (workspaceId: string) => string
 ): Promise<void> {
   if (creationPolicy && !creationPolicy.canCreate) {
     logger.warn('No workspaces found and workspace creation is blocked', {
@@ -212,7 +210,7 @@ async function handleNoWorkspaces(
     })
     if (data.workspace?.id) {
       logger.info(`Created default workspace: ${data.workspace.id}`)
-      router.replace(`/workspace/${data.workspace.id}/home`)
+      router.replace(destinationFor(data.workspace.id))
       return
     }
     logger.error('Failed to create default workspace')