+ * Copyright (c) 2010-2011 Kevin Smith

+#include "Swift/Controllers/Roster/RosterItem.h"

+#include "Swift/Controllers/Roster/ContactRosterItem.h"

+#include "Roster/QtOccupantListWidget.h"

+#include "QtSettingsProvider.h"

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

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

+#include "QtFileTransferJSBridge.h"

+#include <boost/cstdint.hpp>

+#include <boost/format.hpp>

+#include <boost/lexical_cast.hpp>

+

+#include <QLabel>

+#include <QInputDialog>

+#include <QPushButton>

+#include <QFileDialog>

+#include <QMenu>

+

+#include <QDebug>

+QtChatWindow::QtChatWindow(const QString &contact, QtChatTheme* theme, UIEventStream* eventStream) : QtTabbable(), contact_(contact), previousMessageWasSelf_(false), previousMessageWasSystem_(false), previousMessageWasPresence_(false), previousMessageWasFileTransfer_(false), eventStream_(eventStream) {

+ isCorrection_ = false;

+ correctionEnabled_ = Maybe;

+ QtSettingsProvider settings;

+

+#ifdef SWIFT_EXPERIMENTAL_FT

+ setAcceptDrops(true);

+#endif

+

+ alertStyleSheet_ = "background: rgb(255, 255, 153); color: black";

+ alertWidget_ = new QWidget(this);

+ QHBoxLayout* alertLayout = new QHBoxLayout(alertWidget_);

+ layout->addWidget(alertWidget_);

+ alertLabel_ = new QLabel(this);

+ alertLayout->addWidget(alertLabel_);

+ alertButton_ = new QPushButton(this);

+ connect (alertButton_, SIGNAL(clicked()), this, SLOT(handleAlertButtonClicked()));

+ alertLayout->addWidget(alertButton_);

+ QPalette palette = alertWidget_->palette();

+ palette.setColor(QPalette::Window, QColor(Qt::yellow));

+ palette.setColor(QPalette::WindowText, QColor(Qt::black));

+ alertWidget_->setStyleSheet(alertStyleSheet_);

+ alertLabel_->setStyleSheet(alertStyleSheet_);

+ alertWidget_->hide();

+

+ QBoxLayout* subjectLayout = new QBoxLayout(QBoxLayout::LeftToRight);

+ subject_ = new QLineEdit(this);

+ subjectLayout->addWidget(subject_);

+ setSubject("");

+ subject_->setReadOnly(true);

+

+ actionButton_ = new QPushButton(tr("Actions"), this);

+ connect(actionButton_, SIGNAL(clicked()), this, SLOT(handleActionButtonClicked()));

+ subjectLayout->addWidget(actionButton_);

+

+ subject_->hide();

+ actionButton_->hide();

+

+ layout->addLayout(subjectLayout);

+

+ logRosterSplitter_ = new QSplitter(this);

+ logRosterSplitter_->setAutoFillBackground(true);

+ layout->addWidget(logRosterSplitter_);

+ logRosterSplitter_->addWidget(messageLog_);

+ treeWidget_ = new QtOccupantListWidget(eventStream_, this);

+ logRosterSplitter_->addWidget(treeWidget_);

+ logRosterSplitter_->setSizePolicy(QSizePolicy::Expanding, QSizePolicy::Expanding);

+ connect(logRosterSplitter_, SIGNAL(splitterMoved(int, int)), this, SLOT(handleSplitterMoved(int, int)));

+ QHBoxLayout* inputBarLayout = new QHBoxLayout();

+ inputBarLayout->setContentsMargins(0,0,0,0);

+ inputBarLayout->setSpacing(2);

+ inputBarLayout->addWidget(input_);

+ correctingLabel_ = new QLabel(tr("Correcting"), this);

+ inputBarLayout->addWidget(correctingLabel_);

+ correctingLabel_->hide();

+ layout->addLayout(inputBarLayout);

+

+ logRosterSplitter_->setFocusProxy(input_);

+ connect(messageLog_, SIGNAL(fontResized(int)), this, SIGNAL(fontResized(int)));

+

+ treeWidget_->onSomethingSelectedChanged.connect(boost::bind(&QtChatWindow::handleOccupantSelectionChanged, this, _1));

+ treeWidget_->onOccupantActionSelected.connect(boost::bind(boost::ref(onOccupantActionSelected), _1, _2));

+

+ fileTransferJS = new QtFileTransferJSBridge();

+ messageLog_->addToJSEnvironment("filetransfer", fileTransferJS);

+ connect(fileTransferJS, SIGNAL(setDescription(QString)), this, SLOT(handleFileTransferSetDescription(QString)));

+ connect(fileTransferJS, SIGNAL(sendRequest(QString)), this, SLOT(handleFileTransferStart(QString)));

+ connect(fileTransferJS, SIGNAL(acceptRequest(QString, QString)), this, SLOT(handleFileTransferAccept(QString, QString)));

+ connect(fileTransferJS, SIGNAL(cancel(QString)), this, SLOT(handleFileTransferCancel(QString)));

+ delete fileTransferJS;

+ if (mucConfigurationWindow) {

+ delete mucConfigurationWindow.data();

+ }

+}

+

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

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

+}

+

+void QtChatWindow::handleFontResized(int fontSizeSteps) {

+ messageLog_->resizeFont(fontSizeSteps);

+}

+

+void QtChatWindow::handleAlertButtonClicked() {

+ onAlertButtonClicked();

+}

+

+void QtChatWindow::setAlert(const std::string& alertText, const std::string& buttonText) {

+ alertLabel_->setText(alertText.c_str());

+ if (buttonText.empty()) {

+ alertButton_->hide();

+ } else {

+ alertButton_->setText(buttonText.c_str());

+ alertButton_->show();

+ }

+ alertWidget_->show();

+}

+

+void QtChatWindow::cancelAlert() {

+ alertWidget_->hide();

+ } else if ((key == Qt::Key_Up) && input_->toPlainText().isEmpty() && !(lastSentMessage_.isEmpty())) {

+ beginCorrection();

+ } else if (key == Qt::Key_Down && isCorrection_ && input_->textCursor().atBlockEnd()) {

+ cancelCorrection();

+ } else if (key == Qt::Key_Down || key == Qt::Key_Up) {

+ /* Drop */

+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_);

+}

+

+void QtChatWindow::cancelCorrection() {

+ cancelAlert();

+ QTextCursor cursor = input_->textCursor();

+ cursor.select(QTextCursor::Document);

+ cursor.removeSelectedText();

+ isCorrection_ = false;

+ correctingLabel_->hide();

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

+}

+

+QByteArray QtChatWindow::getSplitterState() {

+ return logRosterSplitter_->saveState();

+}

+

+void QtChatWindow::handleChangeSplitterState(QByteArray state) {

+ logRosterSplitter_->restoreState(state);

+}

+

+void QtChatWindow::handleSplitterMoved(int, int) {

+ emit splitterMoved();

+}

+

+ treeWidget_->setRosterModel(roster);

+void QtChatWindow::setCorrectionEnabled(Tristate enabled) {

+ correctionEnabled_ = enabled;

+}

+

+ setAcceptDrops(false);

+ subject_->show();

+ actionButton_->show();

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

+ lastLineTracker_.setHasFocus(true);

+ else {

+ lastLineTracker_.setHasFocus(false);

+ }

+ if (!enabled && mucConfigurationWindow) {

+ delete mucConfigurationWindow.data();

+ }

+ }

+ bool appendToPrevious = !previousMessageWasFileTransfer_ && !previousMessageWasSystem_ && !previousMessageWasPresence_ && ((senderIsSelf && previousMessageWasSelf_) || (!senderIsSelf && !previousMessageWasSelf_ && previousSenderName_ == P2QSTRING(senderName)));

+ if (lastLineTracker_.getShouldMoveLastLine()) {

+ /* should this be queued? */

+ messageLog_->addLastSeenLine();

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

+ appendToPrevious = false;

+ }

+ previousMessageWasFileTransfer_ = false;

+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] : "") );

+}

+

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

+ qDebug() << "addFileTransfer";

+ std::string ft_id = id_.generateID();

+

+ std::string htmlString;

+ if (senderIsSelf) {

+ // outgoing

+ htmlString = "Send file: " + filename + " ( " + formatSize(sizeInBytes) + ") </br>" +

+ "<div id='" + ft_id + "'>" +

+ "<input id='discard' type='submit' value='Cancel' onclick='filetransfer.cancel(\"" + ft_id + "\");' />" +

+ "<input id='description' type='submit' value='Set Description' onclick='filetransfer.setDescription(\"" + ft_id + "\");' />" +

+ "<input id='send' type='submit' value='Send' onclick='filetransfer.sendRequest(\"" + ft_id + "\");' />" +

+ "</div>";

+ } else {

+ // incoming

+ htmlString = "Receiving file: " + filename + " ( " + formatSize(sizeInBytes) + ") </br>" +

+ "<div id='" + ft_id + "'>" +

+ "<input id='discard' type='submit' value='Cancel' onclick='filetransfer.cancel(\"" + ft_id + "\");' />" +

+ "<input id='accept' type='submit' value='Accept' onclick='filetransfer.acceptRequest(\"" + ft_id + "\", \"" + filename + "\");' />" +

+ "</div>";

+ }

+

+ //addMessage(message, senderName, senderIsSelf, boost::shared_ptr<SecurityLabel>(), "", boost::posix_time::second_clock::local_time());

+

+ bool appendToPrevious = !previousMessageWasFileTransfer_ && !previousMessageWasSystem_ && !previousMessageWasPresence_ && ((senderIsSelf && previousMessageWasSelf_) || (!senderIsSelf && !previousMessageWasSelf_ && previousSenderName_ == P2QSTRING(senderName)));

+ 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 = id_.generateID();

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

+

+

+ return ft_id;

+}

+

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

+ messageLog_->setFileTransferProgress(QString::fromStdString(id), percentageDone);

+}

+

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

+ messageLog_->setFileTransferStatus(QString::fromStdString(id), state, QString::fromStdString(msg));

+}

+

+void QtChatWindow::handleFileTransferCancel(QString id) {

+ qDebug() << "QtChatWindow::handleFileTransferCancel(" << id << ")";

+ onFileTransferCancel(Q2PSTRING(id));

+}

+

+void QtChatWindow::handleFileTransferSetDescription(QString id) {

+ bool ok = false;

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

+ tr("Description:"), QLineEdit::Normal, "", &ok);

+ if (ok) {

+ descriptions[id] = text;

+ }

+}

+

+void QtChatWindow::handleFileTransferStart(QString id) {

+ qDebug() << "QtChatWindow::handleFileTransferStart(" << id << ")";

+

+ QString text = descriptions.find(id) == descriptions.end() ? QString() : descriptions[id];

+ onFileTransferStart(Q2PSTRING(id), Q2PSTRING(text));

+}

+

+void QtChatWindow::handleFileTransferAccept(QString id, QString filename) {

+ qDebug() << "QtChatWindow::handleFileTransferAccept(" << id << ", " << filename << ")";

+

+ QString path = QFileDialog::getSaveFileName(this, tr("Save File"), filename);

+ if (!path.isEmpty()) {

+ onFileTransferAccept(Q2PSTRING(id), Q2PSTRING(path));

+ }

+}

+

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

+ previousMessageWasFileTransfer_ = false;

+ previousMessageWasFileTransfer_ = false;

+}

+

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

+ if (!id.empty()) {

+ QString messageHTML(Qt::escape(P2QSTRING(message)));

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

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

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

+ }

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

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

+ cancelCorrection();

+ if (parentWidget() == NULL) {

+ QWidget::show();

+ }

+ emit geometryChanged();

+}

+

+void QtChatWindow::dragEnterEvent(QDragEnterEvent *event) {

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

+ // TODO: check whether contact actually supports file transfer

+ event->acceptProposedAction();

+ }

+}

+

+void QtChatWindow::dropEvent(QDropEvent *event) {

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

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

+ } else {

+ addSystemMessage("Sending of multiple files at once isn't supported at this time.");

+ }

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

+}

+

+void QtChatWindow::handleActionButtonClicked() {

+ QMenu contextMenu;

+ QAction* changeSubject = contextMenu.addAction(tr("Change subject"));

+ QAction* configure = contextMenu.addAction(tr("Configure room"));

+ QAction* destroy = contextMenu.addAction(tr("Destroy room"));

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

+ 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 == destroy) {

+ onDestroyRequest();

+ }

+}

+

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

+ if (mucConfigurationWindow) {

+ delete mucConfigurationWindow.data();

+ }

+ mucConfigurationWindow = new QtMUCConfigurationWindow(form);

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

+}

+