-#include "QtChatView.h"

-#include "MessageSnippet.h"

-#include "SystemMessageSnippet.h"

-#include <Swift/QtUI/QtUISettingConstants.h>

-#include <Swiften/StringCodecs/Base64.h>

-#include "QtChatWindowJSBridge.h"

-#include <boost/format.hpp>

-#include <QFileDialog>

-const QString QtChatWindow::ButtonFileTransferCancel = QString("filetransfer-cancel");

-const QString QtChatWindow::ButtonFileTransferSetDescription = QString("filetransfer-setdescription");

-const QString QtChatWindow::ButtonFileTransferSendRequest = QString("filetransfer-sendrequest");

-const QString QtChatWindow::ButtonFileTransferAcceptRequest = QString("filetransfer-acceptrequest");

-const QString QtChatWindow::ButtonMUCInvite = QString("mucinvite");

-

-QtChatWindow::QtChatWindow(const QString &contact, QtChatTheme* theme, UIEventStream* eventStream, SettingsProvider* settings, QMap<QString, QString> emoticons) : QtTabbable(), contact_(contact), previousMessageWasSelf_(false), previousMessageKind_(PreviosuMessageWasNone), eventStream_(eventStream), emoticons_(emoticons) {

- idCounter_ = 0;

- theme_ = theme;

- showEmoticons_ = true;

- messageLog_ = new QtChatView(theme, this);

- treeWidget_ = new QtOccupantListWidget(eventStream_, settings_, this);

-

- jsBridge = new QtChatWindowJSBridge();

- messageLog_->addToJSEnvironment("chatwindow", jsBridge);

- connect(jsBridge, SIGNAL(buttonClicked(QString,QString,QString,QString)), this, SLOT(handleHTMLButtonClicked(QString,QString,QString,QString)));

-

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

- showEmoticons_ = settings_->getSetting(QtUISettingConstants::SHOW_EMOTICONS);

-

- delete jsBridge;

-void QtChatWindow::handleSettingChanged(const std::string& setting) {

- if (setting == QtUISettingConstants::SHOW_EMOTICONS.getKey()) {

- showEmoticons_ = settings_->getSetting(QtUISettingConstants::SHOW_EMOTICONS);

- messageLog_->showEmoticons(showEmoticons_);

- }

-}

-

-void QtChatWindow::handleOccupantSelectionChanged(RosterItem* item) {

- onOccupantSelectionChanged(dynamic_cast<ContactRosterItem*>(item));

-bool QtChatWindow::appendToPreviousCheck(QtChatWindow::PreviousMessageKind messageKind, const std::string& senderName, bool senderIsSelf) const {

- return previousMessageKind_ == messageKind && ((senderIsSelf && previousMessageWasSelf_) || (!senderIsSelf && !previousMessageWasSelf_&& previousSenderName_ == P2QSTRING(senderName)));

- lastLineTracker_.setHasFocus(true);

- lastLineTracker_.setHasFocus(false);

- return addMessage(linkimoticonify(P2QSTRING(message)), senderName, senderIsSelf, label, avatarPath, "", time);

-}

-

-QString QtChatWindow::linkimoticonify(const QString& message) const {

- QString messageHTML(message);

- messageHTML = Qt::escape(messageHTML);

- QMapIterator<QString, QString> it(emoticons_);

- QString textStyle = showEmoticons_ ? "style='display:none'" : "";

- QString imageStyle = showEmoticons_ ? "" : "style='display:none'";

- if (messageHTML.length() < 500) {

- while (it.hasNext()) {

- it.next();

- messageHTML.replace(it.key(), "<span class='swift_emoticon_image' " + imageStyle + "><img src='" + it.value() + "'/></span><span class='swift_emoticon_text' " + textStyle + ">"+it.key() + "</span>");

- }

- messageHTML = P2QSTRING(Linkify::linkify(Q2PSTRING(messageHTML)));

- }

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

- return messageHTML;

-std::string QtChatWindow::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) {

- 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(Qt::escape(P2QSTRING(label->getForegroundColor()))).arg(Qt::escape(P2QSTRING(label->getBackgroundColor())));

- htmlString += QString("%1</span> ").arg(Qt::escape(P2QSTRING(label->getDisplayMarking())));

- }

- QString messageHTML(message);

- QString styleSpanStart = style == "" ? "" : "<span style=\"" + style + "\">";

- QString styleSpanEnd = style == "" ? "" : "</span>";

- htmlString += "<span class='swift_inner_message'>" + styleSpanStart + messageHTML + styleSpanEnd + "</span>" ;

-

- bool appendToPrevious = appendToPreviousCheck(PreviousMessageWasMessage, senderName, senderIsSelf);

- if (lastLineTracker_.getShouldMoveLastLine()) {

- /* should this be queued? */

- messageLog_->addLastSeenLine();

- /* if the line is added we should break the snippet */

- appendToPrevious = false;

- }

- QString qAvatarPath = scaledAvatarPath.isEmpty() ? "qrc:/icons/avatar.png" : QUrl::fromLocalFile(scaledAvatarPath).toEncoded();

- std::string id = "id" + boost::lexical_cast<std::string>(idCounter_++);

- messageLog_->addMessage(boost::shared_ptr<ChatSnippet>(new MessageSnippet(htmlString, Qt::escape(P2QSTRING(senderName)), B2QDATE(time), qAvatarPath, senderIsSelf, appendToPrevious, theme_, P2QSTRING(id))));

-

- previousMessageWasSelf_ = senderIsSelf;

- previousSenderName_ = P2QSTRING(senderName);

- previousMessageKind_ = PreviousMessageWasMessage;

- return id;

- 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.") + "'/>";

- messageLog_->displayReceiptInfo(P2QSTRING(id), false);

- break;

- case ChatWindow::Received:

- xml = "";

- messageLog_->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;

- }

- messageLog_->setAckXML(P2QSTRING(id), xml);

- 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;

- }

- messageLog_->setReceiptXML(P2QSTRING(id), xml);

- return addMessage(" *" + linkimoticonify(P2QSTRING(message)) + "*", senderName, senderIsSelf, label, avatarPath, "font-style:italic ", time);

-}

-

-std::string formatSize(const boost::uintmax_t bytes) {

- static const char *siPrefix[] = {"k", "M", "G", "T", "P", "E", "Z", "Y", NULL};

- int power = 0;

- double engBytes = bytes;

- while (engBytes >= 1000) {

- ++power;

- engBytes = engBytes / 1000.0;

- }

- return str( boost::format("%.1lf %sB") % engBytes % (power > 0 ? siPrefix[power-1] : "") );

-}

-

-QString encodeButtonArgument(const QString& str) {

- return Qt::escape(P2QSTRING(Base64::encode(createByteArray(Q2PSTRING(str)))));

-}

-

-QString decodeButtonArgument(const QString& str) {

- return P2QSTRING(byteArrayToString(Base64::decode(Q2PSTRING(str))));

-}

-

-QString QtChatWindow::buildChatWindowButton(const QString& name, const QString& id, const QString& arg1, const QString& arg2, const QString& arg3) {

- 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\");' />").arg(name).arg(id).arg(encodeButtonArgument(arg1)).arg(encodeButtonArgument(arg2)).arg(encodeButtonArgument(arg3));

- return html;

- SWIFT_LOG(debug) << "addFileTransfer" << std::endl;

- QString ft_id = QString("ft%1").arg(P2QSTRING(boost::lexical_cast<std::string>(idCounter_++)));

-

- QString htmlString;

- QString formattedFileSize = P2QSTRING(formatSize(sizeInBytes));

- if (senderIsSelf) {

- // outgoing

- htmlString = tr("Send file") + ": " + 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

- htmlString = tr("Receiving file") + ": " + 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);

- if (lastLineTracker_.getShouldMoveLastLine()) {

- /* should this be queued? */

- messageLog_->addLastSeenLine();

- /* if the line is added we should break the snippet */

- appendToPrevious = false;

- }

- QString qAvatarPath = "qrc:/icons/avatar.png";

- std::string id = "ftmessage" + boost::lexical_cast<std::string>(idCounter_++);

- messageLog_->addMessage(boost::shared_ptr<ChatSnippet>(new MessageSnippet(htmlString, Qt::escape(P2QSTRING(senderName)), B2QDATE(boost::posix_time::second_clock::local_time()), qAvatarPath, senderIsSelf, appendToPrevious, theme_, P2QSTRING(id))));

-

- previousMessageWasSelf_ = senderIsSelf;

- previousSenderName_ = P2QSTRING(senderName);

- previousMessageKind_ = PreviousMessageWasFileTransfer;

- return Q2PSTRING(ft_id);

-void QtChatWindow::handleHTMLButtonClicked(QString id, QString encodedArgument1, QString encodedArgument2, QString encodedArgument3) {

- QString arg1 = decodeButtonArgument(encodedArgument1);

- QString arg2 = decodeButtonArgument(encodedArgument2);

- QString arg3 = decodeButtonArgument(encodedArgument3);

-

- if (id.startsWith(ButtonFileTransferCancel)) {

- QString ft_id = arg1;

- 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];

- 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()) {

- onFileTransferAccept(Q2PSTRING(ft_id), Q2PSTRING(path));

- }

- }

- else if (id.startsWith(ButtonMUCInvite)) {

- QString roomJID = arg1;

- QString password = arg2;

- QString elementID = arg3;

-

- eventStream_->send(boost::make_shared<JoinMUCUIEvent>(Q2PSTRING(roomJID), Q2PSTRING(password)));

- messageLog_->setMUCInvitationJoined(elementID);

- }

- else {

- SWIFT_LOG(debug) << "Unknown HTML button! ( " << Q2PSTRING(id) << " )" << std::endl;

- }

-}

- QString errorMessageHTML(Qt::escape(P2QSTRING(errorMessage)));

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

- messageLog_->addMessage(boost::shared_ptr<ChatSnippet>(new SystemMessageSnippet("<span class=\"error\">" + errorMessageHTML + "</span>", QDateTime::currentDateTime(), false, theme_)));

-

- previousMessageWasSelf_ = false;

- previousMessageKind_ = PreviousMessageWasSystem;

-void QtChatWindow::addSystemMessage(const std::string& message) {

-

- QString messageHTML(P2QSTRING(message));

- messageHTML = linkimoticonify(messageHTML);

- messageLog_->addMessage(boost::shared_ptr<ChatSnippet>(new SystemMessageSnippet(messageHTML, QDateTime::currentDateTime(), false, theme_)));

-

- previousMessageKind_ = PreviousMessageWasSystem;

- replaceMessage(" *" + linkimoticonify(P2QSTRING(message)) + "*", id, time, "font-style:italic ");

- replaceMessage(linkimoticonify(P2QSTRING(message)), id, time, "");

-void QtChatWindow::replaceMessage(const QString& message, const std::string& id, const boost::posix_time::ptime& time, const QString& style) {

- if (!id.empty()) {

- if (isWidgetSelected()) {

- onAllMessagesRead();

- }

-

- QString messageHTML(message);

-

- QString styleSpanStart = style == "" ? "" : "<span style=\"" + style + "\">";

- QString styleSpanEnd = style == "" ? "" : "</span>";

- messageHTML = styleSpanStart + messageHTML + styleSpanEnd;

-

- messageLog_->replaceMessage(messageHTML, P2QSTRING(id), B2QDATE(time));

- }

- else {

- std::cerr << "Trying to replace a message with no id";

-

- QString messageHTML(P2QSTRING(message));

- messageHTML = linkimoticonify(messageHTML);

- messageLog_->addMessage(boost::shared_ptr<ChatSnippet>(new SystemMessageSnippet(messageHTML, QDateTime::currentDateTime(), false, theme_)));

-

- previousMessageKind_ = PreviousMessageWasPresence;

-

- messageLog_->replaceLastMessage(linkimoticonify(P2QSTRING(message)));

- QString htmlString = QObject::tr("You've been invited to enter the %1 room.").arg(P2QSTRING(jid.toString())) + " <br/>";

- if (!reason.empty()) {

- htmlString += QObject::tr("Reason: %1").arg(P2QSTRING(reason)) + "<br/>";

- }

- if (!direct) {

- htmlString += QObject::tr("This person may not have really sent this invitation!") + "<br/>";

- }

-

- QString id = QString(ButtonMUCInvite + "%1").arg(P2QSTRING(boost::lexical_cast<std::string>(idCounter_++)));

-

- htmlString += "<div id='" + id + "'>" +

- buildChatWindowButton(tr("Accept Invite"), ButtonMUCInvite, Qt::escape(P2QSTRING(jid.toString())), Qt::escape(P2QSTRING(password)), id) +

- "</div>";

-

- bool appendToPrevious = appendToPreviousCheck(PreviousMessageWasMUCInvite, senderName, false);

- if (lastLineTracker_.getShouldMoveLastLine()) {

- /* should this be queued? */

- messageLog_->addLastSeenLine();

- /* if the line is added we should break the snippet */

- appendToPrevious = false;

- }

- QString qAvatarPath = "qrc:/icons/avatar.png";

-

- messageLog_->addMessage(boost::shared_ptr<ChatSnippet>(new MessageSnippet(htmlString, Qt::escape(P2QSTRING(senderName)), B2QDATE(boost::posix_time::second_clock::local_time()), qAvatarPath, false, appendToPrevious, theme_, id)));

- previousMessageWasSelf_ = false;

- previousSenderName_ = P2QSTRING(senderName);

- previousMessageKind_ = PreviousMessageWasMUCInvite;