Forked mumble-django project from https://bitbucket.org/Svedrin/mumble-django
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

353 lines
11 KiB

  1. # -*- coding: utf-8 -*-
  2. """
  3. * Copyright (C) 2009, withgod <withgod@sourceforge.net>
  4. * Michael "Svedrin" Ziegler <diese-addy@funzt-halt.net>
  5. *
  6. * Mumble-Django is free software; you can redistribute it and/or modify
  7. * it under the terms of the GNU General Public License as published by
  8. * the Free Software Foundation; either version 2 of the License, or
  9. * (at your option) any later version.
  10. *
  11. * This package is distributed in the hope that it will be useful,
  12. * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13. * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  14. * GNU General Public License for more details.
  15. """
  16. from os.path import join
  17. from PIL import Image
  18. from struct import pack, unpack
  19. from zlib import compress, decompress
  20. from django.conf import settings
  21. from mctl import MumbleCtlBase
  22. import Ice
  23. def MumbleCtlIce( connstring ):
  24. version = settings.SLICE_VERSION;
  25. slice = settings.SLICE;
  26. if not slice:
  27. slice = join(
  28. settings.MUMBLE_DJANGO_ROOT, 'pyweb', 'mumble',
  29. 'Murmur_%d-%d-%d.ice' % (version[0], version[1], version[2] )
  30. );
  31. Ice.loadSlice( slice )
  32. ice = Ice.initialize()
  33. import Murmur
  34. prx = ice.stringToProxy( connstring.encode("utf-8") )
  35. meta = Murmur.MetaPrx.checkedCast(prx)
  36. murmurversion = meta.getVersion();
  37. if murmurversion[0] != version[0] or murmurversion[1] != version[1] or murmurversion[2] != version[2]:
  38. raise TypeError(
  39. "Detected Murmur version %d.%d.%d, for which I am not configured." % ( murmurversion[0], murmurversion[1], murmurversion[2] )
  40. );
  41. if murmurversion[0] == murmurversion[1] == 1 and murmurversion[2] <= 8:
  42. return MumbleCtlIce_118( connstring, meta );
  43. elif murmurversion[0] == 1 and murmurversion[1] == 2 and murmurversion[2] == 0:
  44. return MumbleCtlIce_120( connstring, meta );
  45. class MumbleCtlIce_118(MumbleCtlBase):
  46. method = "ICE";
  47. def __init__( self, connstring, meta ):
  48. self.proxy = connstring;
  49. self.meta = meta;
  50. def _getIceServerObject(self, srvid):
  51. return self.meta.getServer(srvid);
  52. def getBootedServers(self):
  53. ret = []
  54. for x in self.meta.getBootedServers():
  55. ret.append(x.id())
  56. return ret
  57. def getVersion( self ):
  58. return self.meta.getVersion();
  59. def getAllServers(self):
  60. ret = []
  61. for x in self.meta.getAllServers():
  62. ret.append(x.id())
  63. return ret
  64. def getRegisteredPlayers(self, srvid):
  65. users = self._getIceServerObject(srvid).getRegisteredPlayers('')
  66. ret = []
  67. for user in users:
  68. ret.append([user.playerid, self.setUnicodeFlag(user.name), self.setUnicodeFlag(user.email), self.setUnicodeFlag(user.pw)])
  69. return ret
  70. def getChannels(self, srvid):
  71. chans = self._getIceServerObject(srvid).getChannels()
  72. ret = []
  73. for x in chans:
  74. chan = chans[x]
  75. ret.append([chan.id, self.setUnicodeFlag(chan.name), chan.parent, chan.links])
  76. return ret
  77. def getPlayers(self, srvid):
  78. users = self._getIceServerObject(srvid).getPlayers()
  79. ret = []
  80. for x in users:
  81. user = users[x]
  82. ret.append([user.session, user.mute, user.deaf, user.suppressed, user.selfMute, user.selfDeaf, user.channel, user.playerid, self.setUnicodeFlag(user.name), user.onlinesecs, user.bytespersec])
  83. return ret
  84. def getACL(self, srvid, identifier):
  85. import Murmur
  86. acls = self._getIceServerObject(srvid).getACL(identifier)
  87. ret = []
  88. for x in acls:
  89. if isinstance(x, list):
  90. tmp = []
  91. for y in x:
  92. if y.__class__ is Murmur.ACL:
  93. tmp.append([y.applyHere, y.applySubs, y.inherited, y.playerid, self.setUnicodeFlag(y.group), y.allow, y.deny])
  94. elif y.__class__ is Murmur.Group:
  95. tmp.append([self.setUnicodeFlag(y.name), y.inherited, y.inherit, y.inheritable, y.add, y.remove, y.members])
  96. ret.append(tmp)
  97. else:
  98. ret.append(x)
  99. return ret
  100. def getDefaultConf(self):
  101. return self.setUnicodeFlag(self.meta.getDefaultConf())
  102. def getAllConf(self, srvid):
  103. return self.setUnicodeFlag(self._getIceServerObject(srvid).getAllConf())
  104. def newServer(self):
  105. return self.meta.newServer().id()
  106. def isBooted( self, srvid ):
  107. return bool( self._getIceServerObject(srvid).isRunning() );
  108. def start( self, srvid ):
  109. self._getIceServerObject(srvid).start();
  110. def stop( self, srvid ):
  111. self._getIceServerObject(srvid).stop();
  112. def deleteServer( self, srvid ):
  113. if self._getIceServerObject(srvid).isRunning():
  114. self._getIceServerObject(srvid).stop()
  115. self._getIceServerObject(srvid).delete()
  116. def setSuperUserPassword(self, srvid, value):
  117. self.meta.setSuperUserPassword(srvid, value)
  118. def setConf(self, srvid, key, value):
  119. value = value.encode("utf-8")
  120. #print "%s server %s=%s (%s/%s)" % (srvid, key, value, type(key), type(value))
  121. self._getIceServerObject(srvid).setConf(key, value)
  122. def registerPlayer(self, srvid, name):
  123. return self._getIceServerObject(srvid).registerPlayer(name)
  124. def unregisterPlayer(self, srvid, mumbleid):
  125. self._getIceServerObject(srvid).unregisterPlayer(mumbleid)
  126. def setRegistration(self, srvid, mumbleid, name, email, password):
  127. user = self._getIceServerObject(srvid).getRegistration(mumbleid)
  128. user.name = name
  129. user.email = email
  130. user.pw = password
  131. #print user
  132. # update*r*egistration r is lowercase...
  133. return self._getIceServerObject(srvid).updateregistration(user)
  134. def setACL(self, srvid, acl):
  135. import Murmur
  136. newacls = [];
  137. newgroups = [];
  138. for curr_acl in acl.acls:
  139. new_acl = Murmur.ACL();
  140. new_acl.applyHere = curr_acl['applyHere'];
  141. new_acl.applySubs = curr_acl['applySubs'];
  142. new_acl.inherited = curr_acl['inherited'];
  143. new_acl.playerid = curr_acl['playerid'];
  144. new_acl.group = curr_acl['group'].encode( "UTF-8" );
  145. new_acl.allow = curr_acl['allow'];
  146. new_acl.deny = curr_acl['deny'];
  147. newacls.append( new_acl );
  148. for curr_group in acl.groups:
  149. new_group = Murmur.Group()
  150. new_group.name = curr_group['name'].encode( "UTF-8" );
  151. new_group.inherited = curr_group['inherited'];
  152. new_group.inherit = curr_group['inherit'];
  153. new_group.inheritable = curr_group['inheritable'];
  154. new_group.add = curr_group['add'];
  155. new_group.remove = curr_group['remove'];
  156. new_group.members = curr_group['members'];
  157. newgroups.append( new_group );
  158. self._getIceServerObject(srvid).setACL( acl.channelId, newacls, newgroups, acl.inherit );
  159. def getTexture(self, srvid, mumbleid):
  160. texture = self._getIceServerObject(srvid).getTexture(mumbleid)
  161. if len(texture) == 0:
  162. raise ValueError( "No Texture has been set." );
  163. # this returns a list of bytes.
  164. decompressed = decompress( texture );
  165. # iterate over 4 byte chunks of the string
  166. imgdata = "";
  167. for idx in range( 0, len(decompressed), 4 ):
  168. # read 4 bytes = BGRA and convert to RGBA
  169. # manual wrote getTexture returns "Textures are stored as zlib compress()ed 600x60 32-bit RGBA data."
  170. # http://mumble.sourceforge.net/slice/Murmur/Server.html#getTexture
  171. # but return values BGRA X(
  172. bgra = unpack( "4B", decompressed[idx:idx+4] );
  173. imgdata += pack( "4B", bgra[2], bgra[1], bgra[0], bgra[3] );
  174. # return an 600x60 RGBA image object created from the data
  175. return Image.fromstring( "RGBA", ( 600, 60 ), imgdata);
  176. def setTexture(self, srvid, mumbleid, infile):
  177. # open image, convert to RGBA, and resize to 600x60
  178. img = Image.open( infile ).convert( "RGBA" ).transform( ( 600, 60 ), Image.EXTENT, ( 0, 0, 600, 60 ) );
  179. # iterate over the list and pack everything into a string
  180. bgrastring = "";
  181. for ent in list( img.getdata() ):
  182. # ent is in RGBA format, but Murmur wants BGRA (ARGB inverse), so stuff needs
  183. # to be reordered when passed to pack()
  184. bgrastring += pack( "4B", ent[2], ent[1], ent[0], ent[3] );
  185. # compress using zlib
  186. compressed = compress( bgrastring );
  187. # pack the original length in 4 byte big endian, and concat the compressed
  188. # data to it to emulate qCompress().
  189. texture = pack( ">L", len(bgrastring) ) + compressed;
  190. # finally call murmur and set the texture
  191. self._getIceServerObject(srvid).setTexture(mumbleid, texture)
  192. @staticmethod
  193. def setUnicodeFlag(data):
  194. ret = ''
  195. if isinstance(data, tuple) or isinstance(data, list) or isinstance(data, dict):
  196. ret = {}
  197. for key in data.keys():
  198. ret[MumbleCtlIce_118.setUnicodeFlag(key)] = MumbleCtlIce_118.setUnicodeFlag(data[key])
  199. else:
  200. ret = unicode(data, 'utf-8')
  201. return ret
  202. class MumbleCtlIce_120(MumbleCtlIce_118):
  203. def getRegisteredPlayers(self, srvid):
  204. users = self._getIceServerObject(srvid).getRegisteredUsers('')
  205. ret = []
  206. for user in users:
  207. ret.append([user.playerid, self.setUnicodeFlag(user.name), self.setUnicodeFlag(user.email), self.setUnicodeFlag(user.pw)])
  208. return ret
  209. def getChannels(self, srvid):
  210. chans = self._getIceServerObject(srvid).getChannels()
  211. ret = []
  212. for x in chans:
  213. chan = chans[x]
  214. ret.append([chan.id, self.setUnicodeFlag(chan.name), chan.parent, chan.links, chan.description])
  215. return ret
  216. def getPlayers(self, srvid):
  217. serv = self._getIceServerObject(srvid);
  218. users = serv.getUsers()
  219. ret = []
  220. for x in users:
  221. user = users[x]
  222. ret.append([user.session, user.mute, user.deaf, user.suppressed, user.selfMute, user.selfDeaf, user.channel, user.playerid, self.setUnicodeFlag(user.name), user.onlinesecs, user.bytespersec])
  223. return ret
  224. def registerPlayer(self, srvid, name):
  225. return self._getIceServerObject(srvid).registerUser(name)
  226. def unregisterPlayer(self, srvid, mumbleid):
  227. self._getIceServerObject(srvid).unregisterUser(mumbleid)
  228. def setRegistration(self, srvid, mumbleid, name, email, password):
  229. user = self._getIceServerObject(srvid).getRegistration(mumbleid)
  230. user.name = name
  231. user.email = email
  232. user.pw = password
  233. return self._getIceServerObject(srvid).updateRegistration(user)
  234. def getACL(self, srvid, identifier):
  235. import Murmur
  236. acls = self._getIceServerObject(srvid).getACL(identifier)
  237. ret = []
  238. for x in acls:
  239. if isinstance(x, list):
  240. tmp = []
  241. for y in x:
  242. if y.__class__ is Murmur.ACL:
  243. tmp.append([y.applyHere, y.applySubs, y.inherited, y.userid, self.setUnicodeFlag(y.group), y.allow, y.deny])
  244. elif y.__class__ is Murmur.Group:
  245. tmp.append([self.setUnicodeFlag(y.name), y.inherited, y.inherit, y.inheritable, y.add, y.remove, y.members])
  246. ret.append(tmp)
  247. else:
  248. ret.append(x)
  249. return ret
  250. def setACL(self, srvid, acl):
  251. import Murmur
  252. newacls = [];
  253. newgroups = [];
  254. for curr_acl in acl.acls:
  255. new_acl = Murmur.ACL();
  256. new_acl.applyHere = curr_acl['applyHere'];
  257. new_acl.applySubs = curr_acl['applySubs'];
  258. new_acl.inherited = curr_acl['inherited'];
  259. new_acl.userid = curr_acl['playerid'];
  260. new_acl.group = curr_acl['group'].encode( "UTF-8" );
  261. new_acl.allow = curr_acl['allow'];
  262. new_acl.deny = curr_acl['deny'];
  263. newacls.append( new_acl );
  264. for curr_group in acl.groups:
  265. new_group = Murmur.Group()
  266. new_group.name = curr_group['name'].encode( "UTF-8" );
  267. new_group.inherited = curr_group['inherited'];
  268. new_group.inherit = curr_group['inherit'];
  269. new_group.inheritable = curr_group['inheritable'];
  270. new_group.add = curr_group['add'];
  271. new_group.remove = curr_group['remove'];
  272. new_group.members = curr_group['members'];
  273. newgroups.append( new_group );
  274. self._getIceServerObject(srvid).setACL( acl.channelId, newacls, newgroups, acl.inherit );