main.go

package main

import (
	"embed"
	"fmt"
	"html/template"
	"io/fs"
	"net/http"
	"os/exec"

	"github.com/rs/zerolog/log"
)

//go:embed static
var staticFiles embed.FS

//go:embed templates/index.html templates/success.html
var templatesFS embed.FS

func main() {
	http.HandleFunc("/", serveHTML)
	http.HandleFunc("/success", serveSuccessHTML)
	http.HandleFunc("/api/reboot", rebootRaspberryPi)

	staticFS, err := fs.Sub(staticFiles, "static")
	if err != nil {
		panic(err)
	}
	http.Handle("/static/", http.StripPrefix("/static", http.FileServer(http.FS(staticFS))))

	log.Info().Msg("Starting server on port 3001...")
	if err := http.ListenAndServe(":3001", nil); err != nil {
		panic(err)
	}
}

func serveHTML(w http.ResponseWriter, r *http.Request) {
	tmpl, err := template.ParseFS(templatesFS, "templates/index.html")
	if err != nil {
		log.Error().Err(err).Msg("Failed to load index template")
		http.Error(w, "Failed to load template", http.StatusInternalServerError)
		return
	}

	err = tmpl.Execute(w, nil)
	if err != nil {
		log.Error().Err(err).Msg("Failed to render index template")
		http.Error(w, "Failed to render template", http.StatusInternalServerError)
		return
	}
}

func serveSuccessHTML(w http.ResponseWriter, r *http.Request) {
	tmpl, err := template.ParseFS(templatesFS, "templates/success.html")
	if err != nil {
		log.Error().Err(err).Msg("Failed to load success template")
		http.Error(w, "Failed to load template", http.StatusInternalServerError)
		return
	}

	err = tmpl.Execute(w, nil)
	if err != nil {
		log.Error().Err(err).Msg("Failed to render success template")
		http.Error(w, "Failed to render template", http.StatusInternalServerError)
		return
	}
}

func rebootRaspberryPi(w http.ResponseWriter, r *http.Request) {
	if r.Method != "POST" {
		http.Error(w, "Method not allowed", http.StatusMethodNotAllowed)
		return
	}

	cmd := exec.Command("sudo", "shutdown", "-r")
	if err := cmd.Run(); err != nil {
		log.Error().Err(err).Msg("Failed to reboot Pi")
		http.Error(w, "Failed to reboot Pi", http.StatusInternalServerError)
		return
	}

	fmt.Fprint(w, "Rebooting...")
}