Sanitize topic message
[quassel.git] / src / qtui / chatlinemodelitem.cpp
index b657f39..480c1b5 100644 (file)
@@ -1,5 +1,5 @@
 /***************************************************************************
- *   Copyright (C) 2005-08 by the Quassel Project                          *
+ *   Copyright (C) 2005-2013 by the Quassel Project                        *
  *   devel@quassel-irc.org                                                 *
  *                                                                         *
  *   This program is free software; you can redistribute it and/or modify  *
@@ -15,7 +15,7 @@
  *   You should have received a copy of the GNU General Public License     *
  *   along with this program; if not, write to the                         *
  *   Free Software Foundation, Inc.,                                       *
- *   59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.             *
+ *   51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.         *
  ***************************************************************************/
 
 #include <QFontMetrics>
 #include "chatlinemodelitem.h"
 #include "chatlinemodel.h"
 #include "qtui.h"
-#include "uistyle.h"
+#include "qtuistyle.h"
 
-ChatLineModelItem::ChatLineModelItem(const Message &msg) : MessageModelItem(msg) {
-  QtUiStyle::StyledMessage m = QtUi::style()->styleMessage(msg);
+// This Struct is taken from Harfbuzz. We use it only to calc it's size.
+// we use a shared memory region so we do not have to malloc a buffer area for every line
+typedef struct {
+    /*HB_LineBreakType*/ unsigned lineBreakType  : 2;
+    /*HB_Bool*/ unsigned whiteSpace              : 1;     /* A unicode whitespace character, except NBSP, ZWNBSP */
+    /*HB_Bool*/ unsigned charStop                : 1;     /* Valid cursor position (for left/right arrow) */
+    /*HB_Bool*/ unsigned wordBoundary            : 1;
+    /*HB_Bool*/ unsigned sentenceBoundary        : 1;
+    unsigned unused                  : 2;
+} HB_CharAttributes_Dummy;
 
-  _timestamp.plainText = m.timestamp.plainText;
-  _sender.plainText = m.sender.plainText;
-  _contents.plainText = m.contents.plainText;
+unsigned char *ChatLineModelItem::TextBoundaryFinderBuffer = (unsigned char *)malloc(512 * sizeof(HB_CharAttributes_Dummy));
+int ChatLineModelItem::TextBoundaryFinderBufferSize = 512 * (sizeof(HB_CharAttributes_Dummy) / sizeof(unsigned char));
+
+// ****************************************
+// the actual ChatLineModelItem
+// ****************************************
+ChatLineModelItem::ChatLineModelItem(const Message &msg)
+    : MessageModelItem(),
+    _styledMsg(msg)
+{
+    if (!msg.sender().contains('!'))
+        _styledMsg.setFlags(msg.flags() |= Message::ServerMsg);
+}
 
-  _timestamp.formatList = m.timestamp.formatList;
-  _sender.formatList = m.sender.formatList;
-  _contents.formatList = m.contents.formatList;
 
-  computeWrapList();
+bool ChatLineModelItem::setData(int column, const QVariant &value, int role)
+{
+    switch (role) {
+    case MessageModel::FlagsRole:
+        _styledMsg.setFlags((Message::Flags)value.toUInt());
+        return true;
+    default:
+        return MessageModelItem::setData(column, value, role);
+    }
 }
 
 
-QVariant ChatLineModelItem::data(int column, int role) const {
-  const ChatLinePart *part;
+QVariant ChatLineModelItem::data(int column, int role) const
+{
+    if (role == ChatLineModel::MsgLabelRole)
+        return messageLabel();
 
-  switch(column) {
-    case ChatLineModel::TimestampColumn: part = &_timestamp; break;
-    case ChatLineModel::SenderColumn:    part = &_sender; break;
-    case ChatLineModel::ContentsColumn:      part = &_contents; break;
-    default: return MessageModelItem::data(column, role);
-  }
+    QVariant variant;
+    MessageModel::ColumnType col = (MessageModel::ColumnType)column;
+    switch (col) {
+    case ChatLineModel::TimestampColumn:
+        variant = timestampData(role);
+        break;
+    case ChatLineModel::SenderColumn:
+        variant = senderData(role);
+        break;
+    case ChatLineModel::ContentsColumn:
+        variant = contentsData(role);
+        break;
+    default:
+        break;
+    }
+    if (!variant.isValid())
+        return MessageModelItem::data(column, role);
+    return variant;
+}
 
-  switch(role) {
+
+QVariant ChatLineModelItem::timestampData(int role) const
+{
+    switch (role) {
     case ChatLineModel::DisplayRole:
-      return part->plainText;
+        return _styledMsg.decoratedTimestamp();
+    case ChatLineModel::EditRole:
+        return _styledMsg.timestamp();
+    case ChatLineModel::BackgroundRole:
+        return backgroundBrush(UiStyle::Timestamp);
+    case ChatLineModel::SelectedBackgroundRole:
+        return backgroundBrush(UiStyle::Timestamp, true);
     case ChatLineModel::FormatRole:
-      return QVariant::fromValue<UiStyle::FormatList>(part->formatList);
-    case ChatLineModel::WrapListRole:
-      if(column != ChatLineModel::ContentsColumn) return QVariant();
-      QVariantList wrapList;
-      typedef QPair<quint16, quint16> WrapPoint;  // foreach can't parse templated params
-      foreach(WrapPoint pair, _wrapList) wrapList << pair.first << pair.second;
-      return wrapList;
-  }
-
-  return MessageModelItem::data(column, role);
+        return QVariant::fromValue<UiStyle::FormatList>(UiStyle::FormatList()
+                                                        << qMakePair((quint16)0, (quint32) UiStyle::formatType(_styledMsg.type()) | UiStyle::Timestamp));
+    }
+    return QVariant();
 }
 
-bool ChatLineModelItem::setData(int column, const QVariant &value, int role) {
-  return false;
+
+QVariant ChatLineModelItem::senderData(int role) const
+{
+    switch (role) {
+    case ChatLineModel::DisplayRole:
+        return _styledMsg.decoratedSender();
+    case ChatLineModel::EditRole:
+        return _styledMsg.plainSender();
+    case ChatLineModel::BackgroundRole:
+        return backgroundBrush(UiStyle::Sender);
+    case ChatLineModel::SelectedBackgroundRole:
+        return backgroundBrush(UiStyle::Sender, true);
+    case ChatLineModel::FormatRole:
+        return QVariant::fromValue<UiStyle::FormatList>(UiStyle::FormatList()
+                                                        << qMakePair((quint16)0, (quint32) UiStyle::formatType(_styledMsg.type()) | UiStyle::Sender));
+    }
+    return QVariant();
 }
 
-void ChatLineModelItem::computeWrapList() {
-  WrapList wplist;  // use a temp list which we'll later copy into a QVector for efficiency
-  QTextBoundaryFinder finder(QTextBoundaryFinder::Word, _contents.plainText);
-  int idx;
-  int flistidx = -1;
-  int fmtend = -1;
-  QFontMetricsF *metrics;
-  QPair<quint16, quint16> wp(0, 0);
-  do {
-    idx = finder.toNextBoundary();
-    if(idx < 0) idx = _contents.plainText.length();
-    else if(finder.boundaryReasons() != QTextBoundaryFinder::StartWord) continue;
-    int start = wp.first;
-    while(start < idx) {
-      if(fmtend <= start) {
-        flistidx++;
-        fmtend = _contents.formatList.count() > flistidx+1 ? _contents.formatList[flistidx+1].first
-                                                           : _contents.plainText.length();
-        metrics = QtUi::style()->fontMetrics(_contents.formatList[flistidx].second);
-        Q_ASSERT(fmtend > start);
-      }
-      int i = qMin(idx, fmtend);
-      wp.second += metrics->width(_contents.plainText.mid(start, i - start));
-      start = i;
+
+QVariant ChatLineModelItem::contentsData(int role) const
+{
+    switch (role) {
+    case ChatLineModel::DisplayRole:
+    case ChatLineModel::EditRole:
+        return _styledMsg.plainContents();
+    case ChatLineModel::BackgroundRole:
+        return backgroundBrush(UiStyle::Contents);
+    case ChatLineModel::SelectedBackgroundRole:
+        return backgroundBrush(UiStyle::Contents, true);
+    case ChatLineModel::FormatRole:
+        return QVariant::fromValue<UiStyle::FormatList>(_styledMsg.contentsFormatList());
+    case ChatLineModel::WrapListRole:
+        if (_wrapList.isEmpty())
+            computeWrapList();
+        return QVariant::fromValue<ChatLineModel::WrapList>(_wrapList);
     }
-    wplist.append(wp);
-    wp.first = idx;
-  } while(idx < _contents.plainText.length());
-
-  // A QVector needs less space than a QList
-  _wrapList.resize(wplist.count());
-  for(int i = 0; i < wplist.count(); i++) {
-    _wrapList[i] = wplist.at(i);
-  }
+    return QVariant();
+}
+
+
+quint32 ChatLineModelItem::messageLabel() const
+{
+    quint32 label = _styledMsg.senderHash() << 16;
+    if (_styledMsg.flags() & Message::Self)
+        label |= UiStyle::OwnMsg;
+    if (_styledMsg.flags() & Message::Highlight)
+        label |= UiStyle::Highlight;
+    return label;
 }
 
+
+QVariant ChatLineModelItem::backgroundBrush(UiStyle::FormatType subelement, bool selected) const
+{
+    QTextCharFormat fmt = QtUi::style()->format(UiStyle::formatType(_styledMsg.type()) | subelement, messageLabel() | (selected ? UiStyle::Selected : 0));
+    if (fmt.hasProperty(QTextFormat::BackgroundBrush))
+        return QVariant::fromValue<QBrush>(fmt.background());
+    return QVariant();
+}
+
+
+void ChatLineModelItem::computeWrapList() const
+{
+    QString text = _styledMsg.plainContents();
+    int length = text.length();
+    if (!length)
+        return;
+
+    QList<ChatLineModel::Word> wplist; // use a temp list which we'll later copy into a QVector for efficiency
+    QTextBoundaryFinder finder(QTextBoundaryFinder::Line, _styledMsg.plainContents().unicode(), length,
+        TextBoundaryFinderBuffer, TextBoundaryFinderBufferSize);
+
+    int idx;
+    int oldidx = 0;
+    ChatLineModel::Word word;
+    word.start = 0;
+    qreal wordstartx = 0;
+
+    QTextLayout layout(_styledMsg.plainContents());
+    QTextOption option;
+    option.setWrapMode(QTextOption::NoWrap);
+    layout.setTextOption(option);
+
+    layout.setAdditionalFormats(QtUi::style()->toTextLayoutList(_styledMsg.contentsFormatList(), length, messageLabel()));
+    layout.beginLayout();
+    QTextLine line = layout.createLine();
+    line.setNumColumns(length);
+    layout.endLayout();
+
+    while ((idx = finder.toNextBoundary()) >= 0 && idx <= length) {
+        // QTextBoundaryFinder has inconsistent behavior in Qt version up to and including 4.6.3 (at least).
+        // It doesn't point to the position we should break, but to the character before that.
+        // Unfortunately Qt decided to fix this by changing the behavior of QTBF, so now we have to add a version
+        // check. At the time of this writing, I'm still trying to get this reverted upstream...
+        //
+        // cf. https://bugs.webkit.org/show_bug.cgi?id=31076 and Qt commit e6ac173
+        static int needWorkaround = -1;
+        if (needWorkaround < 0) {
+            needWorkaround = 0;
+            QStringList versions = QString(qVersion()).split('.');
+            if (versions.count() == 3 && versions.at(0).toInt() == 4) {
+                if (versions.at(1).toInt() <= 6 && versions.at(2).toInt() <= 3)
+                    needWorkaround = 1;
+            }
+        }
+        if (needWorkaround == 1) {
+            if (idx < length)
+                idx++;
+        }
+
+        if (idx == oldidx)
+            continue;
+
+        word.start = oldidx;
+        int wordend = idx;
+        for (; wordend > word.start; wordend--) {
+            if (!text.at(wordend-1).isSpace())
+                break;
+        }
+
+        qreal wordendx = line.cursorToX(wordend);
+        qreal trailingendx = line.cursorToX(idx);
+        word.endX = wordendx;
+        word.width = wordendx - wordstartx;
+        word.trailing = trailingendx - wordendx;
+        wordstartx = trailingendx;
+        wplist.append(word);
+
+        oldidx = idx;
+    }
+
+    // A QVector needs less space than a QList
+    _wrapList.resize(wplist.count());
+    for (int i = 0; i < wplist.count(); i++) {
+        _wrapList[i] = wplist.at(i);
+    }
+}