this repo has no description
tangled.org
1package state
2
3import (
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
31type 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
41func 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
109func (s *State) TID() string {
110 return s.tidClock.Next().String()
111}
112
113func (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
153func (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
162func (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
197func (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
230func (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
320func (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
357func (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
372func (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
391func (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
466func (s *State) RemoveMember(w http.ResponseWriter, r *http.Request) {
467}
468
469func (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
553func (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
580func (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
663func (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
678func (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
719func (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}