diff libervia/backend/plugins/plugin_misc_radiocol.py @ 4071:4b842c1fb686

refactoring: renamed `sat` package to `libervia.backend`
author Goffi <goffi@goffi.org>
date Fri, 02 Jun 2023 11:49:51 +0200
parents sat/plugins/plugin_misc_radiocol.py@524856bd7b19
children 0d7bb4df2343
line wrap: on
line diff
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/libervia/backend/plugins/plugin_misc_radiocol.py	Fri Jun 02 11:49:51 2023 +0200
@@ -0,0 +1,370 @@
+#!/usr/bin/env python3
+
+
+# SAT plugin for managing Radiocol
+# Copyright (C) 2009-2021 Jérôme Poisson (goffi@goffi.org)
+
+# This program is free software: you can redistribute it and/or modify
+# it under the terms of the GNU Affero 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 Affero General Public License for more details.
+
+# You should have received a copy of the GNU Affero General Public License
+# along with this program.  If not, see <http://www.gnu.org/licenses/>.
+
+from libervia.backend.core.i18n import _, D_
+from libervia.backend.core.constants import Const as C
+from libervia.backend.core.log import getLogger
+
+log = getLogger(__name__)
+from twisted.words.xish import domish
+from twisted.internet import reactor
+from twisted.words.protocols.jabber import jid
+from twisted.internet import defer
+from libervia.backend.core import exceptions
+import os.path
+import copy
+import time
+from os import unlink
+
+try:
+    from mutagen.oggvorbis import OggVorbis, OggVorbisHeaderError
+    from mutagen.mp3 import MP3, HeaderNotFoundError
+    from mutagen.easyid3 import EasyID3
+    from mutagen.id3 import ID3NoHeaderError
+except ImportError:
+    raise exceptions.MissingModule(
+        "Missing module Mutagen, please download/install from https://bitbucket.org/lazka/mutagen"
+    )
+
+
+NC_RADIOCOL = "http://www.goffi.org/protocol/radiocol"
+RADIOC_TAG = "radiocol"
+
+PLUGIN_INFO = {
+    C.PI_NAME: "Radio collective plugin",
+    C.PI_IMPORT_NAME: "Radiocol",
+    C.PI_TYPE: "Exp",
+    C.PI_PROTOCOLS: [],
+    C.PI_DEPENDENCIES: ["XEP-0045", "XEP-0249", "ROOM-GAME"],
+    C.PI_MAIN: "Radiocol",
+    C.PI_HANDLER: "yes",
+    C.PI_DESCRIPTION: _("""Implementation of radio collective"""),
+}
+
+
+# Number of songs needed in the queue before we start playing
+QUEUE_TO_START = 2
+# Maximum number of songs in the queue (the song being currently played doesn't count)
+QUEUE_LIMIT = 2
+
+
+class Radiocol(object):
+    def inherit_from_room_game(self, host):
+        global RoomGame
+        RoomGame = host.plugins["ROOM-GAME"].__class__
+        self.__class__ = type(
+            self.__class__.__name__, (self.__class__, RoomGame, object), {}
+        )
+
+    def __init__(self, host):
+        log.info(_("Radio collective initialization"))
+        self.inherit_from_room_game(host)
+        RoomGame._init_(
+            self,
+            host,
+            PLUGIN_INFO,
+            (NC_RADIOCOL, RADIOC_TAG),
+            game_init={
+                "queue": [],
+                "upload": True,
+                "playing": None,
+                "playing_time": 0,
+                "to_delete": {},
+            },
+        )
+        self.host = host
+        host.bridge.add_method(
+            "radiocol_launch",
+            ".plugin",
+            in_sign="asss",
+            out_sign="",
+            method=self._prepare_room,
+            async_=True,
+        )
+        host.bridge.add_method(
+            "radiocol_create",
+            ".plugin",
+            in_sign="sass",
+            out_sign="",
+            method=self._create_game,
+        )
+        host.bridge.add_method(
+            "radiocol_song_added",
+            ".plugin",
+            in_sign="sss",
+            out_sign="",
+            method=self._radiocol_song_added,
+            async_=True,
+        )
+        host.bridge.add_signal(
+            "radiocol_players", ".plugin", signature="ssass"
+        )  # room_jid, referee, players, profile
+        host.bridge.add_signal(
+            "radiocol_started", ".plugin", signature="ssasais"
+        )  # room_jid, referee, players, [QUEUE_TO_START, QUEUE_LIMIT], profile
+        host.bridge.add_signal(
+            "radiocol_song_rejected", ".plugin", signature="sss"
+        )  # room_jid, reason, profile
+        host.bridge.add_signal(
+            "radiocol_preload", ".plugin", signature="ssssssss"
+        )  # room_jid, timestamp, filename, title, artist, album, profile
+        host.bridge.add_signal(
+            "radiocol_play", ".plugin", signature="sss"
+        )  # room_jid, filename, profile
+        host.bridge.add_signal(
+            "radiocol_no_upload", ".plugin", signature="ss"
+        )  # room_jid, profile
+        host.bridge.add_signal(
+            "radiocol_upload_ok", ".plugin", signature="ss"
+        )  # room_jid, profile
+
+    def __create_preload_elt(self, sender, song_added_elt):
+        preload_elt = copy.deepcopy(song_added_elt)
+        preload_elt.name = "preload"
+        preload_elt["sender"] = sender
+        preload_elt["timestamp"] = str(time.time())
+        # attributes filename, title, artist, album, length have been copied
+        # XXX: the frontend should know the temporary directory where file is put
+        return preload_elt
+
+    def _radiocol_song_added(self, referee_s, song_path, profile):
+        return self.radiocol_song_added(jid.JID(referee_s), song_path, profile)
+
+    def radiocol_song_added(self, referee, song_path, profile):
+        """This method is called by libervia when a song has been uploaded
+        @param referee (jid.JID): JID of the referee in the room (room userhost + '/' + nick)
+        @param song_path (unicode): absolute path of the song added
+        @param profile_key (unicode): %(doc_profile_key)s
+        @return: a Deferred instance
+        """
+        # XXX: this is a Q&D way for the proof of concept. In the future, the song should
+        #     be streamed to the backend using XMPP file copy
+        #     Here we cheat because we know we are on the same host, and we don't
+        #     check data. Referee will have to parse the song himself to check it
+        try:
+            if song_path.lower().endswith(".mp3"):
+                actual_song = MP3(song_path)
+                try:
+                    song = EasyID3(song_path)
+
+                    class Info(object):
+                        def __init__(self, length):
+                            self.length = length
+
+                    song.info = Info(actual_song.info.length)
+                except ID3NoHeaderError:
+                    song = actual_song
+            else:
+                song = OggVorbis(song_path)
+        except (OggVorbisHeaderError, HeaderNotFoundError):
+            # this file is not ogg vorbis nor mp3, we reject it
+            self.delete_file(song_path)  # FIXME: same host trick (see note above)
+            return defer.fail(
+                exceptions.DataError(
+                    D_(
+                        "The uploaded file has been rejected, only Ogg Vorbis and MP3 songs are accepted."
+                    )
+                )
+            )
+
+        attrs = {
+            "filename": os.path.basename(song_path),
+            "title": song.get("title", ["Unknown"])[0],
+            "artist": song.get("artist", ["Unknown"])[0],
+            "album": song.get("album", ["Unknown"])[0],
+            "length": str(song.info.length),
+        }
+        radio_data = self.games[
+            referee.userhostJID()
+        ]  # FIXME: referee comes from Libervia's client side, it's unsecure
+        radio_data["to_delete"][
+            attrs["filename"]
+        ] = (
+            song_path
+        )  # FIXME: works only because of the same host trick, see the note under the docstring
+        return self.send(referee, ("", "song_added"), attrs, profile=profile)
+
+    def play_next(self, room_jid, profile):
+        """"Play next song in queue if exists, and put a timer
+        which trigger after the song has been played to play next one"""
+        # TODO: songs need to be erased once played or found invalids
+        #      ==> unlink done the Q&D way with the same host trick (see above)
+        radio_data = self.games[room_jid]
+        if len(radio_data["players"]) == 0:
+            log.debug(_("No more participants in the radiocol: cleaning data"))
+            radio_data["queue"] = []
+            for filename in radio_data["to_delete"]:
+                self.delete_file(filename, radio_data)
+            radio_data["to_delete"] = {}
+        queue = radio_data["queue"]
+        if not queue:
+            # nothing left to play, we need to wait for uploads
+            radio_data["playing"] = None
+            return
+        song = queue.pop(0)
+        filename, length = song["filename"], float(song["length"])
+        self.send(room_jid, ("", "play"), {"filename": filename}, profile=profile)
+        radio_data["playing"] = song
+        radio_data["playing_time"] = time.time()
+
+        if not radio_data["upload"] and len(queue) < QUEUE_LIMIT:
+            # upload is blocked and we now have resources to get more, we reactivate it
+            self.send(room_jid, ("", "upload_ok"), profile=profile)
+            radio_data["upload"] = True
+
+        reactor.callLater(length, self.play_next, room_jid, profile)
+        # we wait more than the song length to delete the file, to manage poorly reactive networks/clients
+        reactor.callLater(
+            length + 90, self.delete_file, filename, radio_data
+        )  # FIXME: same host trick (see above)
+
+    def delete_file(self, filename, radio_data=None):
+        """
+        Delete a previously uploaded file.
+        @param filename: filename to delete, or full filepath if radio_data is None
+        @param radio_data: current game data
+        @return: True if the file has been deleted
+        """
+        if radio_data:
+            try:
+                file_to_delete = radio_data["to_delete"][filename]
+            except KeyError:
+                log.error(
+                    _("INTERNAL ERROR: can't find full path of the song to delete")
+                )
+                return False
+        else:
+            file_to_delete = filename
+        try:
+            unlink(file_to_delete)
+        except OSError:
+            log.error(
+                _("INTERNAL ERROR: can't find %s on the file system" % file_to_delete)
+            )
+            return False
+        return True
+
+    def room_game_cmd(self, mess_elt, profile):
+        from_jid = jid.JID(mess_elt["from"])
+        room_jid = from_jid.userhostJID()
+        nick = self.host.plugins["XEP-0045"].get_room_nick(room_jid, profile)
+
+        radio_elt = mess_elt.firstChildElement()
+        radio_data = self.games[room_jid]
+        if "queue" in radio_data:
+            queue = radio_data["queue"]
+
+        from_referee = self.is_referee(room_jid, from_jid.resource)
+        to_referee = self.is_referee(room_jid, jid.JID(mess_elt["to"]).user)
+        is_player = self.is_player(room_jid, nick)
+        for elt in radio_elt.elements():
+            if not from_referee and not (to_referee and elt.name == "song_added"):
+                continue  # sender must be referee, expect when a song is submitted
+            if not is_player and (elt.name not in ("started", "players")):
+                continue  # user is in the room but not playing
+
+            if elt.name in (
+                "started",
+                "players",
+            ):  # new game created and/or players list updated
+                players = []
+                for player in elt.elements():
+                    players.append(str(player))
+                signal = (
+                    self.host.bridge.radiocol_started
+                    if elt.name == "started"
+                    else self.host.bridge.radiocol_players
+                )
+                signal(
+                    room_jid.userhost(),
+                    from_jid.full(),
+                    players,
+                    [QUEUE_TO_START, QUEUE_LIMIT],
+                    profile,
+                )
+            elif elt.name == "preload":  # a song is in queue and must be preloaded
+                self.host.bridge.radiocol_preload(
+                    room_jid.userhost(),
+                    elt["timestamp"],
+                    elt["filename"],
+                    elt["title"],
+                    elt["artist"],
+                    elt["album"],
+                    elt["sender"],
+                    profile,
+                )
+            elif elt.name == "play":
+                self.host.bridge.radiocol_play(
+                    room_jid.userhost(), elt["filename"], profile
+                )
+            elif elt.name == "song_rejected":  # a song has been refused
+                self.host.bridge.radiocol_song_rejected(
+                    room_jid.userhost(), elt["reason"], profile
+                )
+            elif elt.name == "no_upload":
+                self.host.bridge.radiocol_no_upload(room_jid.userhost(), profile)
+            elif elt.name == "upload_ok":
+                self.host.bridge.radiocol_upload_ok(room_jid.userhost(), profile)
+            elif elt.name == "song_added":  # a song has been added
+                # FIXME: we are KISS for the proof of concept: every song is added, to a limit of 3 in queue.
+                #       Need to manage some sort of rules to allow peoples to send songs
+                if len(queue) >= QUEUE_LIMIT:
+                    # there are already too many songs in queue, we reject this one
+                    # FIXME: add an error code
+                    self.send(
+                        from_jid,
+                        ("", "song_rejected"),
+                        {"reason": "Too many songs in queue"},
+                        profile=profile,
+                    )
+                    return
+
+                # The song is accepted and added in queue
+                preload_elt = self.__create_preload_elt(from_jid.resource, elt)
+                queue.append(preload_elt)
+
+                if len(queue) >= QUEUE_LIMIT:
+                    # We are at the limit, we refuse new upload until next play
+                    self.send(room_jid, ("", "no_upload"), profile=profile)
+                    radio_data["upload"] = False
+
+                self.send(room_jid, preload_elt, profile=profile)
+                if not radio_data["playing"] and len(queue) == QUEUE_TO_START:
+                    # We have not started playing yet, and we have QUEUE_TO_START
+                    # songs in queue. We can now start the party :)
+                    self.play_next(room_jid, profile)
+            else:
+                log.error(_("Unmanaged game element: %s") % elt.name)
+
+    def get_sync_data_for_player(self, room_jid, nick):
+        game_data = self.games[room_jid]
+        elements = []
+        if game_data["playing"]:
+            preload = copy.deepcopy(game_data["playing"])
+            current_time = game_data["playing_time"] + 1 if self.testing else time.time()
+            preload["filename"] += "#t=%.2f" % (current_time - game_data["playing_time"])
+            elements.append(preload)
+            play = domish.Element(("", "play"))
+            play["filename"] = preload["filename"]
+            elements.append(play)
+        if len(game_data["queue"]) > 0:
+            elements.extend(copy.deepcopy(game_data["queue"]))
+            if len(game_data["queue"]) == QUEUE_LIMIT:
+                elements.append(domish.Element(("", "no_upload")))
+        return elements