veryroundbird.house
1package db
2
3import (
4 "context"
5 "database/sql"
6 "fmt"
7 "log"
8
9 _ "github.com/mattn/go-sqlite3"
10)
11
12type DB struct {
13 *sql.DB
14}
15
16type Execer interface {
17 Query(query string, args ...any) (*sql.Rows, error)
18 QueryContext(ctx context.Context, query string, args ...any) (*sql.Rows, error)
19 QueryRow(query string, args ...any) *sql.Row
20 QueryRowContext(ctx context.Context, query string, args ...any) *sql.Row
21 Exec(query string, args ...any) (sql.Result, error)
22 ExecContext(ctx context.Context, query string, args ...any) (sql.Result, error)
23 Prepare(query string) (*sql.Stmt, error)
24 PrepareContext(ctx context.Context, query string) (*sql.Stmt, error)
25}
26
27func Make(dbPath string) (*DB, error) {
28 db, err := sql.Open("sqlite3", dbPath)
29 if err != nil {
30 return nil, err
31 }
32 _, err = db.Exec(`
33 pragma journal_mode = WAL;
34 pragma synchronous = normal;
35 pragma foreign_keys = on;
36 pragma temp_store = memory;
37 pragma mmap_size = 30000000000;
38 pragma page_size = 32768;
39 pragma auto_vacuum = incremental;
40 pragma busy_timeout = 5000;
41
42 create table if not exists registrations (
43 id integer primary key autoincrement,
44 domain text not null unique,
45 did text not null,
46 secret text not null,
47 created text not null default (strftime('%Y-%m-%dT%H:%M:%SZ', 'now')),
48 registered text
49 );
50 create table if not exists public_keys (
51 id integer primary key autoincrement,
52 did text not null,
53 name text not null,
54 key text not null,
55 created text not null default (strftime('%Y-%m-%dT%H:%M:%SZ', 'now')),
56 unique(did, name, key)
57 );
58 create table if not exists repos (
59 id integer primary key autoincrement,
60 did text not null,
61 name text not null,
62 knot text not null,
63 rkey text not null,
64 at_uri text not null unique,
65 created text not null default (strftime('%Y-%m-%dT%H:%M:%SZ', 'now')),
66 unique(did, name, knot, rkey)
67 );
68 create table if not exists collaborators (
69 id integer primary key autoincrement,
70 did text not null,
71 repo integer not null,
72 foreign key (repo) references repos(id) on delete cascade
73 );
74 create table if not exists follows (
75 user_did text not null,
76 subject_did text not null,
77 rkey text not null,
78 followed_at text not null default (strftime('%Y-%m-%dT%H:%M:%SZ', 'now')),
79 primary key (user_did, subject_did),
80 check (user_did <> subject_did)
81 );
82 create table if not exists issues (
83 id integer primary key autoincrement,
84 owner_did text not null,
85 repo_at text not null,
86 issue_id integer not null,
87 title text not null,
88 body text not null,
89 open integer not null default 1,
90 created text not null default (strftime('%Y-%m-%dT%H:%M:%SZ', 'now')),
91 issue_at text,
92 unique(repo_at, issue_id),
93 foreign key (repo_at) references repos(at_uri) on delete cascade
94 );
95 create table if not exists comments (
96 id integer primary key autoincrement,
97 owner_did text not null,
98 issue_id integer not null,
99 repo_at text not null,
100 comment_id integer not null,
101 comment_at text not null,
102 body text not null,
103 created text not null default (strftime('%Y-%m-%dT%H:%M:%SZ', 'now')),
104 unique(issue_id, comment_id),
105 foreign key (repo_at, issue_id) references issues(repo_at, issue_id) on delete cascade
106 );
107 create table if not exists pulls (
108 -- identifiers
109 id integer primary key autoincrement,
110 pull_id integer not null,
111
112 -- at identifiers
113 repo_at text not null,
114 owner_did text not null,
115 rkey text not null,
116 pull_at text,
117
118 -- content
119 title text not null,
120 body text not null,
121 target_branch text not null,
122 state integer not null default 0 check (state in (0, 1, 2)), -- open, merged, closed
123
124 -- meta
125 created text not null default (strftime('%Y-%m-%dT%H:%M:%SZ', 'now')),
126
127 -- constraints
128 unique(repo_at, pull_id),
129 foreign key (repo_at) references repos(at_uri) on delete cascade
130 );
131
132 -- every pull must have atleast 1 submission: the initial submission
133 create table if not exists pull_submissions (
134 -- identifiers
135 id integer primary key autoincrement,
136 pull_id integer not null,
137
138 -- at identifiers
139 repo_at text not null,
140
141 -- content, these are immutable, and require a resubmission to update
142 round_number integer not null default 0,
143 patch text,
144
145 -- meta
146 created text not null default (strftime('%Y-%m-%dT%H:%M:%SZ', 'now')),
147
148 -- constraints
149 unique(repo_at, pull_id, round_number),
150 foreign key (repo_at, pull_id) references pulls(repo_at, pull_id) on delete cascade
151 );
152
153 create table if not exists pull_comments (
154 -- identifiers
155 id integer primary key autoincrement,
156 pull_id integer not null,
157 submission_id integer not null,
158
159 -- at identifiers
160 repo_at text not null,
161 owner_did text not null,
162 comment_at text not null,
163
164 -- content
165 body text not null,
166
167 -- meta
168 created text not null default (strftime('%Y-%m-%dT%H:%M:%SZ', 'now')),
169
170 -- constraints
171 foreign key (repo_at, pull_id) references pulls(repo_at, pull_id) on delete cascade,
172 foreign key (submission_id) references pull_submissions(id) on delete cascade
173 );
174
175 create table if not exists _jetstream (
176 id integer primary key autoincrement,
177 last_time_us integer not null
178 );
179
180 create table if not exists repo_issue_seqs (
181 repo_at text primary key,
182 next_issue_id integer not null default 1
183 );
184
185 create table if not exists repo_pull_seqs (
186 repo_at text primary key,
187 next_pull_id integer not null default 1
188 );
189
190 create table if not exists stars (
191 id integer primary key autoincrement,
192 starred_by_did text not null,
193 repo_at text not null,
194 rkey text not null,
195 created text not null default (strftime('%Y-%m-%dT%H:%M:%SZ', 'now')),
196 foreign key (repo_at) references repos(at_uri) on delete cascade,
197 unique(starred_by_did, repo_at)
198 );
199
200 create table if not exists emails (
201 id integer primary key autoincrement,
202 did text not null,
203 email text not null,
204 verified integer not null default 0,
205 verification_code text not null,
206 last_sent text not null default (strftime('%Y-%m-%dT%H:%M:%SZ', 'now')),
207 is_primary integer not null default 0,
208 created text not null default (strftime('%Y-%m-%dT%H:%M:%SZ', 'now')),
209 unique(did, email)
210 );
211
212 create table if not exists artifacts (
213 -- id
214 id integer primary key autoincrement,
215 did text not null,
216 rkey text not null,
217
218 -- meta
219 repo_at text not null,
220 tag binary(20) not null,
221 created text not null default (strftime('%Y-%m-%dT%H:%M:%SZ', 'now')),
222
223 -- data
224 blob_cid text not null,
225 name text not null,
226 size integer not null default 0,
227 mimetype string not null default "*/*",
228
229 -- constraints
230 unique(did, rkey), -- record must be unique
231 unique(repo_at, tag, name), -- for a given tag object, each file must be unique
232 foreign key (repo_at) references repos(at_uri) on delete cascade
233 );
234
235 create table if not exists profile (
236 -- id
237 id integer primary key autoincrement,
238 did text not null,
239
240 -- data
241 description text not null,
242 include_bluesky integer not null default 0,
243 location text,
244
245 -- constraints
246 unique(did)
247 );
248 create table if not exists profile_links (
249 -- id
250 id integer primary key autoincrement,
251 did text not null,
252
253 -- data
254 link text not null,
255
256 -- constraints
257 foreign key (did) references profile(did) on delete cascade
258 );
259 create table if not exists profile_stats (
260 -- id
261 id integer primary key autoincrement,
262 did text not null,
263
264 -- data
265 kind text not null check (kind in (
266 "merged-pull-request-count",
267 "closed-pull-request-count",
268 "open-pull-request-count",
269 "open-issue-count",
270 "closed-issue-count",
271 "repository-count"
272 )),
273
274 -- constraints
275 foreign key (did) references profile(did) on delete cascade
276 );
277 create table if not exists profile_pinned_repositories (
278 -- id
279 id integer primary key autoincrement,
280 did text not null,
281
282 -- data
283 at_uri text not null,
284
285 -- constraints
286 unique(did, at_uri),
287 foreign key (did) references profile(did) on delete cascade,
288 foreign key (at_uri) references repos(at_uri) on delete cascade
289 );
290
291 create table if not exists oauth_requests (
292 id integer primary key autoincrement,
293 auth_server_iss text not null,
294 state text not null,
295 did text not null,
296 handle text not null,
297 pds_url text not null,
298 pkce_verifier text not null,
299 dpop_auth_server_nonce text not null,
300 dpop_private_jwk text not null
301 );
302
303 create table if not exists oauth_sessions (
304 id integer primary key autoincrement,
305 did text not null,
306 handle text not null,
307 pds_url text not null,
308 auth_server_iss text not null,
309 access_jwt text not null,
310 refresh_jwt text not null,
311 dpop_pds_nonce text,
312 dpop_auth_server_nonce text not null,
313 dpop_private_jwk text not null,
314 expiry text not null
315 );
316
317 create table if not exists punchcard (
318 did text not null,
319 date text not null, -- yyyy-mm-dd
320 count integer,
321 primary key (did, date)
322 );
323
324 create table if not exists spindles (
325 id integer primary key autoincrement,
326 owner text not null,
327 instance text not null,
328 verified text, -- time of verification
329 created text not null default (strftime('%Y-%m-%dT%H:%M:%SZ', 'now')),
330
331 unique(instance)
332 );
333
334 create table if not exists migrations (
335 id integer primary key autoincrement,
336 name text unique
337 )
338 `)
339 if err != nil {
340 return nil, err
341 }
342
343 // run migrations
344 runMigration(db, "add-description-to-repos", func(tx *sql.Tx) error {
345 tx.Exec(`
346 alter table repos add column description text check (length(description) <= 200);
347 `)
348 return nil
349 })
350
351 runMigration(db, "add-rkey-to-pubkeys", func(tx *sql.Tx) error {
352 // add unconstrained column
353 _, err := tx.Exec(`
354 alter table public_keys
355 add column rkey text;
356 `)
357 if err != nil {
358 return err
359 }
360
361 // backfill
362 _, err = tx.Exec(`
363 update public_keys
364 set rkey = ''
365 where rkey is null;
366 `)
367 if err != nil {
368 return err
369 }
370
371 return nil
372 })
373
374 runMigration(db, "add-rkey-to-comments", func(tx *sql.Tx) error {
375 _, err := tx.Exec(`
376 alter table comments drop column comment_at;
377 alter table comments add column rkey text;
378 `)
379 return err
380 })
381
382 runMigration(db, "add-deleted-and-edited-to-issue-comments", func(tx *sql.Tx) error {
383 _, err := tx.Exec(`
384 alter table comments add column deleted text; -- timestamp
385 alter table comments add column edited text; -- timestamp
386 `)
387 return err
388 })
389
390 runMigration(db, "add-source-info-to-pulls-and-submissions", func(tx *sql.Tx) error {
391 _, err := tx.Exec(`
392 alter table pulls add column source_branch text;
393 alter table pulls add column source_repo_at text;
394 alter table pull_submissions add column source_rev text;
395 `)
396 return err
397 })
398
399 runMigration(db, "add-source-to-repos", func(tx *sql.Tx) error {
400 _, err := tx.Exec(`
401 alter table repos add column source text;
402 `)
403 return err
404 })
405
406 // disable foreign-keys for the next migration
407 // NOTE: this cannot be done in a transaction, so it is run outside [0]
408 //
409 // [0]: https://sqlite.org/pragma.html#pragma_foreign_keys
410 db.Exec("pragma foreign_keys = off;")
411 runMigration(db, "recreate-pulls-column-for-stacking-support", func(tx *sql.Tx) error {
412 _, err := tx.Exec(`
413 create table pulls_new (
414 -- identifiers
415 id integer primary key autoincrement,
416 pull_id integer not null,
417
418 -- at identifiers
419 repo_at text not null,
420 owner_did text not null,
421 rkey text not null,
422
423 -- content
424 title text not null,
425 body text not null,
426 target_branch text not null,
427 state integer not null default 0 check (state in (0, 1, 2, 3)), -- closed, open, merged, deleted
428
429 -- source info
430 source_branch text,
431 source_repo_at text,
432
433 -- stacking
434 stack_id text,
435 change_id text,
436 parent_change_id text,
437
438 -- meta
439 created text not null default (strftime('%Y-%m-%dT%H:%M:%SZ', 'now')),
440
441 -- constraints
442 unique(repo_at, pull_id),
443 foreign key (repo_at) references repos(at_uri) on delete cascade
444 );
445
446 insert into pulls_new (
447 id, pull_id,
448 repo_at, owner_did, rkey,
449 title, body, target_branch, state,
450 source_branch, source_repo_at,
451 created
452 )
453 select
454 id, pull_id,
455 repo_at, owner_did, rkey,
456 title, body, target_branch, state,
457 source_branch, source_repo_at,
458 created
459 FROM pulls;
460
461 drop table pulls;
462 alter table pulls_new rename to pulls;
463 `)
464 return err
465 })
466 db.Exec("pragma foreign_keys = on;")
467
468 // run migrations
469 runMigration(db, "add-spindle-to-repos", func(tx *sql.Tx) error {
470 tx.Exec(`
471 alter table repos add column spindle text;
472 `)
473 return nil
474 })
475
476 return &DB{db}, nil
477}
478
479type migrationFn = func(*sql.Tx) error
480
481func runMigration(d *sql.DB, name string, migrationFn migrationFn) error {
482 tx, err := d.Begin()
483 if err != nil {
484 return err
485 }
486 defer tx.Rollback()
487
488 var exists bool
489 err = tx.QueryRow("select exists (select 1 from migrations where name = ?)", name).Scan(&exists)
490 if err != nil {
491 return err
492 }
493
494 if !exists {
495 // run migration
496 err = migrationFn(tx)
497 if err != nil {
498 log.Printf("Failed to run migration %s: %v", name, err)
499 return err
500 }
501
502 // mark migration as complete
503 _, err = tx.Exec("insert into migrations (name) values (?)", name)
504 if err != nil {
505 log.Printf("Failed to mark migration %s as complete: %v", name, err)
506 return err
507 }
508
509 // commit the transaction
510 if err := tx.Commit(); err != nil {
511 return err
512 }
513
514 log.Printf("migration %s applied successfully", name)
515 } else {
516 log.Printf("skipped migration %s, already applied", name)
517 }
518
519 return nil
520}
521
522type filter struct {
523 key string
524 arg any
525 cmp string
526}
527
528func newFilter(key, cmp string, arg any) filter {
529 return filter{
530 key: key,
531 arg: arg,
532 cmp: cmp,
533 }
534}
535
536func FilterEq(key string, arg any) filter { return newFilter(key, "=", arg) }
537func FilterNotEq(key string, arg any) filter { return newFilter(key, "<>", arg) }
538func FilterGte(key string, arg any) filter { return newFilter(key, ">=", arg) }
539func FilterLte(key string, arg any) filter { return newFilter(key, "<=", arg) }
540func FilterIs(key string, arg any) filter { return newFilter(key, "is", arg) }
541func FilterIsNot(key string, arg any) filter { return newFilter(key, "is not", arg) }
542
543func (f filter) Condition() string {
544 return fmt.Sprintf("%s %s ?", f.key, f.cmp)
545}