api

internal/db/db.go

@@ -30,6 +30,7 @@ type DB interface {
 	GetUserBySession(ctx context.Context, sessionId uuid.UUID) (*models.User, error)
 	GetUserByUsername(ctx context.Context, username string) (*models.User, error)
 	GetUserByApiKey(ctx context.Context, key string) (*models.User, error)
+	GetInterest(ctx context.Context, opts GetInterestOpts) ([]InterestBucket, error)
 
 	// Save
 

internal/db/opts.go

@@ -153,3 +153,10 @@ type GetExportPageOpts struct {
 	TrackID    int32
 	Limit      int32
 }
+
+type GetInterestOpts struct {
+	Buckets  int
+	AlbumID  int32
+	ArtistID int32
+	TrackID  int32
+}

internal/db/psql/interest.go

@@ -0,0 +1,70 @@
+package psql
+
+import (
+	"context"
+	"errors"
+	"fmt"
+
+	"github.com/gabehf/koito/internal/db"
+	"github.com/gabehf/koito/internal/repository"
+)
+
+func (d *Psql) GetInterest(ctx context.Context, opts db.GetInterestOpts) ([]db.InterestBucket, error) {
+	if opts.Buckets == 0 {
+		return nil, errors.New("GetInterest: bucket count must be provided")
+	}
+
+	ret := make([]db.InterestBucket, opts.Buckets)
+
+	if opts.ArtistID != 0 {
+		resp, err := d.q.GetGroupedListensFromArtist(ctx, repository.GetGroupedListensFromArtistParams{
+			ArtistID:    opts.ArtistID,
+			BucketCount: opts.Buckets,
+		})
+		if err != nil {
+			return nil, fmt.Errorf("GetInterest: GetGroupedListensFromArtist: %w", err)
+		}
+		for i, v := range resp {
+			ret[i] = db.InterestBucket{
+				BucketStart: v.BucketStart,
+				BucketEnd:   v.BucketEnd,
+				ListenCount: v.ListenCount,
+			}
+		}
+		return ret, nil
+	} else if opts.AlbumID != 0 {
+		resp, err := d.q.GetGroupedListensFromRelease(ctx, repository.GetGroupedListensFromReleaseParams{
+			ReleaseID:   opts.AlbumID,
+			BucketCount: opts.Buckets,
+		})
+		if err != nil {
+			return nil, fmt.Errorf("GetInterest: GetGroupedListensFromRelease: %w", err)
+		}
+		for i, v := range resp {
+			ret[i] = db.InterestBucket{
+				BucketStart: v.BucketStart,
+				BucketEnd:   v.BucketEnd,
+				ListenCount: v.ListenCount,
+			}
+		}
+		return ret, nil
+	} else if opts.TrackID != 0 {
+		resp, err := d.q.GetGroupedListensFromTrack(ctx, repository.GetGroupedListensFromTrackParams{
+			ID:          opts.TrackID,
+			BucketCount: opts.Buckets,
+		})
+		if err != nil {
+			return nil, fmt.Errorf("GetInterest: GetGroupedListensFromTrack: %w", err)
+		}
+		for i, v := range resp {
+			ret[i] = db.InterestBucket{
+				BucketStart: v.BucketStart,
+				BucketEnd:   v.BucketEnd,
+				ListenCount: v.ListenCount,
+			}
+		}
+		return ret, nil
+	} else {
+		return nil, errors.New("GetInterest: artist id, album id, or track id must be provided")
+	}
+}

internal/db/types.go

@@ -44,3 +44,9 @@ type ExportItem struct {
 	ReleaseAliases     []models.Alias
 	Artists            []models.ArtistWithFullAliases
 }
+
+type InterestBucket struct {
+	BucketStart time.Time `json:"bucket_start"`
+	BucketEnd   time.Time `json:"bucket_end"`
+	ListenCount int64     `json:"listen_count"`
+}