import configparser
import json
import time
import unittest
import warnings
from pathlib import Path
from unittest import mock

from requests import Response

from ytmusicapi.setup import main, setup  # noqa: E402
from ytmusicapi.ytmusic import YTMusic  # noqa: E402


def get_resource(file: str) -> str:
    data_dir = Path(__file__).parent
    return data_dir.joinpath(file).as_posix()


config = configparser.RawConfigParser()
config.read(get_resource("test.cfg"), "utf-8")

sample_album = "MPREb_4pL8gzRtw1p"  # Eminem - Revival
sample_video = "hpSrLjc5SMs"  # Oasis - Wonderwall
sample_playlist = "PL6bPxvf5dW5clc3y9wAoslzqUrmkZ5c-u"  # very large playlist

headers_oauth = get_resource(config["auth"]["headers_oauth"])
headers_browser = get_resource(config["auth"]["headers_file"])


class TestYTMusic(unittest.TestCase):

    @classmethod
    def setUpClass(cls):
        warnings.filterwarnings(action="ignore", message="unclosed", category=ResourceWarning)
        with YTMusic(requests_session=False) as yt:
            assert isinstance(yt, YTMusic)
        cls.yt = YTMusic()
        cls.yt_oauth = YTMusic(headers_oauth)
        cls.yt_auth = YTMusic(headers_browser, location="GB")
        cls.yt_brand = YTMusic(config["auth"]["headers"], config["auth"]["brand_account"])
        cls.yt_empty = YTMusic(config["auth"]["headers_empty"],
                               config["auth"]["brand_account_empty"])

    @mock.patch("sys.argv", ["ytmusicapi", "browser", "--file", headers_browser])
    def test_setup_browser(self):
        headers = setup(headers_browser, config["auth"]["headers_raw"])
        self.assertGreaterEqual(len(headers), 2)
        headers_raw = config["auth"]["headers_raw"].split("\n")
        with mock.patch("builtins.input", side_effect=(headers_raw + [EOFError()])):
            headers = main()
            self.assertGreaterEqual(len(headers), 2)

    @mock.patch("requests.Response.json")
    @mock.patch("requests.Session.post")
    @mock.patch("sys.argv", ["ytmusicapi", "oauth", "--file", headers_oauth])
    def test_setup_oauth(self, session_mock, json_mock):
        session_mock.return_value = Response()
        json_mock.side_effect = [
            json.loads(config["auth"]["oauth_code"]),
            json.loads(config["auth"]["oauth_token"]),
        ]
        with mock.patch("builtins.input", return_value="y"):
            main()
            self.assertTrue(Path(headers_oauth).exists())

        json_mock.side_effect = None
        with open(headers_oauth, mode="r", encoding="utf8") as headers:
            string_headers = headers.read()
            self.yt_oauth = YTMusic(string_headers)

    ###############
    # BROWSING
    ###############

    def test_get_home(self):
        result = self.yt.get_home(limit=6)
        self.assertGreaterEqual(len(result), 6)
        result = self.yt_auth.get_home(limit=15)
        self.assertGreaterEqual(len(result), 15)

    def test_search(self):
        query = "edm playlist"
        self.assertRaises(Exception, self.yt_auth.search, query, filter="song")
        self.assertRaises(Exception, self.yt_auth.search, query, scope="upload")
        queries = ["Maylssi", "qllwlwl", "heun"]
        for q in queries:
            with self.subTest():
                results = self.yt_brand.search(q)
                self.assertListEqual(["resultType" in r for r in results], [True] * len(results))
                self.assertGreater(len(results), 10)
                results = self.yt.search(q)
                self.assertGreater(len(results), 10)
        results = self.yt_auth.search("Martin Stig Andersen - Deteriation", ignore_spelling=True)
        self.assertGreater(len(results), 0)

    def test_search_filters(self):
        query = "hip hop playlist"
        results = self.yt_auth.search(query, filter="songs")
        self.assertGreater(len(results), 10)
        self.assertTrue(all(item['resultType'] == 'song' for item in results))
        results = self.yt_auth.search(query, filter="videos")
        self.assertGreater(len(results), 10)
        self.assertTrue(all(item['resultType'] == 'video' for item in results))
        results = self.yt_auth.search(query, filter="albums", limit=40)
        self.assertGreater(len(results), 20)
        self.assertTrue(all(item['resultType'] == 'album' for item in results))
        results = self.yt_auth.search("project-2", filter="artists", ignore_spelling=True)
        self.assertGreater(len(results), 10)
        self.assertTrue(all(item['resultType'] == 'artist' for item in results))
        results = self.yt_auth.search("classical music", filter="playlists")
        self.assertGreater(len(results), 10)
        self.assertTrue(all(item['resultType'] == 'playlist' for item in results))
        results = self.yt_auth.search("clasical music", filter="playlists", ignore_spelling=True)
        self.assertGreater(len(results), 10)
        results = self.yt_auth.search("clasic rock",
                                      filter="community_playlists",
                                      ignore_spelling=True)
        self.assertGreater(len(results), 10)
        self.assertTrue(all(item['resultType'] == 'playlist' for item in results))
        results = self.yt_auth.search("hip hop", filter="featured_playlists")
        self.assertGreater(len(results), 10)
        self.assertTrue(all(item['resultType'] == 'playlist' for item in results))
        results = self.yt_auth.search("some user", filter="profiles")
        self.assertGreater(len(results), 10)
        self.assertTrue(all(item['resultType'] == 'profile' for item in results))

    def test_search_uploads(self):
        self.assertRaises(
            Exception,
            self.yt.search,
            "audiomachine",
            filter="songs",
            scope="uploads",
            limit=40,
        )
        results = self.yt_auth.search("audiomachine", scope="uploads", limit=40)
        self.assertGreater(len(results), 20)

    def test_search_library(self):
        results = self.yt_oauth.search("garrix", scope="library")
        self.assertGreater(len(results), 5)
        results = self.yt_auth.search("bergersen", filter="songs", scope="library", limit=40)
        self.assertGreater(len(results), 10)
        results = self.yt_auth.search("garrix", filter="albums", scope="library", limit=40)
        self.assertGreaterEqual(len(results), 4)
        results = self.yt_auth.search("garrix", filter="artists", scope="library", limit=40)
        self.assertGreaterEqual(len(results), 1)
        results = self.yt_auth.search("garrix", filter="playlists", scope="library")
        self.assertGreaterEqual(len(results), 1)
        self.assertRaises(Exception,
                          self.yt_auth.search,
                          "beatles",
                          filter="community_playlists",
                          scope="library",
                          limit=40)
        self.assertRaises(Exception,
                          self.yt_auth.search,
                          "beatles",
                          filter="featured_playlists",
                          scope="library",
                          limit=40)

    def test_get_artist(self):
        results = self.yt.get_artist("MPLAUCmMUZbaYdNH0bEd1PAlAqsA")
        self.assertEqual(len(results), 14)

        # test correctness of related artists
        related = results["related"]["results"]
        self.assertEqual(
            len([
                x for x in related
                if set(x.keys()) == {"browseId", "subscribers", "title", "thumbnails"}
            ]),
            len(related),
        )

        results = self.yt.get_artist("UCLZ7tlKC06ResyDmEStSrOw")  # no album year
        self.assertGreaterEqual(len(results), 11)

    def test_get_artist_albums(self):
        artist = self.yt.get_artist("UCj5ZiBBqpe0Tg4zfKGHEFuQ")
        results = self.yt.get_artist_albums(artist["albums"]["browseId"],
                                            artist["albums"]["params"])
        self.assertGreater(len(results), 0)

    def test_get_artist_singles(self):
        artist = self.yt.get_artist("UCAeLFBCQS7FvI8PvBrWvSBg")
        results = self.yt.get_artist_albums(artist["singles"]["browseId"],
                                            artist["singles"]["params"])
        self.assertGreater(len(results), 0)

    def test_get_user(self):
        results = self.yt.get_user("UC44hbeRoCZVVMVg5z0FfIww")
        self.assertEqual(len(results), 3)

    def test_get_user_playlists(self):
        results = self.yt.get_user("UCPVhZsC2od1xjGhgEc2NEPQ")
        results = self.yt.get_user_playlists("UCPVhZsC2od1xjGhgEc2NEPQ",
                                             results["playlists"]["params"])
        self.assertGreater(len(results), 100)

    def test_get_album_browse_id(self):
        browse_id = self.yt.get_album_browse_id("OLAK5uy_nMr9h2VlS-2PULNz3M3XVXQj_P3C2bqaY")
        self.assertEqual(browse_id, sample_album)

    def test_get_album(self):
        results = self.yt_auth.get_album(sample_album)
        self.assertGreaterEqual(len(results), 9)
        self.assertTrue(results["tracks"][0]["isExplicit"])
        self.assertIn("feedbackTokens", results["tracks"][0])
        self.assertEqual(len(results["other_versions"]), 2)
        results = self.yt.get_album("MPREb_BQZvl3BFGay")
        self.assertEqual(len(results["tracks"]), 7)

    def test_get_song(self):
        song = self.yt_oauth.get_song(config["uploads"]["private_upload_id"])  # private upload
        self.assertEqual(len(song), 5)
        song = self.yt.get_song(sample_video)
        self.assertGreaterEqual(len(song["streamingData"]["adaptiveFormats"]), 10)

    def test_get_song_related_content(self):
        song = self.yt_oauth.get_watch_playlist(sample_video)
        song = self.yt_oauth.get_song_related(song["related"])
        self.assertGreaterEqual(len(song), 5)

    def test_get_lyrics(self):
        playlist = self.yt.get_watch_playlist(sample_video)
        lyrics_song = self.yt.get_lyrics(playlist["lyrics"])
        self.assertIsNotNone(lyrics_song["lyrics"])
        self.assertIsNotNone(lyrics_song["source"])

        playlist = self.yt.get_watch_playlist(config["uploads"]["private_upload_id"])
        self.assertIsNone(playlist["lyrics"])
        self.assertRaises(Exception, self.yt.get_lyrics, playlist["lyrics"])

    def test_get_signatureTimestamp(self):
        signatureTimestamp = self.yt.get_signatureTimestamp()
        self.assertIsNotNone(signatureTimestamp)

    def test_set_tasteprofile(self):
        self.assertRaises(Exception, self.yt.set_tasteprofile, "not an artist")
        taste_profile = self.yt.get_tasteprofile()
        self.assertIsNone(self.yt.set_tasteprofile(list(taste_profile)[:5], taste_profile))

        self.assertRaises(Exception, self.yt_brand.set_tasteprofile, ["test", "test2"])
        taste_profile = self.yt_brand.get_tasteprofile()
        self.assertIsNone(self.yt_brand.set_tasteprofile(list(taste_profile)[:1], taste_profile))

    def test_get_tasteprofile(self):
        result = self.yt.get_tasteprofile()
        self.assertGreaterEqual(len(result), 0)

        result = self.yt_oauth.get_tasteprofile()
        self.assertGreaterEqual(len(result), 0)

    def test_get_search_suggestions(self):
        result = self.yt.get_search_suggestions("fade")
        self.assertGreaterEqual(len(result), 0)

        result = self.yt.get_search_suggestions("fade", detailed_runs=True)
        self.assertGreaterEqual(len(result), 0)

    ################
    # EXPLORE
    ################

    def test_get_mood_playlists(self):
        categories = self.yt.get_mood_categories()
        self.assertGreater(len(list(categories)), 0)
        cat = list(categories)[0]
        self.assertGreater(len(categories[cat]), 0)
        playlists = self.yt.get_mood_playlists(categories[cat][0]["params"])
        self.assertGreater(len(playlists), 0)

    def test_get_charts(self):
        charts = self.yt_oauth.get_charts()
        self.assertEqual(len(charts), 4)
        charts = self.yt.get_charts(country="US")
        self.assertEqual(len(charts), 5)
        charts = self.yt.get_charts(country="BE")
        self.assertEqual(len(charts), 4)

    ###############
    # WATCH
    ###############

    def test_get_watch_playlist(self):
        playlist = self.yt_oauth.get_watch_playlist(
            playlistId="RDAMPLOLAK5uy_l_fKDQGOUsk8kbWsm9s86n4-nZNd2JR8Q",
            radio=True,
            limit=90,
        )
        self.assertGreaterEqual(len(playlist["tracks"]), 90)
        playlist = self.yt_oauth.get_watch_playlist("9mWr4c_ig54", limit=50)
        self.assertGreater(len(playlist["tracks"]), 45)
        playlist = self.yt_oauth.get_watch_playlist("UoAf_y9Ok4k")  # private track
        self.assertGreaterEqual(len(playlist["tracks"]), 25)
        playlist = self.yt.get_watch_playlist(
            playlistId="OLAK5uy_lKgoGvlrWhX0EIPavQUXxyPed8Cj38AWc", shuffle=True)
        self.assertEqual(len(playlist["tracks"]), 12)
        playlist = self.yt_brand.get_watch_playlist(playlistId=config["playlists"]["own"],
                                                    shuffle=True)
        self.assertEqual(len(playlist["tracks"]), 4)

    ################
    # LIBRARY
    ################

    def test_get_library_playlists(self):
        playlists = self.yt_oauth.get_library_playlists(50)
        self.assertGreater(len(playlists), 25)

        playlists = self.yt_auth.get_library_playlists(None)
        self.assertGreaterEqual(len(playlists), config.getint("limits", "library_playlists"))

        playlists = self.yt_empty.get_library_playlists()
        self.assertLessEqual(len(playlists), 1)  # "Episodes saved for later"

    def test_get_library_songs(self):
        self.assertRaises(Exception, self.yt_auth.get_library_songs, None, True)
        songs = self.yt_oauth.get_library_songs(100)
        self.assertGreaterEqual(len(songs), 100)
        songs = self.yt_auth.get_library_songs(200, validate_responses=True)
        self.assertGreaterEqual(len(songs), config.getint("limits", "library_songs"))
        songs = self.yt_auth.get_library_songs(order="a_to_z")
        self.assertGreaterEqual(len(songs), 25)
        songs = self.yt_empty.get_library_songs()
        self.assertEqual(len(songs), 0)

    def test_get_library_albums(self):
        albums = self.yt_oauth.get_library_albums(100)
        self.assertGreater(len(albums), 50)
        for album in albums:
            self.assertIn("playlistId", album)
        albums = self.yt_brand.get_library_albums(100, order="a_to_z")
        self.assertGreater(len(albums), 50)
        albums = self.yt_brand.get_library_albums(100, order="z_to_a")
        self.assertGreater(len(albums), 50)
        albums = self.yt_brand.get_library_albums(100, order="recently_added")
        self.assertGreater(len(albums), 50)
        albums = self.yt_empty.get_library_albums()
        self.assertEqual(len(albums), 0)

    def test_get_library_artists(self):
        artists = self.yt_auth.get_library_artists(50)
        self.assertGreater(len(artists), 40)
        artists = self.yt_oauth.get_library_artists(order="a_to_z", limit=50)
        self.assertGreater(len(artists), 40)
        artists = self.yt_brand.get_library_artists(limit=None)
        self.assertGreater(len(artists), config.getint("limits", "library_artists"))
        artists = self.yt_empty.get_library_artists()
        self.assertEqual(len(artists), 0)

    def test_get_library_subscriptions(self):
        artists = self.yt_brand.get_library_subscriptions(50)
        self.assertGreater(len(artists), 40)
        artists = self.yt_brand.get_library_subscriptions(order="z_to_a")
        self.assertGreater(len(artists), 20)
        artists = self.yt_brand.get_library_subscriptions(limit=None)
        self.assertGreater(len(artists), config.getint("limits", "library_subscriptions"))
        artists = self.yt_empty.get_library_subscriptions()
        self.assertEqual(len(artists), 0)

    def test_get_liked_songs(self):
        songs = self.yt_brand.get_liked_songs(200)
        self.assertGreater(len(songs["tracks"]), 100)
        songs = self.yt_empty.get_liked_songs()
        self.assertEqual(songs["trackCount"], 0)

    def test_get_history(self):
        songs = self.yt_oauth.get_history()
        self.assertGreater(len(songs), 0)

    def test_manipulate_history_items(self):
        song = self.yt_auth.get_song(sample_video)
        response = self.yt_auth.add_history_item(song)
        self.assertEqual(response.status_code, 204)
        songs = self.yt_auth.get_history()
        self.assertGreater(len(songs), 0)
        response = self.yt_auth.remove_history_items([songs[0]["feedbackToken"]])
        self.assertIn("feedbackResponses", response)

    def test_rate_song(self):
        response = self.yt_auth.rate_song(sample_video, "LIKE")
        self.assertIn("actions", response)
        response = self.yt_auth.rate_song(sample_video, "INDIFFERENT")
        self.assertIn("actions", response)

    def test_edit_song_library_status(self):
        album = self.yt_brand.get_album(sample_album)
        response = self.yt_brand.edit_song_library_status(
            album["tracks"][2]["feedbackTokens"]["add"])
        self.assertTrue(response["feedbackResponses"][0]["isProcessed"])
        response = self.yt_brand.edit_song_library_status(
            album["tracks"][2]["feedbackTokens"]["remove"])
        self.assertTrue(response["feedbackResponses"][0]["isProcessed"])

    def test_rate_playlist(self):
        response = self.yt_auth.rate_playlist("OLAK5uy_l3g4WcHZsEx_QuEDZzWEiyFzZl6pL0xZ4", "LIKE")
        self.assertIn("actions", response)
        response = self.yt_auth.rate_playlist("OLAK5uy_l3g4WcHZsEx_QuEDZzWEiyFzZl6pL0xZ4",
                                              "INDIFFERENT")
        self.assertIn("actions", response)

    def test_subscribe_artists(self):
        self.yt_auth.subscribe_artists(["UCUDVBtnOQi4c7E8jebpjc9Q", "UCiMhD4jzUqG-IgPzUmmytRQ"])
        self.yt_auth.unsubscribe_artists(["UCUDVBtnOQi4c7E8jebpjc9Q", "UCiMhD4jzUqG-IgPzUmmytRQ"])

    ###############
    # PLAYLISTS
    ###############

    def test_get_playlist_foreign(self):
        self.assertRaises(Exception, self.yt.get_playlist, "PLABC")
        playlist = self.yt.get_playlist("PLPK7133-0ahmzknIfvNUMNJglX-O1rTd2", limit=300, suggestions_limit=7)
        self.assertGreater(len(playlist['duration']), 5)
        self.assertGreater(len(playlist["tracks"]), 200)
        self.assertNotIn("suggestions", playlist)

        self.yt.get_playlist("RDATgXd-")
        self.assertGreaterEqual(len(playlist["tracks"]), 100)

        playlist = self.yt_oauth.get_playlist("PLj4BSJLnVpNyIjbCWXWNAmybc97FXLlTk",
                                              limit=None,
                                              related=True)
        self.assertGreater(len(playlist["tracks"]), 200)
        self.assertEqual(len(playlist["related"]), 0)

    def test_get_playlist_owned(self):
        playlist = self.yt_brand.get_playlist(config["playlists"]["own"],
                                              related=True,
                                              suggestions_limit=21)
        self.assertLess(len(playlist["tracks"]), 100)
        self.assertEqual(len(playlist["suggestions"]), 21)
        self.assertEqual(len(playlist["related"]), 10)

    def test_edit_playlist(self):
        playlist = self.yt_brand.get_playlist(config["playlists"]["own"])
        response = self.yt_brand.edit_playlist(
            playlist["id"],
            title="",
            description="",
            privacyStatus="PRIVATE",
            moveItem=(
                playlist["tracks"][1]["setVideoId"],
                playlist["tracks"][0]["setVideoId"],
            ),
        )
        self.assertEqual(response, "STATUS_SUCCEEDED", "Playlist edit failed")
        self.yt_brand.edit_playlist(
            playlist["id"],
            title=playlist["title"],
            description=playlist["description"],
            privacyStatus=playlist["privacy"],
            moveItem=(
                playlist["tracks"][0]["setVideoId"],
                playlist["tracks"][1]["setVideoId"],
            ),
        )
        self.assertEqual(response, "STATUS_SUCCEEDED", "Playlist edit failed")

    # end to end test adding playlist, adding item, deleting item, deleting playlist
    def test_end2end(self):
        playlistId = self.yt_brand.create_playlist(
            "test",
            "test description",
            source_playlist="OLAK5uy_lGQfnMNGvYCRdDq9ZLzJV2BJL2aHQsz9Y",
        )
        self.assertEqual(len(playlistId), 34, "Playlist creation failed")
        response = self.yt_brand.add_playlist_items(
            playlistId,
            [sample_video, sample_video],
            source_playlist="OLAK5uy_nvjTE32aFYdFN7HCyMv3cGqD3wqBb4Jow",
            duplicates=True,
        )
        self.assertEqual(response["status"], "STATUS_SUCCEEDED", "Adding playlist item failed")
        self.assertGreater(len(response["playlistEditResults"]), 0, "Adding playlist item failed")
        time.sleep(2)
        playlist = self.yt_brand.get_playlist(playlistId, related=True)
        self.assertEqual(len(playlist["tracks"]), 46, "Getting playlist items failed")
        response = self.yt_brand.remove_playlist_items(playlistId, playlist["tracks"])
        self.assertEqual(response, "STATUS_SUCCEEDED", "Playlist item removal failed")
        self.yt_brand.delete_playlist(playlistId)

    ###############
    # UPLOADS
    ###############

    def test_get_library_upload_songs(self):
        results = self.yt_oauth.get_library_upload_songs(50, order="z_to_a")
        self.assertGreater(len(results), 25)

        results = self.yt_empty.get_library_upload_songs(100)
        self.assertEqual(len(results), 0)

    def test_get_library_upload_albums(self):
        results = self.yt_oauth.get_library_upload_albums(50, order="a_to_z")
        self.assertGreater(len(results), 40)

        albums = self.yt_auth.get_library_upload_albums(None)
        self.assertGreaterEqual(len(albums), config.getint("limits", "library_upload_albums"))

        results = self.yt_empty.get_library_upload_albums(100)
        self.assertEqual(len(results), 0)

    def test_get_library_upload_artists(self):
        artists = self.yt_oauth.get_library_upload_artists(None)
        self.assertGreaterEqual(len(artists), config.getint("limits", "library_upload_artists"))

        results = self.yt_auth.get_library_upload_artists(50, order="recently_added")
        self.assertGreaterEqual(len(results), 25)

        results = self.yt_empty.get_library_upload_artists(100)
        self.assertEqual(len(results), 0)

    def test_upload_song(self):
        self.assertRaises(Exception, self.yt_auth.upload_song, "song.wav")
        self.assertRaises(Exception, self.yt_oauth.upload_song, config["uploads"]["file"])
        response = self.yt_auth.upload_song(get_resource(config["uploads"]["file"]))
        self.assertEqual(response.status_code, 409)

    @unittest.skip("Do not delete uploads")
    def test_delete_upload_entity(self):
        results = self.yt_oauth.get_library_upload_songs()
        response = self.yt_oauth.delete_upload_entity(results[0]["entityId"])
        self.assertEqual(response, "STATUS_SUCCEEDED")

    def test_get_library_upload_album(self):
        album = self.yt_oauth.get_library_upload_album(config["uploads"]["private_album_id"])
        self.assertGreater(len(album["tracks"]), 0)

    def test_get_library_upload_artist(self):
        tracks = self.yt_oauth.get_library_upload_artist(config["uploads"]["private_artist_id"],
                                                         100)
        self.assertGreater(len(tracks), 0)


if __name__ == "__main__":
    unittest.main()
