@umituz/react-native-ai-generation-content 1.19.9 → 1.20.1

package/package.json

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

package/src/features/text-to-image/presentation/hooks/useGeneration.ts

@@ -1,9 +1,14 @@
 /**
  * Text-to-Image Generation Hook
- * Orchestrates generation with app-provided callbacks
+ * Uses centralized orchestrator for auth, credits, and error handling
  */
 
-import { useState, useCallback } from "react";
+import { useCallback, useMemo } from "react";
+import {
+  useGenerationOrchestrator,
+  type GenerationStrategy,
+  type AlertMessages,
+} from "../../../../presentation/hooks/generation";
 import type {
   TextToImageFormState,
   TextToImageCallbacks,
@@ -11,91 +16,95 @@ import type {
   TextToImageGenerationRequest,
 } from "../../domain/types";
 
-export interface GenerationState {
-  isGenerating: boolean;
-  progress: number;
-  error: string | null;
-}
+declare const __DEV__: boolean;
 
 export interface UseGenerationOptions {
   formState: TextToImageFormState;
   callbacks: TextToImageCallbacks;
+  userId?: string;
   onPromptCleared?: () => void;
 }
 
+export interface GenerationState {
+  isGenerating: boolean;
+  progress: number;
+  error: string | null;
+}
+
 export interface UseGenerationReturn {
   generationState: GenerationState;
   totalCost: number;
   handleGenerate: () => Promise<TextToImageGenerationResult | null>;
 }
 
-const initialState: GenerationState = {
-  isGenerating: false,
-  progress: 0,
-  error: null,
+const DEFAULT_ALERT_MESSAGES: AlertMessages = {
+  networkError: "No internet connection. Please check your network.",
+  policyViolation: "Content not allowed. Please try again.",
+  saveFailed: "Failed to save. Please try again.",
+  creditFailed: "Credit operation failed. Please try again.",
+  unknown: "An error occurred. Please try again.",
+  authRequired: "Please sign in to continue.",
 };
 
-declare const __DEV__: boolean;
-
 export function useGeneration(options: UseGenerationOptions): UseGenerationReturn {
-  const { formState, callbacks, onPromptCleared } = options;
-  const [generationState, setGenerationState] = useState<GenerationState>(initialState);
+  const { formState, callbacks, userId, onPromptCleared } = options;
 
   const totalCost = callbacks.calculateCost(formState.numImages, formState.selectedModel);
 
-  const handleGenerate = useCallback(async (): Promise<TextToImageGenerationResult | null> => {
-    if (typeof __DEV__ !== "undefined" && __DEV__) {
-       
-      console.log("[TextToImage] handleGenerate called");
-    }
-
-    const trimmedPrompt = formState.prompt.trim();
-
-    if (!trimmedPrompt) {
+  // Build strategy for orchestrator
+  const strategy: GenerationStrategy<TextToImageGenerationRequest, string[]> = useMemo(
+    () => ({
+      execute: async (request) => {
+        if (typeof __DEV__ !== "undefined" && __DEV__) {
+          console.log("[TextToImage] Executing generation:", JSON.stringify(request, null, 2));
+        }
+        const result = await callbacks.executeGeneration(request);
+        if (result.success === false) {
+          throw new Error(result.error);
+        }
+        return result.imageUrls;
+      },
+      getCreditCost: () => totalCost,
+    }),
+    [callbacks, totalCost],
+  );
+
+  // Use orchestrator with auth support
+  const { generate, isGenerating, progress, error } = useGenerationOrchestrator(strategy, {
+    userId,
+    alertMessages: DEFAULT_ALERT_MESSAGES,
+    onCreditsExhausted: () => callbacks.onCreditsRequired?.(totalCost),
+    auth: {
+      isAuthenticated: callbacks.isAuthenticated,
+      onAuthRequired: callbacks.onAuthRequired,
+    },
+    onSuccess: (result) => {
+      const imageUrls = result as string[];
       if (typeof __DEV__ !== "undefined" && __DEV__) {
-         
-        console.log("[TextToImage] No prompt provided");
+        console.log("[TextToImage] Success! Generated", imageUrls.length, "image(s)");
       }
-      setGenerationState((prev) => ({ ...prev, error: "Prompt is required" }));
-      return null;
-    }
-
-    const isAuth = callbacks.isAuthenticated();
-    if (typeof __DEV__ !== "undefined" && __DEV__) {
-       
-      console.log("[TextToImage] isAuthenticated:", isAuth);
-    }
-
-    if (!isAuth) {
+      callbacks.onSuccess?.(imageUrls);
+      onPromptCleared?.();
+    },
+    onError: (err) => {
       if (typeof __DEV__ !== "undefined" && __DEV__) {
-         
-        console.log("[TextToImage] Auth required - calling onAuthRequired");
+        console.log("[TextToImage] Error:", err.message);
       }
-      callbacks.onAuthRequired?.();
-      return null;
-    }
+      callbacks.onError?.(err.message);
+    },
+  });
 
-    const affordable = callbacks.canAfford(totalCost);
-    if (typeof __DEV__ !== "undefined" && __DEV__) {
-       
-      console.log("[TextToImage] canAfford:", affordable, "totalCost:", totalCost);
-    }
+  const handleGenerate = useCallback(async (): Promise<TextToImageGenerationResult | null> => {
+    const trimmedPrompt = formState.prompt.trim();
 
-    if (!affordable) {
+    if (!trimmedPrompt) {
       if (typeof __DEV__ !== "undefined" && __DEV__) {
-         
-        console.log("[TextToImage] Credits required - calling onCreditsRequired");
+        console.log("[TextToImage] No prompt provided");
       }
-      callbacks.onCreditsRequired?.(totalCost);
-      return null;
+      callbacks.onError?.("Prompt is required");
+      return { success: false, error: "Prompt is required" };
     }
 
-    if (typeof __DEV__ !== "undefined" && __DEV__) {
-       
-      console.log("[TextToImage] Starting generation...");
-    }
-    setGenerationState({ isGenerating: true, progress: 0, error: null });
-
     const request: TextToImageGenerationRequest = {
       prompt: trimmedPrompt,
       model: formState.selectedModel ?? undefined,
@@ -109,51 +118,22 @@ export function useGeneration(options: UseGenerationOptions): UseGenerationRetur
     };
 
     if (typeof __DEV__ !== "undefined" && __DEV__) {
-       
-      console.log("[TextToImage] Request:", JSON.stringify(request, null, 2));
-    }
-
-    try {
-      const result = await callbacks.executeGeneration(request);
-      if (typeof __DEV__ !== "undefined" && __DEV__) {
-        const logResult = {
-          success: result.success,
-          imageCount: result.success ? result.imageUrls?.length : 0,
-          error: result.success === false ? result.error : undefined,
-        };
-         
-        console.log("[TextToImage] Result:", JSON.stringify(logResult));
-      }
-
-      if (result.success === true) {
-        if (typeof __DEV__ !== "undefined" && __DEV__) {
-           
-          console.log("[TextToImage] Success! Generated", result.imageUrls?.length, "image(s)");
-        }
-        callbacks.onSuccess?.(result.imageUrls);
-        onPromptCleared?.();
-        setGenerationState({ isGenerating: false, progress: 100, error: null });
-      } else {
-        if (typeof __DEV__ !== "undefined" && __DEV__) {
-           
-          console.log("[TextToImage] Generation failed:", result.error);
-        }
-        setGenerationState({ isGenerating: false, progress: 0, error: result.error });
-        callbacks.onError?.(result.error);
-      }
-
-      return result;
-    } catch (error) {
-      const message = error instanceof Error ? error.message : String(error);
-      if (typeof __DEV__ !== "undefined" && __DEV__) {
-         
-        console.error("[TextToImage] Exception:", message);
-      }
-      setGenerationState({ isGenerating: false, progress: 0, error: message });
-      callbacks.onError?.(message);
-      return null;
+      console.log("[TextToImage] Starting generation...");
     }
-  }, [formState, callbacks, totalCost, onPromptCleared]);
 
-  return { generationState, totalCost, handleGenerate };
+    await generate(request);
+
+    // Return result based on orchestrator state
+    return null; // Result handled via callbacks
+  }, [formState, generate, callbacks]);
+
+  return {
+    generationState: {
+      isGenerating,
+      progress,
+      error: error?.message || null,
+    },
+    totalCost,
+    handleGenerate,
+  };
 }

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

@@ -20,6 +20,10 @@ export type {
   GenerationErrorType,
   AlertMessages,
   UseGenerationOrchestratorReturn,
+  AuthCallbacks,
+  ModerationCallbacks,
+  ModerationResult,
+  CreditCallbacks,
 } from "./types";
 
 export type {

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

@@ -1,6 +1,8 @@
 /**
  * Generation Orchestrator
  * Feature-agnostic hook for AI generation with centralized:
+ * - Auth checking (optional)
+ * - Content moderation (optional)
  * - Credit management
  * - Error handling
  * - Alert display
@@ -32,24 +34,100 @@ export const useGenerationOrchestrator = <TInput, TResult>(
   strategy: GenerationStrategy<TInput, TResult>,
   config: GenerationConfig,
 ): UseGenerationOrchestratorReturn<TInput, TResult> => {
-  const { userId, alertMessages, onCreditsExhausted, onSuccess, onError } = config;
+  const { userId, alertMessages, onCreditsExhausted, onSuccess, onError, auth, moderation, credits } = config;
 
   if (typeof __DEV__ !== "undefined" && __DEV__) {
-    console.log("[Orchestrator] Hook initialized:", { userId, hasAlertMessages: !!alertMessages });
+    console.log("[Orchestrator] Hook initialized:", {
+      userId,
+      hasAuth: !!auth,
+      hasModeration: !!moderation,
+      hasCreditsCallbacks: !!credits,
+    });
   }
 
   const [state, setState] = useState<GenerationState<TResult>>(INITIAL_STATE);
   const isGeneratingRef = useRef(false);
+  const pendingInputRef = useRef<TInput | null>(null);
   const offlineStore = useOfflineStore();
   const { showError, showSuccess } = useAlert();
-  const { checkCredits, deductCredit } = useDeductCredit({ userId, onCreditsExhausted });
+  const defaultCredits = useDeductCredit({ userId, onCreditsExhausted });
+
+  // Use provided credit callbacks or default to useDeductCredit hook
+  const checkCredits = credits?.checkCredits ?? defaultCredits.checkCredits;
+  const deductCredit = credits?.deductCredits ?? defaultCredits.deductCredit;
+  const handleCreditsExhausted = credits?.onCreditsExhausted ?? onCreditsExhausted;
+
+  // Core execution logic (after all checks pass)
+  const executeGeneration = useCallback(
+    async (input: TInput) => {
+      const creditCost = strategy.getCreditCost();
+
+      setState((prev) => ({ ...prev, status: "generating", progress: 10 }));
+      if (typeof __DEV__ !== "undefined" && __DEV__) {
+        console.log("[Orchestrator] 🎨 Starting strategy.execute()");
+      }
+
+      const result = await strategy.execute(input, (progress) => {
+        if (typeof __DEV__ !== "undefined" && __DEV__) {
+          console.log("[Orchestrator] 📊 Progress update:", progress);
+        }
+        setState((prev) => ({ ...prev, progress }));
+      });
+
+      if (typeof __DEV__ !== "undefined" && __DEV__) {
+        console.log("[Orchestrator] ✅ strategy.execute() completed");
+      }
+
+      setState((prev) => ({ ...prev, progress: 70 }));
+
+      // Save result
+      if (strategy.save && userId) {
+        setState((prev) => ({ ...prev, status: "saving" }));
+        if (typeof __DEV__ !== "undefined" && __DEV__) {
+          console.log("[Orchestrator] 💾 Saving result...");
+        }
+        try {
+          await strategy.save(result, userId);
+          if (typeof __DEV__ !== "undefined" && __DEV__) {
+            console.log("[Orchestrator] ✅ Save completed");
+          }
+        } catch (saveErr) {
+          if (typeof __DEV__ !== "undefined" && __DEV__) {
+            console.log("[Orchestrator] ❌ Save failed:", saveErr);
+          }
+          throw createGenerationError("save", "Failed to save", saveErr instanceof Error ? saveErr : undefined);
+        }
+      }
+
+      setState((prev) => ({ ...prev, progress: 90 }));
+
+      // Deduct credit
+      if (typeof __DEV__ !== "undefined" && __DEV__) {
+        console.log("[Orchestrator] 💳 Deducting credit:", creditCost);
+      }
+      await deductCredit(creditCost);
+
+      // Success
+      setState({ status: "success", isGenerating: false, progress: 100, result, error: null });
+
+      if (typeof __DEV__ !== "undefined" && __DEV__) {
+        console.log("[Orchestrator] 🎉 Generation SUCCESS");
+      }
+
+      if (alertMessages.success) {
+        void showSuccess("Success", alertMessages.success);
+      }
+
+      onSuccess?.(result);
+    },
+    [strategy, userId, alertMessages, deductCredit, showSuccess, onSuccess],
+  );
 
   const generate = useCallback(
     async (input: TInput) => {
       if (typeof __DEV__ !== "undefined" && __DEV__) {
         console.log("[Orchestrator] 🚀 generate() called");
         console.log("[Orchestrator] Input:", JSON.stringify(input, null, 2).slice(0, 500));
-        console.log("[Orchestrator] isGeneratingRef:", isGeneratingRef.current);
       }
 
       if (isGeneratingRef.current) {
@@ -60,149 +138,141 @@ export const useGenerationOrchestrator = <TInput, TResult>(
       }
 
       isGeneratingRef.current = true;
+      pendingInputRef.current = input;
       setState({ ...INITIAL_STATE, status: "checking", isGenerating: true });
 
-      if (typeof __DEV__ !== "undefined" && __DEV__) {
-        console.log("[Orchestrator] State set to: checking, isGenerating: true");
-      }
-
       try {
-        // Network check
+        // 1. Auth check (optional)
+        if (auth) {
+          setState((prev) => ({ ...prev, status: "authenticating" }));
+          if (typeof __DEV__ !== "undefined" && __DEV__) {
+            console.log("[Orchestrator] 🔐 Checking authentication...");
+          }
+
+          if (!auth.isAuthenticated()) {
+            if (typeof __DEV__ !== "undefined" && __DEV__) {
+              console.log("[Orchestrator] ❌ Not authenticated");
+            }
+            isGeneratingRef.current = false;
+            setState(INITIAL_STATE);
+            auth.onAuthRequired?.();
+            return;
+          }
+          if (typeof __DEV__ !== "undefined" && __DEV__) {
+            console.log("[Orchestrator] ✅ Authentication passed");
+          }
+        }
+
+        // 2. Network check
         if (!offlineStore.isOnline) {
           if (typeof __DEV__ !== "undefined" && __DEV__) {
             console.log("[Orchestrator] ❌ Network check failed - offline");
           }
           throw createGenerationError("network", "No internet connection");
         }
-
         if (typeof __DEV__ !== "undefined" && __DEV__) {
           console.log("[Orchestrator] ✅ Network check passed");
         }
 
-        // Credit check
+        // 3. Credit check
         const creditCost = strategy.getCreditCost();
         if (typeof __DEV__ !== "undefined" && __DEV__) {
           console.log("[Orchestrator] 💳 Credit cost:", creditCost);
         }
 
         const hasCredits = await checkCredits(creditCost);
-        if (typeof __DEV__ !== "undefined" && __DEV__) {
-          console.log("[Orchestrator] 💳 Has credits:", hasCredits);
-        }
-
         if (!hasCredits) {
           if (typeof __DEV__ !== "undefined" && __DEV__) {
             console.log("[Orchestrator] ❌ No credits, opening paywall");
           }
           isGeneratingRef.current = false;
           setState(INITIAL_STATE);
-          onCreditsExhausted?.();
+          handleCreditsExhausted?.();
           return;
         }
-
-        // Start generation
-        setState((prev) => ({ ...prev, status: "generating", progress: 10 }));
         if (typeof __DEV__ !== "undefined" && __DEV__) {
-          console.log("[Orchestrator] 🎨 Starting strategy.execute()");
+          console.log("[Orchestrator] ✅ Credit check passed");
         }
 
-        const result = await strategy.execute(input, (progress) => {
+        // 4. Content moderation (optional)
+        if (moderation) {
+          setState((prev) => ({ ...prev, status: "moderating" }));
           if (typeof __DEV__ !== "undefined" && __DEV__) {
-            console.log("[Orchestrator] 📊 Progress update:", progress);
+            console.log("[Orchestrator] 🛡️ Checking content moderation...");
           }
-          setState((prev) => ({ ...prev, progress }));
-        });
 
-        if (typeof __DEV__ !== "undefined" && __DEV__) {
-          console.log("[Orchestrator] ✅ strategy.execute() completed");
-          console.log("[Orchestrator] Result type:", typeof result);
-        }
-
-        setState((prev) => ({ ...prev, progress: 70 }));
-
-        // Save result
-        if (strategy.save && userId) {
-          setState((prev) => ({ ...prev, status: "saving" }));
-          if (typeof __DEV__ !== "undefined" && __DEV__) {
-            console.log("[Orchestrator] 💾 Saving result...");
-          }
-          try {
-            await strategy.save(result, userId);
+          const moderationResult = await moderation.checkContent(input);
+          if (!moderationResult.allowed && moderationResult.warnings.length > 0) {
             if (typeof __DEV__ !== "undefined" && __DEV__) {
-              console.log("[Orchestrator] ✅ Save completed");
+              console.log("[Orchestrator] ⚠️ Moderation warnings:", moderationResult.warnings);
             }
-          } catch (saveErr) {
-            if (typeof __DEV__ !== "undefined" && __DEV__) {
-              console.log("[Orchestrator] ❌ Save failed:", saveErr);
+
+            if (moderation.onShowWarning) {
+              // Show warning and let user decide
+              moderation.onShowWarning(
+                moderationResult.warnings,
+                () => {
+                  // User cancelled
+                  if (typeof __DEV__ !== "undefined" && __DEV__) {
+                    console.log("[Orchestrator] User cancelled after moderation warning");
+                  }
+                  isGeneratingRef.current = false;
+                  setState(INITIAL_STATE);
+                },
+                async () => {
+                  // User continued - execute generation
+                  if (typeof __DEV__ !== "undefined" && __DEV__) {
+                    console.log("[Orchestrator] User continued after moderation warning");
+                  }
+                  try {
+                    await executeGeneration(input);
+                  } catch (err) {
+                    const error = parseError(err);
+                    setState({ status: "error", isGenerating: false, progress: 0, result: null, error });
+                    void showError("Error", getAlertMessage(error, alertMessages));
+                    onError?.(error);
+                  } finally {
+                    isGeneratingRef.current = false;
+                  }
+                },
+              );
+              return; // Exit here - callback will handle the rest
             }
-            throw createGenerationError(
-              "save",
-              "Failed to save",
-              saveErr instanceof Error ? saveErr : undefined,
-            );
+            // No warning handler - block the request
+            throw createGenerationError("policy", "Content policy violation");
+          }
+          if (typeof __DEV__ !== "undefined" && __DEV__) {
+            console.log("[Orchestrator] ✅ Moderation passed");
           }
         }
 
-        setState((prev) => ({ ...prev, progress: 90 }));
-
-        // Deduct credit
-        if (typeof __DEV__ !== "undefined" && __DEV__) {
-          console.log("[Orchestrator] 💳 Deducting credit:", creditCost);
-        }
-        await deductCredit(creditCost);
-
-        // Success
-        setState({
-          status: "success",
-          isGenerating: false,
-          progress: 100,
-          result,
-          error: null,
-        });
-
-        if (typeof __DEV__ !== "undefined" && __DEV__) {
-          console.log("[Orchestrator] 🎉 Generation SUCCESS");
-        }
-
-        if (alertMessages.success) {
-          void showSuccess("Success", alertMessages.success);
-        }
-
-        onSuccess?.(result);
+        // 5. Execute generation
+        await executeGeneration(input);
       } catch (err) {
         const error = parseError(err);
-
         if (typeof __DEV__ !== "undefined" && __DEV__) {
           console.log("[Orchestrator] ❌ Generation ERROR:", error);
         }
-
-        setState({
-          status: "error",
-          isGenerating: false,
-          progress: 0,
-          result: null,
-          error,
-        });
-
+        setState({ status: "error", isGenerating: false, progress: 0, result: null, error });
         void showError("Error", getAlertMessage(error, alertMessages));
         onError?.(error);
       } finally {
         isGeneratingRef.current = false;
         if (typeof __DEV__ !== "undefined" && __DEV__) {
-          console.log("[Orchestrator] 🏁 generate() finished, isGeneratingRef reset to false");
+          console.log("[Orchestrator] 🏁 generate() finished");
         }
       }
     },
     [
+      auth,
+      moderation,
       strategy,
-      userId,
       alertMessages,
       offlineStore.isOnline,
       checkCredits,
-      deductCredit,
+      handleCreditsExhausted,
+      executeGeneration,
       showError,
-      showSuccess,
-      onSuccess,
       onError,
     ],
   );

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

@@ -6,6 +6,8 @@
 export type OrchestratorStatus =
   | "idle"
   | "checking"
+  | "authenticating"
+  | "moderating"
   | "generating"
   | "saving"
   | "success"
@@ -30,6 +32,43 @@ export interface AlertMessages {
   creditFailed: string;
   unknown: string;
   success?: string;
+  authRequired?: string;
+}
+
+/** Content moderation result */
+export interface ModerationResult {
+  allowed: boolean;
+  warnings: string[];
+}
+
+/** Auth callbacks for features that need authentication */
+export interface AuthCallbacks {
+  /** Check if user is authenticated */
+  isAuthenticated: () => boolean;
+  /** Called when auth is required */
+  onAuthRequired?: () => void;
+}
+
+/** Moderation callbacks for content filtering */
+export interface ModerationCallbacks {
+  /** Check content for policy violations */
+  checkContent: (input: unknown) => Promise<ModerationResult>;
+  /** Show moderation warning with continue/cancel options */
+  onShowWarning?: (
+    warnings: string[],
+    onCancel: () => void,
+    onContinue: () => void,
+  ) => void;
+}
+
+/** Credit callbacks for features that manage credits via callbacks */
+export interface CreditCallbacks {
+  /** Check if user can afford the cost */
+  checkCredits: (cost: number) => Promise<boolean>;
+  /** Deduct credits after successful generation */
+  deductCredits: (cost: number) => Promise<void>;
+  /** Called when credits are exhausted */
+  onCreditsExhausted?: () => void;
 }
 
 export interface GenerationConfig {
@@ -38,6 +77,12 @@ export interface GenerationConfig {
   onCreditsExhausted?: () => void;
   onSuccess?: (result: unknown) => void;
   onError?: (error: GenerationError) => void;
+  /** Optional auth callbacks - if not provided, auth is skipped */
+  auth?: AuthCallbacks;
+  /** Optional moderation callbacks - if not provided, moderation is skipped */
+  moderation?: ModerationCallbacks;
+  /** Optional credit callbacks - if provided, overrides default useDeductCredit */
+  credits?: CreditCallbacks;
 }
 
 export interface GenerationState<TResult> {