def dealWithUnencryptedMessage(message):
'''Decide what to do with the given unencrypted message'''
if message.messageType == Message.TYPE_CONTACT_REQUEST:
print("Received a contact request from", message.senderId)
# Check config to see whether we accept these or not
if Config.getProperty(Config.KEY_ALLOW_FRIEND_REQUESTS) \
and MessageShuffler._isProfileStatusOk(message.senderId, [None, 'requested', 'untrusted', 'trusted']):
# Call DbClient to store new message in inbox
rowToStore = {"messageType":"contactrequest", "fromId":message.senderId,
"fromName":message.senderName, "messageBody":message.message,
"publicKey":message.publicKey, "timestamp":message.timestamp,
"messageRead":False, "messageReplied":False}
DbClient.addMessageToInbox(rowToStore)
elif message.messageType == Message.TYPE_CONTACT_RESPONSE:
print("It's an unencrypted contact response, so it must be a refusal")
sender = DbClient.getProfile(message.senderId, False)
if MessageShuffler._isProfileStatusOk(message.senderId, ['requested']):
senderName = sender.get("displayName") if sender else ""
ContactMaker.handleReceiveDeny(message.senderId)
# Call DbClient to store new message in inbox
rowToStore = {"messageType":"contactresponse", "fromId":message.senderId,
"fromName":senderName, "messageBody":"", "accepted":False,
"messageRead":False, "messageReplied":False, "timestamp":message.timestamp,
"recipients":MessageShuffler.getOwnTorId()}
DbClient.addMessageToInbox(rowToStore)
else:
print("Hä? It's unencrypted but the message type is", message.messageType)
def servePage(self, view, url, params):
print("Compose: %s, params %s" % (url, ",".join(params)))
if url == "/start":
self.requirePageResources(['default.css'])
parentHash = params.get("reply", None)
recpts = params.get("sendto", None)
# Build list of contacts to whom we can send
userboxes = []
for p in DbClient.getMessageableContacts():
box = Bean()
box.dispName = p['displayName']
box.torid = p['torid']
userboxes.append(box)
pageParams = {"contactlist":userboxes, "parenthash" : parentHash if parentHash else "",
"webcachedir":Config.getWebCacheDir(), "recipientids":recpts}
contents = self.buildPage({'pageTitle' : I18nManager.getText("composemessage.title"),
'pageBody' : self.composetemplate.getHtml(pageParams),
'pageFooter' : "<p>Footer</p>"})
view.setHtml(contents)
# If we've got no friends, then warn, can't send to anyone
if not DbClient.hasFriends():
view.page().mainFrame().evaluateJavaScript("window.alert('No friends :(');")
elif url == "/send":
print("Submit new message with params:", params)
msgBody = params['messagebody'] # TODO: check body isn't empty, throw an exception?
parentHash = params.get("parenthash", None)
recpts = params['sendto']
# Make a corresponding message object and pass it on
msg = message.RegularMessage(sendTo=recpts, messageBody=msgBody, replyToHash=parentHash)
msg.recipients = recpts.split(",")
DbClient.addMessageToOutbox(msg)
# Save a copy of the sent message
sentMessage = {"messageType":"normal", "fromId":DbClient.getOwnTorId(),
"messageBody":msgBody, "timestamp":msg.timestamp, "messageRead":True,
"messageReplied":False, "recipients":recpts, "parentHash":parentHash}
DbClient.addMessageToInbox(sentMessage)
# Close window after successful send
contents = self.buildPage({'pageTitle' : I18nManager.getText("messages.title"),
'pageBody' : self.closingtemplate.getHtml(),
'pageFooter' : "<p>Footer</p>"})
view.setHtml(contents)
def testConversationIdsFromParent(self):
'''Check that the correct conversation id can be found based on a parent hash'''
startId = DbClient.getNewConversationId()
# Does it work on an empty inbox?
DbClient._getInboxTable().remove()
nextId = DbClient.getConversationId("some random hash")
self.assertEqual(startId+1, nextId, "Should increment if hash not found in empty inbox")
nextId = DbClient.getConversationId(None)
self.assertEqual(startId+2, nextId, "Should increment if None hash given")
nextId = DbClient.getConversationId("")
self.assertEqual(startId+3, nextId, "Should increment if empty string hash given")
# Now add a message with a hash
rowToStore = {"messageType":"normal", "fromId":"BCD123EFG456HI78", "timestamp":"2012-01-08-13-18",
"messageBody":"I do like a day beside the seaside", "recipients":"ABC123DEF456GH78"}
DbClient.addMessageToInbox(rowToStore)
nextId = DbClient.getConversationId("some random hash")
self.assertEqual(startId+5, nextId, "Should increment if hash not found in inbox")
nextId = DbClient.getConversationId("7b6dd00818a70515ba560efc5a42aab5")
self.assertEqual(startId+4, nextId, "Should take the old id if hash was found in inbox")
anotherId = DbClient.getConversationId("7b6dd00818a70515ba560efc5a42aab5")
self.assertEqual(anotherId, nextId, "Should reuse the same id if hash was found in inbox")
def dealWithAsymmetricMessage(message):
'''Decide what to do with the given asymmetric message'''
if message.senderId == MessageShuffler.getOwnTorId():
print("*** Shouldn't receive a message from myself!")
return
# Sort message according to type
if message.messageType == Message.TYPE_CONTACT_RESPONSE:
print("Received a contact accept from", message.senderId, "name", message.senderName)
if MessageShuffler._isProfileStatusOk(message.senderId, ['pending', 'requested', 'untrusted']):
print(message.senderName, "'s public key is", message.senderKey)
ContactMaker.handleReceiveAccept(message.senderId, message.senderName, message.senderKey)
# Call DbClient to store new message in inbox
rowToStore = {"messageType":"contactresponse", "fromId":message.senderId,
"fromName":message.senderName, "messageBody":message.introMessage, "accepted":True,
"messageRead":False, "messageReplied":False, "timestamp":message.timestamp,
"recipients":MessageShuffler.getOwnTorId()}
DbClient.addMessageToInbox(rowToStore)
elif MessageShuffler._isProfileStatusOk(message.senderId, [None, 'blocked']):
print("Received a contact response but I didn't send them a request!")
print("Encrypted contents are:", message.encryptedContents)
rowToStore = {"messageType":"contactresponse", "fromId":message.senderId,
"fromName":message.senderName, "messageBody":message.introMessage, "accepted":True,
"timestamp":message.timestamp, "encryptedMsg":message.encryptedContents}
DbClient.addMessageToPendingContacts(rowToStore)
elif message.messageType == Message.TYPE_STATUS_NOTIFY:
if message.online:
print("One of our contacts has just come online- ", message.senderId,
"and hash is", message.profileHash)
prof = DbClient.getProfile(userid=message.senderId, extend=False)
if prof:
storedHash = prof.get("profileHash", "empty")
if message.profileHash != storedHash:
reply = InfoRequestMessage(infoType=InfoRequestMessage.INFO_PROFILE)
reply.recipients = [message.senderId]
DbClient.addMessageToOutbox(reply)
if message.ping:
print("Now sending back a pong, too")
reply = StatusNotifyMessage(online=True, ping=False, profileHash=None)
reply.recipients = [message.senderId]
DbClient.addMessageToOutbox(reply)
else:
print("It's already a pong so I won't reply")
Contacts.comeOnline(message.senderId)
else:
print("One of our contacts is going offline -", message.senderId)
Contacts.goneOffline(message.senderId)
elif message.messageType == Message.TYPE_INFO_REQUEST:
print("I've received an info request message for type", message.infoType)
if MessageShuffler._isProfileStatusOk(message.senderId, ['trusted']):
reply = InfoResponseMessage(message.messageType)
reply.recipients = [message.senderId]
DbClient.addMessageToOutbox(reply)
elif message.messageType == Message.TYPE_INFO_RESPONSE:
if message.profile and MessageShuffler._isProfileStatusOk(message.senderId, ['trusted', 'untrusted']):
if message.profileHash:
message.profile['profileHash'] = message.profileHash
DbClient.updateContact(message.senderId, message.profile)
elif message.messageType == Message.TYPE_ASYM_MESSAGE:
print("It's a general kind of message, this should go in the Inbox, right?")
if MessageShuffler._isProfileStatusOk(message.senderId, ['trusted', 'untrusted']):
Contacts.comeOnline(message.senderId)
else:
# It's another asymmetric message type
print("Hä? What kind of asymmetric message type is that? ", message.messageType)
