chore: initial public commit

engine/handlers/stats.go

@@ -0,0 +1,77 @@
+package handlers
+
+import (
+	"net/http"
+	"strings"
+
+	"github.com/gabehf/koito/internal/db"
+	"github.com/gabehf/koito/internal/logger"
+	"github.com/gabehf/koito/internal/utils"
+)
+
+type StatsResponse struct {
+	ListenCount   int64 `json:"listen_count"`
+	TrackCount    int64 `json:"track_count"`
+	AlbumCount    int64 `json:"album_count"`
+	ArtistCount   int64 `json:"artist_count"`
+	HoursListened int64 `json:"hours_listened"`
+}
+
+func StatsHandler(store db.DB) http.HandlerFunc {
+	return func(w http.ResponseWriter, r *http.Request) {
+		l := logger.FromContext(r.Context())
+		var period db.Period
+		switch strings.ToLower(r.URL.Query().Get("period")) {
+		case "day":
+			period = db.PeriodDay
+		case "week":
+			period = db.PeriodWeek
+		case "month":
+			period = db.PeriodMonth
+		case "year":
+			period = db.PeriodYear
+		case "all_time":
+			period = db.PeriodAllTime
+		default:
+			l.Debug().Msgf("Using default value '%s' for period", db.PeriodDay)
+			period = db.PeriodDay
+		}
+		listens, err := store.CountListens(r.Context(), period)
+		if err != nil {
+			l.Err(err).Send()
+			utils.WriteError(w, "failed to get listens: "+err.Error(), http.StatusInternalServerError)
+			return
+		}
+		tracks, err := store.CountTracks(r.Context(), period)
+		if err != nil {
+			l.Err(err).Send()
+			utils.WriteError(w, "failed to get listens: "+err.Error(), http.StatusInternalServerError)
+			return
+		}
+		albums, err := store.CountAlbums(r.Context(), period)
+		if err != nil {
+			l.Err(err).Send()
+			utils.WriteError(w, "failed to get listens: "+err.Error(), http.StatusInternalServerError)
+			return
+		}
+		artists, err := store.CountArtists(r.Context(), period)
+		if err != nil {
+			l.Err(err).Send()
+			utils.WriteError(w, "failed to get listens: "+err.Error(), http.StatusInternalServerError)
+			return
+		}
+		timeListenedS, err := store.CountTimeListened(r.Context(), period)
+		if err != nil {
+			l.Err(err).Send()
+			utils.WriteError(w, "failed to get listens: "+err.Error(), http.StatusInternalServerError)
+			return
+		}
+		utils.WriteJSON(w, http.StatusOK, StatsResponse{
+			ListenCount:   listens,
+			TrackCount:    tracks,
+			AlbumCount:    albums,
+			ArtistCount:   artists,
+			HoursListened: timeListenedS / 60 / 60,
+		})
+	}
+}