Moved general LB related code to separate package

pkg/listenbrainz/client.go

@@ -0,0 +1,160 @@
+/*
+Copyright © 2023-2025 Philipp Wolfer <phw@uploadedlobster.com>
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in
+all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
+THE SOFTWARE.
+*/
+package listenbrainz
+
+import (
+	"context"
+	"errors"
+	"strconv"
+	"time"
+
+	"github.com/go-resty/resty/v2"
+	"go.uploadedlobster.com/scotty/pkg/ratelimit"
+)
+
+const (
+	listenBrainzBaseURL  = "https://api.listenbrainz.org/1/"
+	DefaultItemsPerGet   = 25
+	MaxItemsPerGet       = 1000
+	MaxListensPerRequest = 1000
+)
+
+type Client struct {
+	HTTPClient *resty.Client
+	MaxResults int
+}
+
+func NewClient(token string, userAgent string) Client {
+	client := resty.New()
+	client.SetBaseURL(listenBrainzBaseURL)
+	client.SetAuthScheme("Token")
+	client.SetAuthToken(token)
+	client.SetHeader("Accept", "application/json")
+	client.SetHeader("User-Agent", userAgent)
+
+	// Handle rate limiting (see https://listenbrainz.readthedocs.io/en/latest/users/api/index.html#rate-limiting)
+	ratelimit.EnableHTTPHeaderRateLimit(client, "X-RateLimit-Reset-In")
+
+	return Client{
+		HTTPClient: client,
+		MaxResults: DefaultItemsPerGet,
+	}
+}
+
+func (c Client) GetListens(ctx context.Context, user string, maxTime time.Time, minTime time.Time) (result GetListensResult, err error) {
+	const path = "/user/{username}/listens"
+	errorResult := ErrorResult{}
+	response, err := c.HTTPClient.R().
+		SetContext(ctx).
+		SetPathParam("username", user).
+		SetQueryParams(map[string]string{
+			"max_ts": strconv.FormatInt(maxTime.Unix(), 10),
+			"min_ts": strconv.FormatInt(minTime.Unix(), 10),
+			"count":  strconv.Itoa(c.MaxResults),
+		}).
+		SetResult(&result).
+		SetError(&errorResult).
+		Get(path)
+
+	if !response.IsSuccess() {
+		err = errors.New(errorResult.Error)
+		return
+	}
+	return
+}
+
+func (c Client) SubmitListens(ctx context.Context, listens ListenSubmission) (result StatusResult, err error) {
+	const path = "/submit-listens"
+	errorResult := ErrorResult{}
+	response, err := c.HTTPClient.R().
+		SetContext(ctx).
+		SetBody(listens).
+		SetResult(&result).
+		SetError(&errorResult).
+		Post(path)
+
+	if !response.IsSuccess() {
+		err = errors.New(errorResult.Error)
+		return
+	}
+	return
+}
+
+func (c Client) GetFeedback(ctx context.Context, user string, status int, offset int) (result GetFeedbackResult, err error) {
+	const path = "/feedback/user/{username}/get-feedback"
+	errorResult := ErrorResult{}
+	response, err := c.HTTPClient.R().
+		SetContext(ctx).
+		SetPathParam("username", user).
+		SetQueryParams(map[string]string{
+			"status":   strconv.Itoa(status),
+			"offset":   strconv.Itoa(offset),
+			"count":    strconv.Itoa(c.MaxResults),
+			"metadata": "true",
+		}).
+		SetResult(&result).
+		SetError(&errorResult).
+		Get(path)
+
+	if !response.IsSuccess() {
+		err = errors.New(errorResult.Error)
+		return
+	}
+	return
+}
+
+func (c Client) SendFeedback(ctx context.Context, feedback Feedback) (result StatusResult, err error) {
+	const path = "/feedback/recording-feedback"
+	errorResult := ErrorResult{}
+	response, err := c.HTTPClient.R().
+		SetContext(ctx).
+		SetBody(feedback).
+		SetResult(&result).
+		SetError(&errorResult).
+		Post(path)
+
+	if !response.IsSuccess() {
+		err = errors.New(errorResult.Error)
+		return
+	}
+	return
+}
+
+func (c Client) Lookup(ctx context.Context, recordingName string, artistName string) (result LookupResult, err error) {
+	const path = "/metadata/lookup"
+	errorResult := ErrorResult{}
+	response, err := c.HTTPClient.R().
+		SetContext(ctx).
+		SetQueryParams(map[string]string{
+			"recording_name": recordingName,
+			"artist_name":    artistName,
+		}).
+		SetResult(&result).
+		SetError(&errorResult).
+		Get(path)
+
+	if !response.IsSuccess() {
+		err = errors.New(errorResult.Error)
+		return
+	}
+	return
+}