#!/usr/bin/env python3

import sys
import tekore as tk
import pyml_config as config

# Set up Spotify connection using client token
conf = (config.client_id, config.client_secret)
token = tk.request_client_token(*conf)
spotify = tk.Spotify(token)

# Get track ID from first argument
trackarg = sys.argv[1]
#print(trackarg)

try: 
	#tk.from_url(trackarg)
	#print("Looks like it's a track URL, we'll use that")
	track = spotify.track(tk.from_url(trackarg)[1])
except:
	#print("Looks like it's not a URL, maybe it's a URI?")
	try: 
		track = spotify.track(trackarg)
	except:
		print("Still failed?!?")
#trackid = tk.from_url(trackarg)[1]
#track = spotify.track(trackid)
#print(track.asbuiltin())

available_markets = track.asbuiltin()['available_markets']

# Get the ISRC code to check for duplicate tracks that will be substituted automatically by Spotify if the submitted version isn't available in a jurisdiction
isrc = track.asbuiltin()['external_ids']['isrc']

# Now check for those additional versions, and append them to the available_markets array. Probably better ways to do this...
search_results = spotify.search('isrc:{}'.format(isrc), types=('track',))
for p in search_results:
	if p.total > 1:
		pagenum = 1
		while pagenum <= p.total:
			itemnum = pagenum - 1
			available_markets += p.items[itemnum].asbuiltin()['available_markets']
			pagenum = pagenum + 1
			p.next

# Clear out duplicates
available_markets = list(dict.fromkeys(available_markets))
# Order it
available_markets.sort()


print(available_markets)