forked from tangled.org/core
this repo has no description
at master 4.1 kB view raw
1package knotserver 2 3import ( 4 "context" 5 "fmt" 6 "log/slog" 7 "net/http" 8 "runtime/debug" 9 10 "github.com/go-chi/chi/v5" 11 "tangled.sh/tangled.sh/core/jetstream" 12 "tangled.sh/tangled.sh/core/knotserver/config" 13 "tangled.sh/tangled.sh/core/knotserver/db" 14 "tangled.sh/tangled.sh/core/rbac" 15) 16 17const ( 18 ThisServer = "thisserver" // resource identifier for rbac enforcement 19) 20 21type Handle struct { 22 c *config.Config 23 db *db.DB 24 jc *jetstream.JetstreamClient 25 e *rbac.Enforcer 26 l *slog.Logger 27 28 // init is a channel that is closed when the knot has been initailized 29 // i.e. when the first user (knot owner) has been added. 30 init chan struct{} 31 knotInitialized bool 32} 33 34func Setup(ctx context.Context, c *config.Config, db *db.DB, e *rbac.Enforcer, jc *jetstream.JetstreamClient, l *slog.Logger) (http.Handler, error) { 35 r := chi.NewRouter() 36 37 h := Handle{ 38 c: c, 39 db: db, 40 e: e, 41 l: l, 42 jc: jc, 43 init: make(chan struct{}), 44 } 45 46 err := e.AddDomain(ThisServer) 47 if err != nil { 48 return nil, fmt.Errorf("failed to setup enforcer: %w", err) 49 } 50 51 err = h.jc.StartJetstream(ctx, h.processMessages) 52 if err != nil { 53 return nil, fmt.Errorf("failed to start jetstream: %w", err) 54 } 55 56 // Check if the knot knows about any Dids; 57 // if it does, it is already initialized and we can repopulate the 58 // Jetstream subscriptions. 59 dids, err := db.GetAllDids() 60 if err != nil { 61 return nil, fmt.Errorf("failed to get all Dids: %w", err) 62 } 63 64 if len(dids) > 0 { 65 h.knotInitialized = true 66 close(h.init) 67 for _, d := range dids { 68 h.jc.AddDid(d) 69 } 70 } 71 72 r.Get("/", h.Index) 73 r.Get("/capabilities", h.Capabilities) 74 r.Get("/version", h.Version) 75 r.Route("/{did}", func(r chi.Router) { 76 // Repo routes 77 r.Route("/{name}", func(r chi.Router) { 78 r.Route("/collaborator", func(r chi.Router) { 79 r.Use(h.VerifySignature) 80 r.Post("/add", h.AddRepoCollaborator) 81 }) 82 83 r.Get("/", h.RepoIndex) 84 r.Get("/info/refs", h.InfoRefs) 85 r.Post("/git-upload-pack", h.UploadPack) 86 r.Get("/compare/{rev1}/{rev2}", h.Compare) // git diff-tree compare of two objects 87 88 r.With(h.VerifySignature).Post("/hidden-ref/{forkRef}/{remoteRef}", h.NewHiddenRef) 89 90 r.Route("/merge", func(r chi.Router) { 91 r.With(h.VerifySignature) 92 r.Post("/", h.Merge) 93 r.Post("/check", h.MergeCheck) 94 }) 95 96 r.Route("/tree/{ref}", func(r chi.Router) { 97 r.Get("/", h.RepoIndex) 98 r.Get("/*", h.RepoTree) 99 }) 100 101 r.Route("/blob/{ref}", func(r chi.Router) { 102 r.Get("/*", h.Blob) 103 }) 104 105 r.Get("/log/{ref}", h.Log) 106 r.Get("/archive/{file}", h.Archive) 107 r.Get("/commit/{ref}", h.Diff) 108 r.Get("/tags", h.Tags) 109 r.Route("/branches", func(r chi.Router) { 110 r.Get("/", h.Branches) 111 r.Get("/{branch}", h.Branch) 112 r.Route("/default", func(r chi.Router) { 113 r.Get("/", h.DefaultBranch) 114 r.With(h.VerifySignature).Put("/", h.SetDefaultBranch) 115 }) 116 }) 117 }) 118 }) 119 120 // Create a new repository. 121 r.Route("/repo", func(r chi.Router) { 122 r.Use(h.VerifySignature) 123 r.Put("/new", h.NewRepo) 124 r.Delete("/", h.RemoveRepo) 125 r.Post("/fork", h.RepoFork) 126 }) 127 128 r.Route("/member", func(r chi.Router) { 129 r.Use(h.VerifySignature) 130 r.Put("/add", h.AddMember) 131 }) 132 133 // Initialize the knot with an owner and public key. 134 r.With(h.VerifySignature).Post("/init", h.Init) 135 136 // Health check. Used for two-way verification with appview. 137 r.With(h.VerifySignature).Get("/health", h.Health) 138 139 // All public keys on the knot. 140 r.Get("/keys", h.Keys) 141 142 return r, nil 143} 144 145// version is set during build time. 146var version string 147 148func (h *Handle) Version(w http.ResponseWriter, r *http.Request) { 149 if version == "" { 150 info, ok := debug.ReadBuildInfo() 151 if !ok { 152 http.Error(w, "failed to read build info", http.StatusInternalServerError) 153 return 154 } 155 156 var modVer string 157 for _, mod := range info.Deps { 158 if mod.Path == "tangled.sh/tangled.sh/knotserver" { 159 version = mod.Version 160 break 161 } 162 } 163 164 if modVer == "" { 165 version = "unknown" 166 } 167 } 168 169 w.Header().Set("Content-Type", "text/plain") 170 fmt.Fprintf(w, "knotserver/%s", version) 171}