635 lines
21 KiB
Python
635 lines
21 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
#################################################################################################
|
|
|
|
import collections
|
|
import logging
|
|
import requests
|
|
import os
|
|
import urllib
|
|
|
|
import xbmc
|
|
import xbmcgui
|
|
import xbmcvfs
|
|
|
|
import clientinfo
|
|
import downloadutils
|
|
import read_embyserver as embyserver
|
|
from utils import window, settings, language as lang, urllib_path, create_id
|
|
|
|
#################################################################################################
|
|
|
|
log = logging.getLogger("EMBY."+__name__)
|
|
|
|
#################################################################################################
|
|
|
|
|
|
class PlayUtils():
|
|
|
|
play_session_id = None
|
|
method = "DirectPlay"
|
|
force_transcode = False
|
|
|
|
|
|
def __init__(self, item, listitem, **kwargs):
|
|
|
|
self.info = kwargs
|
|
|
|
self.item = item
|
|
self.listitem = listitem
|
|
|
|
self.clientInfo = clientinfo.ClientInfo()
|
|
self.emby = embyserver.Read_EmbyServer()
|
|
|
|
self.server = window('emby_server%s' % window('emby_currUser'))
|
|
self.play_session_id = str(create_id()).replace("-", "")
|
|
self.token = downloadutils.DownloadUtils().get_token()
|
|
|
|
def get_play_url(self, force_transcode=False):
|
|
|
|
''' New style to retrieve the best playback method based on sending
|
|
the profile to the server. Based on capabilities the correct path is returned,
|
|
including livestreams that need to be opened by the server
|
|
'''
|
|
|
|
self.force_transcode = force_transcode
|
|
info = self.get_playback_info()
|
|
url = False if info == False else None
|
|
|
|
if info:
|
|
url = info['Path'].encode('utf-8')
|
|
window('emby_%s.play.json' % url, {
|
|
|
|
'playmethod': self.method,
|
|
'playsession_id': self.play_session_id,
|
|
'mediasource_id': info.get('Id') or self.item['Id']
|
|
})
|
|
|
|
if 'RequiredHttpHeaders' in info and 'User-Agent' in info['RequiredHttpHeaders']:
|
|
self.listitem.setProperty('User-Agent', info['RequiredHttpHeaders']['User-Agent'])
|
|
|
|
log.info("playback info: %s", info)
|
|
log.info("play method: %s play url: %s", self.method, url)
|
|
|
|
return url
|
|
|
|
def _is_media_selection(self):
|
|
|
|
if self.item['MediaType'] != 'Video':
|
|
log.debug("MediaType not video detected.")
|
|
return False
|
|
|
|
elif self.item['Type'] == 'TvChannel':
|
|
log.debug("TvChannel detected.")
|
|
return False
|
|
|
|
elif len(self.item['MediaSources']) == 1 and self.item['MediaSources'][0]['Type'] == 'Placeholder':
|
|
log.debug("Placeholder detected.")
|
|
return False
|
|
|
|
elif 'SourceType' in self.item and self.item['SourceType'] != 'Library':
|
|
log.debug("SourceType not library detected.")
|
|
return False
|
|
|
|
return True
|
|
|
|
def get_playback_info(self):
|
|
|
|
# Get the playback info for the current item
|
|
|
|
info = self.emby.get_playback_info(self.item['Id'], self.get_device_profile())
|
|
media_sources = info['MediaSources']
|
|
|
|
# Select the mediasource
|
|
if not media_sources:
|
|
log.error('No media sources found: %s', info)
|
|
return
|
|
|
|
selected_source = media_sources[0]
|
|
|
|
if 'MediaSourceId' in self.info:
|
|
for source in media_sources:
|
|
if source['Id'] == self.info['MediaSourceId']:
|
|
selected_source = source
|
|
break
|
|
|
|
elif not self._is_media_selection():
|
|
log.info("Special media type detected. Skip source selection.")
|
|
|
|
elif len(media_sources) > 1:
|
|
# Offer choices
|
|
sources = []
|
|
for source in media_sources:
|
|
sources.append(source.get('Name', "na"))
|
|
|
|
resp = xbmcgui.Dialog().select("Select the source", sources)
|
|
if resp > -1:
|
|
selected_source = media_sources[resp]
|
|
else:
|
|
log.info("No media source selected.")
|
|
return False
|
|
|
|
return self.get_optimal_track(selected_source)
|
|
|
|
def get_optimal_track(self, source):
|
|
|
|
''' Because we posted our deviceprofile to the server, only streams will be
|
|
returned that can actually be played by this client so no need to check bitrates etc.
|
|
'''
|
|
|
|
# Log filename, used by other addons eg subtitles which require the file name
|
|
window('embyfilename', value=self.get_direct_path(source).encode('utf-8'))
|
|
|
|
if (not self.force_transcode and (self.is_strm(source) or self.is_h265(source) or (settings('playFromStream') == "false" and self.is_file_exists(source)))):
|
|
# Append external subtitles
|
|
if settings('enableExternalSubs') == "true":
|
|
self.set_external_subs(source, source['Path'])
|
|
else:
|
|
source['Path'] = self.get_http_path(source, True if not source['SupportsDirectStream'] else self.force_transcode)
|
|
|
|
log.debug("get source: %s", source)
|
|
return source
|
|
|
|
def is_file_exists(self, source):
|
|
|
|
path = self.get_direct_path(source)
|
|
|
|
if xbmcvfs.exists(path): # or ":" not in path:
|
|
log.info("Path exists.")
|
|
|
|
self.method = "DirectPlay"
|
|
source['Path'] = path
|
|
|
|
return True
|
|
|
|
log.info("Failed to find file.")
|
|
return False
|
|
|
|
def is_strm(self, source):
|
|
|
|
if source['Container'] == "strm" or '.strm' in self.item['Path']:
|
|
log.info('Strm detected.')
|
|
|
|
self.method = "DirectPlay"
|
|
source['Path'] = self.get_direct_path(source)
|
|
|
|
return True
|
|
|
|
return False
|
|
|
|
def is_h265(self, source):
|
|
|
|
if source['MediaStreams']:
|
|
force_transcode = False
|
|
|
|
for stream in source['MediaStreams']:
|
|
if self._is_h265(stream) or self._is_high10(stream):
|
|
force_transcode = True
|
|
break
|
|
|
|
if force_transcode:
|
|
source['Path'] = self.get_http_path(source, True)
|
|
return True
|
|
|
|
return False
|
|
|
|
@classmethod
|
|
def _is_h265(cls, stream):
|
|
|
|
if stream['Type'] == "Video" and stream['Codec'] in ("hevc", "h265"):
|
|
if settings('transcode_h265') == "true":
|
|
log.info("Force transcode h265/hevc detected.")
|
|
return True
|
|
|
|
return False
|
|
|
|
@classmethod
|
|
def _is_high10(cls, stream):
|
|
|
|
if stream.get('Profile') == "High 10":
|
|
if settings('transcodeHi10P') == "true":
|
|
log.info("Force transcode hi10p detected.")
|
|
return True
|
|
|
|
return False
|
|
|
|
def get_direct_path(self, source):
|
|
|
|
path = source['Path']
|
|
|
|
if 'VideoType' in source:
|
|
if source['VideoType'] == "Dvd":
|
|
path = "%s/VIDEO_TS/VIDEO_TS.IFO" % path
|
|
elif source['VideoType'] == "BluRay":
|
|
path = "%s/BDMV/index.bdmv" % path
|
|
|
|
# Assign network protocol
|
|
if path.startswith('\\\\'):
|
|
path = path.replace('\\\\', "smb://", 1)
|
|
path = path.replace('\\\\', "\\")
|
|
path = path.replace('\\', "/")
|
|
|
|
if "://" in path:
|
|
# Protocol needs to be lowercase, otherwise weird things happen.
|
|
protocol = path.split('://')[0]
|
|
path = path.replace(protocol, protocol.lower())
|
|
|
|
return path
|
|
|
|
def get_http_path(self, source, transcode=False):
|
|
|
|
if transcode and settings('ignoreTranscode') and source['MediaStreams']:
|
|
# Specified by user should not be transcoded.
|
|
ignore_codecs = settings('ignoreTranscode').split(',')
|
|
|
|
for stream in source['MediaStreams']:
|
|
if stream['Type'] == "Video" and stream['Codec'] in ignore_codecs:
|
|
log.info("Ignoring transcode for: %s", stream['Codec'])
|
|
transcode = False
|
|
break
|
|
|
|
url = self.get_transcode_url(source) if transcode else self.get_direct_url(source)
|
|
url = self._append_http_url(source, url)
|
|
|
|
return url
|
|
|
|
def get_direct_url(self, source):
|
|
|
|
self.method = "DirectStream"
|
|
|
|
if self.item['Type'] == "Audio":
|
|
url = "%s/emby/Audio/%s/stream.%s?static=true" % (self.server, self.item['Id'], self.item['MediaSources'][0]['Container'])
|
|
else:
|
|
url = "%s/emby/Videos/%s/stream?static=true" % (self.server, self.item['Id'])
|
|
|
|
# Append external subtitles
|
|
if settings('enableExternalSubs') == "true":
|
|
self.set_external_subs(source, url)
|
|
|
|
return url
|
|
|
|
def get_transcode_url(self, source):
|
|
|
|
self.method = "Transcode"
|
|
|
|
if 'TranscodingUrl' in source:
|
|
base, params = source['TranscodingUrl'].split("?")
|
|
url_parsed = params.split("&")
|
|
for i in url_parsed:
|
|
if 'AudioStreamIndex' in i or 'AudioBitrate' in i or 'PlaySessionId' in i or 'MediaSourceId' in i or 'api_key' in i: #handle manually
|
|
url_parsed.remove(i)
|
|
url = "%s/emby%s?%s" % (self.server, base.replace("stream", "master"), '&'.join(url_parsed))
|
|
else:
|
|
item_id = self.item['Id']
|
|
url = urllib_path("%s/emby/Videos/%s/master.m3u8" % (self.server, item_id), {
|
|
|
|
'VideoCodec': "h264",
|
|
'AudioCodec': "ac3",
|
|
'MaxAudioChannels': 6,
|
|
'DeviceId': self.clientInfo.get_device_id(),
|
|
'VideoBitrate': self.get_bitrate() * 1000
|
|
})
|
|
|
|
# Select audio and subtitles
|
|
url += self.get_audio_subs(source)
|
|
|
|
# Limit to 8 bit if user selected transcode Hi10P
|
|
if settings('transcodeHi10P') == "true":
|
|
url += "&MaxVideoBitDepth=8"
|
|
|
|
# Adjust the video resolution
|
|
url += "&maxWidth=%s&maxHeight=%s" % (self.get_resolution())
|
|
|
|
return url
|
|
|
|
def _append_http_url(self, source, url):
|
|
|
|
url += "&MediaSourceId=%s" % source['Id']
|
|
url += "&PlaySessionId=%s" % self.play_session_id
|
|
url += "&api_key=%s" % self.token
|
|
|
|
return url
|
|
|
|
def set_external_subs(self, source, play_url):
|
|
|
|
subs = []
|
|
mapping = {}
|
|
|
|
item_id = self.item['Id']
|
|
streams = source['MediaStreams']
|
|
|
|
if not source['MediaStreams']:
|
|
log.info("No media streams found.")
|
|
return
|
|
|
|
temp = xbmc.translatePath("special://profile/addon_data/plugin.video.emby/temp/").decode('utf-8')
|
|
|
|
''' Since Emby returns all possible tracks together, sort them.
|
|
IsTextSubtitleStream if true, is available to download from server.
|
|
'''
|
|
|
|
kodi_index = 0
|
|
for stream in streams:
|
|
|
|
if stream['Type'] == "Subtitle" and stream['IsExternal'] and stream['IsTextSubtitleStream']:
|
|
index = stream['Index']
|
|
|
|
if 'Path' in stream and xbmcvfs.exists(self.get_direct_path(stream)):
|
|
url = None if self.method == "DirectPlay" else self.get_direct_path(stream)
|
|
elif 'DeliveryUrl' in stream:
|
|
url = self.server + "/emby" + stream['DeliveryUrl']
|
|
else:
|
|
url = self.server + "/emby/Videos/%s/%s/Subtitles/%s/Stream.%s?api_key=%s" % (item_id, source['Id'], index, stream['Codec'], self.token)
|
|
|
|
log.info("Subtitle url: %s", url)
|
|
if url is None:
|
|
continue
|
|
|
|
if 'Language' in stream:
|
|
filename = "Stream.%s.%s" % (stream['Language'].encode('utf-8'), stream['Codec'])
|
|
try:
|
|
subs.append(self._download_external_subs(url, temp, filename))
|
|
except Exception as error:
|
|
log.warn(error)
|
|
subs.append(url)
|
|
else:
|
|
subs.append(url)
|
|
|
|
# Map external subtitles for player.py
|
|
mapping[kodi_index] = index
|
|
kodi_index += 1
|
|
|
|
window('emby_%s.indexMapping.json' % play_url.encode('utf-8'), value=mapping)
|
|
self.listitem.setSubtitles(subs)
|
|
|
|
return
|
|
|
|
@classmethod
|
|
def _download_external_subs(cls, src, dst, filename):
|
|
|
|
if not xbmcvfs.exists(dst):
|
|
xbmcvfs.mkdir(dst)
|
|
|
|
path = os.path.join(dst, filename)
|
|
|
|
try:
|
|
response = requests.get(src, stream=True)
|
|
response.raise_for_status()
|
|
except Exception as e:
|
|
raise
|
|
else:
|
|
response.encoding = 'utf-8'
|
|
with open(path, 'wb') as f:
|
|
f.write(response.content)
|
|
del response
|
|
|
|
return path
|
|
|
|
def get_audio_subs(self, source):
|
|
|
|
''' For transcoding only
|
|
Present the list of audio/subs to select from, before playback starts.
|
|
Returns part of the url to append.
|
|
'''
|
|
|
|
prefs = ""
|
|
streams = source['MediaStreams']
|
|
|
|
audio_streams = collections.OrderedDict()
|
|
subs_streams = collections.OrderedDict()
|
|
|
|
if streams:
|
|
|
|
''' Since Emby returns all possible tracks together, sort them.
|
|
IsTextSubtitleStream if true, is available to download from server.
|
|
'''
|
|
|
|
for stream in streams:
|
|
index = stream['Index']
|
|
stream_type = stream['Type']
|
|
|
|
if stream_type == "Audio":
|
|
codec = stream['Codec']
|
|
channel = stream.get('ChannelLayout', "")
|
|
|
|
if 'Language' in stream:
|
|
track = "%s - %s - %s %s" % (index, stream['Language'], codec, channel)
|
|
else:
|
|
track = "%s - %s %s" % (index, codec, channel)
|
|
|
|
audio_streams[track] = index
|
|
|
|
elif stream_type == "Subtitle":
|
|
|
|
if 'Language' in stream:
|
|
track = "%s - %s" % (index, stream['Language'])
|
|
else:
|
|
track = "%s - %s" % (index, stream['Codec'])
|
|
|
|
if stream['IsDefault']:
|
|
track = "%s - Default" % track
|
|
if stream['IsForced']:
|
|
track = "%s - Forced" % track
|
|
|
|
subs_streams[track] = index
|
|
|
|
dialog = xbmcgui.Dialog()
|
|
skip_dialog = int(settings('skipDialogTranscode') or 0)
|
|
audio_selected = None
|
|
|
|
if self.info.get('AudioStreamIndex'):
|
|
audio_selected = self.info['AudioStreamIndex']
|
|
|
|
elif skip_dialog in (0, 1):
|
|
if len(audio_streams) > 1:
|
|
selection = list(audio_streams.keys())
|
|
resp = dialog.select(lang(33013), selection)
|
|
audio_selected = audio_streams[selection[resp]] if resp else source['DefaultAudioStreamIndex']
|
|
else: # Only one choice
|
|
audio_selected = audio_streams[next(iter(audio_streams))]
|
|
else:
|
|
audio_selected = source['DefaultAudioStreamIndex']
|
|
|
|
prefs += "&AudioStreamIndex=%s" % audio_selected
|
|
prefs += "&AudioBitrate=384000" if streams[audio_selected].get('Channels', 0) > 2 else "&AudioBitrate=192000"
|
|
|
|
if self.info.get('SubtitleStreamIndex'):
|
|
index = self.info['SubtitleStreamIndex']
|
|
|
|
if index:
|
|
server_settings = self.emby.get_server_transcoding_settings()
|
|
if server_settings['EnableSubtitleExtraction'] and streams[index]['SupportsExternalStream']:
|
|
self._get_subtitles(source, index)
|
|
else:
|
|
prefs += "&SubtitleStreamIndex=%s" % index
|
|
|
|
elif skip_dialog in (0, 2) and len(subs_streams):
|
|
selection = list(['No subtitles']) + list(subs_streams.keys())
|
|
resp = dialog.select(lang(33014), selection)
|
|
if resp:
|
|
index = subs_streams[selection[resp]] if resp > -1 else source.get('DefaultSubtitleStreamIndex')
|
|
if index is not None:
|
|
server_settings = self.emby.get_server_transcoding_settings()
|
|
if server_settings['EnableSubtitleExtraction'] and streams[index]['SupportsExternalStream']:
|
|
self._get_subtitles(source, index)
|
|
else:
|
|
prefs += "&SubtitleStreamIndex=%s" % index
|
|
|
|
return prefs
|
|
|
|
def _get_subtitles(self, source, index):
|
|
|
|
url = [("%s/Videos/%s/%s/Subtitles/%s/Stream.srt"
|
|
% (self.server, self.item['Id'], source['Id'], index))]
|
|
|
|
log.info("Set up subtitles: %s %s", index, url)
|
|
self.listitem.setSubtitles(url)
|
|
|
|
def get_bitrate(self):
|
|
|
|
''' Get the addon video quality
|
|
Max bit rate supported by server: 2147483 (max signed 32bit integer)
|
|
'''
|
|
|
|
bitrate = {
|
|
|
|
'0': 664,
|
|
'1': 996,
|
|
'2': 1320,
|
|
'3': 2000,
|
|
'4': 3200,
|
|
'5': 4700,
|
|
'6': 6200,
|
|
'7': 7700,
|
|
'8': 9200,
|
|
'9': 10700,
|
|
'10': 12200,
|
|
'11': 13700,
|
|
'12': 15200,
|
|
'13': 16700,
|
|
'14': 18200,
|
|
'15': 20000,
|
|
'16': 25000,
|
|
'17': 30000,
|
|
'18': 35000,
|
|
'16': 40000,
|
|
'17': 100000,
|
|
'18': 1000000
|
|
}
|
|
return bitrate.get(settings('videoBitrate'), 2147483)
|
|
|
|
def get_device_profile(self):
|
|
return {
|
|
|
|
"Name": "Kodi",
|
|
"MaxStreamingBitrate": self.get_bitrate() * 1000,
|
|
"MusicStreamingTranscodingBitrate": 1280000,
|
|
"TimelineOffsetSeconds": 5,
|
|
|
|
"TranscodingProfiles": [
|
|
{
|
|
"Container": "mp3",
|
|
"AudioCodec": "mp3",
|
|
"Type": "Audio"
|
|
},
|
|
{
|
|
"Container": "m3u8",
|
|
"AudioCodec": "ac3",
|
|
"VideoCodec": "h264",
|
|
"Type": "Video"
|
|
},
|
|
{
|
|
"Container": "jpeg",
|
|
"Type": "Photo"
|
|
}
|
|
],
|
|
|
|
"DirectPlayProfiles": [
|
|
{
|
|
"Type": "Video"
|
|
},
|
|
{
|
|
"Type": "Audio"
|
|
},
|
|
{
|
|
"Type": "Photo"
|
|
}
|
|
],
|
|
|
|
"ResponseProfiles": [],
|
|
"ContainerProfiles": [],
|
|
"CodecProfiles": [],
|
|
|
|
"SubtitleProfiles": [
|
|
{
|
|
"Format": "srt",
|
|
"Method": "External"
|
|
},
|
|
{
|
|
"Format": "srt",
|
|
"Method": "Embed"
|
|
},
|
|
{
|
|
"Format": "ass",
|
|
"Method": "External"
|
|
},
|
|
{
|
|
"Format": "ass",
|
|
"Method": "Embed"
|
|
},
|
|
{
|
|
"Format": "sub",
|
|
"Method": "Embed"
|
|
},
|
|
{
|
|
"Format": "sub",
|
|
"Method": "External"
|
|
},
|
|
{
|
|
"Format": "ssa",
|
|
"Method": "Embed"
|
|
},
|
|
{
|
|
"Format": "ssa",
|
|
"Method": "External"
|
|
},
|
|
{
|
|
"Format": "smi",
|
|
"Method": "Embed"
|
|
},
|
|
{
|
|
"Format": "smi",
|
|
"Method": "External"
|
|
},
|
|
{
|
|
"Format": "pgssub",
|
|
"Method": "Embed"
|
|
},
|
|
{
|
|
"Format": "pgssub",
|
|
"Method": "External"
|
|
},
|
|
{
|
|
"Format": "dvdsub",
|
|
"Method": "Embed"
|
|
},
|
|
{
|
|
"Format": "dvdsub",
|
|
"Method": "External"
|
|
},
|
|
{
|
|
"Format": "pgs",
|
|
"Method": "Embed"
|
|
},
|
|
{
|
|
"Format": "pgs",
|
|
"Method": "External"
|
|
}
|
|
]
|
|
}
|
|
|
|
def get_resolution(self):
|
|
|
|
window = xbmcgui.Window()
|
|
return window.getWidth(), window.getHeight()
|