#!/usr/bin/env python3
import os
from os.path import isfile, join
import readline
import youtube_dl

# MUSIC_DIRECTORY = "~/Music"
MUSIC_DIRECTORY = "./music"

COMMANDS = ['extra', 'extension', 'stuff', 'errors',
            'email', 'foobar', 'foo']
FILES = []
DIRECTORIES = []

# TODO: Pull via command line args
def get_playlist_url():
    # return "https://music.youtube.com/watch?v=Enm0XL7xx_E&feature=share" # Deer Tick
    # return "https://music.youtube.com/watch?v=hJLb0zPBzkE&feature=share" # You Worry Me
    return "https://music.youtube.com/playlist?list=OLAK5uy_lHnMUbm8pKsyMTRQNCrjM2v4CPvIJUWq0&feature=share" # Button the busker


def download_song(song_url):
    """
    Download a song using youtube url and song title
    """

    ydl_opts = {
        'format': 'bestaudio/best',
        'postprocessors': [
            {'key': 'FFmpegExtractAudio','preferredcodec': 'mp3',
             'preferredquality': '192',
            },
            {'key': 'FFmpegMetadata'},
        ],
        'write-info-json': True
    }

    with youtube_dl.YoutubeDL(ydl_opts) as ydl:
        info_dict = ydl.extract_info(song_url, download=True)

    print(info_dict['artist'])


def sort_stuff():
    # read in all the files and directories to move
    get_all_files()

    # setup tab completion
    readline.parse_and_bind("tab: complete")
    readline.set_completer(complete)

    in_directory = False

    # loop over all the files left and
    print(len(FILES))
    for f in FILES:
        in_directory = False
        while not in_directory:
            print(f)
            directory = input('Enter directory: ')
            if directory in DIRECTORIES:
                move_file(directory, f)
                in_directory = True
            if directory == "refresh":
                get_directories()


def complete(text, state):
    for cmd in DIRECTORIES:
        if cmd.startswith(text):
            if not state:
                return cmd
            else:
                state -= 1


def get_all_files():
    files = os.listdir()
    global FILES
    global DIRECTORIES

    for f in files:
        if os.path.isfile(f):
            FILES.append(f)
        elif os.path.isdir(f):
            DIRECTORIES.append(f)
        else:
            raise Exception("File wasn't a file or directory! " + f)
    FILES.sort()


def get_directories():
    files = os.listdir()
    global DIRECTORIES

    for f in files:
        if os.path.isdir(f):
            DIRECTORIES.append(f)


def move_file(d, f):
    # print("directory " + d + " file: " + f)
    print(d + "/" + f)
    os.rename(f, (d + "/" + f))


if __name__ == "__main__":
    # Get the playlist url from youtube music
    playlist_url = get_playlist_url()

    download_song(playlist_url)