p/main.go

package main

import (
	"fmt"
	"html/template"
	"log"
	"net/http"
	"os"
	"strings"
)

type PHandler struct {
	tmpl         *template.Template
	pd           *pdClient
	gc           *garminClient
	garminDeviceIDs []string
	garminSender    string
	mux          *http.ServeMux
}

func NewPHandler(pdRoutingKey, garminMapshareID string, garminDeviceIDs []string, garminSender string) (*PHandler, error) {
	tmpl := template.New("index.html")

	tmpl.Funcs(template.FuncMap{
		"replaceAll": func(o, n, s string) string { return strings.ReplaceAll(s, o, n) },
	})

	tmpl, err := tmpl.ParseFiles("static/index.html")
	if err != nil {
		return nil, fmt.Errorf("static/index.html: %w", err)
	}

	ph := &PHandler{
		tmpl:         tmpl,
		pd:           newPDClient(pdRoutingKey),
		garminDeviceIDs: garminDeviceIDs,
		garminSender: garminSender,
		mux:          http.NewServeMux(),
	}

	if garminMapshareID != "" {
		ph.gc = newGarminClient(garminMapshareID)
	}

	ph.mux.HandleFunc("/{$}", ph.serveRoot)

	return ph, nil
}

func (ph *PHandler) ServeHTTP(w http.ResponseWriter, r *http.Request) {
	ph.mux.ServeHTTP(w, r)
}

func (ph *PHandler) serveRoot(w http.ResponseWriter, r *http.Request) {
	err := r.ParseForm()
	if err != nil {
		sendError(w, http.StatusBadRequest, "Parse form: %s", err)
		return
	}

	log.Printf("%s %s %s", r.RemoteAddr, r.URL.Path, r.Form.Encode())

	m := r.Form.Get("m")
	if m == "" {
		err = ph.tmpl.Execute(w, ph.envs())
		if err != nil {
			sendError(w, http.StatusInternalServerError, "Execute template %s: %s", ph.tmpl.Name(), err)
			return
		}

		return
	}

	err = ph.sendAlert(m)
	if err != nil {
		sendError(w, http.StatusInternalServerError, "%s", err)
		return
	}
	sendResponse(w, "Page sent")
}

func (ph *PHandler) sendAlert(m string) error {
	res := make(chan error, 2)

	target := 0

	if ph.gc != nil {
		go func() {
			err := ph.gc.sendMessage(ph.garminDeviceIDs, ph.garminSender, m)
			if err != nil {
				res <- fmt.Errorf("Error sending to Garmin: %w", err)
			} else {
				res <- nil
			}
		}()

		target += 1
	}

	{
		go func() {
			err := ph.pd.sendAlert(m)
			if err != nil {
				res <- fmt.Errorf("Error sending to PagerDuty: %w", err)
			} else {
				res <- nil
			}
		}()

		target += 1
	}

	for i := 0; i < target; i++ {
		err := <-res
		if err != nil {
			return err
		}
	}

	return nil
}

var allowedEnvs = []string{
	"SHORT_NAME",
	"CONTACT_NAME",
	"CONTACT_PHONE",
	"CONTACT_SMS",
	"CONTACT_IMESSAGE",
	"CONTACT_WHATSAPP",
	"CONTACT_PAGE_EMAIL",
	"GARMIN_MAPSHARE_URL",
}

func (ph *PHandler) envs() map[string]string {
	envs := map[string]string{}

	for _, k := range allowedEnvs {
		v := os.Getenv(k)
		if v != "" {
			envs[k] = v
		}
	}

	return envs
}

func main() {
	pdRoutingKey := os.Getenv("PD_ROUTING_KEY")
	if pdRoutingKey == "" {
		log.Fatalf("please set PD_ROUTING_KEY")
	}

	garminMapshareID := os.Getenv("GARMIN_MAPSHARE_ID")

	garminDeviceIDStr := os.Getenv("GARMIN_DEVICE_IDS")
	if garminMapshareID != "" && garminDeviceIDStr == "" {
		log.Fatalf("please set GARMIN_DEVICE_IDS")
	}
	var garminDeviceIDs []string
	if garminDeviceIDStr != "" {
		for _, id := range strings.Split(garminDeviceIDStr, ",") {
			id = strings.TrimSpace(id)
			if id != "" {
				garminDeviceIDs = append(garminDeviceIDs, id)
			}
		}
	}

	garminSender := os.Getenv("GARMIN_SENDER")
	if garminMapshareID != "" && garminSender == "" {
		log.Fatalf("please set GARMIN_SENDER")
	}

	ph, err := NewPHandler(pdRoutingKey, garminMapshareID, garminDeviceIDs, garminSender)
	if err != nil {
		log.Fatalf("NewPHandler: %s", err)
	}

	http.Handle("/", ph)

	port := os.Getenv("PORT")
	if port == "" {
		port = "80"
	}

	bind := fmt.Sprintf(":%s", port)
	log.Printf("listening on %s", bind)

	if err := http.ListenAndServe(bind, nil); err != nil {
		log.Fatalf("listen: %s", err)
	}
}