transpub/transWhat/session.py

831 lines
28 KiB
Python
Raw Normal View History

2013-06-28 20:45:38 +00:00
import utils
import logging
import urllib
import time
2013-05-31 18:53:21 +00:00
2016-08-10 13:14:46 +00:00
# from PIL import Image
import sys
import os
2017-03-08 17:26:57 +00:00
reload(sys)
sys.setdefaultencoding("utf-8")
from yowsup.layers.protocol_media.mediauploader import MediaUploader
from yowsup.layers.protocol_media.mediadownloader import MediaDownloader
2013-05-31 18:53:21 +00:00
from Spectrum2 import protocol_pb2
2013-06-28 20:45:38 +00:00
from buddy import BuddyList
2013-05-31 18:53:21 +00:00
from threading import Timer
from group import Group
2013-06-28 20:45:38 +00:00
from bot import Bot
2016-01-01 15:31:17 +00:00
import deferred
2016-01-02 14:00:47 +00:00
from deferred import call
from yowsupwrapper import YowsupApp
2013-05-31 18:53:21 +00:00
class MsgIDs:
2016-04-06 20:02:40 +00:00
def __init__(self, xmppId, waId):
self.xmppId = xmppId
self.waId = waId
self.cnt = 0
class Session(YowsupApp):
2016-04-09 22:21:58 +00:00
broadcast_prefix = '\U0001F4E2 '
2013-05-31 18:53:21 +00:00
2015-12-02 21:06:56 +00:00
def __init__(self, backend, user, legacyName, extra):
super(Session, self).__init__()
2013-06-28 20:46:56 +00:00
self.logger = logging.getLogger(self.__class__.__name__)
2016-04-09 22:21:58 +00:00
self.logger.info("Created: %s" % legacyName)
2013-05-31 18:53:21 +00:00
self.backend = backend
self.user = user
2013-06-28 20:46:56 +00:00
self.legacyName = legacyName
2013-05-31 18:53:21 +00:00
self.status = protocol_pb2.STATUS_NONE
self.statusMessage = ''
2013-06-28 20:46:56 +00:00
self.groups = {}
self.gotGroupList = False
# Functions to exectute when logged in via yowsup
self.loginQueue = []
self.joinRoomQueue = []
2013-06-28 20:46:56 +00:00
self.presenceRequested = []
self.offlineQueue = []
2015-09-15 19:29:26 +00:00
self.msgIDs = { }
2013-06-28 20:46:56 +00:00
self.groupOfflineQueue = { }
self.loggedIn = False
2016-02-19 16:31:34 +00:00
self.recvMsgIDs = []
2013-06-28 20:46:56 +00:00
2013-05-31 18:53:21 +00:00
self.timer = None
2013-06-28 20:46:56 +00:00
self.password = None
self.initialized = False
self.lastMsgId = None
2015-09-15 19:29:26 +00:00
self.synced = False
2013-05-31 18:53:21 +00:00
2015-12-01 21:21:38 +00:00
self.buddies = BuddyList(self.legacyName, self.backend, self.user, self)
2013-06-28 20:46:56 +00:00
self.bot = Bot(self)
self.imgMsgId = None
self.imgPath = ""
self.imgBuddy = None
self.imgType = ""
2013-06-28 20:46:56 +00:00
def __del__(self): # handleLogoutRequest
self.logout()
2013-05-31 18:53:21 +00:00
2013-06-28 20:46:56 +00:00
def logout(self):
2016-04-09 22:21:58 +00:00
self.logger.info("%s logged out" % self.user)
super(Session, self).logout()
self.loggedIn = False
2013-06-28 20:46:56 +00:00
def login(self, password):
2016-04-09 22:21:58 +00:00
self.logger.info("%s attempting login" % self.user)
self.password = password
2015-09-15 19:29:26 +00:00
self.shouldBeConncted = True
super(Session, self).login(self.legacyName, self.password)
2013-06-28 20:46:56 +00:00
def _shortenGroupId(self, gid):
# FIXME: might have problems if number begins with 0
return gid
# return '-'.join(hex(int(s))[2:] for s in gid.split('-'))
def _lengthenGroupId(self, gid):
return gid
# FIXME: might have problems if number begins with 0
# return '-'.join(str(int(s, 16)) for s in gid.split('-'))
2013-06-28 20:46:56 +00:00
def updateRoomList(self):
rooms = []
text = []
2013-06-28 20:46:56 +00:00
for room, group in self.groups.iteritems():
rooms.append([self._shortenGroupId(room), group.subject])
text.append(self._shortenGroupId(room) + '@' + self.backend.spectrum_jid + ' :' + group.subject)
2015-09-15 19:29:26 +00:00
2016-04-09 22:21:58 +00:00
self.logger.debug("Got rooms: %s" % rooms)
2013-06-28 20:46:56 +00:00
self.backend.handleRoomList(rooms)
2016-04-09 22:21:58 +00:00
message = "Note, you are a participant of the following groups:\n" + \
"\n".join(text) + "\nIf you do not join them you will lose messages"
2015-11-10 13:00:38 +00:00
#self.bot.send(message)
2013-06-28 20:46:56 +00:00
2017-07-02 14:37:08 +00:00
def _updateGroups(self, response, _):
2016-04-09 22:21:58 +00:00
self.logger.debug('Received groups list %s' % response)
2015-09-06 12:06:16 +00:00
groups = response.getGroups()
for group in groups:
room = group.getId()
2017-07-02 14:37:08 +00:00
# ensure self.groups[room] exists
if room not in self.groups:
owner = group.getOwner().split('@')[0]
subjectOwner = group.getSubjectOwner().split('@')[0]
subject = group.getSubject()
2017-07-02 14:37:08 +00:00
self.groups[room] = Group(room, owner, subject, subjectOwner,
self.backend, self.user)
# add/update room participants
self.groups[room].addParticipants(group.getParticipants().keys(),
2017-07-02 14:37:08 +00:00
self.buddies, self.legacyName)
self.gotGroupList = True
2017-07-02 14:37:08 +00:00
# join rooms
while self.joinRoomQueue:
self.joinRoom(*self.joinRoomQueue.pop(0))
# deliver queued offline messages
for room in self.groupOfflineQueue:
while self.groupOfflineQueue[room]:
msg = self.groupOfflineQueue[room].pop(0)
self.backend.handleMessage(self.user, room, msg[1], msg[0], "",
msg[2])
self.logger.debug("Send queued group message to: %s %s %s" %
(msg[0], msg[1], msg[2]))
# pass update to backend
2015-09-15 19:29:26 +00:00
self.updateRoomList()
def joinRoom(self, room, nick):
if not self.gotGroupList:
self.joinRoomQueue.append((room, nick))
return
room = self._lengthenGroupId(room)
2015-09-15 19:29:26 +00:00
if room in self.groups:
2016-04-09 22:21:58 +00:00
self.logger.info("Joining room: %s room=%s, nick=%s" %
(self.legacyName, room, nick))
2015-09-06 12:06:16 +00:00
2015-09-15 19:29:26 +00:00
group = self.groups[room]
group.joined = True
2015-09-15 19:29:26 +00:00
group.nick = nick
group.participants[self.legacyName] = nick
2015-09-15 19:29:26 +00:00
try:
ownerNick = group.participants[group.subjectOwner]
2015-09-15 19:29:26 +00:00
except KeyError:
ownerNick = group.subjectOwner
group.sendParticipantsToSpectrum(self.legacyName)
self.backend.handleSubject(self.user, self._shortenGroupId(room),
group.subject, ownerNick)
2016-04-09 22:21:58 +00:00
self.logger.debug("Room subject: room=%s, subject=%s" %
(room, group.subject))
self.backend.handleRoomNicknameChanged(
self.user, self._shortenGroupId(room), group.subject
)
2015-09-15 19:29:26 +00:00
else:
2016-04-09 22:21:58 +00:00
self.logger.warn("Room doesn't exist: %s" % room)
2015-09-06 12:06:16 +00:00
def leaveRoom(self, room):
if room in self.groups:
2016-04-09 22:21:58 +00:00
self.logger.info("Leaving room: %s room=%s" % (self.legacyName, room))
group = self.groups[room]
group.joined = False
else:
2016-04-09 22:21:58 +00:00
self.logger.warn("Room doesn't exist: %s. Unable to leave." % room)
def _lastSeen(self, number, seconds):
2016-04-09 22:21:58 +00:00
self.logger.debug("Last seen %s at %s seconds" % (number, seconds))
if seconds < 60:
self.onPresenceAvailable(number)
else:
self.onPresenceUnavailable(number)
2016-02-19 16:31:34 +00:00
def sendReadReceipts(self, buddy):
for _id, _from, participant, t in self.recvMsgIDs:
2016-02-19 17:12:47 +00:00
if _from.split('@')[0] == buddy:
self.sendReceipt(_id, _from, 'read', participant)
self.recvMsgIDs.remove((_id, _from, participant, t))
2016-02-19 16:31:34 +00:00
self.logger.debug("Send read receipt to %s (ID: %s)", _from, _id)
# Called by superclass
def onAuthSuccess(self, status, kind, creation,
expiration, props, nonce, t):
2016-04-09 22:21:58 +00:00
self.logger.info("Auth success: %s" % self.user)
self.backend.handleConnected(self.user)
self.backend.handleBuddyChanged(self.user, "bot", self.bot.name,
2016-04-09 22:21:58 +00:00
["Admin"], protocol_pb2.STATUS_ONLINE)
# Initialisation?
self.requestPrivacyList()
self.requestClientConfig()
self.requestServerProperties()
# ?
self.logger.debug('Requesting groups list')
self.requestGroupsList(self._updateGroups)
# self.requestBroadcastList()
# This should handle, sync, statuses, and presence
self.sendPresence(True)
for func in self.loginQueue:
func()
2015-09-15 19:29:26 +00:00
if self.initialized == False:
self.sendOfflineMessages()
#self.bot.call("welcome")
2015-09-15 19:29:26 +00:00
self.initialized = True
self.loggedIn = True
2015-09-15 19:29:26 +00:00
# Called by superclass
def onAuthFailed(self, reason):
2016-04-09 22:21:58 +00:00
self.logger.info("Auth failed: %s (%s)" % (self.user, reason))
self.backend.handleDisconnected(self.user, 0, reason)
self.password = None
self.loggedIn = False
# Called by superclass
def onDisconnect(self):
self.logger.debug('Disconnected')
self.backend.handleDisconnected(self.user, 0, 'Disconnected for unknown reasons')
# Called by superclass
def onReceipt(self, _id, _from, timestamp, type, participant, offline, items):
2016-04-09 22:21:58 +00:00
self.logger.debug("received receipt, sending ack: %s" %
[ _id, _from, timestamp, type, participant, offline, items ]
)
try:
2016-01-04 08:14:02 +00:00
number = _from.split('@')[0]
self.backend.handleMessageAck(self.user, number, self.msgIDs[_id].xmppId)
self.msgIDs[_id].cnt = self.msgIDs[_id].cnt + 1
if self.msgIDs[_id].cnt == 2:
del self.msgIDs[_id]
except KeyError:
2016-04-09 22:21:58 +00:00
self.logger.error("Message %s not found. Unable to send ack" % _id)
# Called by superclass
def onAck(self, _id, _class, _from, timestamp):
2016-04-09 22:21:58 +00:00
self.logger.debug('received ack: %s' % [ _id, _class, _from, timestamp ])
# Called by superclass
2015-09-15 19:29:26 +00:00
def onTextMessage(self, _id, _from, to, notify, timestamp, participant,
offline, retry, body):
buddy = _from.split('@')[0]
messageContent = body
self.sendReceipt(_id, _from, None, participant)
self.recvMsgIDs.append((_id, _from, participant, timestamp))
2016-04-09 22:21:58 +00:00
self.logger.info("Message received from %s to %s: %s (at ts=%s)" %
(buddy, self.legacyName, messageContent, timestamp))
if participant is not None: # Group message or broadcast
partname = participant.split('@')[0]
if _from.split('@')[1] == 'broadcast': # Broadcast message
message = self.broadcast_prefix + messageContent
self.sendMessageToXMPP(partname, message, timestamp)
else: # Group message
if notify is None:
notify = ""
self.sendGroupMessageToXMPP(buddy, partname, messageContent,
timestamp, notify)
else:
self.sendMessageToXMPP(buddy, messageContent, timestamp)
# Called by superclass
def onImage(self, image):
if image.caption is None:
image.caption = ''
2017-03-08 17:26:05 +00:00
self.onMedia(image, "image")
2016-02-19 16:31:34 +00:00
# Called by superclass
def onAudio(self, audio):
2017-03-08 17:26:05 +00:00
self.onMedia(audio, "audio")
2016-02-19 16:31:34 +00:00
# Called by superclass
def onVideo(self, video):
2017-03-08 17:26:05 +00:00
self.onMedia(video, "video")
def onMedia(self, media, type):
self.logger.debug('Received %s message: %s' % (type, media))
buddy = media._from.split('@')[0]
participant = media.participant
caption = ''
2017-03-08 17:26:05 +00:00
if media.isEncrypted():
self.logger.debug('Received encrypted media message')
if self.backend.specConf is not None and self.backend.specConf.__getitem__("service.web_directory") is not None and self.backend.specConf.__getitem__("service.web_url") is not None :
ipath = "/" + str(media.timestamp) + media.getExtension()
with open(self.backend.specConf.__getitem__("service.web_directory") + ipath,"wb") as f:
f.write(media.getMediaContent())
url = self.backend.specConf.__getitem__("service.web_url") + ipath
else:
self.logger.warn('Received encrypted media: web storage not set in config!')
url = media.url
else:
url = media.url
if type == 'image':
caption = media.caption
if participant is not None: # Group message
partname = participant.split('@')[0]
2017-03-08 17:26:05 +00:00
if media._from.split('@')[1] == 'broadcast': # Broadcast message
self.sendMessageToXMPP(partname, self.broadcast_prefix, media.timestamp)
self.sendMessageToXMPP(partname, url, media.timestamp)
2017-06-06 20:24:03 +00:00
self.sendMessageToXMPP(partname, caption, media.timestamp)
else: # Group message
2017-03-08 17:26:05 +00:00
self.sendGroupMessageToXMPP(buddy, partname, url, media.timestamp)
2017-06-06 20:24:03 +00:00
self.sendGroupMessageToXMPP(buddy, partname, caption, media.timestamp)
else:
2017-03-08 17:26:05 +00:00
self.sendMessageToXMPP(buddy, url, media.timestamp)
2017-06-06 20:24:03 +00:00
self.sendMessageToXMPP(buddy, caption, media.timestamp)
2017-03-08 17:26:05 +00:00
self.sendReceipt(media._id, media._from, None, media.participant)
self.recvMsgIDs.append((media._id, media._from, media.participant, media.timestamp))
2015-09-15 19:29:26 +00:00
def onLocation(self, location):
buddy = location._from.split('@')[0]
2015-09-15 19:29:26 +00:00
latitude = location.getLatitude()
longitude = location.getLongitude()
url = location.getLocationURL()
2016-04-06 20:02:40 +00:00
participant = location.participant
latlong = 'geo:' + latitude + ',' + longitude
2015-09-15 19:29:26 +00:00
self.logger.debug("Location received from %s: %s, %s", (buddy, latitude, longitude))
if participant is not None: # Group message
partname = participant.split('@')[0]
if location._from.split('@')[1] == 'broadcast': # Broadcast message
self.sendMessageToXMPP(partname, self.broadcast_prefix, location.timestamp)
if url is not None:
self.sendMessageToXMPP(partname, url, location.timestamp)
self.sendMessageToXMPP(partname, latlong, location.timestamp)
else: # Group message
if url is not None:
self.sendGroupMessageToXMPP(buddy, partname, url, location.timestamp)
self.sendGroupMessageToXMPP(buddy, partname, latlong, location.timestamp)
else:
if url is not None:
self.sendMessageToXMPP(buddy, url, location.timestamp)
self.sendMessageToXMPP(buddy, latlong, location.timestamp)
self.sendReceipt(location._id, location._from, None, location.participant)
self.recvMsgIDs.append((location._id, location._from, location.participant, location.timestamp))
2016-02-19 16:31:34 +00:00
2015-09-15 19:29:26 +00:00
# Called by superclass
def onVCard(self, _id, _from, name, card_data, to, notify, timestamp, participant):
2016-04-09 22:21:58 +00:00
self.logger.debug('received VCard: %s' %
[ _id, _from, name, card_data, to, notify, timestamp, participant ]
)
message = "Received VCard (not implemented yet)"
buddy = _from.split("@")[0]
if participant is not None: # Group message
partname = participant.split('@')[0]
if _from.split('@')[1] == 'broadcast': # Broadcast message
message = self.broadcast_prefix + message
self.sendMessageToXMPP(partname, message, timestamp)
else: # Group message
self.sendGroupMessageToXMPP(buddy, partname, message, timestamp)
else:
self.sendMessageToXMPP(buddy, message, timestamp)
2015-09-15 19:29:26 +00:00
# self.sendMessageToXMPP(buddy, card_data)
#self.transferFile(buddy, str(name), card_data)
self.sendReceipt(_id, _from, None, participant)
self.recvMsgIDs.append((_id, _from, participant, timestamp))
2016-02-19 16:31:34 +00:00
2015-09-15 19:29:26 +00:00
def transferFile(self, buddy, name, data):
# Not working
2016-04-09 22:21:58 +00:00
self.logger.debug('transfering file: %s' % name)
self.backend.handleFTStart(self.user, buddy, name, len(data))
self.backend.handleFTData(0, data)
self.backend.handleFTFinish(self.user, buddy, name, len(data), 0)
# Called by superclass
def onContactTyping(self, buddy):
2016-04-09 22:21:58 +00:00
self.logger.info("Started typing: %s" % buddy)
if buddy != 'bot':
2015-09-15 19:29:26 +00:00
self.sendPresence(True)
self.backend.handleBuddyTyping(self.user, buddy)
2015-09-15 19:29:26 +00:00
if self.timer != None:
self.timer.cancel()
# Called by superclass
def onContactPaused(self, buddy):
2016-04-09 22:21:58 +00:00
self.logger.info("Paused typing: %s" % buddy)
if buddy != 'bot':
2015-09-15 19:29:26 +00:00
self.backend.handleBuddyTyped(self.user, buddy)
self.timer = Timer(3, self.backend.handleBuddyStoppedTyping,
(self.user, buddy)).start()
# Called by superclass
def onAddedToGroup(self, group):
2016-04-09 22:21:58 +00:00
self.logger.debug("Added to group: %s" % group)
room = group.getGroupId()
owner = group.getCreatorJid(full = False)
subjectOwner = group.getSubjectOwnerJid(full = False)
subject = group.getSubject()
self.groups[room] = Group(room, owner, subject, subjectOwner, self.backend, self.user)
self.groups[room].addParticipants(group.getParticipants(), self.buddies, self.legacyName)
self.bot.send("You have been added to group: %s@%s (%s)"
% (self._shortenGroupId(room), subject, self.backend.spectrum_jid))
2015-11-29 14:32:33 +00:00
# Called by superclass
def onParticipantsAddedToGroup(self, group):
2016-04-09 22:21:58 +00:00
self.logger.debug("Participants added to group: %s" % group)
room = group.getGroupId().split('@')[0]
self.groups[room].addParticipants(group.getParticipants(), self.buddies, self.legacyName)
self.groups[room].sendParticipantsToSpectrum(self.legacyName)
# Called by superclass
def onSubjectChanged(self, room, subject, subjectOwner, timestamp):
self.logger.debug(
2016-04-09 22:21:58 +00:00
"onSubjectChange(rrom=%s, subject=%s, subjectOwner=%s, timestamp=%s)" %
(room, subject, subjectOwner, timestamp)
)
try:
group = self.groups[room]
except KeyError:
2016-04-09 22:21:58 +00:00
self.logger.error("Subject of non-existant group (%s) changed" % group)
else:
group.subject = subject
group.subjectOwner = subjectOwner
if not group.joined:
# We have not joined group so we should not send subject
return
self.backend.handleSubject(self.user, room, subject, subjectOwner)
self.backend.handleRoomNicknameChanged(self.user, room, subject)
2015-11-29 14:32:33 +00:00
# Called by superclass
def onParticipantsRemovedFromGroup(self, room, participants):
2016-04-09 22:21:58 +00:00
self.logger.debug("Participants removed from group: %s, %s" %
(room, participants))
self.groups[room].removeParticipants(participants)
2015-11-29 14:32:33 +00:00
# Called by superclass
def onContactStatusChanged(self, number, status):
2016-04-09 22:21:58 +00:00
self.logger.debug("%s changed their status to %s" % (number, status))
try:
buddy = self.buddies[number]
buddy.statusMsg = status
self.buddies.updateSpectrum(buddy)
except KeyError:
2016-04-09 22:21:58 +00:00
self.logger.debug("%s not in buddy list" % number)
# Called by superclass
def onContactPictureChanged(self, number):
2016-04-09 22:21:58 +00:00
self.logger.debug("%s changed their profile picture" % number)
self.buddies.requestVCard(number)
# Called by superclass
def onContactAdded(self, number, nick):
2016-04-09 22:21:58 +00:00
self.logger.debug("Adding new contact %s (%s)" % (nick, number))
self.updateBuddy(number, nick, [])
# Called by superclass
def onContactRemoved(self, number):
2016-04-09 22:21:58 +00:00
self.logger.debug("Removing contact %s" % number)
self.removeBuddy(number)
def onContactUpdated(self, oldnumber, newnumber):
2016-04-09 22:21:58 +00:00
self.logger.debug("Contact has changed number from %s to %s" %
(oldnumber, newnumber))
if newnumber in self.buddies:
2016-04-09 22:21:58 +00:00
self.logger.warn("Contact %s exists, just updating" % newnumber)
self.buddies.refresh(newnumber)
try:
buddy = self.buddies[oldnumber]
except KeyError:
2016-04-09 22:21:58 +00:00
self.logger.warn("Old contact (%s) not found. Adding new contact (%s)" %
(oldnumber, newnumber))
nick = ""
else:
self.removeBuddy(buddy.number)
nick = buddy.nick
self.updateBuddy(newnumber, nick, [])
def onPresenceReceived(self, _type, name, jid, lastseen):
2016-04-09 22:21:58 +00:00
self.logger.info("Presence received: %s %s %s %s" % (_type, name, jid, lastseen))
buddy = jid.split("@")[0]
try:
buddy = self.buddies[buddy]
2015-11-10 16:04:20 +00:00
except KeyError:
# Sometimes whatsapp send our own presence
if buddy != self.legacyName:
2016-04-09 22:21:58 +00:00
self.logger.error("Buddy not found: %s" % buddy)
2015-11-10 16:04:20 +00:00
return
2016-04-09 22:21:58 +00:00
if (lastseen == buddy.lastseen) and (_type == buddy.presence):
2015-11-10 16:04:20 +00:00
return
2015-12-01 19:33:02 +00:00
if ((lastseen != "deny") and (lastseen != None) and (lastseen != "none")):
2015-11-10 16:04:20 +00:00
buddy.lastseen = int(lastseen)
if (_type == None):
buddy.lastseen = time.time()
2015-11-10 16:04:20 +00:00
buddy.presence = _type
if _type == "unavailable":
self.onPresenceUnavailable(buddy)
else:
self.onPresenceAvailable(buddy)
def onPresenceAvailable(self, buddy):
2016-04-09 22:21:58 +00:00
self.logger.info("Is available: %s" % buddy)
self.buddies.updateSpectrum(buddy)
2015-11-10 16:04:20 +00:00
def onPresenceUnavailable(self, buddy):
2016-04-09 22:21:58 +00:00
self.logger.info("Is unavailable: %s" % buddy)
self.buddies.updateSpectrum(buddy)
2013-06-28 20:46:56 +00:00
# spectrum RequestMethods
2013-05-31 18:53:21 +00:00
def sendTypingStarted(self, buddy):
if buddy != "bot":
2016-04-09 22:21:58 +00:00
self.logger.info("Started typing: %s to %s" % (self.legacyName, buddy))
self.sendTyping(buddy, True)
2016-02-19 16:31:34 +00:00
self.sendReadReceipts(buddy)
# If he is typing he is present
# I really don't know where else to put this.
# Ideally, this should be sent if the user is looking at his client
self.sendPresence(True)
2013-05-31 18:53:21 +00:00
def sendTypingStopped(self, buddy):
if buddy != "bot":
2016-04-09 22:21:58 +00:00
self.logger.info("Stopped typing: %s to %s" % (self.legacyName, buddy))
self.sendTyping(buddy, False)
2016-02-19 16:31:34 +00:00
self.sendReadReceipts(buddy)
2013-05-31 18:53:21 +00:00
def sendImage(self, message, ID, to):
if (".jpg" in message.lower()):
imgType = "jpg"
if (".webp" in message.lower()):
imgType = "webp"
success = deferred.Deferred()
error = deferred.Deferred()
self.downloadMedia(message, success.run, error.run)
# Success
path = success.arg(0)
2016-04-09 22:21:58 +00:00
call(self.logger.info, "Success: Image downloaded to %s" % path)
pathWithExt = path.then(lambda p: p + "." + imgType)
call(os.rename, path, pathWithExt)
pathJpg = path.then(lambda p: p + ".jpg")
if imgType != "jpg":
im = call(Image.open, pathWithExt)
call(im.save, pathJpg)
call(os.remove, pathWithExt)
2016-04-09 22:21:58 +00:00
call(self.logger.info, "Sending image to %s" % to)
waId = deferred.Deferred()
call(super(Session, self).sendImage, to, pathJpg, onSuccess = waId.run)
call(self.setWaId, ID, waId)
waId.when(call, os.remove, pathJpg)
waId.when(self.logger.info, "Image sent")
# Error
error.when(self.logger.info, "Download Error. Sending message as is.")
waId = error.when(self.sendTextMessage, to, message)
call(self.setWaId, ID, waId)
def setWaId(self, XmppId, waId):
self.msgIDs[waId] = MsgIDs(XmppId, waId)
def sendMessageToWA(self, sender, message, ID, xhtml=""):
2016-04-09 22:21:58 +00:00
self.logger.info("Message sent from %s to %s: %s (xhtml=%s)" %
(self.legacyName, sender, message, xhtml))
2015-09-15 19:29:26 +00:00
2016-02-19 16:31:34 +00:00
self.sendReadReceipts(sender)
2013-05-31 18:53:21 +00:00
if sender == "bot":
2015-09-06 12:06:16 +00:00
self.bot.parse(message)
elif "-" in sender: # group msg
if "/" in sender: # directed at single user
room, nick = sender.split("/")
group = self.groups[room]
number = None
for othernumber, othernick in group.participants.iteritems():
if othernick == nick:
number = othernumber
break
if number is not None:
2016-04-09 22:21:58 +00:00
self.logger.debug("Private message sent from %s to %s" % (self.legacyName, number))
waId = self.sendTextMessage(number + '@s.whatsapp.net', message)
self.msgIDs[waId] = MsgIDs( ID, waId)
else:
self.logger.error("Attempted to send private message to non-existent user")
2016-04-09 22:21:58 +00:00
self.logger.debug("%s to %s in %s" % (self.legacyName, nick, room))
2015-09-06 12:06:16 +00:00
else:
room = sender
if message[0] == '\\' and message[:1] != '\\\\':
2016-04-09 22:21:58 +00:00
self.logger.debug("Executing command %s in %s" % (message, room))
self.executeCommand(message, room)
else:
try:
group = self.groups[self._lengthenGroupId(room)]
2016-04-09 22:21:58 +00:00
self.logger.debug("Group Message from %s to %s Groups: %s" %
(group.nick , group , self.groups))
self.backend.handleMessage(
2016-04-09 22:21:58 +00:00
self.user, room, message, group.nick, xhtml=xhtml
)
except KeyError:
2016-04-09 22:21:58 +00:00
self.logger.error('Group not found: %s' % room)
if (".jpg" in message.lower()) or (".webp" in message.lower()):
self.sendImage(message, ID, room + '@g.us')
elif "geo:" in message.lower():
self._sendLocation(room + "@g.us", message, ID)
else:
self.sendTextMessage(room + '@g.us', message)
2015-09-06 12:06:16 +00:00
else: # private msg
buddy = sender
if message.split(" ").pop(0) == "\\lastseen":
2016-04-06 20:02:40 +00:00
self.presenceRequested.append(buddy)
self._requestLastSeen(buddy)
elif message.split(" ").pop(0) == "\\gpp":
self.sendMessageToXMPP(buddy, "Fetching Profile Picture")
self.requestVCard(buddy)
elif (".jpg" in message.lower()) or (".webp" in message.lower()):
self.sendImage(message, ID, buddy + "@s.whatsapp.net")
elif "geo:" in message.lower():
self._sendLocation(buddy + "@s.whatsapp.net", message, ID)
else:
waId = self.sendTextMessage(sender + '@s.whatsapp.net', message)
self.msgIDs[waId] = MsgIDs(ID, waId)
# self.logger.info("WA Message send to %s with ID %s", buddy, waId)
def executeCommand(self, command, room):
if command == '\\leave':
self.logger.debug("Leaving room %s", room)
2016-04-06 20:02:40 +00:00
self.leaveGroup(room) # Leave group on whatsapp side
group = self.groups[room]
2016-04-06 20:02:40 +00:00
group.leaveRoom() # Delete Room on spectrum side
del self.groups[room]
def _requestLastSeen(self, buddy):
2016-04-06 20:02:40 +00:00
def onSuccess(buddy, lastseen):
2015-11-10 16:04:20 +00:00
timestamp = time.localtime(time.localtime()-lastseen)
2016-04-06 20:02:40 +00:00
timestring = time.strftime("%a, %d %b %Y %H:%M:%S", timestamp)
self.sendMessageToXMPP(buddy, "%s (%s) %s" % (timestring, utils.ago(lastseen), str(lastseen)))
2016-04-06 20:02:40 +00:00
def onError(errorIqEntity, originalIqEntity):
self.sendMessageToXMPP(errorIqEntity.getFrom(), "LastSeen Error")
self.requestLastSeen(buddy, onSuccess, onError)
def _sendLocation(self, buddy, message, ID):
2016-04-06 20:02:40 +00:00
latitude,longitude = message.split(':')[1].split(',')
waId = self.sendLocation(buddy, float(latitude), float(longitude))
self.msgIDs[waId] = MsgIDs(ID, waId)
2016-04-06 20:02:40 +00:00
self.logger.info("WA Location Message send to %s with ID %s", buddy, waId)
2013-05-31 18:53:21 +00:00
def sendMessageToXMPP(self, buddy, messageContent, timestamp = "", nickname = ""):
if timestamp:
timestamp = time.strftime("%Y%m%dT%H%M%S", time.gmtime(timestamp))
if self.initialized == False:
2016-04-09 22:21:58 +00:00
self.logger.debug("Message queued from %s to %s: %s" %
(buddy, self.legacyName, messageContent))
self.offlineQueue.append((buddy, messageContent, timestamp))
else:
2016-04-09 22:21:58 +00:00
self.logger.debug("Message sent from %s to %s: %s" % (
buddy, self.legacyName, messageContent))
self.backend.handleMessage(self.user, buddy, messageContent, "",
"", timestamp)
2016-04-09 22:21:58 +00:00
def sendGroupMessageToXMPP(self, room, number, messageContent, timestamp = "", defaultname = ""):
2013-06-28 20:46:56 +00:00
if timestamp:
timestamp = time.strftime("%Y%m%dT%H%M%S", time.gmtime(timestamp))
2013-05-31 18:53:21 +00:00
2013-06-28 20:46:56 +00:00
if self.initialized == False:
2016-04-09 22:21:58 +00:00
self.logger.debug("Group message queued from %s to %s: %s" %
(number, room, messageContent))
2013-05-31 18:53:21 +00:00
2013-06-28 20:46:56 +00:00
if room not in self.groupOfflineQueue:
self.groupOfflineQueue[room] = [ ]
2015-09-15 19:29:26 +00:00
self.groupOfflineQueue[room].append(
(number, messageContent, timestamp)
2015-09-15 19:29:26 +00:00
)
2013-05-31 18:53:21 +00:00
else:
2016-04-09 22:21:58 +00:00
self.logger.debug("Group message sent from %s to %s: %s" %
(number, room, messageContent))
try:
group = self.groups[room]
# Update nickname
try:
if defaultname != "" and group.participants[number] == number:
group.changeNick(number, defaultname)
if self.buddies[number].nick != "":
group.changeNick(number, self.buddies[number].nick)
except KeyError:
pass
nick = group.participants[number]
if group.joined:
self.backend.handleMessage(self.user, room, messageContent,
nick, "", timestamp)
else:
self.bot.send("You have received a message in group: %s@%s"
% (room, self.backend.spectrum_jid))
self.bot.send("Join the group in order to reply")
self.bot.send("%s: %s" % (nick, messageContent))
except KeyError:
self.logger.warn("Group is not in group list")
self.backend.handleMessage(self.user, self._shortenGroupId(room),
messageContent, number, "", timestamp)
2013-06-28 20:46:56 +00:00
def changeStatus(self, status):
if status != self.status:
2016-04-09 22:21:58 +00:00
self.logger.info("Status changed: %s" % status)
2013-06-28 20:46:56 +00:00
self.status = status
2015-09-15 19:29:26 +00:00
if status == protocol_pb2.STATUS_ONLINE \
or status == protocol_pb2.STATUS_FFC:
self.sendPresence(True)
2013-06-28 20:46:56 +00:00
else:
self.sendPresence(False)
2013-06-28 20:46:56 +00:00
def changeStatusMessage(self, statusMessage):
if (statusMessage != self.statusMessage) or (self.initialized == False):
2013-06-28 20:46:56 +00:00
self.statusMessage = statusMessage
2016-04-09 22:21:58 +00:00
self.setStatus(statusMessage)
2016-04-10 00:47:53 +00:00
self.logger.info("Status message changed: %s" % statusMessage)
2013-06-28 20:46:56 +00:00
#if self.initialized == False:
# self.sendOfflineMessages()
# self.bot.call("welcome")
# self.initialized = True
2013-06-28 20:46:56 +00:00
def sendOfflineMessages(self):
# Flush Queues
while self.offlineQueue:
msg = self.offlineQueue.pop(0)
self.backend.handleMessage(self.user, msg[0], msg[1], "", "", msg[2])
2013-06-28 20:46:56 +00:00
2015-12-01 19:33:02 +00:00
# Called when user logs in to initialize the roster
2015-12-01 21:21:38 +00:00
def loadBuddies(self, buddies):
self.buddies.load(buddies)
2015-12-01 19:33:02 +00:00
2013-06-28 20:46:56 +00:00
# also for adding a new buddy
def updateBuddy(self, buddy, nick, groups, image_hash = None):
if buddy != "bot":
self.buddies.update(buddy, nick, groups, image_hash)
2013-05-31 18:53:21 +00:00
def removeBuddy(self, buddy):
if buddy != "bot":
2016-04-09 22:21:58 +00:00
self.logger.info("Buddy removed: %s" % buddy)
2013-06-28 20:46:56 +00:00
self.buddies.remove(buddy)
def requestVCard(self, buddy, ID=None):
self.buddies.requestVCard(buddy, ID)
2013-05-31 18:53:21 +00:00
def createThumb(self, size=100, raw=False):
2016-04-06 20:02:40 +00:00
img = Image.open(self.imgPath)
width, height = img.size
img_thumbnail = self.imgPath + '_thumbnail'
2016-04-06 20:02:40 +00:00
if width > height:
nheight = float(height) / width * size
nwidth = size
else:
nwidth = float(width) / height * size
nheight = size
2016-04-06 20:02:40 +00:00
img.thumbnail((nwidth, nheight), Image.ANTIALIAS)
img.save(img_thumbnail, 'JPEG')
2016-04-06 20:02:40 +00:00
with open(img_thumbnail, 'rb') as imageFile:
raw = base64.b64encode(imageFile.read())
2016-04-06 20:02:40 +00:00
return raw
# Not used
def onLocationReceived(self, messageId, jid, name, preview, latitude, longitude, receiptRequested, isBroadcast):
buddy = jid.split("@")[0]
2016-04-09 22:21:58 +00:00
self.logger.info("Location received from %s: %s, %s" % (buddy, latitude, longitude))
url = "http://maps.google.de?%s" % urllib.urlencode({ "q": "%s %s" % (latitude, longitude) })
self.sendMessageToXMPP(buddy, utils.shorten(url))
2016-04-06 20:02:40 +00:00
if receiptRequested:
self.call("message_ack", (jid, 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)
2016-04-06 20:02:40 +00:00
if receiptRequested:
self.call("subject_ack", (gjid, messageId))
# Yowsup Notifications
def onGroupParticipantRemoved(self, gjid, jid, author, timestamp, messageId, receiptRequested):
room = gjid.split("@")[0]
buddy = jid.split("@")[0]
2016-04-09 22:21:58 +00:00
self.logger.info("Removed %s from room %s" % (buddy, room))
self.backend.handleParticipantChanged(self.user, buddy, room, protocol_pb2.PARTICIPANT_FLAG_NONE, protocol_pb2.STATUS_NONE) # TODO
2016-04-06 20:02:40 +00:00
if receiptRequested: self.call("notification_ack", (gjid, messageId))
def onContactProfilePictureUpdated(self, jid, timestamp, messageId, pictureId, receiptRequested):
# TODO
2016-04-06 20:02:40 +00:00
if receiptRequested:
self.call("notification_ack", (jid, messageId))
def onGroupPictureUpdated(self, jid, author, timestamp, messageId, pictureId, receiptRequested):
# TODO
2016-04-06 20:02:40 +00:00
if receiptRequested:
self.call("notification_ack", (jid, messageId))