@umituz/react-native-ai-generation-content 1.3.0 → 1.4.0

package/package.json

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

package/src/index.ts

@@ -72,6 +72,21 @@ export type {
   WrapperConfig,
 } from "./infrastructure/services";
 
+// =============================================================================
+// INFRASTRUCTURE LAYER - Middleware Factories
+// =============================================================================
+
+export {
+  createCreditCheckMiddleware,
+  createHistoryTrackingMiddleware,
+} from "./infrastructure/middleware";
+
+export type {
+  CreditCheckConfig,
+  HistoryConfig,
+  HistoryEntry,
+} from "./infrastructure/middleware";
+
 // =============================================================================
 // INFRASTRUCTURE LAYER - Utils
 // =============================================================================

package/src/infrastructure/middleware/credit-check.middleware.ts

@@ -0,0 +1,74 @@
+/**
+ * Credit Check Middleware Factory
+ * Generic credit checking with app-provided config
+ */
+
+import type { GenerationMiddleware } from "../../domain/entities";
+
+export interface CreditCheckConfig {
+  /**
+   * Get credit type from operation type
+   * App-specific logic
+   */
+  getCreditType: (operationType: string) => string;
+
+  /**
+   * Check if user has available credits
+   * App provides implementation
+   */
+  checkCredits: (
+    userId: string | undefined,
+    operationType: string,
+  ) => Promise<{
+    success: boolean;
+    error?: string;
+    creditType?: string;
+  }>;
+
+  /**
+   * Deduct credits after successful generation
+   * App provides implementation
+   */
+  deductCredits: (
+    userId: string | undefined,
+    creditType: string,
+  ) => Promise<void>;
+}
+
+/**
+ * Create credit check middleware
+ * Checks credits before generation, deducts after success
+ */
+export function createCreditCheckMiddleware(
+  config: CreditCheckConfig,
+): GenerationMiddleware {
+  return {
+    async beforeGenerate(context) {
+      const operationType =
+        (context.request.input?.type as string) || "default";
+
+      const result = await config.checkCredits(
+        context.userId,
+        operationType,
+      );
+
+      if (!result.success) {
+        throw new Error(result.error || "credits_exhausted");
+      }
+
+      context.metadata = {
+        ...context.metadata,
+        creditType: result.creditType || config.getCreditType(operationType),
+      };
+    },
+
+    async afterGenerate(context) {
+      if (context.result.success && context.metadata?.creditType) {
+        await config.deductCredits(
+          context.userId,
+          context.metadata.creditType as string,
+        );
+      }
+    },
+  };
+}

package/src/infrastructure/middleware/history-tracking.middleware.ts

@@ -0,0 +1,69 @@
+/**
+ * History Tracking Middleware Factory
+ * Generic history saving with app-provided config
+ */
+
+import type { GenerationMiddleware } from "../../domain/entities";
+
+export interface HistoryEntry {
+  type: string;
+  prompt: string;
+  result: string | null;
+  success: boolean;
+  error?: string;
+  userId: string;
+  metadata?: Record<string, unknown>;
+  timestamp: unknown;
+}
+
+export interface HistoryConfig {
+  /**
+   * Save generation to history
+   * App provides storage implementation (Firestore, API, etc.)
+   */
+  saveToHistory: (entry: HistoryEntry) => Promise<void>;
+
+  /**
+   * Create timestamp for history entry
+   * App-specific (serverTimestamp for Firestore, new Date() for API, etc.)
+   */
+  createTimestamp: () => unknown;
+}
+
+/**
+ * Create history tracking middleware
+ * Saves generation history after completion
+ */
+export function createHistoryTrackingMiddleware(
+  config: HistoryConfig,
+): GenerationMiddleware {
+  return {
+    async afterGenerate(context) {
+      if (!context.userId) {
+        return;
+      }
+
+      try {
+        const operationType =
+          (context.request.input?.type as string) || "default";
+        const prompt =
+          (context.request.input?.prompt as string) || "";
+
+        const entry: HistoryEntry = {
+          type: operationType,
+          prompt,
+          result: context.result.success ? String(context.result.data) : null,
+          success: context.result.success,
+          error: context.result.error,
+          userId: context.userId,
+          metadata: context.result.metadata as Record<string, unknown>,
+          timestamp: config.createTimestamp(),
+        };
+
+        await config.saveToHistory(entry);
+      } catch {
+        // Silent fail
+      }
+    },
+  };
+}

package/src/infrastructure/middleware/index.ts

@@ -0,0 +1,15 @@
+/**
+ * Middleware Factories
+ * Generic middleware with app-provided config
+ */
+
+export {
+  createCreditCheckMiddleware,
+  type CreditCheckConfig,
+} from "./credit-check.middleware";
+
+export {
+  createHistoryTrackingMiddleware,
+  type HistoryConfig,
+  type HistoryEntry,
+} from "./history-tracking.middleware";