package client

import (
	"context"
	"fmt"

	"gitea.mrixs.me/Mrixs/yandex-music-api-go/internal/models"
)

// LikeTrack likes a track
func (c *Client) LikeTrack(ctx context.Context, trackID string) error {
	params := map[string]string{
		"track_id": trackID,
	}
	var result map[string]any
	return c.request.Post(ctx, "/users/likes/tracks/add", params, &result)
}

// UnlikeTrack unlikes a track
func (c *Client) UnlikeTrack(ctx context.Context, trackID string) error {
	params := map[string]string{
		"track_id": trackID,
	}
	var result map[string]any
	return c.request.Post(ctx, "/users/likes/tracks/remove", params, &result)
}

// LikeAlbum likes an album
func (c *Client) LikeAlbum(ctx context.Context, albumID string) error {
	params := map[string]string{
		"album_id": albumID,
	}
	var result map[string]any
	return c.request.Post(ctx, "/users/likes/albums/add", params, &result)
}

// UnlikeAlbum unlikes an album
func (c *Client) UnlikeAlbum(ctx context.Context, albumID string) error {
	params := map[string]string{
		"album_id": albumID,
	}
	var result map[string]any
	return c.request.Post(ctx, "/users/likes/albums/remove", params, &result)
}

// LikeArtist likes an artist
func (c *Client) LikeArtist(ctx context.Context, artistID string) error {
	params := map[string]string{
		"artist_id": artistID,
	}
	var result map[string]any
	return c.request.Post(ctx, "/users/likes/artists/add", params, &result)
}

// UnlikeArtist unlikes an artist
func (c *Client) UnlikeArtist(ctx context.Context, artistID string) error {
	params := map[string]string{
		"artist_id": artistID,
	}
	var result map[string]any
	return c.request.Post(ctx, "/users/likes/artists/remove", params, &result)
}

// LikePlaylist likes a playlist
func (c *Client) LikePlaylist(ctx context.Context, kind int) error {
	params := map[string]string{
		"kind": fmt.Sprintf("%d", kind),
	}
	var result map[string]any
	return c.request.Post(ctx, "/users/likes/playlists/add", params, &result)
}

// UnlikePlaylist unlikes a playlist
func (c *Client) UnlikePlaylist(ctx context.Context, kind int) error {
	params := map[string]string{
		"kind": fmt.Sprintf("%d", kind),
	}
	var result map[string]any
	return c.request.Post(ctx, "/users/likes/playlists/remove", params, &result)
}

// GetLikedTracks gets user's liked tracks
func (c *Client) GetLikedTracks(ctx context.Context) ([]*models.Track, error) {
	var result []models.Track
	if err := c.request.Get(ctx, "/users/likes/tracks", nil, &result); err != nil {
		return nil, err
	}

	tracks := make([]*models.Track, len(result))
	for i := range result {
		result[i].SetClient(c)
		tracks[i] = &result[i]
	}
	return tracks, nil
}

// GetLikedAlbums gets user's liked albums
func (c *Client) GetLikedAlbums(ctx context.Context) ([]*models.Album, error) {
	var result []models.Album
	if err := c.request.Get(ctx, "/users/likes/albums", nil, &result); err != nil {
		return nil, err
	}

	albums := make([]*models.Album, len(result))
	for i := range result {
		result[i].SetClient(c)
		albums[i] = &result[i]
	}
	return albums, nil
}

// GetLikedArtists gets user's liked artists
func (c *Client) GetLikedArtists(ctx context.Context) ([]*models.Artist, error) {
	var result []models.Artist
	if err := c.request.Get(ctx, "/users/likes/artists", nil, &result); err != nil {
		return nil, err
	}

	artists := make([]*models.Artist, len(result))
	for i := range result {
		result[i].SetClient(c)
		artists[i] = &result[i]
	}
	return artists, nil
}

// GetLikedPlaylists gets user's liked playlists
func (c *Client) GetLikedPlaylists(ctx context.Context) ([]*models.Playlist, error) {
	var result []models.Playlist
	if err := c.request.Get(ctx, "/users/likes/playlists", nil, &result); err != nil {
		return nil, err
	}

	playlists := make([]*models.Playlist, len(result))
	for i := range result {
		result[i].SetClient(c)
		playlists[i] = &result[i]
	}
	return playlists, nil
}