get pagecount as well

2 files changed, 20 insertions, 6 deletions

diff --git a/apps/cf-ai-backend/src/routes/queue.ts b/apps/cf-ai-backend/src/routes/queue.ts
index c00afb5b..9dcef525 100644
--- a/apps/cf-ai-backend/src/routes/queue.ts
+++ b/apps/cf-ai-backend/src/routes/queue.ts
@@ -15,10 +15,10 @@ export const queue = async (batch: MessageBatch, env: Env): Promise<void> => {
 		headers: {
 			Authorization: `Bearer ${token}`,
 		},
-	}).then((res) => res.json())) as { limit: number; tweetsCount: number; user: string };
+	}).then((res) => res.json())) as { tweetsLimit: number; tweetsCount: number; user: string };
 
-	if (messages.length > limits.limit - limits.tweetsCount) {
-		messages.splice(limits.limit - limits.tweetsCount);
+	if (messages.length > limits.tweetsLimit - limits.tweetsCount) {
+		messages.splice(limits.tweetsLimit - limits.tweetsCount);
 	}
 
 	if (messages.length === 0) {
diff --git a/apps/web/src/app/api/getCount/route.ts b/apps/web/src/app/api/getCount/route.ts
index 05a3aa30..3238e58a 100644
--- a/apps/web/src/app/api/getCount/route.ts
+++ b/apps/web/src/app/api/getCount/route.ts
@@ -46,7 +46,7 @@ export async function GET(req: NextRequest) {
 
   const session = { session: sessionData[0], user: user[0] };
 
-  const count = await db
+  const tweetsCount = await db
     .select({
       count: sql<number>`count(*)`.mapWith(Number),
     })
@@ -58,9 +58,23 @@ export async function GET(req: NextRequest) {
       ),
     );
 
+  const pageCount = await db
+    .select({
+      count: sql<number>`count(*)`.mapWith(Number),
+    })
+    .from(storedContent)
+    .where(
+      and(
+        eq(storedContent.user, session.user.id),
+        eq(storedContent.type, "page"),
+      ),
+    );
+
   return NextResponse.json({
-    tweetsCount: count[0].count,
-    limit: 1000,
+    tweetsCount: tweetsCount[0].count,
+    tweetsLimit: 1000,
+    pageCount: pageCount[0].count,
+    pageLimit: 100,
     user: session.user.email,
   });
 }