feat(cloud): partner integration (#493)

* feat: add partnerLicenseKey and partnerTier to subscription schema

* feat: add repo funcs

* feat: set up partner webhook handler

packages/db/src/repository/subscription.repo.ts

@@ -7,6 +7,7 @@ export const updateById = async (
   db: dbClient,
   subscriptionId: number,
   updates: {
+    plan?: string;
     unlimitedSeats?: boolean;
     status?: string;
     seats?: number | null;
@@ -89,3 +90,47 @@ export const create = async (
   const [result] = await db.insert(subscription).values(data).returning();
   return result;
 };
+
+export const getByPartnerLicenseKey = async (
+  db: dbClient,
+  partnerLicenseKey: string,
+) => {
+  const result = await db.query.subscription.findFirst({
+    where: eq(subscription.partnerLicenseKey, partnerLicenseKey),
+  });
+  return result;
+};
+
+export const upsertByPartnerLicenseKey = async (
+  db: dbClient,
+  partnerLicenseKey: string,
+  data: {
+    plan: string;
+    status: string;
+    partnerTier: number;
+    seats: number | null;
+    unlimitedSeats: boolean;
+    referenceId?: string;
+  },
+) => {
+  const existing = await getByPartnerLicenseKey(db, partnerLicenseKey);
+
+  if (existing) {
+    const [result] = await db
+      .update(subscription)
+      .set({ ...data, updatedAt: new Date() })
+      .where(eq(subscription.partnerLicenseKey, partnerLicenseKey))
+      .returning();
+    return result;
+  }
+
+  const [result] = await db
+    .insert(subscription)
+    .values({
+      partnerLicenseKey,
+      ...data,
+      referenceId: data.referenceId ?? null,
+    })
+    .returning();
+  return result;
+};

packages/db/src/schema/subscriptions.ts

@@ -5,31 +5,42 @@ import {
   integer,
   pgTable,
   timestamp,
+  uniqueIndex,
   varchar,
 } from "drizzle-orm/pg-core";
 
 import { workspaces } from "./workspaces";
 
-export const subscription = pgTable("subscription", {
-  id: bigserial("id", { mode: "number" }).primaryKey(),
-  plan: varchar("plan", { length: 255 }).notNull(),
-  referenceId: varchar("referenceId", { length: 12 }).references(
-    () => workspaces.publicId,
-    { onDelete: "set null" },
-  ),
-  stripeCustomerId: varchar("stripeCustomerId", { length: 255 }),
-  stripeSubscriptionId: varchar("stripeSubscriptionId", { length: 255 }),
-  status: varchar("status", { length: 255 }).notNull(),
-  periodStart: timestamp("periodStart"),
-  periodEnd: timestamp("periodEnd"),
-  cancelAtPeriodEnd: boolean("cancelAtPeriodEnd"),
-  seats: integer("seats"),
-  unlimitedSeats: boolean("unlimitedSeats").default(false).notNull(),
-  trialStart: timestamp("trialStart"),
-  trialEnd: timestamp("trialEnd"),
-  createdAt: timestamp("createdAt").notNull().defaultNow(),
-  updatedAt: timestamp("updatedAt").notNull().defaultNow(),
-}).enableRLS();
+export const subscription = pgTable(
+  "subscription",
+  {
+    id: bigserial("id", { mode: "number" }).primaryKey(),
+    plan: varchar("plan", { length: 255 }).notNull(),
+    referenceId: varchar("referenceId", { length: 12 }).references(
+      () => workspaces.publicId,
+      { onDelete: "set null" },
+    ),
+    stripeCustomerId: varchar("stripeCustomerId", { length: 255 }),
+    stripeSubscriptionId: varchar("stripeSubscriptionId", { length: 255 }),
+    status: varchar("status", { length: 255 }).notNull(),
+    periodStart: timestamp("periodStart"),
+    periodEnd: timestamp("periodEnd"),
+    cancelAtPeriodEnd: boolean("cancelAtPeriodEnd"),
+    seats: integer("seats"),
+    unlimitedSeats: boolean("unlimitedSeats").default(false).notNull(),
+    trialStart: timestamp("trialStart"),
+    trialEnd: timestamp("trialEnd"),
+    partnerLicenseKey: varchar("partnerLicenseKey", { length: 255 }),
+    partnerTier: integer("partnerTier"),
+    createdAt: timestamp("createdAt").notNull().defaultNow(),
+    updatedAt: timestamp("updatedAt").notNull().defaultNow(),
+  },
+  (table) => [
+    uniqueIndex("subscription_partner_license_key_idx").on(
+      table.partnerLicenseKey,
+    ),
+  ],
+).enableRLS();
 
 export const subscriptionsRelations = relations(subscription, ({ one }) => ({
   workspace: one(workspaces, {