initial code commit
This commit is contained in:
parent
80938a3329
commit
528d4ed759
0
Spectrum2/__init__.py
Normal file
0
Spectrum2/__init__.py
Normal file
543
Spectrum2/backend.py
Normal file
543
Spectrum2/backend.py
Normal file
|
@ -0,0 +1,543 @@
|
||||||
|
import protocol_pb2, socket, struct, sys, os
|
||||||
|
|
||||||
|
def WRAP(MESSAGE, TYPE):
|
||||||
|
wrap = protocol_pb2.WrapperMessage()
|
||||||
|
wrap.type = TYPE
|
||||||
|
wrap.payload = MESSAGE
|
||||||
|
return wrap.SerializeToString()
|
||||||
|
|
||||||
|
class SpectrumBackend:
|
||||||
|
"""
|
||||||
|
Creates new NetworkPlugin and connects the Spectrum2 NetworkPluginServer.
|
||||||
|
@param loop: Event loop.
|
||||||
|
@param host: Host where Spectrum2 NetworkPluginServer runs.
|
||||||
|
@param port: Port.
|
||||||
|
"""
|
||||||
|
|
||||||
|
def __init__(self):
|
||||||
|
self.m_pingReceived = False
|
||||||
|
self.m_data = ""
|
||||||
|
self.m_init_res = 0
|
||||||
|
|
||||||
|
def handleMessage(self, user, legacyName, msg, nickname = "", xhtml = "", timestamp = ""):
|
||||||
|
m = protocol_pb2.ConversationMessage()
|
||||||
|
m.userName = user
|
||||||
|
m.buddyName = legacyName
|
||||||
|
m.message = msg
|
||||||
|
m.nickname = nickname
|
||||||
|
m.xhtml = xhtml
|
||||||
|
#m.timestamp = str(timestamp)
|
||||||
|
|
||||||
|
message = WRAP(m.SerializeToString(), protocol_pb2.WrapperMessage.TYPE_CONV_MESSAGE)
|
||||||
|
self.send(message)
|
||||||
|
|
||||||
|
def handleAttention(self, user, buddyName, msg):
|
||||||
|
m = protocol_pb2.ConversationMessage()
|
||||||
|
m.userName = user
|
||||||
|
m.buddyName = buddyName
|
||||||
|
m.message = msg
|
||||||
|
|
||||||
|
message = WRAP(m.SerializeToString(), protocol_pb2.WrapperMessage.TYPE_ATTENTION)
|
||||||
|
self.send(message)
|
||||||
|
|
||||||
|
def handleVCard(self, user, ID, legacyName, fullName, nickname, photo):
|
||||||
|
vcard = protocol_pb2.VCard()
|
||||||
|
vcard.userName = user
|
||||||
|
vcard.buddyName = legacyName
|
||||||
|
vcard.id = ID
|
||||||
|
vcard.fullname = fullName
|
||||||
|
vcard.nickname = nickname
|
||||||
|
vcard.photo = photo
|
||||||
|
|
||||||
|
message = WRAP(vcard.SerializeToString(), protocol_pb2.WrapperMessage.TYPE_VCARD)
|
||||||
|
self.send(message)
|
||||||
|
|
||||||
|
|
||||||
|
def handleSubject(self, user, legacyName, msg, nickname = ""):
|
||||||
|
m = protocol_pb2.ConversationMessage()
|
||||||
|
m.userName = user
|
||||||
|
m.buddyName = legacyName
|
||||||
|
m.message = msg
|
||||||
|
m.nickname = nickname
|
||||||
|
|
||||||
|
message = WRAP(m.SerializeToString(), protocol_pb2.WrapperMessage.TYPE_ROOM_SUBJECT_CHANGED)
|
||||||
|
self.send(message)
|
||||||
|
|
||||||
|
def handleBuddyChanged(self, user, buddyName, alias, groups, status, statusMessage = "", iconHash = "", blocked = False):
|
||||||
|
buddy = protocol_pb2.Buddy()
|
||||||
|
buddy.userName = user
|
||||||
|
buddy.buddyName = buddyName
|
||||||
|
buddy.alias = alias
|
||||||
|
buddy.group.extend(groups)
|
||||||
|
buddy.status = status
|
||||||
|
buddy.statusMessage = statusMessage
|
||||||
|
buddy.iconHash = iconHash
|
||||||
|
buddy.blocked = blocked
|
||||||
|
|
||||||
|
message = WRAP(buddy.SerializeToString(), protocol_pb2.WrapperMessage.TYPE_BUDDY_CHANGED)
|
||||||
|
self.send(message)
|
||||||
|
|
||||||
|
|
||||||
|
def handleBuddyTyping(self, user, buddyName):
|
||||||
|
buddy = protocol_pb2.Buddy()
|
||||||
|
buddy.userName = user
|
||||||
|
buddy.buddyName = buddyName
|
||||||
|
|
||||||
|
message = WRAP(buddy.SerializeToString(), protocol_pb2.WrapperMessage.TYPE_BUDDY_TYPING)
|
||||||
|
self.send(message);
|
||||||
|
|
||||||
|
def handleBuddyTyped(self, user, buddyName):
|
||||||
|
buddy = protocol_pb2.Buddy()
|
||||||
|
buddy.userName = user
|
||||||
|
buddy.buddyName = buddyName
|
||||||
|
|
||||||
|
message = WRAP(buddy.SerializeToString(), protocol_pb2.WrapperMessage.TYPE_BUDDY_TYPED)
|
||||||
|
self.send(message);
|
||||||
|
|
||||||
|
def handleBuddyStoppedTyping(self, user, buddyName):
|
||||||
|
buddy = protocol_pb2.Buddy()
|
||||||
|
buddy.userName = user
|
||||||
|
buddy.buddyName = buddyName
|
||||||
|
|
||||||
|
message = WRAP(buddy.SerializeToString(), protocol_pb2.WrapperMessage.TYPE_BUDDY_STOPPED_TYPING)
|
||||||
|
self.send(message)
|
||||||
|
|
||||||
|
def handleAuthorization(self, user, buddyName):
|
||||||
|
buddy = protocol_pb2.Buddy()
|
||||||
|
buddy.userName = user
|
||||||
|
buddy.buddyName = buddyName
|
||||||
|
|
||||||
|
message = WRAP(buddy.SerializeToString(), protocol_pb2.WrapperMessage.TYPE_AUTH_REQUEST)
|
||||||
|
self.send(message)
|
||||||
|
|
||||||
|
|
||||||
|
def handleConnected(self, user):
|
||||||
|
d = protocol_pb2.Connected()
|
||||||
|
d.user = user
|
||||||
|
|
||||||
|
message = WRAP(d.SerializeToString(), protocol_pb2.WrapperMessage.TYPE_CONNECTED)
|
||||||
|
self.send(message);
|
||||||
|
|
||||||
|
|
||||||
|
def handleDisconnected(self, user, error = 0, msg = ""):
|
||||||
|
d = protocol_pb2.Disconnected()
|
||||||
|
d.user = user
|
||||||
|
d.error = error
|
||||||
|
d.message = msg
|
||||||
|
|
||||||
|
message = WRAP(d.SerializeToString(), protocol_pb2.WrapperMessage.TYPE_DISCONNECTED)
|
||||||
|
self.send(message);
|
||||||
|
|
||||||
|
|
||||||
|
def handleParticipantChanged(self, user, nickname, room, flags, status, statusMessage = "", newname = ""):
|
||||||
|
d = protocol_pb2.Participant()
|
||||||
|
d.userName = user
|
||||||
|
d.nickname = nickname
|
||||||
|
d.room = room
|
||||||
|
d.flag = flags
|
||||||
|
d.newname = newname
|
||||||
|
d.status = status
|
||||||
|
d.statusMessage = statusMessage
|
||||||
|
|
||||||
|
message = WRAP(d.SerializeToString(), protocol_pb2.WrapperMessage.TYPE_PARTICIPANT_CHANGED)
|
||||||
|
self.send(message);
|
||||||
|
|
||||||
|
|
||||||
|
def handleRoomNicknameChanged(self, user, r, nickname):
|
||||||
|
room = protocol_pb2.Room()
|
||||||
|
room.userName = user
|
||||||
|
room.nickname = nickname
|
||||||
|
room.room = r
|
||||||
|
room.password = ""
|
||||||
|
|
||||||
|
message = WRAP(room.SerializeToString(), protocol_pb2.WrapperMessage.TYPE_ROOM_NICKNAME_CHANGED)
|
||||||
|
self.send(message);
|
||||||
|
|
||||||
|
def handleRoomList(self, rooms):
|
||||||
|
roomList = protocol_pb2.RoomList()
|
||||||
|
|
||||||
|
for room in rooms:
|
||||||
|
roomList.room.append(room[0])
|
||||||
|
roomList.name.append(room[1])
|
||||||
|
|
||||||
|
message = WRAP(roomList.SerializeToString(), protocol_pb2.WrapperMessage.TYPE_ROOM_LIST)
|
||||||
|
self.send(message);
|
||||||
|
|
||||||
|
|
||||||
|
def handleFTStart(self, user, buddyName, fileName, size):
|
||||||
|
room = protocol_pb2.File()
|
||||||
|
room.userName = user
|
||||||
|
room.buddyName = buddyName
|
||||||
|
room.fileName = fileName
|
||||||
|
room.size = size
|
||||||
|
|
||||||
|
message = WRAP(room.SerializeToString(), protocol_pb2.WrapperMessage.TYPE_FT_START)
|
||||||
|
self.send(message);
|
||||||
|
|
||||||
|
def handleFTFinish(self, user, buddyName, fileName, size, ftid):
|
||||||
|
room = protocol_pb2.File()
|
||||||
|
room.userName = user
|
||||||
|
room.buddyName = buddyName
|
||||||
|
room.fileName = fileName
|
||||||
|
room.size = size
|
||||||
|
|
||||||
|
# Check later
|
||||||
|
if ftid != 0:
|
||||||
|
room.ftID = ftid
|
||||||
|
|
||||||
|
message = WRAP(room.SerializeToString(), protocol_pb2.WrapperMessage.TYPE_FT_FINISH)
|
||||||
|
self.send(message)
|
||||||
|
|
||||||
|
|
||||||
|
def handleFTData(self, ftID, data):
|
||||||
|
d = protocol_pb2.FileTransferData()
|
||||||
|
d.ftid = ftID
|
||||||
|
d.data = data
|
||||||
|
|
||||||
|
message = WRAP(d.SerializeToString(), protocol_pb2.WrapperMessage.TYPE_FT_DATA);
|
||||||
|
self.send(message)
|
||||||
|
|
||||||
|
def handleBackendConfig(self, section, key, value):
|
||||||
|
c = protocol_pb2.BackendConfig()
|
||||||
|
c.config = "[%s]\n%s = %s\n" % (section, key, value)
|
||||||
|
|
||||||
|
message = WRAP(c.SerializeToString(), protocol_pb2.WrapperMessage.TYPE_BACKEND_CONFIG);
|
||||||
|
self.send(message)
|
||||||
|
|
||||||
|
def handleLoginPayload(self, data):
|
||||||
|
payload = protocol_pb2.Login()
|
||||||
|
if (payload.ParseFromString(data) == False):
|
||||||
|
#TODO: ERROR
|
||||||
|
return
|
||||||
|
self.handleLoginRequest(payload.user, payload.legacyName, payload.password, payload.extraFields)
|
||||||
|
|
||||||
|
def handleLogoutPayload(self, data):
|
||||||
|
payload = protocol_pb2.Logout()
|
||||||
|
if (payload.ParseFromString(data) == False):
|
||||||
|
#TODO: ERROR
|
||||||
|
return
|
||||||
|
self.handleLogoutRequest(payload.user, payload.legacyName)
|
||||||
|
|
||||||
|
def handleStatusChangedPayload(self, data):
|
||||||
|
payload = protocol_pb2.Status()
|
||||||
|
if (payload.ParseFromString(data) == False):
|
||||||
|
#TODO: ERROR
|
||||||
|
return
|
||||||
|
self.handleStatusChangeRequest(payload.userName, payload.status, payload.statusMessage)
|
||||||
|
|
||||||
|
def handleConvMessagePayload(self, data):
|
||||||
|
payload = protocol_pb2.ConversationMessage()
|
||||||
|
if (payload.ParseFromString(data) == False):
|
||||||
|
#TODO: ERROR
|
||||||
|
return
|
||||||
|
self.handleMessageSendRequest(payload.userName, payload.buddyName, payload.message, payload.xhtml)
|
||||||
|
|
||||||
|
def handleAttentionPayload(self, data):
|
||||||
|
payload = protocol_pb2.ConversationMessage()
|
||||||
|
if (payload.ParseFromString(data) == False):
|
||||||
|
#TODO: ERROR
|
||||||
|
return
|
||||||
|
self.handleAttentionRequest(payload.userName, payload.buddyName, payload.message)
|
||||||
|
|
||||||
|
def handleFTStartPayload(self, data):
|
||||||
|
payload = protocol_pb2.File()
|
||||||
|
if (payload.ParseFromString(data) == False):
|
||||||
|
#TODO: ERROR
|
||||||
|
return
|
||||||
|
self.handleFTStartRequest(payload.userName, payload.buddyName, payload.fileName, payload.size, payload.ftID);
|
||||||
|
|
||||||
|
def handleFTFinishPayload(self, data):
|
||||||
|
payload = protocol_pb2.File()
|
||||||
|
if (payload.ParseFromString(data) == False):
|
||||||
|
#TODO: ERROR
|
||||||
|
return
|
||||||
|
self.handleFTFinishRequest(payload.userName, payload.buddyName, payload.fileName, payload.size, payload.ftID)
|
||||||
|
|
||||||
|
def handleFTPausePayload(self, data):
|
||||||
|
payload = protocol_pb2.FileTransferData()
|
||||||
|
if (payload.ParseFromString(data) == False):
|
||||||
|
#TODO: ERROR
|
||||||
|
return
|
||||||
|
self.handleFTPauseRequest(payload.ftID)
|
||||||
|
|
||||||
|
def handleFTContinuePayload(self, data):
|
||||||
|
payload = protocol_pb2.FileTransferData()
|
||||||
|
if (payload.ParseFromString(data) == False):
|
||||||
|
#TODO: ERROR
|
||||||
|
return
|
||||||
|
self.handleFTContinueRequest(payload.ftID)
|
||||||
|
|
||||||
|
def handleJoinRoomPayload(self, data):
|
||||||
|
payload = protocol_pb2.Room()
|
||||||
|
if (payload.ParseFromString(data) == False):
|
||||||
|
#TODO: ERROR
|
||||||
|
return
|
||||||
|
self.handleJoinRoomRequest(payload.userName, payload.room, payload.nickname, payload.password)
|
||||||
|
|
||||||
|
def handleLeaveRoomPayload(self, data):
|
||||||
|
payload = protocol_pb2.Room()
|
||||||
|
if (payload.ParseFromString(data) == False):
|
||||||
|
#TODO: ERROR
|
||||||
|
return
|
||||||
|
self.handleLeaveRoomRequest(payload.userName, payload.room)
|
||||||
|
|
||||||
|
def handleVCardPayload(self, data):
|
||||||
|
payload = protocol_pb2.VCard()
|
||||||
|
if (payload.ParseFromString(data) == False):
|
||||||
|
#TODO: ERROR
|
||||||
|
return
|
||||||
|
if payload.HasField('photo'):
|
||||||
|
self.handleVCardUpdatedRequest(payload.userName, payload.photo, payload.nickname)
|
||||||
|
elif len(payload.buddyName) > 0:
|
||||||
|
self.handleVCardRequest(payload.userName, payload.buddyName, payload.id)
|
||||||
|
|
||||||
|
def handleBuddyChangedPayload(self, data):
|
||||||
|
payload = protocol_pb2.Buddy()
|
||||||
|
if (payload.ParseFromString(data) == False):
|
||||||
|
#TODO: ERROR
|
||||||
|
return
|
||||||
|
if payload.HasField('blocked'):
|
||||||
|
self.handleBuddyBlockToggled(payload.userName, payload.buddyName, payload.blocked)
|
||||||
|
else:
|
||||||
|
groups = [g for g in payload.group]
|
||||||
|
self.handleBuddyUpdatedRequest(payload.userName, payload.buddyName, payload.alias, groups);
|
||||||
|
|
||||||
|
def handleBuddyRemovedPayload(self, data):
|
||||||
|
payload = protocol_pb2.Buddy()
|
||||||
|
if (payload.ParseFromString(data) == False):
|
||||||
|
#TODO: ERROR
|
||||||
|
return
|
||||||
|
groups = [g for g in payload.group]
|
||||||
|
self.handleBuddyRemovedRequest(payload.userName, payload.buddyName, groups);
|
||||||
|
|
||||||
|
def handleChatStatePayload(self, data, msgType):
|
||||||
|
payload = protocol_pb2.Buddy()
|
||||||
|
if (payload.ParseFromString(data) == False):
|
||||||
|
#TODO: ERROR
|
||||||
|
return
|
||||||
|
if msgType == protocol_pb2.WrapperMessage.TYPE_BUDDY_TYPING:
|
||||||
|
self.handleTypingRequest(payload.userName, payload.buddyName)
|
||||||
|
elif msgType == protocol_pb2.WrapperMessage.TYPE_BUDDY_TYPED:
|
||||||
|
self.handleTypedRequest(payload.userName, payload.buddyName)
|
||||||
|
elif msgType == protocol_pb2.WrapperMessage.TYPE_BUDDY_STOPPED_TYPING:
|
||||||
|
self.handleStoppedTypingRequest(payload.userName, payload.buddyName)
|
||||||
|
|
||||||
|
|
||||||
|
def handleDataRead(self, data):
|
||||||
|
self.m_data += data
|
||||||
|
while len(self.m_data) != 0:
|
||||||
|
expected_size = 0
|
||||||
|
if (len(self.m_data) >= 4):
|
||||||
|
expected_size = struct.unpack('!I', self.m_data[0:4])[0]
|
||||||
|
if (len(self.m_data) - 4 < expected_size):
|
||||||
|
return
|
||||||
|
else:
|
||||||
|
return
|
||||||
|
|
||||||
|
wrapper = protocol_pb2.WrapperMessage()
|
||||||
|
if (wrapper.ParseFromString(self.m_data[4:]) == False):
|
||||||
|
self.m_data = self.m_data[expected_size+4:]
|
||||||
|
return
|
||||||
|
|
||||||
|
self.m_data = self.m_data[4+expected_size:]
|
||||||
|
|
||||||
|
if wrapper.type == protocol_pb2.WrapperMessage.TYPE_LOGIN:
|
||||||
|
self.handleLoginPayload(wrapper.payload)
|
||||||
|
elif wrapper.type == protocol_pb2.WrapperMessage.TYPE_LOGOUT:
|
||||||
|
self.handleLogoutPayload(wrapper.payload)
|
||||||
|
elif wrapper.type == protocol_pb2.WrapperMessage.TYPE_PING:
|
||||||
|
self.sendPong()
|
||||||
|
elif wrapper.type == protocol_pb2.WrapperMessage.TYPE_CONV_MESSAGE:
|
||||||
|
self.handleConvMessagePayload(wrapper.payload)
|
||||||
|
elif wrapper.type == protocol_pb2.WrapperMessage.TYPE_JOIN_ROOM:
|
||||||
|
self.handleJoinRoomPayload(wrapper.payload)
|
||||||
|
elif wrapper.type == protocol_pb2.WrapperMessage.TYPE_LEAVE_ROOM:
|
||||||
|
self.handleLeaveRoomPayload(wrapper.payload)
|
||||||
|
elif wrapper.type == protocol_pb2.WrapperMessage.TYPE_VCARD:
|
||||||
|
self.handleVCardPayload(wrapper.payload)
|
||||||
|
elif wrapper.type == protocol_pb2.WrapperMessage.TYPE_BUDDY_CHANGED:
|
||||||
|
self.handleBuddyChangedPayload(wrapper.payload)
|
||||||
|
elif wrapper.type == protocol_pb2.WrapperMessage.TYPE_BUDDY_REMOVED:
|
||||||
|
self.handleBuddyRemovedPayload(wrapper.payload)
|
||||||
|
elif wrapper.type == protocol_pb2.WrapperMessage.TYPE_STATUS_CHANGED:
|
||||||
|
self.handleStatusChangedPayload(wrapper.payload)
|
||||||
|
elif wrapper.type == protocol_pb2.WrapperMessage.TYPE_BUDDY_TYPING:
|
||||||
|
self.handleChatStatePayload(wrapper.payload, protocol_pb2.WrapperMessage.TYPE_BUDDY_TYPING)
|
||||||
|
elif wrapper.type == protocol_pb2.WrapperMessage.TYPE_BUDDY_TYPED:
|
||||||
|
self.handleChatStatePayload(wrapper.payload, protocol_pb2.WrapperMessage.TYPE_BUDDY_TYPED)
|
||||||
|
elif wrapper.type == protocol_pb2.WrapperMessage.TYPE_BUDDY_STOPPED_TYPING:
|
||||||
|
self.handleChatStatePayload(wrapper.payload, protocol_pb2.WrapperMessage.TYPE_BUDDY_STOPPED_TYPING)
|
||||||
|
elif wrapper.type == protocol_pb2.WrapperMessage.TYPE_ATTENTION:
|
||||||
|
self.handleAttentionPayload(wrapper.payload)
|
||||||
|
elif wrapper.type == protocol_pb2.WrapperMessage.TYPE_FT_START:
|
||||||
|
self.handleFTStartPayload(wrapper.payload)
|
||||||
|
elif wrapper.type == protocol_pb2.WrapperMessage.TYPE_FT_FINISH:
|
||||||
|
self.handleFTFinishPayload(wrapper.payload)
|
||||||
|
elif wrapper.type == protocol_pb2.WrapperMessage.TYPE_FT_PAUSE:
|
||||||
|
self.handleFTPausePayload(wrapper.payload)
|
||||||
|
elif wrapper.type == protocol_pb2.WrapperMessage.TYPE_FT_CONTINUE:
|
||||||
|
self.handleFTContinuePayload(wrapper.payload)
|
||||||
|
elif wrapper.type == protocol_pb2.WrapperMessage.TYPE_EXIT:
|
||||||
|
self.handleExitRequest()
|
||||||
|
elif wrapper.type == protocol_pb2.WrapperMessage.TYPE_RAW_XML:
|
||||||
|
self.handleRawXmlRequest(wrapper.payload)
|
||||||
|
|
||||||
|
def send(self, data):
|
||||||
|
header = struct.pack('!I',len(data))
|
||||||
|
self.sendData(header + data)
|
||||||
|
|
||||||
|
def checkPing(self):
|
||||||
|
if (self.m_pingReceived == False):
|
||||||
|
self.handleExitRequest()
|
||||||
|
self.m_pingReceived = False
|
||||||
|
|
||||||
|
|
||||||
|
def sendPong(self):
|
||||||
|
self.m_pingReceived = True
|
||||||
|
wrap = protocol_pb2.WrapperMessage()
|
||||||
|
wrap.type = protocol_pb2.WrapperMessage.TYPE_PONG
|
||||||
|
message = wrap.SerializeToString()
|
||||||
|
self.send(message)
|
||||||
|
self.sendMemoryUsage()
|
||||||
|
|
||||||
|
|
||||||
|
def sendMemoryUsage(self):
|
||||||
|
stats = protocol_pb2.Stats()
|
||||||
|
|
||||||
|
stats.init_res = self.m_init_res
|
||||||
|
res = 0
|
||||||
|
shared = 0
|
||||||
|
|
||||||
|
e_res, e_shared = self.handleMemoryUsage()
|
||||||
|
|
||||||
|
stats.res = res + e_res
|
||||||
|
stats.shared = shared + e_shared
|
||||||
|
stats.id = str(os.getpid())
|
||||||
|
|
||||||
|
message = WRAP(stats.SerializeToString(), protocol_pb2.WrapperMessage.TYPE_STATS)
|
||||||
|
self.send(message)
|
||||||
|
|
||||||
|
|
||||||
|
def handleLoginRequest(self, user, legacyName, password, extra):
|
||||||
|
"""
|
||||||
|
Called when XMPP user wants to connect legacy network.
|
||||||
|
You should connect him to legacy network and call handleConnected or handleDisconnected function later.
|
||||||
|
@param user: XMPP JID of user for which this event occurs.
|
||||||
|
@param legacyName: Legacy network name of this user used for login.
|
||||||
|
@param password: Legacy network password of this user.
|
||||||
|
"""
|
||||||
|
|
||||||
|
#\msc
|
||||||
|
#NetworkPlugin,YourNetworkPlugin,LegacyNetwork;
|
||||||
|
#NetworkPlugin->YourNetworkPlugin [label="handleLoginRequest(...)", URL="\ref NetworkPlugin::handleLoginRequest()"];
|
||||||
|
#YourNetworkPlugin->LegacyNetwork [label="connect the legacy network"];
|
||||||
|
#--- [label="If password was valid and user is connected and logged in"];
|
||||||
|
#YourNetworkPlugin<-LegacyNetwork [label="connected"];
|
||||||
|
#YourNetworkPlugin->NetworkPlugin [label="handleConnected()", URL="\ref NetworkPlugin::handleConnected()"];
|
||||||
|
#--- [label="else"];
|
||||||
|
#YourNetworkPlugin<-LegacyNetwork [label="disconnected"];
|
||||||
|
#YourNetworkPlugin->NetworkPlugin [label="handleDisconnected()", URL="\ref NetworkPlugin::handleDisconnected()"];
|
||||||
|
#\endmsc
|
||||||
|
|
||||||
|
raise NotImplementedError, "Implement me"
|
||||||
|
|
||||||
|
def handleLogoutRequest(self, user, legacyName):
|
||||||
|
"""
|
||||||
|
Called when XMPP user wants to disconnect legacy network.
|
||||||
|
You should disconnect him from legacy network.
|
||||||
|
@param user: XMPP JID of user for which this event occurs.
|
||||||
|
@param legacyName: Legacy network name of this user used for login.
|
||||||
|
"""
|
||||||
|
|
||||||
|
raise NotImplementedError, "Implement me"
|
||||||
|
|
||||||
|
def handleMessageSendRequest(self, user, legacyName, message, xhtml = ""):
|
||||||
|
"""
|
||||||
|
Called when XMPP user sends message to legacy network.
|
||||||
|
@param user: XMPP JID of user for which this event occurs.
|
||||||
|
@param legacyName: Legacy network name of buddy or room.
|
||||||
|
@param message: Plain text message.
|
||||||
|
@param xhtml: XHTML message.
|
||||||
|
"""
|
||||||
|
|
||||||
|
raise NotImplementedError, "Implement me"
|
||||||
|
|
||||||
|
def handleVCardRequest(self, user, legacyName, ID):
|
||||||
|
""" Called when XMPP user requests VCard of buddy.
|
||||||
|
@param user: XMPP JID of user for which this event occurs.
|
||||||
|
@param legacyName: Legacy network name of buddy whose VCard is requested.
|
||||||
|
@param ID: ID which is associated with this request. You have to pass it to handleVCard function when you receive VCard."""
|
||||||
|
|
||||||
|
#\msc
|
||||||
|
#NetworkPlugin,YourNetworkPlugin,LegacyNetwork;
|
||||||
|
#NetworkPlugin->YourNetworkPlugin [label="handleVCardRequest(...)", URL="\ref NetworkPlugin::handleVCardRequest()"];
|
||||||
|
#YourNetworkPlugin->LegacyNetwork [label="start VCard fetching"];
|
||||||
|
#YourNetworkPlugin<-LegacyNetwork [label="VCard fetched"];
|
||||||
|
#YourNetworkPlugin->NetworkPlugin [label="handleVCard()", URL="\ref NetworkPlugin::handleVCard()"];
|
||||||
|
#\endmsc
|
||||||
|
|
||||||
|
pass
|
||||||
|
|
||||||
|
|
||||||
|
def handleVCardUpdatedRequest(self, user, photo, nickname):
|
||||||
|
"""
|
||||||
|
Called when XMPP user updates his own VCard.
|
||||||
|
You should update the VCard in legacy network too.
|
||||||
|
@param user: XMPP JID of user for which this event occurs.
|
||||||
|
@param photo: Raw photo data.
|
||||||
|
"""
|
||||||
|
pass
|
||||||
|
|
||||||
|
def handleJoinRoomRequest(self, user, room, nickname, pasword):
|
||||||
|
pass
|
||||||
|
|
||||||
|
def handleLeaveRoomRequest(self, user, room):
|
||||||
|
pass
|
||||||
|
|
||||||
|
def handleStatusChangeRequest(self, user, status, statusMessage):
|
||||||
|
pass
|
||||||
|
|
||||||
|
def handleBuddyUpdatedRequest(self, user, buddyName, alias, groups):
|
||||||
|
pass
|
||||||
|
|
||||||
|
def handleBuddyRemovedRequest(self, user, buddyName, groups):
|
||||||
|
pass
|
||||||
|
|
||||||
|
def handleBuddyBlockToggled(self, user, buddyName, blocked):
|
||||||
|
pass
|
||||||
|
|
||||||
|
def handleTypingRequest(self, user, buddyName):
|
||||||
|
pass
|
||||||
|
|
||||||
|
def handleTypedRequest(self, user, buddyName):
|
||||||
|
pass
|
||||||
|
|
||||||
|
def handleStoppedTypingRequest(self, user, buddyName):
|
||||||
|
pass
|
||||||
|
|
||||||
|
def handleAttentionRequest(self, user, buddyName, message):
|
||||||
|
pass
|
||||||
|
|
||||||
|
def handleFTStartRequest(self, user, buddyName, fileName, size, ftID):
|
||||||
|
pass
|
||||||
|
|
||||||
|
def handleFTFinishRequest(self, user, buddyName, fileName, size, ftID):
|
||||||
|
pass
|
||||||
|
|
||||||
|
def handleFTPauseRequest(self, ftID):
|
||||||
|
pass
|
||||||
|
|
||||||
|
def handleFTContinueRequest(self, ftID):
|
||||||
|
pass
|
||||||
|
|
||||||
|
def handleMemoryUsage(self):
|
||||||
|
return (0,0)
|
||||||
|
|
||||||
|
def handleExitRequest(self):
|
||||||
|
sys.exit(1)
|
||||||
|
|
||||||
|
def handleRawXmlRequest(self, xml):
|
||||||
|
pass
|
||||||
|
|
||||||
|
def sendData(self, data):
|
||||||
|
pass
|
34
Spectrum2/iochannel.py
Normal file
34
Spectrum2/iochannel.py
Normal file
|
@ -0,0 +1,34 @@
|
||||||
|
import asyncore, socket
|
||||||
|
|
||||||
|
class IOChannel(asyncore.dispatcher):
|
||||||
|
def __init__(self, host, port, callback):
|
||||||
|
asyncore.dispatcher.__init__(self)
|
||||||
|
|
||||||
|
self.create_socket(socket.AF_INET, socket.SOCK_STREAM)
|
||||||
|
self.connect((host, port))
|
||||||
|
|
||||||
|
self.callback = callback
|
||||||
|
self.buffer = ""
|
||||||
|
|
||||||
|
def sendData(self, data):
|
||||||
|
self.buffer += data
|
||||||
|
|
||||||
|
def handle_connect(self):
|
||||||
|
pass
|
||||||
|
|
||||||
|
def handle_close(self):
|
||||||
|
self.close()
|
||||||
|
|
||||||
|
def handle_read(self):
|
||||||
|
data = self.recv(65536)
|
||||||
|
self.callback(data)
|
||||||
|
|
||||||
|
def handle_write(self):
|
||||||
|
sent = self.send(self.buffer)
|
||||||
|
self.buffer = self.buffer[sent:]
|
||||||
|
|
||||||
|
def writable(self):
|
||||||
|
return (len(self.buffer) > 0)
|
||||||
|
|
||||||
|
def readable(self):
|
||||||
|
return True
|
189
Spectrum2/protocol.proto
Normal file
189
Spectrum2/protocol.proto
Normal file
|
@ -0,0 +1,189 @@
|
||||||
|
package pbnetwork;
|
||||||
|
|
||||||
|
enum ConnectionError {
|
||||||
|
CONNECTION_ERROR_NETWORK_ERROR = 0;
|
||||||
|
CONNECTION_ERROR_INVALID_USERNAME = 1;
|
||||||
|
CONNECTION_ERROR_AUTHENTICATION_FAILED = 2;
|
||||||
|
CONNECTION_ERROR_AUTHENTICATION_IMPOSSIBLE = 3;
|
||||||
|
CONNECTION_ERROR_NO_SSL_SUPPORT = 4;
|
||||||
|
CONNECTION_ERROR_ENCRYPTION_ERROR = 5;
|
||||||
|
CONNECTION_ERROR_NAME_IN_USE = 6;
|
||||||
|
CONNECTION_ERROR_INVALID_SETTINGS = 7;
|
||||||
|
CONNECTION_ERROR_CERT_NOT_PROVIDED = 8;
|
||||||
|
CONNECTION_ERROR_CERT_UNTRUSTED = 9;
|
||||||
|
CONNECTION_ERROR_CERT_EXPIRED = 10;
|
||||||
|
CONNECTION_ERROR_CERT_NOT_ACTIVATED = 11;
|
||||||
|
CONNECTION_ERROR_CERT_HOSTNAME_MISMATCH = 12;
|
||||||
|
CONNECTION_ERROR_CERT_FINGERPRINT_MISMATCH = 13;
|
||||||
|
CONNECTION_ERROR_CERT_SELF_SIGNED = 14;
|
||||||
|
CONNECTION_ERROR_CERT_OTHER_ERROR = 15;
|
||||||
|
CONNECTION_ERROR_OTHER_ERROR = 16;
|
||||||
|
}
|
||||||
|
|
||||||
|
enum StatusType {
|
||||||
|
STATUS_ONLINE = 0;
|
||||||
|
STATUS_AWAY = 1;
|
||||||
|
STATUS_FFC = 2;
|
||||||
|
STATUS_XA = 3;
|
||||||
|
STATUS_DND = 4;
|
||||||
|
STATUS_NONE = 5;
|
||||||
|
STATUS_INVISIBLE = 6;
|
||||||
|
}
|
||||||
|
|
||||||
|
message Connected {
|
||||||
|
required string user = 1;
|
||||||
|
}
|
||||||
|
|
||||||
|
message Disconnected {
|
||||||
|
required string user = 1;
|
||||||
|
required int32 error = 2;
|
||||||
|
optional string message = 3;
|
||||||
|
}
|
||||||
|
|
||||||
|
message Login {
|
||||||
|
required string user = 1;
|
||||||
|
required string legacyName = 2;
|
||||||
|
required string password = 3;
|
||||||
|
repeated string extraFields = 4;
|
||||||
|
}
|
||||||
|
|
||||||
|
message Logout {
|
||||||
|
required string user = 1;
|
||||||
|
required string legacyName = 2;
|
||||||
|
}
|
||||||
|
|
||||||
|
message Buddy {
|
||||||
|
required string userName = 1;
|
||||||
|
required string buddyName = 2;
|
||||||
|
optional string alias = 3;
|
||||||
|
repeated string group = 4;
|
||||||
|
optional StatusType status = 5;
|
||||||
|
optional string statusMessage = 6;
|
||||||
|
optional string iconHash = 7;
|
||||||
|
optional bool blocked = 8;
|
||||||
|
}
|
||||||
|
|
||||||
|
message ConversationMessage {
|
||||||
|
required string userName = 1;
|
||||||
|
required string buddyName = 2;
|
||||||
|
required string message = 3;
|
||||||
|
optional string nickname = 4;
|
||||||
|
optional string xhtml = 5;
|
||||||
|
optional string timestamp = 6;
|
||||||
|
optional bool headline = 7;
|
||||||
|
optional string id = 8;
|
||||||
|
optional bool pm = 9;
|
||||||
|
}
|
||||||
|
|
||||||
|
message Room {
|
||||||
|
required string userName = 1;
|
||||||
|
required string nickname = 2;
|
||||||
|
required string room = 3;
|
||||||
|
optional string password = 4;
|
||||||
|
}
|
||||||
|
|
||||||
|
message RoomList {
|
||||||
|
repeated string room = 1;
|
||||||
|
repeated string name = 2;
|
||||||
|
}
|
||||||
|
|
||||||
|
enum ParticipantFlag {
|
||||||
|
PARTICIPANT_FLAG_NONE = 0;
|
||||||
|
PARTICIPANT_FLAG_MODERATOR = 1;
|
||||||
|
PARTICIPANT_FLAG_CONFLICT = 2;
|
||||||
|
PARTICIPANT_FLAG_BANNED = 4;
|
||||||
|
PARTICIPANT_FLAG_NOT_AUTHORIZED = 8;
|
||||||
|
PARTICIPANT_FLAG_ME = 16;
|
||||||
|
PARTICIPANT_FLAG_KICKED = 32;
|
||||||
|
PARTICIPANT_FLAG_ROOM_NOT_FOUND = 64;
|
||||||
|
}
|
||||||
|
|
||||||
|
message Participant {
|
||||||
|
required string userName = 1;
|
||||||
|
required string room = 2;
|
||||||
|
required string nickname = 3;
|
||||||
|
required int32 flag = 4;
|
||||||
|
required StatusType status = 5;
|
||||||
|
optional string statusMessage = 6;
|
||||||
|
optional string newname = 7;
|
||||||
|
}
|
||||||
|
|
||||||
|
message VCard {
|
||||||
|
required string userName = 1;
|
||||||
|
required string buddyName = 2;
|
||||||
|
required int32 id = 3;
|
||||||
|
optional string fullname = 4;
|
||||||
|
optional string nickname = 5;
|
||||||
|
optional bytes photo = 6;
|
||||||
|
}
|
||||||
|
|
||||||
|
message Status {
|
||||||
|
required string userName = 1;
|
||||||
|
required StatusType status = 3;
|
||||||
|
optional string statusMessage = 4;
|
||||||
|
}
|
||||||
|
|
||||||
|
message Stats {
|
||||||
|
required int32 res = 1;
|
||||||
|
required int32 init_res = 2;
|
||||||
|
required int32 shared = 3;
|
||||||
|
required string id = 4;
|
||||||
|
}
|
||||||
|
|
||||||
|
message File {
|
||||||
|
required string userName = 1;
|
||||||
|
required string buddyName = 2;
|
||||||
|
required string fileName = 3;
|
||||||
|
required int32 size = 4;
|
||||||
|
optional int32 ftID = 5;
|
||||||
|
}
|
||||||
|
|
||||||
|
message FileTransferData {
|
||||||
|
required int32 ftID = 1;
|
||||||
|
required bytes data = 2;
|
||||||
|
}
|
||||||
|
|
||||||
|
message BackendConfig {
|
||||||
|
required string config = 1;
|
||||||
|
}
|
||||||
|
|
||||||
|
message WrapperMessage {
|
||||||
|
enum Type {
|
||||||
|
TYPE_CONNECTED = 1;
|
||||||
|
TYPE_DISCONNECTED = 2;
|
||||||
|
TYPE_LOGIN = 3;
|
||||||
|
TYPE_LOGOUT = 4;
|
||||||
|
TYPE_BUDDY_CHANGED = 6;
|
||||||
|
TYPE_BUDDY_REMOVED = 7;
|
||||||
|
TYPE_CONV_MESSAGE = 8;
|
||||||
|
TYPE_PING = 9;
|
||||||
|
TYPE_PONG = 10;
|
||||||
|
TYPE_JOIN_ROOM = 11;
|
||||||
|
TYPE_LEAVE_ROOM = 12;
|
||||||
|
TYPE_PARTICIPANT_CHANGED = 13;
|
||||||
|
TYPE_ROOM_NICKNAME_CHANGED = 14;
|
||||||
|
TYPE_ROOM_SUBJECT_CHANGED = 15;
|
||||||
|
TYPE_VCARD = 16;
|
||||||
|
TYPE_STATUS_CHANGED = 17;
|
||||||
|
TYPE_BUDDY_TYPING = 18;
|
||||||
|
TYPE_BUDDY_STOPPED_TYPING = 19;
|
||||||
|
TYPE_BUDDY_TYPED = 20;
|
||||||
|
TYPE_AUTH_REQUEST = 21;
|
||||||
|
TYPE_ATTENTION = 22;
|
||||||
|
TYPE_STATS = 23;
|
||||||
|
TYPE_FT_START = 24;
|
||||||
|
TYPE_FT_FINISH = 25;
|
||||||
|
TYPE_FT_DATA = 26;
|
||||||
|
TYPE_FT_PAUSE = 27;
|
||||||
|
TYPE_FT_CONTINUE = 28;
|
||||||
|
TYPE_EXIT = 29;
|
||||||
|
TYPE_BACKEND_CONFIG = 30;
|
||||||
|
TYPE_QUERY = 31;
|
||||||
|
TYPE_ROOM_LIST = 32;
|
||||||
|
TYPE_CONV_MESSAGE_ACK = 33;
|
||||||
|
TYPE_RAW_XML = 34;
|
||||||
|
}
|
||||||
|
required Type type = 1;
|
||||||
|
optional bytes payload = 2;
|
||||||
|
}
|
||||||
|
;
|
1245
Spectrum2/protocol_pb2.py
Normal file
1245
Spectrum2/protocol_pb2.py
Normal file
File diff suppressed because one or more lines are too long
8
group.py
Normal file
8
group.py
Normal file
|
@ -0,0 +1,8 @@
|
||||||
|
|
||||||
|
class Group():
|
||||||
|
|
||||||
|
def __init__(self, id, subject, owner):
|
||||||
|
self.id = id
|
||||||
|
self.subject = subject
|
||||||
|
self.owner = owner
|
||||||
|
self.participants = { }
|
90
roster.py
Normal file
90
roster.py
Normal file
|
@ -0,0 +1,90 @@
|
||||||
|
|
||||||
|
import logging
|
||||||
|
|
||||||
|
class Number():
|
||||||
|
|
||||||
|
def __init__(self, number, db):
|
||||||
|
self.number = number
|
||||||
|
self.db = db
|
||||||
|
|
||||||
|
cur = self.db.cursor()
|
||||||
|
cur.execute("SELECT id FROM numbers WHERE number = %s", self.number)
|
||||||
|
if (cur.rowcount):
|
||||||
|
self.id = cur.fetchone()[0]
|
||||||
|
logging.info("sql: found existing number %s (id=%s)", self.number, self.id)
|
||||||
|
else:
|
||||||
|
cur.execute("INSERT INTO numbers (number) VALUES (%s)", (self.number))
|
||||||
|
self.db.commit()
|
||||||
|
self.id = cur.lastrowid
|
||||||
|
logging.info("sql: added new number %s (id=%s)", self.number, self.id)
|
||||||
|
|
||||||
|
def __str__(self):
|
||||||
|
return "%s (id=%s)" % (self.number, self.id)
|
||||||
|
|
||||||
|
|
||||||
|
class Buddy():
|
||||||
|
|
||||||
|
def __init__(self, owner, number, nick, id, db):
|
||||||
|
self.owner = owner
|
||||||
|
self.number = number
|
||||||
|
self.nick = nick
|
||||||
|
self.id = id
|
||||||
|
self.db = db
|
||||||
|
|
||||||
|
def update(self, nick):
|
||||||
|
self.nick = nick
|
||||||
|
cur = self.db.cursor()
|
||||||
|
cur.execute("UPDATE roster SET nick = %s WHERE owner_id = %s AND buddy_id = %s", (self.nick, self.owner.id, self.number.id))
|
||||||
|
self.db.commit()
|
||||||
|
|
||||||
|
def delete(self):
|
||||||
|
cur = self.db.cursor()
|
||||||
|
cur.execute("DELETE FROM roster WHERE owner_id = %s AND buddy_id = %s", (self.owner.id, self.number.id))
|
||||||
|
self.db.commit()
|
||||||
|
self.id = None
|
||||||
|
|
||||||
|
@staticmethod
|
||||||
|
def create(owner, number, nick, db):
|
||||||
|
cur = db.cursor()
|
||||||
|
cur.execute("INSERT INTO roster (owner_id, buddy_id, nick) VALUES (%s, %s, %s)", (owner.id, number.id, nick))
|
||||||
|
db.commit()
|
||||||
|
|
||||||
|
return Buddy(owner, number, nick, cur.lastrowid, db)
|
||||||
|
|
||||||
|
def __str__(self):
|
||||||
|
return "%s (nick=%s, id=%s)" % (self.number, self.nick, self.id)
|
||||||
|
|
||||||
|
class Roster():
|
||||||
|
|
||||||
|
def __init__(self, owner, db):
|
||||||
|
self.db = db
|
||||||
|
self.owner = Number(owner, db)
|
||||||
|
self.buddies = { }
|
||||||
|
|
||||||
|
def load(self):
|
||||||
|
cur = self.db.cursor()
|
||||||
|
cur.execute("""SELECT
|
||||||
|
r.id AS id,
|
||||||
|
nb.number AS number,
|
||||||
|
r.nick AS nick
|
||||||
|
FROM roster AS r
|
||||||
|
LEFT JOIN numbers AS nb
|
||||||
|
ON r.buddy_id = nb.id
|
||||||
|
WHERE
|
||||||
|
r.owner_id = %s""", self.owner.id)
|
||||||
|
|
||||||
|
for i in range(cur.rowcount):
|
||||||
|
id, number, nick, = cur.fetchone()
|
||||||
|
buddy = Buddy(self.owner, Number(number, self.db), nick.decode('latin1'), id, self.db)
|
||||||
|
self.buddies[number] = buddy
|
||||||
|
logging.info("roster load: %s", buddy)
|
||||||
|
|
||||||
|
def add(self, number, nick):
|
||||||
|
buddy = Buddy.create(self.owner, Number(number, self.db), nick, self.db)
|
||||||
|
self.buddies[number] = buddy
|
||||||
|
logging.info("roster add: %s <- %s", self.owner, buddy)
|
||||||
|
|
||||||
|
def remove(self, number):
|
||||||
|
logging.info("roster delete: %s", number)
|
||||||
|
self.buddies[number].delete()
|
||||||
|
del self.buddies[number]
|
257
session.py
Normal file
257
session.py
Normal file
|
@ -0,0 +1,257 @@
|
||||||
|
from Yowsup.connectionmanager import YowsupConnectionManager
|
||||||
|
|
||||||
|
from Spectrum2 import protocol_pb2
|
||||||
|
|
||||||
|
from roster import Roster
|
||||||
|
from threading import Timer
|
||||||
|
from group import Group
|
||||||
|
|
||||||
|
|
||||||
|
import base64
|
||||||
|
import logging
|
||||||
|
import urllib
|
||||||
|
import e4u
|
||||||
|
|
||||||
|
class Session:
|
||||||
|
|
||||||
|
def __init__(self, backend, user, legacyName, password, extra, db):
|
||||||
|
logging.info("new session: %s %s", legacyName, extra)
|
||||||
|
|
||||||
|
password = base64.b64decode(bytes(password.encode('utf-8')))
|
||||||
|
|
||||||
|
e4u.load()
|
||||||
|
|
||||||
|
self.db = db
|
||||||
|
self.backend = backend
|
||||||
|
self.user = user
|
||||||
|
self.status = protocol_pb2.STATUS_NONE
|
||||||
|
self.statusMessage = ''
|
||||||
|
self.legacyName = legacyName
|
||||||
|
self.groups = { }
|
||||||
|
self.timer = None
|
||||||
|
|
||||||
|
self.roster = Roster(legacyName, db)
|
||||||
|
self.frontend = YowsupConnectionManager()
|
||||||
|
|
||||||
|
# Events
|
||||||
|
self.listen("auth_success", self.onAuthSuccess)
|
||||||
|
self.listen("auth_fail", self.onAuthFailed)
|
||||||
|
|
||||||
|
self.listen("contact_typing", self.onContactTyping)
|
||||||
|
self.listen("contact_paused", self.onContactPaused)
|
||||||
|
|
||||||
|
self.listen("presence_updated", self.onPrecenceUpdated)
|
||||||
|
self.listen("presence_available", self.onPrecenceAvailable)
|
||||||
|
self.listen("presence_unavailable", self.onPrecenceUnavailable)
|
||||||
|
|
||||||
|
self.listen("message_received", self.onMessageReceived)
|
||||||
|
self.listen("image_received", self.onMediaReceived)
|
||||||
|
self.listen("video_received", self.onMediaReceived)
|
||||||
|
self.listen("audio_received", self.onMediaReceived)
|
||||||
|
self.listen("location_received", self.onLocationReceived)
|
||||||
|
self.listen("vcard_received", self.onVcardReceived)
|
||||||
|
|
||||||
|
self.listen("group_messageReceived", self.onGroupMessageReceived)
|
||||||
|
self.listen("group_gotInfo", self.onGroupGotInfo)
|
||||||
|
self.listen("group_gotParticipants", self.onGroupGotParticipants)
|
||||||
|
self.listen("group_subjectReceived", self.onGroupSubjectReceived)
|
||||||
|
self.listen("notification_groupParticipantAdded", self.onGroupParticipantAdded)
|
||||||
|
self.listen("notification_groupParticipantRemoved", self.onGroupParticipantRemoved)
|
||||||
|
self.call("auth_login", (legacyName, password))
|
||||||
|
|
||||||
|
def __del__(self):
|
||||||
|
self.call("disconnect", ("logout",))
|
||||||
|
|
||||||
|
def _softToUni(self, message):
|
||||||
|
message = unicode(message, "utf-8")
|
||||||
|
return e4u.translate(message, reverse=False, **e4u.SOFTBANK_TRANSLATE_PROFILE)
|
||||||
|
|
||||||
|
def call(self, method, args = ()):
|
||||||
|
self.frontend.methodInterface.call(method, args)
|
||||||
|
|
||||||
|
def listen(self, event, callback):
|
||||||
|
self.frontend.signalInterface.registerListener(event, callback)
|
||||||
|
|
||||||
|
# RequestMethods
|
||||||
|
def sendTypingStarted(self, buddy):
|
||||||
|
logging.info("started typing: %s to %s", self.legacyName, buddy)
|
||||||
|
self.call("typing_send", (buddy + "@s.whatsapp.net",))
|
||||||
|
|
||||||
|
def sendTypingStopped(self, buddy):
|
||||||
|
logging.info("stopped typing: %s to %s", self.legacyName, buddy)
|
||||||
|
self.call("typing_paused", (buddy + "@s.whatsapp.net",))
|
||||||
|
|
||||||
|
def sendMessage(self, sender, message):
|
||||||
|
logging.info("message send to %s: %s", sender, message)
|
||||||
|
message = message.encode("utf-8")
|
||||||
|
|
||||||
|
if ("-" in sender): # group msg
|
||||||
|
if ("/" in sender):
|
||||||
|
room, buddy = sender.split("/")
|
||||||
|
self.call("message_send", (buddy + "@s.whatsapp.net", message))
|
||||||
|
else:
|
||||||
|
room = sender
|
||||||
|
self.backend.handleMessage(self.user, room, message, "me") # TODO
|
||||||
|
self.call("message_send", (room + "@g.us", message))
|
||||||
|
else: # private msg
|
||||||
|
buddy = sender
|
||||||
|
self.call("message_send", (buddy + "@s.whatsapp.net", message))
|
||||||
|
|
||||||
|
def changeStatus(self, status):
|
||||||
|
if (status == self.status): return
|
||||||
|
|
||||||
|
logging.info("status changed: %s", status)
|
||||||
|
self.status = status
|
||||||
|
if (status == protocol_pb2.STATUS_ONLINE):
|
||||||
|
self.call("presence_sendAvailable")
|
||||||
|
elif (status == protocol_pb2.STATUS_FFC):
|
||||||
|
self.call("presence_sendAvailableForChat")
|
||||||
|
else:
|
||||||
|
self.call("presence_sendUnavailable")
|
||||||
|
|
||||||
|
def changeStatusMessage(self, statusMessage):
|
||||||
|
if (statusMessage == self.statusMessage): return
|
||||||
|
|
||||||
|
logging.info("status message changed: %s", statusMessage)
|
||||||
|
self.statusMessage = statusMessage
|
||||||
|
self.call("profile_setStatus", (statusMessage.encode("utf-8"),))
|
||||||
|
|
||||||
|
def updateBuddy(self, buddy, nick, groups):
|
||||||
|
if self.roster.buddies.has_key(buddy):
|
||||||
|
buddy = self.roster.buddies[buddy]
|
||||||
|
buddy.update(nick)
|
||||||
|
logging.info("buddy renamed: %s", buddy)
|
||||||
|
self.call("presence_request", (buddy.number.number + "@s.whatsapp.net",))
|
||||||
|
else:
|
||||||
|
self.roster.add(buddy, nick)
|
||||||
|
self.call("presence_request", (buddy + "@s.whatsapp.net",))
|
||||||
|
self.backend.handleBuddyChanged(self.user, buddy, nick, [], protocol_pb2.STATUS_NONE)
|
||||||
|
|
||||||
|
def removeBuddy(self, buddy):
|
||||||
|
self.roster.remove(buddy)
|
||||||
|
|
||||||
|
def joinRoom(self, room):
|
||||||
|
self.call("group_getParticipants", (room + "@g.us",))
|
||||||
|
|
||||||
|
# EventHandlers
|
||||||
|
def onAuthSuccess(self, user):
|
||||||
|
logging.info("auth success: %s", user)
|
||||||
|
self.backend.handleConnected(self.user)
|
||||||
|
self.call("ready")
|
||||||
|
|
||||||
|
self.call("group_getGroups", ("participating",))
|
||||||
|
|
||||||
|
self.roster.load()
|
||||||
|
for number, buddy in self.roster.buddies.iteritems():
|
||||||
|
logging.info("request presence: %s", buddy)
|
||||||
|
self.call("presence_request", (buddy.number.number + "@s.whatsapp.net",))
|
||||||
|
self.backend.handleBuddyChanged(self.user, buddy.number.number, buddy.nick, [], protocol_pb2.STATUS_NONE)
|
||||||
|
|
||||||
|
def onAuthFailed(self, user, reason):
|
||||||
|
logging.info("auth failed: %s (%s)", user, reason)
|
||||||
|
self.backend.handleDisconnected(self.user, 0, reason)
|
||||||
|
|
||||||
|
def onMessageReceived(self, messageId, jid, messageContent, timestamp, receiptRequested, pushName, isBroadCast):
|
||||||
|
buddy = jid.split("@")[0]
|
||||||
|
logging.info("message received from %s: %s", buddy, messageContent)
|
||||||
|
self.backend.handleMessage(self.user, buddy, self._softToUni(messageContent), timestamp=timestamp)
|
||||||
|
if receiptRequested: self.call("message_ack", (jid, messageId))
|
||||||
|
|
||||||
|
def onMediaReceived(self, messageId, jid, preview, url, size, receiptRequested, isBroadcast):
|
||||||
|
buddy = jid.split("@")[0]
|
||||||
|
logging.info("message received from %s: %s", buddy, url)
|
||||||
|
self.backend.handleMessage(self.user, buddy, url)
|
||||||
|
if receiptRequested: self.call("message_ack", (jid, messageId))
|
||||||
|
|
||||||
|
def onLocationReceived(self, messageId, jid, name, preview, latitude, longitude, receiptRequested, isBroadcast):
|
||||||
|
buddy = jid.split("@")[0]
|
||||||
|
logging.info("location received from %s: %s, %s", buddy, latitude, longitude)
|
||||||
|
self.backend.handleMessage(self.user, buddy, "http://maps.google.de?%s" % urllib.urlencode({ "q": "%s %s" % (latitude, longitude) }))
|
||||||
|
if receiptRequested: self.call("message_ack", (jid, messageId))
|
||||||
|
|
||||||
|
def onVcardReceived(self, messageId, jid, name, data, receiptRequested, isBroadcast): # TODO
|
||||||
|
buddy = jid.split("@")[0]
|
||||||
|
logging.info("vcard received from %s", buddy)
|
||||||
|
self.backend.handleMessage(self.user, buddy, "Received VCard (not implemented yet)")
|
||||||
|
if receiptRequested: self.call("message_ack", (jid, messageId))
|
||||||
|
|
||||||
|
def onContactTyping(self, jid):
|
||||||
|
buddy = jid.split("@")[0]
|
||||||
|
logging.info("started typing: %s", buddy)
|
||||||
|
self.backend.handleBuddyTyping(self.user, buddy)
|
||||||
|
if (self.timer != None): self.timer.cancel()
|
||||||
|
|
||||||
|
def onContactPaused(self, jid):
|
||||||
|
buddy = jid.split("@")[0]
|
||||||
|
logging.info("paused typing: %s", buddy)
|
||||||
|
self.backend.handleBuddyTyped(self.user, jid.split("@")[0])
|
||||||
|
self.timer = Timer(3, self.backend.handleBuddyStoppedTyping, (self.user, buddy)).start()
|
||||||
|
|
||||||
|
def onPrecenceUpdated(self, jid, lastseen):
|
||||||
|
buddy = jid.split("@")[0]
|
||||||
|
logging.info("lastseen: %s %d secs ago", buddy, lastseen)
|
||||||
|
if (lastseen < 60): self.onPrecenceAvailable(jid)
|
||||||
|
else: self.onPrecenceUnavailable(jid)
|
||||||
|
|
||||||
|
def onPrecenceAvailable(self, jid):
|
||||||
|
buddy = jid.split("@")[0]
|
||||||
|
if (self.roster.buddies.has_key(buddy)):
|
||||||
|
buddy = self.roster.buddies[buddy]
|
||||||
|
logging.info("is available: %s", buddy)
|
||||||
|
self.backend.handleBuddyChanged(self.user, buddy.number.number, buddy.nick, [], protocol_pb2.STATUS_ONLINE)
|
||||||
|
|
||||||
|
def onPrecenceUnavailable(self, jid):
|
||||||
|
buddy = jid.split("@")[0]
|
||||||
|
if (self.roster.buddies.has_key(buddy)):
|
||||||
|
buddy = self.roster.buddies[buddy]
|
||||||
|
logging.info("is unavailable: %s", buddy)
|
||||||
|
self.backend.handleBuddyChanged(self.user, buddy.number.number, buddy.nick, [], protocol_pb2.STATUS_XA)
|
||||||
|
|
||||||
|
def onGroupGotInfo(self, gjid, owner, subject, subjectOwner, subjectTimestamp, creationTimestamp):
|
||||||
|
room = gjid.split("@")[0]
|
||||||
|
|
||||||
|
if self.groups.has_key(room):
|
||||||
|
room = self.groups[room]
|
||||||
|
room.owner = owner
|
||||||
|
room.subject = subject
|
||||||
|
else:
|
||||||
|
self.groups = Group(room, subject, owner)
|
||||||
|
|
||||||
|
self.backend.handleRoomList([[room, subject]])
|
||||||
|
|
||||||
|
def onGroupGotParticipants(self, gjid, jids):
|
||||||
|
room = gjid.split("@")[0]
|
||||||
|
|
||||||
|
for jid in jids:
|
||||||
|
buddy = jid.split("@")[0]
|
||||||
|
logging.info("added %s to room %s", buddy, room)
|
||||||
|
self.backend.handleParticipantChanged(self.user, buddy, room, protocol_pb2.PARTICIPANT_FLAG_NONE, protocol_pb2.STATUS_ONLINE)
|
||||||
|
|
||||||
|
self.backend.handleParticipantChanged(self.user, self.legacyName, room, protocol_pb2.PARTICIPANT_FLAG_ME, protocol_pb2.STATUS_ONLINE)
|
||||||
|
# TODO check status and moderator
|
||||||
|
|
||||||
|
def onGroupMessageReceived(self, messageId, gjid, jid, messageContent, timestamp, receiptRequested, pushName):
|
||||||
|
buddy = jid.split("@")[0]
|
||||||
|
room = gjid.split("@")[0]
|
||||||
|
|
||||||
|
logging.info("group message received in %s from %s: %s", room, buddy, messageContent)
|
||||||
|
self.backend.handleMessage(self.user, room, self._softToUni(messageContent), buddy, timestamp=timestamp)
|
||||||
|
if receiptRequested: self.call("message_ack", (gjid, messageId))
|
||||||
|
|
||||||
|
def onGroupSubjectReceived(self, messageId, gjid, jid, subject, timestamp, receiptRequested):
|
||||||
|
room = gjid.split("@")[0]
|
||||||
|
buddy = jid.split("@")[0]
|
||||||
|
|
||||||
|
self.backend.handleSubject(self.user, room, subject, buddy)
|
||||||
|
if receiptRequested: self.call("subject_ack", (gjid, messageId))
|
||||||
|
|
||||||
|
def onGroupParticipantAdded(self, gjid, jid, author, timestamp, messageId, receiptRequested):
|
||||||
|
room = gjid.split("@")[0]
|
||||||
|
buddy = jid.split("@")[0]
|
||||||
|
|
||||||
|
self.backend.handleParticipantChanged(self.user, buddy, room, protocol_pb2.PARTICIPANT_FLAG_NONE, protocol_pb2.STATUS_ONLINE)
|
||||||
|
# if receiptRequested: self.call("message_ack", (gjid, messageId))
|
||||||
|
|
||||||
|
def onGroupParticipantRemoved(gjid, jid, author, timestamp, messageId, receiptRequested):
|
||||||
|
self.backend.handleParticipantChanged(self.user, buddy, room, protocol_pb2.PARTICIPANT_FLAG_NONE, protocol_pb2.STATUS_NONE) # TODO
|
||||||
|
# if receiptRequested: self.call("message_ack", (gjid, messageId))
|
89
whatsappbackend.py
Normal file
89
whatsappbackend.py
Normal file
|
@ -0,0 +1,89 @@
|
||||||
|
from Spectrum2.backend import SpectrumBackend
|
||||||
|
from Spectrum2 import protocol_pb2
|
||||||
|
|
||||||
|
from session import Session
|
||||||
|
|
||||||
|
import logging
|
||||||
|
|
||||||
|
class WhatsAppBackend(SpectrumBackend):
|
||||||
|
def __init__(self, io, db):
|
||||||
|
SpectrumBackend.__init__(self)
|
||||||
|
self.io = io
|
||||||
|
self.db = db
|
||||||
|
self.sessions = { }
|
||||||
|
|
||||||
|
# self.handleBackendConfig("features", "muc", "true")
|
||||||
|
# self.handleBackendConfig("features", "rawxml", "true")
|
||||||
|
|
||||||
|
logging.info("initialized backend")
|
||||||
|
|
||||||
|
# RequestsHandlers
|
||||||
|
def handleLoginRequest(self, user, legacyName, password, extra):
|
||||||
|
self.sessions[user] = Session(self, user, legacyName, password, extra, self.db)
|
||||||
|
|
||||||
|
def handleLogoutRequest(self, user, legacyName):
|
||||||
|
del self.sessions[user]
|
||||||
|
|
||||||
|
def handleMessageSendRequest(self, user, buddy, message, xhtml = ""):
|
||||||
|
self.sessions[user].sendMessage(buddy, message)
|
||||||
|
|
||||||
|
def handleVCardRequest(self, user, buddy, ID):
|
||||||
|
# TODO
|
||||||
|
pass
|
||||||
|
|
||||||
|
def handleVCardUpdatedRequest(self, user, photo, nickname):
|
||||||
|
# TODO
|
||||||
|
pass
|
||||||
|
|
||||||
|
def handleJoinRoomRequest(self, user, room, nickname, pasword):
|
||||||
|
self.sessions[user].joinRoom(room)
|
||||||
|
|
||||||
|
def handleLeaveRoomRequest(self, user, room):
|
||||||
|
pass
|
||||||
|
|
||||||
|
def handleStatusChangeRequest(self, user, status, statusMessage):
|
||||||
|
if (len(statusMessage)):
|
||||||
|
self.sessions[user].changeStatusMessage(statusMessage)
|
||||||
|
self.sessions[user].changeStatus(status)
|
||||||
|
|
||||||
|
def handleBuddyUpdatedRequest(self, user, buddy, nick, groups):
|
||||||
|
self.sessions[user].updateBuddy(buddy, nick, groups)
|
||||||
|
|
||||||
|
def handleBuddyRemovedRequest(self, user, buddy, groups):
|
||||||
|
self.sessions[user].removeBuddy(buddy)
|
||||||
|
|
||||||
|
def handleBuddyBlockToggled(self, user, buddy, blocked):
|
||||||
|
pass
|
||||||
|
|
||||||
|
def handleTypingRequest(self, user, buddy):
|
||||||
|
self.sessions[user].sendTypingStarted(buddy)
|
||||||
|
|
||||||
|
def handleTypedRequest(self, user, buddy):
|
||||||
|
self.sessions[user].sendTypingStopped(buddy)
|
||||||
|
|
||||||
|
def handleStoppedTypingRequest(self, user, buddy):
|
||||||
|
self.sessions[user].sendTypingStopped(buddy)
|
||||||
|
|
||||||
|
def handleAttentionRequest(self, user, buddy, message):
|
||||||
|
# TODO
|
||||||
|
pass
|
||||||
|
|
||||||
|
def handleFTStartRequest(self, user, buddy, fileName, size, ftID):
|
||||||
|
pass
|
||||||
|
|
||||||
|
def handleFTFinishRequest(self, user, buddy, fileName, size, ftID):
|
||||||
|
pass
|
||||||
|
|
||||||
|
def handleFTPauseRequest(self, ftID):
|
||||||
|
pass
|
||||||
|
|
||||||
|
def handleFTContinueRequest(self, ftID):
|
||||||
|
pass
|
||||||
|
|
||||||
|
def handleRawXmlRequest(self, xml):
|
||||||
|
pass
|
||||||
|
|
||||||
|
def sendData(self, data):
|
||||||
|
self.io.sendData(data)
|
||||||
|
|
||||||
|
|
Loading…
Reference in a new issue