core: limit notebooks to 20 on basic plan

prathmesh703 · Aug 16, 2024 · cf4a7a3 · cf4a7a3
1 parent 86e661c
commit cf4a7a3
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/packages/core/src/collections/notebooks.ts b/packages/core/src/collections/notebooks.ts
@@ -25,6 +25,7 @@ import { SQLCollection } from "../database/sql-collection";
 import { isFalse } from "../database";
 import { sql } from "kysely";
 import { deleteItems } from "../utils/array";
+import { CHECK_IDS, checkIsUserPremium } from "../common";
 
 export class Notebooks implements ICollection {
   name = "notebooks";
@@ -59,6 +60,13 @@ export class Notebooks implements ICollection {
     if (oldNotebook && isTrashItem(oldNotebook))
       throw new Error("Cannot modify trashed notebooks.");
 
+    if (
+      !oldNotebook &&
+      (await this.all.count()) >= 20 &&
+      !(await checkIsUserPremium(CHECK_IDS.notebookAdd))
+    )
+      return;
+
     const mergedNotebook: Partial<Notebook> = {
       ...oldNotebook,
       ...notebookArg