Home

ugit @b039da654c21994ef23b218c635e0dc17494f3ae - refs - log -
-
https://git.jolheiser.com/ugit.git
The code powering this h*ckin' site
ugit / internal / http / http.go
- raw
  1
  2
  3
  4
  5
  6
  7
  8
  9
 10
 11
 12
 13
 14
 15
 16
 17
 18
 19
 20
 21
 22
 23
 24
 25
 26
 27
 28
 29
 30
 31
 32
 33
 34
 35
 36
 37
 38
 39
 40
 41
 42
 43
 44
 45
 46
 47
 48
 49
 50
 51
 52
 53
 54
 55
 56
 57
 58
 59
 60
 61
 62
 63
 64
 65
 66
 67
 68
 69
 70
 71
 72
 73
 74
 75
 76
 77
 78
 79
 80
 81
 82
 83
 84
 85
 86
 87
 88
 89
 90
 91
 92
 93
 94
 95
 96
 97
 98
 99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
package http

import (
	"fmt"
	"net/http"
	"net/url"
	"strings"

	"go.jolheiser.com/ugit/assets"
	"go.jolheiser.com/ugit/internal/git"
	"go.jolheiser.com/ugit/internal/html"
	"go.jolheiser.com/ugit/internal/http/httperr"

	"github.com/go-chi/chi/v5"
	"github.com/go-chi/chi/v5/middleware"
)

// Server is the container struct for the HTTP server
type Server struct {
	port int
	mux  *chi.Mux
}

// ListenAndServe simply wraps http.ListenAndServe to contain the functionality here
func (s Server) ListenAndServe() error {
	return http.ListenAndServe(fmt.Sprintf("localhost:%d", s.port), s.mux)
}

// Settings is the configuration for the HTTP server
type Settings struct {
	Title       string
	Description string
	CloneURL    string
	Port        int
	RepoDir     string
	Profile     Profile
}

// Profile is the index profile
type Profile struct {
	Username string
	Email    string
	Links    []Link
}

// Link is a profile link
type Link struct {
	Name string
	URL  string
}

func (s Settings) goGet(repo string) string {
	u, _ := url.Parse(s.CloneURL)
	return fmt.Sprintf(`<!DOCTYPE html><title>%[1]s</title><meta name="go-import" content="%[2]s/%[1]s git %[3]s/%[1]s.git"><meta name="go-source" content="%[2]s/%[1]s _ %[3]s/%[1]s/tree/main{/dir}/{file}#L{line}">`, repo, u.Hostname(), s.CloneURL)
}

// New returns a new HTTP server
func New(settings Settings) Server {
	mux := chi.NewMux()

	mux.Use(middleware.Logger)
	mux.Use(middleware.Recoverer)

	rh := repoHandler{s: settings}
	mux.Route("/", func(r chi.Router) {
		r.Get("/", httperr.Handler(rh.index))
		r.Route("/{repo}", func(r chi.Router) {
			r.Use(rh.repoMiddleware)
			r.Get("/", func(w http.ResponseWriter, r *http.Request) {
				repo := r.Context().Value(repoCtxKey).(*git.Repo)
				if r.URL.Query().Has("go-get") {
					w.Write([]byte(settings.goGet(repo.Name())))
					return
				}
				if strings.HasSuffix(chi.URLParam(r, "repo"), ".git") {
					http.Redirect(w, r, "/"+repo.Name(), http.StatusFound)
					return
				}
				rh.repoTree("", "").ServeHTTP(w, r)
			})
			r.Get("/tree/{ref}/*", func(w http.ResponseWriter, r *http.Request) {
				rh.repoTree(chi.URLParam(r, "ref"), chi.URLParam(r, "*")).ServeHTTP(w, r)
			})
			r.Get("/refs", httperr.Handler(rh.repoRefs))
			r.Get("/log/{ref}", httperr.Handler(rh.repoLog))
			r.Get("/commit/{commit}", httperr.Handler(rh.repoCommit))
			r.Get("/commit/{commit}.patch", httperr.Handler(rh.repoPatch))

			// Protocol
			r.Get("/info/refs", httperr.Handler(rh.infoRefs))
			r.Post("/git-upload-pack", httperr.Handler(rh.uploadPack))
		})
	})

	mux.Route("/_", func(r chi.Router) {
		r.Get("/favicon.svg", func(w http.ResponseWriter, r *http.Request) {
			w.Header().Set("Content-Type", "image/svg+xml")
			w.Write(assets.LogoIcon)
		})
		r.Get("/tailwind.css", html.TailwindHandler)
	})

	return Server{mux: mux, port: settings.Port}
}

type repoHandler struct {
	s Settings
}

func (rh repoHandler) baseContext() html.BaseContext {
	return html.BaseContext{
		Title:       rh.s.Title,
		Description: rh.s.Description,
	}
}

func (rh repoHandler) repoHeaderContext(repo *git.Repo, r *http.Request) html.RepoHeaderComponentContext {
	ref := chi.URLParam(r, "ref")
	if ref == "" {
		ref, _ = repo.DefaultBranch()
	}
	return html.RepoHeaderComponentContext{
		Description: repo.Meta.Description,
		Name:        chi.URLParam(r, "repo"),
		Ref:         ref,
		CloneURL:    rh.s.CloneURL,
	}
}

// NoopLogger is a no-op logging middleware
func NoopLogger(next http.Handler) http.Handler {
	return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
		next.ServeHTTP(w, r)
	})
}