legit/routes/handler.go

package routes

import (
	"log"
	"net/http"
	"path/filepath"
	"regexp"

	"github.com/alexedwards/flow"
	"github.com/sosedoff/gitkit"
	"icyphox.sh/legit/config"
)

type depsWrapper struct {
	actualDeps deps
	gitsvc     *gitkit.Server
}

// Checks for gitprotocol-http(5) specific query params; if found, passes
// the request on to the git http service, else render the web frontend.
func (dw *depsWrapper) Multiplex(w http.ResponseWriter, r *http.Request) {
	path := flow.Param(r.Context(), "...")
	name := flow.Param(r.Context(), "name")
	name = filepath.Clean(name)
	gitCommand := regexp.MustCompile(`git-(upload|receive)-pack`)

	if path == "info/refs" && gitCommand.MatchString(r.URL.RawQuery) && r.Method == "GET" {
		dw.gitsvc.ServeHTTP(w, r)
	} else if gitCommand.MatchString(path) && r.Method == "POST" {
		dw.gitsvc.ServeHTTP(w, r)
	} else if r.Method == "GET" {
		log.Println("index:", r.URL.String())
		dw.actualDeps.RepoIndex(w, r)
	}
}

func Handlers(c *config.Config) *flow.Mux {
	mux := flow.New()
	d := deps{c}

	gitsvc := gitkit.New(gitkit.Config{
		Dir:        c.Repo.ScanPath,
		AutoCreate: false,
	})
	if err := gitsvc.Setup(); err != nil {
		log.Fatalf("git server: %s", err)
	}

	dw := depsWrapper{actualDeps: d, gitsvc: gitsvc}

	mux.NotFound = http.HandlerFunc(func(w http.ResponseWriter, _ *http.Request) {
		d.Write404(w)
	})

	mux.HandleFunc("/", d.Index, "GET")
	mux.HandleFunc("/:name", dw.Multiplex, "GET", "POST")
	mux.HandleFunc("/:name/tree/:ref/...", d.RepoTree, "GET")
	mux.HandleFunc("/:name/blob/:ref/...", d.FileContent, "GET")
	mux.HandleFunc("/:name/log/:ref", d.Log, "GET")
	mux.HandleFunc("/:name/commit/:ref", d.Diff, "GET")
	mux.HandleFunc("/:name/refs", d.Refs, "GET")
	mux.HandleFunc("/:name/...", dw.Multiplex, "GET", "POST")

	return mux
}