gpodder/share/gpodder/extensions/mpris-listener.py

339 lines
14 KiB
Python

# -*- coding: utf-8 -*-
#
# gPodder extension for listening to notifications from MPRIS-capable
# players and translating them to gPodder's Media Player D-Bus API
#
# Copyright (c) 2013-2014 Dov Feldstern <dovdevel@gmail.com>
#
# This program 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.
#
# This program 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 this program. If not, see <http://www.gnu.org/licenses/>.
import collections
import logging
import time
import urllib.error
import urllib.parse
import urllib.request
import dbus
import dbus.service
import gpodder
logger = logging.getLogger(__name__)
_ = gpodder.gettext
__title__ = _('MPRIS Listener')
__description__ = _('Convert MPRIS notifications to gPodder Media Player D-Bus API')
__authors__ = 'Dov Feldstern <dovdevel@gmail.com>'
__doc__ = 'https://gpodder.github.io/docs/extensions/mprislistener.html'
__category__ = 'desktop-integration'
__only_for__ = 'freedesktop'
USECS_IN_SEC = 1000000
TrackInfo = collections.namedtuple('TrackInfo',
['uri', 'length', 'status', 'pos', 'rate'])
def subsecond_difference(usec1, usec2):
return usec1 is not None and usec2 is not None and abs(usec1 - usec2) < USECS_IN_SEC
class CurrentTrackTracker(object):
'''An instance of this class is responsible for tracking the state of the
currently playing track -- it's playback status, playing position, etc.
'''
def __init__(self, notifier):
self.uri = None
self.length = None
self.pos = None
self.rate = None
self.status = None
self._notifier = notifier
self._prev_notif = ()
def _calc_update(self):
now = time.time()
logger.debug('CurrentTrackTracker: calculating at %d (status: %r)',
now, self.status)
try:
if self.status != 'Playing':
logger.debug('CurrentTrackTracker: not currently playing, no change')
return
if self.pos is None or self.rate is None:
logger.debug('CurrentTrackTracker: unknown pos/rate, no change')
return
logger.debug('CurrentTrackTracker: %f @%f (diff: %f)',
self.pos, self.rate, now - self._last_time)
self.pos = self.pos + self.rate * (now - self._last_time) * USECS_IN_SEC
finally:
self._last_time = now
def update_needed(self, current, updated):
for field in updated:
if field == 'pos':
if not subsecond_difference(updated['pos'], current['pos']):
return True
elif updated[field] != current[field]:
return True
# no unequal field was found, no new info here!
return False
def update(self, **kwargs):
# check if there is any new info here -- if not, no need to update!
cur = self.getinfo()._asdict()
if not self.update_needed(cur, kwargs):
return
# there *is* new info, go ahead and update...
uri = kwargs.pop('uri', None)
if uri is not None:
length = kwargs.pop('length') # don't know how to handle uri with no length
if uri != cur['uri']:
# if this is a new uri, and the previous state was 'Playing',
# notify that the previous track has stopped before updating to
# the new track.
if cur['status'] == 'Playing':
logger.debug('notify Stopped: new uri: old %s new %s',
cur['uri'], uri)
self.notify_stop()
self.uri = uri
self.length = float(length)
if 'pos' in kwargs:
# If the position is being updated, and the current status was Playing
# If the status *is* playing, and *was* playing, but the position
# has changed discontinuously, notify a stop for the old position
if (cur['status'] == 'Playing' and
('status' not in kwargs or kwargs['status'] == 'Playing') and not
subsecond_difference(cur['pos'], kwargs['pos'])):
logger.debug('notify Stopped: playback discontinuity:' +
'calc: %r observed: %r', cur['pos'], kwargs['pos'])
self.notify_stop()
if ((kwargs['pos']) <= 0 and
self.pos is not None and
self.length is not None and
(self.length - USECS_IN_SEC) < self.pos and
self.pos < (self.length + 2 * USECS_IN_SEC)):
logger.debug('pos=0 end of stream (calculated pos: %f/%f [%f])',
self.pos / USECS_IN_SEC, self.length / USECS_IN_SEC,
(self.pos / USECS_IN_SEC) - (self.length / USECS_IN_SEC))
self.pos = self.length
kwargs.pop('pos') # remove 'pos' even though we're not using it
else:
if self.pos is not None and self.length is not None:
logger.debug("%r %r", self.pos, self.length)
logger.debug('pos=0 not end of stream (calculated pos: %f/%f [%f])',
self.pos / USECS_IN_SEC, self.length / USECS_IN_SEC,
(self.pos / USECS_IN_SEC) - (self.length / USECS_IN_SEC))
newpos = kwargs.pop('pos')
self.pos = newpos if newpos >= 0 else 0
if 'status' in kwargs:
self.status = kwargs.pop('status')
if 'rate' in kwargs:
self.rate = kwargs.pop('rate')
if kwargs:
logger.error('unexpected update fields %r', kwargs)
# notify about the current state
if self.status == 'Playing':
self.notify_playing()
else:
logger.debug('notify Stopped: status %r', self.status)
self.notify_stop()
def getinfo(self):
self._calc_update()
return TrackInfo(self.uri, self.length, self.status, self.pos, self.rate)
def notify_stop(self):
self.notify('Stopped')
def notify_playing(self):
self.notify('Playing')
def notify(self, status):
if (self.uri is None or
self.pos is None or
self.status is None or
self.length is None or
self.length <= 0):
return
pos = self.pos // USECS_IN_SEC
parsed_url = urllib.parse.urlparse(self.uri)
if (not parsed_url.scheme) or parsed_url.scheme == 'file':
file_uri = urllib.request.url2pathname(urllib.parse.urlparse(self.uri).path).encode('utf-8')
else:
file_uri = self.uri
total_time = self.length // USECS_IN_SEC
if status == 'Stopped':
end_position = pos
start_position = self._notifier.start_position
if self._prev_notif != (start_position, end_position, total_time, file_uri):
self._notifier.PlaybackStopped(start_position, end_position,
total_time, file_uri)
self._prev_notif = (start_position, end_position, total_time, file_uri)
elif status == 'Playing':
start_position = pos
if self._prev_notif != (start_position, file_uri):
self._notifier.PlaybackStarted(start_position, file_uri)
self._prev_notif = (start_position, file_uri)
self._notifier.start_position = start_position
logger.info('CurrentTrackTracker: %s: %r %s', status, self, file_uri)
def __repr__(self):
return '%s: %s at %d/%d (@%f)' % (
self.uri or 'None',
self.status or 'None',
(self.pos or 0) // USECS_IN_SEC,
(self.length or 0) // USECS_IN_SEC,
self.rate or 0)
class MPRISDBusReceiver(object):
INTERFACE_PROPS = 'org.freedesktop.DBus.Properties'
SIGNAL_PROP_CHANGE = 'PropertiesChanged'
PATH_MPRIS = '/org/mpris/MediaPlayer2'
INTERFACE_MPRIS = 'org.mpris.MediaPlayer2.Player'
SIGNAL_SEEKED = 'Seeked'
OTHER_MPRIS_INTERFACES = ['org.mpris.MediaPlayer2',
'org.mpris.MediaPlayer2.TrackList',
'org.mpris.MediaPlayer2.Playlists']
def __init__(self, bus, notifier):
self.bus = bus
self.cur = CurrentTrackTracker(notifier)
self.bus.add_signal_receiver(self.on_prop_change,
self.SIGNAL_PROP_CHANGE,
self.INTERFACE_PROPS,
None,
self.PATH_MPRIS,
sender_keyword='sender')
self.bus.add_signal_receiver(self.on_seeked,
self.SIGNAL_SEEKED,
self.INTERFACE_MPRIS,
None,
None)
def stop_receiving(self):
self.bus.remove_signal_receiver(self.on_prop_change,
self.SIGNAL_PROP_CHANGE,
self.INTERFACE_PROPS,
None,
self.PATH_MPRIS)
self.bus.remove_signal_receiver(self.on_seeked,
self.SIGNAL_SEEKED,
self.INTERFACE_MPRIS,
None,
None)
def on_prop_change(self, interface_name, changed_properties,
invalidated_properties, path=None, sender=None):
if interface_name != self.INTERFACE_MPRIS:
if interface_name not in self.OTHER_MPRIS_INTERFACES:
logger.warning('unexpected interface: %s, props=%r', interface_name, list(changed_properties.keys()))
return
if sender is None:
logger.warning('No sender associated to D-Bus signal, please report a bug')
return
collected_info = {}
logger.debug("on_prop_change %r", changed_properties.keys())
if 'PlaybackStatus' in changed_properties:
collected_info['status'] = str(changed_properties['PlaybackStatus'])
if 'Metadata' in changed_properties:
logger.debug("Metadata %r", changed_properties['Metadata'].keys())
# on stop there is no xesam:url
if 'xesam:url' in changed_properties['Metadata']:
collected_info['uri'] = changed_properties['Metadata']['xesam:url']
collected_info['length'] = changed_properties['Metadata'].get('mpris:length', 0.0)
if 'Rate' in changed_properties:
collected_info['rate'] = changed_properties['Rate']
# Fix #788 pos=0 when Stopped resulting in not saving position on VLC quit
if changed_properties.get('PlaybackStatus') != 'Stopped':
try:
collected_info['pos'] = self.query_property(sender, 'Position')
except dbus.exceptions.DBusException:
pass
if 'status' not in collected_info:
try:
collected_info['status'] = str(self.query_property(
sender, 'PlaybackStatus'))
except dbus.exceptions.DBusException:
pass
logger.debug('collected info: %r', collected_info)
self.cur.update(**collected_info)
def on_seeked(self, position):
logger.debug('seeked to pos: %f', position)
self.cur.update(pos=position)
def query_property(self, sender, prop):
proxy = self.bus.get_object(sender, self.PATH_MPRIS)
props = dbus.Interface(proxy, self.INTERFACE_PROPS)
return props.Get(self.INTERFACE_MPRIS, prop)
class gPodderNotifier(dbus.service.Object):
def __init__(self, bus, path):
dbus.service.Object.__init__(self, bus, path)
self.start_position = 0
@dbus.service.signal(dbus_interface='org.gpodder.player', signature='us')
def PlaybackStarted(self, start_position, file_uri):
logger.info('PlaybackStarted: %s: %d', file_uri, start_position)
@dbus.service.signal(dbus_interface='org.gpodder.player', signature='uuus')
def PlaybackStopped(self, start_position, end_position, total_time, file_uri):
logger.info('PlaybackStopped: %s: %d--%d/%d',
file_uri, start_position, end_position, total_time)
# Finally, this is the extension, which just pulls this all together
class gPodderExtension:
def __init__(self, container):
self.container = container
self.path = '/org/gpodder/player/notifier'
self.notifier = None
self.rcvr = None
def on_load(self):
if gpodder.dbus_session_bus is None:
logger.debug("dbus session bus not available, not loading")
else:
self.session_bus = gpodder.dbus_session_bus
self.notifier = gPodderNotifier(self.session_bus, self.path)
self.rcvr = MPRISDBusReceiver(self.session_bus, self.notifier)
def on_unload(self):
if self.notifier is not None:
self.notifier.remove_from_connection(self.session_bus, self.path)
if self.rcvr is not None:
self.rcvr.stop_receiving()