@umituz/react-native-ai-generation-content 1.72.6 → 1.72.7

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@umituz/react-native-ai-generation-content",
-  "version": "1.72.6",
+  "version": "1.72.7",
   "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/presentation/hooks/useCreationPersistence.ts

@@ -71,12 +71,9 @@ export function useCreationPersistence(
 
       if (creditCost && onCreditDeduct) {
         onCreditDeduct(creditCost).catch((error) => {
-          // Log credit deduction errors for debugging
           if (typeof __DEV__ !== "undefined" && __DEV__) {
             console.error("[CreationPersistence] Credit deduction failed:", error);
           }
-          // Don't throw - credit deduction failure shouldn't block completion
-          // But we should notify user or retry in production
         });
       }
     },

package/src/domains/creations/presentation/hooks/useProcessingJobsPoller.ts

@@ -89,6 +89,19 @@ export function useProcessingJobsPoller(
         if (typeof __DEV__ !== "undefined" && __DEV__) console.log("[ProcessingJobsPoller] Completed:", creation.id, urls);
 
         const uri = urls.videoUrl || urls.imageUrl || "";
+
+        // Validate that we have a valid URI before marking as completed
+        if (!uri || uri.trim() === "") {
+          if (typeof __DEV__ !== "undefined" && __DEV__) {
+            console.error("[ProcessingJobsPoller] No valid URI in result:", creation.id);
+          }
+          await repository.update(userId, creation.id, {
+            status: CREATION_STATUS.FAILED,
+            metadata: { error: "No valid result URL received" },
+          });
+          return;
+        }
+
         await repository.update(userId, creation.id, {
           status: CREATION_STATUS.COMPLETED,
           uri,
@@ -111,6 +124,12 @@ export function useProcessingJobsPoller(
     }
   };
 
+  // Use ref to always get latest creations
+  const creationsRef = useRef(creations);
+  useEffect(() => {
+    creationsRef.current = creations;
+  }, [creations]);
+
   useEffect(() => {
     if (!enabled || !userId || processingJobIds.length === 0) {
       if (intervalRef.current) {
@@ -120,9 +139,9 @@ export function useProcessingJobsPoller(
       return;
     }
 
-    // Get current jobs at poll time to avoid stale closures
+    // Get current jobs at poll time from ref to avoid stale closures
     const pollCurrentJobs = () => {
-      const currentJobs = creations.filter(
+      const currentJobs = creationsRef.current.filter(
         (c) => c.status === CREATION_STATUS.PROCESSING && c.requestId && c.model,
       );
       currentJobs.forEach((job) => pollJobRef.current?.(job));
@@ -144,7 +163,7 @@ export function useProcessingJobsPoller(
         intervalRef.current = null;
       }
     };
-  }, [enabled, userId, processingJobIds, creations]);
+  }, [enabled, userId, processingJobIds]);
 
   return {
     processingCount: processingJobs.length,

package/src/domains/generation/infrastructure/flow/useFlowStore.ts

@@ -3,6 +3,8 @@
  */
 
 import { createStore } from "@umituz/react-native-design-system";
+
+declare const __DEV__: boolean;
 import type {
   FlowState,
   FlowActions,
@@ -34,7 +36,13 @@ export const createFlowStore = (config: FlowStoreConfig) => {
   if (config.initialStepIndex !== undefined) {
     initialIndex = Math.max(0, Math.min(config.initialStepIndex, config.steps.length - 1));
   } else if (config.initialStepId) {
-    initialIndex = Math.max(0, config.steps.findIndex((s) => s.id === config.initialStepId));
+    const foundIndex = config.steps.findIndex((s) => s.id === config.initialStepId);
+    if (foundIndex === -1) {
+      if (typeof __DEV__ !== "undefined" && __DEV__) {
+        console.warn(`[FlowStore] Step "${config.initialStepId}" not found, using first step`);
+      }
+    }
+    initialIndex = Math.max(0, foundIndex);
   }
 
   const initialStepId = config.steps[initialIndex]?.id ?? "";
@@ -86,28 +94,30 @@ export const createFlowStore = (config: FlowStoreConfig) => {
       setScenario: (scenario: unknown) => set({ selectedScenario: scenario }),
 
       setPartnerImage: (partnerId: string, image: FlowUploadedImageData | undefined) =>
-        set({ partners: { ...get().partners, [partnerId]: image } }),
+        set((state) => ({ partners: { ...state.partners, [partnerId]: image } })),
 
       setPartnerName: (partnerId: string, name: string) =>
-        set({ partnerNames: { ...get().partnerNames, [partnerId]: name } }),
+        set((state) => ({ partnerNames: { ...state.partnerNames, [partnerId]: name } })),
 
       setTextInput: (text: string) => set({ textInput: text }),
       setVisualStyle: (styleId: string) => set({ visualStyle: styleId }),
 
       setSelectedFeatures: (featureType: string, ids: readonly string[]) =>
-        set({ selectedFeatures: { ...get().selectedFeatures, [featureType]: ids } }),
+        set((state) => ({ selectedFeatures: { ...state.selectedFeatures, [featureType]: ids } })),
 
       setCustomData: (key: string, value: unknown) =>
-        set({ customData: { ...get().customData, [key]: value } }),
+        set((state) => ({ customData: { ...state.customData, [key]: value } })),
 
       startGeneration: () =>
         set({ generationStatus: "preparing", generationProgress: 0, generationError: undefined }),
 
-      updateProgress: (progress: number) =>
+      updateProgress: (progress: number) => {
+        const validProgress = Math.max(0, Math.min(100, progress));
         set({
-          generationProgress: progress,
-          generationStatus: progress >= 100 ? "completed" : progress > 0 ? "generating" : "preparing",
-        }),
+          generationProgress: validProgress,
+          generationStatus: validProgress >= 100 ? "completed" : validProgress > 0 ? "generating" : "preparing",
+        });
+      },
 
       setResult: (result: unknown) =>
         set({ generationResult: result, generationStatus: "completed", generationProgress: 100 }),

package/src/infrastructure/validation/base-validator.ts

@@ -136,6 +136,10 @@ export function validateBase64(input: unknown): ValidationResult {
     return { isValid: false, errors: ["Input must be a string"] };
   }
 
+  if (input.length === 0) {
+    return { isValid: false, errors: ["Base64 string cannot be empty"] };
+  }
+
   const base64Regex = /^[A-Za-z0-9+/]*={0,2}$/;
   if (!base64Regex.test(input)) {
     return { isValid: false, errors: ["Invalid base64 format"] };

package/src/presentation/hooks/flow-state.utils.ts

@@ -48,6 +48,12 @@ export function updatePhotoStep(
   stepIndex: number,
   updates: Partial<PhotoStepData>,
 ): GenerationFlowState {
+  // Validate bounds to prevent sparse arrays and undefined access
+  if (stepIndex < 0 || stepIndex >= state.photoSteps.length) {
+    console.warn(`[FlowState] Invalid stepIndex ${stepIndex}, ignoring update`);
+    return state;
+  }
+
   const newPhotoSteps = [...state.photoSteps];
   newPhotoSteps[stepIndex] = {
     ...newPhotoSteps[stepIndex],

package/src/presentation/hooks/use-generation.ts

@@ -109,6 +109,9 @@ export function useGeneration<T = unknown>(
         if (!abortRef.current && !abortControllerRef.current?.signal.aborted) {
           setIsGenerating(false);
         }
+        if (abortControllerRef.current && !abortControllerRef.current.signal.aborted) {
+          abortControllerRef.current.abort();
+        }
         abortControllerRef.current = null;
       }
     },