760 lines
19 kB
1
package state
2
3
import (
4
"context"
5
"crypto/hmac"
6
"crypto/sha256"
7
"encoding/hex"
8
"encoding/json"
9
"fmt"
10
"log"
11
"net/http"
12
"path/filepath"
13
"strings"
14
"time"
15
16
comatproto "github.com/bluesky-social/indigo/api/atproto"
17
"github.com/bluesky-social/indigo/atproto/syntax"
18
lexutil "github.com/bluesky-social/indigo/lex/util"
19
"github.com/bluesky-social/jetstream/pkg/models"
20
"github.com/go-chi/chi/v5"
21
tangled "github.com/sotangled/tangled/api/tangled"
22
"github.com/sotangled/tangled/appview"
23
"github.com/sotangled/tangled/appview/auth"
24
"github.com/sotangled/tangled/appview/db"
25
"github.com/sotangled/tangled/appview/pages"
26
"github.com/sotangled/tangled/jetstream"
27
"github.com/sotangled/tangled/rbac"
28
)
29
30
type State struct {
31
db *db.DB
32
auth *auth.Auth
33
enforcer *rbac.Enforcer
34
tidClock *syntax.TIDClock
35
pages *pages.Pages
36
resolver *appview.Resolver
37
jc *jetstream.JetstreamClient
38
}
39
40
func Make() (*State, error) {
41
db, err := db.Make(appview.SqliteDbPath)
42
if err != nil {
43
return nil, err
44
}
45
46
auth, err := auth.Make()
47
if err != nil {
48
return nil, err
49
}
50
51
enforcer, err := rbac.NewEnforcer(appview.SqliteDbPath)
52
if err != nil {
53
return nil, err
54
}
55
56
clock := syntax.NewTIDClock(0)
57
58
pgs := pages.NewPages()
59
60
resolver := appview.NewResolver()
61
62
jc, err := jetstream.NewJetstreamClient("appview", []string{tangled.GraphFollowNSID}, nil, db)
63
if err != nil {
64
return nil, fmt.Errorf("failed to create jetstream client: %w", err)
65
}
66
err = jc.StartJetstream(context.Background(), func(ctx context.Context, e *models.Event) error {
67
did := e.Did
68
raw := e.Commit.Record
69
70
switch e.Commit.Collection {
71
case tangled.GraphFollowNSID:
72
record := tangled.GraphFollow{}
73
err := json.Unmarshal(raw, &record)
74
if err != nil {
75
return err
76
}
77
err = db.AddFollow(did, record.Subject, e.Commit.RKey)
78
if err != nil {
79
return fmt.Errorf("failed to add follow to db: %w", err)
80
}
81
}
82
83
return nil
84
})
85
if err != nil {
86
return nil, fmt.Errorf("failed to start jetstream watcher: %w", err)
87
}
88
89
state := &State{
90
db,
91
auth,
92
enforcer,
93
clock,
94
pgs,
95
resolver,
96
jc,
97
}
98
99
return state, nil
100
}
101
102
func (s *State) TID() string {
103
return s.tidClock.Next().String()
104
}
105
106
func (s *State) Login(w http.ResponseWriter, r *http.Request) {
107
ctx := r.Context()
108
109
switch r.Method {
110
case http.MethodGet:
111
err := s.pages.Login(w, pages.LoginParams{})
112
if err != nil {
113
log.Printf("rendering login page: %s", err)
114
}
115
return
116
case http.MethodPost:
117
handle := strings.TrimPrefix(r.FormValue("handle"), "@")
118
appPassword := r.FormValue("app_password")
119
120
resolved, err := s.resolver.ResolveIdent(ctx, handle)
121
if err != nil {
122
s.pages.Notice(w, "login-msg", fmt.Sprintf("\"%s\" is an invalid handle.", handle))
123
return
124
}
125
126
atSession, err := s.auth.CreateInitialSession(ctx, resolved, appPassword)
127
if err != nil {
128
s.pages.Notice(w, "login-msg", "Invalid handle or password.")
129
return
130
}
131
sessionish := auth.CreateSessionWrapper{ServerCreateSession_Output: atSession}
132
133
err = s.auth.StoreSession(r, w, &sessionish, resolved.PDSEndpoint())
134
if err != nil {
135
s.pages.Notice(w, "login-msg", "Failed to login, try again later.")
136
return
137
}
138
139
log.Printf("successfully saved session for %s (%s)", atSession.Handle, atSession.Did)
140
s.pages.HxRedirect(w, "/")
141
return
142
}
143
}
144
145
func (s *State) Timeline(w http.ResponseWriter, r *http.Request) {
146
user := s.auth.GetUser(r)
147
s.pages.Timeline(w, pages.TimelineParams{
148
LoggedInUser: user,
149
})
150
return
151
}
152
153
// requires auth
154
func (s *State) RegistrationKey(w http.ResponseWriter, r *http.Request) {
155
switch r.Method {
156
case http.MethodGet:
157
// list open registrations under this did
158
159
return
160
case http.MethodPost:
161
session, err := s.auth.Store.Get(r, appview.SessionName)
162
if err != nil || session.IsNew {
163
log.Println("unauthorized attempt to generate registration key")
164
http.Error(w, "Forbidden", http.StatusUnauthorized)
165
return
166
}
167
168
did := session.Values[appview.SessionDid].(string)
169
170
// check if domain is valid url, and strip extra bits down to just host
171
domain := r.FormValue("domain")
172
if domain == "" {
173
http.Error(w, "Invalid form", http.StatusBadRequest)
174
return
175
}
176
177
key, err := s.db.GenerateRegistrationKey(domain, did)
178
179
if err != nil {
180
log.Println(err)
181
http.Error(w, "unable to register this domain", http.StatusNotAcceptable)
182
return
183
}
184
185
w.Write([]byte(key))
186
}
187
}
188
189
func (s *State) Keys(w http.ResponseWriter, r *http.Request) {
190
user := chi.URLParam(r, "user")
191
user = strings.TrimPrefix(user, "@")
192
193
if user == "" {
194
w.WriteHeader(http.StatusBadRequest)
195
return
196
}
197
198
id, err := s.resolver.ResolveIdent(r.Context(), user)
199
if err != nil {
200
w.WriteHeader(http.StatusInternalServerError)
201
return
202
}
203
204
pubKeys, err := s.db.GetPublicKeys(id.DID.String())
205
if err != nil {
206
w.WriteHeader(http.StatusNotFound)
207
return
208
}
209
210
if len(pubKeys) == 0 {
211
w.WriteHeader(http.StatusNotFound)
212
return
213
}
214
215
for _, k := range pubKeys {
216
key := strings.TrimRight(k.Key, "\n")
217
w.Write([]byte(fmt.Sprintln(key)))
218
}
219
}
220
221
// create a signed request and check if a node responds to that
222
func (s *State) InitKnotServer(w http.ResponseWriter, r *http.Request) {
223
user := s.auth.GetUser(r)
224
225
domain := chi.URLParam(r, "domain")
226
if domain == "" {
227
http.Error(w, "malformed url", http.StatusBadRequest)
228
return
229
}
230
log.Println("checking ", domain)
231
232
secret, err := s.db.GetRegistrationKey(domain)
233
if err != nil {
234
log.Printf("no key found for domain %s: %s\n", domain, err)
235
return
236
}
237
238
client, err := NewSignedClient(domain, secret)
239
if err != nil {
240
log.Println("failed to create client to ", domain)
241
}
242
243
resp, err := client.Init(user.Did)
244
if err != nil {
245
w.Write([]byte("no dice"))
246
log.Println("domain was unreachable after 5 seconds")
247
return
248
}
249
250
if resp.StatusCode == http.StatusConflict {
251
log.Println("status conflict", resp.StatusCode)
252
w.Write([]byte("already registered, sorry!"))
253
return
254
}
255
256
if resp.StatusCode != http.StatusNoContent {
257
log.Println("status nok", resp.StatusCode)
258
w.Write([]byte("no dice"))
259
return
260
}
261
262
// verify response mac
263
signature := resp.Header.Get("X-Signature")
264
signatureBytes, err := hex.DecodeString(signature)
265
if err != nil {
266
return
267
}
268
269
expectedMac := hmac.New(sha256.New, []byte(secret))
270
expectedMac.Write([]byte("ok"))
271
272
if !hmac.Equal(expectedMac.Sum(nil), signatureBytes) {
273
log.Printf("response body signature mismatch: %x\n", signatureBytes)
274
return
275
}
276
277
// mark as registered
278
err = s.db.Register(domain)
279
if err != nil {
280
log.Println("failed to register domain", err)
281
http.Error(w, err.Error(), http.StatusInternalServerError)
282
return
283
}
284
285
// set permissions for this did as owner
286
reg, err := s.db.RegistrationByDomain(domain)
287
if err != nil {
288
log.Println("failed to register domain", err)
289
http.Error(w, err.Error(), http.StatusInternalServerError)
290
return
291
}
292
293
// add basic acls for this domain
294
err = s.enforcer.AddDomain(domain)
295
if err != nil {
296
log.Println("failed to setup owner of domain", err)
297
http.Error(w, err.Error(), http.StatusInternalServerError)
298
return
299
}
300
301
// add this did as owner of this domain
302
err = s.enforcer.AddOwner(domain, reg.ByDid)
303
if err != nil {
304
log.Println("failed to setup owner of domain", err)
305
http.Error(w, err.Error(), http.StatusInternalServerError)
306
return
307
}
308
309
w.Write([]byte("check success"))
310
}
311
312
func (s *State) KnotServerInfo(w http.ResponseWriter, r *http.Request) {
313
domain := chi.URLParam(r, "domain")
314
if domain == "" {
315
http.Error(w, "malformed url", http.StatusBadRequest)
316
return
317
}
318
319
user := s.auth.GetUser(r)
320
reg, err := s.db.RegistrationByDomain(domain)
321
if err != nil {
322
w.Write([]byte("failed to pull up registration info"))
323
return
324
}
325
326
var members []string
327
if reg.Registered != nil {
328
members, err = s.enforcer.GetUserByRole("server:member", domain)
329
if err != nil {
330
w.Write([]byte("failed to fetch member list"))
331
return
332
}
333
}
334
335
ok, err := s.enforcer.IsServerOwner(user.Did, domain)
336
isOwner := err == nil && ok
337
338
p := pages.KnotParams{
339
LoggedInUser: user,
340
Registration: reg,
341
Members: members,
342
IsOwner: isOwner,
343
}
344
345
s.pages.Knot(w, p)
346
}
347
348
// get knots registered by this user
349
func (s *State) Knots(w http.ResponseWriter, r *http.Request) {
350
// for now, this is just pubkeys
351
user := s.auth.GetUser(r)
352
registrations, err := s.db.RegistrationsByDid(user.Did)
353
if err != nil {
354
log.Println(err)
355
}
356
357
s.pages.Knots(w, pages.KnotsParams{
358
LoggedInUser: user,
359
Registrations: registrations,
360
})
361
}
362
363
// list members of domain, requires auth and requires owner status
364
func (s *State) ListMembers(w http.ResponseWriter, r *http.Request) {
365
domain := chi.URLParam(r, "domain")
366
if domain == "" {
367
http.Error(w, "malformed url", http.StatusBadRequest)
368
return
369
}
370
371
// list all members for this domain
372
memberDids, err := s.enforcer.GetUserByRole("server:member", domain)
373
if err != nil {
374
w.Write([]byte("failed to fetch member list"))
375
return
376
}
377
378
w.Write([]byte(strings.Join(memberDids, "\n")))
379
return
380
}
381
382
// add member to domain, requires auth and requires invite access
383
func (s *State) AddMember(w http.ResponseWriter, r *http.Request) {
384
domain := chi.URLParam(r, "domain")
385
if domain == "" {
386
http.Error(w, "malformed url", http.StatusBadRequest)
387
return
388
}
389
390
memberDid := r.FormValue("member")
391
if memberDid == "" {
392
http.Error(w, "malformed form", http.StatusBadRequest)
393
return
394
}
395
396
memberIdent, err := s.resolver.ResolveIdent(r.Context(), memberDid)
397
if err != nil {
398
w.Write([]byte("failed to resolve member did to a handle"))
399
return
400
}
401
log.Printf("adding %s to %s\n", memberIdent.Handle.String(), domain)
402
403
// announce this relation into the firehose, store into owners' pds
404
client, _ := s.auth.AuthorizedClient(r)
405
currentUser := s.auth.GetUser(r)
406
addedAt := time.Now().Format(time.RFC3339)
407
resp, err := comatproto.RepoPutRecord(r.Context(), client, &comatproto.RepoPutRecord_Input{
408
Collection: tangled.KnotMemberNSID,
409
Repo: currentUser.Did,
410
Rkey: s.TID(),
411
Record: &lexutil.LexiconTypeDecoder{
412
Val: &tangled.KnotMember{
413
Member: memberIdent.DID.String(),
414
Domain: domain,
415
AddedAt: &addedAt,
416
}},
417
})
418
419
// invalid record
420
if err != nil {
421
log.Printf("failed to create record: %s", err)
422
return
423
}
424
log.Println("created atproto record: ", resp.Uri)
425
426
secret, err := s.db.GetRegistrationKey(domain)
427
if err != nil {
428
log.Printf("no key found for domain %s: %s\n", domain, err)
429
return
430
}
431
432
ksClient, err := NewSignedClient(domain, secret)
433
if err != nil {
434
log.Println("failed to create client to ", domain)
435
return
436
}
437
438
ksResp, err := ksClient.AddMember(memberIdent.DID.String())
439
if err != nil {
440
log.Printf("failed to make request to %s: %s", domain, err)
441
return
442
}
443
444
if ksResp.StatusCode != http.StatusNoContent {
445
w.Write([]byte(fmt.Sprint("knotserver failed to add member: ", err)))
446
return
447
}
448
449
err = s.enforcer.AddMember(domain, memberIdent.DID.String())
450
if err != nil {
451
w.Write([]byte(fmt.Sprint("failed to add member: ", err)))
452
return
453
}
454
455
w.Write([]byte(fmt.Sprint("added member: ", memberIdent.Handle.String())))
456
}
457
458
func (s *State) RemoveMember(w http.ResponseWriter, r *http.Request) {
459
}
460
461
func (s *State) AddRepo(w http.ResponseWriter, r *http.Request) {
462
switch r.Method {
463
case http.MethodGet:
464
user := s.auth.GetUser(r)
465
knots, err := s.enforcer.GetDomainsForUser(user.Did)
466
467
if err != nil {
468
s.pages.Notice(w, "repo", "Invalid user account.")
469
return
470
}
471
472
s.pages.NewRepo(w, pages.NewRepoParams{
473
LoggedInUser: user,
474
Knots: knots,
475
})
476
case http.MethodPost:
477
user := s.auth.GetUser(r)
478
479
domain := r.FormValue("domain")
480
if domain == "" {
481
s.pages.Notice(w, "repo", "Invalid form submission—missing knot domain.")
482
return
483
}
484
485
repoName := r.FormValue("name")
486
if repoName == "" {
487
s.pages.Notice(w, "repo", "Invalid repo name.")
488
return
489
}
490
491
ok, err := s.enforcer.E.Enforce(user.Did, domain, domain, "repo:create")
492
if err != nil || !ok {
493
s.pages.Notice(w, "repo", "You do not have permission to create a repo in this knot.")
494
return
495
}
496
497
secret, err := s.db.GetRegistrationKey(domain)
498
if err != nil {
499
s.pages.Notice(w, "repo", fmt.Sprintf("No registration key found for knot %s.", domain))
500
return
501
}
502
503
client, err := NewSignedClient(domain, secret)
504
if err != nil {
505
s.pages.Notice(w, "repo", "Failed to connect to knot server.")
506
return
507
}
508
509
resp, err := client.NewRepo(user.Did, repoName)
510
if err != nil {
511
s.pages.Notice(w, "repo", "Failed to create repository on knot server.")
512
return
513
}
514
if resp.StatusCode != http.StatusNoContent {
515
s.pages.Notice(w, "repo", fmt.Sprintf("Server returned unexpected status: %d", resp.StatusCode))
516
return
517
}
518
519
// add to local db
520
repo := &db.Repo{
521
Did: user.Did,
522
Name: repoName,
523
Knot: domain,
524
}
525
err = s.db.AddRepo(repo)
526
if err != nil {
527
s.pages.Notice(w, "repo", "Failed to save repository information.")
528
return
529
}
530
531
// acls
532
err = s.enforcer.AddRepo(user.Did, domain, filepath.Join(user.Did, repoName))
533
if err != nil {
534
s.pages.Notice(w, "repo", "Failed to set up repository permissions.")
535
return
536
}
537
538
s.pages.HxLocation(w, fmt.Sprintf("/@%s/%s", user.Handle, repoName))
539
return
540
}
541
}
542
543
func (s *State) ProfilePage(w http.ResponseWriter, r *http.Request) {
544
didOrHandle := chi.URLParam(r, "user")
545
if didOrHandle == "" {
546
http.Error(w, "Bad request", http.StatusBadRequest)
547
return
548
}
549
550
ident, err := s.resolver.ResolveIdent(r.Context(), didOrHandle)
551
if err != nil {
552
log.Printf("resolving identity: %s", err)
553
w.WriteHeader(http.StatusNotFound)
554
return
555
}
556
557
repos, err := s.db.GetAllReposByDid(ident.DID.String())
558
if err != nil {
559
log.Printf("getting repos for %s: %s", ident.DID.String(), err)
560
}
561
562
s.pages.ProfilePage(w, pages.ProfilePageParams{
563
LoggedInUser: s.auth.GetUser(r),
564
UserDid: ident.DID.String(),
565
UserHandle: ident.Handle.String(),
566
Repos: repos,
567
})
568
}
569
570
func (s *State) Follow(w http.ResponseWriter, r *http.Request) {
571
currentUser := s.auth.GetUser(r)
572
573
subject := r.URL.Query().Get("subject")
574
if subject == "" {
575
log.Println("invalid form")
576
return
577
}
578
579
subjectIdent, err := s.resolver.ResolveIdent(r.Context(), subject)
580
if err != nil {
581
log.Println("failed to follow, invalid did")
582
}
583
584
if currentUser.Did == subjectIdent.DID.String() {
585
log.Println("cant follow or unfollow yourself")
586
return
587
}
588
589
client, _ := s.auth.AuthorizedClient(r)
590
591
switch r.Method {
592
case http.MethodPost:
593
createdAt := time.Now().Format(time.RFC3339)
594
rkey := s.TID()
595
resp, err := comatproto.RepoPutRecord(r.Context(), client, &comatproto.RepoPutRecord_Input{
596
Collection: tangled.GraphFollowNSID,
597
Repo: currentUser.Did,
598
Rkey: rkey,
599
Record: &lexutil.LexiconTypeDecoder{
600
Val: &tangled.GraphFollow{
601
Subject: subjectIdent.DID.String(),
602
CreatedAt: createdAt,
603
}},
604
})
605
if err != nil {
606
log.Println("failed to create atproto record", err)
607
return
608
}
609
610
err = s.db.AddFollow(currentUser.Did, subjectIdent.DID.String(), rkey)
611
if err != nil {
612
log.Println("failed to follow", err)
613
return
614
}
615
616
log.Println("created atproto record: ", resp.Uri)
617
618
return
619
case http.MethodDelete:
620
// find the record in the db
621
622
follow, err := s.db.GetFollow(currentUser.Did, subjectIdent.DID.String())
623
if err != nil {
624
log.Println("failed to get follow relationship")
625
return
626
}
627
628
resp, err := comatproto.RepoDeleteRecord(r.Context(), client, &comatproto.RepoDeleteRecord_Input{
629
Collection: tangled.GraphFollowNSID,
630
Repo: currentUser.Did,
631
Rkey: follow.RKey,
632
})
633
634
log.Println(resp.Commit.Cid)
635
636
if err != nil {
637
log.Println("failed to unfollow")
638
return
639
}
640
641
err = s.db.DeleteFollow(currentUser.Did, subjectIdent.DID.String())
642
if err != nil {
643
log.Println("failed to delete follow from DB")
644
// this is not an issue, the firehose event might have already done this
645
}
646
647
w.WriteHeader(http.StatusNoContent)
648
return
649
}
650
651
}
652
653
func (s *State) Router() http.Handler {
654
router := chi.NewRouter()
655
656
router.HandleFunc("/*", func(w http.ResponseWriter, r *http.Request) {
657
pat := chi.URLParam(r, "*")
658
if strings.HasPrefix(pat, "did:") || strings.HasPrefix(pat, "@") {
659
s.UserRouter().ServeHTTP(w, r)
660
} else {
661
s.StandardRouter().ServeHTTP(w, r)
662
}
663
})
664
665
return router
666
}
667
668
func (s *State) UserRouter() http.Handler {
669
r := chi.NewRouter()
670
671
// strip @ from user
672
r.Use(StripLeadingAt)
673
674
r.With(ResolveIdent(s)).Route("/{user}", func(r chi.Router) {
675
r.Get("/", s.ProfilePage)
676
r.With(ResolveRepoKnot(s)).Route("/{repo}", func(r chi.Router) {
677
r.Get("/", s.RepoIndex)
678
r.Get("/log/{ref}", s.RepoLog)
679
r.Route("/tree/{ref}", func(r chi.Router) {
680
r.Get("/*", s.RepoTree)
681
})
682
r.Get("/commit/{ref}", s.RepoCommit)
683
r.Get("/branches", s.RepoBranches)
684
r.Get("/tags", s.RepoTags)
685
r.Get("/blob/{ref}/*", s.RepoBlob)
686
687
// These routes get proxied to the knot
688
r.Get("/info/refs", s.InfoRefs)
689
r.Post("/git-upload-pack", s.UploadPack)
690
691
// settings routes, needs auth
692
r.Group(func(r chi.Router) {
693
r.With(RepoPermissionMiddleware(s, "repo:settings")).Route("/settings", func(r chi.Router) {
694
r.Get("/", s.RepoSettings)
695
r.With(RepoPermissionMiddleware(s, "repo:invite")).Put("/collaborator", s.AddCollaborator)
696
})
697
})
698
})
699
})
700
701
r.NotFound(func(w http.ResponseWriter, r *http.Request) {
702
s.pages.Error404(w)
703
})
704
705
return r
706
}
707
708
func (s *State) StandardRouter() http.Handler {
709
r := chi.NewRouter()
710
711
r.Handle("/static/*", s.pages.Static())
712
713
r.Get("/", s.Timeline)
714
715
r.Get("/login", s.Login)
716
r.Post("/login", s.Login)
717
718
r.Route("/knots", func(r chi.Router) {
719
r.Use(AuthMiddleware(s))
720
r.Get("/", s.Knots)
721
r.Post("/key", s.RegistrationKey)
722
723
r.Route("/{domain}", func(r chi.Router) {
724
r.Post("/init", s.InitKnotServer)
725
r.Get("/", s.KnotServerInfo)
726
r.Route("/member", func(r chi.Router) {
727
r.Use(RoleMiddleware(s, "server:owner"))
728
r.Get("/", s.ListMembers)
729
r.Put("/", s.AddMember)
730
r.Delete("/", s.RemoveMember)
731
})
732
})
733
})
734
735
r.Route("/repo", func(r chi.Router) {
736
r.Route("/new", func(r chi.Router) {
737
r.Get("/", s.AddRepo)
738
r.Post("/", s.AddRepo)
739
})
740
// r.Post("/import", s.ImportRepo)
741
})
742
743
r.With(AuthMiddleware(s)).Route("/follow", func(r chi.Router) {
744
r.Post("/", s.Follow)
745
r.Delete("/", s.Follow)
746
})
747
748
r.Route("/settings", func(r chi.Router) {
749
r.Use(AuthMiddleware(s))
750
r.Get("/", s.Settings)
751
r.Put("/keys", s.SettingsKeys)
752
})
753
754
r.Get("/keys/{user}", s.Keys)
755
756
r.NotFound(func(w http.ResponseWriter, r *http.Request) {
757
s.pages.Error404(w)
758
})
759
return r
760
}
761