@umituz/react-native-ai-generation-content 1.44.4 → 1.45.0

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@umituz/react-native-ai-generation-content",
-  "version": "1.44.4",
+  "version": "1.45.0",
   "description": "Provider-agnostic AI generation orchestration for React Native with result preview components",
   "main": "src/index.ts",
   "types": "src/index.ts",

package/src/domains/generation/wizard/infrastructure/strategies/image-generation.strategy.ts

@@ -80,7 +80,7 @@ function applyStyleEnhancements(prompt: string, wizardData: Record<string, unkno
 // ============================================================================
 
 export function createImageStrategy(options: CreateImageStrategyOptions): WizardStrategy {
-  const { scenario } = options;
+  const { scenario, creditCost } = options;
 
   return {
     execute: async (input: unknown) => {
@@ -98,6 +98,6 @@ export function createImageStrategy(options: CreateImageStrategyOptions): Wizard
       return { imageUrl: result.imageUrl };
     },
 
-    getCreditCost: () => 1,
+    getCreditCost: () => creditCost,
   };
 }

package/src/domains/generation/wizard/infrastructure/strategies/image-generation.types.ts

@@ -23,4 +23,6 @@ export interface WizardImageResult {
 export interface CreateImageStrategyOptions {
   readonly scenario: WizardScenarioData;
   readonly collectionName?: string;
+  /** Credit cost for this generation - REQUIRED, determined by the app */
+  readonly creditCost: number;
 }

package/src/domains/generation/wizard/infrastructure/strategies/video-generation.strategy.ts

@@ -70,7 +70,7 @@ export async function buildVideoInput(
 }
 
 export function createVideoStrategy(options: CreateVideoStrategyOptions): WizardStrategy {
-  const { scenario } = options;
+  const { scenario, creditCost } = options;
   const videoFeatureType = getVideoFeatureType(scenario.id);
 
   return {
@@ -95,6 +95,6 @@ export function createVideoStrategy(options: CreateVideoStrategyOptions): Wizard
       return { videoUrl: result.videoUrl };
     },
 
-    getCreditCost: () => 1,
+    getCreditCost: () => creditCost,
   };
 }

package/src/domains/generation/wizard/infrastructure/strategies/video-generation.types.ts

@@ -27,6 +27,8 @@ export interface WizardVideoResult {
 export interface CreateVideoStrategyOptions {
   readonly scenario: WizardScenarioData;
   readonly collectionName?: string;
+  /** Credit cost for this generation - REQUIRED, determined by the app */
+  readonly creditCost: number;
 }
 
 export interface PhotoValidationResult {

package/src/domains/generation/wizard/infrastructure/strategies/wizard-strategy.factory.ts

@@ -18,6 +18,8 @@ export type { WizardStrategy } from "./wizard-strategy.types";
 export interface CreateWizardStrategyOptions {
   readonly scenario: WizardScenarioData;
   readonly collectionName?: string;
+  /** Credit cost for this generation - REQUIRED, determined by the app */
+  readonly creditCost: number;
 }
 
 // ============================================================================
@@ -25,14 +27,14 @@ export interface CreateWizardStrategyOptions {
 // ============================================================================
 
 export function createWizardStrategy(options: CreateWizardStrategyOptions): WizardStrategy {
-  const { scenario, collectionName } = options;
+  const { scenario, collectionName, creditCost } = options;
 
   if (scenario.outputType === "image") {
-    return createImageStrategy({ scenario, collectionName });
+    return createImageStrategy({ scenario, collectionName, creditCost });
   }
 
   // Default to video strategy for video outputType or undefined
-  return createVideoStrategy({ scenario, collectionName });
+  return createVideoStrategy({ scenario, collectionName, creditCost });
 }
 
 // ============================================================================

package/src/domains/generation/wizard/presentation/components/GenericWizardFlow.tsx

@@ -24,6 +24,8 @@ export interface GenericWizardFlowProps {
   readonly scenarioId?: string;
   readonly userId?: string;
   readonly alertMessages: AlertMessages;
+  /** Credit cost for this generation - REQUIRED, determined by the app */
+  readonly creditCost: number;
   readonly skipResultStep?: boolean;
   readonly onStepChange?: (stepId: string, stepType: StepType | string) => void;
   readonly onGenerationStart?: (
@@ -49,6 +51,7 @@ export const GenericWizardFlow: React.FC<GenericWizardFlowProps> = (props) => {
     scenarioId,
     userId,
     alertMessages,
+    creditCost,
     skipResultStep = false,
     onStepChange,
     onGenerationStart,
@@ -104,6 +107,7 @@ export const GenericWizardFlow: React.FC<GenericWizardFlowProps> = (props) => {
       validatedScenario={validatedScenario}
       userId={userId}
       alertMessages={alertMessages}
+      creditCost={creditCost}
       skipResultStep={skipResultStep}
       onStepChange={onStepChange}
       onGenerationStart={onGenerationStart}

package/src/domains/generation/wizard/presentation/components/WizardFlow.types.ts

@@ -17,6 +17,8 @@ export interface BaseWizardFlowProps {
   readonly creditBalance: number;
   /** Are credits loaded */
   readonly isCreditsLoaded: boolean;
+  /** Credit cost for this generation - REQUIRED, determined by the app */
+  readonly creditCost: number;
   /** Show auth modal with callback */
   readonly onShowAuthModal: (callback: () => void) => void;
   /** Show paywall */

package/src/domains/generation/wizard/presentation/components/WizardFlowContent.tsx

@@ -25,6 +25,8 @@ export interface WizardFlowContentProps {
   readonly validatedScenario: WizardScenarioData;
   readonly userId?: string;
   readonly alertMessages: AlertMessages;
+  /** Credit cost for this generation - REQUIRED, determined by the app */
+  readonly creditCost: number;
   readonly skipResultStep?: boolean;
   readonly onStepChange?: (stepId: string, stepType: StepType | string) => void;
   readonly onGenerationStart?: (
@@ -49,6 +51,7 @@ export const WizardFlowContent: React.FC<WizardFlowContentProps> = (props) => {
     validatedScenario,
     userId,
     alertMessages,
+    creditCost,
     skipResultStep = false,
     onStepChange,
     onGenerationStart,
@@ -127,6 +130,7 @@ export const WizardFlowContent: React.FC<WizardFlowContentProps> = (props) => {
     userId,
     isGeneratingStep: currentStep?.type === StepType.GENERATING,
     alertMessages,
+    creditCost,
     onSuccess: handlers.handleGenerationComplete,
     onError: onGenerationError,
     onCreditsExhausted,

package/src/domains/generation/wizard/presentation/hooks/useWizardGeneration.ts

@@ -33,6 +33,7 @@ export const useWizardGeneration = (
     userId,
     isGeneratingStep,
     alertMessages,
+    creditCost,
     onSuccess,
     onError,
     onCreditsExhausted,
@@ -45,8 +46,11 @@ export const useWizardGeneration = (
   // Persistence utility - separate from strategy
   const persistence = useMemo(() => createCreationPersistence(), []);
 
-  // Strategy - only handles execution
-  const strategy = useMemo(() => createWizardStrategy({ scenario }), [scenario]);
+  // Strategy - only handles execution, creditCost is passed from app
+  const strategy = useMemo(
+    () => createWizardStrategy({ scenario, creditCost }),
+    [scenario, creditCost],
+  );
 
   const handleSuccess = useCallback(
     async (result: unknown) => {

package/src/domains/generation/wizard/presentation/hooks/wizard-generation.types.ts

@@ -27,6 +27,8 @@ export interface UseWizardGenerationProps {
   readonly isGeneratingStep: boolean;
   /** Required - alert messages for error states */
   readonly alertMessages: AlertMessages;
+  /** Credit cost for this generation - REQUIRED, determined by the app */
+  readonly creditCost: number;
   readonly onSuccess?: (result: unknown) => void;
   readonly onError?: (error: string) => void;
   readonly onCreditsExhausted?: () => void;

package/src/features/image-to-video/presentation/screens/ImageToVideoWizardFlow.tsx

@@ -33,6 +33,7 @@ export const ImageToVideoWizardFlow: React.FC<ImageToVideoWizardFlowProps> = (pr
     hasPremium,
     creditBalance,
     isCreditsLoaded,
+    creditCost,
     onShowAuthModal,
     onShowPaywall,
     onGenerationComplete,
@@ -63,10 +64,10 @@ export const ImageToVideoWizardFlow: React.FC<ImageToVideoWizardFlowProps> = (pr
   const handleGenerationStart = useCallback(
     (_data: Record<string, unknown>, proceed: () => void) => {
       if (!isAuthenticated) { onShowAuthModal(proceed); return; }
-      if (!hasPremium && isCreditsLoaded && creditBalance < 1) { onShowPaywall(); return; }
+      if (!hasPremium && isCreditsLoaded && creditBalance < creditCost) { onShowPaywall(); return; }
       proceed();
     },
-    [isAuthenticated, hasPremium, creditBalance, isCreditsLoaded, onShowAuthModal, onShowPaywall],
+    [isAuthenticated, hasPremium, creditBalance, creditCost, isCreditsLoaded, onShowAuthModal, onShowPaywall],
   );
 
   return (
@@ -76,6 +77,7 @@ export const ImageToVideoWizardFlow: React.FC<ImageToVideoWizardFlowProps> = (pr
         scenario={scenario}
         userId={userId}
         alertMessages={alertMessages ?? defaultAlerts}
+        creditCost={creditCost}
         onGenerationStart={handleGenerationStart}
         onGenerationComplete={onGenerationComplete}
         onGenerationError={onGenerationError}

package/src/features/text-to-image/presentation/screens/TextToImageWizardFlow.tsx

@@ -33,6 +33,7 @@ export const TextToImageWizardFlow: React.FC<TextToImageWizardFlowProps> = (prop
     hasPremium,
     creditBalance,
     isCreditsLoaded,
+    creditCost,
     onShowAuthModal,
     onShowPaywall,
     onGenerationComplete,
@@ -63,10 +64,10 @@ export const TextToImageWizardFlow: React.FC<TextToImageWizardFlowProps> = (prop
   const handleGenerationStart = useCallback(
     (_data: Record<string, unknown>, proceed: () => void) => {
       if (!isAuthenticated) { onShowAuthModal(proceed); return; }
-      if (!hasPremium && isCreditsLoaded && creditBalance < 1) { onShowPaywall(); return; }
+      if (!hasPremium && isCreditsLoaded && creditBalance < creditCost) { onShowPaywall(); return; }
       proceed();
     },
-    [isAuthenticated, hasPremium, creditBalance, isCreditsLoaded, onShowAuthModal, onShowPaywall],
+    [isAuthenticated, hasPremium, creditBalance, creditCost, isCreditsLoaded, onShowAuthModal, onShowPaywall],
   );
 
   return (
@@ -76,6 +77,7 @@ export const TextToImageWizardFlow: React.FC<TextToImageWizardFlowProps> = (prop
         scenario={scenario}
         userId={userId}
         alertMessages={alertMessages ?? defaultAlerts}
+        creditCost={creditCost}
         onGenerationStart={handleGenerationStart}
         onGenerationComplete={onGenerationComplete}
         onGenerationError={onGenerationError}

package/src/features/text-to-video/presentation/screens/TextToVideoWizardFlow.tsx

@@ -33,6 +33,7 @@ export const TextToVideoWizardFlow: React.FC<TextToVideoWizardFlowProps> = (prop
     hasPremium,
     creditBalance,
     isCreditsLoaded,
+    creditCost,
     onShowAuthModal,
     onShowPaywall,
     onGenerationComplete,
@@ -63,10 +64,10 @@ export const TextToVideoWizardFlow: React.FC<TextToVideoWizardFlowProps> = (prop
   const handleGenerationStart = useCallback(
     (_data: Record<string, unknown>, proceed: () => void) => {
       if (!isAuthenticated) { onShowAuthModal(proceed); return; }
-      if (!hasPremium && isCreditsLoaded && creditBalance < 1) { onShowPaywall(); return; }
+      if (!hasPremium && isCreditsLoaded && creditBalance < creditCost) { onShowPaywall(); return; }
       proceed();
     },
-    [isAuthenticated, hasPremium, creditBalance, isCreditsLoaded, onShowAuthModal, onShowPaywall],
+    [isAuthenticated, hasPremium, creditBalance, creditCost, isCreditsLoaded, onShowAuthModal, onShowPaywall],
   );
 
   return (
@@ -76,6 +77,7 @@ export const TextToVideoWizardFlow: React.FC<TextToVideoWizardFlowProps> = (prop
         scenario={scenario}
         userId={userId}
         alertMessages={alertMessages ?? defaultAlerts}
+        creditCost={creditCost}
         onGenerationStart={handleGenerationStart}
         onGenerationComplete={onGenerationComplete}
         onGenerationError={onGenerationError}

package/src/presentation/hooks/generation/orchestrator.ts

@@ -36,7 +36,7 @@ export const useGenerationOrchestrator = <TInput, TResult>(
 
   const offlineStore = useOfflineStore();
   const { showError, showSuccess } = useAlert();
-  const { deductCredit } = useDeductCredit({ userId, onCreditsExhausted });
+  const { deductCredit, checkCredits } = useDeductCredit({ userId, onCreditsExhausted });
 
   useEffect(() => {
     isMountedRef.current = true;
@@ -93,6 +93,17 @@ export const useGenerationOrchestrator = <TInput, TResult>(
       try {
         if (!offlineStore.isOnline) throw createGenerationError("network", alertMessages.networkError);
 
+        // Pre-validate credits before generation to catch concurrent consumption
+        const creditCost = strategy.getCreditCost();
+        const hasEnoughCredits = await checkCredits(creditCost);
+        if (!hasEnoughCredits) {
+          if (typeof __DEV__ !== "undefined" && __DEV__) {
+            console.log("[Orchestrator] Pre-validation: insufficient credits");
+          }
+          onCreditsExhausted?.();
+          throw createGenerationError("credits", alertMessages.creditFailed);
+        }
+
         return await handleModeration({
           input,
           moderation,
@@ -118,7 +129,7 @@ export const useGenerationOrchestrator = <TInput, TResult>(
         isGeneratingRef.current = false;
       }
     },
-    [moderation, alertMessages, offlineStore.isOnline, executeGeneration, showError, onError, handleLifecycleComplete],
+    [moderation, alertMessages, offlineStore.isOnline, strategy, checkCredits, onCreditsExhausted, executeGeneration, showError, onError, handleLifecycleComplete],
   );
 
   const reset = useCallback(() => {