Merge pull request #249 from giselles-ai/feat/delete-team-migration

feat: Add cascade delete to team-related foreign key constraints
giselles-ai · Dec 18, 2024 · f52650a · f52650a
2 parents d4d45c6 + 650931e
commit f52650a
Show file tree

Hide file tree

Showing 4 changed files with 2,071 additions and 2 deletions.
diff --git a/drizzle/schema.ts b/drizzle/schema.ts
@@ -54,7 +54,7 @@ export const subscriptions = pgTable("subscriptions", {
 	dbId: serial("db_id").primaryKey(),
 	teamDbId: integer("team_db_id")
 		.notNull()
-		.references(() => teams.dbId),
+		.references(() => teams.dbId, { onDelete: "cascade" }),
 	status: text("status").$type<Stripe.Subscription.Status>().notNull(),
 	cancelAtPeriodEnd: boolean("cancel_at_period_end").notNull(),
 	cancelAt: timestamp("cancel_at"),
@@ -113,7 +113,7 @@ export const teamMemberships = pgTable(
 			.references(() => users.dbId),
 		teamDbId: integer("team_db_id")
 			.notNull()
-			.references(() => teams.dbId),
+			.references(() => teams.dbId, { onDelete: "cascade" }),
 		role: text("role").notNull().$type<TeamRole>(),
 	},
 	(teamMembership) => ({

diff --git a/migrations/0019_oval_zaran.sql b/migrations/0019_oval_zaran.sql
@@ -0,0 +1,15 @@
+ALTER TABLE "subscriptions" DROP CONSTRAINT "subscriptions_team_db_id_teams_db_id_fk";
+--> statement-breakpoint
+ALTER TABLE "team_memberships" DROP CONSTRAINT "team_memberships_team_db_id_teams_db_id_fk";
+--> statement-breakpoint
+DO $$ BEGIN
+ ALTER TABLE "subscriptions" ADD CONSTRAINT "subscriptions_team_db_id_teams_db_id_fk" FOREIGN KEY ("team_db_id") REFERENCES "public"."teams"("db_id") ON DELETE cascade ON UPDATE no action;
+EXCEPTION
+ WHEN duplicate_object THEN null;
+END $$;
+--> statement-breakpoint
+DO $$ BEGIN
+ ALTER TABLE "team_memberships" ADD CONSTRAINT "team_memberships_team_db_id_teams_db_id_fk" FOREIGN KEY ("team_db_id") REFERENCES "public"."teams"("db_id") ON DELETE cascade ON UPDATE no action;
+EXCEPTION
+ WHEN duplicate_object THEN null;
+END $$;