+#include "buffermodel.h"
+#include "bufferview.h"
+#include "chatline.h"
+#include "chatlinemodel.h"
+#include "chatview.h"
+#include "contextmenuactionprovider.h"
+#include "icon.h"
+#include "mainwin.h"
+#include "qtui.h"
+#include "qtuistyle.h"
+
+ChatItem::ChatItem(const QRectF &boundingRect, ChatLine *parent)
+ : _parent(parent),
+ _boundingRect(boundingRect),
+ _selectionMode(NoSelection),
+ _selectionStart(-1),
+ _cachedLayout(0)
+{
+}
+
+
+ChatItem::~ChatItem()
+{
+ delete _cachedLayout;
+}
+
+
+ChatLine *ChatItem::chatLine() const
+{
+ return _parent;
+}
+
+
+ChatScene *ChatItem::chatScene() const
+{
+ return chatLine()->chatScene();
+}
+
+
+ChatView *ChatItem::chatView() const
+{
+ return chatScene()->chatView();
+}
+
+
+const QAbstractItemModel *ChatItem::model() const
+{
+ return chatLine()->model();
+}
+
+
+int ChatItem::row() const
+{
+ return chatLine()->row();
+}
+
+
+QPointF ChatItem::mapToLine(const QPointF &p) const
+{
+ return p + pos();
+}
+
+
+QPointF ChatItem::mapFromLine(const QPointF &p) const
+{
+ return p - pos();
+}
+
+
+// relative to the ChatLine
+QPointF ChatItem::mapToScene(const QPointF &p) const
+{
+ return chatLine()->mapToScene(p /* + pos() */);
+}
+
+
+QPointF ChatItem::mapFromScene(const QPointF &p) const
+{
+ return chatLine()->mapFromScene(p) /* - pos() */;
+}
+
+
+QVariant ChatItem::data(int role) const
+{
+ QModelIndex index = model()->index(row(), column());
+ if (!index.isValid()) {
+ qWarning() << "ChatItem::data(): model index is invalid!" << index;
+ return QVariant();
+ }
+ return model()->data(index, role);
+}
+
+
+QTextLayout *ChatItem::layout() const
+{
+ if (_cachedLayout)
+ return _cachedLayout;
+
+ _cachedLayout = new QTextLayout;
+ initLayout(_cachedLayout);
+ chatView()->setHasCache(chatLine());
+ return _cachedLayout;
+}
+
+
+void ChatItem::clearCache()
+{
+ delete _cachedLayout;
+ _cachedLayout = 0;
+}
+
+
+void ChatItem::initLayoutHelper(QTextLayout *layout, QTextOption::WrapMode wrapMode, Qt::Alignment alignment) const
+{
+ Q_ASSERT(layout);
+
+ layout->setText(data(MessageModel::DisplayRole).toString());
+
+ QTextOption option;
+ option.setWrapMode(wrapMode);
+ option.setAlignment(alignment);
+ layout->setTextOption(option);
+
+ QList<QTextLayout::FormatRange> formatRanges
+ = QtUi::style()->toTextLayoutList(formatList(), layout->text().length(), data(ChatLineModel::MsgLabelRole).value<UiStyle::MessageLabel>());
+ layout->setAdditionalFormats(formatRanges);
+}
+
+
+void ChatItem::initLayout(QTextLayout *layout) const
+{
+ initLayoutHelper(layout, QTextOption::NoWrap);
+ doLayout(layout);
+}
+
+
+void ChatItem::doLayout(QTextLayout *layout) const
+{
+ layout->beginLayout();
+ QTextLine line = layout->createLine();
+ if (line.isValid()) {
+ line.setLineWidth(width());
+ line.setPosition(QPointF(0, 0));
+ }
+ layout->endLayout();
+}
+
+
+UiStyle::FormatList ChatItem::formatList() const
+{
+ return data(MessageModel::FormatRole).value<UiStyle::FormatList>();
+}
+
+
+qint16 ChatItem::posToCursor(const QPointF &posInLine) const
+{
+ QPointF pos = mapFromLine(posInLine);
+ if (pos.y() > height())
+ return data(MessageModel::DisplayRole).toString().length();
+ if (pos.y() < 0)
+ return 0;
+
+ for (int l = layout()->lineCount() - 1; l >= 0; l--) {
+ QTextLine line = layout()->lineAt(l);
+ if (pos.y() >= line.y()) {
+ return line.xToCursor(pos.x(), QTextLine::CursorOnCharacter);
+ }
+ }
+ return 0;
+}
+
+
+void ChatItem::paintBackground(QPainter *painter)
+{
+ QVariant bgBrush;
+ if (_selectionMode == FullSelection)
+ bgBrush = data(ChatLineModel::SelectedBackgroundRole);
+ else
+ bgBrush = data(ChatLineModel::BackgroundRole);
+ if (bgBrush.isValid())
+ painter->fillRect(boundingRect(), bgBrush.value<QBrush>());
+}
+
+
+// NOTE: This is not the most time-efficient implementation, but it saves space by not caching unnecessary data
+// This is a deliberate trade-off. (-> selectFmt creation, data() call)
+void ChatItem::paint(QPainter *painter, const QStyleOptionGraphicsItem *option, QWidget *widget)
+{
+ Q_UNUSED(option); Q_UNUSED(widget);
+ painter->save();
+ painter->setClipRect(boundingRect());
+ paintBackground(painter);
+
+ layout()->draw(painter, pos(), additionalFormats(), boundingRect());
+
+ // layout()->draw(painter, QPointF(0,0), formats, boundingRect());
+
+ // Debuging Stuff
+ // uncomment partially or all of the following stuff:
+ //
+ // 0) alternativ painter color for debug stuff
+// if(row() % 2)
+// painter->setPen(Qt::red);
+// else
+// painter->setPen(Qt::blue);
+// 1) draw wordwrap points in the first line
+// if(column() == 2) {
+// ChatLineModel::WrapList wrapList = data(ChatLineModel::WrapListRole).value<ChatLineModel::WrapList>();
+// foreach(ChatLineModel::Word word, wrapList) {
+// if(word.endX > width())
+// break;
+// painter->drawLine(word.endX, 0, word.endX, height());
+// }
+// }
+// 2) draw MsgId over the time column
+// if(column() == 0) {
+// QString msgIdString = QString::number(data(MessageModel::MsgIdRole).value<MsgId>().toLongLong());
+// QPointF bottomPoint = boundingRect().bottomLeft();
+// bottomPoint.ry() -= 2;
+// painter->drawText(bottomPoint, msgIdString);
+// }
+// 3) draw bounding rect
+// painter->drawRect(_boundingRect.adjusted(0, 0, -1, -1));
+
+ painter->restore();
+}
+
+
+void ChatItem::overlayFormat(UiStyle::FormatList &fmtList, quint16 start, quint16 end, UiStyle::FormatType overlayFmt) const
+{
+ for (size_t i = 0; i < fmtList.size(); i++) {
+ int fmtStart = fmtList.at(i).first;
+ int fmtEnd = (i < fmtList.size()-1 ? fmtList.at(i+1).first : data(MessageModel::DisplayRole).toString().length());
+
+ if (fmtEnd <= start)
+ continue;
+ if (fmtStart >= end)
+ break;
+
+ // split the format if necessary
+ if (fmtStart < start) {
+ fmtList.insert(fmtList.begin() + i, fmtList.at(i));
+ fmtList[++i].first = start;
+ }
+ if (end < fmtEnd) {
+ fmtList.insert(fmtList.begin() + i, fmtList.at(i));
+ fmtList[i+1].first = end;
+ }
+
+ fmtList[i].second.type |= overlayFmt;
+ }
+}
+
+
+QVector<QTextLayout::FormatRange> ChatItem::additionalFormats() const
+{
+ // Calculate formats to overlay (only) if there's a selection, and/or a hovered clickable
+ if (!hasSelection() && !hasActiveClickable()) {
+ return {};
+ }
+
+ using Label = UiStyle::MessageLabel;
+ using Format = UiStyle::Format;
+
+ Label itemLabel = data(ChatLineModel::MsgLabelRole).value<Label>();
+ const auto &fmtList = formatList();
+
+ struct LabelFormat {
+ quint16 offset;
+ Format format;
+ Label label;
+ };
+
+ // Transform formatList() into an extended list of LabelFormats
+ std::vector<LabelFormat> labelFmtList;
+ std::transform(fmtList.cbegin(), fmtList.cend(), std::back_inserter(labelFmtList), [itemLabel](const std::pair<quint16, Format> &f) {
+ return LabelFormat{f.first, f.second, itemLabel};
+ });
+ // Append dummy element to avoid special-casing handling the last real format
+ labelFmtList.push_back(LabelFormat{quint16(data(MessageModel::DisplayRole).toString().length()), Format(), itemLabel});
+
+ // Apply the given label to the given range in the format list, splitting formats as necessary
+ auto applyLabel = [&labelFmtList](quint16 start, quint16 end, Label label) {
+ size_t i = 0;
+
+ // Skip unaffected formats
+ for (; i < labelFmtList.size() - 1; ++i) {
+ if (labelFmtList[i+1].offset > start)
+ break;
+ }
+ // Range start doesn't align; split affected format and let the index point to the newly inserted copy
+ if (labelFmtList[i].offset < start) {
+ labelFmtList.insert(labelFmtList.begin() + i, labelFmtList[i]);
+ labelFmtList[++i].offset = start;
+ }
+
+ // Apply label to formats fully affected
+ for (; i < labelFmtList.size() - 1; ++i) {
+ if (labelFmtList[i+1].offset <= end) {
+ labelFmtList[i].label |= label;
+ continue;
+ }
+ // Last affected format, split if end of range doesn't align
+ if (labelFmtList[i+1].offset > end) {
+ labelFmtList.insert(labelFmtList.begin() + i, labelFmtList[i]);
+ labelFmtList[i].label |= label;
+ labelFmtList[i+1].offset = end;
+ }
+ break;
+ }
+ };
+
+ // Apply selection label
+ if (hasSelection()) {
+ quint16 start, end;
+ if (_selectionMode == FullSelection) {
+ start = 0;
+ end = data(MessageModel::DisplayRole).toString().length();
+ }
+ else {
+ start = qMin(_selectionStart, _selectionEnd);
+ end = qMax(_selectionStart, _selectionEnd);
+ }
+ applyLabel(start, end, Label::Selected);
+ }
+
+ // Apply hovered label
+ if (hasActiveClickable()) {
+ applyLabel(activeClickableRange().first, activeClickableRange().second, Label::Hovered);
+ }
+
+ // Add all formats that have an extra label to the additionalFormats list
+ QList<QTextLayout::FormatRange> additionalFormats;
+ for (size_t i = 0; i < labelFmtList.size() - 1; ++i) {
+ if (labelFmtList[i].label != itemLabel) {
+ additionalFormats << QtUi::style()->toTextLayoutList({std::make_pair(labelFmtList[i].offset, labelFmtList[i].format)},
+ labelFmtList[i+1].offset,
+ labelFmtList[i].label);
+ }
+ }
+
+ return additionalFormats.toVector();
+}
+
+
+bool ChatItem::hasSelection() const
+{
+ if (_selectionMode == NoSelection)
+ return false;
+ if (_selectionMode == FullSelection)
+ return true;
+ // partial
+ return _selectionStart != _selectionEnd;
+}
+
+
+QString ChatItem::selection() const
+{
+ if (_selectionMode == FullSelection)
+ return data(MessageModel::DisplayRole).toString();
+ if (_selectionMode == PartialSelection)
+ return data(MessageModel::DisplayRole).toString().mid(qMin(_selectionStart, _selectionEnd), qAbs(_selectionStart - _selectionEnd));
+ return QString();
+}
+
+
+void ChatItem::setSelection(SelectionMode mode, qint16 start, qint16 end)
+{
+ _selectionMode = mode;
+ _selectionStart = start;
+ _selectionEnd = end;
+ chatLine()->update();
+}
+
+
+void ChatItem::setFullSelection()
+{
+ if (_selectionMode != FullSelection) {
+ _selectionMode = FullSelection;
+ chatLine()->update();
+ }
+}
+
+
+void ChatItem::clearSelection()
+{
+ if (_selectionMode != NoSelection) {
+ _selectionMode = NoSelection;
+ chatLine()->update();
+ }
+}
+
+
+void ChatItem::continueSelecting(const QPointF &pos)
+{
+ _selectionMode = PartialSelection;
+ _selectionEnd = posToCursor(pos);
+ chatLine()->update();
+}
+
+
+bool ChatItem::isPosOverSelection(const QPointF &pos) const
+{
+ if (_selectionMode == FullSelection)
+ return true;
+ if (_selectionMode == PartialSelection) {
+ int cursor = posToCursor(pos);
+ return cursor >= qMin(_selectionStart, _selectionEnd) && cursor <= qMax(_selectionStart, _selectionEnd);
+ }
+ return false;
+}
+
+
+bool ChatItem::hasActiveClickable() const
+{
+ return false;
+}
+
+
+std::pair<quint16, quint16> ChatItem::activeClickableRange() const
+{
+ return {};
+}
+
+
+QList<QRectF> ChatItem::findWords(const QString &searchWord, Qt::CaseSensitivity caseSensitive)
+{
+ QList<QRectF> resultList;
+ const QAbstractItemModel *model_ = model();
+ if (!model_)
+ return resultList;
+
+ QString plainText = model_->data(model_->index(row(), column()), MessageModel::DisplayRole).toString();
+ QList<int> indexList;
+ int searchIdx = plainText.indexOf(searchWord, 0, caseSensitive);
+ while (searchIdx != -1) {
+ indexList << searchIdx;
+ searchIdx = plainText.indexOf(searchWord, searchIdx + 1, caseSensitive);
+ }
+
+ foreach(int idx, indexList) {
+ QTextLine line = layout()->lineForTextPosition(idx);
+ qreal x = line.cursorToX(idx);
+ qreal width = line.cursorToX(idx + searchWord.count()) - x;
+ qreal height = line.height();
+ qreal y = height * line.lineNumber();
+ resultList << QRectF(x, y, width, height);
+ }
+
+ return resultList;
+}
+
+
+void ChatItem::handleClick(const QPointF &pos, ChatScene::ClickMode clickMode)
+{
+ // single clicks are already handled by the scene (for clearing the selection)
+ if (clickMode == ChatScene::DragStartClick) {
+ chatScene()->setSelectingItem(this);
+ _selectionStart = _selectionEnd = posToCursor(pos);
+ _selectionMode = NoSelection; // will be set to PartialSelection by mouseMoveEvent
+ chatLine()->update();
+ }
+}
+
+
+void ChatItem::mouseMoveEvent(QGraphicsSceneMouseEvent *event)
+{
+ if (event->buttons() == Qt::LeftButton) {
+ if (boundingRect().contains(event->pos())) {
+ qint16 end = posToCursor(event->pos());
+ if (end != _selectionEnd) {
+ _selectionEnd = end;
+ _selectionMode = (_selectionStart != _selectionEnd ? PartialSelection : NoSelection);
+ chatLine()->update();
+ }
+ }
+ else {
+ setFullSelection();
+ chatScene()->startGlobalSelection(this, event->pos());
+ }
+ event->accept();
+ }
+ else {
+ event->ignore();
+ }
+}
+
+
+void ChatItem::mousePressEvent(QGraphicsSceneMouseEvent *event)
+{
+ if (event->buttons() == Qt::LeftButton)
+ event->accept();
+ else
+ event->ignore();
+}
+
+
+void ChatItem::mouseReleaseEvent(QGraphicsSceneMouseEvent *event)
+{
+ if (_selectionMode != NoSelection && event->button() == Qt::LeftButton) {
+ chatScene()->selectionToClipboard(QClipboard::Selection);
+ event->accept();
+ }
+ else
+ event->ignore();
+}
+
+
+void ChatItem::addActionsToMenu(QMenu *menu, const QPointF &pos)
+{
+ Q_UNUSED(pos);
+
+ GraphicalUi::contextMenuActionProvider()->addActions(menu, chatScene()->filter(), data(MessageModel::BufferIdRole).value<BufferId>());
+}
+
+
+// ************************************************************
+// SenderChatItem
+// ************************************************************
+
+void SenderChatItem::initLayout(QTextLayout *layout) const
+{
+ initLayoutHelper(layout, QTextOption::ManualWrap, Qt::AlignRight);
+ doLayout(layout);
+}
+
+
+void SenderChatItem::paint(QPainter *painter, const QStyleOptionGraphicsItem *option, QWidget *widget)
+{
+ Q_UNUSED(option); Q_UNUSED(widget);
+ painter->save();
+ painter->setClipRect(boundingRect());
+ paintBackground(painter);
+
+ qreal layoutWidth = layout()->minimumWidth();
+ qreal offset = 0;
+ if (chatScene()->senderCutoffMode() == ChatScene::CutoffLeft)
+ offset = qMin(width() - layoutWidth, (qreal)0);
+ else
+ offset = qMax(layoutWidth - width(), (qreal)0);
+
+ if (layoutWidth > width()) {
+ // Draw a nice gradient for longer items
+ // Qt's text drawing with a gradient brush sucks, so we use compositing instead
+ QPixmap pixmap(layout()->boundingRect().toRect().size());
+ pixmap.fill(Qt::transparent);
+
+ QPainter pixPainter(&pixmap);
+ layout()->draw(&pixPainter, QPointF(qMax(offset, (qreal)0), 0), additionalFormats());
+
+ // Create alpha channel mask
+ QLinearGradient gradient;
+ if (offset < 0) {
+ gradient.setStart(0, 0);
+ gradient.setFinalStop(12, 0);
+ gradient.setColorAt(0, Qt::transparent);
+ gradient.setColorAt(1, Qt::white);
+ }
+ else {
+ gradient.setStart(width()-10, 0);
+ gradient.setFinalStop(width(), 0);
+ gradient.setColorAt(0, Qt::white);
+ gradient.setColorAt(1, Qt::transparent);
+ }
+ pixPainter.setCompositionMode(QPainter::CompositionMode_DestinationIn); // gradient's alpha gets applied to the pixmap
+ pixPainter.fillRect(pixmap.rect(), gradient);
+ painter->drawPixmap(pos(), pixmap);
+ }
+ else {
+ layout()->draw(painter, pos(), additionalFormats(), boundingRect());
+ }
+ painter->restore();
+}
+
+
+void SenderChatItem::handleClick(const QPointF &pos, ChatScene::ClickMode clickMode)
+{
+ if (clickMode == ChatScene::DoubleClick) {
+ BufferInfo curBufInfo = Client::networkModel()->bufferInfo(data(MessageModel::BufferIdRole).value<BufferId>());
+ QString nick = data(MessageModel::EditRole).toString();
+ // check if the nick is a valid ircUser
+ if (!nick.isEmpty() && Client::network(curBufInfo.networkId())->ircUser(nick))
+ Client::bufferModel()->switchToOrStartQuery(curBufInfo.networkId(), nick);
+ }
+ else
+ ChatItem::handleClick(pos, clickMode);
+}
+
+
+// ************************************************************
+// ContentsChatItem
+// ************************************************************
+
+ContentsChatItem::ActionProxy ContentsChatItem::_actionProxy;
+
+ContentsChatItem::ContentsChatItem(const QPointF &pos, const qreal &width, ChatLine *parent)
+ : ChatItem(QRectF(pos, QSizeF(width, 0)), parent),
+ _data(0)
+{
+ setPos(pos);
+ setGeometryByWidth(width);
+}
+
+
+QFontMetricsF *ContentsChatItem::fontMetrics() const
+{
+ return QtUi::style()->fontMetrics(data(ChatLineModel::FormatRole).value<UiStyle::FormatList>().at(0).second.type, UiStyle::MessageLabel::None);
+}