added limit

supermemoryai · Apr 14, 2024 · d099d4b · d099d4b
1 parent 0a29564
commit d099d4b
Showing 1 changed file with 12 additions and 2 deletions.
diff --git a/apps/web/src/app/api/store/route.ts b/apps/web/src/app/api/store/route.ts
@@ -82,9 +82,19 @@ export async function POST(req: NextRequest) {
       count: sql<number>`count(*)`.mapWith(Number),
     })
     .from(storedContent)
-    .where(eq(storedContent.user, session.user.id));
+    .where(
+      and(
+        eq(storedContent.user, session.user.id),
+        eq(storedContent.type, "page"),
+      ),
+    );
 
-  console.log("count", count[0].count);
+  if (count[0].count > 100) {
+    return NextResponse.json(
+      { message: "Error", error: "Limit exceeded" },
+      { status: 499 },
+    );
+  }
 
   const { id } = (
     await db