stillbox/pkg/rest/incidents.go

package rest

import (
	"bytes"
	"encoding/json"
	"fmt"
	"net/http"

	"dynatron.me/x/stillbox/internal/forms"
	"dynatron.me/x/stillbox/pkg/database"
	"dynatron.me/x/stillbox/pkg/incidents"
	"dynatron.me/x/stillbox/pkg/incidents/incstore"
	"dynatron.me/x/stillbox/pkg/talkgroups/tgstore"

	"github.com/go-chi/chi/v5"
	"github.com/google/uuid"
)

type incidentsAPI struct {
}

func (ia *incidentsAPI) Subrouter() http.Handler {
	r := chi.NewMux()

	r.Get(`/{id:[a-f0-9-]+}`, ia.getIncident)
	r.Get(`/{id:[a-f0-9-]+}.m3u`, ia.getCallsM3U)

	r.Post(`/create`, ia.createIncident)
	r.Post(`/`, ia.listIncidents)
	r.Post(`/{id:[a-f0-9-]+}/calls`, ia.postCalls)

	r.Put(`/{id:[a-f0-9]+}`, ia.updateIncident)

	r.Delete(`/{id:[a-f0-9]+}`, ia.deleteIncident)

	return r
}

func (ia *incidentsAPI) listIncidents(w http.ResponseWriter, r *http.Request) {
	ctx := r.Context()
	incs := incstore.FromCtx(ctx)

	p := incstore.IncidentsParams{}
	err := forms.Unmarshal(r, &p, forms.WithTag("json"), forms.WithAcceptBlank(), forms.WithOmitEmpty())
	if err != nil {
		wErr(w, r, badRequest(err))
		return
	}

	res := struct {
		Incidents []database.Incident `json:"incidents"`
		Count     int                 `json:"count"`
	}{}

	res.Incidents, res.Count, err = incs.Incidents(ctx, p)
	if err != nil {
		wErr(w, r, autoError(err))
		return
	}

	respond(w, r, res)
}

func (ia *incidentsAPI) createIncident(w http.ResponseWriter, r *http.Request) {
	ctx := r.Context()
	incs := incstore.FromCtx(ctx)

	p := incidents.Incident{}
	err := forms.Unmarshal(r, &p, forms.WithTag("json"), forms.WithAcceptBlank(), forms.WithOmitEmpty())
	if err != nil {
		wErr(w, r, badRequest(err))
		return
	}

	inc, err := incs.CreateIncident(ctx, p)
	if err != nil {
		wErr(w, r, autoError(err))
		return
	}

	respond(w, r, inc)
}

func (ia *incidentsAPI) getIncident(w http.ResponseWriter, r *http.Request) {
	ctx := r.Context()
	incs := incstore.FromCtx(ctx)

	id, err := idOnlyParam(w, r)
	if err != nil {
		return
	}

	inc, err := incs.Incident(ctx, id)
	if err != nil {
		wErr(w, r, autoError(err))
		return
	}

	respond(w, r, inc)
}

func (ia *incidentsAPI) updateIncident(w http.ResponseWriter, r *http.Request) {
	ctx := r.Context()
	incs := incstore.FromCtx(ctx)

	id, err := idOnlyParam(w, r)
	if err != nil {
		return
	}

	p := incstore.UpdateIncidentParams{}
	err = forms.Unmarshal(r, &p, forms.WithTag("json"), forms.WithAcceptBlank(), forms.WithOmitEmpty())
	if err != nil {
		wErr(w, r, badRequest(err))
		return
	}

	inc, err := incs.UpdateIncident(ctx, id, p)
	if err != nil {
		wErr(w, r, autoError(err))
		return
	}

	respond(w, r, inc)
}

func (ia *incidentsAPI) deleteIncident(w http.ResponseWriter, r *http.Request) {
	ctx := r.Context()
	incs := incstore.FromCtx(ctx)

	urlParams := struct {
		ID uuid.UUID `param:"id"`
	}{}

	err := decodeParams(&urlParams, r)
	if err != nil {
		wErr(w, r, badRequest(err))
		return
	}

	err = incs.DeleteIncident(ctx, urlParams.ID)
	if err != nil {
		wErr(w, r, autoError(err))
		return
	}

	w.WriteHeader(http.StatusNoContent)
}

type CallIncidentParams struct {
	Add   []uuid.UUID     `json:"add"`
	Notes json.RawMessage `json:"notes"`

	Remove []uuid.UUID `json:"remove"`
}

func (ia *incidentsAPI) postCalls(w http.ResponseWriter, r *http.Request) {
	ctx := r.Context()
	incs := incstore.FromCtx(ctx)

	id, err := idOnlyParam(w, r)
	if err != nil {
		return
	}

	p := CallIncidentParams{}
	err = forms.Unmarshal(r, &p, forms.WithTag("json"), forms.WithAcceptBlank(), forms.WithOmitEmpty())
	if err != nil {
		wErr(w, r, badRequest(err))
		return
	}

	err = incs.AddRemoveIncidentCalls(ctx, id, p.Add, p.Notes, p.Remove)
	if err != nil {
		wErr(w, r, autoError(err))
		return
	}

	w.WriteHeader(http.StatusNoContent)
}

func (ia *incidentsAPI) getCallsM3U(w http.ResponseWriter, r *http.Request) {
	ctx := r.Context()
	incs := incstore.FromCtx(ctx)
	tgst := tgstore.FromCtx(ctx)

	id, err := idOnlyParam(w, r)
	if err != nil {
		return
	}

	inc, err := incs.Incident(ctx, id)
	if err != nil {
		wErr(w, r, autoError(err))
		return
	}

	var b bytes.Buffer

	callUrl := r.URL
	callUrl.RawQuery = ""
	callUrl.Fragment = ""

	b.WriteString("#EXTM3U\n\n")
	for _, c := range inc.Calls {
		tg, err := tgst.TG(ctx, c.TalkgroupTuple())
		if err != nil {
			wErr(w, r, autoError(err))
			return
		}
		var from string
		if c.Source != 0 {
			from = fmt.Sprintf(" from %d", c.Source)
		}

		callUrl.Path = "/api/call/%s" + c.ID.String()

		fmt.Fprintf(w, "#EXTINF:%d,%s%s (%s)\n%s\n\n",
			c.Duration.Seconds(),
			tg.StringTag(true),
			from,
			c.DateTime.Format("15:04 01/02"),
			callUrl,
		)
	}

	// Not a lot of agreement on which MIME type to use for non-HLS m3u,
	// let's hope this is good enough
	w.Header().Set("Content-Type", "audio/x-mpegurl")
	w.WriteHeader(http.StatusOK)
	_, _ = b.WriteTo(w)
}