julien.rbrt.fr
forked from
tangled.org/core
Monorepo for Tangled — https://tangled.org
1package knotserver
2
3import (
4 "context"
5 "fmt"
6 "log/slog"
7 "net/http"
8
9 "github.com/go-chi/chi/v5"
10 "tangled.org/core/idresolver"
11 "tangled.org/core/jetstream"
12 "tangled.org/core/knotserver/config"
13 "tangled.org/core/knotserver/db"
14 "tangled.org/core/knotserver/xrpc"
15 "tangled.org/core/log"
16 "tangled.org/core/notifier"
17 "tangled.org/core/rbac"
18 "tangled.org/core/xrpc/serviceauth"
19)
20
21type Knot struct {
22 c *config.Config
23 db *db.DB
24 jc *jetstream.JetstreamClient
25 e *rbac.Enforcer
26 l *slog.Logger
27 n *notifier.Notifier
28 resolver *idresolver.Resolver
29}
30
31func Setup(ctx context.Context, c *config.Config, db *db.DB, e *rbac.Enforcer, jc *jetstream.JetstreamClient, n *notifier.Notifier) (http.Handler, error) {
32 h := Knot{
33 c: c,
34 db: db,
35 e: e,
36 l: log.FromContext(ctx),
37 jc: jc,
38 n: n,
39 resolver: idresolver.DefaultResolver(),
40 }
41
42 err := e.AddKnot(rbac.ThisServer)
43 if err != nil {
44 return nil, fmt.Errorf("failed to setup enforcer: %w", err)
45 }
46
47 // configure owner
48 if err = h.configureOwner(); err != nil {
49 return nil, err
50 }
51 h.l.Info("owner set", "did", h.c.Server.Owner)
52 h.jc.AddDid(h.c.Server.Owner)
53
54 // configure known-dids in jetstream consumer
55 dids, err := h.db.GetAllDids()
56 if err != nil {
57 return nil, fmt.Errorf("failed to get all dids: %w", err)
58 }
59 for _, d := range dids {
60 jc.AddDid(d)
61 }
62
63 err = h.jc.StartJetstream(ctx, h.processMessages)
64 if err != nil {
65 return nil, fmt.Errorf("failed to start jetstream: %w", err)
66 }
67
68 return h.Router(), nil
69}
70
71func (h *Knot) Router() http.Handler {
72 r := chi.NewRouter()
73
74 r.Use(h.RequestLogger)
75
76 r.Get("/", func(w http.ResponseWriter, r *http.Request) {
77 w.Write([]byte("This is a knot server. More info at https://tangled.sh"))
78 })
79
80 r.Route("/{did}", func(r chi.Router) {
81 r.Route("/{name}", func(r chi.Router) {
82 // routes for git operations
83 r.Get("/info/refs", h.InfoRefs)
84 r.Post("/git-upload-pack", h.UploadPack)
85 r.Post("/git-receive-pack", h.ReceivePack)
86 })
87 })
88
89 // xrpc apis
90 r.Mount("/xrpc", h.XrpcRouter())
91
92 // Socket that streams git oplogs
93 r.Get("/events", h.Events)
94
95 return r
96}
97
98func (h *Knot) XrpcRouter() http.Handler {
99 serviceAuth := serviceauth.NewServiceAuth(h.l, h.resolver, h.c.Server.Did().String())
100
101 l := log.SubLogger(h.l, "xrpc")
102
103 xrpc := &xrpc.Xrpc{
104 Config: h.c,
105 Db: h.db,
106 Ingester: h.jc,
107 Enforcer: h.e,
108 Logger: l,
109 Notifier: h.n,
110 Resolver: h.resolver,
111 ServiceAuth: serviceAuth,
112 }
113
114 return xrpc.Router()
115}
116
117func (h *Knot) configureOwner() error {
118 cfgOwner := h.c.Server.Owner
119
120 rbacDomain := "thisserver"
121
122 existing, err := h.e.GetKnotUsersByRole("server:owner", rbacDomain)
123 if err != nil {
124 return err
125 }
126
127 switch len(existing) {
128 case 0:
129 // no owner configured, continue
130 case 1:
131 // find existing owner
132 existingOwner := existing[0]
133
134 // no ownership change, this is okay
135 if existingOwner == h.c.Server.Owner {
136 break
137 }
138
139 // remove existing owner
140 if err = h.db.RemoveDid(existingOwner); err != nil {
141 return err
142 }
143 if err = h.e.RemoveKnotOwner(rbacDomain, existingOwner); err != nil {
144 return err
145 }
146
147 default:
148 return fmt.Errorf("more than one owner in DB, try deleting %q and starting over", h.c.Server.DBPath)
149 }
150
151 if err = h.db.AddDid(cfgOwner); err != nil {
152 return fmt.Errorf("failed to add owner to DB: %w", err)
153 }
154 if err := h.e.AddKnotOwner(rbacDomain, cfgOwner); err != nil {
155 return fmt.Errorf("failed to add owner to RBAC: %w", err)
156 }
157
158 return nil
159}