diff options
Diffstat (limited to 'packages/trpc/routers')
| -rw-r--r-- | packages/trpc/routers/admin.ts | 4 | ||||
| -rw-r--r-- | packages/trpc/routers/subscriptions.ts | 10 |
2 files changed, 13 insertions, 1 deletions
diff --git a/packages/trpc/routers/admin.ts b/packages/trpc/routers/admin.ts index 2935f2e8..c4fc9d3d 100644 --- a/packages/trpc/routers/admin.ts +++ b/packages/trpc/routers/admin.ts @@ -361,6 +361,10 @@ export const adminAppRouter = router({ updateData.storageQuota = input.storageQuota; } + if (input.browserCrawlingEnabled !== undefined) { + updateData.browserCrawlingEnabled = input.browserCrawlingEnabled; + } + if (Object.keys(updateData).length === 0) { throw new TRPCError({ code: "BAD_REQUEST", diff --git a/packages/trpc/routers/subscriptions.ts b/packages/trpc/routers/subscriptions.ts index 4915a225..1cd34d30 100644 --- a/packages/trpc/routers/subscriptions.ts +++ b/packages/trpc/routers/subscriptions.ts @@ -90,12 +90,14 @@ async function syncStripeDataToDatabase(customerId: string, db: Context["db"]) { }) .where(eq(subscriptions.stripeCustomerId, customerId)); - // Update user quotas to free tier limits + // Update user quotas to free tier limits and disable browser crawling await trx .update(users) .set({ bookmarkQuota: serverConfig.quotas.free.bookmarkLimit, storageQuota: serverConfig.quotas.free.assetSizeBytes, + browserCrawlingEnabled: + serverConfig.quotas.free.browserCrawlingEnabled, }) .where(eq(users.id, existingSubscription.userId)); }); @@ -129,19 +131,25 @@ async function syncStripeDataToDatabase(customerId: string, db: Context["db"]) { .where(eq(subscriptions.stripeCustomerId, customerId)); if (subData.status === "active" || subData.status === "trialing") { + // Enable paid tier quotas and browser crawling await trx .update(users) .set({ bookmarkQuota: serverConfig.quotas.paid.bookmarkLimit, storageQuota: serverConfig.quotas.paid.assetSizeBytes, + browserCrawlingEnabled: + serverConfig.quotas.paid.browserCrawlingEnabled, }) .where(eq(users.id, existingSubscription.userId)); } else { + // Set free tier quotas and disable browser crawling await trx .update(users) .set({ bookmarkQuota: serverConfig.quotas.free.bookmarkLimit, storageQuota: serverConfig.quotas.free.assetSizeBytes, + browserCrawlingEnabled: + serverConfig.quotas.free.browserCrawlingEnabled, }) .where(eq(users.id, existingSubscription.userId)); } |
