]> kaliko git repositories - mpd-sima.git/blobdiff - sima/core.py
Fixed bug in uniq plugin
[mpd-sima.git] / sima / core.py
index 9c96cb521ae0ab82b0535ab8d1ea0be91cf2b759..60f6b2b08a0f85ba7854d204e4127ba5723edd9b 100644 (file)
-#!/usr/bin/env python3
 # -*- coding: utf-8 -*-
+# Copyright (c) 2009, 2010, 2011, 2013, 2014, 2015 Jack Kaliko <kaliko@azylum.org>
+#
+#  This file is part of sima
+#
+#  sima is free software: you can redistribute it and/or modify
+#  it under the terms of the GNU General Public License as published by
+#  the Free Software Foundation, either version 3 of the License, or
+#  (at your option) any later version.
+#
+#  sima is distributed in the hope that it will be useful,
+#  but WITHOUT ANY WARRANTY; without even the implied warranty of
+#  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+#  GNU General Public License for more details.
+#
+#  You should have received a copy of the GNU General Public License
+#  along with sima.  If not, see <http://www.gnu.org/licenses/>.
+#
+#
+"""Core Object dealing with plugins and player client
+"""
+
+import time
+
+from collections import deque
+from logging import getLogger
 
 from .client import PlayerClient
+from .client import PlayerError, PlayerUnHandledError
+from .lib.simadb import SimaDB
+from .lib.daemon import Daemon
+from .utils.utils import SigHup
 
-class Sima(object):
+class Sima(Daemon):
     """Main class, plugin and player management
     """
 
-    def __init__(self):
-        self.plugins = list()
-        self.player = None
-        self.connect_player()
+    def __init__(self, conf):
+        ## Set daemon
+        Daemon.__init__(self, conf.get('daemon', 'pidfile'))
+        self.enabled = True
+        self.config = conf
+        self.sdb = SimaDB(db_path=conf.get('sima', 'db_file'))
+        PlayerClient.database = self.sdb
+        self.log = getLogger('sima')
+        self._plugins = list()
+        self._core_plugins = list()
+        self.player = self.__get_player()  # Player client
+        self.short_history = deque(maxlen=60)
+
+    def __get_player(self):
+        """Instanciate the player"""
+        host = self.config.get('MPD', 'host')
+        port = self.config.get('MPD', 'port')
+        pswd = self.config.get('MPD', 'password', fallback=None)
+        return PlayerClient(host, port, pswd)
+
+    def add_history(self):
+        """Handle local, in memory, short history"""
+        self.short_history.appendleft(self.player.current)
 
     def register_plugin(self, plugin_class):
-        self.plugins.append(plugin_class(self))
+        """Registers plugin in Sima instance..."""
+        plgn = plugin_class(self)
+        prio = int(plgn.priority)
+        self._plugins.append((prio, plgn))
+
+    def register_core_plugin(self, plugin_class):
+        """Registers core plugins"""
+        plgn = plugin_class(self)
+        prio = int(plgn.priority)
+        self._core_plugins.append((prio, plgn))
 
     def foreach_plugin(self, method, *args, **kwds):
+        """Plugin's callbacks dispatcher"""
+        for plugin in self.core_plugins:
+            getattr(plugin, method)(*args, **kwds)
         for plugin in self.plugins:
+            #self.log.debug('dispatching {0} to {1}'.format(method, plugin))
             getattr(plugin, method)(*args, **kwds)
 
-    def connect_player(self):
-        """Instanciate player client and connect it
+    @property
+    def core_plugins(self):
+        return [plugin[1] for plugin in
+                sorted(self._core_plugins, key=lambda pl: pl[0], reverse=True)]
+
+    @property
+    def plugins(self):
+        return [plugin[1] for plugin in sorted(self._plugins, key=lambda pl: pl[0], reverse=True)]
+
+    def need_tracks(self):
+        """Is the player in need for tracks"""
+        if not self.enabled:
+            self.log.debug('Queueing disabled!')
+            return False
+        queue = self.player.queue
+        queue_trigger = self.config.getint('sima', 'queue_length')
+        self.log.debug('Currently {0} track(s) ahead. (target {1})'.format(
+                       len(queue), queue_trigger))
+        if len(queue) < queue_trigger:
+            return True
+        return False
+
+    def queue(self):
+        to_add = list()
+        for plugin in self.plugins:
+            self.log.info('running {}'.format(plugin))
+            pl_candidates = getattr(plugin, 'callback_need_track')()
+            if pl_candidates:
+                to_add.extend(pl_candidates)
+            if to_add:
+                break
+        for track in to_add:
+            self.player.add(track)
+
+    def reconnect_player(self):
+        """Trying to reconnect cycling through longer timeout
+        cycle : 5s 10s 1m 5m 20m 1h
         """
-        self.player = PlayerClient()  # Player client
-        self.player.connect()
+        sleepfor = [5, 10, 60, 300, 1200, 3600]
+        while True:
+            tmp = sleepfor.pop(0)
+            sleepfor.append(tmp)
+            self.log.info('Trying to reconnect in {:>4d} seconds'.format(tmp))
+            time.sleep(tmp)
+            try:
+                self.player.connect()
+            except PlayerError as err:
+                self.log.debug(err)
+                continue
+            except PlayerUnHandledError as err:
+                #TODO: unhandled Player exceptions
+                self.log.warning('Unhandled player exception: %s' % err)
+            self.log.info('Got reconnected')
+            break
+        self.foreach_plugin('start')
+
+    def hup_handler(self, signum, frame):
+        self.log.warning('Caught a sighup!')
+        # Cleaning pending command
+        self.player.clean()
+        self.foreach_plugin('shutdown')
+        self.player.disconnect()
+        raise SigHup('SIGHUP caught!')
 
     def shutdown(self):
         """General shutdown method
         """
-        self.player.disconnect()
+        self.log.warning('Starting shutdown.')
+        # Cleaning pending command
+        self.player.clean()
         self.foreach_plugin('shutdown')
+        self.player.disconnect()
+
+        self.log.info('The way is shut, it was made by those who are dead. '
+                      'And the dead keep it…')
+        self.log.info('bye...')
 
     def run(self):
-        """Dispatching callbacks to plugins
         """
-        print(self.player.status())
+        """
+        try:
+            self.log.info('Connecting MPD: {0}:{1}'.format(*self.player._mpd))
+            self.player.connect()
+            self.foreach_plugin('start')
+        except (PlayerError, PlayerUnHandledError) as err:
+            self.log.warning('Player: {}'.format(err))
+            self.reconnect_player()
         while 42:
-            # hanging here untill a monitored event is raised in the player
-            changed = self.player.monitor()
-            print(changed)
-            print(self.player.current)
-            if 'playlist' in changed:
-                self.foreach_plugin('callback_playlist')
-            if 'player' in changed:
-                pass
-
+            try:
+                self.loop()
+            except PlayerUnHandledError as err:
+                #TODO: unhandled Player exceptions
+                self.log.warning('Unhandled player exception: {}'.format(err))
+                del self.player
+                self.player = PlayerClient()
+                time.sleep(10)
+            except PlayerError as err:
+                self.log.warning('Player error: %s' % err)
+                self.reconnect_player()
+                del self.changed
 
+    def loop(self):
+        """Dispatching callbacks to plugins
+        """
+        # hanging here untill a monitored event is raised in the player
+        if getattr(self, 'changed', False): # first iteration exception
+            self.changed = self.player.monitor()
+        else:  # first iteration goes through else
+            self.changed = ['playlist', 'player', 'skipped']
+        self.log.debug('changed: {}'.format(', '.join(self.changed)))
+        if 'playlist' in self.changed:
+            self.foreach_plugin('callback_playlist')
+        if ('player' in self.changed
+            or 'options' in self.changed):
+            self.foreach_plugin('callback_player')
+        if 'database' in self.changed:
+            self.foreach_plugin('callback_player_database')
+        if 'skipped' in self.changed:
+            if self.player.state == 'play':
+                self.log.info('Playing: {}'.format(self.player.current))
+                self.add_history()
+                self.foreach_plugin('callback_next_song')
+        if self.need_tracks():
+            self.queue()
 
 # VIM MODLINE
 # vim: ai ts=4 sw=4 sts=4 expandtab