@umituz/react-native-ai-generation-content 1.83.46 → 1.83.48

package/package.json

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

package/src/domains/creations/domain/constants/creation-fields.constants.ts

@@ -49,6 +49,7 @@ export const CREATION_FIELDS = {
   // AI provider metadata
   REQUEST_ID: "requestId" as const,
   MODEL: "model" as const,
+  PROVIDER: "provider" as const,
 } as const;
 
 /** Union type of all field names */

package/src/domains/creations/domain/entities/Creation.ts

@@ -25,8 +25,8 @@ export interface Creation {
   readonly uri: string;
   readonly type: string;
   readonly prompt?: string;
+  readonly provider?: string;
   readonly metadata?: Record<string, unknown>;
-  readonly originalUri?: string;
   readonly createdAt: Date;
   readonly isShared: boolean;
   readonly isFavorite: boolean;
@@ -53,12 +53,8 @@ interface FirebaseTimestamp {
 export interface CreationDocument {
   readonly uri?: string;
   readonly prompt?: string;
+  readonly provider?: string;
   readonly metadata?: Record<string, unknown>;
-  readonly originalImage?: string;
-  readonly originalImageUrl?: string;
-  readonly transformedImage?: string;
-  readonly transformedImageUrl?: string;
-  readonly transformationType?: string;
   readonly type?: string;
   readonly status?: string;
   readonly output?: CreationOutput | null;
@@ -92,21 +88,15 @@ export function mapDocumentToCreation(
 ): Creation {
   const creationDate = toDate(data.createdAt) ?? new Date();
 
-  // Get URI from output or direct fields
-  const uri = data.output?.imageUrl ||
-              data.output?.videoUrl ||
-              data.transformedImageUrl ||
-              data.transformedImage ||
-              data.uri ||
-              "";
+  const uri = data.output?.imageUrl || data.output?.videoUrl || data.uri || "";
 
   return {
     id,
     uri,
-    type: data.transformationType || data.type || "unknown",
+    type: data.type || "unknown",
     prompt: data.prompt,
+    provider: data.provider,
     metadata: data.metadata,
-    originalUri: data.originalImageUrl || data.originalImage,
     createdAt: creationDate,
     isShared: data.isShared ?? false,
     isFavorite: data.isFavorite ?? false,

package/src/domains/creations/infrastructure/repositories/creation-create.operations.ts

@@ -25,6 +25,7 @@ export async function createCreation(
     ...(creation.status !== undefined && { status: creation.status }),
     ...(creation.output !== undefined && { output: creation.output }),
     ...(creation.prompt !== undefined && { prompt: creation.prompt }),
+    ...(creation.provider !== undefined && { provider: creation.provider }),
     ...(creation.requestId !== undefined && { requestId: creation.requestId }),
     ...(creation.model !== undefined && { model: creation.model }),
     ...(creation.startedAt !== undefined && { startedAt: creation.startedAt }),

package/src/domains/generation/wizard/infrastructure/utils/creation-save-operations.ts

@@ -25,6 +25,7 @@ export async function saveAsProcessing(
     uri: "",
     type: data.scenarioId,
     prompt: data.prompt,
+    provider: data.provider,
     status: "processing" as const,
     createdAt: new Date(),
     startedAt,
@@ -39,7 +40,6 @@ export async function saveAsProcessing(
       ...(data.resolution && { resolution: data.resolution }),
       ...(data.creditCost && { creditCost: data.creditCost }),
       ...(data.aspectRatio && { aspectRatio: data.aspectRatio }),
-      ...(data.provider && { provider: data.provider }),
       ...(data.outputType && { outputType: data.outputType }),
     },
   });

package/src/domains/generation/wizard/infrastructure/utils/creation-update-operations.ts

@@ -4,6 +4,7 @@
  */
 
 import type { ICreationsRepository } from "../../../../creations/domain/repositories";
+import type { Creation, CreationOutput } from "../../../../creations/domain/entities/Creation";
 import type { CompletedCreationData } from "./creation-persistence.types";
 
 
@@ -16,10 +17,11 @@ export async function updateToCompleted(
   creationId: string,
   data: CompletedCreationData
 ): Promise<void> {
-  const output: { imageUrl?: string; videoUrl?: string; thumbnailUrl?: string } = {};
-  if (data.imageUrl) output.imageUrl = data.imageUrl;
-  if (data.videoUrl) output.videoUrl = data.videoUrl;
-  if (data.thumbnailUrl) output.thumbnailUrl = data.thumbnailUrl;
+  const output: CreationOutput = {
+    ...(data.imageUrl && { imageUrl: data.imageUrl }),
+    ...(data.videoUrl && { videoUrl: data.videoUrl }),
+    ...(data.thumbnailUrl && { thumbnailUrl: data.thumbnailUrl }),
+  };
 
   const completedAt = new Date();
   const durationMs =
@@ -33,7 +35,7 @@ export async function updateToCompleted(
     output,
     completedAt,
     ...(durationMs !== undefined && { durationMs }),
-  } as Partial<import("../../../../creations/domain/entities/Creation").Creation>);
+  } as Partial<Creation>);
 
   if (typeof __DEV__ !== "undefined" && __DEV__) {
     console.log("[CreationPersistence] Updated to completed", { creationId });
@@ -53,7 +55,7 @@ export async function updateToFailed(
     status: "failed" as const,
     metadata: { error },
     completedAt: new Date(),
-  } as Partial<import("../../../../creations/domain/entities/Creation").Creation>);
+  } as Partial<Creation>);
 
   if (typeof __DEV__ !== "undefined" && __DEV__) {
     console.log("[CreationPersistence] Updated to failed", { creationId, error });

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

@@ -32,6 +32,8 @@ export interface GenericWizardFlowProps {
   readonly creditCost: number;
   /** Calculator function provided by APP - package calls this to get dynamic cost */
   readonly calculateCredits?: CreditCalculatorFn;
+  /** Called after successful generation to deduct credits — provided by the app */
+  readonly deductCredits?: (cost: number) => Promise<boolean>;
   readonly skipResultStep?: boolean;
   readonly onStepChange?: (stepId: string, stepType: StepType | string) => void;
   readonly onGenerationStart?: (
@@ -61,6 +63,7 @@ export const GenericWizardFlow: React.FC<GenericWizardFlowProps> = (props) => {
     alertMessages,
     creditCost,
     calculateCredits,
+    deductCredits,
     skipResultStep = false,
     onStepChange,
     onGenerationStart,
@@ -119,6 +122,7 @@ export const GenericWizardFlow: React.FC<GenericWizardFlowProps> = (props) => {
       alertMessages={alertMessages}
       creditCost={creditCost}
       calculateCredits={calculateCredits}
+      deductCredits={deductCredits}
       skipResultStep={skipResultStep}
       onStepChange={onStepChange}
       onGenerationStart={onGenerationStart}

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

@@ -34,6 +34,8 @@ interface WizardFlowContentProps {
   readonly creditCost: number;
   /** Calculator function provided by APP - package calls this to get dynamic cost */
   readonly calculateCredits?: CreditCalculatorFn;
+  /** Called after successful generation to deduct credits — provided by the app */
+  readonly deductCredits?: (cost: number) => Promise<boolean>;
   readonly skipResultStep?: boolean;
   readonly onStepChange?: (stepId: string, stepType: StepType | string) => void;
   readonly onGenerationStart?: (
@@ -60,8 +62,9 @@ export const WizardFlowContent: React.FC<WizardFlowContentProps> = (props) => {
     modelConfig,
     userId,
     alertMessages,
-    creditCost, // Still needed for initial feature gate in parent
-    calculateCredits, // Calculator function from APP
+    creditCost,
+    calculateCredits,
+    deductCredits,
     skipResultStep = false,
     onStepChange,
     onGenerationStart,
@@ -188,6 +191,7 @@ export const WizardFlowContent: React.FC<WizardFlowContentProps> = (props) => {
     isGeneratingStep: currentStep?.type === StepType.GENERATING,
     alertMessages,
     creditCost: calculatedCreditCost,
+    deductCredits,
     onSuccess: handlers.handleGenerationComplete,
     onError: handlers.handleGenerationError,
     onCreditsExhausted,

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

@@ -12,6 +12,7 @@ export interface UseVideoQueueGenerationProps {
   readonly persistence: CreationPersistence;
   readonly strategy: WizardStrategy;
   readonly creditCost?: number;
+  readonly deductCredits?: (cost: number) => Promise<boolean>;
   readonly onSuccess?: (result: unknown) => void;
   readonly onError?: (error: string) => void;
 }

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

@@ -20,6 +20,7 @@ interface UsePhotoBlockingGenerationProps {
   readonly persistence: CreationPersistence;
   readonly strategy: WizardStrategy;
   readonly creditCost?: number;
+  readonly deductCredits?: (cost: number) => Promise<boolean>;
   readonly alertMessages: AlertMessages;
   readonly onSuccess?: (result: unknown) => void;
   readonly onError?: (error: string) => void;
@@ -39,6 +40,7 @@ export function usePhotoBlockingGeneration(
     scenario,
     persistence,
     creditCost,
+    deductCredits,
     strategy,
     alertMessages,
     onSuccess,
@@ -67,9 +69,19 @@ export function usePhotoBlockingGeneration(
       }
 
       creationIdRef.current = null;
+
+      // Deduct credits after successful generation
+      if (deductCredits && creditCost) {
+        await deductCredits(creditCost).catch((err) => {
+          if (typeof __DEV__ !== "undefined" && __DEV__) {
+            console.error("[PhotoBlockingGeneration] deductCredits error:", err);
+          }
+        });
+      }
+
       onSuccess?.(result);
     },
-    [userId, persistence, onSuccess],
+    [userId, persistence, deductCredits, creditCost, onSuccess],
   );
 
   const handleError = useCallback(

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

@@ -16,7 +16,7 @@ import type {
 } from "./use-video-queue-generation.types";
 
 export function useVideoQueueGeneration(props: UseVideoQueueGenerationProps): UseVideoQueueGenerationReturn {
-  const { userId, scenario, persistence, strategy, creditCost, onSuccess, onError } = props;
+  const { userId, scenario, persistence, strategy, creditCost, deductCredits, onSuccess, onError } = props;
 
   const creationIdRef = useRef<string | null>(null);
   const requestIdRef = useRef<string | null>(null);
@@ -111,6 +111,15 @@ export function useVideoQueueGeneration(props: UseVideoQueueGenerationProps): Us
 
       resetRefs();
 
+      // Deduct credits after successful generation
+      if (deductCredits && creditCost) {
+        await deductCredits(creditCost).catch((err) => {
+          if (typeof __DEV__ !== "undefined" && __DEV__) {
+            console.error("[VideoQueue] deductCredits error:", err);
+          }
+        });
+      }
+
       if (typeof __DEV__ !== "undefined" && __DEV__) {
         console.log("[VideoQueue] 🎯 Calling onSuccess callback now...", { persistenceSucceeded });
       }
@@ -120,7 +129,7 @@ export function useVideoQueueGeneration(props: UseVideoQueueGenerationProps): Us
         console.log("[VideoQueue] ✅ onSuccess callback completed");
       }
     },
-    [userId, persistence, onSuccess, onError, resetRefs, clearPolling],
+    [userId, persistence, deductCredits, creditCost, onSuccess, onError, resetRefs, clearPolling],
   );
 
   const handleError = useCallback(

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

@@ -28,6 +28,7 @@ export const useWizardGeneration = (props: UseWizardGenerationProps): UseWizardG
     isGeneratingStep,
     alertMessages,
     creditCost,
+    deductCredits,
     onSuccess,
     onError,
     onCreditsExhausted,
@@ -53,6 +54,7 @@ export const useWizardGeneration = (props: UseWizardGenerationProps): UseWizardG
     persistence,
     strategy,
     creditCost,
+    deductCredits,
     onSuccess,
     onError,
   });
@@ -63,6 +65,7 @@ export const useWizardGeneration = (props: UseWizardGenerationProps): UseWizardG
     persistence,
     strategy,
     creditCost,
+    deductCredits,
     alertMessages,
     onSuccess,
     onError,

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

@@ -36,6 +36,8 @@ export interface UseWizardGenerationProps {
   readonly alertMessages: AlertMessages;
   /** Credit cost for this generation - REQUIRED, determined by the app */
   readonly creditCost: number;
+  /** Called after successful generation to deduct credits — provided by the app */
+  readonly deductCredits?: (cost: number) => Promise<boolean>;
   readonly onSuccess?: (result: unknown) => void;
   readonly onError?: (error: string) => void;
   readonly onCreditsExhausted?: () => void;