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}