from DataBase import *
from SpotifyWebAPI import fetch_data
import cmd

playlists = get_data('playlistTracks')
top_artists = get_data('top_artists')
top_tracks = get_data('top_tracks')
user_tracks = get_data('tracks')


def get_playlist_names(pl):
    return [name for name, items in pl.items()]


def print_playlist_tracks(pl_id: int):
    pls = list(playlists)
    name = pls[pl_id]
    pl = playlists[name]
    print(name)
    for track in pl:
        track_data = track['track']
        track_name = track_data['name']
        artists = get_artists_str(track_data['artists'])
        print(f"   '{track_name}' - '{artists}' ")


def print_top_artists():
    for artist in top_artists:
        name = artist['name']
        print(f"   '{name}'")


def print_top_tracks():
    for track in top_tracks:
        track_name = track['name']
        artists = get_artists_str(track['artists'])
        print(f"   '{track_name}' - '{artists}' ")


def print_stats():
    print(f" tracks - {len(user_tracks)}")
    print(f" playlists - {len(playlists)}")

    for name, tracks in playlists.items():
        print(f"    '{name}' - {len(tracks)}")

    print(f" top tracks - {len(top_tracks)}")
    print(f" top artists - {len(top_artists)}")


def print_tracks():
    track_idx = 0
    for trackItem in user_tracks:
        track = trackItem['track']
        print(f"    {track_idx}:  '{track['name']}' by '{get_artists_str(track['artists'])}'")
        track_idx += 1


class Interpreter(cmd.Cmd):
    intro = "Welcome to the interactive command loop. Type help or ? to list commands.\n"
    prompt = "(spotify) "

    def do_playlists(self, arg):
        """prints user playlists"""
        pl_id = 0
        for pl in get_playlist_names(playlists):
            print(f"{pl_id} - {pl}")
            pl_id += 1

    def do_playlist_tracks(self, arg):
        """prints playlist [name]"""
        try:
            pl_id = int(arg.split()[0])
            print_playlist_tracks(pl_id)
        except IndexError:
            print("Invalid input")

    def do_top_artists(self, arg):
        """Top artists"""
        print_top_artists()

    def do_top_tracks(self, arg):
        """Top tracks"""
        print_top_tracks()

    def do_stat(self, arg):
        """Print stats"""
        print_stats()

    def do_tracks(self, arg):
        """Print tracks"""
        print_tracks()

    def do_fetch(self, arg):
        """Fetch all spotify data"""
        try:
            fetch_data()

            global playlists, top_artists, top_tracks, user_tracks

            playlists = get_data('playlistTracks')
            top_artists = get_data('top_artists')
            top_tracks = get_data('top_tracks')
            user_tracks = get_data('tracks')

        except ValueError:
            print("Can not fetch the data.")

    def do_exit(self, arg):
        """Exit the command loop: exit"""
        print("Goodbye!")
        return True


if __name__ == '__main__':
    Interpreter().cmdloop()