forked from tangled.org/core
Something went wrong. Try again.
5.1 kB · 211 lines
Go
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212package knotserver
import ( "context" "fmt" "log/slog" "net/http" "runtime/debug"
"github.com/go-chi/chi/v5" "tangled.sh/tangled.sh/core/idresolver" "tangled.sh/tangled.sh/core/jetstream" "tangled.sh/tangled.sh/core/knotserver/config" "tangled.sh/tangled.sh/core/knotserver/db" "tangled.sh/tangled.sh/core/knotserver/xrpc" tlog "tangled.sh/tangled.sh/core/log" "tangled.sh/tangled.sh/core/notifier" "tangled.sh/tangled.sh/core/rbac")
type Handle struct { c *config.Config db *db.DB jc *jetstream.JetstreamClient e *rbac.Enforcer l *slog.Logger n *notifier.Notifier resolver *idresolver.Resolver
// init is a channel that is closed when the knot has been initailized // i.e. when the first user (knot owner) has been added. init chan struct{} knotInitialized bool}
func Setup(ctx context.Context, c *config.Config, db *db.DB, e *rbac.Enforcer, jc *jetstream.JetstreamClient, l *slog.Logger, n *notifier.Notifier) (http.Handler, error) { r := chi.NewRouter()
h := Handle{ c: c, db: db, e: e, l: l, jc: jc, n: n, resolver: idresolver.DefaultResolver(), init: make(chan struct{}), }
err := e.AddKnot(rbac.ThisServer) if err != nil { return nil, fmt.Errorf("failed to setup enforcer: %w", err) }
err = h.jc.StartJetstream(ctx, h.processMessages) if err != nil { return nil, fmt.Errorf("failed to start jetstream: %w", err) }
// Check if the knot knows about any Dids; // if it does, it is already initialized and we can repopulate the // Jetstream subscriptions. dids, err := db.GetAllDids() if err != nil { return nil, fmt.Errorf("failed to get all Dids: %w", err) }
if len(dids) > 0 { h.knotInitialized = true close(h.init) for _, d := range dids { h.jc.AddDid(d) } }
r.Get("/", h.Index) r.Get("/capabilities", h.Capabilities) r.Get("/version", h.Version) r.Route("/{did}", func(r chi.Router) { // Repo routes r.Route("/{name}", func(r chi.Router) { r.Route("/collaborator", func(r chi.Router) { r.Use(h.VerifySignature) r.Post("/add", h.AddRepoCollaborator) })
r.Route("/languages", func(r chi.Router) { r.With(h.VerifySignature) r.Get("/", h.RepoLanguages) r.Get("/{ref}", h.RepoLanguages) })
r.Get("/", h.RepoIndex) r.Get("/info/refs", h.InfoRefs) r.Post("/git-upload-pack", h.UploadPack) r.Post("/git-receive-pack", h.ReceivePack) r.Get("/compare/{rev1}/{rev2}", h.Compare) // git diff-tree compare of two objects
r.With(h.VerifySignature).Post("/hidden-ref/{forkRef}/{remoteRef}", h.NewHiddenRef)
r.Route("/merge", func(r chi.Router) { r.With(h.VerifySignature) r.Post("/", h.Merge) r.Post("/check", h.MergeCheck) })
r.Route("/tree/{ref}", func(r chi.Router) { r.Get("/", h.RepoIndex) r.Get("/*", h.RepoTree) })
r.Route("/blob/{ref}", func(r chi.Router) { r.Get("/*", h.Blob) })
r.Route("/raw/{ref}", func(r chi.Router) { r.Get("/*", h.BlobRaw) })
r.Get("/log/{ref}", h.Log) r.Get("/archive/{file}", h.Archive) r.Get("/commit/{ref}", h.Diff) r.Get("/tags", h.Tags) r.Route("/branches", func(r chi.Router) { r.Get("/", h.Branches) r.Get("/{branch}", h.Branch) r.Route("/default", func(r chi.Router) { r.Get("/", h.DefaultBranch) r.With(h.VerifySignature).Put("/", h.SetDefaultBranch) }) }) }) })
// xrpc apis r.Mount("/xrpc", h.XrpcRouter())
// Create a new repository. r.Route("/repo", func(r chi.Router) { r.Use(h.VerifySignature) r.Put("/new", h.NewRepo) r.Delete("/", h.RemoveRepo) r.Route("/fork", func(r chi.Router) { r.Post("/", h.RepoFork) r.Post("/sync/{branch}", h.RepoForkSync) r.Get("/sync/{branch}", h.RepoForkAheadBehind) }) })
r.Route("/member", func(r chi.Router) { r.Use(h.VerifySignature) r.Put("/add", h.AddMember) })
// Socket that streams git oplogs r.Get("/events", h.Events)
// Initialize the knot with an owner and public key. r.With(h.VerifySignature).Post("/init", h.Init)
// Health check. Used for two-way verification with appview. r.With(h.VerifySignature).Get("/health", h.Health)
// All public keys on the knot. r.Get("/keys", h.Keys)
return r, nil}
func (h *Handle) XrpcRouter() http.Handler { logger := tlog.New("knots")
xrpc := &xrpc.Xrpc{ Config: h.c, Db: h.db, Ingester: h.jc, Enforcer: h.e, Logger: logger, Notifier: h.n, Resolver: h.resolver, } return xrpc.Router()}
// version is set during build time.var version string
func (h *Handle) Version(w http.ResponseWriter, r *http.Request) { if version == "" { info, ok := debug.ReadBuildInfo() if !ok { http.Error(w, "failed to read build info", http.StatusInternalServerError) return }
var modVer string for _, mod := range info.Deps { if mod.Path == "tangled.sh/tangled.sh/knotserver" { version = mod.Version break } }
if modVer == "" { version = "unknown" } }
w.Header().Set("Content-Type", "text/plain; charset=utf-8") fmt.Fprintf(w, "knotserver/%s", version)}