# -*- coding: utf-8 -*-
# Copyright 2017 Leo Moll and Dominik Schlösser
#

# -- Imports ------------------------------------------------
import xbmcplugin, xbmcgui

from resources.lib.film import Film
from resources.lib.settings import Settings

# -- Classes ------------------------------------------------
class FilmUI( Film ):
	def __init__( self, plugin, sortmethods = None ):
		self.plugin			= plugin
		self.handle			= plugin.addon_handle
		self.settings		= Settings()
		self.sortmethods	= sortmethods if sortmethods is not None else [ xbmcplugin.SORT_METHOD_TITLE, xbmcplugin.SORT_METHOD_DATE, xbmcplugin.SORT_METHOD_DURATION, xbmcplugin.SORT_METHOD_SIZE ]
		self.showshows		= False
		self.showchannels	= False

	def Begin( self, showshows, showchannels ):
		self.showshows		= showshows
		self.showchannels	= showchannels
		# xbmcplugin.setContent( self.handle, 'tvshows' )
		for method in self.sortmethods:
			xbmcplugin.addSortMethod( self.handle, method )

	def Add( self, alttitle = None, totalItems = None ):
		# get the best url
		videourl = self.url_video_hd if ( self.url_video_hd != "" and self.settings.preferhd ) else self.url_video if self.url_video != "" else self.url_video_sd
		videohds = " (HD)" if ( self.url_video_hd != "" and self.settings.preferhd ) else ""
		# exit if no url supplied
		if videourl == "":
			return

		if alttitle is not None:
			resultingtitle = alttitle
		else:
			if self.showshows:
				resultingtitle = self.show + ': ' + self.title
			else:
				resultingtitle = self.title
			if self.showchannels:
				resultingtitle += ' [' + self.channel + ']'

		infoLabels = {
			'title' : resultingtitle + videohds,
			'sorttitle' : resultingtitle,
			'tvshowtitle' : self.show,
			'plot' : self.description
		}

		if self.size > 0:
			infoLabels['size'] = self.size * 1024 * 1024

		if self.seconds > 0:
			infoLabels['duration'] = self.seconds

		if self.aired is not None:
			airedstring = '%s' % self.aired
			if airedstring[:4] != '1970':
				infoLabels['date']		= airedstring[8:10] + '-' + airedstring[5:7] + '-' + airedstring[:4]
				infoLabels['aired']		= airedstring
				infoLabels['dateadded']	= airedstring

		li = xbmcgui.ListItem( resultingtitle, self.description )
		li.setInfo( type = 'video', infoLabels = infoLabels )
		li.setProperty( 'IsPlayable', 'true' )

		# create context menu
		contextmenu = []
		if self.size > 0:
			# Download video
			contextmenu.append( (
				self.plugin.language( 30921 ),
				'RunPlugin({})'.format( self.plugin.build_url( { 'mode': "download", 'id': self.id, 'quality': 1 } ) )
			) )
			if self.url_video_hd:
				# Download SD video
				contextmenu.append( (
					self.plugin.language( 30923 ),
					'RunPlugin({})'.format( self.plugin.build_url( { 'mode': "download", 'id': self.id, 'quality': 2 } ) )
				) )
			if self.url_video_sd:
				# Download SD video
				contextmenu.append( (
					self.plugin.language( 30922 ),
					'RunPlugin({})'.format( self.plugin.build_url( { 'mode': "download", 'id': self.id, 'quality': 0 } ) )
				) )
		# Add to queue
		# TODO: Enable later
#		contextmenu.append( (
#			self.plugin.language( 30924 ),
#			'RunPlugin({})'.format( self.plugin.build_url( { 'mode': "enqueue", 'id': self.id } ) )
#		) )
		li.addContextMenuItems( contextmenu )

		if totalItems is not None:
			xbmcplugin.addDirectoryItem(
				handle		= self.handle,
				url			= videourl,
				listitem	= li,
				isFolder	= False,
				totalItems	= totalItems
			)
		else:
			xbmcplugin.addDirectoryItem(
				handle		= self.handle,
				url			= videourl,
				listitem	= li,
				isFolder	= False
			)

	def End( self ):
		xbmcplugin.endOfDirectory( self.handle, cacheToDisc = False )