@taskforcehq/taskforce 0.3.303 → 0.3.305

package/dist/components/views/PlansPage.js

@@ -7,18 +7,15 @@ import { PlanComparisonPage } from './PlanComparisonPage';
 import { AppShellHeader } from './AppShellHeader';
 import { resolveTaskforceCloudEnvironmentFromBaseUrl } from '../../config/cloudEnvironment';
 import { ACCOUNT_PROFILE_SUMMARY_PATH, invalidateAccountProfileSummaryCache, loadAccountProfileSummaryCached } from '../../utils/accountProfileSummaryCache';
+import { isCommercialEntitlementActive, resolveCommercialLifecyclePresentation } from '../../utils/commercialLifecyclePresentation';
 const TRANSIENT_CHECKOUT_QUERY_CLEAR_DELAY_MS = 1600;
-const ACTIVE_ENTITLEMENT_STATES = new Set(['active', 'trialing', 'grace']);
-const GATE_MESSAGES = {
-    planSelectionRequired: 'Choose a plan to continue.',
-    checkoutPending: 'Complete checkout to continue.',
-    misconfigured: 'Onboarding policy is misconfigured. Contact support or a system administrator.',
-    missingEntitlement: 'No entitlement is linked to this account.'
-};
-const BILLING_CONFLICT_MESSAGES = {
-    CHECKOUT_SESSION_EXPIRED: 'Your previous checkout expired. Choose a plan to continue.',
-    CHECKOUT_SESSION_SUPERSEDED: 'A newer checkout replaced your previous checkout. Choose a plan to continue.'
-};
+const CHECKOUT_CONFIRM_RETRY_DELAYS_MS = [0, 500, 1000, 2000];
+function shouldRetryCheckoutConfirmation(statusCode, errorCode) {
+    if (statusCode === 401) {
+        return true;
+    }
+    return String(errorCode || '').trim().toUpperCase() === 'CHECKOUT_SESSION_INCOMPLETE';
+}
 function parseInterval(raw) {
     return raw === 'year' ? 'year' : 'month';
 }
@@ -38,13 +35,14 @@ function buildEmbeddedPlansPath(params = {}) {
     }
     return `/?${query.toString()}`;
 }
-export function PlansPage({ isAuthenticated, authUserId, runtimeMode, authSessionResolved, currentTheme = 'dark', projectName, currentWorkspaceId, apiBaseUrl, connectedEnvironmentSource, resolveCloudAuthUrl, embedded = false, shellOwnsScroll = false }) {
+export function PlansPage({ isAuthenticated, authUserId, runtimeMode, authSessionResolved, currentTheme = 'dark', projectName, currentWorkspaceId, apiBaseUrl, connectedEnvironmentSource, resolveCloudAuthUrl, embedded = false, shellOwnsScroll = false, onAccountProfileSummaryChange, onContinueToTaskforce, continueBusy = false }) {
     const navigate = useNavigate();
     const location = useLocation();
     const [accountProfileSummary, setAccountProfileSummary] = useState(null);
     const [actionBusy, setActionBusy] = useState(false);
     const [feedback, setFeedback] = useState(null);
     const checkoutStartRan = useRef(false);
+    const checkoutSuccessHandledRef = useRef(null);
     const transientQueryClearTimeoutRef = useRef(null);
     const query = useMemo(() => new URLSearchParams(location.search), [location.search]);
     const checkoutState = String(query.get('checkout') || '').trim().toLowerCase();
@@ -89,32 +87,9 @@ export function PlansPage({ isAuthenticated, authUserId, runtimeMode, authSessio
         ].map((value) => String(value || '').trim()).filter((value, index, all) => value && all.indexOf(value) === index);
         return endpoints;
     }, [normalizedApiBaseUrl]);
-    const hasResolvedActiveEntitlement = useMemo(() => {
-        const entitlementState = String(accountProfileSummary?.entitlementState || '').trim().toLowerCase();
-        return ACTIVE_ENTITLEMENT_STATES.has(entitlementState);
-    }, [accountProfileSummary?.entitlementState]);
-    const gateMessage = useMemo(() => {
-        if (hasResolvedActiveEntitlement)
-            return null;
-        const billingConflictCode = String(accountProfileSummary?.billingConflictCode || '').trim().toUpperCase();
-        if (billingConflictCode === 'CHECKOUT_SESSION_EXPIRED')
-            return BILLING_CONFLICT_MESSAGES.CHECKOUT_SESSION_EXPIRED;
-        if (billingConflictCode === 'CHECKOUT_SESSION_SUPERSEDED')
-            return BILLING_CONFLICT_MESSAGES.CHECKOUT_SESSION_SUPERSEDED;
-        const summaryMessage = String(accountProfileSummary?.message || '').trim();
-        if (summaryMessage)
-            return summaryMessage;
-        const effectiveGateState = String(accountProfileSummary?.gate || gateState || '').trim().toLowerCase();
-        if (effectiveGateState === 'plan_selection_required')
-            return GATE_MESSAGES.planSelectionRequired;
-        if (effectiveGateState === 'checkout_pending')
-            return GATE_MESSAGES.checkoutPending;
-        if (effectiveGateState === 'misconfigured')
-            return GATE_MESSAGES.misconfigured;
-        if (effectiveGateState === 'missing_entitlement')
-            return GATE_MESSAGES.missingEntitlement;
-        return null;
-    }, [accountProfileSummary?.billingConflictCode, accountProfileSummary?.gate, accountProfileSummary?.message, gateState, hasResolvedActiveEntitlement]);
+    const lifecyclePresentation = useMemo(() => resolveCommercialLifecyclePresentation(accountProfileSummary, gateState), [accountProfileSummary, gateState]);
+    const hasResolvedActiveEntitlement = useMemo(() => lifecyclePresentation.allowReturnToApp, [lifecyclePresentation.allowReturnToApp]);
+    const gateMessage = useMemo(() => lifecyclePresentation.message, [lifecyclePresentation.message]);
     const clearTransientQueryState = useCallback(() => {
         const next = new URLSearchParams(location.search);
         next.delete('checkout');
@@ -148,25 +123,24 @@ export function PlansPage({ isAuthenticated, authUserId, runtimeMode, authSessio
                 force: options?.force === true
             });
             setAccountProfileSummary(payload);
+            onAccountProfileSummaryChange?.(payload);
             return payload;
         }
         catch {
             // best effort only
             return null;
         }
-    }, [authUserId, buildCloudBillingUrl, isAuthenticated]);
+    }, [authUserId, buildCloudBillingUrl, isAuthenticated, onAccountProfileSummaryChange]);
     const resolveRecoveryMessage = useCallback((summary, fallback) => {
-        const billingConflictCode = String(summary?.billingConflictCode || '').trim().toUpperCase();
-        if (billingConflictCode === 'CHECKOUT_SESSION_EXPIRED')
-            return BILLING_CONFLICT_MESSAGES.CHECKOUT_SESSION_EXPIRED;
-        if (billingConflictCode === 'CHECKOUT_SESSION_SUPERSEDED')
-            return BILLING_CONFLICT_MESSAGES.CHECKOUT_SESSION_SUPERSEDED;
-        const summaryMessage = String(summary?.message || '').trim();
-        if (summaryMessage)
-            return summaryMessage;
+        const message = resolveCommercialLifecyclePresentation(summary, gateState).message;
+        if (message)
+            return message;
         return String(fallback || 'Unable to confirm checkout completion.').trim() || 'Unable to confirm checkout completion.';
+    }, [gateState]);
+    const shouldAutoContinueAfterActivation = useCallback((summary, fallbackGate) => {
+        return !resolveCommercialLifecyclePresentation(summary, fallbackGate).allowReturnToApp;
     }, []);
-    const startCheckout = useCallback(async (planVersionId, interval) => {
+    const startCheckout = useCallback(async (planVersionId, interval, options) => {
         if (!planVersionId) {
             setFeedback({ type: 'error', message: 'Selected plan is missing a default plan version.' });
             return;
@@ -189,13 +163,21 @@ export function PlansPage({ isAuthenticated, authUserId, runtimeMode, authSessio
             if (checkoutState === 'free') {
                 setFeedback({ type: 'success', message: 'Plan selected. Your access is ready.' });
                 invalidateAccountProfileSummaryCache(buildCloudBillingUrl(ACCOUNT_PROFILE_SUMMARY_PATH), { identityKey: authUserId });
-                await loadAccountProfileSummary();
+                const summary = await loadAccountProfileSummary({ force: true });
+                const summaryState = String(summary?.entitlementState || '').trim().toLowerCase();
+                if (options?.autoContinueOnActivation === true && isCommercialEntitlementActive(summaryState)) {
+                    await onContinueToTaskforce?.();
+                }
                 return;
             }
             if (checkoutState === 'updated') {
                 setFeedback({ type: 'success', message: String(payload?.message || 'Your subscription was updated successfully.') });
                 invalidateAccountProfileSummaryCache(buildCloudBillingUrl(ACCOUNT_PROFILE_SUMMARY_PATH), { identityKey: authUserId });
-                await loadAccountProfileSummary();
+                const summary = await loadAccountProfileSummary({ force: true });
+                const summaryState = String(summary?.entitlementState || '').trim().toLowerCase();
+                if (options?.autoContinueOnActivation === true && isCommercialEntitlementActive(summaryState)) {
+                    await onContinueToTaskforce?.();
+                }
                 return;
             }
             if (checkoutState === 'already_current_plan') {
@@ -235,12 +217,13 @@ export function PlansPage({ isAuthenticated, authUserId, runtimeMode, authSessio
         finally {
             setActionBusy(false);
         }
-    }, [authUserId, buildCloudBillingUrl, loadAccountProfileSummary, runtimeMode]);
-    const confirmCheckoutSession = useCallback(async (sessionId) => {
+    }, [authUserId, buildCloudBillingUrl, loadAccountProfileSummary, onContinueToTaskforce, runtimeMode]);
+    const confirmCheckoutSession = useCallback(async (sessionId, options) => {
         if (!sessionId) {
             return {
                 confirmed: false,
-                errorMessage: 'Checkout session is missing.'
+                errorMessage: 'Checkout session is missing.',
+                retryable: false
             };
         }
         try {
@@ -252,22 +235,29 @@ export function PlansPage({ isAuthenticated, authUserId, runtimeMode, authSessio
             });
             const payload = await res.json().catch(() => ({}));
             if (!res.ok) {
+                const errorCode = String(payload?.code || '').trim();
                 return {
                     confirmed: false,
-                    errorMessage: String(payload?.error || 'Unable to confirm checkout completion.')
+                    errorMessage: String(payload?.error || 'Unable to confirm checkout completion.'),
+                    retryable: shouldRetryCheckoutConfirmation(res.status, errorCode)
                 };
             }
             invalidateAccountProfileSummaryCache(buildCloudBillingUrl(ACCOUNT_PROFILE_SUMMARY_PATH), { identityKey: authUserId });
-            await loadAccountProfileSummary();
+            const summary = await loadAccountProfileSummary({ force: true });
+            const summaryState = String(summary?.entitlementState || '').trim().toLowerCase();
+            if (options?.autoContinueOnActivation === true && isCommercialEntitlementActive(summaryState)) {
+                await onContinueToTaskforce?.();
+            }
             return { confirmed: true };
         }
         catch {
             return {
                 confirmed: false,
-                errorMessage: 'Unable to confirm checkout completion.'
+                errorMessage: 'Unable to confirm checkout completion.',
+                retryable: true
             };
         }
-    }, [authUserId, buildCloudBillingUrl, loadAccountProfileSummary]);
+    }, [authUserId, buildCloudBillingUrl, loadAccountProfileSummary, onContinueToTaskforce]);
     const openBillingPortal = useCallback(async () => {
         if (!isAuthenticated) {
             navigate(`/login?mode=login&next=${encodeURIComponent(buildEmbeddedPlansPath())}`);
@@ -315,15 +305,84 @@ export function PlansPage({ isAuthenticated, authUserId, runtimeMode, authSessio
         }
     }, []);
     useEffect(() => {
+        if (checkoutState !== 'success') {
+            checkoutSuccessHandledRef.current = null;
+        }
+    }, [checkoutState]);
+    useEffect(() => {
+        let cancelled = false;
+        const pendingRetryTimeouts = new Set();
+        const waitForRetry = async (delayMs) => {
+            if (delayMs <= 0)
+                return;
+            await new Promise((resolve) => {
+                const timeoutId = window.setTimeout(() => {
+                    pendingRetryTimeouts.delete(timeoutId);
+                    resolve();
+                }, delayMs);
+                pendingRetryTimeouts.add(timeoutId);
+            });
+        };
         if (checkoutState === 'success') {
             checkoutStartRan.current = true;
             setFeedback({ type: 'success', message: 'Checkout completed successfully. Finalizing your plan...' });
+            if (!authSessionResolved) {
+                return () => {
+                    cancelled = true;
+                    pendingRetryTimeouts.forEach((timeoutId) => window.clearTimeout(timeoutId));
+                    pendingRetryTimeouts.clear();
+                };
+            }
+            if (!isAuthenticated) {
+                setFeedback({ type: 'error', message: 'Sign in to finish activating your plan.' });
+                return () => {
+                    cancelled = true;
+                    pendingRetryTimeouts.forEach((timeoutId) => window.clearTimeout(timeoutId));
+                    pendingRetryTimeouts.clear();
+                };
+            }
+            const successFlowKey = querySessionId || '__missing_session__';
+            if (checkoutSuccessHandledRef.current === successFlowKey) {
+                return () => {
+                    cancelled = true;
+                    pendingRetryTimeouts.forEach((timeoutId) => window.clearTimeout(timeoutId));
+                    pendingRetryTimeouts.clear();
+                };
+            }
+            checkoutSuccessHandledRef.current = successFlowKey;
             void (async () => {
-                const confirmation = querySessionId
-                    ? await confirmCheckoutSession(querySessionId)
-                    : { confirmed: false, errorMessage: 'Checkout session is missing.' };
+                const preConfirmationSummary = await loadAccountProfileSummary({ force: true });
+                if (cancelled)
+                    return;
+                let confirmation = querySessionId
+                    ? { confirmed: false, errorMessage: 'Unable to confirm checkout completion.', retryable: true }
+                    : { confirmed: false, errorMessage: 'Checkout session is missing.', retryable: false };
+                for (const delayMs of CHECKOUT_CONFIRM_RETRY_DELAYS_MS) {
+                    await waitForRetry(delayMs);
+                    if (cancelled)
+                        return;
+                    confirmation = querySessionId
+                        ? await confirmCheckoutSession(querySessionId, {
+                            autoContinueOnActivation: shouldAutoContinueAfterActivation(preConfirmationSummary, gateState)
+                        })
+                        : { confirmed: false, errorMessage: 'Checkout session is missing.', retryable: false };
+                    if (cancelled)
+                        return;
+                    if (confirmation.confirmed) {
+                        setFeedback({ type: 'success', message: 'Checkout completed successfully.' });
+                        scheduleTransientQueryClear();
+                        return;
+                    }
+                    if (!confirmation.retryable) {
+                        break;
+                    }
+                }
+                if (cancelled)
+                    return;
                 if (!confirmation.confirmed) {
                     const summary = await loadAccountProfileSummary({ force: true });
+                    if (cancelled)
+                        return;
                     setFeedback({
                         type: 'error',
                         message: resolveRecoveryMessage(summary, confirmation.errorMessage)
@@ -331,10 +390,12 @@ export function PlansPage({ isAuthenticated, authUserId, runtimeMode, authSessio
                     scheduleTransientQueryClear();
                     return;
                 }
-                setFeedback({ type: 'success', message: 'Checkout completed successfully.' });
-                scheduleTransientQueryClear();
             })();
-            return;
+            return () => {
+                cancelled = true;
+                pendingRetryTimeouts.forEach((timeoutId) => window.clearTimeout(timeoutId));
+                pendingRetryTimeouts.clear();
+            };
         }
         if (checkoutState === 'cancel') {
             void (async () => {
@@ -349,29 +410,31 @@ export function PlansPage({ isAuthenticated, authUserId, runtimeMode, authSessio
         }
         if ((checkoutState === 'start' || checkoutState === 'free') && isAuthenticated && queryPlanVersionId && !checkoutStartRan.current) {
             checkoutStartRan.current = true;
-            void startCheckout(queryPlanVersionId, queryInterval);
+            void (async () => {
+                const preCheckoutSummary = await loadAccountProfileSummary({ force: true });
+                await startCheckout(queryPlanVersionId, queryInterval, {
+                    autoContinueOnActivation: shouldAutoContinueAfterActivation(preCheckoutSummary, gateState)
+                });
+            })();
+            return;
+        }
+    }, [authSessionResolved, checkoutState, confirmCheckoutSession, gateState, isAuthenticated, loadAccountProfileSummary, queryInterval, queryPlanVersionId, querySessionId, resolveRecoveryMessage, scheduleTransientQueryClear, shouldAutoContinueAfterActivation, startCheckout]);
+    useEffect(() => {
+        if (checkoutState === 'success' || checkoutState === 'cancel' || checkoutState === 'start' || checkoutState === 'free') {
             return;
         }
         if (gateMessage) {
-            setFeedback({ type: 'error', message: gateMessage });
+            setFeedback({ type: lifecyclePresentation.statusTone, message: gateMessage });
             return;
         }
         setFeedback((current) => {
             if (!current || current.type !== 'error')
                 return current;
-            if (current.message === GATE_MESSAGES.planSelectionRequired)
-                return null;
-            if (current.message === GATE_MESSAGES.misconfigured)
-                return null;
-            if (current.message === GATE_MESSAGES.missingEntitlement)
-                return null;
-            if (current.message === BILLING_CONFLICT_MESSAGES.CHECKOUT_SESSION_EXPIRED)
-                return null;
-            if (current.message === BILLING_CONFLICT_MESSAGES.CHECKOUT_SESSION_SUPERSEDED)
+            if (current.message === lifecyclePresentation.message)
                 return null;
             return current;
         });
-    }, [checkoutState, confirmCheckoutSession, gateMessage, isAuthenticated, loadAccountProfileSummary, queryInterval, queryPlanVersionId, querySessionId, resolveRecoveryMessage, scheduleTransientQueryClear, startCheckout]);
+    }, [checkoutState, gateMessage, lifecyclePresentation.message, lifecyclePresentation.statusTone]);
     const handleSelectPlan = useCallback(async (intent) => {
         if (!isAuthenticated) {
             const nextCheckoutState = intent.pricingType === 'free' ? 'free' : 'start';
@@ -392,13 +455,18 @@ export function PlansPage({ isAuthenticated, authUserId, runtimeMode, authSessio
             await openBillingPortal();
             return;
         }
+        const autoContinueOnActivation = shouldAutoContinueAfterActivation(accountProfileSummary, gateState);
         if (intent.pricingType === 'free') {
-            await startCheckout(intent.planVersionId, intent.interval);
+            await startCheckout(intent.planVersionId, intent.interval, { autoContinueOnActivation });
             return;
         }
-        await startCheckout(intent.planVersionId, intent.interval);
-    }, [isAuthenticated, navigate, openBillingPortal, startCheckout]);
+        await startCheckout(intent.planVersionId, intent.interval, { autoContinueOnActivation });
+    }, [accountProfileSummary, gateState, isAuthenticated, navigate, openBillingPortal, shouldAutoContinueAfterActivation, startCheckout]);
     const handleBack = useCallback(() => {
+        if (hasResolvedActiveEntitlement && onContinueToTaskforce) {
+            void onContinueToTaskforce();
+            return;
+        }
         if (embedded) {
             closeEmbeddedPlansScreen();
             return;
@@ -408,8 +476,9 @@ export function PlansPage({ isAuthenticated, authUserId, runtimeMode, authSessio
             return;
         }
         navigate('/');
-    }, [closeEmbeddedPlansScreen, embedded, navigate]);
-    const topActions = embedded ? (_jsx(_Fragment, { children: isAuthenticated && accountProfileSummary?.stripeCustomerId && (_jsx("button", { className: styles.navAction, onClick: () => { void openBillingPortal(); }, disabled: actionBusy, children: "Manage billing" })) })) : (_jsxs(_Fragment, { children: [_jsx("button", { className: styles.navAction, onClick: handleBack, children: "Back" }), isAuthenticated && accountProfileSummary?.stripeCustomerId && (_jsx("button", { className: styles.navAction, onClick: () => { void openBillingPortal(); }, disabled: actionBusy, children: "Manage billing" })), !isAuthenticated && (_jsxs("button", { className: styles.navAction, onClick: () => navigate(`/login?mode=login&next=${encodeURIComponent(buildEmbeddedPlansPath())}`), children: [_jsx(User, { size: 16, style: { marginRight: '8px', verticalAlign: 'text-bottom' } }), "Sign In"] }))] }));
+    }, [closeEmbeddedPlansScreen, embedded, hasResolvedActiveEntitlement, navigate, onContinueToTaskforce]);
+    const continueAction = hasResolvedActiveEntitlement && onContinueToTaskforce ? (_jsx("button", { className: styles.navAction, onClick: () => { void onContinueToTaskforce(); }, disabled: continueBusy || actionBusy, children: "Take Me to Taskforce" })) : null;
+    const topActions = embedded ? (_jsxs(_Fragment, { children: [continueAction, isAuthenticated && accountProfileSummary?.stripeCustomerId && (_jsx("button", { className: styles.navAction, onClick: () => { void openBillingPortal(); }, disabled: actionBusy, children: "Manage billing" }))] })) : (_jsxs(_Fragment, { children: [_jsx("button", { className: styles.navAction, onClick: handleBack, disabled: continueBusy || (isAuthenticated && !hasResolvedActiveEntitlement), children: hasResolvedActiveEntitlement ? 'Take Me to Taskforce' : 'Back' }), isAuthenticated && accountProfileSummary?.stripeCustomerId && (_jsx("button", { className: styles.navAction, onClick: () => { void openBillingPortal(); }, disabled: actionBusy, children: "Manage billing" })), !isAuthenticated && (_jsxs("button", { className: styles.navAction, onClick: () => navigate(`/login?mode=login&next=${encodeURIComponent(buildEmbeddedPlansPath())}`), children: [_jsx(User, { size: 16, style: { marginRight: '8px', verticalAlign: 'text-bottom' } }), "Sign In"] }))] }));
     const chrome = embedded ? null : (_jsx(AppShellHeader, { projectName: projectName, currentWorkspaceId: currentWorkspaceId, runtimeMode: runtimeMode === 'cloud' ? 'cloud' : 'local', theme: currentTheme, actions: topActions }));
-    return (_jsx(PlanComparisonPage, { chrome: chrome, heading: "Subscription Plans", subtitle: "Manage your subscription and explore available features. Your current plan is highlighted below.", isAuthenticated: isAuthenticated, authSessionResolved: authSessionResolved, currentPlanId: String(accountProfileSummary?.planId || queryPlanId || '').trim() || null, currentPlanVersionId: String(accountProfileSummary?.planVersionId || queryPlanVersionId || '').trim() || null, currentEntitlementState: String(accountProfileSummary?.entitlementState || '').trim() || null, canManageCurrentPlan: Boolean(accountProfileSummary?.stripeCustomerId), pricingEndpoint: pricingEndpoints[0], statusBanner: feedback, onBack: handleBack, backLabel: embedded ? 'Back to Board' : 'Back to App', topActions: embedded ? topActions : null, shellOwnsScroll: shellOwnsScroll, showPageBackButton: !embedded, fallbackPricingSource: fallbackPricingSource, onSelectPlan: handleSelectPlan }));
+    return (_jsx(PlanComparisonPage, { chrome: chrome, heading: "Subscription Plans", subtitle: "Manage your subscription and explore available features. Your current plan is highlighted below.", isAuthenticated: isAuthenticated, authSessionResolved: authSessionResolved, currentPlanId: String(accountProfileSummary?.planId || queryPlanId || '').trim() || null, currentPlanVersionId: String(accountProfileSummary?.planVersionId || queryPlanVersionId || '').trim() || null, currentEntitlementState: String(accountProfileSummary?.entitlementState || '').trim() || null, canManageCurrentPlan: Boolean(accountProfileSummary?.stripeCustomerId), pricingEndpoint: pricingEndpoints[0], statusBanner: feedback, onBack: handleBack, backLabel: hasResolvedActiveEntitlement ? 'Take Me to Taskforce' : (embedded ? 'Back to Board' : 'Back to App'), backDisabled: continueBusy || (isAuthenticated && !hasResolvedActiveEntitlement), topActions: embedded ? topActions : null, shellOwnsScroll: shellOwnsScroll, showPageBackButton: !embedded, fallbackPricingSource: fallbackPricingSource, onSelectPlan: handleSelectPlan }));
 }