plugins radiocol, card_game, quiz: code factorization
This commit is contained in:
souliane
parent
ba4c9158ea
commit
34947f8840
1 changed files with 284 additions and 379 deletions
|
|
@ -29,7 +29,7 @@ from zope.interface import implements
|
|||
import os.path
|
||||
from os import unlink
|
||||
from mutagen.oggvorbis import OggVorbis, OggVorbisHeaderError
|
||||
|
||||
from sat.tools.plugins.games import RoomGame
|
||||
try:
|
||||
from twisted.words.protocols.xmlstream import XMPPHandler
|
||||
except ImportError:
|
||||
|
|
@ -54,14 +54,15 @@ PLUGIN_INFO = {
|
|||
QUEUE_LIMIT = 2
|
||||
|
||||
|
||||
class Radiocol(object):
|
||||
class Radiocol(RoomGame):
|
||||
|
||||
def __init__(self, host):
|
||||
info(_("Radio collective initialization"))
|
||||
RoomGame.__init__(self, host, PLUGIN_INFO, (NC_RADIOCOL, RADIOC_TAG),
|
||||
options={'queue': [], 'upload': True, 'playing': False, 'to_delete': {}})
|
||||
self.host = host
|
||||
self.radios = {}
|
||||
host.bridge.addMethod("radiocolLaunch", ".plugin", in_sign='ass', out_sign='', method=self.radiocolLaunch)
|
||||
host.bridge.addMethod("radiocolCreate", ".plugin", in_sign='ss', out_sign='', method=self.radiocolCreate)
|
||||
host.bridge.addMethod("radiocolLaunch", ".plugin", in_sign='ass', out_sign='', method=self.prepareRoom)
|
||||
host.bridge.addMethod("radiocolCreate", ".plugin", in_sign='ss', out_sign='', method=self.createCollectiveGame)
|
||||
host.bridge.addMethod("radiocolSongAdded", ".plugin", in_sign='sss', out_sign='', method=self.radiocolSongAdded)
|
||||
host.bridge.addSignal("radiocolStarted", ".plugin", signature='sss') # room_jid, referee, profile
|
||||
host.bridge.addSignal("radiocolSongRejected", ".plugin", signature='sss') # room_jid, reason, profile
|
||||
|
|
@ -71,19 +72,6 @@ class Radiocol(object):
|
|||
host.bridge.addSignal("radiocolUploadOk", ".plugin", signature='ss') # room_jid, profile
|
||||
host.trigger.add("MUC user joined", self.userJoinedTrigger)
|
||||
|
||||
def createRadiocolElt(self, to_jid, type="normal"):
|
||||
type = "normal" if to_jid.resource else "groupchat"
|
||||
elt = domish.Element((None, 'message'))
|
||||
elt["to"] = to_jid.full()
|
||||
elt["type"] = type
|
||||
elt.addElement((NC_RADIOCOL, RADIOC_TAG))
|
||||
return elt
|
||||
|
||||
def __create_started_elt(self):
|
||||
"""Create a game_started domish element"""
|
||||
started_elt = domish.Element((None, 'started'))
|
||||
return started_elt
|
||||
|
||||
def __create_preload_elt(self, sender, filename, title, artist, album):
|
||||
preload_elt = domish.Element((None, 'preload'))
|
||||
preload_elt['sender'] = sender
|
||||
|
|
@ -93,82 +81,6 @@ class Radiocol(object):
|
|||
preload_elt['album'] = album
|
||||
return preload_elt
|
||||
|
||||
def userJoinedTrigger(self, room, user, profile):
|
||||
"""This trigger is used to check if we are waiting people in this room,
|
||||
and to create a game if everybody is here"""
|
||||
room_jid = room.occupantJID.userhost()
|
||||
if room_jid in self.radios and self.radios[room_jid]["referee"] == room.occupantJID.full():
|
||||
#we are in a radiocol room, let's start the party !
|
||||
mess = self.createRadiocolElt(jid.JID(room_jid + '/' + user.nick))
|
||||
mess.firstChildElement().addChild(self.__create_started_elt())
|
||||
self.host.profiles[profile].xmlstream.send(mess)
|
||||
return True
|
||||
|
||||
def radiocolLaunch(self, occupants, profile_key='@DEFAULT@'):
|
||||
"""Launch a game: helper method to create a room, invite occupants, and create the radiocol
|
||||
@param occupants: list for occupants jid"""
|
||||
debug(_('Launching radiocol'))
|
||||
profile = self.host.memory.getProfileName(profile_key)
|
||||
if not profile:
|
||||
error(_("Unknown profile"))
|
||||
return
|
||||
|
||||
def radiocolRoomJoined(room):
|
||||
print "radiocolRoomJoined"
|
||||
_room_jid = room.occupantJID.userhostJID()
|
||||
self.radiocolCreate(_room_jid.userhost(), profile_key=profile)
|
||||
for occupant in occupants:
|
||||
self.host.plugins["XEP-0249"].invite(jid.JID(occupant), room.occupantJID.userhostJID(), {"game": "Radiocol"}, profile)
|
||||
|
||||
def after_init(ignore):
|
||||
room_name = "sat_radiocol_%s" % self.host.plugins["XEP-0045"].getUniqueName(profile_key)
|
||||
print "\n\n===> room_name:", room_name
|
||||
muc_service = None
|
||||
for service in self.host.memory.getServerServiceEntities("conference", "text", profile):
|
||||
if not ".irc." in service.userhost():
|
||||
#FIXME:
|
||||
#This awfull ugly hack is here to avoid an issue with openfire: the irc gateway
|
||||
#use "conference/text" identity (instead of "conference/irc"), there is certainly a better way
|
||||
#to manage this, but this hack fill do it for test purpose
|
||||
muc_service = service
|
||||
break
|
||||
if not muc_service:
|
||||
error(_("Can't find a MUC service"))
|
||||
return
|
||||
|
||||
_jid, xmlstream = self.host.getJidNStream(profile)
|
||||
d = self.host.plugins["XEP-0045"].join(jid.JID("%s@%s" % (room_name, muc_service.userhost())), _jid.user, {}, profile)
|
||||
d.addCallback(radiocolRoomJoined)
|
||||
|
||||
client = self.host.getClient(profile)
|
||||
if not client:
|
||||
error(_('No client for this profile key: %s') % profile_key)
|
||||
return
|
||||
client.client_initialized.addCallback(after_init)
|
||||
|
||||
def radiocolCreate(self, room_jid_param, profile_key='@DEFAULT@'):
|
||||
"""Create a new game
|
||||
@param room_jid_param: jid of the room
|
||||
@param profile_key: %(doc_profile_key)s"""
|
||||
debug(_("Creating Radiocol"))
|
||||
room_jid = jid.JID(room_jid_param)
|
||||
profile = self.host.memory.getProfileName(profile_key)
|
||||
if not profile:
|
||||
error(_("profile %s is unknown") % profile_key)
|
||||
return
|
||||
if room_jid in self.radios:
|
||||
warning(_("Radiocol already started in room %s") % room_jid.userhost())
|
||||
else:
|
||||
room_nick = self.host.plugins["XEP-0045"].getRoomNick(room_jid.userhost(), profile)
|
||||
if not room_nick:
|
||||
error('Internal error')
|
||||
return
|
||||
referee = room_jid.userhost() + '/' + room_nick
|
||||
self.radios[room_jid.userhost()] = {'referee': referee, 'queue': [], 'upload': True, 'playing': False, 'occupants_data': {}, 'to_delete': {}}
|
||||
mess = self.createRadiocolElt(jid.JID(room_jid.userhost()))
|
||||
mess.firstChildElement().addChild(self.__create_started_elt())
|
||||
self.host.profiles[profile].xmlstream.send(mess)
|
||||
|
||||
def radiocolSongAdded(self, referee, song_path, profile):
|
||||
"""This method is called by libervia when a song has been uploaded
|
||||
@param room_jid_param: jid of the room
|
||||
|
|
@ -189,7 +101,7 @@ class Radiocol(object):
|
|||
unlink(song_path) # FIXME: same host trick (see note above)
|
||||
self.host.bridge.radiocolSongRejected(jid.JID(referee).userhost(),
|
||||
"Uploaded file is not Ogg Vorbis song, only Ogg Vorbis songs are acceptable", profile)
|
||||
"""mess = self.createRadiocolElt(jid.JID(referee))
|
||||
"""mess = self.createGameElt(jid.JID(referee))
|
||||
reject_elt = mess.firstChildElement().addElement(('','song_rejected'))
|
||||
reject_elt['sender'] = client.jid
|
||||
reject_elt['reason'] = "Uploaded file is not Ogg Vorbis song, only Ogg Vorbis songs are acceptable"
|
||||
|
|
@ -200,7 +112,7 @@ class Radiocol(object):
|
|||
artist = song.get("artist", ["Unknown"])[0]
|
||||
album = song.get("album", ["Unknown"])[0]
|
||||
length = song.info.length
|
||||
mess = self.createRadiocolElt(jid.JID(referee))
|
||||
mess = self.createGameElt(jid.JID(referee))
|
||||
added_elt = mess.firstChildElement().addElement(('', 'song_added'))
|
||||
added_elt['filename'] = filename = os.path.basename(song_path)
|
||||
added_elt['title'] = title
|
||||
|
|
@ -209,7 +121,7 @@ class Radiocol(object):
|
|||
added_elt['length'] = str(length)
|
||||
self.host.profiles[profile].xmlstream.send(mess)
|
||||
|
||||
radio_data = self.radios[jid.JID(referee).userhost()] # FIXME: referee comes from Libervia's client side, it's unsecure
|
||||
radio_data = self.games[jid.JID(referee).userhost()] # FIXME: referee comes from Libervia's client side, it's unsecure
|
||||
radio_data['to_delete'][filename] = song_path # FIXME: works only because of the same host trick, see the note under the docstring
|
||||
|
||||
def playNext(self, room_jid, profile):
|
||||
|
|
@ -219,7 +131,7 @@ class Radiocol(object):
|
|||
# and clean the datas/stop the playlist if it's not the case
|
||||
#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.radios[room_jid.userhost()]
|
||||
radio_data = self.games[room_jid.userhost()]
|
||||
queue = radio_data['queue']
|
||||
if not queue:
|
||||
#nothing left to play, we need to wait for uploads
|
||||
|
|
@ -227,14 +139,14 @@ class Radiocol(object):
|
|||
return
|
||||
|
||||
filename, length = queue.pop(0)
|
||||
mess = self.createRadiocolElt(room_jid)
|
||||
mess = self.createGameElt(room_jid)
|
||||
play_elt = mess.firstChildElement().addElement(('', 'play'))
|
||||
play_elt['filename'] = filename
|
||||
self.host.profiles[profile].xmlstream.send(mess)
|
||||
|
||||
if not radio_data['upload'] and len(queue) < QUEUE_LIMIT:
|
||||
#upload is blocked and we now have resources to get more, we reactivate it
|
||||
mess = self.createRadiocolElt(room_jid)
|
||||
mess = self.createGameElt(room_jid)
|
||||
no_upload_elt = mess.firstChildElement().addElement(('', 'upload_ok'))
|
||||
self.host.profiles[profile].xmlstream.send(mess)
|
||||
radio_data['upload'] = True
|
||||
|
|
@ -254,8 +166,7 @@ class Radiocol(object):
|
|||
from_jid = jid.JID(mess_elt['from'])
|
||||
room_jid = jid.JID(from_jid.userhost())
|
||||
radio_elt = mess_elt.firstChildElement()
|
||||
radio_data = self.radios[room_jid.userhost()]
|
||||
occupants_data = radio_data['occupants_data']
|
||||
radio_data = self.games[room_jid.userhost()]
|
||||
queue = radio_data['queue']
|
||||
|
||||
for elt in radio_elt.elements():
|
||||
|
|
@ -278,7 +189,7 @@ class Radiocol(object):
|
|||
|
||||
if len(queue) >= QUEUE_LIMIT:
|
||||
#there are already too many songs in queue, we reject this one
|
||||
mess = self.createRadiocolElt(room_jid)
|
||||
mess = self.createGameElt(room_jid)
|
||||
reject_elt = mess.firstChildElement().addElement(('', 'song_rejected'))
|
||||
reject_elt['sender'] = from_jid.resource
|
||||
reject_elt['reason'] = "Too many songs in queue"
|
||||
|
|
@ -291,13 +202,13 @@ class Radiocol(object):
|
|||
|
||||
if len(queue) >= QUEUE_LIMIT:
|
||||
#We are at the limit, we refuse new upload until next play
|
||||
mess = self.createRadiocolElt(room_jid)
|
||||
mess = self.createGameElt(room_jid)
|
||||
no_upload_elt = mess.firstChildElement().addElement(('', 'no_upload'))
|
||||
#FIXME: add an error code
|
||||
self.host.profiles[profile].xmlstream.send(mess)
|
||||
radio_data['upload'] = False
|
||||
|
||||
mess = self.createRadiocolElt(room_jid)
|
||||
mess = self.createGameElt(room_jid)
|
||||
preload_elt = self.__create_preload_elt(from_jid.resource,
|
||||
elt['filename'],
|
||||
elt['title'],
|
||||
|
|
|
|||
Loading…
Reference in a new issue