forked from tangled.org/core
Monorepo for Tangled
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(c.Server.PlcUrl), 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.CORS) 75 r.Use(h.RequestLogger) 76 77 r.Get("/", func(w http.ResponseWriter, r *http.Request) { 78 w.Write([]byte("This is a knot server. More info at https://tangled.sh")) 79 }) 80 81 r.Route("/{did}", func(r chi.Router) { 82 r.Route("/{name}", func(r chi.Router) { 83 // routes for git operations 84 r.Get("/info/refs", h.InfoRefs) 85 r.Post("/git-upload-archive", h.UploadArchive) 86 r.Post("/git-upload-pack", h.UploadPack) 87 r.Post("/git-receive-pack", h.ReceivePack) 88 }) 89 }) 90 91 // xrpc apis 92 r.Mount("/xrpc", h.XrpcRouter()) 93 94 // Socket that streams git oplogs 95 r.Get("/events", h.Events) 96 97 return r 98} 99 100func (h *Knot) XrpcRouter() http.Handler { 101 serviceAuth := serviceauth.NewServiceAuth(h.l, h.resolver, h.c.Server.Did().String()) 102 103 l := log.SubLogger(h.l, "xrpc") 104 105 xrpc := &xrpc.Xrpc{ 106 Config: h.c, 107 Db: h.db, 108 Ingester: h.jc, 109 Enforcer: h.e, 110 Logger: l, 111 Notifier: h.n, 112 Resolver: h.resolver, 113 ServiceAuth: serviceAuth, 114 } 115 116 return xrpc.Router() 117} 118 119func (h *Knot) configureOwner() error { 120 cfgOwner := h.c.Server.Owner 121 122 rbacDomain := "thisserver" 123 124 existing, err := h.e.GetKnotUsersByRole("server:owner", rbacDomain) 125 if err != nil { 126 return err 127 } 128 129 switch len(existing) { 130 case 0: 131 // no owner configured, continue 132 case 1: 133 // find existing owner 134 existingOwner := existing[0] 135 136 // no ownership change, this is okay 137 if existingOwner == h.c.Server.Owner { 138 break 139 } 140 141 // remove existing owner 142 if err = h.db.RemoveDid(existingOwner); err != nil { 143 return err 144 } 145 if err = h.e.RemoveKnotOwner(rbacDomain, existingOwner); err != nil { 146 return err 147 } 148 149 default: 150 return fmt.Errorf("more than one owner in DB, try deleting %q and starting over", h.c.Server.DBPath) 151 } 152 153 if err = h.db.AddDid(cfgOwner); err != nil { 154 return fmt.Errorf("failed to add owner to DB: %w", err) 155 } 156 if err := h.e.AddKnotOwner(rbacDomain, cfgOwner); err != nil { 157 return fmt.Errorf("failed to add owner to RBAC: %w", err) 158 } 159 160 return nil 161}