tjh.dev
forked from
tangled.org/core
this repo has no description
Revert "appview/db: transactions all the things"
This reverts commit 4d8e06b62e423f07df5623af237973e9a4384b7a.
oppi.li
+2
-2
appview/db/db.go
+2
-2
appview/db/db.go
+4
-11
appview/db/pubkeys.go
+4
-11
appview/db/pubkeys.go
···
1
package db
2
3
import (
4
-
"database/sql"
5
"encoding/json"
6
"time"
7
)
8
9
func (d *DB) AddPublicKey(did, name, key string) error {
10
query := `insert into public_keys (did, name, key) values (?, ?, ?)`
11
-
_, err := d.Db.Exec(query, did, name, key)
12
-
return err
13
-
}
14
-
15
-
func (d *DB) AddPublicKeyTx(tx *sql.Tx, did, name, key string) error {
16
-
query := `insert into public_keys (did, name, key) values (?, ?, ?)`
17
-
_, err := tx.Exec(query, did, name, key)
18
return err
19
}
20
21
func (d *DB) RemovePublicKey(did string) error {
22
query := `delete from public_keys where did = ?`
23
-
_, err := d.Db.Exec(query, did)
24
return err
25
}
26
···
45
func (d *DB) GetAllPublicKeys() ([]PublicKey, error) {
46
var keys []PublicKey
47
48
-
rows, err := d.Db.Query(`select key, name, did, created from public_keys`)
49
if err != nil {
50
return nil, err
51
}
···
71
func (d *DB) GetPublicKeys(did string) ([]PublicKey, error) {
72
var keys []PublicKey
73
74
-
rows, err := d.Db.Query(`select did, key, name, created from public_keys where did = ?`, did)
75
if err != nil {
76
return nil, err
77
}
···
1
package db
2
3
import (
4
"encoding/json"
5
"time"
6
)
7
8
func (d *DB) AddPublicKey(did, name, key string) error {
9
query := `insert into public_keys (did, name, key) values (?, ?, ?)`
10
+
_, err := d.db.Exec(query, did, name, key)
11
return err
12
}
13
14
func (d *DB) RemovePublicKey(did string) error {
15
query := `delete from public_keys where did = ?`
16
+
_, err := d.db.Exec(query, did)
17
return err
18
}
19
···
38
func (d *DB) GetAllPublicKeys() ([]PublicKey, error) {
39
var keys []PublicKey
40
41
+
rows, err := d.db.Query(`select key, name, did, created from public_keys`)
42
if err != nil {
43
return nil, err
44
}
···
64
func (d *DB) GetPublicKeys(did string) ([]PublicKey, error) {
65
var keys []PublicKey
66
67
+
rows, err := d.db.Query(`select did, key, name, created from public_keys where did = ?`, did)
68
if err != nil {
69
return nil, err
70
}
+5
-58
appview/db/registration.go
+5
-58
appview/db/registration.go
···
35
func (d *DB) RegistrationsByDid(did string) ([]Registration, error) {
36
var registrations []Registration
37
38
-
rows, err := d.Db.Query(`
39
select domain, did, created, registered from registrations
40
where did = ?
41
`, did)
···
75
var registeredAt *int64
76
var registration Registration
77
78
-
err := d.Db.QueryRow(`
79
-
select domain, did, created, registered from registrations
80
-
where domain = ?
81
-
`, domain).Scan(®istration.Domain, ®istration.ByDid, &createdAt, ®isteredAt)
82
-
83
-
if err != nil {
84
-
if err == sql.ErrNoRows {
85
-
return nil, nil
86
-
} else {
87
-
return nil, err
88
-
}
89
-
}
90
-
91
-
createdAtTime := time.Unix(*createdAt, 0)
92
-
var registeredAtTime *time.Time
93
-
if registeredAt != nil {
94
-
x := time.Unix(*registeredAt, 0)
95
-
registeredAtTime = &x
96
-
}
97
-
98
-
registration.Created = &createdAtTime
99
-
registration.Registered = registeredAtTime
100
-
101
-
return ®istration, nil
102
-
}
103
-
104
-
func (d *DB) RegistrationByDomainTx(tx *sql.Tx, domain string) (*Registration, error) {
105
-
var createdAt *int64
106
-
var registeredAt *int64
107
-
var registration Registration
108
-
109
-
err := tx.QueryRow(`
110
select domain, did, created, registered from registrations
111
where domain = ?
112
`, domain).Scan(®istration.Domain, ®istration.ByDid, &createdAt, ®isteredAt)
···
153
154
secret := uuid.New().String()
155
156
-
_, err = d.Db.Exec(`
157
insert into registrations (domain, did, secret)
158
values (?, ?, ?)
159
on conflict(domain) do update set did = excluded.did, secret = excluded.secret
···
167
}
168
169
func (d *DB) GetRegistrationKey(domain string) (string, error) {
170
-
res := d.Db.QueryRow(`select secret from registrations where domain = ?`, domain)
171
-
172
-
var secret string
173
-
err := res.Scan(&secret)
174
-
if err != nil || secret == "" {
175
-
return "", err
176
-
}
177
-
178
-
return secret, nil
179
-
}
180
-
181
-
func (d *DB) GetRegistrationKeyTx(tx *sql.Tx, domain string) (string, error) {
182
-
res := tx.QueryRow(`select secret from registrations where domain = ?`, domain)
183
184
var secret string
185
err := res.Scan(&secret)
···
191
}
192
193
func (d *DB) Register(domain string) error {
194
-
_, err := d.Db.Exec(`
195
-
update registrations
196
-
set registered = strftime('%s', 'now')
197
-
where domain = ?;
198
-
`, domain)
199
-
200
-
return err
201
-
}
202
-
203
-
func (d *DB) RegisterTx(tx *sql.Tx, domain string) error {
204
-
_, err := tx.Exec(`
205
update registrations
206
set registered = strftime('%s', 'now')
207
where domain = ?;
···
35
func (d *DB) RegistrationsByDid(did string) ([]Registration, error) {
36
var registrations []Registration
37
38
+
rows, err := d.db.Query(`
39
select domain, did, created, registered from registrations
40
where did = ?
41
`, did)
···
75
var registeredAt *int64
76
var registration Registration
77
78
+
err := d.db.QueryRow(`
79
select domain, did, created, registered from registrations
80
where domain = ?
81
`, domain).Scan(®istration.Domain, ®istration.ByDid, &createdAt, ®isteredAt)
···
122
123
secret := uuid.New().String()
124
125
+
_, err = d.db.Exec(`
126
insert into registrations (domain, did, secret)
127
values (?, ?, ?)
128
on conflict(domain) do update set did = excluded.did, secret = excluded.secret
···
136
}
137
138
func (d *DB) GetRegistrationKey(domain string) (string, error) {
139
+
res := d.db.QueryRow(`select secret from registrations where domain = ?`, domain)
140
141
var secret string
142
err := res.Scan(&secret)
···
148
}
149
150
func (d *DB) Register(domain string) error {
151
+
_, err := d.db.Exec(`
152
update registrations
153
set registered = strftime('%s', 'now')
154
where domain = ?;
+4
-11
appview/db/repos.go
+4
-11
appview/db/repos.go
···
1
package db
2
3
-
import "database/sql"
4
-
5
type Repo struct {
6
Did string
7
Name string
···
12
func (d *DB) GetAllReposByDid(did string) ([]Repo, error) {
13
var repos []Repo
14
15
-
rows, err := d.Db.Query(`select did, name, knot, created from repos where did = ?`, did)
16
if err != nil {
17
return nil, err
18
}
···
38
func (d *DB) GetRepo(did, name string) (*Repo, error) {
39
var repo Repo
40
41
-
row := d.Db.QueryRow(`select did, name, knot, created from repos where did = ? and name = ?`, did, name)
42
var createdAt *int64
43
if err := row.Scan(&repo.Did, &repo.Name, &repo.Knot, &createdAt); err != nil {
44
return nil, err
···
49
}
50
51
func (d *DB) AddRepo(repo *Repo) error {
52
-
_, err := d.Db.Exec(`insert into repos (did, name, knot) values (?, ?, ?)`, repo.Did, repo.Name, repo.Knot)
53
-
return err
54
-
}
55
-
56
-
func (d *DB) AddRepoTx(tx *sql.Tx, repo *Repo) error {
57
-
_, err := tx.Exec(`insert into repos (did, name, knot) values (?, ?, ?)`, repo.Did, repo.Name, repo.Knot)
58
return err
59
}
60
61
func (d *DB) RemoveRepo(did, name, knot string) error {
62
-
_, err := d.Db.Exec(`delete from repos where did = ? and name = ? and knot = ?`, did, name, knot)
63
return err
64
}
···
1
package db
2
3
type Repo struct {
4
Did string
5
Name string
···
10
func (d *DB) GetAllReposByDid(did string) ([]Repo, error) {
11
var repos []Repo
12
13
+
rows, err := d.db.Query(`select did, name, knot, created from repos where did = ?`, did)
14
if err != nil {
15
return nil, err
16
}
···
36
func (d *DB) GetRepo(did, name string) (*Repo, error) {
37
var repo Repo
38
39
+
row := d.db.QueryRow(`select did, name, knot, created from repos where did = ? and name = ?`, did, name)
40
var createdAt *int64
41
if err := row.Scan(&repo.Did, &repo.Name, &repo.Knot, &createdAt); err != nil {
42
return nil, err
···
47
}
48
49
func (d *DB) AddRepo(repo *Repo) error {
50
+
_, err := d.db.Exec(`insert into repos (did, name, knot) values (?, ?, ?)`, repo.Did, repo.Name, repo.Knot)
51
return err
52
}
53
54
func (d *DB) RemoveRepo(did, name, knot string) error {
55
+
_, err := d.db.Exec(`delete from repos where did = ? and name = ? and knot = ?`, did, name, knot)
56
return err
57
}
+32
-100
appview/state/state.go
+32
-100
appview/state/state.go
···
200
return
201
}
202
203
-
// Start transaction
204
-
tx, err := s.db.Db.Begin()
205
-
if err != nil {
206
-
log.Printf("failed to start transaction: %s", err)
207
-
http.Error(w, "Internal server error", http.StatusInternalServerError)
208
-
return
209
-
}
210
-
defer tx.Rollback() // Will rollback if not committed
211
-
212
-
if err := s.db.AddPublicKeyTx(tx, did, name, key); err != nil {
213
log.Printf("adding public key: %s", err)
214
return
215
}
···
232
return
233
}
234
235
-
// If everything succeeded, commit the transaction
236
-
if err := tx.Commit(); err != nil {
237
-
log.Printf("failed to commit transaction: %s", err)
238
-
http.Error(w, "Internal server error", http.StatusInternalServerError)
239
-
return
240
-
}
241
-
242
log.Println("created atproto record: ", resp.Uri)
243
244
return
···
256
}
257
log.Println("checking ", domain)
258
259
-
// Start transaction
260
-
tx, err := s.db.Db.Begin()
261
-
if err != nil {
262
-
log.Printf("failed to start transaction: %s", err)
263
-
http.Error(w, "Internal server error", http.StatusInternalServerError)
264
-
return
265
-
}
266
-
defer tx.Rollback() // Will rollback if not committed
267
-
268
-
secret, err := s.db.GetRegistrationKeyTx(tx, domain)
269
if err != nil {
270
log.Printf("no key found for domain %s: %s\n", domain, err)
271
return
···
310
return
311
}
312
313
-
// mark as registered within transaction
314
-
err = s.db.RegisterTx(tx, domain)
315
if err != nil {
316
log.Println("failed to register domain", err)
317
http.Error(w, err.Error(), http.StatusInternalServerError)
318
return
319
}
320
321
-
// set permissions for this did as owner within transaction
322
-
reg, err := s.db.RegistrationByDomainTx(tx, domain)
323
if err != nil {
324
log.Println("failed to register domain", err)
325
http.Error(w, err.Error(), http.StatusInternalServerError)
326
return
327
}
328
329
-
// add basic acls for this domain within transaction
330
err = s.enforcer.AddDomain(domain)
331
if err != nil {
332
log.Println("failed to setup owner of domain", err)
···
334
return
335
}
336
337
-
// add this did as owner of this domain within transaction
338
err = s.enforcer.AddOwner(domain, reg.ByDid)
339
if err != nil {
340
log.Println("failed to setup owner of domain", err)
341
http.Error(w, err.Error(), http.StatusInternalServerError)
342
-
return
343
-
}
344
-
345
-
// Commit transaction
346
-
if err := tx.Commit(); err != nil {
347
-
log.Printf("failed to commit transaction: %s", err)
348
-
http.Error(w, "Internal server error", http.StatusInternalServerError)
349
return
350
}
351
···
443
}
444
log.Printf("adding %s to %s\n", memberIdent.Handle.String(), domain)
445
446
-
// Start transaction
447
-
tx, err := s.db.Db.Begin()
448
if err != nil {
449
-
log.Printf("failed to start transaction: %s", err)
450
-
http.Error(w, "Internal server error", http.StatusInternalServerError)
451
return
452
}
453
-
defer tx.Rollback() // Will rollback if not committed
454
455
-
// Get registration key within transaction
456
-
secret, err := s.db.GetRegistrationKeyTx(tx, domain)
457
if err != nil {
458
log.Printf("no key found for domain %s: %s\n", domain, err)
459
return
460
}
461
462
-
// Make the external call to the knot server
463
ksClient, err := NewSignedClient(domain, secret)
464
if err != nil {
465
log.Println("failed to create client to ", domain)
···
468
469
ksResp, err := ksClient.AddMember(memberIdent.DID.String(), []string{})
470
if err != nil {
471
-
log.Printf("failed to make request to %s: %s", domain, err)
472
-
return
473
}
474
475
if ksResp.StatusCode != http.StatusNoContent {
···
477
return
478
}
479
480
-
// Create ATProto record within transaction
481
-
client, _ := s.auth.AuthorizedClient(r)
482
-
currentUser := s.auth.GetUser(r)
483
-
addedAt := time.Now().Format(time.RFC3339)
484
-
resp, err := comatproto.RepoPutRecord(r.Context(), client, &comatproto.RepoPutRecord_Input{
485
-
Collection: tangled.KnotMemberNSID,
486
-
Repo: currentUser.Did,
487
-
Rkey: s.TID(),
488
-
Record: &lexutil.LexiconTypeDecoder{
489
-
Val: &tangled.KnotMember{
490
-
Member: memberIdent.DID.String(),
491
-
Domain: domain,
492
-
AddedAt: &addedAt,
493
-
}},
494
-
})
495
-
if err != nil {
496
-
log.Printf("failed to create record: %s", err)
497
-
return
498
-
}
499
-
500
-
// Update RBAC within transaction
501
err = s.enforcer.AddMember(domain, memberIdent.DID.String())
502
if err != nil {
503
w.Write([]byte(fmt.Sprint("failed to add member: ", err)))
504
return
505
}
506
507
-
// If everything succeeded, commit the transaction
508
-
if err := tx.Commit(); err != nil {
509
-
log.Printf("failed to commit transaction: %s", err)
510
-
http.Error(w, "Internal server error", http.StatusInternalServerError)
511
-
return
512
-
}
513
-
514
-
log.Println("created atproto record: ", resp.Uri)
515
w.Write([]byte(fmt.Sprint("added member: ", memberIdent.Handle.String())))
516
}
517
···
545
return
546
}
547
548
-
// Start transaction
549
-
tx, err := s.db.Db.Begin()
550
-
if err != nil {
551
-
log.Printf("failed to start transaction: %s", err)
552
-
http.Error(w, "Internal server error", http.StatusInternalServerError)
553
-
return
554
-
}
555
-
defer tx.Rollback() // Will rollback if not committed
556
-
557
-
secret, err := s.db.GetRegistrationKeyTx(tx, domain)
558
if err != nil {
559
log.Printf("no key found for domain %s: %s\n", domain, err)
560
return
···
563
client, err := NewSignedClient(domain, secret)
564
if err != nil {
565
log.Println("failed to create client to ", domain)
566
-
return
567
}
568
569
resp, err := client.NewRepo(user.Did, repoName)
···
576
return
577
}
578
579
-
// add to local db within transaction
580
repo := &db.Repo{
581
Did: user.Did,
582
Name: repoName,
583
Knot: domain,
584
}
585
-
err = s.db.AddRepoTx(tx, repo)
586
if err != nil {
587
log.Println("failed to add repo to db", err)
588
return
589
}
590
591
-
// acls within transaction
592
err = s.enforcer.AddRepo(user.Did, domain, filepath.Join(user.Did, repoName))
593
if err != nil {
594
log.Println("failed to set up acls", err)
595
-
return
596
-
}
597
-
598
-
// Commit transaction
599
-
if err := tx.Commit(); err != nil {
600
-
log.Printf("failed to commit transaction: %s", err)
601
-
http.Error(w, "Internal server error", http.StatusInternalServerError)
602
return
603
}
604
···
200
return
201
}
202
203
+
if err := s.db.AddPublicKey(did, name, key); err != nil {
204
log.Printf("adding public key: %s", err)
205
return
206
}
···
223
return
224
}
225
226
log.Println("created atproto record: ", resp.Uri)
227
228
return
···
240
}
241
log.Println("checking ", domain)
242
243
+
secret, err := s.db.GetRegistrationKey(domain)
244
if err != nil {
245
log.Printf("no key found for domain %s: %s\n", domain, err)
246
return
···
285
return
286
}
287
288
+
// mark as registered
289
+
err = s.db.Register(domain)
290
if err != nil {
291
log.Println("failed to register domain", err)
292
http.Error(w, err.Error(), http.StatusInternalServerError)
293
return
294
}
295
296
+
// set permissions for this did as owner
297
+
reg, err := s.db.RegistrationByDomain(domain)
298
if err != nil {
299
log.Println("failed to register domain", err)
300
http.Error(w, err.Error(), http.StatusInternalServerError)
301
return
302
}
303
304
+
// add basic acls for this domain
305
err = s.enforcer.AddDomain(domain)
306
if err != nil {
307
log.Println("failed to setup owner of domain", err)
···
309
return
310
}
311
312
+
// add this did as owner of this domain
313
err = s.enforcer.AddOwner(domain, reg.ByDid)
314
if err != nil {
315
log.Println("failed to setup owner of domain", err)
316
http.Error(w, err.Error(), http.StatusInternalServerError)
317
return
318
}
319
···
411
}
412
log.Printf("adding %s to %s\n", memberIdent.Handle.String(), domain)
413
414
+
// announce this relation into the firehose, store into owners' pds
415
+
client, _ := s.auth.AuthorizedClient(r)
416
+
currentUser := s.auth.GetUser(r)
417
+
addedAt := time.Now().Format(time.RFC3339)
418
+
resp, err := comatproto.RepoPutRecord(r.Context(), client, &comatproto.RepoPutRecord_Input{
419
+
Collection: tangled.KnotMemberNSID,
420
+
Repo: currentUser.Did,
421
+
Rkey: s.TID(),
422
+
Record: &lexutil.LexiconTypeDecoder{
423
+
Val: &tangled.KnotMember{
424
+
Member: memberIdent.DID.String(),
425
+
Domain: domain,
426
+
AddedAt: &addedAt,
427
+
}},
428
+
})
429
+
// invalid record
430
if err != nil {
431
+
log.Printf("failed to create record: %s", err)
432
return
433
}
434
+
log.Println("created atproto record: ", resp.Uri)
435
436
+
secret, err := s.db.GetRegistrationKey(domain)
437
if err != nil {
438
log.Printf("no key found for domain %s: %s\n", domain, err)
439
return
440
}
441
442
ksClient, err := NewSignedClient(domain, secret)
443
if err != nil {
444
log.Println("failed to create client to ", domain)
···
447
448
ksResp, err := ksClient.AddMember(memberIdent.DID.String(), []string{})
449
if err != nil {
450
+
log.Printf("failet to make request to %s: %s", domain, err)
451
}
452
453
if ksResp.StatusCode != http.StatusNoContent {
···
455
return
456
}
457
458
err = s.enforcer.AddMember(domain, memberIdent.DID.String())
459
if err != nil {
460
w.Write([]byte(fmt.Sprint("failed to add member: ", err)))
461
return
462
}
463
464
w.Write([]byte(fmt.Sprint("added member: ", memberIdent.Handle.String())))
465
}
466
···
494
return
495
}
496
497
+
secret, err := s.db.GetRegistrationKey(domain)
498
if err != nil {
499
log.Printf("no key found for domain %s: %s\n", domain, err)
500
return
···
503
client, err := NewSignedClient(domain, secret)
504
if err != nil {
505
log.Println("failed to create client to ", domain)
506
}
507
508
resp, err := client.NewRepo(user.Did, repoName)
···
515
return
516
}
517
518
+
// add to local db
519
repo := &db.Repo{
520
Did: user.Did,
521
Name: repoName,
522
Knot: domain,
523
}
524
+
err = s.db.AddRepo(repo)
525
if err != nil {
526
log.Println("failed to add repo to db", err)
527
return
528
}
529
530
+
// acls
531
err = s.enforcer.AddRepo(user.Did, domain, filepath.Join(user.Did, repoName))
532
if err != nil {
533
log.Println("failed to set up acls", err)
534
return
535
}
536