@umituz/react-native-subscription 2.43.20 → 2.44.0

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@umituz/react-native-subscription",
-  "version": "2.43.20",
+  "version": "2.44.0",
   "description": "Complete subscription management with RevenueCat, paywall UI, and credits system for React Native apps",
   "main": "./src/index.ts",
   "types": "./src/index.ts",

package/src/domains/subscription/presentation/components/ManagedSubscriptionFlow.tsx

@@ -1,6 +1,14 @@
-import React, { useState, useEffect, useCallback } from "react";
+/**
+ * ManagedSubscriptionFlow
+ *
+ * State machine-based flow orchestration.
+ * No complex if/else - clean state transitions.
+ */
+
+import React, { useEffect, useCallback } from "react";
 import type { NavigationProp } from "@react-navigation/native";
 import type { ImageSourcePropType } from "react-native";
+import { View } from "react-native";
 import { SplashScreen, useSplashFlow } from "@umituz/react-native-design-system/molecules";
 import { OnboardingScreen } from "@umituz/react-native-design-system/onboarding";
 import { OfflineBanner } from "@umituz/react-native-design-system/offline";
@@ -8,10 +16,10 @@ import { useAppDesignTokens } from "@umituz/react-native-design-system/theme";
 import { usePremiumStatus } from "../../presentation/usePremiumStatus";
 import { usePremiumPackages } from "../../presentation/usePremiumPackages";
 import { usePremiumActions } from "../../presentation/usePremiumActions";
-import { usePaywallOrchestrator } from "../../../paywall/hooks/usePaywallOrchestrator";
-import { PaywallFeedbackScreen } from "../../../subscription/presentation/components/feedback/PaywallFeedbackScreen";
-import { PaywallFeedbackTranslations } from "../../../subscription/presentation/components/feedback/PaywallFeedbackScreen.types";
-import { PaywallTranslations, PaywallLegalUrls, SubscriptionFeature } from "../../../paywall/entities/types";
+import { useSubscriptionFlowStore, SubscriptionFlowStatus } from "../useSubscriptionFlow";
+import { PaywallFeedbackScreen } from "./feedback/PaywallFeedbackScreen";
+import type { PaywallFeedbackTranslations } from "./feedback/PaywallFeedbackScreen.types";
+import type { PaywallTranslations, PaywallLegalUrls, SubscriptionFeature } from "../../../paywall/entities/types";
 import { usePaywallFeedbackSubmit } from "../../../../presentation/hooks/feedback/useFeedbackSubmit";
 import { PaywallScreen } from "../../../paywall/components/PaywallScreen";
 
@@ -66,19 +74,124 @@ export interface ManagedSubscriptionFlowProps {
   };
 }
 
-/**
- * ManagedSubscriptionFlow
- * 
- * A high-level layout component that orchestrates the entire application flow:
- * Splash -> Onboarding -> [Managed Paywall Screens] -> Main Application Stack.
- * 
- * Use this to reduce AppNavigator boilerplate to nearly zero.
- */
 import {
   SubscriptionFlowProvider,
   useSubscriptionFlowStatus
 } from "../providers/SubscriptionFlowProvider";
-import { useSubscriptionFlowStore, SubscriptionFlowStatus } from "../useSubscriptionFlow";
+
+// ============================================================================
+// STATE MACHINE COMPONENTS
+// ============================================================================
+
+interface StateComponentProps {
+  status: SubscriptionFlowStatus;
+  tokens: any;
+  onboardingConfig: ManagedSubscriptionFlowProps["onboarding"];
+  paywallConfig: ManagedSubscriptionFlowProps["paywall"];
+  feedbackConfig: ManagedSubscriptionFlowProps["feedback"];
+  isPremium: boolean;
+  packages: any[];
+  credits: number | null;
+  isSyncing: boolean;
+  onPurchasePackage: (pkgId: string) => Promise<any>;
+  onRestorePurchase: () => Promise<any>;
+  navigation: NavigationProp<any>;
+}
+
+const InitializingState: React.FC<{ tokens: any; splash?: ManagedSubscriptionFlowProps["splash"] }> = ({ tokens, splash }) => (
+  <SplashScreen
+    appName={splash?.appName || "Loading..."}
+    tagline={splash?.tagline || "Please wait while we set things up"}
+    colors={tokens.colors}
+  />
+);
+
+const OnboardingState: React.FC<{
+  config: ManagedSubscriptionFlowProps["onboarding"];
+  onComplete: () => void;
+}> = ({ config, onComplete }) => (
+  <OnboardingScreen
+    slides={config.slides}
+    onComplete={onComplete}
+    showSkipButton={config.showSkipButton ?? true}
+    showBackButton={config.showBackButton ?? true}
+    showProgressBar={config.showProgressBar ?? true}
+    themeColors={config.themeColors}
+    translations={config.translations}
+  />
+);
+
+const PaywallState: React.FC<{
+  config: ManagedSubscriptionFlowProps["paywall"];
+  packages: any[];
+  isPremium: boolean;
+  credits: number | null;
+  isSyncing: boolean;
+  onPurchase: (pkgId: string) => Promise<any>;
+  onRestore: () => Promise<any>;
+  onClose: (purchased: boolean) => void;
+}> = ({ config, packages, isPremium, credits, isSyncing, onPurchase, onRestore, onClose }) => {
+  const [purchaseSuccessful, setPurchaseSuccessful] = React.useState(false);
+
+  const handlePurchase = async (pkgId: string) => {
+    const result = await onPurchase(pkgId);
+    if (result?.success) {
+      setPurchaseSuccessful(true);
+    }
+    return result;
+  };
+
+  const handleClose = () => {
+    onClose(purchaseSuccessful);
+  };
+
+  return (
+    <PaywallScreen
+      translations={config.translations}
+      legalUrls={config.legalUrls}
+      features={config.features}
+      bestValueIdentifier={config.bestValueIdentifier}
+      creditsLabel={config.creditsLabel}
+      heroImage={config.heroImage}
+      source="onboarding"
+      packages={packages}
+      isPremium={isPremium}
+      credits={credits}
+      isSyncing={isSyncing}
+      onPurchase={handlePurchase}
+      onRestore={onRestore}
+      onClose={handleClose}
+    />
+  );
+};
+
+const FeedbackState: React.FC<{
+  config: ManagedSubscriptionFlowProps["feedback"];
+  onClose: () => void;
+}> = ({ config, onClose }) => {
+  const { submit: internalSubmit } = usePaywallFeedbackSubmit();
+
+  const handleSubmit = async (data: { reason: string; otherText?: string }) => {
+    if (config.onSubmit) {
+      await config.onSubmit(data);
+    } else {
+      const description = data.otherText ? `${data.reason}: ${data.otherText}` : data.reason;
+      await internalSubmit(description);
+    }
+  };
+
+  return (
+    <PaywallFeedbackScreen
+      onClose={onClose}
+      onSubmit={handleSubmit}
+      translations={config.translations}
+    />
+  );
+};
+
+// ============================================================================
+// MAIN COMPONENT
+// ============================================================================
 
 const ManagedSubscriptionFlowInner = React.memo<ManagedSubscriptionFlowProps>(({
   children,
@@ -96,165 +209,111 @@ const ManagedSubscriptionFlowInner = React.memo<ManagedSubscriptionFlowProps>(({
     duration: splash?.duration || 0,
   });
 
-  // Hooks for paywall (called at component level, not inside conditional render)
-  const { isPremium, isSyncing, credits } = usePremiumStatus();
+  // Premium hooks
+  const { isPremium, isSyncing, credits, isLoading: isPremiumLoading } = usePremiumStatus();
   const { packages } = usePremiumPackages();
-  const { purchasePackage: originalPurchasePackage, restorePurchase } = usePremiumActions();
-  const { closePostOnboardingPaywall } = useSubscriptionFlowStore();
+  const { purchasePackage, restorePurchase } = usePremiumActions();
 
-  // Track if purchase was successful to avoid showing feedback
-  const [purchaseSuccessful, setPurchaseSuccessful] = useState(false);
+  // Store actions
+  const completeOnboarding = useSubscriptionFlowStore((s) => s.completeOnboarding);
+  const showPaywall = useSubscriptionFlowStore((s) => s.showPaywall);
+  const completePaywall = useSubscriptionFlowStore((s) => s.completePaywall);
+  const hideFeedback = useSubscriptionFlowStore((s) => s.hideFeedback);
+  const showFeedbackScreen = useSubscriptionFlowStore((s) => s.showFeedbackScreen);
 
-  // Wrap purchasePackage to track success
-  const purchasePackage = useCallback(async (pkgId: string) => {
-    const result = await originalPurchasePackage(pkgId);
-    if (result?.success) {
-      setPurchaseSuccessful(true);
-    }
-    return result;
-  }, [originalPurchasePackage]);
+  const showFeedback = useSubscriptionFlowStore((s) => s.showFeedback);
 
-  // Wrap onClose to pass purchase status
-  const handleClose = useCallback(() => {
-    closePostOnboardingPaywall({ purchased: purchaseSuccessful });
-  }, [closePostOnboardingPaywall, purchaseSuccessful]);
+  // ========================================================================
+  // STATE TRANSITIONS
+  // ========================================================================
 
-  const [isNavReady, setIsNavReady] = useState(false);
-
-  // Mark navigation tree as ready after splash and localization
+  // CHECK_PREMIUM state transition logic
   useEffect(() => {
-    if (isSplashComplete && islocalizationReady) {
-      const timer = setTimeout(() => setIsNavReady(true), 500);
-      return () => clearTimeout(timer);
-    }
-  }, [isSplashComplete, islocalizationReady]);
-
-  const {
-    flowState,
-    setShowFeedback,
-    completeOnboarding
-  } = usePaywallOrchestrator({
-    navigation,
-    isNavReady,
-    isLocalizationReady: islocalizationReady,
-    translations: paywall.translations,
-    features: paywall.features,
-    legalUrls: paywall.legalUrls,
-    heroImage: paywall.heroImage,
-    bestValueIdentifier: paywall.bestValueIdentifier,
-    creditsLabel: paywall.creditsLabel,
-    disableNavigation: true, // Paywall is rendered inline, don't navigate
-  });
-
-  const { submit: internalSubmit } = usePaywallFeedbackSubmit();
+    if (status === SubscriptionFlowStatus.CHECK_PREMIUM && !isPremiumLoading) {
+      const paywallShown = useSubscriptionFlowStore.getState().paywallShown;
 
-  const handleFeedbackSubmit = async (data: { reason: string; otherText?: string }) => {
-    if (feedback.onSubmit) {
-      await feedback.onSubmit(data);
-    } else {
-      const description = data.otherText ? `${data.reason}: ${data.otherText}` : data.reason;
-      await internalSubmit(description);
+      if (isPremium) {
+        // User is premium, go to ready
+        completePaywall(true);
+      } else if (!paywallShown) {
+        // User not premium and paywall not shown, show paywall
+        showPaywall();
+      } else {
+        // Paywall already shown, go to ready
+        completePaywall(false);
+      }
     }
-  };
+  }, [status, isPremium, isPremiumLoading, showPaywall, completePaywall]);
 
-  // 1. Loading / Initialization View
-  if (status === SubscriptionFlowStatus.INITIALIZING || !islocalizationReady) {
-    if (__DEV__) {
-       console.log('[ManagedSubscriptionFlow] ⏳ Rendering Initialization state', {
-         status,
-         islocalizationReady,
-         hasSplashConfig: !!splash,
-         isSplashComplete
-       });
+  // Show feedback when needed
+  useEffect(() => {
+    if (status === SubscriptionFlowStatus.READY && showFeedback) {
+      showFeedbackScreen();
     }
+  }, [status, showFeedback, showFeedbackScreen]);
 
-    // Even if no splash config provided, we should show a basic splash to avoid white screen
-    return (
-      <SplashScreen
-        appName={splash?.appName || "Loading..."}
-        tagline={splash?.tagline || "Please wait while we set things up"}
-        colors={tokens.colors}
-      />
-    );
-  }
+  // ========================================================================
+  // RENDER
+  // ========================================================================
 
-  if (__DEV__) {
-    console.log('[ManagedSubscriptionFlow] 🔄 Rendering Main state', {
-      status,
-      isSplashComplete,
-      islocalizationReady,
-      showFeedback: flowState.showFeedback
-    });
+  // Wait for localization
+  if (!islocalizationReady || status === SubscriptionFlowStatus.INITIALIZING) {
+    return <InitializingState tokens={tokens} splash={splash} />;
   }
 
-  // 2. Onboarding View
-  if (status === SubscriptionFlowStatus.ONBOARDING) {
-    return (
-      <OnboardingScreen
-        slides={onboarding.slides}
-        onComplete={completeOnboarding}
-        showSkipButton={onboarding.showSkipButton ?? true}
-        showBackButton={onboarding.showBackButton ?? true}
-        showProgressBar={onboarding.showProgressBar ?? true}
-        themeColors={onboarding.themeColors}
-        translations={onboarding.translations}
-      />
-    );
-  }
-
-  // 2.5. Post-Onboarding Paywall View
-  if (status === SubscriptionFlowStatus.POST_ONBOARDING_PAYWALL) {
-    if (__DEV__) {
-      console.log('[ManagedSubscriptionFlow] 💳 Rendering Post-Onboarding Paywall', {
-        hasPackages: flowState.showPostOnboardingPaywall
-      });
-    }
+  // Render by state
+  switch (status) {
+    case SubscriptionFlowStatus.ONBOARDING:
+      return <OnboardingState config={onboarding} onComplete={completeOnboarding} />;
 
-    return (
-      <PaywallScreen
-        translations={paywall.translations}
-        legalUrls={paywall.legalUrls}
-        features={paywall.features}
-        bestValueIdentifier={paywall.bestValueIdentifier}
-        creditsLabel={paywall.creditsLabel}
-        heroImage={paywall.heroImage}
-        source="onboarding"
-        packages={packages}
-        isPremium={isPremium}
-        credits={credits}
-        isSyncing={isSyncing}
-        onPurchase={purchasePackage}
-        onRestore={restorePurchase}
-        onClose={handleClose}
-      />
-    );
-  }
+    case SubscriptionFlowStatus.CHECK_PREMIUM:
+      // Show loading while checking premium
+      return <InitializingState tokens={tokens} splash={splash} />;
 
-  // 3. Application Content + Overlays
-  return (
-    <>
-      {children}
-      
-      {offline && (
-        <OfflineBanner
-          visible={offline.isOffline}
-          message={offline.message}
-          backgroundColor={offline.backgroundColor || tokens.colors.error}
-          position={offline.position || "top"}
+    case SubscriptionFlowStatus.POST_ONBOARDING_PAYWALL:
+      return (
+        <PaywallState
+          config={paywall}
+          packages={packages}
+          isPremium={isPremium}
+          credits={credits}
+          isSyncing={isSyncing}
+          onPurchase={purchasePackage}
+          onRestore={restorePurchase}
+          onClose={(purchased) => completePaywall(purchased)}
         />
-      )}
+      );
 
-      {flowState.showFeedback && (
-        <PaywallFeedbackScreen
-          onClose={() => setShowFeedback(false)}
-          onSubmit={handleFeedbackSubmit}
-          translations={feedback.translations}
-        />
-      )}
-    </>
-  );
+    case SubscriptionFlowStatus.READY:
+      return (
+        <>
+          {children}
+
+          {offline && (
+            <OfflineBanner
+              visible={offline.isOffline}
+              message={offline.message}
+              backgroundColor={offline.backgroundColor || tokens.colors.error}
+              position={offline.position || "top"}
+            />
+          )}
+
+          {showFeedback && (
+            <FeedbackState
+              config={feedback}
+              onClose={hideFeedback}
+            />
+          )}
+        </>
+      );
+
+    default:
+      return <InitializingState tokens={tokens} splash={splash} />;
+  }
 });
 
+ManagedSubscriptionFlowInner.displayName = "ManagedSubscriptionFlowInner";
+
 export const ManagedSubscriptionFlow: React.FC<ManagedSubscriptionFlowProps> = (props) => {
   return (
     <SubscriptionFlowProvider>

package/src/domains/subscription/presentation/components/feedback/FeedbackOption.styles.ts

@@ -11,24 +11,24 @@ export const feedbackOptionStyles = StyleSheet.create({
     marginRight: 12,
   },
   radioButton: {
-    width: 22,
-    height: 22,
-    borderRadius: 11,
-    borderWidth: 2,
+    width: 26,
+    height: 26,
+    borderRadius: 13,
+    borderWidth: 2.5,
     justifyContent: "center",
     alignItems: "center",
   },
   radioButtonInner: {
-    width: 12,
-    height: 12,
-    borderRadius: 6,
+    width: 14,
+    height: 14,
+    borderRadius: 7,
   },
   inputContainer: {
     padding: 12,
   },
   textInput: {
-    fontSize: 15,
-    borderWidth: 1,
-    borderColor: "#ccc",
+    fontSize: 16,
+    borderWidth: 1.5,
+    lineHeight: 22,
   },
 });

package/src/domains/subscription/presentation/components/feedback/FeedbackOption.tsx

@@ -20,10 +20,11 @@ export const FeedbackOption: React.FC<FeedbackOptionProps> = React.memo(({
   const tokens = useAppDesignTokens();
 
   const containerStyle = {
-    marginBottom: tokens.spacing.sm,
     backgroundColor: tokens.colors.surfaceSecondary,
     borderRadius: tokens.borderRadius.md,
     overflow: "hidden" as const,
+    borderWidth: 1,
+    borderColor: isSelected ? tokens.colors.primary : "transparent",
   };
 
   return (

package/src/domains/subscription/presentation/components/feedback/PaywallFeedbackScreen.tsx

@@ -141,21 +141,22 @@ PaywallFeedbackScreen.displayName = "PaywallFeedbackScreen";
 
 const createScreenStyles = (
     tokens: {
-        colors: { backgroundPrimary: string; border: string };
-        spacing: { xl: number; sm: number; md: number };
+        colors: { backgroundPrimary: string; border: string; onPrimary: string; textDisabled: string; surfaceSecondary: string; primary: string };
+        spacing: { xl: number; sm: number; md: number; lg: number };
     },
     _insets: { top: number; bottom: number }
 ) => ({
     container: {
         flex: 1,
+        opacity: 1,
     },
     closeBtn: {
         position: 'absolute' as const,
         top: 12,
         right: 12,
-        width: 36,
-        height: 36,
-        borderRadius: 18,
+        width: 40,
+        height: 40,
+        borderRadius: 20,
         zIndex: 1000,
         justifyContent: 'center' as const,
         alignItems: 'center' as const,
@@ -164,21 +165,22 @@ const createScreenStyles = (
         flex: 1,
     },
     scrollContent: {
-        paddingTop: 60,
-        paddingBottom: 100,
+        paddingTop: 80,
+        paddingBottom: 120,
     },
     header: {
         paddingHorizontal: tokens.spacing.xl,
-        marginBottom: tokens.spacing.xl,
+        marginBottom: tokens.spacing.xl + 8,
     },
     title: {
-        marginBottom: tokens.spacing.sm,
+        marginBottom: tokens.spacing.md + 4,
     },
     subtitle: {
-        lineHeight: 22,
+        lineHeight: 24,
     },
     optionsContainer: {
         paddingHorizontal: tokens.spacing.xl,
+        gap: tokens.spacing.md,
     },
     footer: {
         position: 'absolute' as const,
@@ -186,17 +188,26 @@ const createScreenStyles = (
         left: 0,
         right: 0,
         paddingHorizontal: tokens.spacing.xl,
-        paddingTop: tokens.spacing.md,
+        paddingTop: tokens.spacing.lg,
+        paddingBottom: tokens.spacing.lg,
         backgroundColor: tokens.colors.backgroundPrimary,
         borderTopWidth: 1,
         borderTopColor: tokens.colors.border,
+        opacity: 1,
     },
     submitButton: {
-        borderRadius: 14,
-        paddingVertical: 16,
+        borderRadius: 16,
+        paddingVertical: 18,
         alignItems: 'center' as const,
+        shadowColor: "#000",
+        shadowOffset: { width: 0, height: 2 },
+        shadowOpacity: 0.1,
+        shadowRadius: 4,
+        elevation: 3,
     },
     submitText: {
-        fontWeight: "600" as const,
+        fontWeight: "700" as const,
+        fontSize: 17,
+        letterSpacing: 0.3,
     },
 });

package/src/domains/subscription/presentation/useSubscriptionFlow.ts

@@ -1,7 +1,8 @@
 /**
- * Subscription Flow Store
- * Manages the high-level application flow: Splash -> Onboarding -> Paywall -> Main App.
- * Uses @umituz/react-native-design-system's storage utility for standardized persistence.
+ * Subscription Flow State Machine
+ *
+ * Single source of truth for app flow state.
+ * Clean state transitions without complex if/else logic.
  */
 
 import { createStore } from "@umituz/react-native-design-system/storage";
@@ -10,9 +11,9 @@ import { subscriptionEventBus, FLOW_EVENTS } from "../../../shared/infrastructur
 export enum SubscriptionFlowStatus {
   INITIALIZING = "INITIALIZING",
   ONBOARDING = "ONBOARDING",
-  PAYWALL = "PAYWALL",
-  READY = "READY",
+  CHECK_PREMIUM = "CHECK_PREMIUM",
   POST_ONBOARDING_PAYWALL = "POST_ONBOARDING_PAYWALL",
+  READY = "READY",
 }
 
 export enum SyncStatus {
@@ -23,28 +24,46 @@ export enum SyncStatus {
 }
 
 export interface SubscriptionFlowState {
+  // Flow state
   status: SubscriptionFlowStatus;
+
+  // Sync state
   syncStatus: SyncStatus;
   syncError: string | null;
-  isInitialized: boolean;
+
+  // Onboarding state
   isOnboardingComplete: boolean;
-  showPostOnboardingPaywall: boolean;
-  showFeedback: boolean;
+
+  // Paywall state
   paywallShown: boolean;
+
+  // Feedback state
+  showFeedback: boolean;
+
+  // Auth modal state
   isAuthModalOpen: boolean;
+
+  // Initialization flag
+  isInitialized: boolean;
 }
 
 export interface SubscriptionFlowActions {
-  completeOnboarding: () => Promise<void>;
-  closePostOnboardingPaywall: (params?: { purchased: boolean }) => Promise<void>;
-  closeFeedback: () => void;
+  // Flow actions
+  completeOnboarding: () => void;
+  checkPremiumStatus: () => void;
+  showPaywall: () => void;
+  completePaywall: (purchased: boolean) => void;
+  showFeedbackScreen: () => void;
+  hideFeedback: () => void;
+
+  // Auth actions
   setAuthModalOpen: (open: boolean) => void;
-  markPaywallShown: () => Promise<void>;
-  setShowFeedback: (show: boolean) => void;
-  resetFlow: () => Promise<void>;
-  setInitialized: (initialized: boolean) => void;
-  setStatus: (status: SubscriptionFlowStatus) => void;
+
+  // Sync actions
   setSyncStatus: (status: SyncStatus, error?: string | null) => void;
+
+  // Reset
+  resetFlow: () => void;
 }
 
 export type SubscriptionFlowStore = SubscriptionFlowState & SubscriptionFlowActions;
@@ -53,73 +72,103 @@ const initialState: SubscriptionFlowState = {
   status: SubscriptionFlowStatus.INITIALIZING,
   syncStatus: SyncStatus.IDLE,
   syncError: null,
-  isInitialized: false,
   isOnboardingComplete: false,
-  showPostOnboardingPaywall: false,
-  showFeedback: false,
   paywallShown: false,
+  showFeedback: false,
   isAuthModalOpen: false,
+  isInitialized: false,
 };
 
+/**
+ * State transition rules:
+ *
+ * INITIALIZING -> ONBOARDING (first launch)
+ * INITIALIZING -> CHECK_PREMIUM (onboarding already done)
+ *
+ * ONBOARDING -> CHECK_PREMIUM (onboarding completed)
+ *
+ * CHECK_PREMIUM -> READY (user is premium)
+ * CHECK_PREMIUM -> POST_ONBOARDING_PAYWALL (user not premium, paywall not shown)
+ * CHECK_PREMIUM -> READY (user not premium but paywall already shown)
+ *
+ * POST_ONBOARDING_PAYWALL -> READY (paywall closed)
+ *
+ * READY -> READY (stays ready, shows overlays when needed)
+ */
 export const useSubscriptionFlowStore = createStore<SubscriptionFlowState, SubscriptionFlowActions>({
   name: "subscription-flow-storage",
   initialState,
-  persist: false,
+  persist: true,
   onRehydrate: (state) => {
     if (!state.isInitialized) {
       state.setInitialized(true);
+
+      // First time: show onboarding
+      state.setStatus(SubscriptionFlowStatus.INITIALIZING);
+    } else if (state.isOnboardingComplete) {
+      // Onboarding done, check premium status
+      state.setStatus(SubscriptionFlowStatus.CHECK_PREMIUM);
+    } else {
+      // Show onboarding
+      state.setStatus(SubscriptionFlowStatus.ONBOARDING);
     }
   },
   actions: (set) => ({
-    completeOnboarding: async () => {
+    setInitialized: (initialized: boolean) => set({ isInitialized: initialized }),
+    setStatus: (status: SubscriptionFlowStatus) => set({ status }),
+
+    completeOnboarding: () => {
       set({
         isOnboardingComplete: true,
-        showPostOnboardingPaywall: true,
-        status: SubscriptionFlowStatus.POST_ONBOARDING_PAYWALL,
+        status: SubscriptionFlowStatus.CHECK_PREMIUM,
       });
       subscriptionEventBus.emit(FLOW_EVENTS.ONBOARDING_COMPLETED, { timestamp: Date.now() });
     },
-    closePostOnboardingPaywall: async (params?: { purchased: boolean }) => {
-      const purchased = params?.purchased ?? false;
+
+    checkPremiumStatus: () => {
+      // This is a transient state - the component will check isPremium
+      // and transition accordingly
+      set({ status: SubscriptionFlowStatus.CHECK_PREMIUM });
+    },
+
+    showPaywall: () => {
       set({
-        showPostOnboardingPaywall: false,
+        status: SubscriptionFlowStatus.POST_ONBOARDING_PAYWALL,
         paywallShown: true,
+      });
+      subscriptionEventBus.emit(FLOW_EVENTS.PAYWALL_SHOWN, { timestamp: Date.now() });
+    },
+
+    completePaywall: (purchased: boolean) => {
+      set({
         status: SubscriptionFlowStatus.READY,
-        showFeedback: !purchased, // Only show feedback if NOT purchased
+        paywallShown: true,
+        showFeedback: !purchased, // Only show feedback if not purchased
       });
       subscriptionEventBus.emit(FLOW_EVENTS.PAYWALL_CLOSED, {
         timestamp: Date.now(),
         purchased
       });
     },
-    closeFeedback: () => set({ showFeedback: false }),
+
+    showFeedbackScreen: () => set({ showFeedback: true }),
+    hideFeedback: () => set({ showFeedback: false }),
+
     setAuthModalOpen: (open: boolean) => set({ isAuthModalOpen: open }),
-    setShowFeedback: (show: boolean) => set({ showFeedback: show }),
-    markPaywallShown: async () => {
-      set({ paywallShown: true });
-      subscriptionEventBus.emit(FLOW_EVENTS.PAYWALL_SHOWN, { timestamp: Date.now() });
-    },
-    setInitialized: (initialized: boolean) => set((state) => {
-      if (state.isInitialized === initialized) return state;
-      return { isInitialized: initialized };
-    }),
-    setStatus: (status: SubscriptionFlowStatus) => set((state) => {
-      if (state.status === status) return state;
-      return { status };
-    }),
-    setSyncStatus: (syncStatus: SyncStatus, syncError: string | null = null) => 
+
+    setSyncStatus: (syncStatus: SyncStatus, syncError: string | null = null) =>
       set({ syncStatus, syncError }),
-    resetFlow: async () => {
+
+    resetFlow: () => {
       set({
         status: SubscriptionFlowStatus.INITIALIZING,
         syncStatus: SyncStatus.IDLE,
         syncError: null,
-        isInitialized: false, // Reset isInitialized to allow fresh initialization
         isOnboardingComplete: false,
-        showPostOnboardingPaywall: false,
-        showFeedback: false,
         paywallShown: false,
+        showFeedback: false,
         isAuthModalOpen: false,
+        isInitialized: false,
       });
     },
   }),