gonotes/cmd/server/main.go

package main

import (
	"flag"
	"log"
	"net"
	"net/http"
	"os"
	"time"

	"forgejo.gwairfelin.com/max/gonotes/internal/conf"
	"forgejo.gwairfelin.com/max/gonotes/internal/middleware"
	"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")

	cacheExpiration, err := time.ParseDuration("24h")
	if err != nil {
		log.Fatal(err)
	}

	etag := middleware.NewETag("static", cacheExpiration)

	router.Handle("/", logger(http.RedirectHandler("/notes/", http.StatusFound)))
	router.Handle("/notes/", logger(http.StripPrefix("/notes", notesRouter)))
	router.Handle(
		"/static/",
		logger(
			middleware.StaticEtagMiddleware(
				*etag,
				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)
		}
	})
}