package web
import (
"net/http"
"net/url"
"path"
"text/template"
"github.com/charmbracelet/soft-serve/server/backend"
"github.com/charmbracelet/soft-serve/server/config"
"github.com/charmbracelet/soft-serve/server/utils"
"github.com/prometheus/client_golang/prometheus"
"github.com/prometheus/client_golang/prometheus/promauto"
"goji.io/pattern"
)
var goGetCounter = promauto.NewCounterVec(prometheus.CounterOpts{
Namespace: "soft_serve",
Subsystem: "http",
Name: "go_get_total",
Help: "The total number of go get requests",
}, []string{"repo"})
var repoIndexHTMLTpl = template.Must(template.New("index").Parse(`
Redirecting to docs at godoc.org/{{ .ImportRoot }}/{{ .Repo }}...
`))
// GoGetHandler handles go get requests.
type GoGetHandler struct {
cfg *config.Config
be backend.Backend
}
var _ http.Handler = (*GoGetHandler)(nil)
func (g GoGetHandler) ServeHTTP(w http.ResponseWriter, r *http.Request) {
repo := pattern.Path(r.Context())
repo = utils.SanitizeRepo(repo)
be := g.be.WithContext(r.Context())
// Handle go get requests.
//
// Always return a 200 status code, even if the repo doesn't exist.
//
// https://golang.org/cmd/go/#hdr-Remote_import_paths
// https://go.dev/ref/mod#vcs-branch
if r.URL.Query().Get("go-get") == "1" {
repo := repo
importRoot, err := url.Parse(g.cfg.HTTP.PublicURL)
if err != nil {
http.Error(w, err.Error(), http.StatusInternalServerError)
return
}
// find the repo
for {
if _, err := be.Repository(repo); err == nil {
break
}
if repo == "" || repo == "." || repo == "/" {
return
}
repo = path.Dir(repo)
}
if err := repoIndexHTMLTpl.Execute(w, struct {
Repo string
Config *config.Config
ImportRoot string
}{
Repo: url.PathEscape(repo),
Config: g.cfg,
ImportRoot: importRoot.Host,
}); err != nil {
http.Error(w, err.Error(), http.StatusInternalServerError)
return
}
goGetCounter.WithLabelValues(repo).Inc()
return
}
http.NotFound(w, r)
}