mirror of
https://git.sr.ht/~phw/scotty
synced 2025-06-01 19:38:34 +02:00
282 lines
8 KiB
Go
282 lines
8 KiB
Go
/*
|
|
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 (
|
|
"strconv"
|
|
"time"
|
|
|
|
"go.uploadedlobster.com/mbtypes"
|
|
"golang.org/x/exp/constraints"
|
|
)
|
|
|
|
type GetListensResult struct {
|
|
Payload GetListenPayload `json:"payload"`
|
|
}
|
|
|
|
type GetListenPayload struct {
|
|
Count int `json:"count"`
|
|
UserName string `json:"user_id"`
|
|
LatestListenTimestamp int64 `json:"latest_listen_ts"`
|
|
OldestListenTimestamp int64 `json:"oldest_listen_ts"`
|
|
Listens []Listen `json:"listens"`
|
|
}
|
|
|
|
type listenType string
|
|
|
|
const (
|
|
PlayingNow listenType = "playing_now"
|
|
Single listenType = "single"
|
|
Import listenType = "import"
|
|
)
|
|
|
|
type ListenSubmission struct {
|
|
ListenType listenType `json:"listen_type"`
|
|
Payload []Listen `json:"payload"`
|
|
}
|
|
|
|
type Listen struct {
|
|
InsertedAt float64 `json:"inserted_at,omitempty"`
|
|
ListenedAt int64 `json:"listened_at"`
|
|
RecordingMSID string `json:"recording_msid,omitempty"`
|
|
UserName string `json:"user_name,omitempty"`
|
|
TrackMetadata Track `json:"track_metadata"`
|
|
}
|
|
|
|
type Track struct {
|
|
TrackName string `json:"track_name,omitempty"`
|
|
ArtistName string `json:"artist_name,omitempty"`
|
|
ReleaseName string `json:"release_name,omitempty"`
|
|
RecordingMSID string `json:"recording_msid,omitempty"`
|
|
AdditionalInfo map[string]any `json:"additional_info,omitempty"`
|
|
MBIDMapping *MBIDMapping `json:"mbid_mapping,omitempty"`
|
|
}
|
|
|
|
type MBIDMapping struct {
|
|
ArtistMBIDs []mbtypes.MBID `json:"artist_mbids,omitempty"`
|
|
Artists []Artist `json:"artists,omitempty"`
|
|
RecordingMBID mbtypes.MBID `json:"recording_mbid,omitempty"`
|
|
RecordingName string `json:"recording_name,omitempty"`
|
|
ReleaseMBID mbtypes.MBID `json:"release_mbid,omitempty"`
|
|
CAAID int `json:"caa_id,omitempty"`
|
|
CAAReleaseMBID mbtypes.MBID `json:"caa_release_mbid,omitempty"`
|
|
}
|
|
|
|
type Artist struct {
|
|
ArtistCreditName string `json:"artist_credit_name,omitempty"`
|
|
ArtistMBID mbtypes.MBID `json:"artist_mbid,omitempty"`
|
|
JoinPhrase string `json:"join_phrase,omitempty"`
|
|
}
|
|
|
|
type GetFeedbackResult struct {
|
|
Count int `json:"count"`
|
|
TotalCount int `json:"total_count"`
|
|
Offset int `json:"offset"`
|
|
Feedback []Feedback `json:"feedback"`
|
|
}
|
|
|
|
type Feedback struct {
|
|
Created int64 `json:"created,omitempty"`
|
|
RecordingMBID mbtypes.MBID `json:"recording_mbid,omitempty"`
|
|
RecordingMSID mbtypes.MBID `json:"recording_msid,omitempty"`
|
|
Score int `json:"score,omitempty"`
|
|
TrackMetadata *Track `json:"track_metadata,omitempty"`
|
|
UserName string `json:"user_id,omitempty"`
|
|
}
|
|
|
|
type LookupResult struct {
|
|
ArtistCreditName string `json:"artist_credit_name"`
|
|
ReleaseName string `json:"release_name"`
|
|
RecordingName string `json:"recording_name"`
|
|
RecordingMBID mbtypes.MBID `json:"recording_mbid"`
|
|
ReleaseMBID mbtypes.MBID `json:"release_mbid"`
|
|
ArtistMBIDs []mbtypes.MBID `json:"artist_mbids"`
|
|
}
|
|
|
|
type RecordingMetadataRequest struct {
|
|
RecordingMBIDs []mbtypes.MBID `json:"recording_mbids"`
|
|
Includes string `json:"inc,omitempty"`
|
|
}
|
|
|
|
// Result for a recording metadata lookup
|
|
type RecordingMetadataResult map[mbtypes.MBID]RecordingMetadata
|
|
|
|
type RecordingMetadata struct {
|
|
Artist struct {
|
|
Name string `json:"name"`
|
|
ArtistCreditID int `json:"artist_credit_id"`
|
|
Artists []struct {
|
|
Name string `json:"name"`
|
|
Area string `json:"area"`
|
|
ArtistMBID mbtypes.MBID `json:"artist_mbid"`
|
|
JoinPhrase string `json:"join_phrase"`
|
|
BeginYear int `json:"begin_year"`
|
|
Type string `json:"type"`
|
|
// todo rels
|
|
} `json:"artists"`
|
|
} `json:"artist"`
|
|
Recording struct {
|
|
Name string `json:"name"`
|
|
Length int `json:"length"`
|
|
// TODO rels
|
|
} `json:"recording"`
|
|
Release struct {
|
|
Name string `json:"name"`
|
|
AlbumArtistName string `json:"album_artist_name"`
|
|
Year int `json:"year"`
|
|
MBID mbtypes.MBID `json:"mbid"`
|
|
ReleaseGroupMBID mbtypes.MBID `json:"release_group_mbid"`
|
|
CAAID int `json:"caa_id"`
|
|
CAAReleaseMBID mbtypes.MBID `json:"caa_release_mbid"`
|
|
} `json:"release"`
|
|
}
|
|
|
|
type StatusResult struct {
|
|
Status string `json:"status"`
|
|
}
|
|
|
|
type ErrorResult struct {
|
|
Code int `json:"code"`
|
|
Error string `json:"error"`
|
|
}
|
|
|
|
func (t Track) Duration() time.Duration {
|
|
info := t.AdditionalInfo
|
|
millisecondsF, ok := tryGetFloat[float64](info, "duration_ms")
|
|
if ok {
|
|
return time.Duration(int64(millisecondsF * float64(time.Millisecond)))
|
|
}
|
|
|
|
millisecondsI, ok := tryGetInteger[int64](info, "duration_ms")
|
|
if ok {
|
|
return time.Duration(millisecondsI * int64(time.Millisecond))
|
|
}
|
|
|
|
secondsF, ok := tryGetFloat[float64](info, "duration")
|
|
if ok {
|
|
return time.Duration(int64(secondsF * float64(time.Second)))
|
|
}
|
|
|
|
secondsI, ok := tryGetInteger[int64](info, "duration")
|
|
if ok {
|
|
return time.Duration(secondsI * int64(time.Second))
|
|
}
|
|
|
|
return time.Duration(0)
|
|
}
|
|
|
|
func (t Track) TrackNumber() int {
|
|
value, ok := tryGetInteger[int](t.AdditionalInfo, "tracknumber")
|
|
if ok {
|
|
return value
|
|
}
|
|
return 0
|
|
}
|
|
|
|
func (t Track) DiscNumber() int {
|
|
value, ok := tryGetInteger[int](t.AdditionalInfo, "discnumber")
|
|
if ok {
|
|
return value
|
|
}
|
|
return 0
|
|
}
|
|
|
|
func (t Track) ISRC() mbtypes.ISRC {
|
|
return mbtypes.ISRC(tryGetValueOrEmpty[string](t.AdditionalInfo, "isrc"))
|
|
}
|
|
|
|
func (t Track) RecordingMBID() mbtypes.MBID {
|
|
mbid := mbtypes.MBID(tryGetValueOrEmpty[string](t.AdditionalInfo, "recording_mbid"))
|
|
if mbid == "" && t.MBIDMapping != nil {
|
|
return t.MBIDMapping.RecordingMBID
|
|
} else {
|
|
return mbid
|
|
}
|
|
}
|
|
|
|
func (t Track) ReleaseMBID() mbtypes.MBID {
|
|
mbid := mbtypes.MBID(tryGetValueOrEmpty[string](t.AdditionalInfo, "release_mbid"))
|
|
if mbid == "" && t.MBIDMapping != nil {
|
|
return t.MBIDMapping.ReleaseMBID
|
|
} else {
|
|
return mbid
|
|
}
|
|
}
|
|
|
|
func (t Track) ReleaseGroupMBID() mbtypes.MBID {
|
|
return mbtypes.MBID(tryGetValueOrEmpty[string](t.AdditionalInfo, "release_group_mbid"))
|
|
}
|
|
|
|
func tryGetValueOrEmpty[T any](dict map[string]any, key string) T {
|
|
var result T
|
|
value, ok := dict[key].(T)
|
|
if ok {
|
|
result = value
|
|
}
|
|
return result
|
|
}
|
|
|
|
func tryGetFloat[T constraints.Float](dict map[string]any, key string) (T, bool) {
|
|
valueFloat64, ok := dict[key].(float64)
|
|
if ok {
|
|
return T(valueFloat64), ok
|
|
}
|
|
|
|
valueFloat32, ok := dict[key].(float32)
|
|
if ok {
|
|
return T(valueFloat32), ok
|
|
}
|
|
|
|
valueStr, ok := dict[key].(string)
|
|
if ok {
|
|
valueFloat64, err := strconv.ParseFloat(valueStr, 64)
|
|
if err == nil {
|
|
return T(valueFloat64), ok
|
|
}
|
|
}
|
|
|
|
return 0, false
|
|
}
|
|
|
|
func tryGetInteger[T constraints.Integer](dict map[string]any, key string) (T, bool) {
|
|
valueInt64, ok := dict[key].(int64)
|
|
if ok {
|
|
return T(valueInt64), ok
|
|
}
|
|
|
|
valueInt32, ok := dict[key].(int32)
|
|
if ok {
|
|
return T(valueInt32), ok
|
|
}
|
|
|
|
valueInt, ok := dict[key].(int)
|
|
if ok {
|
|
return T(valueInt), ok
|
|
}
|
|
|
|
valueFloat, ok := tryGetFloat[float64](dict, key)
|
|
if ok {
|
|
return T(valueFloat), ok
|
|
}
|
|
|
|
return 0, false
|
|
}
|