#!/usr/bin/env python3
import os.path as path
from os import mkdir, rmdir
import json

import deemix.utils.localpaths as localpaths

import spotipy
from spotipy.oauth2 import SpotifyClientCredentials

class SpotifyHelper:
	def __init__(self):
		self.credentials = {}
		self.spotifyEnabled = False
		self.sp = None
		self.initCredentials()

	def initCredentials(self):
		configFolder = localpaths.getConfigFolder()
		if not path.isdir(configFolder):
			mkdir(configFolder)
		if not path.isfile(path.join(configFolder, 'authCredentials.json')):
			with open(path.join(configFolder, 'authCredentials.json'), 'w') as f:
				json.dump({'clientId': "", 'clientSecret': ""}, f, indent=2)
		with open(path.join(configFolder, 'authCredentials.json'), 'r') as credentialsFile:
			self.credentials = json.load(credentialsFile)
		self.checkCredentials()

	def checkCredentials(self):
		if self.credentials['clientId'] == "" or self.credentials['clientSecret'] == "":
			spotifyEnabled = False
		else:
			try:
				self.createSpotifyConnection()
				self.sp.user_playlists('spotify')
				self.spotifyEnabled = True
			except Exception as e:
				self.spotifyEnabled = False
		return self.spotifyEnabled

	def getCredentials(self):
		return self.credentials

	def setCredentials(self, spotifyCredentials):
		configFolder = localpaths.getConfigFolder()
		with open(path.join(configFolder, 'authCredentials.json'), 'w') as f:
			json.dump(spotifyCredentials, f, indent=2)
		self.credentials = spotifyCredentials
		self.checkCredentials()

	def createSpotifyConnection(self):
		client_credentials_manager = SpotifyClientCredentials(client_id=self.credentials['clientId'], client_secret=self.credentials['clientSecret'])
		self.sp = spotipy.Spotify(client_credentials_manager=client_credentials_manager)

	def _convert_playlist_structure(self, spotify_obj):
		if len(spotify_obj['images']):
			url = spotify_obj['images'][0]['url']
		else:
			url = "https://e-cdns-images.dzcdn.net/images/cover/d41d8cd98f00b204e9800998ecf8427e/75x75-000000-80-0-0.jpg"
		deezer_obj = {
			'checksum': spotify_obj['snapshot_id'],
			'collaborative': spotify_obj['collaborative'],
			'creation_date': "????-00-00",
			'creator': {'id': spotify_obj['owner']['id'], 'name': spotify_obj['owner']['display_name'], 'tracklist': spotify_obj['owner']['href'], 'type': "user"},
			'description': spotify_obj['description'],
			'duration': 0,
			'fans': spotify_obj['followers']['total'],
			'id': spotify_obj['id'],
			'is_loved_track': False,
			'link': spotify_obj['external_urls']['spotify'],
			'nb_tracks': spotify_obj['tracks']['total'],
			'picture': url,
			'picture_big': url,
			'picture_medium': url,
			'picture_small': url,
			'picture_xl': url,
			'public': spotify_obj['public'],
			'share': spotify_obj['external_urls']['spotify'],
			'title': spotify_obj['name'],
			'tracklist': spotify_obj['tracks']['href'],
			'type': "playlist"
		}
		return deezer_obj

	def get_trackid_spotify(self, dz, track_id, fallbackSearch, spotifyTrack=None):
		if not self.spotifyEnabled:
			raise spotifyFeaturesNotEnabled
		if not spotifyTrack:
			spotify_track = self.sp.track(track_id)
		else:
			spotify_track = spotifyTrack
		dz_track = 0
		if 'external_ids' in spotify_track and 'isrc' in spotify_track['external_ids']:
			try:
				dz_track = dz.get_track_by_ISRC(spotify_track['external_ids']['isrc'])
				dz_track = dz_track['id'] if 'id' in dz_track else 0
			except:
				dz_track = dz.get_track_from_metadata(spotify_track['artists'][0]['name'], spotify_track['name'], spotify_track['album']['name']) if fallbackSearch else 0
		elif fallbackSearch:
			dz_track = dz.get_track_from_metadata(spotify_track['artists'][0]['name'], spotify_track['name'], spotify_track['album']['name'])
		return dz_track

	def get_albumid_spotify(self, dz, album_id):
		if not self.spotifyEnabled:
			raise spotifyFeaturesNotEnabled
		spotify_album = self.sp.album(album_id)
		dz_album = 0
		if 'external_ids' in spotify_album and 'upc' in spotify_album['external_ids']:
			try:
				dz_album = dz.get_album_by_UPC(spotify_album['external_ids']['upc'])
				dz_album = dz_album['id'] if 'id' in dz_album else 0
			except:
				try:
					dz_album = dz.get_album_by_UPC(int(spotify_album['external_ids']['upc']))
					dz_album = dz_album['id'] if 'id' in dz_album else 0
				except:
					dz_album = 0
		return dz_album

	def convert_spotify_playlist(self, dz, playlist_id, settings):
		if not self.spotifyEnabled:
			raise spotifyFeaturesNotEnabled
		spotify_playlist = self.sp.playlist(playlist_id)
		result = {
			'title': spotify_playlist['name'],
			'artist': spotify_playlist['owner']['display_name'],
			'size': spotify_playlist['tracks']['total'],
			'downloaded': 0,
			'failed': 0,
			'progress': 0,
			'type': 'spotify_playlist',
			'settings': settings or {},
			'id': playlist_id
		}
		if len(spotify_playlist['images']):
			result['cover'] = spotify_playlist['images'][0]['url']
		else:
			result['cover'] = "https://e-cdns-images.dzcdn.net/images/cover/d41d8cd98f00b204e9800998ecf8427e/75x75-000000-80-0-0.jpg"
		playlistAPI = self._convert_playlist_structure(spotify_playlist)
		playlistAPI['various_artist'] = dz.get_artist(5080)
		tracklist = spotify_playlist['tracks']['items']
		result['collection'] = []
		while spotify_playlist['tracks']['next']:
			spotify_playlist['tracks'] = self.sp.next(spotify_playlist['tracks'])
			tracklist += spotify_playlist['tracks']['items']
		totalSize = len(tracklist)
		for pos, track in enumerate(tracklist, start=1):
			trackID = self.get_trackid_spotify(dz, 0, settings['fallbackSearch'], track['track'])
			if trackID == 0:
				deezerTrack = {
					'SNG_ID': 0,
					'SNG_TITLE': track['track']['name'],
					'DURATION': 0,
					'MD5_ORIGIN': 0,
					'MEDIA_VERSION': 0,
					'FILESIZE': 0,
					'ALB_TITLE': track['track']['album']['name'],
					'ALB_PICTURE': "",
					'ART_ID': 0,
					'ART_NAME': track['track']['artists'][0]['name']
				}
			else:
				deezerTrack = dz.get_track_gw(trackID)
			deezerTrack['_EXTRA_PLAYLIST'] = playlistAPI
			deezerTrack['POSITION'] = pos
			deezerTrack['SIZE'] = totalSize
			deezerTrack['FILENAME_TEMPLATE'] = settings['playlistTracknameTemplate']
			result['collection'].append(deezerTrack)
		return result

class spotifyFeaturesNotEnabled(Exception):
	pass