feat(billing): add subscriptions and plans support

.env.example

@@ -94,3 +94,4 @@ NEXT_PUBLIC_POSTHOG_HOST=https://app.posthog.com
 
 STRIPE_SECRET_KEY=sk_test_your_stripe_secret_key
 STRIPE_WEBHOOK_SECRET=whsec_your_webhook_secret
+STRIPE_PRO_PRICE_ID=price_your_pro_plan_price_id

apps/api/.env.example

@@ -1,5 +1,6 @@
 STRIPE_SECRET_KEY=sk_test_123
 STRIPE_WEBHOOK_SECRET=whsec_123
+STRIPE_PRO_PRICE_ID=price_12345678
 
 VITE_POSTHOG_KEY=phc_123
 VITE_POSTHOG_HOST=https://app.posthog.com

apps/api/src/routes/index.ts

@@ -8,6 +8,7 @@ import { logs } from "./logs";
 import organization from "./organization";
 import { payments } from "./payments";
 import projects from "./projects";
+import { subscriptions } from "./subscriptions";
 import { user } from "./user";
 
 import type { ServerTypes } from "../vars";
@@ -40,3 +41,4 @@ routes.route("/projects", projects);
 
 routes.route("/orgs", organization);
 routes.route("/payments", payments);
+routes.route("/subscriptions", subscriptions);

apps/api/src/routes/keys-provider.e2e.ts

@@ -20,6 +20,7 @@ describe("e2e tests for provider keys", () => {
 		await db.insert(tables.organization).values({
 			id: "test-org-id",
 			name: "Test Organization",
+			plan: "pro",
 		});
 
 		await db.insert(tables.userOrganization).values({
@@ -32,6 +33,7 @@ describe("e2e tests for provider keys", () => {
 			id: "test-project-id",
 			name: "Test Project",
 			organizationId: "test-org-id",
+			mode: "api-keys",
 		});
 	});
 

apps/api/src/routes/keys-provider.spec.ts

@@ -18,6 +18,7 @@ describe("provider keys route", () => {
 		await db.insert(tables.organization).values({
 			id: "test-org-id",
 			name: "Test Organization",
+			plan: "pro",
 		});
 
 		// Associate user with organization

apps/api/src/routes/keys-provider.ts

@@ -116,6 +116,16 @@ keysProvider.openapi(create, async (c) => {
 		});
 	}
 
+	const organization = userOrgs[0].organization;
+
+	// Check if organization has pro plan for provider keys
+	if (organization?.plan !== "pro") {
+		throw new HTTPException(403, {
+			message:
+				"Provider keys are only available on the Pro plan. Please upgrade to use your own API keys.",
+		});
+	}
+
 	// Check if a provider key already exists for this provider and organization
 	const existingKey = await db.query.providerKey.findFirst({
 		where: {

apps/api/src/routes/organization.ts

@@ -14,6 +14,8 @@ const organizationSchema = z.object({
 	updatedAt: z.date(),
 	name: z.string(),
 	credits: z.string(),
+	plan: z.enum(["free", "pro"]),
+	planExpiresAt: z.date().nullable(),
 	status: z.enum(["active", "inactive", "deleted"]).nullable(),
 });
 

apps/api/src/routes/payments.ts

@@ -330,15 +330,13 @@ payments.openapi(setDefaultPaymentMethod, async (c) => {
 			.update(tables.paymentMethod)
 			.set({
 				isDefault: false,
-				updatedAt: new Date(),
 			})
 			.where(eq(tables.paymentMethod.organizationId, organizationId));
 
 		await db
 			.update(tables.paymentMethod)
 			.set({
 				isDefault: true,
-				updatedAt: new Date(),
 			})
 			.where(eq(tables.paymentMethod.id, paymentMethodId));