dwelling-radio/internal/handlers/handlers.go

package handlers

import (
	"dwelling-radio/internal/configuration"
	"dwelling-radio/internal/radio"
	"dwelling-radio/pkg/logging"
	"dwelling-radio/pkg/utils"
	"embed"
	"encoding/json"
	"fmt"
	"html/template"
	"io/fs"
	"net/http"

	"github.com/Joker/jade"
)

var compiledTemplates map[string]*template.Template

//go:embed web/assets
var assetsDir embed.FS

//go:embed web/templates
var templatesDir embed.FS

type NotFoundData struct {
	MainSite string
}

type IndexData struct {
	MainSite string
	Status   *radio.IcecastStatus
	Songs    []radio.Song
}

type RadioHandlers struct {
	conf   *configuration.Configuration
	logErr *logging.Logger
}

func NewRadioHandlers(conf *configuration.Configuration, lErr *logging.Logger) *RadioHandlers {
	compileTemplates(lErr)

	return &RadioHandlers{
		conf:   conf,
		logErr: lErr}
}

func (h *RadioHandlers) AssetsFS() http.FileSystem {
	f, _ := fs.Sub(assetsDir, "web/assets")
	return http.FS(f)
}

func (h *RadioHandlers) Index(w http.ResponseWriter, r *http.Request) {
	rad, err := radio.IcecastGetStatus(h.conf.Icecast.URL)
	if err != nil {
		h.logErr.Println("failed to get Icecast status:", err)
		rad = &radio.IcecastStatus{}
	}

	if err := compiledTemplates["index"].Execute(w, &IndexData{
		MainSite: utils.MainSite(r.Host),
		Status:   rad,
		Songs: radio.IcecastLastPlayedSongs(h.conf.ListLastNSongs,
			h.conf.Icecast.Playlist),
	}); err != nil {
		w.WriteHeader(http.StatusInternalServerError)
		h.logErr.Fatalln("failed to execute Index template:", err)
	}
}

func (h *RadioHandlers) Stats(w http.ResponseWriter, r *http.Request) {
	st, err := radio.IcecastGetStatus(h.conf.Icecast.URL)
	if err != nil {
		st = &radio.IcecastStatus{}
	}

	json.NewEncoder(w).Encode(st)
}

func (h *RadioHandlers) LastSong(w http.ResponseWriter, r *http.Request) {
	songs := radio.IcecastLastPlayedSongs(1, h.conf.Icecast.Playlist)
	json.NewEncoder(w).Encode(songs[0])
}

func (h *RadioHandlers) Playlist(w http.ResponseWriter, r *http.Request) {
	w.Header().Add("Content-Disposition", "attachment; filename=\"radio.arav.top.m3u\"")

	pf, _ := assetsDir.Open("radio.arav.top.m3u")
	defer pf.Close()

	fmt.Fprint(w, pf)
}

func compileTemplates(lErr *logging.Logger) {
	compiledTemplates = make(map[string]*template.Template)

	t, _ := fs.Sub(templatesDir, "web/templates")
	templatesFS := http.FS(t)

	indexStr, err := jade.ParseFileFromFileSystem("index.jade", templatesFS)
	if err != nil {
		lErr.Fatalln(err)
	}

	indexTpl, err := template.New("index").Parse(indexStr)
	if err != nil {
		lErr.Fatalln(err)
	}

	compiledTemplates["index"] = indexTpl
}