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

package/package.json

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

package/src/domain/entities/index.ts

@@ -7,3 +7,4 @@ export * from "./error.types";
 export * from "./generation.types";
 export * from "./polling.types";
 export * from "./progress.types";
+export * from "./middleware.types";

package/src/domain/entities/middleware.types.ts

@@ -0,0 +1,56 @@
+/**
+ * Middleware Types
+ * Generic middleware pattern for generation workflow
+ */
+
+import type { GenerationRequest, GenerationResult } from "./generation.types";
+
+/**
+ * Context passed to middleware hooks
+ */
+export interface MiddlewareContext {
+  request: GenerationRequest;
+  userId?: string;
+  metadata?: Record<string, unknown>;
+}
+
+/**
+ * Result context passed to afterGenerate hook
+ */
+export interface MiddlewareResultContext<T> extends MiddlewareContext {
+  result: GenerationResult<T>;
+}
+
+/**
+ * Middleware hook executed before generation
+ * Can throw error to prevent generation
+ */
+export type BeforeGenerateHook = (
+  context: MiddlewareContext,
+) => Promise<void> | void;
+
+/**
+ * Middleware hook executed after generation (success or failure)
+ * Cannot modify result, used for logging/tracking
+ */
+export type AfterGenerateHook<T = unknown> = (
+  context: MiddlewareResultContext<T>,
+) => Promise<void> | void;
+
+/**
+ * Complete middleware configuration
+ */
+export interface GenerationMiddleware {
+  /** Hook executed before generation starts */
+  beforeGenerate?: BeforeGenerateHook;
+  /** Hook executed after generation completes (success or failure) */
+  afterGenerate?: AfterGenerateHook;
+}
+
+/**
+ * Middleware chain configuration
+ */
+export interface MiddlewareChain {
+  /** Array of middleware to execute in order */
+  middleware: GenerationMiddleware[];
+}

package/src/index.ts

@@ -42,6 +42,12 @@ export type {
   PollingOptions,
   ProgressStageConfig,
   ProgressConfig,
+  MiddlewareContext,
+  MiddlewareResultContext,
+  BeforeGenerateHook,
+  AfterGenerateHook,
+  GenerationMiddleware,
+  MiddlewareChain,
 } from "./domain/entities";
 
 export { DEFAULT_POLLING_CONFIG, DEFAULT_PROGRESS_STAGES } from "./domain/entities";
@@ -55,14 +61,32 @@ export {
   generationOrchestrator,
   pollJob,
   createJobPoller,
+  generationWrapper,
+  createGenerationWrapper,
 } from "./infrastructure/services";
 
 export type {
   OrchestratorConfig,
   PollJobOptions,
   PollJobResult,
+  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";

package/src/infrastructure/services/generation-wrapper.service.ts

@@ -0,0 +1,148 @@
+/**
+ * Generation Wrapper Service
+ * Wraps orchestrator with middleware support
+ * Enables credit checking, moderation, history via middleware pattern
+ */
+
+import type {
+  GenerationRequest,
+  GenerationResult,
+  GenerationMiddleware,
+  MiddlewareContext,
+  MiddlewareResultContext,
+} from "../../domain/entities";
+import { generationOrchestrator } from "./generation-orchestrator.service";
+import type { OrchestratorConfig } from "./generation-orchestrator.service";
+
+export interface WrapperConfig {
+  /** Middleware to execute before/after generation */
+  middleware?: GenerationMiddleware[];
+  /** Orchestrator configuration */
+  orchestratorConfig?: OrchestratorConfig;
+}
+
+class GenerationWrapperService {
+  private middleware: GenerationMiddleware[] = [];
+  private orchestratorConfig?: OrchestratorConfig;
+
+  /**
+   * Configure wrapper with middleware
+   */
+  configure(config: WrapperConfig): void {
+    this.middleware = config.middleware || [];
+    this.orchestratorConfig = config.orchestratorConfig;
+
+    if (this.orchestratorConfig) {
+      generationOrchestrator.configure(this.orchestratorConfig);
+    }
+  }
+
+  /**
+   * Add middleware to chain
+   */
+  use(middleware: GenerationMiddleware): void {
+    this.middleware.push(middleware);
+  }
+
+  /**
+   * Clear all middleware
+   */
+  clearMiddleware(): void {
+    this.middleware = [];
+  }
+
+  /**
+   * Generate with middleware support
+   */
+  async generate<T = unknown>(
+    request: GenerationRequest,
+    userId?: string,
+    metadata?: Record<string, unknown>,
+  ): Promise<GenerationResult<T>> {
+    const startTime = Date.now();
+
+    const context: MiddlewareContext = {
+      request,
+      userId,
+      metadata,
+    };
+
+    try {
+      await this.executeBeforeHooks(context);
+
+      const result = await generationOrchestrator.generate<T>(request);
+
+      const resultContext: MiddlewareResultContext<T> = {
+        ...context,
+        result,
+      };
+
+      await this.executeAfterHooks(resultContext);
+
+      return result;
+    } catch (error) {
+      const errorResult: GenerationResult<T> = {
+        success: false,
+        error: error instanceof Error ? error.message : String(error),
+        metadata: {
+          model: request.model || "unknown",
+          startTime,
+          endTime: Date.now(),
+          duration: Date.now() - startTime,
+        },
+      };
+
+      const resultContext: MiddlewareResultContext<T> = {
+        ...context,
+        result: errorResult,
+      };
+
+      await this.executeAfterHooks(resultContext).catch(() => {
+        // Silent fail on afterHooks error
+      });
+
+      return errorResult;
+    }
+  }
+
+  /**
+   * Execute all beforeGenerate hooks
+   */
+  private async executeBeforeHooks(
+    context: MiddlewareContext,
+  ): Promise<void> {
+    for (const mw of this.middleware) {
+      if (mw.beforeGenerate) {
+        await mw.beforeGenerate(context);
+      }
+    }
+  }
+
+  /**
+   * Execute all afterGenerate hooks
+   */
+  private async executeAfterHooks<T>(
+    context: MiddlewareResultContext<T>,
+  ): Promise<void> {
+    for (const mw of this.middleware) {
+      if (mw.afterGenerate) {
+        await mw.afterGenerate(context);
+      }
+    }
+  }
+}
+
+export const generationWrapper = new GenerationWrapperService();
+
+/**
+ * Create new wrapper instance
+ */
+export function createGenerationWrapper(
+  config?: WrapperConfig,
+): GenerationWrapperService {
+  const wrapper = new GenerationWrapperService();
+  if (config) {
+    wrapper.configure(config);
+  }
+  return wrapper;
+}

package/src/infrastructure/services/index.ts

@@ -7,3 +7,8 @@ export { generationOrchestrator } from "./generation-orchestrator.service";
 export type { OrchestratorConfig } from "./generation-orchestrator.service";
 export { pollJob, createJobPoller } from "./job-poller.service";
 export type { PollJobOptions, PollJobResult } from "./job-poller.service";
+export {
+  generationWrapper,
+  createGenerationWrapper,
+} from "./generation-wrapper.service";
+export type { WrapperConfig } from "./generation-wrapper.service";