feat: Implement payment provider abstraction, Drizzle ORM database service, and core application utilities and stores.

Author: SanjeevSaniel

package-lock.json

@@ -54,6 +54,7 @@
     "@sentry/nextjs": "^10.27.0",
     "@stripe/stripe-js": "^8.0.0",
     "@tabler/icons-react": "^3.34.1",
+    "@upstash/ratelimit": "^2.0.5",
     "@vercel/kv": "^3.0.0",
     "aceternity-ui": "^0.2.2",
     "ai": "^5.0.28",
@@ -75,6 +76,8 @@
     "next": "15.5.2",
     "next-themes": "^0.4.6",
     "openai": "^5.16.0",
+    "pino": "^10.1.0",
+    "pino-pretty": "^13.1.3",
     "razorpay": "^2.9.6",
     "react": "19.1.0",
     "react-dom": "19.1.0",
@@ -101,6 +104,9 @@
     "eslint": "^9",
     "eslint-config-next": "15.5.2",
     "tailwindcss": "^4.1.13",
-    "typescript": "^5"
+    "typescript": "^5",
+    "vitest": "^2.1.8",
+    "@testing-library/react": "^16.0.1",
+    "@testing-library/jest-dom": "^6.6.3"
   }
 }

package.json

@@ -127,14 +127,17 @@ export function cleanupExpiredImages(): number {
 /**
  * Get session info (without the image data)
  */
-export function getSessionInfo(chatId: string): Omit<ImageSession, 'imageBase64'> | null {
+export function getSessionInfo(
+  chatId: string,
+): Omit<ImageSession, 'imageBase64'> | null {
   const session = imageSessionStore.get(chatId);
 
   if (!session) {
     return null;
   }
 
   const { imageBase64, ...info } = session;
+  void imageBase64;
   return info;
 }
 
@@ -143,7 +146,7 @@ export function getSessionInfo(chatId: string): Omit<ImageSession, 'imageBase64'
  */
 export function hasSessionImage(chatId: string, userId: string): boolean {
   const session = imageSessionStore.get(chatId);
-  return session !== null && session.userId === userId;
+  return !!session && session.userId === userId;
 }
 
 // Run cleanup every 15 minutes

src/lib/image-session-store.ts

@@ -7,7 +7,7 @@ const logger = pino({
     asObject: true,
   },
   formatters: {
-    level: (label) => {
+    level: (label: string) => {
       return { level: label };
     },
   },

src/lib/logger.ts

@@ -36,7 +36,7 @@ export function getMemoryClient(): MemoryClient {
 export async function addMemory(
   messages: Array<{ role: 'user' | 'assistant' | 'system'; content: string }>,
   userId: string,
-  chatId?: string
+  chatId?: string,
 ) {
   try {
     const client = getMemoryClient();
@@ -64,16 +64,25 @@ export async function addMemory(
     // Add timestamp for debugging
     metadata.timestamp = new Date().toISOString();
 
-    console.log(`📝 [MEM0] Adding ${messages.length} messages for user ${userId}, chat ${chatId || 'N/A'}`);
+    console.log(
+      `📝 [MEM0] Adding ${messages.length} messages for user ${userId}, chat ${
+        chatId || 'N/A'
+      }`,
+    );
     console.log(`📝 [MEM0] Metadata:`, JSON.stringify(metadata));
 
     // Log message summary for debugging
     messages.forEach((msg, idx) => {
       console.log(`  ${idx + 1}. ${msg.role}: ${msg.content.slice(0, 50)}...`);
     });
 
+    // Filter out system messages as they might not be supported or needed for memory
+    const validMessages = messages.filter(
+      (msg) => msg.role !== 'system',
+    ) as Array<{ role: 'user' | 'assistant'; content: string }>;
+
     // Call Mem0 API
-    const result = await client.add(messages, metadata);
+    const result = await client.add(validMessages, metadata);
 
     console.log('✅ [MEM0] Memory stored successfully');
     console.log('📊 [MEM0] Result:', JSON.stringify(result, null, 2));
@@ -105,7 +114,7 @@ export async function searchMemory(
   query: string,
   userId: string,
   chatId?: string,
-  limit: number = 10
+  limit: number = 10,
 ) {
   try {
     const client = getMemoryClient();
@@ -118,16 +127,12 @@ export async function searchMemory(
 
     // Build filters for searching memories
     const filters: any = {
-      OR: [
-        { user_id: userId }
-      ]
+      OR: [{ user_id: userId }],
     };
 
     // Optionally filter by specific chat
     if (chatId) {
-      filters.AND = [
-        { chat_id: chatId }
-      ];
+      filters.AND = [{ chat_id: chatId }];
     }
 
     console.log(`🔍 [MEM0] Searching memory...`);
@@ -140,10 +145,12 @@ export async function searchMemory(
     const results = await client.search(query, {
       api_version: 'v2',
       filters: filters,
-      limit: limit
+      limit: limit,
     });
 
-    console.log(`✅ [MEM0] Search completed: found ${results?.length || 0} memories`);
+    console.log(
+      `✅ [MEM0] Search completed: found ${results?.length || 0} memories`,
+    );
 
     if (results && results.length > 0) {
       console.log('📋 [MEM0] Sample results:');
@@ -176,10 +183,7 @@ export async function getChatMemories(userId: string, chatId: string) {
     const client = getMemoryClient();
 
     const filters = {
-      AND: [
-        { user_id: userId },
-        { chat_id: chatId }
-      ]
+      AND: [{ user_id: userId }, { chat_id: chatId }],
     };
 
     console.log(`📚 [MEM0] Retrieving all memories for chat ${chatId}`);
@@ -188,7 +192,7 @@ export async function getChatMemories(userId: string, chatId: string) {
     const results = await client.search('', {
       api_version: 'v2',
       filters: filters,
-      limit: 50 // Get more memories for full context
+      limit: 50, // Get more memories for full context
     });
 
     console.log(`✅ [MEM0] Retrieved ${results?.length || 0} chat memories`);
@@ -205,13 +209,15 @@ export async function getChatMemories(userId: string, chatId: string) {
  */
 export async function deleteUserMemories(userId: string) {
   try {
-    const client = getMemoryClient();
+    getMemoryClient(); // Ensure client is initialized
 
     console.log(`🗑️ [MEM0] Deleting memories for user ${userId}`);
 
     // Note: The actual delete implementation depends on mem0ai API
     // You may need to adjust this based on the latest API
-    console.warn('[MEM0] Delete functionality needs to be implemented based on mem0ai API');
+    console.warn(
+      '[MEM0] Delete functionality needs to be implemented based on mem0ai API',
+    );
 
     return { success: true, message: 'Delete not yet implemented' };
   } catch (error) {

src/lib/mem0-client.ts

@@ -1,4 +1,4 @@
-import { nanoBananaConfig, IMAGE_GENERATION_CONFIG } from '@/config/ai';
+import { nanoBananaConfig } from '@/config/ai';
 
 interface NanoBananaResponse {
   success: boolean;
@@ -46,7 +46,7 @@ export class NanoBananaService {
    */
   async generateThumbnail(
     prompt: string,
-    options: GenerationOptions = {}
+    options: GenerationOptions = {},
   ): Promise<NanoBananaResponse> {
     const startTime = Date.now();
 
@@ -63,15 +63,17 @@ export class NanoBananaService {
       const response = await fetch(`${this.baseUrl}/images/generations`, {
         method: 'POST',
         headers: {
-          'Authorization': `Bearer ${this.apiKey}`,
+          Authorization: `Bearer ${this.apiKey}`,
           'Content-Type': 'application/json',
         },
         body: JSON.stringify(requestBody),
       });
 
       if (!response.ok) {
         const errorData = await response.text();
-        throw new Error(`Nano Banana API error: ${response.status} - ${errorData}`);
+        throw new Error(
+          `Nano Banana API error: ${response.status} - ${errorData}`,
+        );
       }
 
       const data = await response.json();
@@ -96,7 +98,8 @@ export class NanoBananaService {
 
       return {
         success: false,
-        error: error instanceof Error ? error.message : 'Unknown error occurred',
+        error:
+          error instanceof Error ? error.message : 'Unknown error occurred',
         metadata: {
           model: options.model || nanoBananaConfig.models.flash,
           prompt: prompt,
@@ -113,13 +116,13 @@ export class NanoBananaService {
   async generateThumbnailVariations(
     prompt: string,
     count: number = 2,
-    options: GenerationOptions = {}
+    options: GenerationOptions = {},
   ): Promise<NanoBananaResponse[]> {
     const promises = Array.from({ length: count }, (_, index) =>
       this.generateThumbnail(prompt, {
         ...options,
         seed: options.seed ? options.seed + index : undefined,
-      })
+      }),
     );
 
     return Promise.all(promises);
@@ -140,7 +143,9 @@ export class NanoBananaService {
     ];
 
     // Check if prompt already contains thumbnail-specific terms
-    const hasThumbTerms = /thumbnail|youtube|clickbait|title|preview/i.test(prompt);
+    const hasThumbTerms = /thumbnail|youtube|clickbait|title|preview/i.test(
+      prompt,
+    );
 
     if (!hasThumbTerms) {
       prompt += ', thumbnail style';
@@ -149,7 +154,7 @@ export class NanoBananaService {
     // Add enhancements if not already present
     const lowercasePrompt = prompt.toLowerCase();
     const missingEnhancements = enhancements.filter(
-      enhancement => !lowercasePrompt.includes(enhancement)
+      (enhancement) => !lowercasePrompt.includes(enhancement),
     );
 
     if (missingEnhancements.length > 0) {
@@ -166,7 +171,7 @@ export class NanoBananaService {
     try {
       const response = await fetch(`${this.baseUrl}/models`, {
         headers: {
-          'Authorization': `Bearer ${this.apiKey}`,
+          Authorization: `Bearer ${this.apiKey}`,
         },
       });
 
@@ -184,7 +189,7 @@ export class NanoBananaService {
     try {
       const response = await fetch(`${this.baseUrl}/models`, {
         headers: {
-          'Authorization': `Bearer ${this.apiKey}`,
+          Authorization: `Bearer ${this.apiKey}`,
         },
       });
 
@@ -193,12 +198,16 @@ export class NanoBananaService {
       }
 
       const data = await response.json();
-      return data.data?.map((model: any) => model.id) || [nanoBananaConfig.models.flash];
+      return (
+        data.data?.map((model: any) => model.id) || [
+          nanoBananaConfig.models.flash,
+        ]
+      );
     } catch (error) {
       console.error('Error fetching available models:', error);
       return [nanoBananaConfig.models.flash]; // Default fallback
     }
   }
 }
 
-export default NanoBananaService;
+export default NanoBananaService;

src/lib/nano-banana.ts

@@ -8,6 +8,7 @@
 import Stripe from 'stripe';
 import Razorpay from 'razorpay';
 import { razorpayConfig } from '@/config/payment';
+import crypto from 'crypto';
 
 // Payment provider types
 export type PaymentProvider = 'stripe' | 'razorpay';
@@ -239,7 +240,7 @@ export function verifyWebhookSignature(
   }
 
   if (PROVIDER === 'razorpay') {
-    const crypto = require('crypto');
+
     const expectedSignature = crypto
       .createHmac('sha256', secret)
       .update(payload)