import os
from mutagen.easyid3 import EasyID3
from mutagen.mp3 import MP3
import requests, ast, time, math, argparse, json

loading = ["-","\\","|","/"]

parser=argparse.ArgumentParser(description="Options for the generation of the song database")
parser.add_argument('-k','--apikey', help='String: LastFM api key', default="")
parser.add_argument('-m', '--mode', help='new/update: Remake database or update current', default= "update")
parser.add_argument('-a', '--art', help="True/False: Add art to the database using LastFm (takes minimum 0.25s per song)", default="True")
parser.add_argument('-d','--directory',help="Directory of the song files (USING FORWARD SLASHES)", default="./sound")
args = parser.parse_args()
apikeylastfm = args.apikey
soundLocation = args.directory
# if you want to set the api key/sound directory permenantly for your setup just uncomment the next line
# apikeylastfm = "KeyHere"
# soundLocation = "directoryHere"
songFiles = os.listdir(soundLocation)
if args.mode == "update":
    try:   
        with open('songDatabase.json', 'r') as handle:
            songDatabaseList = json.load(handle)
    except:
        songDatabaseList=[]

    for i in songDatabaseList:
        try:
            songFiles.index(i["file"]) != -1
        except:
            print("deleted: " + i["file"] + " from database")
            songDatabaseList.pop(songDatabaseList.index(i))
            
    for i in songDatabaseList:
        songFiles.pop(songFiles.index(i["file"]))
    print("new songs: " + str(songFiles))
elif args.mode=="new":
    songDatabaseList = []

if args.art.lower() == "true":
    x = len(songFiles)*0.25
    if x > 60:
        print("ETA "+ str(x/60) + " minutes")
    else:
        print("ETA "+ str(x) + " seconds")

for i in songFiles:
    try:
        song = EasyID3(soundLocation+"/"+i)
        title = song['title'][0]
        artist = song['artist'][0]
    except:
        try:
            song = i.split("_")
            title = song[0]
            artist = song[1].split(".")[0]
        except:
            title = i
            artist = None
    if args.art.lower() == "true" and not(args.apikey == ""):
        try:
            image = ast.literal_eval(requests.post(url="http://ws.audioscrobbler.com/2.0/?method=track.getInfo&api_key="+apikeylastfm+"&artist="+artist+"&track="+title+"&format=json").text)["track"]["album"]["image"][1]["#text"]
            if image == "":
                image = ast.literal_eval(requests.post(url="http://ws.audioscrobbler.com/2.0/?method=track.getInfo&api_key="+apikeylastfm+"&artist="+artist+"&track="+title+"&format=json").text)["track"]["album"]["image"][2]["#text"]
                if image == "":
                    image = None
            time.sleep(0.25)
        except:
            image=None
    else:
        image=None
    try:
        length = math.ceil(MP3(soundLocation+"/"+i).info.length)
    except:
        length = 0
    if len(songFiles) != 1:
        index = (songFiles.index(i))%4
        print("\r" + str(loading[index] + str(math.floor((songFiles.index(i)/(len(songFiles)-1))*100))+ "%"), end='', flush=True)
    # each "song" is stored as a dictionary containing the below stuff, and each dictionary is put into a list
    songDatabaseList.append({"file":i,"title":title,"artist":artist,"art":image,"length":length})
    
with open('songDatabase.json', 'w') as handle:
    json.dump(songDatabaseList, handle)