mirror of
https://git.sr.ht/~phw/scotty
synced 2025-04-16 10:09:28 +02:00
98 lines
3 KiB
Go
98 lines
3 KiB
Go
/*
|
|
Copyright © 2023 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 (
|
|
"slices"
|
|
"time"
|
|
|
|
"github.com/spf13/viper"
|
|
"go.uploadedlobster.com/scotty/models"
|
|
)
|
|
|
|
type ListenBrainzApiBackend struct {
|
|
client Client
|
|
username string
|
|
}
|
|
|
|
func (b ListenBrainzApiBackend) FromConfig(config *viper.Viper) models.Backend {
|
|
b.client = NewClient(config.GetString("token"))
|
|
b.client.MaxResults = MaxItemsPerGet
|
|
b.username = config.GetString("username")
|
|
return b
|
|
}
|
|
|
|
func (b ListenBrainzApiBackend) ExportListens(oldestTimestamp time.Time) ([]models.Listen, error) {
|
|
maxTime := time.Now()
|
|
minTime := time.Unix(0, 0)
|
|
listens := make([]models.Listen, 0)
|
|
|
|
out:
|
|
for {
|
|
result, err := b.client.GetListens(b.username, maxTime, minTime)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
count := len(result.Payload.Listens)
|
|
if count == 0 {
|
|
break
|
|
}
|
|
|
|
// Set maxTime to the oldest returned listen
|
|
maxTime = time.Unix(result.Payload.Listens[count-1].ListenedAt, 0)
|
|
|
|
for _, listen := range result.Payload.Listens {
|
|
if listen.ListenedAt > oldestTimestamp.Unix() {
|
|
listens = append(listens, ListenFromListenBrainz(listen))
|
|
} else {
|
|
// result contains listens older then oldestTimestamp,
|
|
// we can stop requesting more
|
|
break out
|
|
}
|
|
}
|
|
}
|
|
|
|
slices.Reverse(listens)
|
|
return listens, nil
|
|
}
|
|
|
|
func ListenFromListenBrainz(lbListen Listen) models.Listen {
|
|
track := lbListen.TrackMetadata
|
|
listen := models.Listen{
|
|
ListenedAt: time.Unix(lbListen.ListenedAt, 0),
|
|
UserName: lbListen.UserName,
|
|
Track: models.Track{
|
|
TrackName: track.TrackName,
|
|
ReleaseName: track.ReleaseName,
|
|
ArtistNames: []string{track.ArtistName},
|
|
Duration: track.Duration(),
|
|
TrackNumber: track.TrackNumber(),
|
|
RecordingMbid: models.MBID(track.RecordingMbid()),
|
|
ReleaseMbid: models.MBID(track.ReleaseMbid()),
|
|
ReleaseGroupMbid: models.MBID(track.ReleaseGroupMbid()),
|
|
Isrc: track.Isrc(),
|
|
AdditionalInfo: track.AdditionalInfo,
|
|
},
|
|
}
|
|
return listen
|
|
}
|