bla

git-svn-id: http://xapek.org/svn/common/omegle@1423 d0e8fea9-7529-0410-93fb-d39fd5b9c1dd

1 files changed, 135 insertions, 70 deletions

diff --git a/icq_roombot.py b/icq_roombot.py
index ebb700c..c2a07fb 100644
--- a/icq_roombot.py
+++ b/icq_roombot.py
@@ -8,93 +8,158 @@ from random import randint
 uin = '370496181'
 password = 'megahal123'
 
+class Member(object):
+    def __init__(self,id,name):
+        self.id = id
+        self.name = name
+
+class Room(object):
+    def __init__(self):
+        self.members = []
+	self.msg_cbs = []
+
+    def addMessageCallback(self,cb):
+        self.msg_cbs.append(cb)
+
+    def removeMessageCallback(self,cb):
+        self.msg_cbs.remove(cb)
+    
+    def join(self,member):
+        self.members.append(member)
+
+    def leave(self,member):
+        self.members.remove(member)
+
+    def getMemberById(self,id):
+        l=filter(lambda member: member.id == id, self.members)
+        if len(l) > 0:
+            return l[0]
+        else:
+            return None
+
+    instance = None
+    @classmethod
+    def getInstance(klass):
+        if klass.instance == None:
+            klass.instance = Room()
+        return klass.instance
+	
 CAP_TYPING  = '\x56\x3f\xc8\x09\x0b\x6f\x41\xbd\x9f\x79\x42\x26\x09\xdf\xa2\xf3'
 class icqBot( oscar.BOSConnection ):
-	capabilities = [oscar.CAP_CHAT,CAP_TYPING]
+    capabilities = [oscar.CAP_CHAT,CAP_TYPING]
  
-	def initDone( self ):
-		self.requestSelfInfo()
-		self.requestSSI().addCallback(self.gotBuddyList)
-		self.members = dict()
+    def initDone( self ):
+        self.requestSelfInfo()
+        self.requestSSI().addCallback(self.gotBuddyList)
+        self.room = Room.getInstance()
+        self.members = dict()
 
-	def oscar_04_14(self,snac):
-		pass #TODO: handle typing event - see omegle_icq.py
+    def oscar_04_14(self,snac):
+        pass #TODO: handle typing event - see omegle_icq.py
 
-	def updateBuddy(self, user):
-		print "Update buddy %s" % user
+    def updateBuddy(self, user):
+        print "Update buddy %s" % user
  
-	def gotBuddyList( self, buddylist ):
-		self.activateSSI()
-		self.setProfile("Forget ICQ, MSN, Yahoo and the other shitty protocols! Use Jabber!")
-		self.setIdleTime( 0 )
-		self.clientReady()
-		for user in buddylist[0][0].users:
-			self.sendAuthorizationResponse(user.name, True, '')
+    def gotBuddyList( self, buddylist ):
+        self.activateSSI()
+        self.setProfile("Forget ICQ, MSN, Yahoo and the other shitty protocols! Use Jabber!")
+        self.setIdleTime( 0 )
+        self.clientReady()
+        for user in buddylist[0][0].users:
+            self.sendAuthorizationResponse(user.name, True, '')
  
-		print buddylist
+        print buddylist
  
-	def receiveMessage( self, user, multiparts, flags ):
-		try:
-			msg = str(multiparts[0][0])
-			if msg.strip() in ("/leave", "/LEAVE"):
-				self.members.pop(user.name)
-				self.sendMessage(user.name, "You left")
-				for member in self.members.keys():
-					self.sendMessage(user.name, "%s(%s) left"%(member,self.members[member]))
-			elif msg.strip().startswith("/list"):
-				names = ", ".join(
-					map(lambda key: "%s(%s)"%(self.members[key],key), self.members.keys()))
-				self.sendMessage(user.name, "Connected to: " + names)
-			elif msg.strip().startswith("/join"):
-				nick = msg.split("/join")[-1:][0] or user.name
-				self.sendMessage(user.name, "========= Joined as %s - to leave type >/leave<" % nick)
-				self.sendMessage(user.name, "Connected to: " + ", ".join(self.members.values()))
-				for member in self.members.keys():
-					self.sendMessage(member, "======== %s(%s) (re-)joined" % (nick,user.name))
-				self.members.update( ( (user.name, nick), ) )
-			elif user.name in self.members.keys():
-				if msg.strip().startswith("/invite"):
-					who = msg.split("/invite")[-1:][0] or False
-					if who:
-						self.sendMessage(who, "%s invite you to join this room" % user.name)
-						self.sendMessage(user.name, "Invitation sent to %s" % who)
-				else:
-					for member in self.members.keys():
-						if member == user.name:
-							continue
-						self.sendMessage(member, "%s: %s"%(self.members[user.name], msg)  )
-			else:
-				if randint(0,5) == 3:
-					self.sendMessage(user.name, "Use '/join NICKNAME', '/list' and '/leave'")
-
-			self.sendAuthorizationResponse(user.name, True, '')
-		except Exception,e:
-			print "error: %s %s\n%s" % (user,multiparts,e)
-		except:
-			pass
-
-	def chatReceiveMessage( self, chat, user, message ):
-		self.receiveMessage( self, user, message, 0 )
+    def _getMessage(self,multiparts):
+        message = None
+        if len(multiparts[0]) == 2:
+            if multiparts[0][1] == "unicode":
+                message = unicode(multiparts[0][0])
+            else:
+                message = multiparts[0][0].decode(multiparts[0][1],'latin1')
+        else:
+            message = str(multiparts[0][0])
 
+        #filter qip \x00CHAR\x00CHAR - qip sucks at unicode?
+        message = filter(lambda x: x!=u"\x00", message)
+        return message
+        
+    def sendMessage(self, user, messageUtf8):
+        message = messageUtf8.encode('latin1','replace')
+        oscar.BOSConnection.sendMessage(self,user,message)
+    def receiveMessage( self, user, multiparts, flags ):
+        try:
+            try:
+                message = self._getMessage(multiparts)
+            except Exception,e:
+                print e
+                self.sendMessage(user.name, str(e))
+                return
+
+            if self.room.getMemberById(user.name) != None:
+                member = self.room.getMemberById(user.name)
+
+                if message.strip() == "/leave":
+                    self.room.leave(member)
+                    self.sendMessage(user.name, 'You left')
+                    for otherMember in self.room.members:
+                        self.sendMessage(otherMember.id, "===== %s/%s left"%(member.name,member.id) )
+                elif message.strip() == "/invite":
+                    self.sendMessage(member.id, "===== geht nicht")
+                elif message.strip() == "/nick":
+                    member.name = message.strip().split("/nick")[1]
+                elif message.strip() == "/msg":
+                    self.sendMessage(member.id, "===== geht nicht")
+                elif message.strip() == "/list":
+                    self.sendMessage(member.id, "===== List: %s"%", ".join(map(lambda m: "%s/%s"%(m.id,m.name),self.room.members)) )
+                elif message.strip().startswith("/"):
+                    self.sendMessage(member.id, "===== kennich nicht")
+                else:
+                    for otherMember in self.room.members:
+                        if otherMember.id == user.name:
+                            continue
+                        self.sendMessage(otherMember.id, message.strip())
+            else:
+                if message.strip().startswith("/join"):
+                    nick = "%s/%s"%(message.strip().split("/join")[1],user.name)
+                    member = Member(user.name, nick)
+                    self.room.join(member)
+                    for otherMember in self.room.members:
+                        self.sendMessage(otherMember.id, "===== %s/%s joined" % (member.name,member.id))
+                else:
+                    if randint(0,5) == 3:
+                        self.sendMessage(user.name, "Use '/join Nickname' or ' /join Nickname'")
+
+            self.sendAuthorizationResponse(user.name, True, '')
+        except Exception,e:
+            print "error: %s %s\n%s" % (user,multiparts,e)
+
+    def chatReceiveMessage( self, chat, user, message ):
+        self.receiveMessage( self, user, message, 0 )
+    """
 	def receiveSendFileRequest(self, *args):
 #	def receiveSendFileRequest(self, user, file, description, cookie):
 		print args
+                """
  
-	def sendAuthorizationResponse(self, uin, success, responsString):
-		packet  = struct.pack("b", len(uin)) + uin
-		if success:
-			packet += struct.pack("b", 1)
-		else:
-			packet += struct.pack("b", 0)
-		packet += struct.pack(">H", len(responsString)) + responsString
-		self.sendSNACnr(0x13, 0x1a, packet)
+    def sendAuthorizationResponse(self, uin, success, responsString):
+        packet  = struct.pack("b", len(uin)) + uin
+        if success:
+            packet += struct.pack("b", 1)
+        else:
+            packet += struct.pack("b", 0)
+        packet += struct.pack(">H", len(responsString)) + responsString
+        self.sendSNACnr(0x13, 0x1a, packet)
+
+    def connectionLost(self,*args,**kwargs):
+        print "Connection lost"
+        oscar.BOSConnection.connectionLost(self,*args,**kwargs)
 
-	def connectionLost(self,*args,**kwargs):
-		print "Connection lost"
+		
 
 
 class OscarCrap( oscar.OscarAuthenticator ):
-	BOSClass = icqBot
+    BOSClass = icqBot
 
 
 protocol.ClientCreator( reactor, OscarCrap, uin, password, icq=1 ).connectTCP( 'login.icq.com', 5238 )