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

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@umituz/react-native-ai-generation-content",
-  "version": "1.44.4",
+  "version": "1.44.5",
   "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

@@ -79,8 +79,24 @@ function applyStyleEnhancements(prompt: string, wizardData: Record<string, unkno
 // Strategy Factory
 // ============================================================================
 
+/**
+ * Calculate credit cost based on scenario input type
+ * - dual: 2 credits (2 photos)
+ * - single: 1 credit (1 photo)
+ * - text: 1 credit (text-to-image)
+ */
+function calculateCreditCost(inputType?: string): number {
+  if (inputType === "dual") {
+    return 2;
+  }
+  return 1;
+}
+
 export function createImageStrategy(options: CreateImageStrategyOptions): WizardStrategy {
-  const { scenario } = options;
+  const { scenario, creditCost } = options;
+
+  // Use explicit creditCost if provided, otherwise calculate from inputType
+  const resolvedCreditCost = creditCost ?? calculateCreditCost(scenario.inputType);
 
   return {
     execute: async (input: unknown) => {
@@ -98,6 +114,6 @@ export function createImageStrategy(options: CreateImageStrategyOptions): Wizard
       return { imageUrl: result.imageUrl };
     },
 
-    getCreditCost: () => 1,
+    getCreditCost: () => resolvedCreditCost,
   };
 }

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;
+  /** Override credit cost (defaults to inputType-based calculation: dual=2, single/text=1) */
+  readonly creditCost?: number;
 }

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

@@ -69,10 +69,26 @@ export async function buildVideoInput(
   };
 }
 
+/**
+ * Calculate credit cost based on scenario input type
+ * - dual: 2 credits (2 photos)
+ * - single: 1 credit (1 photo)
+ * - text: 1 credit (text-to-video)
+ */
+function calculateCreditCost(inputType?: string): number {
+  if (inputType === "dual") {
+    return 2;
+  }
+  return 1;
+}
+
 export function createVideoStrategy(options: CreateVideoStrategyOptions): WizardStrategy {
-  const { scenario } = options;
+  const { scenario, creditCost } = options;
   const videoFeatureType = getVideoFeatureType(scenario.id);
 
+  // Use explicit creditCost if provided, otherwise calculate from inputType
+  const resolvedCreditCost = creditCost ?? calculateCreditCost(scenario.inputType);
+
   return {
     execute: async (input: unknown) => {
       const videoInput = input as WizardVideoInput;
@@ -95,6 +111,6 @@ export function createVideoStrategy(options: CreateVideoStrategyOptions): Wizard
       return { videoUrl: result.videoUrl };
     },
 
-    getCreditCost: () => 1,
+    getCreditCost: () => resolvedCreditCost,
   };
 }

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;
+  /** Override credit cost (defaults to inputType-based calculation: dual=2, single/text=1) */
+  readonly creditCost?: number;
 }
 
 export interface PhotoValidationResult {

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(() => {