/*
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
}