PlexKodiConnect/resources/lib/kodimonitor.py

#!/usr/bin/env python
# -*- coding: utf-8 -*-
"""
PKC Kodi Monitoring implementation
"""
from __future__ import absolute_import, division, unicode_literals
from logging import getLogger
from json import loads
import copy

import xbmc
import xbmcgui

from .plex_db import PlexDB
from . import kodi_db
from .downloadutils import DownloadUtils as DU
from . import utils, timing, plex_functions as PF, json_rpc as js
from . import playqueue as PQ, backgroundthread, app, variables as v

LOG = getLogger('PLEX.kodimonitor')

# "Start from beginning", "Play from beginning"
STRINGS = (utils.lang(12021).encode('utf-8'),
           utils.lang(12023).encode('utf-8'))


class MonitorError(Exception):
    """
    Exception we raise for all errors associated with xbmc.Monitor
    """
    pass


class KodiMonitor(xbmc.Monitor):
    """
    PKC implementation of the Kodi Monitor class. Invoke only once.
    """
    def __init__(self):
        self._already_slept = False
        # Info to the currently playing item
        self.playerid = None
        self.playlistid = None
        self.playqueue = None
        for playerid in app.PLAYSTATE.player_states:
            app.PLAYSTATE.player_states[playerid] = copy.deepcopy(app.PLAYSTATE.template)
            app.PLAYSTATE.old_player_states[playerid] = copy.deepcopy(app.PLAYSTATE.template)
        xbmc.Monitor.__init__(self)
        LOG.info("Kodi monitor started.")

    def onScanStarted(self, library):
        """
        Will be called when Kodi starts scanning the library
        """
        LOG.debug("Kodi library scan %s running.", library)

    def onScanFinished(self, library):
        """
        Will be called when Kodi finished scanning the library
        """
        LOG.debug("Kodi library scan %s finished.", library)

    def onSettingsChanged(self):
        """
        Monitor the PKC settings for changes made by the user
        """
        LOG.debug('PKC settings change detected')
        # Assume that the user changed something so we can try to reconnect
        # app.APP.suspend = False
        # app.APP.resume_threads(block=False)

    def onNotification(self, sender, method, data):
        """
        Called when a bunch of different stuff happens on the Kodi side
        """
        if data:
            data = loads(data, 'utf-8')
            LOG.debug("Method: %s Data: %s", method, data)

        if method == "Player.OnPlay":
            with app.APP.lock_playqueues:
                self.on_play(data)
        elif method == "Player.OnStop":
            if data.get('end'):
                with app.APP.lock_playqueues:
                    _playback_cleanup(ended=True)
            else:
                with app.APP.lock_playqueues:
                    _playback_cleanup()
        elif method == 'Playlist.OnAdd':
            self._playlist_onadd(data)
        elif method == 'Playlist.OnRemove':
            self._playlist_onremove(data)
        elif method == 'Playlist.OnClear':
            self._playlist_onclear(data)
        elif method == "VideoLibrary.OnUpdate":
            # Manually marking as watched/unwatched
            playcount = data.get('playcount')
            item = data.get('item')
            if playcount is None or item is None:
                return
            try:
                kodi_id = item['id']
                kodi_type = item['type']
            except (KeyError, TypeError):
                LOG.info("Item is invalid for playstate update.")
                return
            # Send notification to the server.
            with PlexDB() as plexdb:
                db_item = plexdb.item_by_kodi_id(kodi_id, kodi_type)
            if not db_item:
                LOG.error("Could not find plex_id in plex database for a "
                          "video library update")
            else:
                # notify the server
                if playcount > 0:
                    PF.scrobble(db_item['plex_id'], 'watched')
                else:
                    PF.scrobble(db_item['plex_id'], 'unwatched')
        elif method == "VideoLibrary.OnRemove":
            pass
        elif method == "System.OnSleep":
            # Connection is going to sleep
            LOG.info("Marking the server as offline. SystemOnSleep activated.")
        elif method == "System.OnWake":
            # Allow network to wake up
            self.waitForAbort(10)
            app.CONN.online = False
        elif method == "GUI.OnScreensaverDeactivated":
            if utils.settings('dbSyncScreensaver') == "true":
                self.waitForAbort(5)
                app.SYNC.run_lib_scan = 'full'
        elif method == "System.OnQuit":
            LOG.info('Kodi OnQuit detected - shutting down')
            app.APP.stop_pkc = True

    def _playlist_onadd(self, data):
        '''
        Called when a new item is added to a Kodi playqueue
        '''
        if 'item' in data and data['item'].get('type') == v.KODI_TYPE_SHOW:
            # Hitting the "browse" button on tv show info dialog
            # Hence show the tv show directly
            xbmc.executebuiltin("Dialog.Close(all, true)")
            js.activate_window('videos',
                               'videodb://tvshows/titles/%s/' % data['item']['id'])
            return

        if data['position'] == 0:
            self.playlistid = data['playlistid']
        if app.PLAYSTATE.playlist_start_pos == data['position']:
            LOG.debug('Playlist ready')
            app.PLAYSTATE.playlist_ready = True
            app.PLAYSTATE.playlist_start_pos = None

    def _playlist_onremove(self, data):
        """
        Called if an item is removed from a Kodi playlist. Example data dict:
        {
            u'playlistid': 1,
            u'position': 0
        }
        """
        pass

    def _playlist_onclear(self, data):
        """
        Called if a Kodi playlist is cleared. Example data dict:
        {
            u'playlistid': 1,
        }
        Let's NOT use this as Kodi's responses when e.g. playing an entire
        folder are NOT threadsafe: Playlist.OnAdd might be added first, then
        Playlist.OnClear might be received LATER
        """
        if self.playlistid == data['playlistid']:
            LOG.debug('Resetting autoplay')
            app.PLAYSTATE.autoplay = False
        # playqueue = PQ.PLAYQUEUES[data['playlistid']]
        # if not playqueue.is_pkc_clear():
        #     playqueue.pkc_edit = True
        #     playqueue.clear(kodi=False)
        # else:
        #     LOG.debug('Detected PKC clear - ignoring')

    @staticmethod
    def _get_ids(kodi_id, kodi_type, path):
        """
        Returns the tuple (plex_id, plex_type) or (None, None)
        """
        # No Kodi id returned by Kodi, even if there is one. Ex: Widgets
        plex_id = None
        plex_type = None
        # If using direct paths and starting playback from a widget
        if not kodi_id and kodi_type and path:
            kodi_id, _ = kodi_db.kodiid_from_filename(path, kodi_type)
        if kodi_id:
            with PlexDB() as plexdb:
                db_item = plexdb.item_by_kodi_id(kodi_id, kodi_type)
            if db_item:
                plex_id = db_item['plex_id']
                plex_type = db_item['plex_type']
        return plex_id, plex_type

    def _json_item(self, playerid):
        """
        Uses JSON RPC to get the playing item's info and returns the tuple
            kodi_id, kodi_type, path
        or None each time if not found.
        """
        if not self._already_slept:
            # SLEEP before calling this for the first time just after playback
            # start as Kodi updates this info very late!! Might get previous
            # element otherwise
            self._already_slept = True
            self.waitForAbort(1)
        try:
            json_item = js.get_item(playerid)
        except KeyError:
            LOG.debug('No playing item returned by Kodi')
            return None, None, None
        LOG.debug('Kodi playing item properties: %s', json_item)
        return (json_item.get('id'),
                json_item.get('type'),
                json_item.get('file'))

    def _get_playerid(self, data):
        """
        Sets self.playerid with an int 0, 1 [or 2] or raises MonitorError
            0: usually video
            1: usually audio
        """
        try:
            self.playerid = data['player']['playerid']
        except (TypeError, KeyError):
            LOG.info('Aborting playback report - data invalid for updates: %s',
                     data)
            raise MonitorError()
        if self.playerid == -1:
            # Kodi might return -1 for "last player"
            try:
                self.playerid = js.get_player_ids()[0]
            except IndexError:
                LOG.error('Coud not get playerid for data: %s', data)
                raise MonitorError()

    def _check_playing_item(self, data):
        """
        Returns a PF.PlaylistItem() for the currently playing item
        Raises MonitorError or IndexError if we need to init the PKC playqueue
        """
        info = js.get_player_props(self.playerid)
        LOG.debug('Current info for player %s: %s', self.playerid, info)
        position = info['position'] if info['position'] != -1 else 0
        kodi_playlist = js.playlist_get_items(self.playerid)
        LOG.debug('Current Kodi playlist: %s', kodi_playlist)
        playlistitem = PQ.PlaylistItem(kodi_item=kodi_playlist[position])
        if isinstance(self.playqueue.items[0], PQ.PlaylistItemDummy):
            # This dummy item will be deleted by webservice soon - it won't
            # play
            LOG.debug('Dummy item detected')
            position = 1
        elif playlistitem != self.playqueue.items[position]:
            LOG.debug('Different playqueue items: %s vs. %s ',
                      playlistitem, self.playqueue.items[position])
            raise MonitorError()
        # Return the PKC playqueue item - contains more info
        return self.playqueue.items[position]

    def _load_playerstate(self, item):
        """
        Pass in a PF.PlaylistItem(). Will then set the currently playing
        state with app.PLAYSTATE.player_states[self.playerid]
        """
        if self.playqueue.id:
            container_key = '/playQueues/%s' % self.playqueue.id
        else:
            container_key = '/library/metadata/%s' % item.plex_id
        status = app.PLAYSTATE.player_states[self.playerid]
        # Remember that this player has been active
        app.PLAYSTATE.active_players.add(self.playerid)
        status.update(js.get_player_props(self.playerid))
        status['container_key'] = container_key
        status['file'] = item.file
        status['kodi_id'] = item.kodi_id
        status['kodi_type'] = item.kodi_type
        status['plex_id'] = item.plex_id
        status['plex_type'] = item.plex_type
        status['playmethod'] = item.playmethod
        status['playcount'] = item.playcount
        LOG.debug('Set player state for player %s: %s', self.playerid, status)

    def on_play(self, data):
        """
        Called whenever playback is started. Example data:
        {
            u'item': {u'type': u'movie', u'title': u''},
            u'player': {u'playerid': 1, u'speed': 1}
        }
        Unfortunately when using Widgets, Kodi doesn't tell us shit
        """
        # Some init
        self._already_slept = False
        self.playerid = None
        # Get the type of media we're playing
        try:
            self._get_playerid(data)
        except MonitorError:
            return
        self.playqueue = PQ.PLAYQUEUES[self.playerid]
        LOG.debug('Current PKC playqueue: %s', self.playqueue)
        item = None
        try:
            item = self._check_playing_item(data)
        except (MonitorError, IndexError):
            LOG.debug('Detected that we need to initialize the PKC playqueue')

        if not item:
            # Initialize the PKC playqueue
            # Yet TODO
            LOG.debug('Need to initialize Plex and PKC playqueue')
            if not kodi_id or not kodi_type:
                kodi_id, kodi_type, path = self._json_item(playerid)
            plex_id, plex_type = self._get_ids(kodi_id, kodi_type, path)
            if not plex_id:
                LOG.debug('No Plex id obtained - aborting playback report')
                app.PLAYSTATE.player_states[playerid] = copy.deepcopy(app.PLAYSTATE.template)
                return
            playlistitem = PQ.PlaylistItem(plex_id=plex_id,
                                           grab_xml=True)
            playlistitem.file = path
            self.playqueue.init(playlistitem)
            # Set the Plex container key (e.g. using the Plex playqueue)
            container_key = None
            if info['playlistid'] != -1:
                # -1 is Kodi's answer if there is no playlist
                container_key = PQ.PLAYQUEUES[playerid].id
            if container_key is not None:
                container_key = '/playQueues/%s' % container_key
            elif plex_id is not None:
                container_key = '/library/metadata/%s' % plex_id
        self._load_playerstate(item)


def _playback_cleanup(ended=False):
    """
    PKC cleanup after playback ends/is stopped. Pass ended=True if Kodi
    completely finished playing an item (because we will get and use wrong
    timing data otherwise)
    """
    LOG.debug('playback_cleanup called. Active players: %s',
              app.PLAYSTATE.active_players)
    # We might have saved a transient token from a user flinging media via
    # Companion (if we could not use the playqueue to store the token)
    app.CONN.plex_transient_token = None
    LOG.debug('Playstate is: %s', app.PLAYSTATE.player_states)
    for playerid in app.PLAYSTATE.active_players:
        status = app.PLAYSTATE.player_states[playerid]
        # Remember the last played item later
        app.PLAYSTATE.old_player_states[playerid] = copy.deepcopy(status)
        # Stop transcoding
        if status['playmethod'] == 'Transcode':
            LOG.debug('Tell the PMS to stop transcoding')
            DU().downloadUrl(
                '{server}/video/:/transcode/universal/stop',
                parameters={'session': v.PKC_MACHINE_IDENTIFIER})
        if status['plex_type'] in v.PLEX_VIDEOTYPES:
            # Bookmarks are not be pickup up correctly, so let's do them
            # manually. Applies to addon paths, but direct paths might have
            # started playback via PMS
            _record_playstate(status, ended)
        # Reset the player's status
        app.PLAYSTATE.player_states[playerid] = copy.deepcopy(app.PLAYSTATE.template)
    # As all playback has halted, reset the players that have been active
    app.PLAYSTATE.active_players = set()
    LOG.info('Finished PKC playback cleanup')


def _record_playstate(status, ended):
    if not status['plex_id']:
        LOG.debug('No Plex id found to record playstate for status %s', status)
        return
    if status['plex_type'] not in v.PLEX_VIDEOTYPES:
        LOG.debug('Not messing with non-video entries')
        return
    with PlexDB() as plexdb:
        db_item = plexdb.item_by_id(status['plex_id'], status['plex_type'])
    if not db_item:
        # Item not (yet) in Kodi library
        LOG.debug('No playstate update due to Plex id not found: %s', status)
        return
    totaltime = float(timing.kodi_time_to_millis(status['totaltime'])) / 1000
    if ended:
        progress = 0.99
        time = v.IGNORE_SECONDS_AT_START + 1
    else:
        time = float(timing.kodi_time_to_millis(status['time'])) / 1000
        try:
            progress = time / totaltime
        except ZeroDivisionError:
            progress = 0.0
        LOG.debug('Playback progress %s (%s of %s seconds)',
                  progress, time, totaltime)
    playcount = status['playcount']
    last_played = timing.kodi_now()
    if playcount is None:
        LOG.debug('playcount not found, looking it up in the Kodi DB')
        with kodi_db.KodiVideoDB() as kodidb:
            playcount = kodidb.get_playcount(db_item['kodi_fileid'])
        playcount = 0 if playcount is None else playcount
    if time < v.IGNORE_SECONDS_AT_START:
        LOG.debug('Ignoring playback less than %s seconds',
                  v.IGNORE_SECONDS_AT_START)
        # Annoying Plex bug - it'll reset an already watched video to unwatched
        playcount = None
        last_played = None
        time = 0
    elif progress >= v.MARK_PLAYED_AT:
        LOG.debug('Recording entirely played video since progress > %s',
                  v.MARK_PLAYED_AT)
        playcount += 1
        time = 0
    with kodi_db.KodiVideoDB() as kodidb:
        kodidb.set_resume(db_item['kodi_fileid'],
                          time,
                          totaltime,
                          playcount,
                          last_played)
        if 'kodi_fileid_2' in db_item and db_item['kodi_fileid_2']:
            # Dirty hack for our episodes
            kodidb.set_resume(db_item['kodi_fileid_2'],
                              time,
                              totaltime,
                              playcount,
                              last_played)
    # We might need to reconsider cleaning the file/path table in the future
    # _clean_file_table()
    # Update the current view to show e.g. an up-to-date progress bar and use
    # the latest resume point info
    if xbmc.getCondVisibility('Container.Content(musicvideos)'):
        # Prevent cursor from moving
        xbmc.executebuiltin('Container.Refresh')
    else:
        # Update widgets
        xbmc.executebuiltin('UpdateLibrary(video)')
        if xbmc.getCondVisibility('Window.IsMedia'):
            xbmc.executebuiltin('Container.Refresh')
    # Hack to force "in progress" widget to appear if it wasn't visible before
    if (app.APP.force_reload_skin and
            xbmc.getCondVisibility('Window.IsVisible(Home.xml)')):
        LOG.debug('Refreshing skin to update widgets')
        xbmc.executebuiltin('ReloadSkin()')


def _clean_file_table():
    """
    If we associate a playing video e.g. pointing to plugin://... to an existing
    Kodi library item, Kodi will add an additional entry for this (additional)
    path plugin:// in the file table. This leads to all sorts of wierd behavior.
    This function tries for at most 5 seconds to clean the file table.
    """
    LOG.debug('Start cleaning Kodi files table')
    # app.APP.monitor.waitForAbort(1)
    try:
        with kodi_db.KodiVideoDB() as kodidb:
            file_ids = list(kodidb.obsolete_file_ids())
            LOG.debug('Obsolete kodi file_ids: %s', file_ids)
            for file_id in file_ids:
                kodidb.remove_file(file_id)
    except utils.OperationalError:
        LOG.debug('Database was locked, unable to clean file table')
    else:
        LOG.debug('Done cleaning up Kodi file table')


class ContextMonitor(backgroundthread.KillableThread):
    """
    Detect the resume dialog for widgets. Could also be used to detect
    external players (see Emby implementation)

    Let's not register this thread because it won't quit due to
    xbmc.getCondVisibility
    It should still exit at some point due to xbmc.abortRequested
    """
    def run(self):
        LOG.info("----===## Starting ContextMonitor ##===----")
        # app.APP.register_thread(self)
        try:
            self._run()
        finally:
            # app.APP.deregister_thread(self)
            LOG.info("##===---- ContextMonitor Stopped ----===##")

    def _run(self):
        while not self.isCanceled():
            # The following function will block if called while PKC should
            # exit!
            if xbmc.getCondVisibility('Window.IsVisible(DialogContextMenu.xml)'):
                if xbmc.getInfoLabel('Control.GetLabel(1002)') in STRINGS:
                    # Remember that the item IS indeed resumable
                    control = int(xbmcgui.Window(10106).getFocusId())
                    app.PLAYSTATE.resume_playback = True if control == 1001 else False
                else:
                    # Different context menu is displayed
                    app.PLAYSTATE.resume_playback = None
            xbmc.sleep(100)