| d12933e | | | 1 | import type { FastifyInstance } from "fastify"; |
| d12933e | | | 2 | import { z } from "zod"; |
| d12933e | | | 3 | |
| d12933e | | | 4 | const createDiffSchema = z.object({ |
| d12933e | | | 5 | title: z.string().min(1).max(256), |
| d12933e | | | 6 | description: z.string().optional(), |
| 2ec6868 | | | 7 | head_commit: z.string().min(1), |
| 2ec6868 | | | 8 | base_commit: z.string().optional(), |
| d12933e | | | 9 | }); |
| d12933e | | | 10 | |
| d12933e | | | 11 | const updateDiffSchema = z.object({ |
| d12933e | | | 12 | title: z.string().min(1).max(256).optional(), |
| d12933e | | | 13 | description: z.string().optional(), |
| d12933e | | | 14 | status: z.enum(["open", "closed"]).optional(), |
| 2ec6868 | | | 15 | head_commit: z.string().min(1).optional(), |
| d12933e | | | 16 | }); |
| d12933e | | | 17 | |
| d12933e | | | 18 | const createCommentSchema = z.object({ |
| d12933e | | | 19 | body: z.string().min(1), |
| d12933e | | | 20 | file_path: z.string().optional(), |
| d12933e | | | 21 | line_number: z.number().int().positive().optional(), |
| d12933e | | | 22 | side: z.enum(["left", "right"]).optional(), |
| d12933e | | | 23 | commit_sha: z.string().optional(), |
| d12933e | | | 24 | parent_id: z.number().int().positive().optional(), |
| d12933e | | | 25 | }); |
| d12933e | | | 26 | |
| d12933e | | | 27 | const createReviewSchema = z.object({ |
| d12933e | | | 28 | status: z.enum(["approved", "changes_requested"]), |
| d12933e | | | 29 | body: z.string().optional(), |
| d12933e | | | 30 | }); |
| d12933e | | | 31 | |
| d12933e | | | 32 | export async function diffRoutes(app: FastifyInstance) { |
| d12933e | | | 33 | // List diffs for a repo |
| d12933e | | | 34 | app.get<{ |
| d12933e | | | 35 | Params: { owner: string; repo: string }; |
| d12933e | | | 36 | Querystring: { status?: string }; |
| d12933e | | | 37 | }>( |
| d12933e | | | 38 | "/:owner/:repo/diffs", |
| d12933e | | | 39 | async (request, reply) => { |
| d12933e | | | 40 | const { owner, repo } = request.params; |
| d12933e | | | 41 | const status = request.query.status ?? "open"; |
| d12933e | | | 42 | const db = (app as any).db; |
| d12933e | | | 43 | |
| d12933e | | | 44 | const repoRow = db |
| d12933e | | | 45 | .prepare(` |
| d12933e | | | 46 | SELECT id FROM repos_with_owner |
| d12933e | | | 47 | WHERE owner_name = ? AND name = ? |
| d12933e | | | 48 | `) |
| d12933e | | | 49 | .get(owner, repo) as any; |
| d12933e | | | 50 | |
| d12933e | | | 51 | if (!repoRow) { |
| d12933e | | | 52 | return reply.code(404).send({ error: "Repository not found" }); |
| d12933e | | | 53 | } |
| d12933e | | | 54 | |
| d12933e | | | 55 | const diffs = db |
| d12933e | | | 56 | .prepare(` |
| d12933e | | | 57 | SELECT d.*, u.username as author_name, u.display_name as author_display_name |
| d12933e | | | 58 | FROM diffs d |
| d12933e | | | 59 | JOIN users u ON d.author_id = u.id |
| d12933e | | | 60 | WHERE d.repo_id = ? AND d.status = ? |
| d12933e | | | 61 | ORDER BY d.updated_at DESC |
| d12933e | | | 62 | `) |
| d12933e | | | 63 | .all(repoRow.id, status); |
| d12933e | | | 64 | |
| d12933e | | | 65 | return { diffs }; |
| d12933e | | | 66 | } |
| d12933e | | | 67 | ); |
| d12933e | | | 68 | |
| d12933e | | | 69 | // Create diff |
| d12933e | | | 70 | app.post<{ Params: { owner: string; repo: string } }>( |
| d12933e | | | 71 | "/:owner/:repo/diffs", |
| d12933e | | | 72 | { |
| d12933e | | | 73 | preHandler: [(app as any).authenticate], |
| d12933e | | | 74 | handler: async (request, reply) => { |
| d12933e | | | 75 | const parsed = createDiffSchema.safeParse(request.body); |
| d12933e | | | 76 | if (!parsed.success) { |
| d12933e | | | 77 | return reply.code(400).send({ error: parsed.error.flatten() }); |
| d12933e | | | 78 | } |
| d12933e | | | 79 | |
| d12933e | | | 80 | const { owner, repo } = request.params; |
| d12933e | | | 81 | const user = request.user as any; |
| d12933e | | | 82 | const db = (app as any).db; |
| d12933e | | | 83 | |
| d12933e | | | 84 | const repoRow = db |
| d12933e | | | 85 | .prepare(` |
| 55e9501 | | | 86 | SELECT id FROM repos_with_owner |
| 55e9501 | | | 87 | WHERE owner_name = ? AND name = ? |
| d12933e | | | 88 | `) |
| d12933e | | | 89 | .get(owner, repo) as any; |
| d12933e | | | 90 | |
| d12933e | | | 91 | if (!repoRow) { |
| d12933e | | | 92 | return reply.code(404).send({ error: "Repository not found" }); |
| d12933e | | | 93 | } |
| d12933e | | | 94 | |
| 2ec6868 | | | 95 | const { title, description, head_commit, base_commit } = parsed.data; |
| 2ec6868 | | | 96 | |
| 2ec6868 | | | 97 | // Idempotent: if an open diff already exists for this commit, return it |
| 2ec6868 | | | 98 | const existing = db |
| 2ec6868 | | | 99 | .prepare(` |
| 2ec6868 | | | 100 | SELECT * FROM diffs |
| 2ec6868 | | | 101 | WHERE repo_id = ? AND head_commit = ? AND status = 'open' |
| 2ec6868 | | | 102 | `) |
| 2ec6868 | | | 103 | .get(repoRow.id, head_commit) as any; |
| 2ec6868 | | | 104 | if (existing) { |
| 2ec6868 | | | 105 | return reply.code(200).send({ diff: existing }); |
| 2ec6868 | | | 106 | } |
| 2ec6868 | | | 107 | |
| d12933e | | | 108 | // Get next diff number for this repo |
| d12933e | | | 109 | const maxNumber = db |
| d12933e | | | 110 | .prepare("SELECT MAX(number) as max_num FROM diffs WHERE repo_id = ?") |
| d12933e | | | 111 | .get(repoRow.id) as any; |
| d12933e | | | 112 | const nextNumber = (maxNumber?.max_num ?? 0) + 1; |
| d12933e | | | 113 | |
| d12933e | | | 114 | const result = db |
| d12933e | | | 115 | .prepare(` |
| 2ec6868 | | | 116 | INSERT INTO diffs (repo_id, number, title, description, author_id, head_commit, base_commit) |
| d12933e | | | 117 | VALUES (?, ?, ?, ?, ?, ?, ?) |
| d12933e | | | 118 | `) |
| d12933e | | | 119 | .run( |
| d12933e | | | 120 | repoRow.id, |
| d12933e | | | 121 | nextNumber, |
| d12933e | | | 122 | title, |
| d12933e | | | 123 | description ?? "", |
| d12933e | | | 124 | user.id, |
| 2ec6868 | | | 125 | head_commit, |
| 2ec6868 | | | 126 | base_commit ?? null |
| d12933e | | | 127 | ); |
| d12933e | | | 128 | |
| d12933e | | | 129 | const diff = db |
| d12933e | | | 130 | .prepare("SELECT * FROM diffs WHERE id = ?") |
| d12933e | | | 131 | .get(result.lastInsertRowid); |
| d12933e | | | 132 | |
| d12933e | | | 133 | return reply.code(201).send({ diff }); |
| d12933e | | | 134 | }, |
| d12933e | | | 135 | } |
| d12933e | | | 136 | ); |
| d12933e | | | 137 | |
| d12933e | | | 138 | // Get single diff |
| d12933e | | | 139 | app.get<{ Params: { owner: string; repo: string; number: string } }>( |
| d12933e | | | 140 | "/:owner/:repo/diffs/:number", |
| d12933e | | | 141 | async (request, reply) => { |
| d12933e | | | 142 | const { owner, repo, number } = request.params; |
| d12933e | | | 143 | const db = (app as any).db; |
| d12933e | | | 144 | |
| d12933e | | | 145 | const diff = db |
| d12933e | | | 146 | .prepare(` |
| d12933e | | | 147 | SELECT d.*, u.username as author_name, u.display_name as author_display_name |
| d12933e | | | 148 | FROM diffs d |
| d12933e | | | 149 | JOIN users u ON d.author_id = u.id |
| 55e9501 | | | 150 | JOIN repos_with_owner rwo ON d.repo_id = rwo.id |
| 55e9501 | | | 151 | WHERE rwo.owner_name = ? AND rwo.name = ? AND d.number = ? |
| d12933e | | | 152 | `) |
| d12933e | | | 153 | .get(owner, repo, parseInt(number)) as any; |
| d12933e | | | 154 | |
| d12933e | | | 155 | if (!diff) { |
| d12933e | | | 156 | return reply.code(404).send({ error: "Diff not found" }); |
| d12933e | | | 157 | } |
| d12933e | | | 158 | |
| d12933e | | | 159 | // Get comments |
| d12933e | | | 160 | const comments = db |
| d12933e | | | 161 | .prepare(` |
| d12933e | | | 162 | SELECT c.*, u.username as author_name, u.display_name as author_display_name |
| d12933e | | | 163 | FROM comments c |
| d12933e | | | 164 | JOIN users u ON c.author_id = u.id |
| d12933e | | | 165 | WHERE c.diff_id = ? |
| d12933e | | | 166 | ORDER BY c.created_at ASC |
| d12933e | | | 167 | `) |
| d12933e | | | 168 | .all(diff.id); |
| d12933e | | | 169 | |
| d12933e | | | 170 | // Get reviews |
| d12933e | | | 171 | const reviews = db |
| d12933e | | | 172 | .prepare(` |
| d12933e | | | 173 | SELECT rv.*, u.username as reviewer_name, u.display_name as reviewer_display_name |
| d12933e | | | 174 | FROM reviews rv |
| d12933e | | | 175 | JOIN users u ON rv.reviewer_id = u.id |
| d12933e | | | 176 | WHERE rv.diff_id = ? |
| d12933e | | | 177 | ORDER BY rv.created_at DESC |
| d12933e | | | 178 | `) |
| d12933e | | | 179 | .all(diff.id); |
| d12933e | | | 180 | |
| d12933e | | | 181 | return { diff, comments, reviews }; |
| d12933e | | | 182 | } |
| d12933e | | | 183 | ); |
| d12933e | | | 184 | |
| d12933e | | | 185 | // Update diff |
| d12933e | | | 186 | app.patch<{ Params: { owner: string; repo: string; number: string } }>( |
| d12933e | | | 187 | "/:owner/:repo/diffs/:number", |
| d12933e | | | 188 | { |
| d12933e | | | 189 | preHandler: [(app as any).authenticate], |
| d12933e | | | 190 | handler: async (request, reply) => { |
| d12933e | | | 191 | const parsed = updateDiffSchema.safeParse(request.body); |
| d12933e | | | 192 | if (!parsed.success) { |
| d12933e | | | 193 | return reply.code(400).send({ error: parsed.error.flatten() }); |
| d12933e | | | 194 | } |
| d12933e | | | 195 | |
| d12933e | | | 196 | const { owner, repo, number } = request.params; |
| d12933e | | | 197 | const db = (app as any).db; |
| d12933e | | | 198 | |
| d12933e | | | 199 | const diff = db |
| d12933e | | | 200 | .prepare(` |
| d12933e | | | 201 | SELECT d.* FROM diffs d |
| 55e9501 | | | 202 | JOIN repos_with_owner rwo ON d.repo_id = rwo.id |
| 55e9501 | | | 203 | WHERE rwo.owner_name = ? AND rwo.name = ? AND d.number = ? |
| d12933e | | | 204 | `) |
| d12933e | | | 205 | .get(owner, repo, parseInt(number)) as any; |
| d12933e | | | 206 | |
| d12933e | | | 207 | if (!diff) { |
| d12933e | | | 208 | return reply.code(404).send({ error: "Diff not found" }); |
| d12933e | | | 209 | } |
| d12933e | | | 210 | |
| d12933e | | | 211 | const updates = parsed.data; |
| d12933e | | | 212 | const setClauses: string[] = ["updated_at = datetime('now')"]; |
| d12933e | | | 213 | const values: any[] = []; |
| d12933e | | | 214 | |
| d12933e | | | 215 | if (updates.title) { |
| d12933e | | | 216 | setClauses.push("title = ?"); |
| d12933e | | | 217 | values.push(updates.title); |
| d12933e | | | 218 | } |
| d12933e | | | 219 | if (updates.description !== undefined) { |
| d12933e | | | 220 | setClauses.push("description = ?"); |
| d12933e | | | 221 | values.push(updates.description); |
| d12933e | | | 222 | } |
| d12933e | | | 223 | if (updates.status) { |
| d12933e | | | 224 | setClauses.push("status = ?"); |
| d12933e | | | 225 | values.push(updates.status); |
| d12933e | | | 226 | } |
| 2ec6868 | | | 227 | if (updates.head_commit) { |
| 2ec6868 | | | 228 | setClauses.push("head_commit = ?"); |
| 2ec6868 | | | 229 | values.push(updates.head_commit); |
| 2ec6868 | | | 230 | } |
| d12933e | | | 231 | |
| d12933e | | | 232 | values.push(diff.id); |
| d12933e | | | 233 | |
| d12933e | | | 234 | db.prepare( |
| d12933e | | | 235 | `UPDATE diffs SET ${setClauses.join(", ")} WHERE id = ?` |
| d12933e | | | 236 | ).run(...values); |
| d12933e | | | 237 | |
| d12933e | | | 238 | const updated = db |
| d12933e | | | 239 | .prepare("SELECT * FROM diffs WHERE id = ?") |
| d12933e | | | 240 | .get(diff.id); |
| d12933e | | | 241 | |
| d12933e | | | 242 | return { diff: updated }; |
| d12933e | | | 243 | }, |
| d12933e | | | 244 | } |
| d12933e | | | 245 | ); |
| d12933e | | | 246 | |
| 2ec6868 | | | 247 | // Land a diff (pushrebase commit onto main) |
| d12933e | | | 248 | app.post<{ Params: { owner: string; repo: string; number: string } }>( |
| 2ec6868 | | | 249 | "/:owner/:repo/diffs/:number/land", |
| d12933e | | | 250 | { |
| d12933e | | | 251 | preHandler: [(app as any).authenticate], |
| d12933e | | | 252 | handler: async (request, reply) => { |
| d12933e | | | 253 | const { owner, repo, number } = request.params; |
| d12933e | | | 254 | const user = request.user as any; |
| d12933e | | | 255 | const db = (app as any).db; |
| d12933e | | | 256 | |
| d12933e | | | 257 | const diff = db |
| d12933e | | | 258 | .prepare(` |
| d12933e | | | 259 | SELECT d.* FROM diffs d |
| 55e9501 | | | 260 | JOIN repos_with_owner rwo ON d.repo_id = rwo.id |
| 55e9501 | | | 261 | WHERE rwo.owner_name = ? AND rwo.name = ? AND d.number = ? |
| d12933e | | | 262 | `) |
| d12933e | | | 263 | .get(owner, repo, parseInt(number)) as any; |
| d12933e | | | 264 | |
| d12933e | | | 265 | if (!diff) { |
| d12933e | | | 266 | return reply.code(404).send({ error: "Diff not found" }); |
| d12933e | | | 267 | } |
| d12933e | | | 268 | |
| d12933e | | | 269 | if (diff.status !== "open") { |
| d12933e | | | 270 | return reply |
| d12933e | | | 271 | .code(400) |
| d12933e | | | 272 | .send({ error: "Diff is not open" }); |
| d12933e | | | 273 | } |
| d12933e | | | 274 | |
| 2ec6868 | | | 275 | // Land the commit onto main via grove_bridge (Mononoke pushrebase) |
| d12933e | | | 276 | const bridgeUrl = process.env.GROVE_BRIDGE_URL ?? "http://localhost:3100"; |
| d12933e | | | 277 | try { |
| d12933e | | | 278 | const res = await fetch(`${bridgeUrl}/repos/${repo}/land`, { |
| d12933e | | | 279 | method: "POST", |
| d12933e | | | 280 | headers: { "Content-Type": "application/json" }, |
| d12933e | | | 281 | body: JSON.stringify({ |
| 2ec6868 | | | 282 | source_bookmark: diff.head_commit, |
| 2ec6868 | | | 283 | target_bookmark: "main", |
| d12933e | | | 284 | }), |
| d12933e | | | 285 | }); |
| d12933e | | | 286 | if (!res.ok) { |
| d12933e | | | 287 | const body = await res.json().catch(() => ({ error: "Land failed" })); |
| 2ec6868 | | | 288 | return reply.code(409).send({ error: body.error || "Land failed" }); |
| d12933e | | | 289 | } |
| d12933e | | | 290 | } catch (err: unknown) { |
| 2ec6868 | | | 291 | const msg = err instanceof Error ? err.message : "Land failed"; |
| d12933e | | | 292 | return reply.code(502).send({ error: msg }); |
| d12933e | | | 293 | } |
| d12933e | | | 294 | |
| d12933e | | | 295 | db.prepare(` |
| d12933e | | | 296 | UPDATE diffs |
| 2ec6868 | | | 297 | SET status = 'landed', landed_at = datetime('now'), landed_by = ?, updated_at = datetime('now') |
| d12933e | | | 298 | WHERE id = ? |
| d12933e | | | 299 | `).run(user.id, diff.id); |
| d12933e | | | 300 | |
| d12933e | | | 301 | const updated = db |
| d12933e | | | 302 | .prepare("SELECT * FROM diffs WHERE id = ?") |
| d12933e | | | 303 | .get(diff.id); |
| d12933e | | | 304 | |
| d12933e | | | 305 | return { diff: updated }; |
| d12933e | | | 306 | }, |
| d12933e | | | 307 | } |
| d12933e | | | 308 | ); |
| d12933e | | | 309 | |
| d12933e | | | 310 | // Add comment to diff |
| d12933e | | | 311 | app.post<{ Params: { owner: string; repo: string; number: string } }>( |
| d12933e | | | 312 | "/:owner/:repo/diffs/:number/comments", |
| d12933e | | | 313 | { |
| d12933e | | | 314 | preHandler: [(app as any).authenticate], |
| d12933e | | | 315 | handler: async (request, reply) => { |
| d12933e | | | 316 | const parsed = createCommentSchema.safeParse(request.body); |
| d12933e | | | 317 | if (!parsed.success) { |
| d12933e | | | 318 | return reply.code(400).send({ error: parsed.error.flatten() }); |
| d12933e | | | 319 | } |
| d12933e | | | 320 | |
| d12933e | | | 321 | const { owner, repo, number } = request.params; |
| d12933e | | | 322 | const user = request.user as any; |
| d12933e | | | 323 | const db = (app as any).db; |
| d12933e | | | 324 | |
| d12933e | | | 325 | const diff = db |
| d12933e | | | 326 | .prepare(` |
| d12933e | | | 327 | SELECT d.id FROM diffs d |
| 55e9501 | | | 328 | JOIN repos_with_owner rwo ON d.repo_id = rwo.id |
| 55e9501 | | | 329 | WHERE rwo.owner_name = ? AND rwo.name = ? AND d.number = ? |
| d12933e | | | 330 | `) |
| d12933e | | | 331 | .get(owner, repo, parseInt(number)) as any; |
| d12933e | | | 332 | |
| d12933e | | | 333 | if (!diff) { |
| d12933e | | | 334 | return reply.code(404).send({ error: "Diff not found" }); |
| d12933e | | | 335 | } |
| d12933e | | | 336 | |
| d12933e | | | 337 | const { body, file_path, line_number, side, commit_sha, parent_id } = |
| d12933e | | | 338 | parsed.data; |
| d12933e | | | 339 | |
| d12933e | | | 340 | const result = db |
| d12933e | | | 341 | .prepare(` |
| d12933e | | | 342 | INSERT INTO comments (diff_id, author_id, body, file_path, line_number, side, commit_sha, parent_id) |
| d12933e | | | 343 | VALUES (?, ?, ?, ?, ?, ?, ?, ?) |
| d12933e | | | 344 | `) |
| d12933e | | | 345 | .run(diff.id, user.id, body, file_path, line_number, side, commit_sha, parent_id); |
| d12933e | | | 346 | |
| d12933e | | | 347 | const comment = db |
| d12933e | | | 348 | .prepare(` |
| d12933e | | | 349 | SELECT c.*, u.username as author_name |
| d12933e | | | 350 | FROM comments c JOIN users u ON c.author_id = u.id |
| d12933e | | | 351 | WHERE c.id = ? |
| d12933e | | | 352 | `) |
| d12933e | | | 353 | .get(result.lastInsertRowid); |
| d12933e | | | 354 | |
| d12933e | | | 355 | return reply.code(201).send({ comment }); |
| d12933e | | | 356 | }, |
| d12933e | | | 357 | } |
| d12933e | | | 358 | ); |
| d12933e | | | 359 | |
| d12933e | | | 360 | // Submit review |
| d12933e | | | 361 | app.post<{ Params: { owner: string; repo: string; number: string } }>( |
| d12933e | | | 362 | "/:owner/:repo/diffs/:number/reviews", |
| d12933e | | | 363 | { |
| d12933e | | | 364 | preHandler: [(app as any).authenticate], |
| d12933e | | | 365 | handler: async (request, reply) => { |
| d12933e | | | 366 | const parsed = createReviewSchema.safeParse(request.body); |
| d12933e | | | 367 | if (!parsed.success) { |
| d12933e | | | 368 | return reply.code(400).send({ error: parsed.error.flatten() }); |
| d12933e | | | 369 | } |
| d12933e | | | 370 | |
| d12933e | | | 371 | const { owner, repo, number } = request.params; |
| d12933e | | | 372 | const user = request.user as any; |
| d12933e | | | 373 | const db = (app as any).db; |
| d12933e | | | 374 | |
| d12933e | | | 375 | const diff = db |
| d12933e | | | 376 | .prepare(` |
| d12933e | | | 377 | SELECT d.id FROM diffs d |
| 55e9501 | | | 378 | JOIN repos_with_owner rwo ON d.repo_id = rwo.id |
| 55e9501 | | | 379 | WHERE rwo.owner_name = ? AND rwo.name = ? AND d.number = ? |
| d12933e | | | 380 | `) |
| d12933e | | | 381 | .get(owner, repo, parseInt(number)) as any; |
| d12933e | | | 382 | |
| d12933e | | | 383 | if (!diff) { |
| d12933e | | | 384 | return reply.code(404).send({ error: "Diff not found" }); |
| d12933e | | | 385 | } |
| d12933e | | | 386 | |
| d12933e | | | 387 | const { status, body } = parsed.data; |
| d12933e | | | 388 | |
| d12933e | | | 389 | const result = db |
| d12933e | | | 390 | .prepare(` |
| d12933e | | | 391 | INSERT INTO reviews (diff_id, reviewer_id, status, body) |
| d12933e | | | 392 | VALUES (?, ?, ?, ?) |
| d12933e | | | 393 | `) |
| d12933e | | | 394 | .run(diff.id, user.id, status, body); |
| d12933e | | | 395 | |
| d12933e | | | 396 | const review = db |
| d12933e | | | 397 | .prepare(` |
| d12933e | | | 398 | SELECT rv.*, u.username as reviewer_name |
| d12933e | | | 399 | FROM reviews rv JOIN users u ON rv.reviewer_id = u.id |
| d12933e | | | 400 | WHERE rv.id = ? |
| d12933e | | | 401 | `) |
| d12933e | | | 402 | .get(result.lastInsertRowid); |
| d12933e | | | 403 | |
| d12933e | | | 404 | return reply.code(201).send({ review }); |
| d12933e | | | 405 | }, |
| d12933e | | | 406 | } |
| d12933e | | | 407 | ); |
| d12933e | | | 408 | } |