blog/router/router.go

package router

import (
	"net"
	"net/http"

	"go.jolheiser.com/blog/post"
	"go.jolheiser.com/blog/static"

	"github.com/go-chi/chi/v5"
	"github.com/go-chi/chi/v5/middleware"
	"github.com/rs/zerolog/log"
)

func New(blog *post.Blog) *chi.Mux {
	m := chi.NewMux()

	m.Use(middleware.Logger)
	m.Use(middleware.Recoverer)

	m.Get("/", indexHandler(blog))
	m.Get("/{post}", fileHandler(blog))
	m.Route("/_", func(r chi.Router) {
		r.Get("/reload", reloadHandler(blog))
		r.Get("/sakura.css", static.SakuraCSS)
	})

	return m
}

func indexHandler(blog *post.Blog) http.HandlerFunc {
	return func(w http.ResponseWriter, r *http.Request) {
		if err := static.IndexTemplate.Execute(w, map[string]interface{}{
			"Blog": blog,
		}); err != nil {
			log.Error().Err(err).Msg("could not execute template")
		}
	}
}

func fileHandler(blog *post.Blog) http.HandlerFunc {
	return func(w http.ResponseWriter, r *http.Request) {
		postName := chi.URLParam(r, "post")
		p, ok := blog.Post(postName)
		if !ok {
			w.WriteHeader(http.StatusNotFound)
			return
		}

		if err := static.PostTemplate.Execute(w, map[string]interface{}{
			"Post": p,
		}); err != nil {
			log.Error().Err(err).Msg("could not execute template")
		}
	}
}

func reloadHandler(blog *post.Blog) http.HandlerFunc {
	return func(w http.ResponseWriter, r *http.Request) {
		host, _, err := net.SplitHostPort(r.RemoteAddr)
		if err != nil {
			w.WriteHeader(http.StatusUnauthorized)
			return
		}

		ip := net.ParseIP(host)
		if ip == nil || !ip.IsLoopback() {
			w.WriteHeader(http.StatusUnauthorized)
			return
		}

		log.Info().Msg("reloading posts")
		if err := blog.Scan(); err != nil {
			http.Error(w, "could not re-scan", http.StatusInternalServerError)
			log.Error().Err(err).Msg("could not re-scan")
		}
	}
}