#!/usr/bin/env python3

import sys
import os
import requests
import glob
from mido import MidiFile
from configparser import ConfigParser

url = os.environ.get("API_URL")

def getOrCreateAlbum(name, artistId):
	if not name:
		return
	res = requests.post(f"{url}/album", json={
		"name": name,
		"artist": artistId,
	})
	out = res.json()
	print(out)
	return out["id"]

def getOrCreateGenre(names):
	ids = []
	for name in names.split(","):
		res = requests.post(f"{url}/genre", json={
			"name": name,
		})
		out = res.json()
		print(out)
		ids += [out["id"]]
	#TODO handle multiple genres
	return ids[0]

def getOrCreateArtist(name):
	res = requests.post(f"{url}/artist", json={
		"name": name,
	})
	out = res.json()
	print(out)
	return out["id"]

def populateFile(path, midi, mxl):
	config = ConfigParser()
	config.read(path)
	mid = MidiFile(midi)
	metadata = config["Metadata"];
	difficulties = dict(config["Difficulties"])
	difficulties["length"] = round((mid.length), 2)
	artistId = getOrCreateArtist(metadata["Artist"])
	print(f"Populating {metadata['Name']}")
	res = requests.post(f"{url}/song", json={
		"name": metadata["Name"],
		"midiPath": f"/assets/{midi}",
		"musicXmlPath": f"/assets/{mxl}",
		"difficulties": difficulties,
		"artist": artistId,
		"album": getOrCreateAlbum(metadata["Album"], artistId),
		"genre": getOrCreateGenre(metadata["Genre"]),
		"illustrationPath": f"/assets/{os.path.commonpath([midi, mxl])}/illustration.png"
	})
	print(res.json())

def main():
	global url
	if url == None:
		url = "http://localhost:3000"
	print("Searching for files...")
	for file in glob.glob("**/*.ini", recursive=True):
		print(f"File found: {file}")
		path = os.path.splitext(file)[0]
		populateFile(file, path + ".midi", path + ".mxl")

if __name__ == "__main__":
	exit(main())