From bf6d04c4eaffc390c1c0cbfc6d73f33f98132f38 Mon Sep 17 00:00:00 2001
From: arkml <arkm85@gmail.com>
Date: Fri, 26 Sep 2025 15:24:55 +0530
Subject: [PATCH] fixed billing

---
 apps/rowboat/app/api/uploaded-images/route.ts | 34 ++++++++++++-------
 1 file changed, 22 insertions(+), 12 deletions(-)

diff --git a/apps/rowboat/app/api/uploaded-images/route.ts b/apps/rowboat/app/api/uploaded-images/route.ts
index 3e941c13..35e7854e 100644
--- a/apps/rowboat/app/api/uploaded-images/route.ts
+++ b/apps/rowboat/app/api/uploaded-images/route.ts
@@ -3,10 +3,9 @@ import { S3Client, PutObjectCommand } from '@aws-sdk/client-s3';
 import crypto from 'crypto';
 import { tempBinaryCache } from '@/src/application/services/temp-binary-cache';
 import { GoogleGenerativeAI } from '@google/generative-ai';
-import { UsageTracker } from '@/app/lib/billing';
-import { logUsage } from '@/app/actions/billing.actions';
+import { UsageTracker, getCustomerForUserId, logUsage as libLogUsage } from '@/app/lib/billing';
 import { authCheck } from '@/app/actions/auth.actions';
-import { USE_AUTH } from '@/app/lib/feature_flags';
+import { USE_AUTH, USE_BILLING } from '@/app/lib/feature_flags';
 
 // POST /api/uploaded-images
 // Accepts an image file (multipart/form-data, field name: "file")
@@ -15,9 +14,10 @@ import { USE_AUTH } from '@/app/lib/feature_flags';
 export async function POST(request: NextRequest) {
   try {
     // Require authentication if enabled
+    let currentUser: any | null = null;
     try {
       if (USE_AUTH) {
-        await authCheck();
+        currentUser = await authCheck();
       }
     } catch (_) {
       return NextResponse.json({ error: 'Unauthorized' }, { status: 401 });
@@ -101,11 +101,16 @@ export async function POST(request: NextRequest) {
 
       const url = `/api/uploaded-images/${imageId}`;
 
-      // Log usage to billing if available
+      // Log usage to billing similar to rag-worker
       try {
-        const items = usageTracker.flush();
-        if (items.length > 0) {
-          await logUsage({ items });
+        if (USE_BILLING && currentUser) {
+          const customer = await getCustomerForUserId(currentUser.id);
+          if (customer) {
+            const items = usageTracker.flush();
+            if (items.length > 0) {
+              await libLogUsage(customer.id, { items });
+            }
+          }
         }
       } catch (_) {
         // ignore billing logging errors
@@ -118,11 +123,16 @@ export async function POST(request: NextRequest) {
     const ttlSec = 10 * 60; // 10 minutes
     const id = tempBinaryCache.put(buf, mime, ttlSec * 1000);
     const url = `/api/tmp-images/${id}`;
-    // Log usage to billing if available
+    // Log usage to billing similar to rag-worker
     try {
-      const items = usageTracker.flush();
-      if (items.length > 0) {
-        await logUsage({ items });
+      if (USE_BILLING && currentUser) {
+        const customer = await getCustomerForUserId(currentUser.id);
+        if (customer) {
+          const items = usageTracker.flush();
+          if (items.length > 0) {
+            await libLogUsage(customer.id, { items });
+          }
+        }
       }
     } catch (_) {
       // ignore billing logging errors