/ opencodehub / src / db / schema / pr-states.ts
Swadhin Biswas feat(api): add runner, analytics, packages, and security services · 3/6/2026
28f7954
108 lines · 3.7 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
/**
 * Custom PR States Schema
 * Allows repositories to define custom workflow states for PRs
 */

import { pgTable, text, timestamp, integer, boolean } from "drizzle-orm/pg-core";
import { relations } from "drizzle-orm";
import { repositories } from "./repositories";
import { users } from "./users";
import { teams } from "./teams";

/**
 * PR State Definitions
 * Custom states that can be assigned to PRs in a repository
 */
export const prStateDefinitions = pgTable("pr_state_definitions", {
    id: text("id").primaryKey(),
    repositoryId: text("repository_id")
        .notNull()
        .references(() => repositories.id, { onDelete: "cascade" }),

    // State info
    name: text("name").notNull(), // e.g., "needs_work", "ready_to_merge", "on_hold"
    displayName: text("display_name").notNull(), // e.g., "Needs Work", "Ready to Merge"
    description: text("description"),
    color: text("color").notNull().default("#6B7280"), // Hex color for UI
    icon: text("icon"), // Icon name for UI

    // Ordering and behavior
    order: integer("order").notNull().default(0),
    isDefault: boolean("is_default").default(false), // Default state for new PRs
    isFinal: boolean("is_final").default(false), // States like "merged" or "closed"
    allowMerge: boolean("allow_merge").default(false), // Can merge from this state
    requireCodeOwner: boolean("require_code_owner").default(false), // Must satisfy CODEOWNERS before entering this state

    createdAt: timestamp("created_at").notNull().defaultNow(),
    updatedAt: timestamp("updated_at").notNull().defaultNow(),
});

/**
 * Required Reviewers Per State
 * Defines who must approve before transitioning to a state
 */
export const prStateReviewers = pgTable("pr_state_reviewers", {
    id: text("id").primaryKey(),
    stateDefinitionId: text("state_definition_id")
        .notNull()
        .references(() => prStateDefinitions.id, { onDelete: "cascade" }),

    // Reviewer target - one of these
    userId: text("user_id").references(() => users.id, { onDelete: "cascade" }),
    teamId: text("team_id").references(() => teams.id, { onDelete: "cascade" }),

    // How many approvals needed (for team)
    requiredCount: integer("required_count").default(1),

    createdAt: timestamp("created_at").notNull().defaultNow(),
});

/**
 * PR State Transitions
 * Track state change history
 */
export const prStateTransitions = pgTable("pr_state_transitions", {
    id: text("id").primaryKey(),
    pullRequestId: text("pull_request_id").notNull(),

    fromState: text("from_state"),
    toState: text("to_state").notNull(),

    changedById: text("changed_by_id")
        .notNull()
        .references(() => users.id),

    reason: text("reason"),
    createdAt: timestamp("created_at").notNull().defaultNow(),
});

// Relations
export const prStateDefinitionsRelations = relations(prStateDefinitions, ({ one, many }) => ({
    repository: one(repositories, {
        fields: [prStateDefinitions.repositoryId],
        references: [repositories.id],
    }),
    reviewers: many(prStateReviewers),
}));

export const prStateReviewersRelations = relations(prStateReviewers, ({ one }) => ({
    stateDefinition: one(prStateDefinitions, {
        fields: [prStateReviewers.stateDefinitionId],
        references: [prStateDefinitions.id],
    }),
    user: one(users, {
        fields: [prStateReviewers.userId],
        references: [users.id],
    }),
    team: one(teams, {
        fields: [prStateReviewers.teamId],
        references: [teams.id],
    }),
}));

// Types
export type PRStateDefinition = typeof prStateDefinitions.$inferSelect;
export type NewPRStateDefinition = typeof prStateDefinitions.$inferInsert;
export type PRStateReviewer = typeof prStateReviewers.$inferSelect;
export type PRStateTransition = typeof prStateTransitions.$inferSelect;