feature/adding-log-file #1
|
|
@ -0,0 +1,30 @@
|
||||||
|
package main
|
||||||
|
|
||||||
|
import (
|
||||||
|
"fmt"
|
||||||
|
"net/http"
|
||||||
|
)
|
||||||
|
|
||||||
|
type Application struct {
|
||||||
|
Config *Config
|
||||||
|
Mux *http.ServeMux
|
||||||
|
Log *LogFile
|
||||||
|
}
|
||||||
|
|
||||||
|
func (app *Application) routes() {
|
||||||
|
m := http.NewServeMux()
|
||||||
|
|
||||||
|
m.HandleFunc("/healthcheck", healthcheck)
|
||||||
|
m.HandleFunc("/reloadRules", reloadRules(app.Config))
|
||||||
|
m.HandleFunc("/", serveLogger(app.Log)(serveRules(app.Config)))
|
||||||
|
|
||||||
|
app.Mux = m
|
||||||
|
}
|
||||||
|
|
||||||
|
func (app *Application) Setup(port string) *http.Server {
|
||||||
|
app.routes()
|
||||||
|
return &http.Server{
|
||||||
|
Addr: fmt.Sprintf(":%s", port),
|
||||||
|
Handler: app.Mux,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,19 @@
|
||||||
|
package main
|
||||||
|
|
||||||
|
import (
|
||||||
|
"embed"
|
||||||
|
"html/template"
|
||||||
|
)
|
||||||
|
|
||||||
|
// go:embed templates/*
|
||||||
|
var tmpls embed.FS
|
||||||
|
|
||||||
|
func GetServeHtml() *template.Template {
|
||||||
|
data, _ := tmpls.ReadFile("success.html")
|
||||||
|
return template.Must(template.New("main").Parse(string(data)))
|
||||||
|
}
|
||||||
|
|
||||||
|
func GetDefaultHtml() []byte {
|
||||||
|
data, _ := tmpls.ReadFile("default.html")
|
||||||
|
return data
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,71 @@
|
||||||
|
package main
|
||||||
|
|
||||||
|
import (
|
||||||
|
"bytes"
|
||||||
|
"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 {
|
||||||
|
errorLog.Printf("Cannot reload rules: %+v", err)
|
||||||
|
http.Error(w, err.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" {
|
||||||
|
w.Write(GetDefaultHtml())
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
nameOfPkg := r.Host + r.URL.Path
|
||||||
|
|
||||||
|
var vanityUrl, proto, repoUrl string
|
||||||
|
for _, rule := range c.MappingRules.Mappings {
|
||||||
|
if strings.HasPrefix(strings.ToLower(nameOfPkg), strings.ToLower(rule.VanityUrl+"/")) {
|
||||||
|
repo := strings.Replace(strings.ToLower(nameOfPkg), strings.ToLower(rule.VanityUrl), "", -1)
|
||||||
|
repo = strings.Split(repo, "/")[1]
|
||||||
|
|
||||||
|
vanityUrl = rule.VanityUrl + "/" + repo
|
||||||
|
repoUrl = rule.RealUrl + "/" + repo
|
||||||
|
proto = rule.Protocol
|
||||||
|
|
||||||
|
break
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
d := ImportRuleStruct{
|
||||||
|
VanityUrl: vanityUrl,
|
||||||
|
Proto: proto,
|
||||||
|
RepoUrl: repoUrl,
|
||||||
|
}
|
||||||
|
tmpl := GetServeHtml()
|
||||||
|
|
||||||
|
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())
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,23 @@
|
||||||
|
package main
|
||||||
|
|
||||||
|
import (
|
||||||
|
"net/http"
|
||||||
|
)
|
||||||
|
|
||||||
|
// serveLogger is a logging middleware for serving. It generates logs for
|
||||||
|
// requests sent to the server.
|
||||||
|
func serveLogger(l *LogFile) func(http.HandlerFunc) http.HandlerFunc {
|
||||||
|
return func(next http.HandlerFunc) http.HandlerFunc {
|
||||||
|
return func(w http.ResponseWriter, r *http.Request) {
|
||||||
|
l.WriteLog(r)
|
||||||
|
next(w, r)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// func serveLogger(logger *LogFile, next http.Handler) http.Handler {
|
||||||
|
// return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
||||||
|
// logger.WriteLog(r)
|
||||||
|
// next.ServeHTTP(w, r)
|
||||||
|
// })
|
||||||
|
// }
|
||||||
|
|
@ -0,0 +1,6 @@
|
||||||
|
<html>
|
||||||
|
<head></head>
|
||||||
|
<body>
|
||||||
|
<h5>Nothing here. Move along.</h5>
|
||||||
|
</body>
|
||||||
|
</html>
|
||||||
|
|
@ -0,0 +1,7 @@
|
||||||
|
<html>
|
||||||
|
<head>
|
||||||
|
<meta http-equiv="Content-Type" content="text/html; charset=utf-8"/>
|
||||||
|
<meta name="go-import" content="{{.VanityUrl}} {{.Proto}} {{.RepoUrl}}">
|
||||||
|
<meta name="go-source" content="{{.VanityUrl}} {{.RepoUrl}} {{.RepoUrl}}/tree/main{/dir} {{.RepoUrl}}/blob/main{/dir}/{file}#L{line}">
|
||||||
|
</head>
|
||||||
|
</html>
|
||||||
Loading…
Reference in New Issue