#!/usr/bin/env python3
import datetime
import glob
import json
import music_tag
import os
from os.path import isfile, join
from pathlib import Path
import readline
import sys
import youtube_dl


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

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

AUDIO_OPTIONS = {
        'format': 'bestaudio/best',
        'cookiefile': 'cookies.txt',
        'outtmpl': MUSIC_DIRECTORY + '%(title)s.%(ext)s',
        'postprocessors': [
            {
                'key': 'FFmpegExtractAudio',
                'preferredcodec': 'mp3',
                'preferredquality': '192',
            },
            {'key': 'FFmpegMetadata'},
        ],
        'writeinfojson': True
    }

ytdl = youtube_dl.YoutubeDL(AUDIO_OPTIONS)


# TODO: Make this better with argparse
def get_playlist_url():
    return sys.argv[1]


def download_song(song_url):
    """
    Download a song using youtube url and song title
    """
    return ytdl.extract_info(song_url, download=True)


def write_metadata_to_song_file(file, metadata):
    f = music_tag.load_file(file)

    f['name'] = metadata['title']
    f['artist'] = metadata['artist']
    f['album'] = metadata['album']
    f['year'] = format_youtube_date(metadata['release_date'])

    f.save()


def format_youtube_date(date):
    fmt = "%Y%m%d"
    d = datetime.datetime.strptime(date, fmt)
    return d.year


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(directory):
    things = glob.glob(os.path.join(directory, '*.mp3'))

    files = []

    for thing in things:
        if os.path.isfile(thing):
            files.append(thing)

    return files


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

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


def move_file(file, metadata):
    artist = 'unknownartist'
    album = 'unknownalbum'
    title = 'unknownsong'

    if ('artist' in metadata.keys()) and (metadata['artist'] is not None):
        artist = metadata['artist'].replace(' ', '_').lower()

    if ('album' in metadata.keys()) and (metadata['album'] is not None):
        album = metadata['album'].replace(' ', '_').lower()

    if ('title' in metadata.keys()) and (metadata['title'] is not None):
        title = metadata['title'].replace(' ', '_').lower()

    final_directory = os.path.join(
        MUSIC_DIRECTORY,
        artist,
        album)

    Path(final_directory).mkdir(parents=True, exist_ok=True)

    os.rename(
        file,
        os.path.join(final_directory, artist + '_' + title + '.mp3'))


def get_video_urls_in_playlist(playlist_url):
    videos = ytdl.extract_info(playlist_url, download=False)

    urls = []
    for vid in videos['entries']:
        if 'webpage_url' in vid.keys() and vid['webpage_url'] is not None:
            urls.append(vid['webpage_url'])

    return urls


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

    urls = get_video_urls_in_playlist(playlist_url)

    for url in urls:
        try:
            # TODO: make this continue downloading if a song fails
            download_song(url)
        except Exception as ex:
            # TODO: Handle this a different way
            print(ex)

    files = get_all_files(MUSIC_DIRECTORY)

    for f in files:
        json_data = None
        with open(f.replace('.mp3', '.info.json')) as json_file:
            json_data = json.load(json_file)
            write_metadata_to_song_file(f, json_data)
        move_file(f, json_data)