66 lines
1.4 KiB
Go
66 lines
1.4 KiB
Go
package main
|
|
|
|
import (
|
|
"flag"
|
|
"log"
|
|
"net"
|
|
"net/http"
|
|
"os"
|
|
|
|
"forgejo.gwairfelin.com/max/gonotes/internal/conf"
|
|
"forgejo.gwairfelin.com/max/gonotes/internal/notes/views"
|
|
)
|
|
|
|
func main() {
|
|
var confFile string
|
|
|
|
flag.StringVar(&confFile, "c", "/etc/gonotes/conf.toml", "Specify path to config file.")
|
|
flag.Parse()
|
|
|
|
conf.LoadConfig(confFile)
|
|
|
|
log.SetOutput(os.Stdout)
|
|
|
|
router := http.NewServeMux()
|
|
notesRouter := views.GetRoutes("/notes")
|
|
|
|
router.Handle("/", logger(http.RedirectHandler("/notes/", http.StatusFound)))
|
|
router.Handle("/notes/", logger(http.StripPrefix("/notes", notesRouter)))
|
|
router.Handle(
|
|
"/static/",
|
|
logger(
|
|
http.FileServer(http.FS(conf.Static)),
|
|
),
|
|
)
|
|
|
|
listener, err := net.Listen(conf.Conf.Protocol, conf.Conf.Address)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
log.Fatal(http.Serve(listener, router))
|
|
}
|
|
|
|
type loggingResponseWriter struct {
|
|
http.ResponseWriter
|
|
statusCode int
|
|
}
|
|
|
|
func NewLoggingResponseWriter(w http.ResponseWriter) *loggingResponseWriter {
|
|
return &loggingResponseWriter{w, http.StatusOK}
|
|
}
|
|
|
|
func (w *loggingResponseWriter) WriteHeader(code int) {
|
|
w.statusCode = code
|
|
w.ResponseWriter.WriteHeader(code)
|
|
}
|
|
|
|
func logger(next http.Handler) http.Handler {
|
|
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
lwr := NewLoggingResponseWriter(w)
|
|
next.ServeHTTP(lwr, r)
|
|
|
|
if conf.Conf.LogAccess {
|
|
log.Printf("%s %s %s%s %d", r.RemoteAddr, r.Method, r.Host, r.URL.Path, lwr.statusCode)
|
|
}
|
|
})
|
|
}
|