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