/*

* Copyright (c) 2010-2014 Kevin Smith

* Licensed under the GNU General Public License v3.

* See Documentation/Licenses/GPLv3.txt for more information.

*/

#include <Swift/Controllers/Chat/ChatController.h>

#include <boost/bind.hpp>

#include <boost/smart_ptr/make_shared.hpp>

#include <stdio.h>

#include <Swiften/Avatars/AvatarManager.h>

#include <Swiften/Base/Algorithm.h>

#include <Swiften/Base/DateTime.h>

#include <Swiften/Base/foreach.h>

#include <Swiften/Base/format.h>

#include <Swiften/Base/Log.h>

#include <Swiften/Chat/ChatStateNotifier.h>

#include <Swiften/Chat/ChatStateTracker.h>

#include <Swiften/Client/ClientBlockListManager.h>

#include <Swiften/Client/NickResolver.h>

#include <Swiften/Client/StanzaChannel.h>

#include <Swiften/Disco/EntityCapsProvider.h>

#include <Swiften/Elements/DeliveryReceipt.h>

#include <Swiften/Elements/DeliveryReceiptRequest.h>

#include <Swiften/Elements/Idle.h>

#include <Swiften/FileTransfer/FileTransferManager.h>

#include <Swift/Controllers/Intl.h>

#include <Swift/Controllers/UIInterfaces/ChatWindowFactory.h>

#include <Swift/Controllers/XMPPEvents/EventController.h>

#include <Swift/Controllers/FileTransfer/FileTransferController.h>

#include <Swift/Controllers/StatusUtil.h>

#include <Swift/Controllers/UIEvents/UIEventStream.h>

#include <Swift/Controllers/UIEvents/SendFileUIEvent.h>

#include <Swift/Controllers/UIEvents/AcceptWhiteboardSessionUIEvent.h>

#include <Swift/Controllers/UIEvents/CancelWhiteboardSessionUIEvent.h>

#include <Swift/Controllers/UIEvents/ShowWhiteboardUIEvent.h>

#include <Swift/Controllers/UIEvents/RequestChangeBlockStateUIEvent.h>

#include <Swift/Controllers/UIEvents/InviteToMUCUIEvent.h>

#include <Swift/Controllers/UIEvents/RequestInviteToMUCUIEvent.h>

#include <Swift/Controllers/SettingConstants.h>

#include <Swift/Controllers/Highlighter.h>

#include <Swift/Controllers/Chat/ChatMessageParser.h>

namespace Swift {

/**

* The controller does not gain ownership of the stanzaChannel, nor the factory.

*/

ChatController::ChatController(const JID& self, StanzaChannel* stanzaChannel, IQRouter* iqRouter, ChatWindowFactory* chatWindowFactory, const JID &contact, NickResolver* nickResolver, PresenceOracle* presenceOracle, AvatarManager* avatarManager, bool isInMUC, bool useDelayForLatency, UIEventStream* eventStream, EventController* eventController, TimerFactory* timerFactory, EntityCapsProvider* entityCapsProvider, bool userWantsReceipts, SettingsProvider* settings, HistoryController* historyController, MUCRegistry* mucRegistry, HighlightManager* highlightManager, ClientBlockListManager* clientBlockListManager, ChatMessageParser* chatMessageParser, AutoAcceptMUCInviteDecider* autoAcceptMUCInviteDecider)

: ChatControllerBase(self, stanzaChannel, iqRouter, chatWindowFactory, contact, presenceOracle, avatarManager, useDelayForLatency, eventStream, eventController, timerFactory, entityCapsProvider, historyController, mucRegistry, highlightManager, chatMessageParser, autoAcceptMUCInviteDecider), eventStream_(eventStream), userWantsReceipts_(userWantsReceipts), settings_(settings), clientBlockListManager_(clientBlockListManager) {

isInMUC_ = isInMUC;

lastWasPresence_ = false;

chatStateNotifier_ = new ChatStateNotifier(stanzaChannel, contact, entityCapsProvider);

chatStateTracker_ = new ChatStateTracker();

nickResolver_ = nickResolver;

presenceOracle_->onPresenceChange.connect(boost::bind(&ChatController::handlePresenceChange, this, _1));

chatStateTracker_->onChatStateChange.connect(boost::bind(&ChatWindow::setContactChatState, chatWindow_, _1));

stanzaChannel_->onStanzaAcked.connect(boost::bind(&ChatController::handleStanzaAcked, this, _1));

nickResolver_->onNickChanged.connect(boost::bind(&ChatController::handleContactNickChanged, this, _1, _2));

std::string nick = nickResolver_->jidToNick(toJID_);

chatWindow_->setName(nick);

std::string startMessage;

Presence::ref theirPresence;

if (isInMUC) {

startMessage = str(format(QT_TRANSLATE_NOOP("", "Starting chat with %1% in chatroom %2%")) % nick % contact.toBare().toString());

theirPresence = presenceOracle->getLastPresence(contact);

} else {

startMessage = str(format(QT_TRANSLATE_NOOP("", "Starting chat with %1% - %2%")) % nick % contact.toBare().toString());

theirPresence = contact.isBare() ? presenceOracle->getHighestPriorityPresence(contact.toBare()) : presenceOracle->getLastPresence(contact);

}

Idle::ref idle;

if (theirPresence && (idle = theirPresence->getPayload<Idle>())) {

startMessage += str(format(QT_TRANSLATE_NOOP("", ", who has been idle since %1%")) % dateTimeToLocalString(idle->getSince()));

}

startMessage += ": " + statusShowTypeToFriendlyName(theirPresence ? theirPresence->getShow() : StatusShow::None);

if (theirPresence && !theirPresence->getStatus().empty()) {

startMessage += " (" + theirPresence->getStatus() + ")";

}

lastShownStatus_ = theirPresence ? theirPresence->getShow() : StatusShow::None;

chatStateNotifier_->setContactIsOnline(theirPresence && theirPresence->getType() == Presence::Available);

startMessage += ".";

chatWindow_->addSystemMessage(chatMessageParser_->parseMessageBody(startMessage), ChatWindow::DefaultDirection);

chatWindow_->onUserTyping.connect(boost::bind(&ChatStateNotifier::setUserIsTyping, chatStateNotifier_));

chatWindow_->onUserCancelsTyping.connect(boost::bind(&ChatStateNotifier::userCancelledNewMessage, chatStateNotifier_));

chatWindow_->onFileTransferStart.connect(boost::bind(&ChatController::handleFileTransferStart, this, _1, _2));

chatWindow_->onFileTransferAccept.connect(boost::bind(&ChatController::handleFileTransferAccept, this, _1, _2));

chatWindow_->onFileTransferCancel.connect(boost::bind(&ChatController::handleFileTransferCancel, this, _1));

chatWindow_->onSendFileRequest.connect(boost::bind(&ChatController::handleSendFileRequest, this, _1));

chatWindow_->onWhiteboardSessionAccept.connect(boost::bind(&ChatController::handleWhiteboardSessionAccept, this));

chatWindow_->onWhiteboardSessionCancel.connect(boost::bind(&ChatController::handleWhiteboardSessionCancel, this));

chatWindow_->onWhiteboardWindowShow.connect(boost::bind(&ChatController::handleWhiteboardWindowShow, this));

chatWindow_->onBlockUserRequest.connect(boost::bind(&ChatController::handleBlockUserRequest, this));

chatWindow_->onUnblockUserRequest.connect(boost::bind(&ChatController::handleUnblockUserRequest, this));

chatWindow_->onInviteToChat.connect(boost::bind(&ChatController::handleInviteToChat, this, _1));

handleBareJIDCapsChanged(toJID_);

settings_->onSettingChanged.connect(boost::bind(&ChatController::handleSettingChanged, this, _1));

eventStream_->onUIEvent.connect(boost::bind(&ChatController::handleUIEvent, this, _1));

}

void ChatController::handleContactNickChanged(const JID& jid, const std::string& /*oldNick*/) {

if (jid.toBare() == toJID_.toBare()) {

chatWindow_->setName(nickResolver_->jidToNick(jid));

}

}

ChatController::~ChatController() {

eventStream_->onUIEvent.disconnect(boost::bind(&ChatController::handleUIEvent, this, _1));

settings_->onSettingChanged.disconnect(boost::bind(&ChatController::handleSettingChanged, this, _1));

nickResolver_->onNickChanged.disconnect(boost::bind(&ChatController::handleContactNickChanged, this, _1, _2));

delete chatStateNotifier_;

delete chatStateTracker_;

}

JID ChatController::getBaseJID() {

return isInMUC_ ? toJID_ : ChatControllerBase::getBaseJID();

}

void ChatController::cancelReplaces() {

lastWasPresence_ = false;

}

void ChatController::handleBareJIDCapsChanged(const JID& /*jid*/) {

DiscoInfo::ref disco = entityCapsProvider_->getCaps(toJID_);

if (disco) {

if (disco->hasFeature(DiscoInfo::MessageCorrectionFeature)) {

chatWindow_->setCorrectionEnabled(ChatWindow::Yes);

} else {

chatWindow_->setCorrectionEnabled(ChatWindow::No);

}

if (disco->hasFeature(DiscoInfo::MessageDeliveryReceiptsFeature)) {

contactSupportsReceipts_ = ChatWindow::Yes;

} else {

contactSupportsReceipts_ = ChatWindow::No;

}

if (FileTransferManager::isSupportedBy(disco)) {

chatWindow_->setFileTransferEnabled(ChatWindow::Yes);

} else {

chatWindow_->setFileTransferEnabled(ChatWindow::No);

}

} else {

SWIFT_LOG(debug) << "No disco info :(" << std::endl;

chatWindow_->setCorrectionEnabled(ChatWindow::Maybe);

contactSupportsReceipts_ = ChatWindow::Maybe;

}

checkForDisplayingDisplayReceiptsAlert();

}

void ChatController::setToJID(const JID& jid) {

chatStateNotifier_->setContact(jid);

ChatControllerBase::setToJID(jid);

Presence::ref presence;

if (isInMUC_) {

presence = presenceOracle_->getLastPresence(jid);

} else {

presence = jid.isBare() ? presenceOracle_->getHighestPriorityPresence(jid.toBare()) : presenceOracle_->getLastPresence(jid);

}

chatStateNotifier_->setContactIsOnline(presence && presence->getType() == Presence::Available);

handleBareJIDCapsChanged(toJID_);

}

void ChatController::setAvailableServerFeatures(boost::shared_ptr<DiscoInfo> info) {

ChatControllerBase::setAvailableServerFeatures(info);

if (iqRouter_->isAvailable() && info->hasFeature(DiscoInfo::BlockingCommandFeature)) {

boost::shared_ptr<BlockList> blockList = clientBlockListManager_->getBlockList();

blockingOnStateChangedConnection_ = blockList->onStateChanged.connect(boost::bind(&ChatController::handleBlockingStateChanged, this));

blockingOnItemAddedConnection_ = blockList->onItemAdded.connect(boost::bind(&ChatController::handleBlockingItemAdded, this, _1));

blockingOnItemRemovedConnection_ = blockList->onItemRemoved.connect(boost::bind(&ChatController::handleBlockingItemRemoved, this, _1));

handleBlockingStateChanged();

}

}

bool ChatController::isIncomingMessageFromMe(boost::shared_ptr<Message>) {

return false;

}

void ChatController::preHandleIncomingMessage(boost::shared_ptr<MessageEvent> messageEvent) {

if (messageEvent->isReadable()) {

chatWindow_->flash();

lastWasPresence_ = false;

}

boost::shared_ptr<Message> message = messageEvent->getStanza();

JID from = message->getFrom();

if (!from.equals(toJID_, JID::WithResource)) {

if (toJID_.equals(from, JID::WithoutResource) && toJID_.isBare()){

setToJID(from);

}

}

chatStateTracker_->handleMessageReceived(message);

chatStateNotifier_->receivedMessageFromContact(!!message->getPayload<ChatState>());

// handle XEP-0184 Message Receipts

// incomming receipts

if (boost::shared_ptr<DeliveryReceipt> receipt = message->getPayload<DeliveryReceipt>()) {

SWIFT_LOG(debug) << "received receipt for id: " << receipt->getReceivedID() << std::endl;

if (requestedReceipts_.find(receipt->getReceivedID()) != requestedReceipts_.end()) {

chatWindow_->setMessageReceiptState(requestedReceipts_[receipt->getReceivedID()], ChatWindow::ReceiptReceived);

requestedReceipts_.erase(receipt->getReceivedID());

}

// incomming errors in response to send out receipts

} else if (message->getPayload<DeliveryReceiptRequest>() && (message->getType() == Message::Error)) {

if (requestedReceipts_.find(message->getID()) != requestedReceipts_.end()) {

chatWindow_->setMessageReceiptState(requestedReceipts_[message->getID()], ChatWindow::ReceiptFailed);

requestedReceipts_.erase(message->getID());

}

// incoming receipt requests

} else if (message->getPayload<DeliveryReceiptRequest>()) {

if (receivingPresenceFromUs_) {

boost::shared_ptr<Message> receiptMessage = boost::make_shared<Message>();

receiptMessage->setTo(toJID_);

receiptMessage->addPayload(boost::make_shared<DeliveryReceipt>(message->getID()));

stanzaChannel_->sendMessage(receiptMessage);

}

}

}

void ChatController::postHandleIncomingMessage(boost::shared_ptr<MessageEvent> messageEvent, const HighlightAction& highlight) {

eventController_->handleIncomingEvent(messageEvent);

if (!messageEvent->getConcluded()) {

highlighter_->handleHighlightAction(highlight);

}

}

void ChatController::preSendMessageRequest(boost::shared_ptr<Message> message) {

chatStateNotifier_->addChatStateRequest(message);

if (userWantsReceipts_ && (contactSupportsReceipts_ != ChatWindow::No) && message) {

message->addPayload(boost::make_shared<DeliveryReceiptRequest>());

}

}

void ChatController::setContactIsReceivingPresence(bool isReceivingPresence) {

receivingPresenceFromUs_ = isReceivingPresence;

}

void ChatController::handleSettingChanged(const std::string& settingPath) {

if (settingPath == SettingConstants::REQUEST_DELIVERYRECEIPTS.getKey()) {

userWantsReceipts_ = settings_->getSetting(SettingConstants::REQUEST_DELIVERYRECEIPTS);

checkForDisplayingDisplayReceiptsAlert();

}

}

void ChatController::checkForDisplayingDisplayReceiptsAlert() {

if (userWantsReceipts_ && (contactSupportsReceipts_ == ChatWindow::No)) {

chatWindow_->setAlert(QT_TRANSLATE_NOOP("", "This chat doesn't support delivery receipts."));

} else if (userWantsReceipts_ && (contactSupportsReceipts_ == ChatWindow::Maybe)) {

chatWindow_->setAlert(QT_TRANSLATE_NOOP("", "This chat may not support delivery receipts. You might not receive delivery receipts for the messages you sent."));

} else {

chatWindow_->cancelAlert();

}

}

void ChatController::handleBlockingStateChanged() {

boost::shared_ptr<BlockList> blockList = clientBlockListManager_->getBlockList();

if (blockList->getState() == BlockList::Available) {

if (isInMUC_ ? blockList->isBlocked(toJID_) : blockList->isBlocked(toJID_.toBare())) {

chatWindow_->setAlert(QT_TRANSLATE_NOOP("", "You've currently blocked this contact. To continue your conversation you have to unblock the contact first."));

chatWindow_->setInputEnabled(false);

chatWindow_->setBlockingState(ChatWindow::IsBlocked);

} else {

chatWindow_->setBlockingState(ChatWindow::IsUnblocked);

}

}

}

void ChatController::handleBlockingItemAdded(const JID& jid) {

if (toJID_ == (isInMUC_ ? jid: jid.toBare())) {

chatWindow_->setAlert(QT_TRANSLATE_NOOP("", "You've currently blocked this contact. To continue your conversation you have to unblock the contact first."));

chatWindow_->setInputEnabled(false);

chatWindow_->setBlockingState(ChatWindow::IsBlocked);

}

}

void ChatController::handleBlockingItemRemoved(const JID& jid) {

if (toJID_ == (isInMUC_ ? jid: jid.toBare())) {

// FIXME: Support for different types of alerts.

chatWindow_->cancelAlert();

chatWindow_->setInputEnabled(true);

chatWindow_->setBlockingState(ChatWindow::IsUnblocked);

}

}

void ChatController::handleBlockUserRequest() {

if (isInMUC_) {

eventStream_->send(boost::make_shared<RequestChangeBlockStateUIEvent>(RequestChangeBlockStateUIEvent::Blocked, toJID_));

} else {

eventStream_->send(boost::make_shared<RequestChangeBlockStateUIEvent>(RequestChangeBlockStateUIEvent::Blocked, toJID_.toBare()));

}

}

void ChatController::handleUnblockUserRequest() {

if (isInMUC_) {

eventStream_->send(boost::make_shared<RequestChangeBlockStateUIEvent>(RequestChangeBlockStateUIEvent::Unblocked, toJID_));

} else {

eventStream_->send(boost::make_shared<RequestChangeBlockStateUIEvent>(RequestChangeBlockStateUIEvent::Unblocked, toJID_.toBare()));

}

}

void ChatController::handleInviteToChat(const std::vector<JID>& droppedJIDs) {

boost::shared_ptr<UIEvent> event(new RequestInviteToMUCUIEvent(toJID_.toBare(), droppedJIDs, RequestInviteToMUCUIEvent::Impromptu));

eventStream_->send(event);

}

void ChatController::handleUIEvent(boost::shared_ptr<UIEvent> event) {

boost::shared_ptr<InviteToMUCUIEvent> inviteEvent = boost::dynamic_pointer_cast<InviteToMUCUIEvent>(event);

if (inviteEvent && inviteEvent->getRoom() == toJID_.toBare()) {

onConvertToMUC(detachChatWindow(), inviteEvent->getInvites(), inviteEvent->getReason());

}

}

void ChatController::postSendMessage(const std::string& body, boost::shared_ptr<Stanza> sentStanza) {

boost::shared_ptr<Replace> replace = sentStanza->getPayload<Replace>();

if (replace) {

eraseIf(unackedStanzas_, PairSecondEquals<boost::shared_ptr<Stanza>, std::string>(myLastMessageUIID_));

replaceMessage(body, myLastMessageUIID_, boost::posix_time::microsec_clock::universal_time(), HighlightAction());

} else {

myLastMessageUIID_ = addMessage(body, QT_TRANSLATE_NOOP("", "me"), true, labelsEnabled_ ? chatWindow_->getSelectedSecurityLabel().getLabel() : boost::shared_ptr<SecurityLabel>(), avatarManager_->getAvatarPath(selfJID_), boost::posix_time::microsec_clock::universal_time(), HighlightAction());

}

if (stanzaChannel_->getStreamManagementEnabled() && !myLastMessageUIID_.empty() ) {

chatWindow_->setAckState(myLastMessageUIID_, ChatWindow::Pending);

unackedStanzas_[sentStanza] = myLastMessageUIID_;

}

if (sentStanza->getPayload<DeliveryReceiptRequest>()) {

requestedReceipts_[sentStanza->getID()] = myLastMessageUIID_;

chatWindow_->setMessageReceiptState(myLastMessageUIID_, ChatWindow::ReceiptRequested);

}

lastWasPresence_ = false;

chatStateNotifier_->userSentMessage();

}

void ChatController::handleStanzaAcked(boost::shared_ptr<Stanza> stanza) {

std::map<boost::shared_ptr<Stanza>, std::string>::iterator unackedStanza = unackedStanzas_.find(stanza);

if (unackedStanza != unackedStanzas_.end()) {

chatWindow_->setAckState(unackedStanza->second, ChatWindow::Received);

unackedStanzas_.erase(unackedStanza);

}

}

void ChatController::setOnline(bool online) {

if (!online) {

std::map<boost::shared_ptr<Stanza>, std::string>::iterator it = unackedStanzas_.begin();

for ( ; it != unackedStanzas_.end(); ++it) {

chatWindow_->setAckState(it->second, ChatWindow::Failed);

}

unackedStanzas_.clear();

Presence::ref fakeOffline(new Presence());

fakeOffline->setFrom(toJID_);

fakeOffline->setType(Presence::Unavailable);

chatStateTracker_->handlePresenceChange(fakeOffline);

}

ChatControllerBase::setOnline(online);

}

void ChatController::handleNewFileTransferController(FileTransferController* ftc) {

std::string nick = senderDisplayNameFromMessage(ftc->getOtherParty());

std::string ftID = ftc->setChatWindow(chatWindow_, nick);

ftControllers[ftID] = ftc;

}

void ChatController::handleWhiteboardSessionRequest(bool senderIsSelf) {

lastWbID_ = chatWindow_->addWhiteboardRequest(senderIsSelf);

}

void ChatController::handleWhiteboardStateChange(const ChatWindow::WhiteboardSessionState state) {

chatWindow_->setWhiteboardSessionStatus(lastWbID_, state);

}

void ChatController::handleFileTransferCancel(std::string id) {

SWIFT_LOG(debug) << "handleFileTransferCancel(" << id << ")" << std::endl;

if (ftControllers.find(id) != ftControllers.end()) {

ftControllers[id]->cancel();

} else {

std::cerr << "unknown file transfer UI id" << std::endl;

}

}

void ChatController::handleFileTransferStart(std::string id, std::string description) {

SWIFT_LOG(debug) << "handleFileTransferStart(" << id << ", " << description << ")" << std::endl;

if (ftControllers.find(id) != ftControllers.end()) {

ftControllers[id]->start(description);

} else {

std::cerr << "unknown file transfer UI id" << std::endl;

}

}

void ChatController::handleFileTransferAccept(std::string id, std::string filename) {

SWIFT_LOG(debug) << "handleFileTransferAccept(" << id << ", " << filename << ")" << std::endl;

if (ftControllers.find(id) != ftControllers.end()) {

ftControllers[id]->accept(filename);

} else {

std::cerr << "unknown file transfer UI id" << std::endl;

}

}

void ChatController::handleSendFileRequest(std::string filename) {

SWIFT_LOG(debug) << "ChatController::handleSendFileRequest(" << filename << ")" << std::endl;

eventStream_->send(boost::make_shared<SendFileUIEvent>(getToJID(), filename));

}

void ChatController::handleWhiteboardSessionAccept() {

eventStream_->send(boost::make_shared<AcceptWhiteboardSessionUIEvent>(toJID_));

}

void ChatController::handleWhiteboardSessionCancel() {

eventStream_->send(boost::make_shared<CancelWhiteboardSessionUIEvent>(toJID_));

}

void ChatController::handleWhiteboardWindowShow() {

eventStream_->send(boost::make_shared<ShowWhiteboardUIEvent>(toJID_));

}

std::string ChatController::senderDisplayNameFromMessage(const JID& from) {

return nickResolver_->jidToNick(from);

}

std::string ChatController::getStatusChangeString(boost::shared_ptr<Presence> presence) {

std::string nick = senderDisplayNameFromMessage(presence->getFrom());

std::string response;

if (!presence || presence->getType() == Presence::Unavailable || presence->getType() == Presence::Error) {

response = QT_TRANSLATE_NOOP("", "%1% has gone offline");

} else if (presence->getType() == Presence::Available) {

StatusShow::Type show = presence->getShow();

if (show == StatusShow::Online || show == StatusShow::FFC) {

response = QT_TRANSLATE_NOOP("", "%1% has become available");

} else if (show == StatusShow::Away || show == StatusShow::XA) {

response = QT_TRANSLATE_NOOP("", "%1% has gone away");

} else if (show == StatusShow::DND) {

response = QT_TRANSLATE_NOOP("", "%1% is now busy");

}

}

Idle::ref idle;

if ((idle = presence->getPayload<Idle>())) {

response += str(format(QT_TRANSLATE_NOOP("", " and has been idle since %1%")) % dateTimeToLocalString(idle->getSince()));

}

if (!response.empty()) {

response = str(format(response) % nick);

}

if (!presence->getStatus().empty()) {

response += " (" + presence->getStatus() + ")";

}

return response + ".";

}

void ChatController::handlePresenceChange(boost::shared_ptr<Presence> newPresence) {

bool me = false;

if (toJID_.isBare()) {

newPresence = presenceOracle_->getHighestPriorityPresence(toJID_);

if ((newPresence ? newPresence->getShow() : StatusShow::None) != lastShownStatus_) {

me = true;

}

} else if (toJID_.equals(newPresence->getFrom(), JID::WithResource)) {

me = true;

}

if (!me) {

return;

}

if (!newPresence) {

newPresence = boost::make_shared<Presence>();

newPresence->setType(Presence::Unavailable);

}

lastShownStatus_ = newPresence->getShow();

chatStateTracker_->handlePresenceChange(newPresence);

chatStateNotifier_->setContactIsOnline(newPresence->getType() == Presence::Available);

std::string newStatusChangeString = getStatusChangeString(newPresence);

if (newStatusChangeString != lastStatusChangeString_) {

if (lastWasPresence_) {

} else {

chatWindow_->addPresenceMessage(chatMessageParser_->parseMessageBody(newStatusChangeString), ChatWindow::DefaultDirection);

}

lastStatusChangeString_ = newStatusChangeString;

lastWasPresence_ = true;

}

}

boost::optional<boost::posix_time::ptime> ChatController::getMessageTimestamp(boost::shared_ptr<Message> message) const {

return message->getTimestamp();

}

void ChatController::logMessage(const std::string& message, const JID& fromJID, const JID& toJID, const boost::posix_time::ptime& timeStamp, bool /* isIncoming */) {

HistoryMessage::Type type;

if (mucRegistry_->isMUC(fromJID.toBare()) || mucRegistry_->isMUC(toJID.toBare())) {

type = HistoryMessage::PrivateMessage;

}

else {

type = HistoryMessage::Chat;

}

if (historyController_) {

historyController_->addMessage(message, fromJID, toJID, type, timeStamp);

}

}

ChatWindow* ChatController::detachChatWindow() {

chatWindow_->onUserTyping.disconnect(boost::bind(&ChatStateNotifier::setUserIsTyping, chatStateNotifier_));

chatWindow_->onUserCancelsTyping.disconnect(boost::bind(&ChatStateNotifier::userCancelledNewMessage, chatStateNotifier_));

return ChatControllerBase::detachChatWindow();

}

}

/*

* Copyright (c) 2010-2014 Kevin Smith

* Licensed under the GNU General Public License v3.

* See Documentation/Licenses/GPLv3.txt for more information.

*/

#include <Swift/Controllers/Chat/MUCController.h>

#include <boost/bind.hpp>

#include <boost/regex.hpp>

#include <boost/algorithm/string.hpp>

#include <Swiften/Avatars/AvatarManager.h>

#include <Swiften/Base/foreach.h>

#include <Swiften/Base/foreach.h>

#include <Swiften/Base/format.h>

#include <Swiften/Base/Log.h>

#include <Swiften/Client/StanzaChannel.h>

#include <Swiften/Disco/EntityCapsProvider.h>

#include <Swiften/Elements/Delay.h>

#include <Swiften/MUC/MUC.h>

#include <Swiften/Network/Timer.h>

#include <Swiften/Network/TimerFactory.h>

#include <Swiften/Roster/XMPPRoster.h>

#include <SwifTools/TabComplete.h>

#include <Swift/Controllers/Chat/ChatMessageParser.h>

#include <Swift/Controllers/Highlighter.h>

#include <Swift/Controllers/Intl.h>

#include <Swift/Controllers/Roster/ContactRosterItem.h>

#include <Swift/Controllers/Roster/GroupRosterItem.h>

#include <Swift/Controllers/Roster/ItemOperations/SetAvatar.h>

#include <Swift/Controllers/Roster/ItemOperations/SetPresence.h>

#include <Swift/Controllers/Roster/ItemOperations/SetMUC.h>

#include <Swift/Controllers/Roster/Roster.h>

#include <Swift/Controllers/Roster/RosterVCardProvider.h>

#include <Swift/Controllers/UIEvents/InviteToMUCUIEvent.h>

#include <Swift/Controllers/UIEvents/RequestAddUserDialogUIEvent.h>

#include <Swift/Controllers/UIEvents/RequestChatUIEvent.h>

#include <Swift/Controllers/UIEvents/RequestInviteToMUCUIEvent.h>

#include <Swift/Controllers/UIEvents/ShowProfileForRosterItemUIEvent.h>

#include <Swift/Controllers/UIEvents/UIEventStream.h>

#include <Swift/Controllers/UIInterfaces/ChatWindow.h>

#include <Swift/Controllers/UIInterfaces/ChatWindowFactory.h>

#include <Swift/Controllers/XMPPEvents/EventController.h>

#define MUC_JOIN_WARNING_TIMEOUT_MILLISECONDS 60000

namespace Swift {

/**

* The controller does not gain ownership of the stanzaChannel, nor the factory.

*/

MUCController::MUCController (

const JID& self,

MUC::ref muc,

const boost::optional<std::string>& password,

const std::string &nick,

StanzaChannel* stanzaChannel,

IQRouter* iqRouter,

ChatWindowFactory* chatWindowFactory,

PresenceOracle* presenceOracle,

AvatarManager* avatarManager,

UIEventStream* uiEventStream,

bool useDelayForLatency,

TimerFactory* timerFactory,

EventController* eventController,

EntityCapsProvider* entityCapsProvider,

XMPPRoster* roster,

HistoryController* historyController,

MUCRegistry* mucRegistry,

HighlightManager* highlightManager,

ChatMessageParser* chatMessageParser,

bool isImpromptu,

AutoAcceptMUCInviteDecider* autoAcceptMUCInviteDecider,

VCardManager* vcardManager) :

ChatControllerBase(self, stanzaChannel, iqRouter, chatWindowFactory, muc->getJID(), presenceOracle, avatarManager, useDelayForLatency, uiEventStream, eventController, timerFactory, entityCapsProvider, historyController, mucRegistry, highlightManager, chatMessageParser, autoAcceptMUCInviteDecider), muc_(muc), nick_(nick), desiredNick_(nick), password_(password), renameCounter_(0), isImpromptu_(isImpromptu), isImpromptuAlreadyConfigured_(false) {

parting_ = true;

joined_ = false;

lastWasPresence_ = false;

shouldJoinOnReconnect_ = true;

doneGettingHistory_ = false;

events_ = uiEventStream;

xmppRoster_ = roster;

roster_ = new Roster(false, true);

rosterVCardProvider_ = new RosterVCardProvider(roster_, vcardManager, JID::WithResource);

completer_ = new TabComplete();

chatWindow_->setRosterModel(roster_);

chatWindow_->setTabComplete(completer_);

chatWindow_->onClosed.connect(boost::bind(&MUCController::handleWindowClosed, this));

chatWindow_->onOccupantSelectionChanged.connect(boost::bind(&MUCController::handleWindowOccupantSelectionChanged, this, _1));

chatWindow_->onOccupantActionSelected.connect(boost::bind(&MUCController::handleActionRequestedOnOccupant, this, _1, _2));

chatWindow_->onChangeSubjectRequest.connect(boost::bind(&MUCController::handleChangeSubjectRequest, this, _1));

chatWindow_->onBookmarkRequest.connect(boost::bind(&MUCController::handleBookmarkRequest, this));

chatWindow_->onConfigureRequest.connect(boost::bind(&MUCController::handleConfigureRequest, this, _1));

chatWindow_->onConfigurationFormCancelled.connect(boost::bind(&MUCController::handleConfigurationCancelled, this));

chatWindow_->onDestroyRequest.connect(boost::bind(&MUCController::handleDestroyRoomRequest, this));

chatWindow_->onInviteToChat.connect(boost::bind(&MUCController::handleInvitePersonToThisMUCRequest, this, _1));

chatWindow_->onGetAffiliationsRequest.connect(boost::bind(&MUCController::handleGetAffiliationsRequest, this));

chatWindow_->onChangeAffiliationsRequest.connect(boost::bind(&MUCController::handleChangeAffiliationsRequest, this, _1));

muc_->onJoinComplete.connect(boost::bind(&MUCController::handleJoinComplete, this, _1));

muc_->onJoinFailed.connect(boost::bind(&MUCController::handleJoinFailed, this, _1));

muc_->onOccupantJoined.connect(boost::bind(&MUCController::handleOccupantJoined, this, _1));

muc_->onOccupantPresenceChange.connect(boost::bind(&MUCController::handleOccupantPresenceChange, this, _1));

muc_->onOccupantLeft.connect(boost::bind(&MUCController::handleOccupantLeft, this, _1, _2, _3));

muc_->onRoleChangeFailed.connect(boost::bind(&MUCController::handleOccupantRoleChangeFailed, this, _1, _2, _3));

muc_->onAffiliationListReceived.connect(boost::bind(&MUCController::handleAffiliationListReceived, this, _1, _2));

muc_->onConfigurationFailed.connect(boost::bind(&MUCController::handleConfigurationFailed, this, _1));

muc_->onConfigurationFormReceived.connect(boost::bind(&MUCController::handleConfigurationFormReceived, this, _1));

highlighter_->setMode(isImpromptu_ ? Highlighter::ChatMode : Highlighter::MUCMode);

highlighter_->setNick(nick_);

if (timerFactory) {

loginCheckTimer_ = boost::shared_ptr<Timer>(timerFactory->createTimer(MUC_JOIN_WARNING_TIMEOUT_MILLISECONDS));

loginCheckTimer_->onTick.connect(boost::bind(&MUCController::handleJoinTimeoutTick, this));

loginCheckTimer_->start();

}

if (isImpromptu) {

muc_->onUnlocked.connect(boost::bind(&MUCController::handleRoomUnlocked, this));

chatWindow_->convertToMUC(ChatWindow::ImpromptuMUC);

} else {

muc_->onOccupantRoleChanged.connect(boost::bind(&MUCController::handleOccupantRoleChanged, this, _1, _2, _3));

muc_->onOccupantAffiliationChanged.connect(boost::bind(&MUCController::handleOccupantAffiliationChanged, this, _1, _2, _3));

chatWindow_->convertToMUC(ChatWindow::StandardMUC);

chatWindow_->setName(muc->getJID().getNode());

}

setOnline(true);

if (avatarManager_ != NULL) {

avatarChangedConnection_ = (avatarManager_->onAvatarChanged.connect(boost::bind(&MUCController::handleAvatarChanged, this, _1)));

}

handleBareJIDCapsChanged(muc->getJID());

eventStream_->onUIEvent.connect(boost::bind(&MUCController::handleUIEvent, this, _1));

}

MUCController::~MUCController() {

eventStream_->onUIEvent.disconnect(boost::bind(&MUCController::handleUIEvent, this, _1));

chatWindow_->setRosterModel(NULL);

delete rosterVCardProvider_;

delete roster_;

if (loginCheckTimer_) {

loginCheckTimer_->stop();

}

chatWindow_->setTabComplete(NULL);

delete completer_;

}

void MUCController::cancelReplaces() {

lastWasPresence_ = false;

}

void MUCController::handleWindowOccupantSelectionChanged(ContactRosterItem* item) {

std::vector<ChatWindow::OccupantAction> actions;

if (item) {

MUCOccupant::Affiliation affiliation = muc_->getOccupant(getNick()).getAffiliation();

MUCOccupant::Role role = muc_->getOccupant(getNick()).getRole();

if (role == MUCOccupant::Moderator && !isImpromptu_)

{

if (affiliation == MUCOccupant::Admin || affiliation == MUCOccupant::Owner) {

actions.push_back(ChatWindow::Ban);

}

actions.push_back(ChatWindow::Kick);

actions.push_back(ChatWindow::MakeModerator);

actions.push_back(ChatWindow::MakeParticipant);

actions.push_back(ChatWindow::MakeVisitor);

}

// Add contact is available only if the real JID is also available

if (muc_->getOccupant(item->getJID().getResource()).getRealJID()) {

actions.push_back(ChatWindow::AddContact);

}

actions.push_back(ChatWindow::ShowProfile);

}

chatWindow_->setAvailableOccupantActions(actions);

}

void MUCController::handleActionRequestedOnOccupant(ChatWindow::OccupantAction action, ContactRosterItem* item) {

JID mucJID = item->getJID();

MUCOccupant occupant = muc_->getOccupant(mucJID.getResource());

JID realJID;

if (occupant.getRealJID()) {

realJID = occupant.getRealJID().get();

}

switch (action) {

case ChatWindow::Kick: muc_->kickOccupant(mucJID);break;

case ChatWindow::Ban: muc_->changeAffiliation(realJID, MUCOccupant::Outcast);break;

case ChatWindow::MakeModerator: muc_->changeOccupantRole(mucJID, MUCOccupant::Moderator);break;

case ChatWindow::MakeParticipant: muc_->changeOccupantRole(mucJID, MUCOccupant::Participant);break;

case ChatWindow::MakeVisitor: muc_->changeOccupantRole(mucJID, MUCOccupant::Visitor);break;

case ChatWindow::AddContact: if (occupant.getRealJID()) events_->send(boost::make_shared<RequestAddUserDialogUIEvent>(realJID, occupant.getNick()));break;

case ChatWindow::ShowProfile: events_->send(boost::make_shared<ShowProfileForRosterItemUIEvent>(mucJID));break;

}

}

void MUCController::handleBareJIDCapsChanged(const JID& /*jid*/) {

ChatWindow::Tristate support = ChatWindow::Yes;

bool any = false;

foreach (const std::string& nick, currentOccupants_) {

DiscoInfo::ref disco = entityCapsProvider_->getCaps(toJID_.toBare().toString() + "/" + nick);

if (disco && disco->hasFeature(DiscoInfo::MessageCorrectionFeature)) {

any = true;

} else {

support = ChatWindow::Maybe;

}

}

if (!any) {

support = ChatWindow::No;

}

chatWindow_->setCorrectionEnabled(support);

}

/**

* Join the MUC if not already in it.

*/

void MUCController::rejoin() {

if (parting_) {

joined_ = false;

parting_ = false;

if (password_) {

muc_->setPassword(*password_);

}

//FIXME: check for received activity

#ifdef SWIFT_EXPERIMENTAL_HISTORY

if (lastActivity_ == boost::posix_time::not_a_date_time && historyController_) {

lastActivity_ = historyController_->getLastTimeStampFromMUC(selfJID_, toJID_);

}

#endif

if (lastActivity_ == boost::posix_time::not_a_date_time) {

muc_->joinAs(nick_);

}

else {

muc_->joinWithContextSince(nick_, lastActivity_);

}

}

}

bool MUCController::isJoined() {

return joined_;

}

const std::string& MUCController::getNick() {

return nick_;

}

const boost::optional<std::string> MUCController::getPassword() const {

return password_;

}

bool MUCController::isImpromptu() const {

return isImpromptu_;

}

std::map<std::string, JID> MUCController::getParticipantJIDs() const {

std::map<std::string, JID> participants;

typedef std::pair<std::string, MUCOccupant> MUCOccupantPair;

std::map<std::string, MUCOccupant> occupants = muc_->getOccupants();

foreach(const MUCOccupantPair& occupant, occupants) {

if (occupant.first != nick_) {

participants[occupant.first] = occupant.second.getRealJID().is_initialized() ? occupant.second.getRealJID().get().toBare() : JID();

}

}

return participants;

}

void MUCController::sendInvites(const std::vector<JID>& jids, const std::string& reason) const {

foreach (const JID& jid, jids) {

muc_->invitePerson(jid, reason, isImpromptu_);

}

}

void MUCController::handleJoinTimeoutTick() {

receivedActivity();

chatWindow_->addSystemMessage(chatMessageParser_->parseMessageBody(str(format(QT_TRANSLATE_NOOP("", "Room %1% is not responding. This operation may never complete.")) % toJID_.toString())), ChatWindow::DefaultDirection);

}

void MUCController::receivedActivity() {

if (loginCheckTimer_) {

loginCheckTimer_->stop();

}

}

#pragma clang diagnostic push

#pragma clang diagnostic ignored "-Wswitch-enum"

void MUCController::handleJoinFailed(boost::shared_ptr<ErrorPayload> error) {

receivedActivity();

std::string errorMessage = QT_TRANSLATE_NOOP("", "Unable to enter this room");

std::string rejoinNick;

if (error) {

switch (error->getCondition()) {

case ErrorPayload::Conflict:

rejoinNick = nick_ + "_";

errorMessage = str(format(QT_TRANSLATE_NOOP("", "Unable to enter this room as %1%, retrying as %2%")) % nick_ % rejoinNick);

break;

case ErrorPayload::JIDMalformed:

errorMessage += ": ";

errorMessage += QT_TRANSLATE_NOOP("", "No nickname specified");

break;

case ErrorPayload::NotAuthorized:

errorMessage += ": ";

errorMessage += QT_TRANSLATE_NOOP("", "The correct room password is needed");

break;

case ErrorPayload::RegistrationRequired:

errorMessage += ": ";

errorMessage += QT_TRANSLATE_NOOP("", "Only members may enter");

break;

case ErrorPayload::Forbidden:

errorMessage += ": ";

errorMessage += QT_TRANSLATE_NOOP("", "You are banned from the room");

break;

case ErrorPayload::ServiceUnavailable:

errorMessage += ": ";

errorMessage += QT_TRANSLATE_NOOP("", "The room is full");

break;

case ErrorPayload::ItemNotFound:

errorMessage += ": ";

errorMessage += QT_TRANSLATE_NOOP("", "The room does not exist");

break;

default: break;

}

}

errorMessage = str(format(QT_TRANSLATE_NOOP("", "Couldn't join room: %1%.")) % errorMessage);

chatWindow_->addErrorMessage(chatMessageParser_->parseMessageBody(errorMessage));

parting_ = true;

if (!rejoinNick.empty() && renameCounter_ < 10) {

renameCounter_++;

setNick(rejoinNick);

rejoin();

}

}

#pragma clang diagnostic pop

void MUCController::handleJoinComplete(const std::string& nick) {

receivedActivity();

renameCounter_ = 0;

joined_ = true;

std::string joinMessage;

if (isImpromptu_) {

joinMessage = str(format(QT_TRANSLATE_NOOP("", "You have joined the chat as %1%.")) % nick);

} else {

joinMessage = str(format(QT_TRANSLATE_NOOP("", "You have entered room %1% as %2%.")) % toJID_.toString() % nick);

}

setNick(nick);

#ifdef SWIFT_EXPERIMENTAL_HISTORY

addRecentLogs();

#endif

clearPresenceQueue();

shouldJoinOnReconnect_ = true;

setEnabled(true);

if (isImpromptu_) {

setAvailableRoomActions(MUCOccupant::NoAffiliation, MUCOccupant::Participant);

} else {

MUCOccupant occupant = muc_->getOccupant(nick);

setAvailableRoomActions(occupant.getAffiliation(), occupant.getRole());

}

onUserJoined();

if (isImpromptu_) {

setImpromptuWindowTitle();

}

}

void MUCController::handleAvatarChanged(const JID& jid) {

if (parting_ || !jid.equals(toJID_, JID::WithoutResource)) {

return;

}

roster_->applyOnItems(SetAvatar(jid, avatarManager_->getAvatarPath(jid), JID::WithResource));

}

void MUCController::handleWindowClosed() {

parting_ = true;

shouldJoinOnReconnect_ = false;

muc_->part();

onUserLeft();

}

void MUCController::handleOccupantJoined(const MUCOccupant& occupant) {

if (nick_ != occupant.getNick()) {

completer_->addWord(occupant.getNick());

}

receivedActivity();

JID jid(nickToJID(occupant.getNick()));

JID realJID;

if (occupant.getRealJID()) {

realJID = occupant.getRealJID().get();

}

currentOccupants_.insert(occupant.getNick());

NickJoinPart event(occupant.getNick(), Join);

appendToJoinParts(joinParts_, event);

MUCOccupant::Role role = MUCOccupant::Participant;

MUCOccupant::Affiliation affiliation = MUCOccupant::NoAffiliation;

if (!isImpromptu_) {

role = occupant.getRole();

affiliation = occupant.getAffiliation();

}

std::string groupName(roleToGroupName(role));

roster_->addContact(jid, realJID, occupant.getNick(), groupName, avatarManager_->getAvatarPath(jid));

roster_->applyOnItems(SetMUC(jid, role, affiliation));

roster_->getGroup(groupName)->setManualSort(roleToSortName(role));

if (joined_) {

std::string joinString;

if (role != MUCOccupant::NoRole && role != MUCOccupant::Participant) {

joinString = str(format(QT_TRANSLATE_NOOP("", "%1% has entered the %3% as a %2%.")) % occupant.getNick() % roleToFriendlyName(role) % (isImpromptu_ ? QT_TRANSLATE_NOOP("", "chat") : QT_TRANSLATE_NOOP("", "room")));

}

else {

joinString = str(format(QT_TRANSLATE_NOOP("", "%1% has entered the %2%.")) % occupant.getNick() % (isImpromptu_ ? QT_TRANSLATE_NOOP("", "chat") : QT_TRANSLATE_NOOP("", "room")));

}

if (shouldUpdateJoinParts()) {

updateJoinParts();

} else {

addPresenceMessage(joinString);

}

if (isImpromptu_) {

setImpromptuWindowTitle();

onActivity("");

}

}

if (avatarManager_ != NULL) {

handleAvatarChanged(jid);

}

}

void MUCController::addPresenceMessage(const std::string& message) {

lastWasPresence_ = true;

chatWindow_->addPresenceMessage(chatMessageParser_->parseMessageBody(message), ChatWindow::DefaultDirection);

}

void MUCController::setAvailableRoomActions(const MUCOccupant::Affiliation& affiliation, const MUCOccupant::Role& role)

{

std::vector<ChatWindow::RoomAction> actions;

if (role <= MUCOccupant::Participant) {

actions.push_back(ChatWindow::ChangeSubject);

}

if (affiliation == MUCOccupant::Owner) {

actions.push_back(ChatWindow::Configure);

}

if (affiliation <= MUCOccupant::Admin) {

actions.push_back(ChatWindow::Affiliations);

}

if (affiliation == MUCOccupant::Owner) {

actions.push_back(ChatWindow::Destroy);

}

if (role <= MUCOccupant::Visitor) {

actions.push_back(ChatWindow::Invite);

}

chatWindow_->setAvailableRoomActions(actions);

}

void MUCController::clearPresenceQueue() {

lastWasPresence_ = false;

joinParts_.clear();

}

std::string MUCController::roleToFriendlyName(MUCOccupant::Role role) {

switch (role) {

case MUCOccupant::Moderator: return QT_TRANSLATE_NOOP("", "moderator");

case MUCOccupant::Participant: return QT_TRANSLATE_NOOP("", "participant");

case MUCOccupant::Visitor: return QT_TRANSLATE_NOOP("", "visitor");

case MUCOccupant::NoRole: return "";

}

assert(false);

return "";

}

std::string MUCController::roleToSortName(MUCOccupant::Role role) {

switch (role) {

case MUCOccupant::Moderator: return "1";

case MUCOccupant::Participant: return "2";

case MUCOccupant::Visitor: return "3";

case MUCOccupant::NoRole: return "4";

}

assert(false);

return "5";

}

JID MUCController::nickToJID(const std::string& nick) {

return JID(toJID_.getNode(), toJID_.getDomain(), nick);

}

bool MUCController::messageTargetsMe(boost::shared_ptr<Message> message) {

std::string stringRegexp(".*\\b" + boost::to_lower_copy(nick_) + "\\b.*");

boost::regex myRegexp(stringRegexp);

return boost::regex_match(boost::to_lower_copy(message->getBody()), myRegexp);

}

void MUCController::preHandleIncomingMessage(boost::shared_ptr<MessageEvent> messageEvent) {

if (messageEvent->getStanza()->getType() == Message::Groupchat) {

lastActivity_ = boost::posix_time::microsec_clock::universal_time();

}

clearPresenceQueue();

boost::shared_ptr<Message> message = messageEvent->getStanza();

if (joined_ && messageEvent->getStanza()->getFrom().getResource() != nick_ && messageTargetsMe(message) && !message->getPayload<Delay>() && messageEvent->isReadable()) {

chatWindow_->flash();

}

else {

messageEvent->setTargetsMe(false);

}

if (messageEvent->isReadable() && isImpromptu_) {

chatWindow_->flash(); /* behave like a regular char*/

}

if (joined_) {

std::string nick = message->getFrom().getResource();

if (nick != nick_ && currentOccupants_.find(nick) != currentOccupants_.end()) {

completer_->addWord(nick);

}

}

/*Buggy implementations never send the status code, so use an incoming message as a hint that joining's done (e.g. the old ejabberd on psi-im.org).*/

receivedActivity();

joined_ = true;

if (message->hasSubject() && message->getBody().empty()) {

chatWindow_->addSystemMessage(chatMessageParser_->parseMessageBody(str(format(QT_TRANSLATE_NOOP("", "The room subject is now: %1%")) % message->getSubject())), ChatWindow::DefaultDirection);;

chatWindow_->setSubject(message->getSubject());

doneGettingHistory_ = true;

}

if (!doneGettingHistory_ && !message->getPayload<Delay>()) {

doneGettingHistory_ = true;

}

if (!doneGettingHistory_) {

checkDuplicates(message);

messageEvent->conclude();

}

}

void MUCController::postHandleIncomingMessage(boost::shared_ptr<MessageEvent> messageEvent, const HighlightAction& highlight) {

boost::shared_ptr<Message> message = messageEvent->getStanza();

if (joined_ && messageEvent->getStanza()->getFrom().getResource() != nick_ && !message->getPayload<Delay>()) {

if (messageTargetsMe(message) || isImpromptu_) {

eventController_->handleIncomingEvent(messageEvent);

if (!messageEvent->getConcluded()) {

highlighter_->handleHighlightAction(highlight);

}

}

}

}

void MUCController::handleOccupantRoleChanged(const std::string& nick, const MUCOccupant& occupant, const MUCOccupant::Role& oldRole) {

clearPresenceQueue();

receivedActivity();

JID jid(nickToJID(nick));

roster_->removeContactFromGroup(jid, roleToGroupName(oldRole));

JID realJID;

if (occupant.getRealJID()) {

realJID = occupant.getRealJID().get();

}

std::string group(roleToGroupName(occupant.getRole()));

roster_->addContact(jid, realJID, nick, group, avatarManager_->getAvatarPath(jid));

roster_->getGroup(group)->setManualSort(roleToSortName(occupant.getRole()));

roster_->applyOnItems(SetMUC(jid, occupant.getRole(), occupant.getAffiliation()));

chatWindow_->addSystemMessage(chatMessageParser_->parseMessageBody(str(format(QT_TRANSLATE_NOOP("", "%1% is now a %2%")) % nick % roleToFriendlyName(occupant.getRole()))), ChatWindow::DefaultDirection);

if (nick == nick_) {

setAvailableRoomActions(occupant.getAffiliation(), occupant.getRole());

}

}

void MUCController::handleOccupantAffiliationChanged(const std::string& nick, const MUCOccupant::Affiliation& affiliation, const MUCOccupant::Affiliation& /*oldAffiliation*/)

{

if (nick == nick_) {

setAvailableRoomActions(affiliation, muc_->getOccupant(nick_).getRole());

}

JID jid(nickToJID(nick));

MUCOccupant occupant = muc_->getOccupant(nick);

roster_->applyOnItems(SetMUC(jid, occupant.getRole(), affiliation));

}

std::string MUCController::roleToGroupName(MUCOccupant::Role role) {

std::string result;

switch (role) {

case MUCOccupant::Moderator: result = QT_TRANSLATE_NOOP("", "Moderators"); break;

case MUCOccupant::Participant: result = QT_TRANSLATE_NOOP("", "Participants"); break;

case MUCOccupant::Visitor: result = QT_TRANSLATE_NOOP("", "Visitors"); break;

case MUCOccupant::NoRole: result = QT_TRANSLATE_NOOP("", "Occupants"); break;

}

return result;

}

void MUCController::setOnline(bool online) {

ChatControllerBase::setOnline(online);

if (!online) {

muc_->part();

parting_ = true;

processUserPart();

} else {

if (shouldJoinOnReconnect_) {

renameCounter_ = 0;

if (isImpromptu_) {

chatWindow_->addSystemMessage(chatMessageParser_->parseMessageBody(QT_TRANSLATE_NOOP("", "Trying to join chat")), ChatWindow::DefaultDirection);

} else {

chatWindow_->addSystemMessage(chatMessageParser_->parseMessageBody(str(format(QT_TRANSLATE_NOOP("", "Trying to enter room %1%")) % toJID_.toString())), ChatWindow::DefaultDirection);

}

if (loginCheckTimer_) {

loginCheckTimer_->start();

}

setNick(desiredNick_);

rejoin();

}

}

}

void MUCController::processUserPart() {

roster_->removeAll();

/* handleUserLeft won't throw a part back up unless this is called

when it doesn't yet know we've left - which only happens on

disconnect, so call with disconnect here so if the signal does

bubble back up, it'll be with the right type.*/

muc_->handleUserLeft(MUC::Disconnect);

setEnabled(false);

}

bool MUCController::shouldUpdateJoinParts() {

return lastWasPresence_;

}

void MUCController::handleOccupantLeft(const MUCOccupant& occupant, MUC::LeavingType type, const std::string& reason) {

NickJoinPart event(occupant.getNick(), Part);

appendToJoinParts(joinParts_, event);

currentOccupants_.erase(occupant.getNick());

completer_->removeWord(occupant.getNick());

std::string partMessage;

bool clearAfter = false;

if (occupant.getNick() != nick_) {

std::string partType;

switch (type) {

case MUC::LeaveKick: clearPresenceQueue(); clearAfter = true; partType = " (kicked)"; break;

case MUC::LeaveBan: clearPresenceQueue(); clearAfter = true; partType = " (banned)"; break;

case MUC::LeaveNotMember: clearPresenceQueue(); clearAfter = true; partType = " (no longer a member)"; break;

case MUC::LeaveDestroy:

case MUC::Disconnect:

case MUC::LeavePart: break;

}

if (isImpromptu_) {

partMessage = str(format(QT_TRANSLATE_NOOP("", "%1% has left the chat%2%")) % occupant.getNick() % partType);

} else {

partMessage = str(format(QT_TRANSLATE_NOOP("", "%1% has left the room%2%")) % occupant.getNick() % partType);

}

}

else if (isImpromptu_) {

switch (type) {

case MUC::LeaveKick:

case MUC::LeaveBan: clearPresenceQueue(); clearAfter = true; partMessage = QT_TRANSLATE_NOOP("", "You have been removed from this chat"); break;

case MUC::LeaveNotMember: clearPresenceQueue(); clearAfter = true; partMessage = QT_TRANSLATE_NOOP("", "You have been removed from this chat"); break;

case MUC::LeaveDestroy: clearPresenceQueue(); clearAfter = true; partMessage = QT_TRANSLATE_NOOP("", "This chat has ended"); break;

case MUC::Disconnect:

case MUC::LeavePart: partMessage = QT_TRANSLATE_NOOP("", "You have left the chat");

}

}

else {

switch (type) {

case MUC::LeaveKick: clearPresenceQueue(); clearAfter = true; partMessage = QT_TRANSLATE_NOOP("", "You have been kicked out of the room"); break;

case MUC::LeaveBan: clearPresenceQueue(); clearAfter = true; partMessage = QT_TRANSLATE_NOOP("", "You have been banned from the room"); break;

case MUC::LeaveNotMember: clearPresenceQueue(); clearAfter = true; partMessage = QT_TRANSLATE_NOOP("", "You are no longer a member of the room and have been removed"); break;

case MUC::LeaveDestroy: clearPresenceQueue(); clearAfter = true; partMessage = QT_TRANSLATE_NOOP("", "The room has been destroyed"); break;

case MUC::Disconnect:

case MUC::LeavePart: partMessage = QT_TRANSLATE_NOOP("", "You have left the room");

}

}

if (!reason.empty()) {

partMessage += " (" + reason + ")";

}

partMessage += ".";

if (occupant.getNick() != nick_) {

if (shouldUpdateJoinParts()) {

updateJoinParts();

} else {

addPresenceMessage(partMessage);

}

roster_->removeContact(JID(toJID_.getNode(), toJID_.getDomain(), occupant.getNick()));

} else {

addPresenceMessage(partMessage);

parting_ = true;

processUserPart();

}

if (clearAfter) {

clearPresenceQueue();

}

if (isImpromptu_) {

setImpromptuWindowTitle();

}

}

void MUCController::handleOccupantPresenceChange(boost::shared_ptr<Presence> presence) {

receivedActivity();

roster_->applyOnItems(SetPresence(presence, JID::WithResource));

}

bool MUCController::isIncomingMessageFromMe(boost::shared_ptr<Message> message) {

JID from = message->getFrom();

return nick_ == from.getResource();

}

std::string MUCController::senderDisplayNameFromMessage(const JID& from) {

return from.getResource();

}

void MUCController::preSendMessageRequest(boost::shared_ptr<Message> message) {

message->setType(Swift::Message::Groupchat);

}

boost::optional<boost::posix_time::ptime> MUCController::getMessageTimestamp(boost::shared_ptr<Message> message) const {

return message->getTimestampFrom(toJID_);

}

void MUCController::updateJoinParts() {

}

void MUCController::appendToJoinParts(std::vector<NickJoinPart>& joinParts, const NickJoinPart& newEvent) {

std::vector<NickJoinPart>::iterator it = joinParts.begin();

bool matched = false;

for (; it != joinParts.end(); ++it) {

if ((*it).nick == newEvent.nick) {

matched = true;

JoinPart type = (*it).type;

switch (newEvent.type) {

case Join: type = (type == Part) ? PartThenJoin : Join; break;

case Part: type = (type == Join) ? JoinThenPart : Part; break;

case PartThenJoin: break;

case JoinThenPart: break;

}

(*it).type = type;

break;

}

}

if (!matched) {

joinParts.push_back(newEvent);

}

}

std::string MUCController::concatenateListOfNames(const std::vector<NickJoinPart>& joinParts) {

std::string result;

for (size_t i = 0; i < joinParts.size(); i++) {

if (i > 0) {

if (i < joinParts.size() - 1) {

result += ", ";

} else {

result += QT_TRANSLATE_NOOP("", " and ");

}

}

NickJoinPart event = joinParts[i];

result += event.nick;

}

return result;

}

std::string MUCController::generateJoinPartString(const std::vector<NickJoinPart>& joinParts, bool isImpromptu) {

std::vector<NickJoinPart> sorted[4];

std::string eventStrings[4];

foreach (NickJoinPart event, joinParts) {

sorted[event.type].push_back(event);

}

std::string result;

std::vector<JoinPart> populatedEvents;

for (size_t i = 0; i < 4; i++) {

std::string names = concatenateListOfNames(sorted[i]);

if (!names.empty()) {

std::string eventString;

switch (i) {

case Join:

if (sorted[i].size() > 1) {

eventString = (isImpromptu ? QT_TRANSLATE_NOOP("", "%1% have joined the chat") : QT_TRANSLATE_NOOP("", "%1% have entered the room"));

}

else {

eventString = (isImpromptu ? QT_TRANSLATE_NOOP("", "%1% has joined the chat") : QT_TRANSLATE_NOOP("", "%1% has entered the room"));

}

break;

case Part:

if (sorted[i].size() > 1) {

eventString = (isImpromptu ? QT_TRANSLATE_NOOP("", "%1% have left the chat") : QT_TRANSLATE_NOOP("", "%1% have left the room"));

}

else {

eventString = (isImpromptu ? QT_TRANSLATE_NOOP("", "%1% have left the chat") : QT_TRANSLATE_NOOP("", "%1% has left the room"));

}

break;

case JoinThenPart:

if (sorted[i].size() > 1) {

eventString = (isImpromptu ? QT_TRANSLATE_NOOP("", "%1% have joined then left the chat") : QT_TRANSLATE_NOOP("", "%1% have entered then left the room"));

}

else {

eventString = (isImpromptu ? QT_TRANSLATE_NOOP("", "%1% has joined then left the chat") : QT_TRANSLATE_NOOP("", "%1% has entered then left the room"));

}

break;

case PartThenJoin:

if (sorted[i].size() > 1) {

eventString = (isImpromptu ? QT_TRANSLATE_NOOP("", "%1% have left then returned to the chat") : QT_TRANSLATE_NOOP("", "%1% have left then returned to the room"));

}

else {

eventString = (isImpromptu ? QT_TRANSLATE_NOOP("", "%1% has left then returned to the chat") : QT_TRANSLATE_NOOP("", "%1% has left then returned to the room"));

}

break;

}

populatedEvents.push_back(static_cast<JoinPart>(i));

eventStrings[i] = str(boost::format(eventString) % names);

}

}

for (size_t i = 0; i < populatedEvents.size(); i++) {

if (i > 0) {

if (i < populatedEvents.size() - 1) {

result += ", ";

} else {

result += QT_TRANSLATE_NOOP("", " and ");

}

}

result += eventStrings[populatedEvents[i]];

}

return result;

}

void MUCController::handleChangeSubjectRequest(const std::string& subject) {

muc_->changeSubject(subject);

}

void MUCController::handleBookmarkRequest() {

const JID jid = muc_->getJID();

MUCBookmark bookmark(jid, jid.toBare().toString());

bookmark.setPassword(password_);

bookmark.setNick(nick_);

chatWindow_->showBookmarkWindow(bookmark);

}

void MUCController::handleConfigureRequest(Form::ref form) {

if (form) {

muc_->configureRoom(form);

}

else {

muc_->requestConfigurationForm();

}

}

void MUCController::handleConfigurationFailed(ErrorPayload::ref error) {

std::string errorMessage = getErrorMessage(error);

errorMessage = str(format(QT_TRANSLATE_NOOP("", "Room configuration failed: %1%.")) % errorMessage);

chatWindow_->addErrorMessage(chatMessageParser_->parseMessageBody(errorMessage));

}

void MUCController::handleOccupantRoleChangeFailed(ErrorPayload::ref error, const JID&, MUCOccupant::Role) {

std::string errorMessage = getErrorMessage(error);

errorMessage = str(format(QT_TRANSLATE_NOOP("", "Occupant role change failed: %1%.")) % errorMessage);

chatWindow_->addErrorMessage(chatMessageParser_->parseMessageBody(errorMessage));

}

void MUCController::configureAsImpromptuRoom(Form::ref form) {

muc_->configureRoom(buildImpromptuRoomConfiguration(form));

isImpromptuAlreadyConfigured_ = true;

onImpromptuConfigCompleted();

}

void MUCController::handleConfigurationFormReceived(Form::ref form) {

if (isImpromptu_) {

if (!isImpromptuAlreadyConfigured_) {

configureAsImpromptuRoom(form);

}

} else {

chatWindow_->showRoomConfigurationForm(form);

}

}

void MUCController::handleConfigurationCancelled() {

muc_->cancelConfigureRoom();

}

void MUCController::handleDestroyRoomRequest() {

muc_->destroyRoom();

}

void MUCController::handleInvitePersonToThisMUCRequest(const std::vector<JID>& jidsToInvite) {

RequestInviteToMUCUIEvent::ImpromptuMode mode = isImpromptu_ ? RequestInviteToMUCUIEvent::Impromptu : RequestInviteToMUCUIEvent::NotImpromptu;

boost::shared_ptr<UIEvent> event(new RequestInviteToMUCUIEvent(muc_->getJID(), jidsToInvite, mode));

eventStream_->send(event);

}

void MUCController::handleUIEvent(boost::shared_ptr<UIEvent> event) {

boost::shared_ptr<InviteToMUCUIEvent> inviteEvent = boost::dynamic_pointer_cast<InviteToMUCUIEvent>(event);

if (inviteEvent && inviteEvent->getRoom() == muc_->getJID()) {

foreach (const JID& jid, inviteEvent->getInvites()) {

muc_->invitePerson(jid, inviteEvent->getReason(), isImpromptu_);

}

}

}

void MUCController::handleGetAffiliationsRequest() {

muc_->requestAffiliationList(MUCOccupant::Owner);

muc_->requestAffiliationList(MUCOccupant::Admin);

muc_->requestAffiliationList(MUCOccupant::Member);

muc_->requestAffiliationList(MUCOccupant::Outcast);

}

typedef std::pair<MUCOccupant::Affiliation, JID> AffiliationChangePair;

void MUCController::handleChangeAffiliationsRequest(const std::vector<std::pair<MUCOccupant::Affiliation, JID> >& changes) {

std::set<JID> addedJIDs;

foreach (const AffiliationChangePair& change, changes) {

if (change.first != MUCOccupant::NoAffiliation) {

addedJIDs.insert(change.second);

}

}

foreach (const AffiliationChangePair& change, changes) {

if (change.first != MUCOccupant::NoAffiliation || addedJIDs.find(change.second) == addedJIDs.end()) {

muc_->changeAffiliation(change.second, change.first);

}

}

}

void MUCController::handleAffiliationListReceived(MUCOccupant::Affiliation affiliation, const std::vector<JID>& jids) {

chatWindow_->setAffiliations(affiliation, jids);

}

void MUCController::logMessage(const std::string& message, const JID& fromJID, const JID& toJID, const boost::posix_time::ptime& timeStamp, bool isIncoming) {

// log only incoming messages

if (isIncoming && historyController_) {

historyController_->addMessage(message, fromJID, toJID, HistoryMessage::Groupchat, timeStamp);

}

}

void MUCController::addRecentLogs() {

if (!historyController_) {

return;

}

joinContext_ = historyController_->getMUCContext(selfJID_, toJID_, lastActivity_);

foreach (const HistoryMessage& message, joinContext_) {

bool senderIsSelf = nick_ == message.getFromJID().getResource();

// the chatWindow uses utc timestamps

addMessage(message.getMessage(), senderDisplayNameFromMessage(message.getFromJID()), senderIsSelf, boost::shared_ptr<SecurityLabel>(new SecurityLabel()), avatarManager_->getAvatarPath(message.getFromJID()), message.getTime() - boost::posix_time::hours(message.getOffset()), HighlightAction());

}

}

void MUCController::checkDuplicates(boost::shared_ptr<Message> newMessage) {

std::string body = newMessage->getBody();

JID jid = newMessage->getFrom();

boost::optional<boost::posix_time::ptime> time = newMessage->getTimestamp();

reverse_foreach (const HistoryMessage& message, joinContext_) {

boost::posix_time::ptime messageTime = message.getTime() - boost::posix_time::hours(message.getOffset());

if (time && time < messageTime) {

break;

}

if (time && time != messageTime) {

continue;

}

if (message.getFromJID() != jid) {

continue;

}

if (message.getMessage() != body) {

continue;

}

// Mark the message as unreadable

newMessage->setBody("");

}

}

void MUCController::setNick(const std::string& nick) {

nick_ = nick;

highlighter_->setNick(nick_);

}

Form::ref MUCController::buildImpromptuRoomConfiguration(Form::ref roomConfigurationForm) {

Form::ref result = boost::make_shared<Form>(Form::SubmitType);

std::string impromptuConfigs[] = { "muc#roomconfig_enablelogging", "muc#roomconfig_persistentroom", "muc#roomconfig_publicroom", "muc#roomconfig_whois"};

std::set<std::string> impromptuConfigsMissing(impromptuConfigs, impromptuConfigs + 4);

foreach (boost::shared_ptr<FormField> field, roomConfigurationForm->getFields()) {

boost::shared_ptr<FormField> resultField;

if (field->getName() == "muc#roomconfig_enablelogging") {

resultField = boost::make_shared<FormField>(FormField::BooleanType, "0");

}

if (field->getName() == "muc#roomconfig_persistentroom") {

resultField = boost::make_shared<FormField>(FormField::BooleanType, "0");

}

if (field->getName() == "muc#roomconfig_publicroom") {

resultField = boost::make_shared<FormField>(FormField::BooleanType, "0");

}

if (field->getName() == "muc#roomconfig_whois") {

resultField = boost::make_shared<FormField>(FormField::ListSingleType, "anyone");

}

if (field->getName() == "FORM_TYPE") {

resultField = boost::make_shared<FormField>(FormField::HiddenType, "http://jabber.org/protocol/muc#roomconfig");

}

if (resultField) {

impromptuConfigsMissing.erase(field->getName());

resultField->setName(field->getName());

result->addField(resultField);

}

}

foreach (const std::string& config, impromptuConfigsMissing) {

if (config == "muc#roomconfig_publicroom") {

chatWindow_->addSystemMessage(chatMessageParser_->parseMessageBody(QT_TRANSLATE_NOOP("", "This server doesn't support hiding your chat from other users.")), ChatWindow::DefaultDirection);

} else if (config == "muc#roomconfig_whois") {

chatWindow_->addSystemMessage(chatMessageParser_->parseMessageBody(QT_TRANSLATE_NOOP("", "This server doesn't support sharing people's real identity in this chat.")), ChatWindow::DefaultDirection);

}

}

return result;

}

void MUCController::setImpromptuWindowTitle() {

std::string title;

typedef std::pair<std::string, MUCOccupant> StringMUCOccupantPair;

std::map<std::string, MUCOccupant> occupants = muc_->getOccupants();

if (occupants.size() <= 1) {

title = QT_TRANSLATE_NOOP("", "Empty Chat");

} else {

foreach (StringMUCOccupantPair pair, occupants) {

if (pair.first != nick_) {

title += (title.empty() ? "" : ", ") + pair.first;

}

}

}

chatWindow_->setName(title);

}

void MUCController::handleRoomUnlocked() {

// Handle buggy MUC implementations where the joined room already exists and is unlocked.

// Configure the room again in this case.

if (!isImpromptuAlreadyConfigured_) {

if (isImpromptu_ && (muc_->getOccupant(nick_).getAffiliation() == MUCOccupant::Owner)) {

muc_->requestConfigurationForm();

} else if (isImpromptu_) {

onImpromptuConfigCompleted();

}

}

}

}

/*

* Copyright (c) 2010-2014 Kevin Smith

* Licensed under the GNU General Public License v3.

* See Documentation/Licenses/GPLv3.txt for more information.

*/

#pragma once

#include <vector>

#include <string>

#include <boost/optional.hpp>

#include <boost/shared_ptr.hpp>

#include <boost/make_shared.hpp>

#include <boost/date_time/posix_time/posix_time.hpp>

#include <Swiften/Base/boost_bsignals.h>

#include <Swiften/Elements/SecurityLabelsCatalog.h>

#include <Swiften/Elements/ChatState.h>

#include <Swiften/Elements/Form.h>

#include <Swiften/Elements/MUCOccupant.h>

#include <Swiften/MUC/MUCBookmark.h>

#include <Swift/Controllers/HighlightManager.h>

namespace Swift {

class AvatarManager;

class TreeWidget;

class Roster;

class TabComplete;

class RosterItem;

class ContactRosterItem;

class FileTransferController;

class UserSearchWindow;

class ChatWindow {

public:

class ChatMessagePart {

public:

virtual ~ChatMessagePart() {}

};

class ChatMessage {

public:

ChatMessage() {}

ChatMessage(const std::string& text) {

append(boost::make_shared<ChatTextMessagePart>(text));

}

void append(const boost::shared_ptr<ChatMessagePart>& part) {

parts_.push_back(part);

}

const std::vector<boost::shared_ptr<ChatMessagePart> >& getParts() const {

return parts_;

}

private:

std::vector<boost::shared_ptr<ChatMessagePart> > parts_;

};

class ChatTextMessagePart : public ChatMessagePart {

public:

ChatTextMessagePart(const std::string& text) : text(text) {}

std::string text;

};

class ChatURIMessagePart : public ChatMessagePart {

public:

ChatURIMessagePart(const std::string& target) : target(target) {}

std::string target;

};

class ChatEmoticonMessagePart : public ChatMessagePart {

public:

std::string imagePath;

std::string alternativeText;

};

class ChatHighlightingMessagePart : public ChatMessagePart {

public:

std::string foregroundColor;

std::string backgroundColor;

std::string text;

};

enum AckState {Pending, Received, Failed};

enum ReceiptState {ReceiptRequested, ReceiptReceived, ReceiptFailed};

enum Tristate {Yes, No, Maybe};

enum OccupantAction {Kick, Ban, MakeModerator, MakeParticipant, MakeVisitor, AddContact, ShowProfile};

enum RoomAction {ChangeSubject, Configure, Affiliations, Destroy, Invite};

enum FileTransferState {WaitingForAccept, Negotiating, Transferring, Canceled, Finished, FTFailed};

enum WhiteboardSessionState {WhiteboardAccepted, WhiteboardTerminated, WhiteboardRejected};

enum BlockingState {BlockingUnsupported, IsBlocked, IsUnblocked};

enum Direction { UnknownDirection, DefaultDirection };

enum MUCType { StandardMUC, ImpromptuMUC };

ChatWindow() {}

virtual ~ChatWindow() {}

/** Add message to window.

* @return id of added message (for acks).

*/

virtual std::string addMessage(const ChatMessage& message, const std::string& senderName, bool senderIsSelf, boost::shared_ptr<SecurityLabel> label, const std::string& avatarPath, const boost::posix_time::ptime& time, const HighlightAction& highlight) = 0;

/** Adds action to window.

* @return id of added message (for acks);

*/

virtual std::string addAction(const ChatMessage& message, const std::string& senderName, bool senderIsSelf, boost::shared_ptr<SecurityLabel> label, const std::string& avatarPath, const boost::posix_time::ptime& time, const HighlightAction& highlight) = 0;

virtual void addSystemMessage(const ChatMessage& message, Direction direction) = 0;

virtual void addPresenceMessage(const ChatMessage& message, Direction direction) = 0;

virtual void addErrorMessage(const ChatMessage& message) = 0;

virtual void replaceMessage(const ChatMessage& message, const std::string& id, const boost::posix_time::ptime& time, const HighlightAction& highlight) = 0;

virtual void replaceWithAction(const ChatMessage& message, const std::string& id, const boost::posix_time::ptime& time, const HighlightAction& highlight) = 0;

// File transfer related stuff

virtual std::string addFileTransfer(const std::string& senderName, bool senderIsSelf, const std::string& filename, const boost::uintmax_t sizeInBytes) = 0;

virtual void setFileTransferProgress(std::string, const int percentageDone) = 0;

virtual void setFileTransferStatus(std::string, const FileTransferState state, const std::string& msg = "") = 0;

virtual void addMUCInvitation(const std::string& senderName, const JID& jid, const std::string& reason, const std::string& password, bool direct = true, bool isImpromptu = false, bool isContinuation = false) = 0;

virtual std::string addWhiteboardRequest(bool senderIsSelf) = 0;

virtual void setWhiteboardSessionStatus(std::string id, const ChatWindow::WhiteboardSessionState state) = 0;

// message receipts

virtual void setMessageReceiptState(const std::string& id, ChatWindow::ReceiptState state) = 0;

virtual void setContactChatState(ChatState::ChatStateType state) = 0;

virtual void setName(const std::string& name) = 0;

virtual void show() = 0;

virtual void activate() = 0;

virtual void setAvailableSecurityLabels(const std::vector<SecurityLabelsCatalog::Item>& labels) = 0;

virtual void setSecurityLabelsEnabled(bool enabled) = 0;

virtual void setCorrectionEnabled(Tristate enabled) = 0;

virtual void setFileTransferEnabled(Tristate enabled) = 0;

virtual void setUnreadMessageCount(int count) = 0;

virtual void convertToMUC(MUCType mucType) = 0;

// virtual TreeWidget *getTreeWidget() = 0;

virtual void setSecurityLabelsError() = 0;

virtual SecurityLabelsCatalog::Item getSelectedSecurityLabel() = 0;

virtual void setInputEnabled(bool enabled) = 0;

virtual void setRosterModel(Roster* model) = 0;

virtual void setTabComplete(TabComplete* completer) = 0;

virtual void setAckState(const std::string& id, AckState state) = 0;

virtual void flash() = 0;

virtual void setSubject(const std::string& subject) = 0;

virtual void setAffiliations(MUCOccupant::Affiliation, const std::vector<JID>&) = 0;

virtual void setAvailableRoomActions(const std::vector<RoomAction> &actions) = 0;

virtual void setBlockingState(BlockingState state) = 0;

virtual void setCanInitiateImpromptuChats(bool supportsImpromptu) = 0;

virtual void showBookmarkWindow(const MUCBookmark& bookmark) = 0;

/**

* Set an alert on the window.

* @param alertText Description of alert (required).

* @param buttonText Button text to use (optional, no button is shown if empty).

*/

virtual void setAlert(const std::string& alertText, const std::string& buttonText = "") = 0;

/**

* Removes an alert.

*/

virtual void cancelAlert() = 0;

/**

* Actions that can be performed on the selected occupant.

*/

virtual void setAvailableOccupantActions(const std::vector<OccupantAction>& actions) = 0;

/**

* A room configuration has been requested, show the form.

* If the form is cancelled, must emit onConfigurationFormCancelled().

*/

virtual void showRoomConfigurationForm(Form::ref) = 0;

boost::signal<void ()> onClosed;

boost::signal<void ()> onAllMessagesRead;

boost::signal<void (const std::string&, bool isCorrection)> onSendMessageRequest;

boost::signal<void ()> onSendCorrectionMessageRequest;

boost::signal<void ()> onUserTyping;

boost::signal<void ()> onUserCancelsTyping;

boost::signal<void ()> onAlertButtonClicked;

boost::signal<void (ContactRosterItem*)> onOccupantSelectionChanged;

boost::signal<void (ChatWindow::OccupantAction, ContactRosterItem*)> onOccupantActionSelected;

boost::signal<void (const std::string&)> onChangeSubjectRequest;

boost::signal<void ()> onBookmarkRequest;

boost::signal<void (Form::ref)> onConfigureRequest;

boost::signal<void ()> onDestroyRequest;

boost::signal<void (const std::vector<JID>&)> onInviteToChat;

boost::signal<void ()> onConfigurationFormCancelled;

boost::signal<void ()> onGetAffiliationsRequest;

boost::signal<void (MUCOccupant::Affiliation, const JID&)> onSetAffiliationRequest;

boost::signal<void (const std::vector<std::pair<MUCOccupant::Affiliation, JID> >& changes)> onChangeAffiliationsRequest;

boost::signal<void ()> onLogCleared;

// File transfer related

boost::signal<void (std::string /* id */)> onFileTransferCancel;

boost::signal<void (std::string /* id */, std::string /* description */)> onFileTransferStart;

boost::signal<void (std::string /* id */, std::string /* path */)> onFileTransferAccept;

boost::signal<void (std::string /* path */)> onSendFileRequest;

//Whiteboard related

boost::signal<void ()> onWhiteboardSessionAccept;

boost::signal<void ()> onWhiteboardSessionCancel;

boost::signal<void ()> onWhiteboardWindowShow;

// Blocking Command related

boost::signal<void ()> onBlockUserRequest;

boost::signal<void ()> onUnblockUserRequest;

};

}

/*

* Copyright (c) 2010-2014 Kevin Smith

* Licensed under the GNU General Public License v3.

* See Documentation/Licenses/GPLv3.txt for more information.

*/

#pragma once

#include <boost/shared_ptr.hpp>

#include <Swift/Controllers/UIInterfaces/ChatWindow.h>

#include <Swiften/Base/foreach.h>

namespace Swift {

class MockChatWindow : public ChatWindow {

public:

MockChatWindow() : labelsEnabled_(false) {}

virtual ~MockChatWindow();

virtual std::string addMessage(const ChatMessage& message, const std::string& /*senderName*/, bool /*senderIsSelf*/, boost::shared_ptr<SecurityLabel> /*label*/, const std::string& /*avatarPath*/, const boost::posix_time::ptime& /*time*/, const HighlightAction& /*highlight*/) {

lastMessageBody_ = bodyFromMessage(message); return "id";}

virtual std::string addAction(const ChatMessage& /*message*/, const std::string& /*senderName*/, bool /*senderIsSelf*/, boost::shared_ptr<SecurityLabel> /*label*/, const std::string& /*avatarPath*/, const boost::posix_time::ptime& /*time*/, const HighlightAction& /*highlight*/) {return "id";}

virtual void addSystemMessage(const ChatMessage& /*message*/, Direction /*direction*/) {}

virtual void addPresenceMessage(const ChatMessage& /*message*/, Direction /*direction*/) {}

virtual void addErrorMessage(const ChatMessage& /*message*/) {}

virtual void replaceMessage(const ChatMessage& /*message*/, const std::string& /*id*/, const boost::posix_time::ptime& /*time*/, const HighlightAction& /*highlight*/) {}

virtual void replaceWithAction(const ChatMessage& /*message*/, const std::string& /*id*/, const boost::posix_time::ptime& /*time*/, const HighlightAction& /*highlight*/) {}

// File transfer related stuff

virtual std::string addFileTransfer(const std::string& /*senderName*/, bool /*senderIsSelf*/,const std::string& /*filename*/, const boost::uintmax_t /*sizeInBytes*/) { return 0; }

virtual void setFileTransferProgress(std::string /*id*/, const int /*alreadyTransferedBytes*/) { }

virtual void setFileTransferStatus(std::string /*id*/, const FileTransferState /*state*/, const std::string& /*msg*/) { }

virtual void setMessageReceiptState(const std::string &/* id */, ReceiptState /* state */) { }

virtual void setContactChatState(ChatState::ChatStateType /*state*/) {}

virtual void setName(const std::string& name) {name_ = name;}

virtual void show() {}

virtual void activate() {}

virtual void setAvailableSecurityLabels(const std::vector<SecurityLabelsCatalog::Item>& labels) {labels_ = labels;}

virtual void setSecurityLabelsEnabled(bool enabled) {labelsEnabled_ = enabled;}

virtual void setUnreadMessageCount(int /*count*/) {}

virtual void convertToMUC(MUCType /*mucType*/) {}

virtual void setSecurityLabelsError() {}

virtual SecurityLabelsCatalog::Item getSelectedSecurityLabel() {return label_;}

virtual void setInputEnabled(bool /*enabled*/) {}

virtual void setRosterModel(Roster* roster) { roster_ = roster; }

Roster* getRosterModel() { return roster_; }

virtual void setTabComplete(TabComplete*) {}

void setAckState(const std::string& /*id*/, AckState /*state*/) {}

virtual void flash() {}

virtual void setAlert(const std::string& /*alertText*/, const std::string& /*buttonText*/) {}

virtual void cancelAlert() {}

virtual void setCorrectionEnabled(Tristate /*enabled*/) {}

virtual void setFileTransferEnabled(Tristate /*enabled*/) {}

void setAvailableOccupantActions(const std::vector<OccupantAction>&/* actions*/) {}

void setSubject(const std::string& /*subject*/) {}

virtual void showRoomConfigurationForm(Form::ref) {}

virtual void addMUCInvitation(const std::string& /*senderName*/, const JID& /*jid*/, const std::string& /*reason*/, const std::string& /*password*/, bool = true, bool = false, bool = false) {}

virtual std::string addWhiteboardRequest(bool) {return "";}

virtual void setWhiteboardSessionStatus(std::string, const ChatWindow::WhiteboardSessionState){}

virtual void setAffiliations(MUCOccupant::Affiliation, const std::vector<JID>&) {}

virtual void setAvailableRoomActions(const std::vector<RoomAction> &) {}

virtual void setBlockingState(BlockingState) {}

virtual void setCanInitiateImpromptuChats(bool /*supportsImpromptu*/) {}

virtual void showBookmarkWindow(const MUCBookmark& /*bookmark*/) {}

std::string bodyFromMessage(const ChatMessage& message) {

boost::shared_ptr<ChatTextMessagePart> text;

foreach (boost::shared_ptr<ChatMessagePart> part, message.getParts()) {

if ((text = boost::dynamic_pointer_cast<ChatTextMessagePart>(part))) {

return text->text;

}

}

return "";

}

std::string name_;

std::string lastMessageBody_;

std::vector<SecurityLabelsCatalog::Item> labels_;

bool labelsEnabled_;

SecurityLabelsCatalog::Item label_;

Roster* roster_;

};

}

/*

* Copyright (c) 2013 Kevin Smith

* Licensed under the GNU General Public License v3.

* See Documentation/Licenses/GPLv3.txt for more information.

*/

#pragma once

#include <string>

#include <boost/shared_ptr.hpp>

#include <boost/date_time/posix_time/posix_time.hpp>

#include <QWidget>

#include <Swift/Controllers/UIInterfaces/ChatWindow.h>

namespace Swift {

class HighlightAction;

class SecurityLabel;

class QtChatView : public QWidget {

Q_OBJECT

public:

QtChatView(QWidget* parent);

virtual ~QtChatView();

/** Add message to window.

* @return id of added message (for acks).

*/

virtual std::string addMessage(const ChatWindow::ChatMessage& message, const std::string& senderName, bool senderIsSelf, boost::shared_ptr<SecurityLabel> label, const std::string& avatarPath, const boost::posix_time::ptime& time, const HighlightAction& highlight) = 0;

/** Adds action to window.

* @return id of added message (for acks);

*/

virtual std::string addAction(const ChatWindow::ChatMessage& message, const std::string& senderName, bool senderIsSelf, boost::shared_ptr<SecurityLabel> label, const std::string& avatarPath, const boost::posix_time::ptime& time, const HighlightAction& highlight) = 0;

virtual void addSystemMessage(const ChatWindow::ChatMessage& message, ChatWindow::Direction direction) = 0;

virtual void addPresenceMessage(const ChatWindow::ChatMessage& message, ChatWindow::Direction direction) = 0;

virtual void addErrorMessage(const ChatWindow::ChatMessage& message) = 0;

virtual void replaceMessage(const ChatWindow::ChatMessage& message, const std::string& id, const boost::posix_time::ptime& time, const HighlightAction& highlight) = 0;

virtual void replaceWithAction(const ChatWindow::ChatMessage& message, const std::string& id, const boost::posix_time::ptime& time, const HighlightAction& highlight) = 0;

virtual void setAckState(const std::string& id, ChatWindow::AckState state) = 0;

virtual std::string addFileTransfer(const std::string& senderName, bool senderIsSelf, const std::string& filename, const boost::uintmax_t sizeInBytes) = 0;

virtual void setFileTransferProgress(std::string, const int percentageDone) = 0;

virtual void setFileTransferStatus(std::string, const ChatWindow::FileTransferState state, const std::string& msg = "") = 0;

virtual void addMUCInvitation(const std::string& senderName, const JID& jid, const std::string& reason, const std::string& password, bool direct, bool isImpromptu, bool isContinuation) = 0;

virtual std::string addWhiteboardRequest(const QString& contact, bool senderIsSelf) = 0;

virtual void setWhiteboardSessionStatus(const std::string& id, const ChatWindow::WhiteboardSessionState state) = 0;

virtual void setMessageReceiptState(const std::string& id, ChatWindow::ReceiptState state) = 0;

virtual void showEmoticons(bool show) = 0;

virtual void addLastSeenLine() = 0;

public slots:

virtual void resizeFont(int fontSizeSteps) = 0;

virtual void scrollToBottom() = 0;

virtual void handleKeyPressEvent(QKeyEvent* event) = 0;

};

}

/* Drop */

} else {

messageLog_->handleKeyPressEvent(event);

}

}

void QtChatWindow::beginCorrection() {

if (correctionEnabled_ == ChatWindow::Maybe) {

setAlert(Q2PSTRING(tr("This chat may not support message correction. If you send a correction anyway, it may appear as a duplicate message")));

} else if (correctionEnabled_ == ChatWindow::No) {

setAlert(Q2PSTRING(tr("This chat does not support message correction. If you send a correction anyway, it will appear as a duplicate message")));

}

QTextCursor cursor = input_->textCursor();

cursor.select(QTextCursor::Document);

cursor.beginEditBlock();

cursor.insertText(QString(lastSentMessage_));

cursor.endEditBlock();

isCorrection_ = true;

correctingLabel_->show();

input_->setStyleSheet(alertStyleSheet_);

labelsWidget_->setEnabled(false);

}

void QtChatWindow::cancelCorrection() {

cancelAlert();

QTextCursor cursor = input_->textCursor();

cursor.select(QTextCursor::Document);

cursor.removeSelectedText();

isCorrection_ = false;

correctingLabel_->hide();

input_->setStyleSheet(qApp->styleSheet());

labelsWidget_->setEnabled(true);

}

QByteArray QtChatWindow::getSplitterState() {

return logRosterSplitter_->saveState();

}

void QtChatWindow::handleChangeSplitterState(QByteArray state) {

logRosterSplitter_->restoreState(state);

}

void QtChatWindow::handleSplitterMoved(int, int) {

emit splitterMoved();

}

void QtChatWindow::tabComplete() {

if (!completer_) {

return;

}

QTextCursor cursor;

if (tabCompleteCursor_.hasSelection()) {

cursor = tabCompleteCursor_;

} else {

cursor = input_->textCursor();

while(cursor.movePosition(QTextCursor::Left, QTextCursor::KeepAnchor) && cursor.document()->characterAt(cursor.position() - 1) != ' ') { }

}

QString root = cursor.selectedText();

if (root.isEmpty()) {

return;

}

QString suggestion = P2QSTRING(completer_->completeWord(Q2PSTRING(root)));

if (root == suggestion) {

return;

}

tabCompletion_ = true;

cursor.beginEditBlock();

cursor.removeSelectedText();

int oldPosition = cursor.position();

cursor.insertText(suggestion);

tabCompleteCursor_ = cursor;

tabCompleteCursor_.setPosition(oldPosition, QTextCursor::KeepAnchor);

cursor.endEditBlock();

tabCompletion_ = false;

}

void QtChatWindow::setRosterModel(Roster* roster) {

treeWidget_->setRosterModel(roster);

}

void QtChatWindow::setAvailableSecurityLabels(const std::vector<SecurityLabelsCatalog::Item>& labels) {

delete labelModel_;

labelModel_ = new LabelModel();

labelModel_->availableLabels_ = labels;

int i = 0;

int defaultIndex = 0;

labelsWidget_->setModel(labelModel_);

foreach (SecurityLabelsCatalog::Item label, labels) {

if (label.getIsDefault()) {

defaultIndex = i;

break;

}

i++;

}

labelsWidget_->setCurrentIndex(defaultIndex);

}

void QtChatWindow::handleCurrentLabelChanged(int index) {

if (static_cast<size_t>(index) >= labelModel_->availableLabels_.size()) {

qDebug() << "User selected a label that doesn't exist";

return;

}

const SecurityLabelsCatalog::Item& label = labelModel_->availableLabels_[index];

if (label.getLabel()) {

QPalette palette = labelsWidget_->palette();

//palette.setColor(QPalette::Base, P2QSTRING(label.getLabel()->getBackgroundColor()));

palette.setColor(labelsWidget_->backgroundRole(), P2QSTRING(label.getLabel()->getBackgroundColor()));

palette.setColor(labelsWidget_->foregroundRole(), P2QSTRING(label.getLabel()->getForegroundColor()));

labelsWidget_->setPalette(palette);

midBar_->setPalette(palette);

labelsWidget_->setAutoFillBackground(true);

}

else {

labelsWidget_->setAutoFillBackground(false);

labelsWidget_->setPalette(defaultLabelsPalette_);

midBar_->setPalette(defaultLabelsPalette_);

}

}

void QtChatWindow::setSecurityLabelsError() {

labelsWidget_->setEnabled(false);

}

void QtChatWindow::setSecurityLabelsEnabled(bool enabled) {

if (enabled) {

labelsWidget_->setEnabled(true);

labelsWidget_->show();

} else {

labelsWidget_->hide();

}

}

void QtChatWindow::setCorrectionEnabled(Tristate enabled) {

correctionEnabled_ = enabled;

}

void QtChatWindow::setFileTransferEnabled(Tristate enabled) {

fileTransferEnabled_ = enabled;

}

SecurityLabelsCatalog::Item QtChatWindow::getSelectedSecurityLabel() {

assert(labelsWidget_->isEnabled());

assert(labelsWidget_->currentIndex() >= 0 && static_cast<size_t>(labelsWidget_->currentIndex()) < labelModel_->availableLabels_.size());

return labelModel_->availableLabels_[labelsWidget_->currentIndex()];

}

void QtChatWindow::closeEvent(QCloseEvent* event) {

event->accept();

emit windowClosing();

onClosed();

}

void QtChatWindow::convertToMUC(MUCType mucType) {

impromptu_ = (mucType == ImpromptuMUC);

treeWidget_->setMessageTarget(impromptu_ ? QtTreeWidget::MessageDisplayJID : QtTreeWidget::MessageDefaultJID);

isMUC_ = true;

treeWidget_->show();

subject_->setVisible(!impromptu_);

}

void QtChatWindow::qAppFocusChanged(QWidget* /*old*/, QWidget* /*now*/) {

if (isWidgetSelected()) {

lastLineTracker_.setHasFocus(true);

input_->setFocus();

onAllMessagesRead();

}

else {

lastLineTracker_.setHasFocus(false);

}

}

void QtChatWindow::setInputEnabled(bool enabled) {

inputEnabled_ = enabled;

if (!enabled) {

if (mucConfigurationWindow_) {

delete mucConfigurationWindow_.data();

}

if (affiliationEditor_) {

delete affiliationEditor_.data();

}

}

}

void QtChatWindow::showEvent(QShowEvent* event) {

emit windowOpening();

QWidget::showEvent(event);

}

void QtChatWindow::setUnreadMessageCount(int count) {

if (unreadCount_ != count) {

unreadCount_ = count;

updateTitleWithUnreadCount();

emit countUpdated();

}

}

void QtChatWindow::setContactChatState(ChatState::ChatStateType state) {

contactIsTyping_ = (state == ChatState::Composing);

emit titleUpdated();

}

QtTabbable::AlertType QtChatWindow::getWidgetAlertState() {

if (contactIsTyping_) {

return ImpendingActivity;

}

if (unreadCount_ > 0) {

return WaitingActivity;

}

return NoActivity;

}

void QtChatWindow::setName(const std::string& name) {

contact_ = P2QSTRING(name);

updateTitleWithUnreadCount();

}

void QtChatWindow::updateTitleWithUnreadCount() {

if (isWindow()) {

setWindowTitle(unreadCount_ > 0 ? QString("(%1) %2").arg(unreadCount_).arg(contact_) : contact_);

} else {

setWindowTitle(contact_);

}

emit titleUpdated();

}

void QtChatWindow::flash() {

emit requestFlash();

}

int QtChatWindow::getCount() {

return unreadCount_;

}

void QtChatWindow::returnPressed() {

if (!inputEnabled_) {

return;

}

messageLog_->scrollToBottom();

lastSentMessage_ = QString(input_->toPlainText());

onSendMessageRequest(Q2PSTRING(input_->toPlainText()), isCorrection_);

inputClearing_ = true;

input_->clear();

cancelCorrection();

inputClearing_ = false;

}

void QtChatWindow::handleInputChanged() {

if (inputClearing_) {

return;

}

if (input_->toPlainText().isEmpty()) {

onUserCancelsTyping();

} else {

onUserTyping();

}

}

void QtChatWindow::handleCursorPositionChanged() {

if (tabCompletion_) {

return;

}

tabCompleteCursor_.clearSelection();

}

void QtChatWindow::show() {

if (parentWidget() == NULL) {

QWidget::show();

}

emit windowOpening();

}

void QtChatWindow::activate() {

if (isWindow()) {

QWidget::show();

}

emit wantsToActivate();

input_->setFocus();

}

void QtChatWindow::resizeEvent(QResizeEvent*) {

emit geometryChanged();

}

void QtChatWindow::moveEvent(QMoveEvent*) {

emit geometryChanged();

}

void QtChatWindow::dragEnterEvent(QDragEnterEvent *event) {

if (event->mimeData()->hasUrls() && event->mimeData()->urls().size() == 1) {

// TODO: check whether contact actually supports file transfer

if (!isMUC_) {

event->acceptProposedAction();

}

} else if (event->mimeData()->hasFormat("application/vnd.swift.contact-jid-list")) {

if (isMUC_ || supportsImpromptuChat_) {

event->acceptProposedAction();

}

}

}

void QtChatWindow::dropEvent(QDropEvent *event) {

if (fileTransferEnabled_ == ChatWindow::Yes && event->mimeData()->hasUrls()) {

if (event->mimeData()->urls().size() == 1) {

onSendFileRequest(Q2PSTRING(event->mimeData()->urls().at(0).toLocalFile()));

} else {

std::string messageText(Q2PSTRING(tr("Sending of multiple files at once isn't supported at this time.")));

ChatMessage message;

message.append(boost::make_shared<ChatTextMessagePart>(messageText));

addSystemMessage(message, DefaultDirection);

}

} else if (event->mimeData()->hasFormat("application/vnd.swift.contact-jid-list")) {

QByteArray dataBytes = event->mimeData()->data("application/vnd.swift.contact-jid-list");

QDataStream dataStream(&dataBytes, QIODevice::ReadOnly);

std::vector<JID> invites;

while (!dataStream.atEnd()) {

QString jidString;

dataStream >> jidString;

invites.push_back(Q2PSTRING(jidString));

}

onInviteToChat(invites);

}

}

void QtChatWindow::setAvailableOccupantActions(const std::vector<OccupantAction>& actions) {

treeWidget_->setAvailableOccupantActions(actions);

}

void QtChatWindow::setSubject(const std::string& subject) {

//subject_->setVisible(!subject.empty());

subject_->setText(P2QSTRING(subject));

subject_->setToolTip(P2QSTRING(subject));

subject_->setCursorPosition(0);

}

void QtChatWindow::handleActionButtonClicked() {

QMenu contextMenu;

QAction* changeSubject = NULL;

QAction* configure = NULL;

QAction* affiliations = NULL;

QAction* destroy = NULL;

QAction* invite = NULL;

QAction* block = NULL;

QAction* unblock = NULL;

if (availableRoomActions_.empty()) {

if (blockingState_ == IsBlocked) {

unblock = contextMenu.addAction(tr("Unblock"));

} else if (blockingState_ == IsUnblocked) {

block = contextMenu.addAction(tr("Block"));

}

if (supportsImpromptuChat_) {

invite = contextMenu.addAction(tr("Invite person to this chat…"));

}

} else {

foreach(ChatWindow::RoomAction availableAction, availableRoomActions_)

{

if (impromptu_) {

// hide options we don't need in impromptu chats

if (availableAction == ChatWindow::ChangeSubject ||

availableAction == ChatWindow::Configure ||

availableAction == ChatWindow::Affiliations ||

availableAction == ChatWindow::Destroy) {

continue;

}

}

switch(availableAction)

{

case ChatWindow::ChangeSubject: changeSubject = contextMenu.addAction(tr("Change subject…")); break;

case ChatWindow::Configure: configure = contextMenu.addAction(tr("Configure room…")); break;

case ChatWindow::Affiliations: affiliations = contextMenu.addAction(tr("Edit affiliations…")); break;

case ChatWindow::Destroy: destroy = contextMenu.addAction(tr("Destroy room")); break;

case ChatWindow::Invite: invite = contextMenu.addAction(tr("Invite person to this room…")); break;

}

}

}

QAction* bookmark = contextMenu.addAction(tr("Add boomark..."));

QAction* result = contextMenu.exec(QCursor::pos());

if (result == NULL) {

/* Skip processing. Note that otherwise, because the actions could be null they could match */

}

else if (result == changeSubject) {

bool ok;

QString subject = QInputDialog::getText(this, tr("Change room subject"), tr("New subject:"), QLineEdit::Normal, subject_->text(), &ok);

if (ok) {

onChangeSubjectRequest(Q2PSTRING(subject));

}

}

else if (result == configure) {

onConfigureRequest(Form::ref());

}

else if (result == affiliations) {

if (!affiliationEditor_) {

onGetAffiliationsRequest();

affiliationEditor_ = new QtAffiliationEditor(this);

connect(affiliationEditor_, SIGNAL(accepted()), this, SLOT(handleAffiliationEditorAccepted()));

}

affiliationEditor_->show();

}

else if (result == destroy) {

QMessageBox msgBox;

msgBox.setWindowTitle(tr("Confirm room destruction"));

msgBox.setText(tr("Are you sure you want to destroy the room?"));

msgBox.setInformativeText(tr("This will destroy the room."));

msgBox.setStandardButtons(QMessageBox::Yes | QMessageBox::No);

msgBox.setDefaultButton(QMessageBox::No);

if (msgBox.exec() == QMessageBox::Yes) {

onDestroyRequest();

}

}

else if (result == invite) {

onInviteToChat(std::vector<JID>());

}

else if (result == block) {

onBlockUserRequest();

}

else if (result == unblock) {

onUnblockUserRequest();

} else if (result == bookmark) {

onBookmarkRequest();

}

}

void QtChatWindow::handleAffiliationEditorAccepted() {

onChangeAffiliationsRequest(affiliationEditor_->getChanges());

}

void QtChatWindow::setAffiliations(MUCOccupant::Affiliation affiliation, const std::vector<JID>& jids) {

if (!affiliationEditor_) return;

affiliationEditor_->setAffiliations(affiliation, jids);

}

void QtChatWindow::setAvailableRoomActions(const std::vector<RoomAction>& actions) {

availableRoomActions_ = actions;

}

void QtChatWindow::setBlockingState(BlockingState state) {

blockingState_ = state;

}

void QtChatWindow::setCanInitiateImpromptuChats(bool supportsImpromptu) {

supportsImpromptuChat_ = supportsImpromptu;

}

void QtChatWindow::showBookmarkWindow(const MUCBookmark& bookmark) {

QtAddBookmarkWindow* window = new QtAddBookmarkWindow(eventStream_, bookmark);

window->show();

}

void QtChatWindow::showRoomConfigurationForm(Form::ref form) {

if (mucConfigurationWindow_) {

delete mucConfigurationWindow_.data();

}

mucConfigurationWindow_ = new QtMUCConfigurationWindow(form);

mucConfigurationWindow_->onFormComplete.connect(boost::bind(boost::ref(onConfigureRequest), _1));

mucConfigurationWindow_->onFormCancelled.connect(boost::bind(boost::ref(onConfigurationFormCancelled)));

}

void QtChatWindow::handleAppendedToLog() {

if (lastLineTracker_.getShouldMoveLastLine()) {

/* should this be queued? */

messageLog_->addLastSeenLine();

}

if (isWidgetSelected()) {

onAllMessagesRead();

}

}

void QtChatWindow::addMUCInvitation(const std::string& senderName, const JID& jid, const std::string& reason, const std::string& password, bool direct, bool isImpromptu, bool isContinuation) {

handleAppendedToLog();

messageLog_->addMUCInvitation(senderName, jid, reason, password, direct, isImpromptu, isContinuation);

}

std::string QtChatWindow::addMessage(const ChatMessage& message, const std::string& senderName, bool senderIsSelf, boost::shared_ptr<SecurityLabel> label, const std::string& avatarPath, const boost::posix_time::ptime& time, const HighlightAction& highlight) {

handleAppendedToLog();

return messageLog_->addMessage(message, senderName, senderIsSelf, label, avatarPath, time, highlight);

}

std::string QtChatWindow::addAction(const ChatMessage& message, const std::string& senderName, bool senderIsSelf, boost::shared_ptr<SecurityLabel> label, const std::string& avatarPath, const boost::posix_time::ptime& time, const HighlightAction& highlight) {

handleAppendedToLog();

return messageLog_->addAction(message, senderName, senderIsSelf, label, avatarPath, time, highlight);

}

void QtChatWindow::addSystemMessage(const ChatMessage& message, Direction direction) {

handleAppendedToLog();

messageLog_->addSystemMessage(message, direction);

}

void QtChatWindow::addPresenceMessage(const ChatMessage& message, Direction direction) {

handleAppendedToLog();

messageLog_->addPresenceMessage(message, direction);

}

void QtChatWindow::addErrorMessage(const ChatMessage& message) {

handleAppendedToLog();

messageLog_->addErrorMessage(message);

}

void QtChatWindow::replaceMessage(const ChatMessage& message, const std::string& id, const boost::posix_time::ptime& time, const HighlightAction& highlight) {

handleAppendedToLog();

messageLog_->replaceMessage(message, id, time, highlight);

}

void QtChatWindow::replaceWithAction(const ChatMessage& message, const std::string& id, const boost::posix_time::ptime& time, const HighlightAction& highlight) {

handleAppendedToLog();

messageLog_->replaceWithAction(message, id, time, highlight);

}

std::string QtChatWindow::addFileTransfer(const std::string& senderName, bool senderIsSelf, const std::string& filename, const boost::uintmax_t sizeInBytes) {

handleAppendedToLog();

return messageLog_->addFileTransfer(senderName, senderIsSelf, filename, sizeInBytes);

}

void QtChatWindow::setFileTransferProgress(std::string id, const int percentageDone) {

messageLog_->setFileTransferProgress(id, percentageDone);

}

void QtChatWindow::setFileTransferStatus(std::string id, const FileTransferState state, const std::string& msg) {

messageLog_->setFileTransferStatus(id, state, msg);

}

std::string QtChatWindow::addWhiteboardRequest(bool senderIsSelf) {

handleAppendedToLog();

return messageLog_->addWhiteboardRequest(contact_, senderIsSelf);

}

void QtChatWindow::setWhiteboardSessionStatus(std::string id, const ChatWindow::WhiteboardSessionState state) {

messageLog_->setWhiteboardSessionStatus(id, state);

}

}

void QtChatWindow::setAckState(const std::string& id, AckState state) {

messageLog_->setAckState(id, state);

}

void QtChatWindow::setMessageReceiptState(const std::string& id, ChatWindow::ReceiptState state) {

messageLog_->setMessageReceiptState(id, state);

}

}

/*

* Copyright (c) 2010-2014 Kevin Smith

* Licensed under the GNU General Public License v3.

* See Documentation/Licenses/GPLv3.txt for more information.

*/

#pragma once

#include <map>

#include <QPointer>

#include <QTextCursor>

#include <QMap>

#include <SwifTools/LastLineTracker.h>

#include <Swift/Controllers/UIInterfaces/ChatWindow.h>

#include <Swift/QtUI/ChatSnippet.h>

#include <Swift/QtUI/QtAffiliationEditor.h>

#include <Swift/QtUI/QtMUCConfigurationWindow.h>

#include <Swift/QtUI/QtSwiftUtil.h>

#include <Swift/QtUI/QtTabbable.h>

class QTextEdit;

class QLineEdit;

class QComboBox;

class QLabel;

class QSplitter;

class QPushButton;

namespace Swift {

class QtChatView;

class QtOccupantListWidget;

class QtChatTheme;

class TreeWidget;

class QtTextEdit;

class UIEventStream;

class QtChatWindowJSBridge;

class SettingsProvider;

class LabelModel : public QAbstractListModel {

Q_OBJECT

public:

LabelModel(QObject* parent = NULL) : QAbstractListModel(parent) {}

virtual int rowCount(const QModelIndex& /*index*/) const {

return static_cast<int>(availableLabels_.size());

}

virtual QVariant data(const QModelIndex& index, int role) const {

if (!index.isValid()) {

return QVariant();

}

boost::shared_ptr<SecurityLabel> label = availableLabels_[index.row()].getLabel();

if (label && role == Qt::TextColorRole) {

return P2QSTRING(label->getForegroundColor());

}

if (label && role == Qt::TextColorRole) {

return P2QSTRING(label->getBackgroundColor());

}

if (role == Qt::DisplayRole) {

std::string selector = availableLabels_[index.row()].getSelector();

std::string displayMarking = label ? label->getDisplayMarking() : "";

QString labelName = selector.empty() ? displayMarking.c_str() : selector.c_str();

return labelName;

}

return QVariant();

}

std::vector<SecurityLabelsCatalog::Item> availableLabels_;

};

class QtChatWindow : public QtTabbable, public ChatWindow {

Q_OBJECT

public:

QtChatWindow(const QString &contact, QtChatTheme* theme, UIEventStream* eventStream, SettingsProvider* settings);

~QtChatWindow();

std::string addMessage(const ChatMessage& message, const std::string &senderName, bool senderIsSelf, boost::shared_ptr<SecurityLabel> label, const std::string& avatarPath, const boost::posix_time::ptime& time, const HighlightAction& highlight);

std::string addAction(const ChatMessage& message, const std::string &senderName, bool senderIsSelf, boost::shared_ptr<SecurityLabel> label, const std::string& avatarPath, const boost::posix_time::ptime& time, const HighlightAction& highlight);

void addSystemMessage(const ChatMessage& message, Direction direction);

void addPresenceMessage(const ChatMessage& message, Direction direction);

void addErrorMessage(const ChatMessage& message);

void replaceMessage(const ChatMessage& message, const std::string& id, const boost::posix_time::ptime& time, const HighlightAction& highlight);

void replaceWithAction(const ChatMessage& message, const std::string& id, const boost::posix_time::ptime& time, const HighlightAction& highlight);

// File transfer related stuff

std::string addFileTransfer(const std::string& senderName, bool senderIsSelf, const std::string& filename, const boost::uintmax_t sizeInBytes);

void setFileTransferProgress(std::string id, const int percentageDone);

void setFileTransferStatus(std::string id, const FileTransferState state, const std::string& msg);

std::string addWhiteboardRequest(bool senderIsSelf);

void setWhiteboardSessionStatus(std::string id, const ChatWindow::WhiteboardSessionState state);

void show();

void activate();

void setUnreadMessageCount(int count);

void convertToMUC(MUCType mucType);

// TreeWidget *getTreeWidget();

void setAvailableSecurityLabels(const std::vector<SecurityLabelsCatalog::Item>& labels);

void setSecurityLabelsEnabled(bool enabled);

void setSecurityLabelsError();

SecurityLabelsCatalog::Item getSelectedSecurityLabel();

void setName(const std::string& name);

void setInputEnabled(bool enabled);

QtTabbable::AlertType getWidgetAlertState();

void setContactChatState(ChatState::ChatStateType state);

void setRosterModel(Roster* roster);

void setTabComplete(TabComplete* completer);

int getCount();

void setAckState(const std::string& id, AckState state);

// message receipts

void setMessageReceiptState(const std::string& id, ChatWindow::ReceiptState state);

void flash();

QByteArray getSplitterState();

virtual void setAvailableOccupantActions(const std::vector<OccupantAction>& actions);

void setSubject(const std::string& subject);

void showRoomConfigurationForm(Form::ref);

void addMUCInvitation(const std::string& senderName, const JID& jid, const std::string& reason, const std::string& password, bool direct = true, bool isImpromptu = false, bool isContinuation = false);

void setAffiliations(MUCOccupant::Affiliation, const std::vector<JID>&);

void setAvailableRoomActions(const std::vector<RoomAction>& actions);

void setBlockingState(BlockingState state);

virtual void setCanInitiateImpromptuChats(bool supportsImpromptu);

virtual void showBookmarkWindow(const MUCBookmark& bookmark);

public slots:

void handleChangeSplitterState(QByteArray state);

void handleFontResized(int fontSizeSteps);

void setAlert(const std::string& alertText, const std::string& buttonText = "");

void cancelAlert();

void setCorrectionEnabled(Tristate enabled);

void setFileTransferEnabled(Tristate enabled);

signals:

void geometryChanged();

void splitterMoved();

void fontResized(int);

protected slots:

void qAppFocusChanged(QWidget* old, QWidget* now);

void closeEvent(QCloseEvent* event);

void resizeEvent(QResizeEvent* event);

void moveEvent(QMoveEvent* event);

void dragEnterEvent(QDragEnterEvent *event);

void dropEvent(QDropEvent *event);

protected:

void showEvent(QShowEvent* event);

private slots:

void handleLogCleared();

void returnPressed();

void handleInputChanged();

void handleCursorPositionChanged();

void handleKeyPressEvent(QKeyEvent* event);

void handleSplitterMoved(int pos, int index);

void handleAlertButtonClicked();

void handleActionButtonClicked();

void handleAffiliationEditorAccepted();

void handleCurrentLabelChanged(int);

private:

void updateTitleWithUnreadCount();

void tabComplete();

void beginCorrection();

void cancelCorrection();

void handleSettingChanged(const std::string& setting);

void handleOccupantSelectionChanged(RosterItem* item);

void handleAppendedToLog();

int unreadCount_;

bool contactIsTyping_;

LastLineTracker lastLineTracker_;

QString contact_;

QString lastSentMessage_;

QTextCursor tabCompleteCursor_;

QtChatView* messageLog_;

QtChatTheme* theme_;

QtTextEdit* input_;

QWidget* midBar_;

QBoxLayout* subjectLayout_;

QComboBox* labelsWidget_;

QtOccupantListWidget* treeWidget_;

QLabel* correctingLabel_;

QLabel* alertLabel_;

QWidget* alertWidget_;

QPushButton* alertButton_;

TabComplete* completer_;

QLineEdit* subject_;

bool isCorrection_;

bool inputClearing_;

bool tabCompletion_;

UIEventStream* eventStream_;

bool inputEnabled_;

QSplitter *logRosterSplitter_;

Tristate correctionEnabled_;

Tristate fileTransferEnabled_;

QString alertStyleSheet_;

QPointer<QtMUCConfigurationWindow> mucConfigurationWindow_;

QPointer<QtAffiliationEditor> affiliationEditor_;

SettingsProvider* settings_;

std::vector<ChatWindow::RoomAction> availableRoomActions_;

QPalette defaultLabelsPalette_;

LabelModel* labelModel_;

BlockingState blockingState_;

bool impromptu_;

bool isMUC_;

bool supportsImpromptuChat_;

};

}

/*

* Copyright (c) 2013-2014 Kevin Smith and Remko Tronçon

* Licensed under the GNU General Public License v3.

* See Documentation/Licenses/GPLv3.txt for more information.

*/

#include <Swift/QtUI/QtPlainChatView.h>

#include <QTextEdit>

#include <QScrollBar>

#include <QVBoxLayout>

#include <QPushButton>

#include <QLabel>

#include <QDialog>

#include <QProgressBar>

#include <QFileDialog>

#include <QInputDialog>

#include <QMenu>

#include <Swiften/Base/foreach.h>

#include <Swiften/Base/FileSize.h>

#include <Swift/Controllers/UIEvents/UIEventStream.h>

#include <Swift/Controllers/UIEvents/JoinMUCUIEvent.h>

#include <Swift/QtUI/ChatSnippet.h>

#include <Swift/QtUI/QtSwiftUtil.h>

#include <Swift/QtUI/QtUtilities.h>

namespace Swift {

QtPlainChatView::QtPlainChatView(QtChatWindow *window, UIEventStream* eventStream)

: QtChatView(window), window_(window), eventStream_(eventStream), idGenerator_(0) {

QVBoxLayout* mainLayout = new QVBoxLayout(this);

mainLayout->setSpacing(0);

mainLayout->setContentsMargins(0,0,0,0);

log_ = new LogTextEdit(this);

log_->setReadOnly(true);

log_->setAccessibleName(tr("Chat Messages"));

mainLayout->addWidget(log_);

}

QtPlainChatView::~QtPlainChatView() {

}

QString chatMessageToString(const ChatWindow::ChatMessage& message) {

QString result;

foreach (boost::shared_ptr<ChatWindow::ChatMessagePart> part, message.getParts()) {

boost::shared_ptr<ChatWindow::ChatTextMessagePart> textPart;

boost::shared_ptr<ChatWindow::ChatURIMessagePart> uriPart;

boost::shared_ptr<ChatWindow::ChatEmoticonMessagePart> emoticonPart;

boost::shared_ptr<ChatWindow::ChatHighlightingMessagePart> highlightPart;

if ((textPart = boost::dynamic_pointer_cast<ChatWindow::ChatTextMessagePart>(part))) {

QString text = QtUtilities::htmlEscape(P2QSTRING(textPart->text));

text.replace("\n","<br/>");

result += text;

continue;

}

if ((uriPart = boost::dynamic_pointer_cast<ChatWindow::ChatURIMessagePart>(part))) {

QString uri = QtUtilities::htmlEscape(P2QSTRING(uriPart->target));

result += "<a href='" + uri + "' >" + uri + "</a>";

continue;

}

if ((emoticonPart = boost::dynamic_pointer_cast<ChatWindow::ChatEmoticonMessagePart>(part))) {

result += P2QSTRING(emoticonPart->alternativeText);

continue;

}

if ((highlightPart = boost::dynamic_pointer_cast<ChatWindow::ChatHighlightingMessagePart>(part))) {

//FIXME: Maybe do something here. Anything, really.

continue;

}

}

return result;

}

std::string QtPlainChatView::addMessage(const ChatWindow::ChatMessage& message, const std::string& senderName, bool senderIsSelf, boost::shared_ptr<SecurityLabel> label, const std::string& /*avatarPath*/, const boost::posix_time::ptime& time, const HighlightAction& /*highlight*/) {

QString text = "<p>";

if (label) {

text += P2QSTRING(label->getLabel()) + "<br/>";

}

QString name = senderIsSelf ? "you" : P2QSTRING(senderName);

text += QString(tr("At %1 %2 said:")).arg(ChatSnippet::timeToEscapedString(B2QDATE(time))).arg(name) + "<br/>";

text += chatMessageToString(message);

text += "</p>";

log_->append(text);

const std::string idx = senderIsSelf ? "" : senderName;

lastMessageLabel_[idx] = label;

return idx;

}

std::string QtPlainChatView::addAction(const ChatWindow::ChatMessage& message, const std::string& senderName, bool senderIsSelf, boost::shared_ptr<SecurityLabel> label, const std::string& /*avatarPath*/, const boost::posix_time::ptime& time, const HighlightAction& /*highlight*/) {

QString text = "<p>";

if (label) {

text += P2QSTRING(label->getLabel()) + "<br/>";

}

QString name = senderIsSelf ? "you" : P2QSTRING(senderName);

text += QString(tr("At %1 <i>%2 ")).arg(ChatSnippet::timeToEscapedString(B2QDATE(time))).arg(name);

text += chatMessageToString(message);

text += "</i></p>";

log_->append(text);

const std::string idx = senderIsSelf ? "" : senderName;

lastMessageLabel_[idx] = label;

return idx;

}

void QtPlainChatView::addSystemMessage(const ChatWindow::ChatMessage& message, ChatWindow::Direction /*direction*/)

{

QString text = "<p><i>";

text += chatMessageToString(message);

text += "</i></p>";

log_->append(text);

}

void QtPlainChatView::addPresenceMessage(const ChatWindow::ChatMessage& message, ChatWindow::Direction /*direction*/)

{

QString text = "<p><i>";

text += chatMessageToString(message);

text += "</i></p>";

log_->append(text);

}

void QtPlainChatView::addErrorMessage(const ChatWindow::ChatMessage& message)

{

QString text = "<p><i>";

text += chatMessageToString(message);

text += "</i></p>";

log_->append(text);

}

void QtPlainChatView::replaceMessage(const ChatWindow::ChatMessage& message, const std::string& id, const boost::posix_time::ptime& time, const HighlightAction& /*highlight*/)

{

QString text = "<p>";

if (lastMessageLabel_[id]) {

text += P2QSTRING(lastMessageLabel_[id]->getLabel()) + "<br/>";

}

QString name = id.empty() ? "you" : P2QSTRING(id);

text += QString(tr("At %1 %2 corrected the last message to:")).arg(ChatSnippet::timeToEscapedString(B2QDATE(time))).arg(name) + "<br/>";

text += chatMessageToString(message);

text += "</p>";

log_->append(text);

}

void QtPlainChatView::replaceWithAction(const ChatWindow::ChatMessage& message, const std::string& id, const boost::posix_time::ptime& time, const HighlightAction& /*highlight*/)

{

QString text = "<p>";

if (lastMessageLabel_[id]) {

text += P2QSTRING(lastMessageLabel_[id]->getLabel()) + "<br/>";

}

QString name = id.empty() ? "you" : P2QSTRING(id);

text += QString(tr("At %1 %2 corrected the last action to: <i>")).arg(ChatSnippet::timeToEscapedString(B2QDATE(time))).arg(name);

text += chatMessageToString(message);

text += "</i></p>";

log_->append(text);

}

{

QString text = "<p>The last message was corrected to:<br/>";

text += chatMessageToString(message);

text += "</p>";

log_->append(text);

}

void QtPlainChatView::setAckState(const std::string& /*id*/, ChatWindow::AckState state)

{

if (state == ChatWindow::Failed) {

addSystemMessage(ChatWindow::ChatMessage("Message delivery failed due to disconnection from server."), ChatWindow::DefaultDirection);

}

}

std::string QtPlainChatView::addFileTransfer(const std::string& senderName, bool senderIsSelf, const std::string& filename, const boost::uintmax_t sizeInBytes)

{

const std::string ftId = "ft" + boost::lexical_cast<std::string>(idGenerator_++);

const std::string sizeString = formatSize(sizeInBytes);

FileTransfer* transfer;

if (senderIsSelf) {

QString description = QInputDialog::getText(this, tr("File transfer description"),

tr("Description:"), QLineEdit::Normal, "");

/* NOTE: it is not possible to abort if description is not provided, since we must always return a valid transfer id */

const std::string message = std::string() + "Confirm file transfer: <i>" + filename + " (" + sizeString + " bytes)</i>";

transfer = new FileTransfer(this, senderIsSelf, ftId, filename, ChatWindow::WaitingForAccept, Q2PSTRING(description), message, true);

addSystemMessage(ChatWindow::ChatMessage("Preparing to start file transfer..."), ChatWindow::DefaultDirection);

} else { /* incoming transfer */

const std::string message = std::string() + "Incoming file transfer: <i>" + filename + " (" + sizeString + " bytes)</i>";

transfer = new FileTransfer(this, senderIsSelf, ftId, filename, ChatWindow::WaitingForAccept, "", message, true);

addSystemMessage("Incoming file transfer from " + senderName + "...", ChatWindow::DefaultDirection);

}

fileTransfers_[ftId] = transfer;

layout()->addWidget(transfer->dialog_);

return ftId;

}

void QtPlainChatView::setFileTransferProgress(std::string id, const int percentageDone)

{

FileTransferMap::iterator transfer = fileTransfers_.find(id);

if (transfer != fileTransfers_.end()) {

transfer->second->bar_->setValue(percentageDone);

}

}

void QtPlainChatView::setFileTransferStatus(std::string id, const ChatWindow::FileTransferState state, const std::string& msg)

{

FileTransferMap::iterator transferIter = fileTransfers_.find(id);

if (transferIter == fileTransfers_.end()) {

return;

}

/* store the layout index so we can restore it to the same location */

FileTransfer* oldTransfer = transferIter->second;

const int layoutIndex = layout()->indexOf(oldTransfer->dialog_);

layout()->removeWidget(oldTransfer->dialog_);

const std::string &label = (!msg.empty() ? msg : oldTransfer->message_);

FileTransfer* transfer = new FileTransfer(this, oldTransfer->senderIsSelf_, oldTransfer->ftId_, oldTransfer->filename_, state, oldTransfer->description_, label, false);

fileTransfers_[oldTransfer->ftId_] = transfer; /* replace the transfer object for this file id */

delete oldTransfer;

/* insert the new dialog at the old position in the layout list */

QBoxLayout* parentLayout = dynamic_cast<QBoxLayout*>(layout());

assert(parentLayout);

parentLayout->insertWidget(layoutIndex, transfer->dialog_);

/* log the transfer end result as a system message */

if (state == ChatWindow::Finished) {

addSystemMessage(ChatWindow::ChatMessage("The file transfer completed successfully."), ChatWindow::DefaultDirection);

} else if (state == ChatWindow::Canceled) {

addSystemMessage(ChatWindow::ChatMessage("The file transfer was canceled."), ChatWindow::DefaultDirection);

} else if (state == ChatWindow::FTFailed) {

addSystemMessage(ChatWindow::ChatMessage("The file transfer failed."), ChatWindow::DefaultDirection);

}

}

void QtPlainChatView::addMUCInvitation(const std::string& senderName, const JID& jid, const std::string& /*reason*/, const std::string& password, bool /*direct*/, bool isImpromptu, bool isContinuation)

{

PopupDialog *invite = new PopupDialog(this);

QLabel* statusLabel = new QLabel;

std::string msg = senderName + " has invited you to join " + jid.toString() + "...";

statusLabel->setText(P2QSTRING(msg));

invite->layout_->addWidget(statusLabel);

invite->layout_->addWidget(new QLabel); /* padding */

AcceptMUCInviteAction* accept = new AcceptMUCInviteAction(invite, "Accept", jid, senderName, password, isImpromptu, isContinuation);

connect(accept, SIGNAL(clicked()), SLOT(acceptMUCInvite()));

invite->layout_->addWidget(accept);

AcceptMUCInviteAction* reject = new AcceptMUCInviteAction(invite, "Reject", jid, senderName, password, isImpromptu, isContinuation);

connect(reject, SIGNAL(clicked()), SLOT(rejectMUCInvite()));

invite->layout_->addWidget(reject);

statusLabel->setText(P2QSTRING(msg));

layout()->addWidget(invite->dialog_);

addSystemMessage(ChatWindow::ChatMessage(msg), ChatWindow::DefaultDirection);

}

void QtPlainChatView::scrollToBottom()

{

log_->ensureCursorVisible();

log_->verticalScrollBar()->setValue(log_->verticalScrollBar()->maximum());

}

void QtPlainChatView::fileTransferAccept()

{

FileTransfer::Action* action = dynamic_cast<FileTransfer::Action*>(sender());

if (!action) {

return;

}

FileTransferMap::iterator transferIter = fileTransfers_.find(action->id_);

if (transferIter == fileTransfers_.end()) {

return;

}

FileTransfer* transfer = transferIter->second;

if (transfer->senderIsSelf_) { /* if we are the sender, kick of the transfer */

window_->onFileTransferStart(transfer->ftId_, transfer->description_);

} else { /* ask the user where to save the file first */

QString path = QFileDialog::getSaveFileName(this, tr("Save File"), P2QSTRING(transfer->filename_));

if (path.isEmpty()) {

fileTransferReject(); /* because the user did not select a desintation path */

return;

}

window_->onFileTransferAccept(transfer->ftId_, Q2PSTRING(path));

}

setFileTransferStatus(transfer->ftId_, ChatWindow::Negotiating, transfer->message_);

}

void QtPlainChatView::fileTransferReject()

{

FileTransfer::Action* action = dynamic_cast<FileTransfer::Action*>(sender());

if (action) {

window_->onFileTransferCancel(action->id_);

fileTransferFinish();

}

}

void QtPlainChatView::fileTransferFinish()

{

FileTransfer::Action* action = dynamic_cast<FileTransfer::Action*>(sender());

if (action) {

FileTransferMap::iterator transferIter = fileTransfers_.find(action->id_);

if (transferIter != fileTransfers_.end()) {

delete transferIter->second; /* cause the dialog to close */

fileTransfers_.erase(transferIter);

}

}

}

void QtPlainChatView::acceptMUCInvite()

{

AcceptMUCInviteAction *action = dynamic_cast<AcceptMUCInviteAction*>(sender());

if (action) {

eventStream_->send(boost::make_shared<JoinMUCUIEvent>(action->jid_.toString(), action->password_, boost::optional<std::string>(), false, false, action->isImpromptu_, action->isContinuation_));

delete action->parent_;

}

}

void QtPlainChatView::rejectMUCInvite()

{

AcceptMUCInviteAction *action = dynamic_cast<AcceptMUCInviteAction*>(sender());

if (action) {

/* NOTE: no action required to reject an invite? */

delete action->parent_;

}

}

QtPlainChatView::FileTransfer::FileTransfer(QtPlainChatView* parent, bool senderIsSelf, const std::string& ftId, const std::string& filename, const ChatWindow::FileTransferState state, const std::string &desc, const std::string& msg, bool initializing)

: PopupDialog(parent), bar_(0), senderIsSelf_(senderIsSelf), ftId_(ftId), filename_(filename), description_(desc), message_(msg), initializing_(initializing)

{

QHBoxLayout* layout = new QHBoxLayout;

QLabel* statusLabel = new QLabel;

layout_->addWidget(statusLabel);

layout_->addWidget(new QLabel); /* padding */

if (initializing_) {

FileTransfer::Action* accept = new FileTransfer::Action(senderIsSelf?"Confirm":"Accept", ftId);

parent->connect(accept, SIGNAL(clicked()), SLOT(fileTransferAccept()));

layout_->addWidget(accept);

FileTransfer::Action* reject = new FileTransfer::Action(senderIsSelf?"Cancel":"Reject", ftId);

parent->connect(reject, SIGNAL(clicked()), SLOT(fileTransferReject()));

layout_->addWidget(reject);

statusLabel->setText(P2QSTRING(msg));

return;

}

std::string status = msg;

switch (state) {

case ChatWindow::WaitingForAccept: {

status = "Waiting for user to accept <i>" + filename + "</i>...";

FileTransfer::Action* cancel = new FileTransfer::Action("Cancel", ftId);

parent->connect(cancel, SIGNAL(clicked()), SLOT(fileTransferReject()));

layout_->addWidget(cancel);

break;

}

case ChatWindow::Negotiating: {

status = "Preparing to transfer <i>" + filename + "</i>...";

FileTransfer::Action* cancel = new FileTransfer::Action("Cancel", ftId);

parent->connect(cancel, SIGNAL(clicked()), SLOT(fileTransferReject()));

layout_->addWidget(cancel);

break;

}

case ChatWindow::Transferring: {

status = "Transferring <i>" + filename + "</i>...";

bar_ = new QProgressBar;

bar_->setRange(0, 100);

bar_->setValue(0);

layout->addWidget(bar_);

FileTransfer::Action* cancel = new FileTransfer::Action("Cancel", ftId);

parent->connect(cancel, SIGNAL(clicked()), SLOT(fileTransferReject()));

layout_->addWidget(cancel);

break;

}

case ChatWindow::Canceled: {

status = "File <i>" + filename + "</i> was canceled.";

FileTransfer::Action* finish = new FileTransfer::Action("Hide", ftId);

parent->connect(finish, SIGNAL(clicked()), SLOT(fileTransferFinish()));

layout_->addWidget(finish);

break;

}

case ChatWindow::Finished: {

status = "File <i>" + filename + "</i> was transfered successfully.";

FileTransfer::Action* finish = new FileTransfer::Action("Hide", ftId);

parent->connect(finish, SIGNAL(clicked()), SLOT(fileTransferFinish()));

layout_->addWidget(finish);

break;

}

case ChatWindow::FTFailed: {

status = "File transfer failed: <i>" + filename + "</i>";

FileTransfer::Action* finish = new FileTransfer::Action("Hide", ftId);

parent->connect(finish, SIGNAL(clicked()), SLOT(fileTransferFinish()));

layout_->addWidget(finish);

break;

}

}

statusLabel->setText(P2QSTRING(status));

}

void QtPlainChatView::LogTextEdit::contextMenuEvent(QContextMenuEvent *event)

{

QMenu *menu = createStandardContextMenu();

menu->exec(event->globalPos());

delete menu;

}

}

/*

* Copyright (c) 2013-2014 Kevin Smith and Remko Tronçon

* Licensed under the GNU General Public License v3.

* See Documentation/Licenses/GPLv3.txt for more information.

*/

#pragma once

#include <string>

#include <boost/shared_ptr.hpp>

#include <boost/date_time/posix_time/posix_time.hpp>

#include <QWidget>

#include <QTextEdit>

#include <Swift/Controllers/UIInterfaces/ChatWindow.h>

#include <Swift/QtUI/QtChatView.h>

#include <Swift/QtUI/QtChatWindow.h>

class QTextEdit;

class QProgressBar;

namespace Swift {

class HighlightAction;

class SecurityLabel;

class QtPlainChatView : public QtChatView {

Q_OBJECT

public:

QtPlainChatView(QtChatWindow *window, UIEventStream* eventStream);

virtual ~QtPlainChatView();

/** Add message to window.

* @return id of added message (for acks).

*/

virtual std::string addMessage(const ChatWindow::ChatMessage& /*message*/, const std::string& /*senderName*/, bool /*senderIsSelf*/, boost::shared_ptr<SecurityLabel> /*label*/, const std::string& /*avatarPath*/, const boost::posix_time::ptime& /*time*/, const HighlightAction& /*highlight*/);

/** Adds action to window.

* @return id of added message (for acks);

*/

virtual std::string addAction(const ChatWindow::ChatMessage& /*message*/, const std::string& /*senderName*/, bool /*senderIsSelf*/, boost::shared_ptr<SecurityLabel> /*label*/, const std::string& /*avatarPath*/, const boost::posix_time::ptime& /*time*/, const HighlightAction& /*highlight*/);

virtual void addSystemMessage(const ChatWindow::ChatMessage& /*message*/, ChatWindow::Direction /*direction*/);

virtual void addPresenceMessage(const ChatWindow::ChatMessage& /*message*/, ChatWindow::Direction /*direction*/);

virtual void addErrorMessage(const ChatWindow::ChatMessage& /*message*/);

virtual void replaceMessage(const ChatWindow::ChatMessage& /*message*/, const std::string& /*id*/, const boost::posix_time::ptime& /*time*/, const HighlightAction& /*highlight*/);

virtual void replaceWithAction(const ChatWindow::ChatMessage& /*message*/, const std::string& /*id*/, const boost::posix_time::ptime& /*time*/, const HighlightAction& /*highlight*/);

virtual void setAckState(const std::string& /*id*/, ChatWindow::AckState /*state*/);

virtual std::string addFileTransfer(const std::string& /*senderName*/, bool /*senderIsSelf*/, const std::string& /*filename*/, const boost::uintmax_t /*sizeInBytes*/);

virtual void setFileTransferProgress(std::string, const int /*percentageDone*/);

virtual void setFileTransferStatus(std::string, const ChatWindow::FileTransferState /*state*/, const std::string& /*msg*/ = "");

virtual void addMUCInvitation(const std::string& /*senderName*/, const JID& /*jid*/, const std::string& /*reason*/, const std::string& /*password*/, bool /*direct*/, bool /*isImpromptu*/, bool /*isContinuation*/);

virtual std::string addWhiteboardRequest(const QString& /*contact*/, bool /*senderIsSelf*/) {return "";};

virtual void setWhiteboardSessionStatus(const std::string& /*id*/, const ChatWindow::WhiteboardSessionState /*state*/) {};

virtual void setMessageReceiptState(const std::string& /*id*/, ChatWindow::ReceiptState /*state*/) {};

virtual void showEmoticons(bool /*show*/) {};

virtual void addLastSeenLine() {};

public slots:

virtual void resizeFont(int /*fontSizeSteps*/) {};

virtual void scrollToBottom();

virtual void handleKeyPressEvent(QKeyEvent* /*event*/) {};

virtual void fileTransferAccept();

virtual void fileTransferReject();

virtual void fileTransferFinish();

virtual void acceptMUCInvite();

virtual void rejectMUCInvite();

private:

struct PopupDialog {

PopupDialog(QtPlainChatView* parent) {

dialog_ = new QFrame(parent);

dialog_->setFrameShape(QFrame::Panel);

dialog_->setFrameShadow(QFrame::Raised);

layout_ = new QHBoxLayout;

dialog_->setLayout(layout_);

}

virtual ~PopupDialog() {

delete dialog_;

}

QFrame* dialog_;

QHBoxLayout* layout_;

};

struct AcceptMUCInviteAction : public QPushButton {

AcceptMUCInviteAction(PopupDialog* parent, const std::string& text, const JID& jid, const std::string& senderName, const std::string& password, bool isImpromptu, bool isContinuation)

: QPushButton(P2QSTRING(text)), parent_(parent), jid_(jid), senderName_(senderName), password_(password), isImpromptu_(isImpromptu), isContinuation_(isContinuation) {}

PopupDialog *parent_;

JID jid_;

std::string senderName_;

std::string password_;

bool isImpromptu_;

bool isContinuation_;

};

struct FileTransfer : public PopupDialog {

struct Action : QPushButton {

Action(const std::string& text, const std::string& id)

: QPushButton(P2QSTRING(text)), id_(id) {}

std::string id_;

};

FileTransfer(QtPlainChatView* parent, bool senderIsSelf, const std::string& ftId, const std::string& filename, const ChatWindow::FileTransferState state, const std::string& desc, const std::string& msg, bool initializing);

QProgressBar* bar_;

bool senderIsSelf_;

std::string ftId_;

std::string filename_;

std::string description_;

std::string message_;

bool initializing_;

};

class LogTextEdit : public QTextEdit {

public:

LogTextEdit(QWidget* parent) : QTextEdit(parent) {}

virtual ~LogTextEdit() {}

virtual void contextMenuEvent(QContextMenuEvent *event);

};

typedef std::map<std::string, FileTransfer*> FileTransferMap;

QtChatWindow* window_;

UIEventStream* eventStream_;

LogTextEdit* log_;

FileTransferMap fileTransfers_;

std::map<std::string, boost::shared_ptr<SecurityLabel> > lastMessageLabel_;

int idGenerator_;

};

}

/*

* Copyright (c) 2010-2014 Remko Tronçon

* Licensed under the GNU General Public License v3.

* See Documentation/Licenses/GPLv3.txt for more information.

*/

#include "QtWebKitChatView.h"

#include <QtDebug>

#include <QEventLoop>

#include <QFile>

#include <QDesktopServices>

#include <QVBoxLayout>

#include <QWebFrame>

#include <QKeyEvent>

#include <QStackedWidget>

#include <QTimer>

#include <QMessageBox>

#include <QApplication>

#include <QInputDialog>

#include <QFileDialog>

#include <Swiften/Base/Log.h>

#include <Swiften/Base/FileSize.h>

#include <Swiften/StringCodecs/Base64.h>

#include <Swift/Controllers/UIEvents/UIEventStream.h>

#include <Swift/Controllers/UIEvents/JoinMUCUIEvent.h>

#include <Swift/QtUI/QtWebView.h>

#include <Swift/QtUI/QtChatWindow.h>

#include <Swift/QtUI/QtChatWindowJSBridge.h>

#include <Swift/QtUI/QtScaledAvatarCache.h>

#include <Swift/QtUI/QtSwiftUtil.h>

#include <Swift/QtUI/QtUtilities.h>

#include <Swift/QtUI/MessageSnippet.h>

#include <Swift/QtUI/SystemMessageSnippet.h>

namespace Swift {

const QString QtWebKitChatView::ButtonWhiteboardSessionCancel = QString("whiteboard-cancel");

const QString QtWebKitChatView::ButtonWhiteboardSessionAcceptRequest = QString("whiteboard-acceptrequest");

const QString QtWebKitChatView::ButtonWhiteboardShowWindow = QString("whiteboard-showwindow");

const QString QtWebKitChatView::ButtonFileTransferCancel = QString("filetransfer-cancel");

const QString QtWebKitChatView::ButtonFileTransferSetDescription = QString("filetransfer-setdescription");

const QString QtWebKitChatView::ButtonFileTransferSendRequest = QString("filetransfer-sendrequest");

const QString QtWebKitChatView::ButtonFileTransferAcceptRequest = QString("filetransfer-acceptrequest");

const QString QtWebKitChatView::ButtonMUCInvite = QString("mucinvite");

QtWebKitChatView::QtWebKitChatView(QtChatWindow* window, UIEventStream* eventStream, QtChatTheme* theme, QWidget* parent, bool disableAutoScroll) : QtChatView(parent), window_(window), eventStream_(eventStream), fontSizeSteps_(0), disableAutoScroll_(disableAutoScroll), previousMessageKind_(PreviosuMessageWasNone), previousMessageWasSelf_(false), showEmoticons_(false), insertingLastLine_(false), idCounter_(0) {

theme_ = theme;

QVBoxLayout* mainLayout = new QVBoxLayout(this);

mainLayout->setSpacing(0);

mainLayout->setContentsMargins(0,0,0,0);

webView_ = new QtWebView(this);

connect(webView_, SIGNAL(linkClicked(const QUrl&)), SLOT(handleLinkClicked(const QUrl&)));

connect(webView_, SIGNAL(loadFinished(bool)), SLOT(handleViewLoadFinished(bool)));

connect(webView_, SIGNAL(gotFocus()), SIGNAL(gotFocus()));

connect(webView_, SIGNAL(clearRequested()), SLOT(handleClearRequested()));

connect(webView_, SIGNAL(fontGrowRequested()), SLOT(increaseFontSize()));

connect(webView_, SIGNAL(fontShrinkRequested()), SLOT(decreaseFontSize()));

#if defined (Q_OS_UNIX) && !defined(Q_OS_MAC)

/* To give a border on Linux, where it looks bad without */

QStackedWidget* stack = new QStackedWidget(this);

stack->addWidget(webView_);

stack->setFrameStyle(QFrame::StyledPanel | QFrame::Sunken);

stack->setLineWidth(2);

mainLayout->addWidget(stack);

#else

mainLayout->addWidget(webView_);

#endif

#ifdef SWIFT_EXPERIMENTAL_FT

setAcceptDrops(true);

#endif

webPage_ = new QWebPage(this);

webPage_->setLinkDelegationPolicy(QWebPage::DelegateAllLinks);

if (Log::getLogLevel() == Log::debug) {

webPage_->settings()->setAttribute(QWebSettings::DeveloperExtrasEnabled, true);

}

webView_->setPage(webPage_);

connect(webPage_, SIGNAL(selectionChanged()), SLOT(copySelectionToClipboard()));

connect(webPage_, SIGNAL(scrollRequested(int, int, const QRect&)), SLOT(handleScrollRequested(int, int, const QRect&)));

viewReady_ = false;

isAtBottom_ = true;

resetView();

jsBridge = new QtChatWindowJSBridge();

addToJSEnvironment("chatwindow", jsBridge);

connect(jsBridge, SIGNAL(buttonClicked(QString,QString,QString,QString,QString,QString)), this, SLOT(handleHTMLButtonClicked(QString,QString,QString,QString,QString,QString)));

}

QtWebKitChatView::~QtWebKitChatView() {

delete jsBridge;

}

void QtWebKitChatView::handleClearRequested() {

QMessageBox messageBox(this);

messageBox.setWindowTitle(tr("Clear log"));

messageBox.setText(tr("You are about to clear the contents of your chat log."));

messageBox.setInformativeText(tr("Are you sure?"));

messageBox.setStandardButtons(QMessageBox::Yes | QMessageBox::No);

messageBox.setDefaultButton(QMessageBox::Yes);

int button = messageBox.exec();

if (button == QMessageBox::Yes) {

logCleared();

resetView();

}

}

void QtWebKitChatView::handleKeyPressEvent(QKeyEvent* event) {

webView_->keyPressEvent(event);

}

void QtWebKitChatView::addMessageBottom(boost::shared_ptr<ChatSnippet> snippet) {

if (viewReady_) {

addToDOM(snippet);

} else {

/* If this asserts, the previous queuing code was necessary and should be reinstated */

assert(false);

}

}

void QtWebKitChatView::addMessageTop(boost::shared_ptr<ChatSnippet> snippet) {

// save scrollbar maximum value

if (!topMessageAdded_) {

scrollBarMaximum_ = webPage_->mainFrame()->scrollBarMaximum(Qt::Vertical);

}

topMessageAdded_ = true;

QWebElement continuationElement = firstElement_.findFirst("#insert");

bool insert = snippet->getAppendToPrevious();

bool fallback = continuationElement.isNull();

boost::shared_ptr<ChatSnippet> newSnippet = (insert && fallback) ? snippet->getContinuationFallbackSnippet() : snippet;

QWebElement newElement = snippetToDOM(newSnippet);

if (insert && !fallback) {

Q_ASSERT(!continuationElement.isNull());

continuationElement.replace(newElement);

} else {

continuationElement.removeFromDocument();

topInsertPoint_.prependOutside(newElement);

}

firstElement_ = newElement;

if (lastElement_.isNull()) {

lastElement_ = firstElement_;

}

if (fontSizeSteps_ != 0) {

double size = 1.0 + 0.2 * fontSizeSteps_;

QString sizeString(QString().setNum(size, 'g', 3) + "em");

const QWebElementCollection spans = firstElement_.findAll("span.swift_resizable");

Q_FOREACH (QWebElement span, spans) {

span.setStyleProperty("font-size", sizeString);

}

}

}

QWebElement QtWebKitChatView::snippetToDOM(boost::shared_ptr<ChatSnippet> snippet) {

QWebElement newElement = newInsertPoint_.clone();

newElement.setInnerXml(snippet->getContent());

Q_ASSERT(!newElement.isNull());

return newElement;

}

void QtWebKitChatView::addToDOM(boost::shared_ptr<ChatSnippet> snippet) {

//qDebug() << snippet->getContent();

rememberScrolledToBottom();

bool insert = snippet->getAppendToPrevious();

QWebElement continuationElement = lastElement_.findFirst("#insert");

bool fallback = insert && continuationElement.isNull();

boost::shared_ptr<ChatSnippet> newSnippet = (insert && fallback) ? snippet->getContinuationFallbackSnippet() : snippet;

QWebElement newElement = snippetToDOM(newSnippet);

if (insert && !fallback) {

Q_ASSERT(!continuationElement.isNull());

continuationElement.replace(newElement);

} else {

continuationElement.removeFromDocument();

newInsertPoint_.prependOutside(newElement);

}

lastElement_ = newElement;

if (fontSizeSteps_ != 0) {

double size = 1.0 + 0.2 * fontSizeSteps_;

QString sizeString(QString().setNum(size, 'g', 3) + "em");

const QWebElementCollection spans = lastElement_.findAll("span.swift_resizable");

Q_FOREACH (QWebElement span, spans) {

span.setStyleProperty("font-size", sizeString);

}

}

//qDebug() << "-----------------";

//qDebug() << webPage_->mainFrame()->toHtml();

}

void QtWebKitChatView::addLastSeenLine() {

/* if the line is added we should break the snippet */

insertingLastLine_ = true;

if (lineSeparator_.isNull()) {

lineSeparator_ = newInsertPoint_.clone();

lineSeparator_.setInnerXml(QString("<hr/>"));

newInsertPoint_.prependOutside(lineSeparator_);

}

else {

QWebElement lineSeparatorC = lineSeparator_.clone();

lineSeparatorC.removeFromDocument();

}

newInsertPoint_.prependOutside(lineSeparator_);

}

assert(viewReady_);

rememberScrolledToBottom();

assert(!lastElement_.isNull());

QWebElement replace = lastElement_.findFirst("span.swift_message");

assert(!replace.isNull());

QString old = lastElement_.toOuterXml();

replace.setInnerXml(ChatSnippet::escape(newMessage));

}

void QtWebKitChatView::replaceLastMessage(const QString& newMessage, const QString& note) {

rememberScrolledToBottom();

QWebElement replace = lastElement_.findFirst("span.swift_time");

assert(!replace.isNull());

replace.setInnerXml(ChatSnippet::escape(note));

}

QString QtWebKitChatView::getLastSentMessage() {

return lastElement_.toPlainText();

}

void QtWebKitChatView::addToJSEnvironment(const QString& name, QObject* obj) {

webView_->page()->currentFrame()->addToJavaScriptWindowObject(name, obj);

}

void QtWebKitChatView::replaceMessage(const QString& newMessage, const QString& id, const QDateTime& editTime) {

rememberScrolledToBottom();

QWebElement message = document_.findFirst("#" + id);

if (!message.isNull()) {

QWebElement replaceContent = message.findFirst("span.swift_inner_message");

assert(!replaceContent.isNull());

QString old = replaceContent.toOuterXml();

replaceContent.setInnerXml(ChatSnippet::escape(newMessage));

QWebElement replaceTime = message.findFirst("span.swift_time");

assert(!replaceTime.isNull());

old = replaceTime.toOuterXml();

replaceTime.setInnerXml(ChatSnippet::escape(tr("%1 edited").arg(ChatSnippet::timeToEscapedString(editTime))));

}

else {

qWarning() << "Trying to replace element with id " << id << " but it's not there.";

}

}

void QtWebKitChatView::showEmoticons(bool show) {

showEmoticons_ = show;

{

const QWebElementCollection spans = document_.findAll("span.swift_emoticon_image");

Q_FOREACH (QWebElement span, spans) {

span.setStyleProperty("display", show ? "inline" : "none");

}

}

{

const QWebElementCollection spans = document_.findAll("span.swift_emoticon_text");

Q_FOREACH (QWebElement span, spans) {

span.setStyleProperty("display", show ? "none" : "inline");

}

}

}

void QtWebKitChatView::copySelectionToClipboard() {

if (!webPage_->selectedText().isEmpty()) {

webPage_->triggerAction(QWebPage::Copy);

}

}

void QtWebKitChatView::setAckXML(const QString& id, const QString& xml) {

QWebElement message = document_.findFirst("#" + id);

/* Deliberately not asserting here, so that when we start expiring old messages it won't hit us */

if (message.isNull()) return;

QWebElement ackElement = message.findFirst("span.swift_ack");

assert(!ackElement.isNull());

ackElement.setInnerXml(xml);

}

void QtWebKitChatView::setReceiptXML(const QString& id, const QString& xml) {

QWebElement message = document_.findFirst("#" + id);

if (message.isNull()) return;

QWebElement receiptElement = message.findFirst("span.swift_receipt");

assert(!receiptElement.isNull());

receiptElement.setInnerXml(xml);

}

void QtWebKitChatView::displayReceiptInfo(const QString& id, bool showIt) {

QWebElement message = document_.findFirst("#" + id);

if (message.isNull()) return;

QWebElement receiptElement = message.findFirst("span.swift_receipt");

assert(!receiptElement.isNull());

receiptElement.setStyleProperty("display", showIt ? "inline" : "none");

}

void QtWebKitChatView::rememberScrolledToBottom() {

isAtBottom_ = webPage_->mainFrame()->scrollBarValue(Qt::Vertical) >= (webPage_->mainFrame()->scrollBarMaximum(Qt::Vertical) - 1);

}

void QtWebKitChatView::scrollToBottom() {

isAtBottom_ = true;

webPage_->mainFrame()->setScrollBarValue(Qt::Vertical, webPage_->mainFrame()->scrollBarMaximum(Qt::Vertical));

webView_->update(); /* Work around redraw bug in some versions of Qt. */

}

void QtWebKitChatView::handleFrameSizeChanged() {

if (topMessageAdded_) {

// adjust new scrollbar position

int newMaximum = webPage_->mainFrame()->scrollBarMaximum(Qt::Vertical);

webPage_->mainFrame()->setScrollBarValue(Qt::Vertical, newMaximum - scrollBarMaximum_);

topMessageAdded_ = false;

}

if (isAtBottom_ && !disableAutoScroll_) {

scrollToBottom();

}

}

void QtWebKitChatView::handleLinkClicked(const QUrl& url) {

QDesktopServices::openUrl(url);

}

void QtWebKitChatView::handleViewLoadFinished(bool ok) {

Q_ASSERT(ok);

viewReady_ = true;

}

void QtWebKitChatView::increaseFontSize(int numSteps) {

//qDebug() << "Increasing";

fontSizeSteps_ += numSteps;

emit fontResized(fontSizeSteps_);

}

void QtWebKitChatView::decreaseFontSize() {

fontSizeSteps_--;

if (fontSizeSteps_ < 0) {

fontSizeSteps_ = 0;

}

emit fontResized(fontSizeSteps_);

}

void QtWebKitChatView::resizeFont(int fontSizeSteps) {

fontSizeSteps_ = fontSizeSteps;

double size = 1.0 + 0.2 * fontSizeSteps_;

QString sizeString(QString().setNum(size, 'g', 3) + "em");

//qDebug() << "Setting to " << sizeString;

const QWebElementCollection spans = document_.findAll("span.swift_resizable");

Q_FOREACH (QWebElement span, spans) {

span.setStyleProperty("font-size", sizeString);

}

webView_->setFontSizeIsMinimal(size == 1.0);

}

void QtWebKitChatView::resetView() {

lastElement_ = QWebElement();

firstElement_ = lastElement_;

topMessageAdded_ = false;

scrollBarMaximum_ = 0;

QString pageHTML = theme_->getTemplate();

pageHTML.replace("==bodyBackground==", "background-color:#e3e3e3");

pageHTML.replace(pageHTML.indexOf("%@"), 2, theme_->getBase());

if (pageHTML.count("%@") > 3) {

pageHTML.replace(pageHTML.indexOf("%@"), 2, theme_->getMainCSS());

}

pageHTML.replace(pageHTML.indexOf("%@"), 2, "Variants/Blue on Green.css");

pageHTML.replace(pageHTML.indexOf("%@"), 2, ""/*headerSnippet.getContent()*/);

pageHTML.replace(pageHTML.indexOf("%@"), 2, ""/*footerSnippet.getContent()*/);

QEventLoop syncLoop;

connect(webView_, SIGNAL(loadFinished(bool)), &syncLoop, SLOT(quit()));

webPage_->mainFrame()->setHtml(pageHTML);

while (!viewReady_) {

QTimer t;

t.setSingleShot(true);

connect(&t, SIGNAL(timeout()), &syncLoop, SLOT(quit()));

t.start(50);

syncLoop.exec();

}

document_ = webPage_->mainFrame()->documentElement();

resetTopInsertPoint();

QWebElement chatElement = document_.findFirst("#Chat");

newInsertPoint_ = chatElement.clone();

newInsertPoint_.setOuterXml("<div id='swift_insert'/>");

chatElement.appendInside(newInsertPoint_);

Q_ASSERT(!newInsertPoint_.isNull());

scrollToBottom();

connect(webPage_->mainFrame(), SIGNAL(contentsSizeChanged(const QSize&)), this, SLOT(handleFrameSizeChanged()), Qt::UniqueConnection);

}

static QWebElement findElementWithID(QWebElement document, QString elementName, QString id) {

QWebElementCollection elements = document.findAll(elementName);

Q_FOREACH(QWebElement element, elements) {

if (element.attribute("id") == id) {

return element;

}

}

return QWebElement();

}

void QtWebKitChatView::setFileTransferProgress(QString id, const int percentageDone) {

QWebElement ftElement = findElementWithID(document_, "div", id);

if (ftElement.isNull()) {

SWIFT_LOG(debug) << "Tried to access FT UI via invalid id!" << std::endl;

return;

}

QWebElement progressBar = ftElement.findFirst("div.progressbar");

progressBar.setStyleProperty("width", QString::number(percentageDone) + "%");

QWebElement progressBarValue = ftElement.findFirst("div.progressbar-value");

progressBarValue.setInnerXml(QString::number(percentageDone) + " %");

}

void QtWebKitChatView::setFileTransferStatus(QString id, const ChatWindow::FileTransferState state, const QString& /* msg */) {

QWebElement ftElement = findElementWithID(document_, "div", id);

if (ftElement.isNull()) {

SWIFT_LOG(debug) << "Tried to access FT UI via invalid id! id = " << Q2PSTRING(id) << std::endl;

return;

}

QString newInnerHTML = "";

if (state == ChatWindow::WaitingForAccept) {

newInnerHTML = tr("Waiting for other side to accept the transfer.") + "<br/>" +

buildChatWindowButton(tr("Cancel"), ButtonFileTransferCancel, id);

}

if (state == ChatWindow::Negotiating) {

// replace with text "Negotiaging" + Cancel button

newInnerHTML = tr("Negotiating...") + "<br/>" +

buildChatWindowButton(tr("Cancel"), ButtonFileTransferCancel, id);

}

else if (state == ChatWindow::Transferring) {

// progress bar + Cancel Button

newInnerHTML = "<div style=\"position: relative; width: 90%; height: 20px; border: 2px solid grey; -webkit-border-radius: 10px;\">"

"<div class=\"progressbar\" style=\"width: 0%; height: 100%; background: #AAA; -webkit-border-radius: 6px;\">"

"<div class=\"progressbar-value\" style=\"position: absolute; top: 0px; left: 0px; width: 100%; text-align: center; padding-top: 2px;\">"

"0%"

"</div>"

"</div>"

"</div>" +

buildChatWindowButton(tr("Cancel"), ButtonFileTransferCancel, id);

}

else if (state == ChatWindow::Canceled) {

newInnerHTML = tr("Transfer has been canceled!");

}

else if (state == ChatWindow::Finished) {

// text "Successful transfer"

newInnerHTML = tr("Transfer completed successfully.");

}

else if (state == ChatWindow::FTFailed) {

newInnerHTML = tr("Transfer failed.");

}

ftElement.setInnerXml(newInnerHTML);

}

void QtWebKitChatView::setWhiteboardSessionStatus(QString id, const ChatWindow::WhiteboardSessionState state) {

QWebElement divElement = findElementWithID(document_, "div", id);

QString newInnerHTML;

if (state == ChatWindow::WhiteboardAccepted) {

newInnerHTML = tr("Started whiteboard chat") + "<br/>" + buildChatWindowButton(tr("Show whiteboard"), ButtonWhiteboardShowWindow, id);

} else if (state == ChatWindow::WhiteboardTerminated) {

newInnerHTML = tr("Whiteboard chat has been canceled");

} else if (state == ChatWindow::WhiteboardRejected) {

newInnerHTML = tr("Whiteboard chat request has been rejected");

}

divElement.setInnerXml(newInnerHTML);

}

void QtWebKitChatView::setMUCInvitationJoined(QString id) {

QWebElement divElement = findElementWithID(document_, "div", id);

QWebElement buttonElement = findElementWithID(divElement, "input", "mucinvite");

if (!buttonElement.isNull()) {

buttonElement.setAttribute("value", tr("Return to room"));

}

}

void QtWebKitChatView::handleScrollRequested(int, int dy, const QRect&) {

rememberScrolledToBottom();

int pos = webPage_->mainFrame()->scrollBarValue(Qt::Vertical) - dy;

emit scrollRequested(pos);

if (pos == 0) {

emit scrollReachedTop();

}

else if (pos == webPage_->mainFrame()->scrollBarMaximum(Qt::Vertical)) {

emit scrollReachedBottom();

}

}

int QtWebKitChatView::getSnippetPositionByDate(const QDate& date) {

QWebElement message = webPage_->mainFrame()->documentElement().findFirst(".date" + date.toString(Qt::ISODate));

return message.geometry().top();

}

void QtWebKitChatView::resetTopInsertPoint() {

QWebElement continuationElement = firstElement_.findFirst("#insert");

continuationElement.removeFromDocument();

firstElement_ = QWebElement();

topInsertPoint_.removeFromDocument();

QWebElement chatElement = document_.findFirst("#Chat");

topInsertPoint_ = chatElement.clone();

topInsertPoint_.setOuterXml("<div id='swift_insert'/>");

chatElement.prependInside(topInsertPoint_);

}

std::string QtWebKitChatView::addMessage(

const ChatWindow::ChatMessage& message,

const std::string& senderName,

bool senderIsSelf,

boost::shared_ptr<SecurityLabel> label,

const std::string& avatarPath,

const boost::posix_time::ptime& time,

const HighlightAction& highlight) {

return addMessage(chatMessageToHTML(message), senderName, senderIsSelf, label, avatarPath, "", time, highlight, ChatSnippet::getDirection(message));

}

QString QtWebKitChatView::chatMessageToHTML(const ChatWindow::ChatMessage& message) {

QString result;

foreach (boost::shared_ptr<ChatWindow::ChatMessagePart> part, message.getParts()) {

boost::shared_ptr<ChatWindow::ChatTextMessagePart> textPart;

boost::shared_ptr<ChatWindow::ChatURIMessagePart> uriPart;

boost::shared_ptr<ChatWindow::ChatEmoticonMessagePart> emoticonPart;

boost::shared_ptr<ChatWindow::ChatHighlightingMessagePart> highlightPart;

if ((textPart = boost::dynamic_pointer_cast<ChatWindow::ChatTextMessagePart>(part))) {

QString text = QtUtilities::htmlEscape(P2QSTRING(textPart->text));

text.replace("\n","<br/>");

result += text;

continue;

}

if ((uriPart = boost::dynamic_pointer_cast<ChatWindow::ChatURIMessagePart>(part))) {

QString uri = QtUtilities::htmlEscape(P2QSTRING(uriPart->target));

result += "<a href='" + uri + "' >" + uri + "</a>";

continue;

}

if ((emoticonPart = boost::dynamic_pointer_cast<ChatWindow::ChatEmoticonMessagePart>(part))) {

QString textStyle = showEmoticons_ ? "style='display:none'" : "";

QString imageStyle = showEmoticons_ ? "" : "style='display:none'";

result += "<span class='swift_emoticon_image' " + imageStyle + "><img src='" + P2QSTRING(emoticonPart->imagePath) + "'/></span><span class='swift_emoticon_text' " + textStyle + ">" + QtUtilities::htmlEscape(P2QSTRING(emoticonPart->alternativeText)) + "</span>";

continue;

}

if ((highlightPart = boost::dynamic_pointer_cast<ChatWindow::ChatHighlightingMessagePart>(part))) {

//FIXME: Maybe do something here. Anything, really.

continue;

}

}

return result;

}

QString QtWebKitChatView::getHighlightSpanStart(const HighlightAction& highlight) {

QString color = QtUtilities::htmlEscape(P2QSTRING(highlight.getTextColor()));

QString background = QtUtilities::htmlEscape(P2QSTRING(highlight.getTextBackground()));

if (color.isEmpty()) {

color = "black";

}

if (background.isEmpty()) {

background = "yellow";

}

return QString("<span style=\"color: %1; background: %2\">").arg(color).arg(background);

}

std::string QtWebKitChatView::addMessage(

const QString& message,

const std::string& senderName,

bool senderIsSelf,

boost::shared_ptr<SecurityLabel> label,

const std::string& avatarPath,

const QString& style,

const boost::posix_time::ptime& time,

const HighlightAction& highlight,

ChatSnippet::Direction direction) {

QString scaledAvatarPath = QtScaledAvatarCache(32).getScaledAvatarPath(avatarPath.c_str());

QString htmlString;

if (label) {

htmlString = QString("<span style=\"border: thin dashed grey; padding-left: .5em; padding-right: .5em; color: %1; background-color: %2; font-size: 90%; margin-right: .5em; \" class='swift_label'>").arg(QtUtilities::htmlEscape(P2QSTRING(label->getForegroundColor()))).arg(QtUtilities::htmlEscape(P2QSTRING(label->getBackgroundColor())));

htmlString += QString("%1</span> ").arg(QtUtilities::htmlEscape(P2QSTRING(label->getDisplayMarking())));

}

QString styleSpanStart = style == "" ? "" : "<span style=\"" + style + "\">";

QString styleSpanEnd = style == "" ? "" : "</span>";

QString highlightSpanStart = highlight.highlightText() ? getHighlightSpanStart(highlight) : "";

QString highlightSpanEnd = highlight.highlightText() ? "</span>" : "";

htmlString += "<span class='swift_inner_message'>" + styleSpanStart + highlightSpanStart + message + highlightSpanEnd + styleSpanEnd + "</span>" ;

bool appendToPrevious = appendToPreviousCheck(PreviousMessageWasMessage, senderName, senderIsSelf);

QString qAvatarPath = scaledAvatarPath.isEmpty() ? "qrc:/icons/avatar.png" : QUrl::fromLocalFile(scaledAvatarPath).toEncoded();

std::string id = "id" + boost::lexical_cast<std::string>(idCounter_++);

addMessageBottom(boost::make_shared<MessageSnippet>(htmlString, QtUtilities::htmlEscape(P2QSTRING(senderName)), B2QDATE(time), qAvatarPath, senderIsSelf, appendToPrevious, theme_, P2QSTRING(id), direction));

previousMessageWasSelf_ = senderIsSelf;

previousSenderName_ = P2QSTRING(senderName);

previousMessageKind_ = PreviousMessageWasMessage;

return id;

}

std::string QtWebKitChatView::addAction(const ChatWindow::ChatMessage& message, const std::string &senderName, bool senderIsSelf, boost::shared_ptr<SecurityLabel> label, const std::string& avatarPath, const boost::posix_time::ptime& time, const HighlightAction& highlight) {

return addMessage(" *" + chatMessageToHTML(message) + "*", senderName, senderIsSelf, label, avatarPath, "font-style:italic ", time, highlight, ChatSnippet::getDirection(message));

}

static QString encodeButtonArgument(const QString& str) {

return QtUtilities::htmlEscape(P2QSTRING(Base64::encode(createByteArray(Q2PSTRING(str)))));

}

static QString decodeButtonArgument(const QString& str) {

return P2QSTRING(byteArrayToString(Base64::decode(Q2PSTRING(str))));

}

QString QtWebKitChatView::buildChatWindowButton(const QString& name, const QString& id, const QString& arg1, const QString& arg2, const QString& arg3, const QString& arg4, const QString& arg5) {

QRegExp regex("[A-Za-z][A-Za-z0-9\\-\\_]+");

Q_ASSERT(regex.exactMatch(id));

QString html = QString("<input id='%2' type='submit' value='%1' onclick='chatwindow.buttonClicked(\"%2\", \"%3\", \"%4\", \"%5\", \"%6\", \"%7\");' />").arg(name).arg(id).arg(encodeButtonArgument(arg1)).arg(encodeButtonArgument(arg2)).arg(encodeButtonArgument(arg3)).arg(encodeButtonArgument(arg4)).arg(encodeButtonArgument(arg5));

return html;

}

std::string QtWebKitChatView::addFileTransfer(const std::string& senderName, bool senderIsSelf, const std::string& filename, const boost::uintmax_t sizeInBytes) {

SWIFT_LOG(debug) << "addFileTransfer" << std::endl;

QString ft_id = QString("ft%1").arg(P2QSTRING(boost::lexical_cast<std::string>(idCounter_++)));

QString actionText;

QString htmlString;

QString formattedFileSize = P2QSTRING(formatSize(sizeInBytes));

if (senderIsSelf) {

// outgoing

actionText = tr("Send file");

htmlString = actionText + ": " + P2QSTRING(filename) + " ( " + formattedFileSize + ") <br/>" +

"<div id='" + ft_id + "'>" +

buildChatWindowButton(tr("Cancel"), ButtonFileTransferCancel, ft_id) +

buildChatWindowButton(tr("Set Description"), ButtonFileTransferSetDescription, ft_id) +

buildChatWindowButton(tr("Send"), ButtonFileTransferSendRequest, ft_id) +

"</div>";

} else {

// incoming

actionText = tr("Receiving file");

htmlString = actionText + ": " + P2QSTRING(filename) + " ( " + formattedFileSize + ") <br/>" +

"<div id='" + ft_id + "'>" +

buildChatWindowButton(tr("Cancel"), ButtonFileTransferCancel, ft_id) +

buildChatWindowButton(tr("Accept"), ButtonFileTransferAcceptRequest, ft_id, P2QSTRING(filename)) +

"</div>";

}

//addMessage(message, senderName, senderIsSelf, boost::shared_ptr<SecurityLabel>(), "", boost::posix_time::second_clock::local_time());

bool appendToPrevious = appendToPreviousCheck(PreviousMessageWasFileTransfer, senderName, senderIsSelf);

QString qAvatarPath = "qrc:/icons/avatar.png";

std::string id = "ftmessage" + boost::lexical_cast<std::string>(idCounter_++);

addMessageBottom(boost::make_shared<MessageSnippet>(htmlString, QtUtilities::htmlEscape(P2QSTRING(senderName)), B2QDATE(boost::posix_time::second_clock::local_time()), qAvatarPath, senderIsSelf, appendToPrevious, theme_, P2QSTRING(id), ChatSnippet::getDirection(actionText)));

previousMessageWasSelf_ = senderIsSelf;

previousSenderName_ = P2QSTRING(senderName);

previousMessageKind_ = PreviousMessageWasFileTransfer;

return Q2PSTRING(ft_id);

}

void QtWebKitChatView::setFileTransferProgress(std::string id, const int percentageDone) {

setFileTransferProgress(P2QSTRING(id), percentageDone);

}

void QtWebKitChatView::setFileTransferStatus(std::string id, const ChatWindow::FileTransferState state, const std::string& msg) {

setFileTransferStatus(P2QSTRING(id), state, P2QSTRING(msg));

}

std::string QtWebKitChatView::addWhiteboardRequest(const QString& contact, bool senderIsSelf) {

QString wb_id = QString("wb%1").arg(P2QSTRING(boost::lexical_cast<std::string>(idCounter_++)));

QString htmlString;

QString actionText;

if (senderIsSelf) {

actionText = tr("Starting whiteboard chat");

htmlString = "<div id='" + wb_id + "'>" + actionText + "<br />"+

buildChatWindowButton(tr("Cancel"), ButtonWhiteboardSessionCancel, wb_id) +

"</div>";

} else {

actionText = tr("%1 would like to start a whiteboard chat");

htmlString = "<div id='" + wb_id + "'>" + actionText.arg(QtUtilities::htmlEscape(contact)) + ": <br/>" +

buildChatWindowButton(tr("Cancel"), ButtonWhiteboardSessionCancel, wb_id) +

buildChatWindowButton(tr("Accept"), ButtonWhiteboardSessionAcceptRequest, wb_id) +

"</div>";

}

QString qAvatarPath = "qrc:/icons/avatar.png";

std::string id = "wbmessage" + boost::lexical_cast<std::string>(idCounter_++);

addMessageBottom(boost::make_shared<MessageSnippet>(htmlString, QtUtilities::htmlEscape(contact), B2QDATE(boost::posix_time::second_clock::local_time()), qAvatarPath, false, false, theme_, P2QSTRING(id), ChatSnippet::getDirection(actionText)));

previousMessageWasSelf_ = false;

previousSenderName_ = contact;

return Q2PSTRING(wb_id);

}

void QtWebKitChatView::setWhiteboardSessionStatus(const std::string& id, const ChatWindow::WhiteboardSessionState state) {

setWhiteboardSessionStatus(P2QSTRING(id), state);

}

void QtWebKitChatView::handleHTMLButtonClicked(QString id, QString encodedArgument1, QString encodedArgument2, QString encodedArgument3, QString encodedArgument4, QString encodedArgument5) {

QString arg1 = decodeButtonArgument(encodedArgument1);

QString arg2 = decodeButtonArgument(encodedArgument2);

QString arg3 = decodeButtonArgument(encodedArgument3);

QString arg4 = decodeButtonArgument(encodedArgument4);

QString arg5 = decodeButtonArgument(encodedArgument5);

if (id.startsWith(ButtonFileTransferCancel)) {

QString ft_id = arg1;

window_->onFileTransferCancel(Q2PSTRING(ft_id));

}

else if (id.startsWith(ButtonFileTransferSetDescription)) {

QString ft_id = arg1;

bool ok = false;

QString text = QInputDialog::getText(this, tr("File transfer description"),

tr("Description:"), QLineEdit::Normal, "", &ok);

if (ok) {

descriptions_[ft_id] = text;

}

}

else if (id.startsWith(ButtonFileTransferSendRequest)) {

QString ft_id = arg1;

QString text = descriptions_.find(ft_id) == descriptions_.end() ? QString() : descriptions_[ft_id];

window_->onFileTransferStart(Q2PSTRING(ft_id), Q2PSTRING(text));

}

else if (id.startsWith(ButtonFileTransferAcceptRequest)) {

QString ft_id = arg1;

QString filename = arg2;

QString path = QFileDialog::getSaveFileName(this, tr("Save File"), filename);

if (!path.isEmpty()) {

window_->onFileTransferAccept(Q2PSTRING(ft_id), Q2PSTRING(path));

}

}

else if (id.startsWith(ButtonWhiteboardSessionAcceptRequest)) {

QString id = arg1;

setWhiteboardSessionStatus(id, ChatWindow::WhiteboardAccepted);

window_->onWhiteboardSessionAccept();

}

else if (id.startsWith(ButtonWhiteboardSessionCancel)) {

QString id = arg1;

setWhiteboardSessionStatus(id, ChatWindow::WhiteboardTerminated);

window_->onWhiteboardSessionCancel();

}

else if (id.startsWith(ButtonWhiteboardShowWindow)) {

QString id = arg1;

window_->onWhiteboardWindowShow();

}

else if (id.startsWith(ButtonMUCInvite)) {

QString roomJID = arg1;

QString password = arg2;

QString elementID = arg3;

QString isImpromptu = arg4;

QString isContinuation = arg5;

eventStream_->send(boost::make_shared<JoinMUCUIEvent>(Q2PSTRING(roomJID), Q2PSTRING(password), boost::optional<std::string>(), false, false, isImpromptu.contains("true"), isContinuation.contains("true")));

setMUCInvitationJoined(elementID);

}

else {

SWIFT_LOG(debug) << "Unknown HTML button! ( " << Q2PSTRING(id) << " )" << std::endl;

}

}

void QtWebKitChatView::addErrorMessage(const ChatWindow::ChatMessage& errorMessage) {

if (window_->isWidgetSelected()) {

window_->onAllMessagesRead();

}

QString errorMessageHTML(chatMessageToHTML(errorMessage));

addMessageBottom(boost::make_shared<SystemMessageSnippet>("<span class=\"error\">" + errorMessageHTML + "</span>", QDateTime::currentDateTime(), false, theme_, ChatSnippet::getDirection(errorMessage)));

previousMessageWasSelf_ = false;

previousMessageKind_ = PreviousMessageWasSystem;

}

void QtWebKitChatView::addSystemMessage(const ChatWindow::ChatMessage& message, ChatWindow::Direction direction) {

if (window_->isWidgetSelected()) {

window_->onAllMessagesRead();

}

QString messageHTML = chatMessageToHTML(message);

addMessageBottom(boost::make_shared<SystemMessageSnippet>(messageHTML, QDateTime::currentDateTime(), false, theme_, getActualDirection(message, direction)));

previousMessageKind_ = PreviousMessageWasSystem;

}

void QtWebKitChatView::replaceWithAction(const ChatWindow::ChatMessage& message, const std::string& id, const boost::posix_time::ptime& time, const HighlightAction& highlight) {

replaceMessage(" *" + chatMessageToHTML(message) + "*", id, time, "font-style:italic ", highlight);

}

void QtWebKitChatView::replaceMessage(const ChatWindow::ChatMessage& message, const std::string& id, const boost::posix_time::ptime& time, const HighlightAction& highlight) {

replaceMessage(chatMessageToHTML(message), id, time, "", highlight);

}

void QtWebKitChatView::replaceMessage(const QString& message, const std::string& id, const boost::posix_time::ptime& time, const QString& style, const HighlightAction& highlight) {

if (!id.empty()) {

if (window_->isWidgetSelected()) {

window_->onAllMessagesRead();

}

QString messageHTML(message);

QString styleSpanStart = style == "" ? "" : "<span style=\"" + style + "\">";

QString styleSpanEnd = style == "" ? "" : "</span>";

QString highlightSpanStart = highlight.highlightText() ? getHighlightSpanStart(highlight) : "";

QString highlightSpanEnd = highlight.highlightText() ? "</span>" : "";

messageHTML = styleSpanStart + highlightSpanStart + messageHTML + highlightSpanEnd + styleSpanEnd;

replaceMessage(messageHTML, P2QSTRING(id), B2QDATE(time));

}

else {

std::cerr << "Trying to replace a message with no id";

}

}

void QtWebKitChatView::addPresenceMessage(const ChatWindow::ChatMessage& message, ChatWindow::Direction direction) {

if (window_->isWidgetSelected()) {

window_->onAllMessagesRead();

}

QString messageHTML = chatMessageToHTML(message);

addMessageBottom(boost::make_shared<SystemMessageSnippet>(messageHTML, QDateTime::currentDateTime(), false, theme_, getActualDirection(message, direction)));

previousMessageKind_ = PreviousMessageWasPresence;

}

}

void QtWebKitChatView::addMUCInvitation(const std::string& senderName, const JID& jid, const std::string& reason, const std::string& password, bool direct, bool isImpromptu, bool isContinuation) {

if (window_->isWidgetSelected()) {

window_->onAllMessagesRead();

}

QString message;

if (isImpromptu) {

message = QObject::tr("You've been invited to join a chat.") + "\n";

} else {

message = QObject::tr("You've been invited to enter the %1 room.").arg(P2QSTRING(jid.toString())) + "\n";

}

QString htmlString = message;

if (!reason.empty()) {

htmlString += QObject::tr("Reason: %1").arg(P2QSTRING(reason)) + "\n";

}

if (!direct) {

htmlString += QObject::tr("This person may not have really sent this invitation!") + "\n";

}

htmlString = chatMessageToHTML(ChatWindow::ChatMessage(Q2PSTRING(htmlString)));

QString id = QString(ButtonMUCInvite + "%1").arg(P2QSTRING(boost::lexical_cast<std::string>(idCounter_++)));

htmlString += "<div id='" + id + "'>" +

buildChatWindowButton(chatMessageToHTML(ChatWindow::ChatMessage(Q2PSTRING((tr("Accept Invite"))))), ButtonMUCInvite, QtUtilities::htmlEscape(P2QSTRING(jid.toString())), QtUtilities::htmlEscape(P2QSTRING(password)), id, QtUtilities::htmlEscape(isImpromptu ? "true" : "false"), QtUtilities::htmlEscape(isContinuation ? "true" : "false")) +

"</div>";

bool appendToPrevious = appendToPreviousCheck(PreviousMessageWasMUCInvite, senderName, false);

QString qAvatarPath = "qrc:/icons/avatar.png";

addMessageBottom(boost::make_shared<MessageSnippet>(htmlString, QtUtilities::htmlEscape(P2QSTRING(senderName)), B2QDATE(boost::posix_time::second_clock::local_time()), qAvatarPath, false, appendToPrevious, theme_, id, ChatSnippet::getDirection(message)));

previousMessageWasSelf_ = false;

previousSenderName_ = P2QSTRING(senderName);

previousMessageKind_ = PreviousMessageWasMUCInvite;

}

void QtWebKitChatView::setAckState(std::string const& id, ChatWindow::AckState state) {

QString xml;

switch (state) {

case ChatWindow::Pending:

xml = "<img src='qrc:/icons/throbber.gif' title='" + tr("This message has not been received by your server yet.") + "'/>";

displayReceiptInfo(P2QSTRING(id), false);

break;

case ChatWindow::Received:

xml = "";

displayReceiptInfo(P2QSTRING(id), true);

break;

case ChatWindow::Failed: xml = "<img src='qrc:/icons/error.png' title='" + tr("This message may not have been transmitted.") + "'/>"; break;

}

setAckXML(P2QSTRING(id), xml);

}

void QtWebKitChatView::setMessageReceiptState(const std::string& id, ChatWindow::ReceiptState state) {

QString xml;

switch (state) {

case ChatWindow::ReceiptReceived:

xml = "<img src='qrc:/icons/check.png' title='" + tr("The receipt for this message has been received.") + "'/>";

break;

case ChatWindow::ReceiptRequested:

xml = "<img src='qrc:/icons/warn.png' title='" + tr("The receipt for this message has not yet been received. The recipient(s) might not have received this message.") + "'/>";

break;

case ChatWindow::ReceiptFailed:

xml = "<img src='qrc:/icons/error.png' title='" + tr("Failed to transmit message to the receipient(s).") + "'/>";

}

setReceiptXML(P2QSTRING(id), xml);

}

bool QtWebKitChatView::appendToPreviousCheck(PreviousMessageKind messageKind, const std::string& senderName, bool senderIsSelf) {

bool result = previousMessageKind_ == messageKind && ((senderIsSelf && previousMessageWasSelf_) || (!senderIsSelf && !previousMessageWasSelf_&& previousSenderName_ == P2QSTRING(senderName)));

if (insertingLastLine_) {

insertingLastLine_ = false;

return false;

}

return result;

}

ChatSnippet::Direction QtWebKitChatView::getActualDirection(const ChatWindow::ChatMessage& message, ChatWindow::Direction direction) {

if (direction == ChatWindow::DefaultDirection) {

return QCoreApplication::translate("QApplication", "QT_LAYOUT_DIRECTION") == "RTL" ? ChatSnippet::RTL : ChatSnippet::LTR;

}

else {

return ChatSnippet::getDirection(message);

}

}

// void QtWebKitChatView::setShowEmoticons(bool value) {

// showEmoticons_ = value;

// }

}

/*

* Copyright (c) 2010-2013 Remko Tronçon

* Licensed under the GNU General Public License v3.

* See Documentation/Licenses/GPLv3.txt for more information.

*/

#pragma once

#include <QString>

#include <QWidget>

#include <QList>

#include <QWebElement>

#include <boost/shared_ptr.hpp>

#include <Swiften/Base/Override.h>

#include <Swift/Controllers/UIInterfaces/ChatWindow.h>

#include <Swift/QtUI/ChatSnippet.h>

#include <Swift/QtUI/QtChatView.h>

class QWebPage;

class QUrl;

class QDate;

namespace Swift {

class QtWebView;

class QtChatTheme;

class QtChatWindowJSBridge;

class UIEventStream;

class QtChatWindow;

class QtWebKitChatView : public QtChatView {

Q_OBJECT

public:

static const QString ButtonWhiteboardSessionCancel;

static const QString ButtonWhiteboardSessionAcceptRequest;

static const QString ButtonWhiteboardShowWindow;

static const QString ButtonFileTransferCancel;

static const QString ButtonFileTransferSetDescription;

static const QString ButtonFileTransferSendRequest;

static const QString ButtonFileTransferAcceptRequest;

static const QString ButtonMUCInvite;

public:

QtWebKitChatView(QtChatWindow* window, UIEventStream* eventStream, QtChatTheme* theme, QWidget* parent, bool disableAutoScroll = false);

~QtWebKitChatView();

/** Add message to window.

* @return id of added message (for acks).

*/

virtual std::string addMessage(const ChatWindow::ChatMessage& message, const std::string& senderName, bool senderIsSelf, boost::shared_ptr<SecurityLabel> label, const std::string& avatarPath, const boost::posix_time::ptime& time, const HighlightAction& highlight) SWIFTEN_OVERRIDE;

/** Adds action to window.

* @return id of added message (for acks);

*/

virtual std::string addAction(const ChatWindow::ChatMessage& message, const std::string& senderName, bool senderIsSelf, boost::shared_ptr<SecurityLabel> label, const std::string& avatarPath, const boost::posix_time::ptime& time, const HighlightAction& highlight) SWIFTEN_OVERRIDE;

virtual void addSystemMessage(const ChatWindow::ChatMessage& message, ChatWindow::Direction direction) SWIFTEN_OVERRIDE;

virtual void addPresenceMessage(const ChatWindow::ChatMessage& message, ChatWindow::Direction direction) SWIFTEN_OVERRIDE;

virtual void addErrorMessage(const ChatWindow::ChatMessage& message) SWIFTEN_OVERRIDE;

virtual void replaceMessage(const ChatWindow::ChatMessage& message, const std::string& id, const boost::posix_time::ptime& time, const HighlightAction& highlight) SWIFTEN_OVERRIDE;

virtual void replaceWithAction(const ChatWindow::ChatMessage& message, const std::string& id, const boost::posix_time::ptime& time, const HighlightAction& highlight) SWIFTEN_OVERRIDE;

void setAckState(const std::string& id, ChatWindow::AckState state);

virtual std::string addFileTransfer(const std::string& senderName, bool senderIsSelf, const std::string& filename, const boost::uintmax_t sizeInBytes) SWIFTEN_OVERRIDE;

virtual void setFileTransferProgress(std::string, const int percentageDone) SWIFTEN_OVERRIDE;

virtual void setFileTransferStatus(std::string, const ChatWindow::FileTransferState state, const std::string& msg = "") SWIFTEN_OVERRIDE;

virtual void addMUCInvitation(const std::string& senderName, const JID& jid, const std::string& reason, const std::string& password, bool direct, bool isImpromptu, bool isContinuation) SWIFTEN_OVERRIDE;

virtual std::string addWhiteboardRequest(const QString& contact, bool senderIsSelf) SWIFTEN_OVERRIDE;

virtual void setWhiteboardSessionStatus(const std::string& id, const ChatWindow::WhiteboardSessionState state) SWIFTEN_OVERRIDE;

virtual void setMessageReceiptState(const std::string& id, ChatWindow::ReceiptState state) SWIFTEN_OVERRIDE;

virtual void showEmoticons(bool show) SWIFTEN_OVERRIDE;

void addMessageTop(boost::shared_ptr<ChatSnippet> snippet);

void addMessageBottom(boost::shared_ptr<ChatSnippet> snippet);

int getSnippetPositionByDate(const QDate& date); // FIXME : This probably shouldn't have been public

void addLastSeenLine();

private: // previously public, now private

void replaceLastMessage(const QString& newMessage, const QString& note);

void replaceMessage(const QString& newMessage, const QString& id, const QDateTime& time);

void rememberScrolledToBottom();

void setAckXML(const QString& id, const QString& xml);

void setReceiptXML(const QString& id, const QString& xml);

void displayReceiptInfo(const QString& id, bool showIt);

QString getLastSentMessage();

void addToJSEnvironment(const QString&, QObject*);

void setFileTransferProgress(QString id, const int percentageDone);

void setFileTransferStatus(QString id, const ChatWindow::FileTransferState state, const QString& msg);

void setWhiteboardSessionStatus(QString id, const ChatWindow::WhiteboardSessionState state);

void setMUCInvitationJoined(QString id);

signals:

void gotFocus();

void fontResized(int);

void logCleared();

void scrollRequested(int pos);

void scrollReachedTop();

void scrollReachedBottom();

public slots:

void copySelectionToClipboard();

void handleLinkClicked(const QUrl&);

void resetView();

void resetTopInsertPoint();

void increaseFontSize(int numSteps = 1);

void decreaseFontSize();

void resizeFont(int fontSizeSteps);

void scrollToBottom();

void handleKeyPressEvent(QKeyEvent* event);

private slots:

void handleViewLoadFinished(bool);

void handleFrameSizeChanged();

void handleClearRequested();

void handleScrollRequested(int dx, int dy, const QRect& rectToScroll);

void handleHTMLButtonClicked(QString id, QString arg1, QString arg2, QString arg3, QString arg4, QString arg5);

private:

enum PreviousMessageKind {

PreviosuMessageWasNone,

PreviousMessageWasMessage,

PreviousMessageWasSystem,

PreviousMessageWasPresence,

PreviousMessageWasFileTransfer,

PreviousMessageWasMUCInvite

};

std::string addMessage(

const QString& message,

const std::string& senderName,

bool senderIsSelf,

boost::shared_ptr<SecurityLabel> label,

const std::string& avatarPath,

const QString& style,

const boost::posix_time::ptime& time,

const HighlightAction& highlight,

ChatSnippet::Direction direction);

void replaceMessage(

const QString& message,

const std::string& id,

const boost::posix_time::ptime& time,

const QString& style,

const HighlightAction& highlight);

bool appendToPreviousCheck(PreviousMessageKind messageKind, const std::string& senderName, bool senderIsSelf);

static ChatSnippet::Direction getActualDirection(const ChatWindow::ChatMessage& message, ChatWindow::Direction direction);

QString chatMessageToHTML(const ChatWindow::ChatMessage& message);

QString getHighlightSpanStart(const HighlightAction& highlight);

static QString buildChatWindowButton(const QString& name, const QString& id, const QString& arg1 = QString(), const QString& arg2 = QString(), const QString& arg3 = QString(), const QString& arg4 = QString(), const QString& arg5 = QString());

private:

void headerEncode();

void messageEncode();

void addToDOM(boost::shared_ptr<ChatSnippet> snippet);

QWebElement snippetToDOM(boost::shared_ptr<ChatSnippet> snippet);

QtChatWindow* window_;

UIEventStream* eventStream_;

bool viewReady_;

bool isAtBottom_;

bool topMessageAdded_;

int scrollBarMaximum_;

QtWebView* webView_;

QWebPage* webPage_;

int fontSizeSteps_;

QtChatTheme* theme_;

QWebElement newInsertPoint_;

QWebElement topInsertPoint_;

QWebElement lineSeparator_;

QWebElement lastElement_;

QWebElement firstElement_;

QWebElement document_;

bool disableAutoScroll_;

QtChatWindowJSBridge* jsBridge;

PreviousMessageKind previousMessageKind_;

bool previousMessageWasSelf_;

bool showEmoticons_;

bool insertingLastLine_;

int idCounter_;

QString previousSenderName_;

std::map<QString, QString> descriptions_;

};

}