From 1b27f32cc21c4f85e475795c1033f4ad2d6cba52 Mon Sep 17 00:00:00 2001
From: Anirudh Oppiliappan <x@icyphox.sh>
Date: Wed, 14 Dec 2022 21:14:34 +0530
Subject: routes: multiplex git and web ui

---
 routes/handler.go | 41 ++++++++++++++++++++++++++++++++++++++++-
 1 file changed, 40 insertions(+), 1 deletion(-)

(limited to 'routes')

diff --git a/routes/handler.go b/routes/handler.go
index 3abc4fd..83f893f 100644
--- a/routes/handler.go
+++ b/routes/handler.go
@@ -1,26 +1,65 @@
 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.Index(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", d.RepoIndex, "GET")
+	mux.HandleFunc("/:name", dw.Multiplex, "GET", "POST")
+	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")
+
 	return mux
 }
-- 
cgit