#!/usr/bin/env python3

from gmusicapi import Mobileclient
import argparse
import sys
import requests

creds_path=sys.path[0] + '/mobileclient.cred'

def authenticate(api):
    creds = api.perform_oauth(storage_filepath=creds_path, open_browser=False)

def transferLibrary(gpm_api, mudbase_api):
    songs = gpm_api.get_all_songs()
    
    # Determine all unique artists
    artists = sorted(set([song['artist'] for song in songs if 'artist' in song]))

    # Determine all genres
    genres = sorted(set([song['genre'] for song in songs if 'genre' in song]))

    # Store the artist index of each song
    songArtistIdxs = [ artists.index(song['artist']) for song in songs ]

    # Store the genre index of each song
    songArtistIdxs = [ genres.index(song['genre']) for song in songs ]
    
    # Determine all unique albums per artist
    artistAlbums = [ sorted(set([ song['album'] for song in songs if song['artist'] == artist ])) for artist in artists ]

    # Determine store ID for all artists
    def getArtistStoreIds(song):
        if 'artistId' in song:
            return [ song['artistId'][0] ]
        return [];
    artistStoreIds = [ [ getArtistStoreIds(song) for song in songs if song['artist'] == artist ][0] for artist in artists ]

    # Create GPM import tag
    gpmTagIdResponse = requests.post(mudbase_api + '/tag', data = {
        'name': 'GPM Import'
    }).json()
    print(f"Created tag \"GPM Import\", response: {gpmTagIdResponse}")

    # Create genres and store their mudbase Ids
    genreRootResponse = requests.post(mudbase_api + '/tag', data = {
        'name': 'Genre'
    }).json()
    print(f"Created tag \"Genre\", response: {genreRootResponse}")
    genreMudbaseIds = []
    for idx,genre in enumerate(genres):
        response = requests.post(mudbase_api + '/tag', data = {
            'name': genre,
            'parentId': genreRootResponse['id'] 
        }).json()
        print(f"Created tag \"Genre/{genre}\", response: {response}")
        genreMudbaseIds.append(response['id'])

    # Create artists and store their mudbase Ids
    artistMudbaseIds = []
    for idx,artist in enumerate(artists):
        response = requests.post(mudbase_api + '/artist', json = {
            'name': artist,
            'storeLinks': [ 'https://play.google.com/music/m/' + id for id in artistStoreIds[idx] ],
            'tagIds': [ gpmTagIdResponse['id'] ]
        }).json()
        print(f"Created artist \"{artist}\", response: {response}")
        artistMudbaseIds.append(response['id'])

    # Create songs
    def getSongStoreIds(song):
        if 'storeId' in song:
            return [ song['storeId'] ]
        return [];
    for song in songs:
        artistMudbaseId = artistMudbaseIds[ artists.index(song['artist']) ]
        genreMudbaseId = genreMudbaseIds[ genres.index(song['genre']) ]
        response = requests.post(mudbase_api + '/song', json = {
            'title': song['title'],
            'artistIds': [ artistMudbaseId ],
            'tagIds' : [ genreMudbaseId, gpmTagIdResponse['id'] ],
            'storeLinks': [ 'https://play.google.com/music/m/' + id for id in getSongStoreIds(song) ],
        }).json()
        print(f"Created song \"{song['title']}\" with artist ID {artistMudbaseId}, response: {response}")

api = Mobileclient()

parser = argparse.ArgumentParser(description="Import Google Music library into MudBase.")
parser.add_argument('--authenticate', help="Generate credentials for authentication", action="store_true")
parser.add_argument('mudbase_api', help="Address for the Mudbase back-end API")

args = parser.parse_args()

if args.authenticate:
    authenticate(api)
else:
    api.oauth_login(Mobileclient.FROM_MAC_ADDRESS, oauth_credentials=creds_path)
    transferLibrary(api, args.mudbase_api)