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