@umituz/react-native-ai-generation-content 1.17.307 → 1.17.308

package/package.json

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

package/src/features/couple-future/presentation/hooks/useCoupleFutureGeneration.ts

@@ -1,5 +1,9 @@
+/**
+ * useCoupleFutureGeneration Hook
+ * Couple future generation with centralized credit management
+ */
+
 import { useCallback } from "react";
-import { useDeductCredit } from "@umituz/react-native-subscription";
 import { usePhotoGeneration } from "../../../../presentation/hooks/usePhotoGeneration";
 import { executeCoupleFuture } from "../../infrastructure/executor";
 import type { CoupleFutureInput } from "../../domain/types";
@@ -20,6 +24,7 @@ export interface UseCoupleFutureGenerationConfig<
     input: TInput,
   ) => Promise<TResult> | TResult;
   buildCreation?: (result: TResult, input: TInput) => Creation | null;
+  onCreditsExhausted?: () => void;
   onSuccess?: (result: TResult) => void;
   onError?: (error: string) => void;
   alertMessages: {
@@ -41,19 +46,25 @@ export const useCoupleFutureGeneration = <
     userId,
     processResult,
     buildCreation,
+    onCreditsExhausted,
     onSuccess,
     onError,
     alertMessages,
   } = config;
 
-  const { checkCredits, deductCredit } = useDeductCredit({ userId });
   const repository = useCallback(
     () => createCreationsRepository("creations"),
     [],
   );
 
   const generationConfig: PhotoGenerationConfig<TInput, TResult, void> = {
-    generate: async (input: TInput, onProgress?: (progress: number) => void) => {
+    userId,
+    creditCost: 1,
+    onCreditsExhausted,
+    generate: async (
+      input: TInput,
+      onProgress?: (progress: number) => void,
+    ) => {
       const result = await executeCoupleFuture(
         {
           partnerABase64: input.partnerABase64,
@@ -69,7 +80,6 @@ export const useCoupleFutureGeneration = <
 
       return processResult(result.imageUrl, input);
     },
-
     save: async (result: TResult, input: TInput) => {
       if (!userId || !buildCreation) {
         return;
@@ -79,9 +89,6 @@ export const useCoupleFutureGeneration = <
         await repository().create(userId, creation);
       }
     },
-
-    checkCredits: () => checkCredits(1),
-    deductCredits: () => deductCredit(1).then(() => {}),
     onSuccess,
     onError: (error: PhotoGenerationError) => {
       onError?.(error.message);

package/src/presentation/hooks/index.ts

@@ -29,6 +29,13 @@ export type {
   UsePhotoGenerationReturn,
 } from "./usePhotoGeneration";
 
+export { useGenerationCredits, CreditError } from "./useGenerationCredits";
+export type {
+  UseGenerationCreditsParams,
+  UseGenerationCreditsReturn,
+  GenerationCreditsError,
+} from "./useGenerationCredits";
+
 export type {
   PhotoGenerationInput,
   PhotoGenerationResult,

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

@@ -29,14 +29,25 @@ export interface AlertMessages {
 }
 
 export interface PhotoGenerationConfig<TInput, TResult, TSaveInput> {
+  /** User ID for credit operations - required for credit check/deduct */
+  userId: string | undefined;
+  /** Credit cost per generation (default: 1) */
+  creditCost?: number;
+  /** Generation function */
   generate: (input: TInput, onProgress?: (progress: number) => void) => Promise<TResult>;
+  /** Save result function */
   save?: (result: TResult, input: TInput) => Promise<TSaveInput>;
+  /** Build metadata for tracking */
   buildMetadata?: (input: TInput) => Record<string, unknown>;
-  checkCredits?: () => Promise<boolean>;
-  deductCredits?: () => Promise<void>;
+  /** Called when credits are exhausted */
+  onCreditsExhausted?: () => void;
+  /** Success callback */
   onSuccess?: (result: TResult) => void;
+  /** Error callback */
   onError?: (error: PhotoGenerationError) => void;
+  /** Save complete callback */
   onSaveComplete?: (saveResult: TSaveInput) => void;
+  /** Alert messages for errors */
   alertMessages: AlertMessages;
 }
 

package/src/presentation/hooks/useGenerationCredits.ts

@@ -0,0 +1,77 @@
+/**
+ * useGenerationCredits Hook
+ * Centralized credit management for all AI generation features
+ * Provides server-side credit validation before generation
+ */
+
+import { useCallback } from "react";
+import { useDeductCredit } from "@umituz/react-native-subscription";
+
+export interface UseGenerationCreditsParams {
+  userId: string | undefined;
+  onCreditsExhausted?: () => void;
+}
+
+export interface GenerationCreditsError {
+  type: "no_credits" | "deduct_failed";
+  message: string;
+}
+
+export interface UseGenerationCreditsReturn {
+  /** Server-side credit check */
+  checkCredits: (cost?: number) => Promise<boolean>;
+  /** Deduct credits after successful generation */
+  deductCredits: (cost?: number) => Promise<boolean>;
+  /** Whether credit operation is in progress */
+  isProcessing: boolean;
+  /** Execute generation with automatic credit check and deduct */
+  withCredits: <T>(
+    generate: () => Promise<T>,
+    cost?: number,
+  ) => Promise<T>;
+}
+
+export const useGenerationCredits = ({
+  userId,
+  onCreditsExhausted,
+}: UseGenerationCreditsParams): UseGenerationCreditsReturn => {
+  const { checkCredits, deductCredit, isDeducting } = useDeductCredit({
+    userId,
+    onCreditsExhausted,
+  });
+
+  const withCredits = useCallback(
+    async <T>(generate: () => Promise<T>, cost: number = 1): Promise<T> => {
+      const hasCredits = await checkCredits(cost);
+      if (!hasCredits) {
+        throw new CreditError("no_credits", "Insufficient credits");
+      }
+
+      const result = await generate();
+
+      await deductCredit(cost);
+
+      return result;
+    },
+    [checkCredits, deductCredit],
+  );
+
+  return {
+    checkCredits,
+    deductCredits: deductCredit,
+    isProcessing: isDeducting,
+    withCredits,
+  };
+};
+
+class CreditError extends Error {
+  constructor(
+    public type: GenerationCreditsError["type"],
+    message: string,
+  ) {
+    super(message);
+    this.name = "CreditError";
+  }
+}
+
+export { CreditError };

package/src/presentation/hooks/usePhotoGeneration.ts

@@ -1,10 +1,12 @@
 /**
  * usePhotoGeneration Hook
  * Generic hook for photo-based AI generation workflows
+ * Uses centralized credit management
  */
 
 import { useState, useCallback, useRef } from "react";
 import { useOfflineStore, useAlert } from "@umituz/react-native-design-system";
+import { useGenerationCredits, CreditError } from "./useGenerationCredits";
 import type {
   PhotoGenerationConfig,
   PhotoGenerationState,
@@ -12,7 +14,8 @@ import type {
   PhotoGenerationStatus,
 } from "./photo-generation.types";
 
-export interface UsePhotoGenerationReturn<TInput, TResult> extends PhotoGenerationState<TResult> {
+export interface UsePhotoGenerationReturn<TInput, TResult>
+  extends PhotoGenerationState<TResult> {
   generate: (input: TInput) => Promise<void>;
   reset: () => void;
   status: PhotoGenerationStatus;
@@ -22,10 +25,11 @@ export const usePhotoGeneration = <TInput, TResult, TSaveInput = unknown>(
   config: PhotoGenerationConfig<TInput, TResult, TSaveInput>,
 ): UsePhotoGenerationReturn<TInput, TResult> => {
   const {
+    userId,
+    creditCost = 1,
     generate: generateFn,
     save: saveFn,
-    checkCredits,
-    deductCredits,
+    onCreditsExhausted,
     onSuccess,
     onError,
     onSaveComplete,
@@ -44,6 +48,11 @@ export const usePhotoGeneration = <TInput, TResult, TSaveInput = unknown>(
   const offlineStore = useOfflineStore();
   const { showError } = useAlert();
 
+  const { checkCredits, deductCredits } = useGenerationCredits({
+    userId,
+    onCreditsExhausted,
+  });
+
   const createError = useCallback(
     (
       type: PhotoGenerationError["type"],
@@ -68,31 +77,24 @@ export const usePhotoGeneration = <TInput, TResult, TSaveInput = unknown>(
       setStatus("validating");
 
       try {
-        // Check network connectivity
         if (!offlineStore.isOnline) {
           throw createError("network_error", "No internet connection");
         }
 
-        // Check credits
-        if (checkCredits) {
-          const hasCredits = await checkCredits();
-          if (!hasCredits) {
-            throw createError("credit_failed", "Insufficient credits");
-          }
+        const hasCredits = await checkCredits(creditCost);
+        if (!hasCredits) {
+          throw createError("credit_failed", "Insufficient credits");
         }
 
         setStatus("generating");
         setState((prev) => ({ ...prev, progress: 20 }));
 
-        // Generate without timeout - let AI provider handle its own timeout
-        // Pass progress callback to allow provider to report real progress
         const result = await generateFn(input, (newProgress) => {
           setState((prev) => ({ ...prev, progress: newProgress }));
         });
 
         setState((prev) => ({ ...prev, progress: 60 }));
 
-        // Save result
         if (saveFn) {
           setStatus("saving");
           try {
@@ -109,14 +111,7 @@ export const usePhotoGeneration = <TInput, TResult, TSaveInput = unknown>(
 
         setState((prev) => ({ ...prev, progress: 80 }));
 
-        // Deduct credits after successful generation
-        if (deductCredits) {
-          try {
-            await deductCredits();
-          } catch {
-            // Silently fail credit deduction as generation succeeded
-          }
-        }
+        await deductCredits(creditCost);
 
         setState({
           isGenerating: false,
@@ -129,13 +124,28 @@ export const usePhotoGeneration = <TInput, TResult, TSaveInput = unknown>(
       } catch (err: unknown) {
         let generationError: PhotoGenerationError;
 
-        if (err && typeof err === "object" && "type" in err && "message" in err) {
+        if (err instanceof CreditError) {
+          generationError = createError("credit_failed", err.message);
+        } else if (
+          err &&
+          typeof err === "object" &&
+          "type" in err &&
+          "message" in err
+        ) {
           generationError = err as PhotoGenerationError;
         } else if (err instanceof Error) {
           if (err.name === "ContentPolicyViolationError") {
-            generationError = createError("policy_violation", "Content policy violation", err);
+            generationError = createError(
+              "policy_violation",
+              "Content policy violation",
+              err,
+            );
           } else {
-            generationError = createError("unknown", err.message || "Generation failed", err);
+            generationError = createError(
+              "unknown",
+              err.message || "Generation failed",
+              err,
+            );
           }
         } else {
           generationError = createError("unknown", "Generation failed");
@@ -150,11 +160,15 @@ export const usePhotoGeneration = <TInput, TResult, TSaveInput = unknown>(
         setStatus("error");
 
         const errorMessage =
-          generationError.type === "network_error" ? alertMessages.networkError :
-            generationError.type === "policy_violation" ? alertMessages.policyViolation :
-              generationError.type === "save_failed" ? alertMessages.saveFailed :
-                generationError.type === "credit_failed" ? alertMessages.creditFailed :
-                  alertMessages.unknown;
+          generationError.type === "network_error"
+            ? alertMessages.networkError
+            : generationError.type === "policy_violation"
+              ? alertMessages.policyViolation
+              : generationError.type === "save_failed"
+                ? alertMessages.saveFailed
+                : generationError.type === "credit_failed"
+                  ? alertMessages.creditFailed
+                  : alertMessages.unknown;
 
         void showError("Error", errorMessage);
         onError?.(generationError);
@@ -167,13 +181,14 @@ export const usePhotoGeneration = <TInput, TResult, TSaveInput = unknown>(
       saveFn,
       checkCredits,
       deductCredits,
+      creditCost,
       onSuccess,
       onError,
       onSaveComplete,
       createError,
       offlineStore,
       alertMessages,
-      showError
+      showError,
     ],
   );