Changeset 123

Timestamp:

10/30/07 21:06:00 (2 years ago)

Author:

omega

Message:

* possibilité de lancer un random chat en parler au composant, ça envoie une invite mais on ne peut pas encore rejoindre le salon

Files:

• randomchat/randomchat.py (modified) (4 diffs)

randomchat/randomchat.py

@@ -3 +3 @@
 
 import xmpp
+import random
+import sha
+import time
+
 import config
 
@@ -40 +44 @@
         self.db.execute("INSERT INTO randomchat (jid) VALUES (?)", (jid,))
         self.db.commit()
+        self.users_list[jid] = PRS_OFFLINE
 
     def remove_user(self, jid):
         self.db.execute("DELETE FROM randomchat WHERE jid=?", (jid,))
         self.db.commit()
+        del self.users_list[jid]
 
-    def get_chat_partner(jid):
-        pass
+    def get_chat_partner(self, jid):
+        user = jid.getStripped()
+        if len(self.users_list) < 2:
+            return None
+        else:
+            users = self.users_list.keys()
+            other = random.choice(users)
+            while other == user:
+                other = random.choice(users)
+            return other
+
+class MUC:
+    def __init__(self, selfjid, con):
+        self.jid = selfjid
+        self.con = con
+        self.users = {}
+
+    def invite(self, room, jid):
+        invite = xmpp.Message(to = jid, frm = room+'@'+self.jid, body = 'Vous avez été invité dans le salon '+room+'@'+self.jid, payload = [ 
+            xmpp.Node('x', attrs={'xmlns': xmpp.NS_MUC+'#user'}, payload = [ 
+                xmpp.Node('invite', attrs={'to': room+'@'+self.jid}, payload = [ 
+                    xmpp.Node('reason', payload = ['Random chat !'] ) 
+                ])
+            ])
+        ])
+        self.con.send(invite)
+        self.users[jid.getStripped()] = room
+
+    def invite_users(self, jid1, jid2):
+        room = sha.sha(str(jid1)+str(jid2)+str(time.time())).hexdigest()
+        self.users[jid1] = room
+        self.users[jid2] = room
+        self.invite(room,jid1)
+        self.invite(room,jid2)
+
 
 
@@ -54 +93 @@
         self.con = xmpp.Component(server, port)
         self.jid = jid
+        self.muc = MUC(jid, self.con)
         if not self.con.connect():
             raise Exception('Unable to connect to %s:%s' % (server, port))
@@ -95 +135 @@
 
     def message_handler(self, con, msg):
-        self.init_chat(msg.getFrom())
+        if not self.init_chat(msg.getFrom()):
+            con.send(xmpp.Message(to = msg.getFrom(), frm = msg.getTo(), body = 'Pas de participant trouvé', typ = msg.getType()))
 
     def init_chat(self, jid):
         other_jid = self.users.get_chat_partner(jid)
+        if other_jid is None:
+            return False
+        self.muc.invite_users(jid, xmpp.JID(other_jid))
+        return True