80 lines
1.8 KiB
Go
80 lines
1.8 KiB
Go
package main
|
|
|
|
import (
|
|
"bytes"
|
|
"fmt"
|
|
"net/http"
|
|
"strings"
|
|
)
|
|
|
|
func healthcheck(w http.ResponseWriter, r *http.Request) {
|
|
w.Write([]byte("ok"))
|
|
}
|
|
|
|
func reloadRules(c *Config) http.HandlerFunc {
|
|
return func(w http.ResponseWriter, r *http.Request) {
|
|
err := c.LoadMappingFile("")
|
|
if err != nil {
|
|
e := fmt.Errorf("annot reload rules: %+v", err)
|
|
// errorLog.Printf("Cannot reload rules: %+v", err)
|
|
http.Error(w, e.Error(), http.StatusInternalServerError)
|
|
return
|
|
}
|
|
w.Write([]byte("ok"))
|
|
}
|
|
}
|
|
|
|
func serveRules(c *Config) http.HandlerFunc {
|
|
return func(w http.ResponseWriter, r *http.Request) {
|
|
if r.Method != http.MethodGet {
|
|
http.Error(w, http.StatusText(http.StatusMethodNotAllowed), http.StatusInternalServerError)
|
|
return
|
|
}
|
|
|
|
// if go-get param is absent, return nothing
|
|
if r.FormValue("go-get") != "1" {
|
|
data, err := GetDefaultHtml()
|
|
if err != nil {
|
|
http.Error(w, err.Error(), http.StatusInternalServerError)
|
|
return
|
|
}
|
|
w.Write(data)
|
|
return
|
|
}
|
|
|
|
nameOfPkg := r.Host + r.URL.Path
|
|
|
|
var vanityUrl, proto, repoUrl string
|
|
for _, rule := range c.MappingRules.Mappings {
|
|
if strings.HasPrefix(strings.ToLower(rule.VanityUrl+"/"), strings.Trim(strings.ToLower(nameOfPkg), " ")) {
|
|
vanityUrl = rule.VanityUrl
|
|
repoUrl = rule.RealUrl
|
|
proto = rule.Protocol
|
|
|
|
break
|
|
}
|
|
}
|
|
|
|
d := ImportRuleStruct{
|
|
VanityUrl: vanityUrl,
|
|
Proto: proto,
|
|
RepoUrl: repoUrl,
|
|
}
|
|
tmpl, err := GetServeHtml()
|
|
if err != nil {
|
|
http.Error(w, err.Error(), http.StatusInternalServerError)
|
|
return
|
|
}
|
|
|
|
var buf bytes.Buffer
|
|
err = tmpl.Execute(&buf, &d)
|
|
if err != nil {
|
|
http.Error(w, err.Error(), http.StatusInternalServerError)
|
|
return
|
|
}
|
|
|
|
w.Header().Set("Cache-Control", "public, max-age=500")
|
|
w.Write(buf.Bytes())
|
|
}
|
|
}
|