qTox/src/chatlog/content/text.cpp

/*
    Copyright (C) 2014 by Project Tox <https://tox.im>

    This file is part of qTox, a Qt-based graphical interface for Tox.

    This program is libre software: you can redistribute it and/or modify
    it under the terms of the GNU General Public License as published by
    the Free Software Foundation, either version 3 of the License, or
    (at your option) any later version.
    This program is distributed in the hope that it will be useful,
    but WITHOUT ANY WARRANTY; without even the implied warranty of
    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.

    See the COPYING file for more details.
*/

#include "text.h"

#include "../customtextdocument.h"
#include "src/misc/smileypack.h"

#include <QFontMetrics>
#include <QPainter>
#include <QPalette>
#include <QDebug>
#include <QTextBlock>
#include <QAbstractTextDocumentLayout>
#include <QApplication>
#include <QGraphicsSceneMouseEvent>
#include <QFontMetrics>
#include <QRegExp>
#include <QDesktopServices>

Text::Text(const QString& txt, QFont font, bool enableElide)
    : ChatLineContent()
    , elide(enableElide)
    , defFont(font)
{
    setText(txt);
    setAcceptedMouseButtons(Qt::LeftButton | Qt::RightButton);

    ensureIntegrity();
    freeResources();
    //setCacheMode(QGraphicsItem::DeviceCoordinateCache);
}

Text::~Text()
{
    delete doc;
}

void Text::setText(const QString& txt)
{
    text = SmileyPack::getInstance().smileyfied(toHtmlChars(txt));

    detectAnchors();
    detectQuotes();

    ensureIntegrity();
    freeResources();
}

void Text::setWidth(qreal w)
{
    if(w == width)
        return;

    width = w;

    if(elide)
    {
        QFontMetrics metrics = QFontMetrics(QFont());
        elidedText = metrics.elidedText(text, Qt::ElideRight, width);
    }

    ensureIntegrity();
    freeResources();
}

void Text::selectionMouseMove(QPointF scenePos)
{
    ensureIntegrity();
    int cur = cursorFromPos(scenePos);
    if(cur >= 0)
    {
        cursor.setPosition(cur, QTextCursor::KeepAnchor);
        selectedText = cursor.selectedText();
    }

    update();
}

void Text::selectionStarted(QPointF scenePos)
{
    ensureIntegrity();
    int cur = cursorFromPos(scenePos);
    if(cur >= 0)
        cursor.setPosition(cur);

    selectedText.clear();
    selectedText.squeeze();
}

void Text::selectionCleared()
{
    ensureIntegrity();
    cursor.setPosition(0);
    selectedText.clear();
    selectedText.squeeze();
    freeResources();

    update();
}

void Text::selectAll()
{
    ensureIntegrity();
    cursor.select(QTextCursor::Document);
    selectedText = text;
    update();
}

bool Text::isOverSelection(QPointF scenePos) const
{
    int cur = cursorFromPos(scenePos);
    if(cur >= 0 && cursor.selectionStart() < cur && cursor.selectionEnd() >= cur)
        return true;

    return false;
}

QString Text::getSelectedText() const
{
    return selectedText;
}

QRectF Text::boundingSceneRect() const
{
    return QRectF(scenePos(), size);
}

QRectF Text::boundingRect() const
{
    return QRectF(QPointF(0, 0), size);
}

void Text::paint(QPainter* painter, const QStyleOptionGraphicsItem* option, QWidget* widget)
{
    if(doc)
    {
        // draw selection
        QAbstractTextDocumentLayout::PaintContext ctx;
        QAbstractTextDocumentLayout::Selection sel;
        sel.cursor = cursor;
        sel.format.setBackground(QApplication::palette().color(QPalette::Highlight));
        sel.format.setForeground(QApplication::palette().color(QPalette::HighlightedText));
        ctx.selections.append(sel);

        // draw text
        doc->documentLayout()->draw(painter, ctx);
    }

    Q_UNUSED(option)
    Q_UNUSED(widget)
}

void Text::visibilityChanged(bool visible)
{
    isVisible = visible;

    if(visible)
        ensureIntegrity();
    else
        freeResources();
}

qreal Text::getAscent() const
{
    return vOffset;
}

void Text::mousePressEvent(QGraphicsSceneMouseEvent *event)
{
    if(event->button() == Qt::LeftButton)
        event->accept(); // grabber
}

void Text::mouseReleaseEvent(QGraphicsSceneMouseEvent *event)
{
    QString anchor = doc->documentLayout()->anchorAt(event->pos());

    // open anchors in browser
    if(!anchor.isEmpty())
        QDesktopServices::openUrl(anchor);
}

QString Text::getText() const
{
    return text;
}

void Text::ensureIntegrity()
{
    if(!doc)
    {
        doc = new CustomTextDocument();
        doc->setDefaultFont(defFont);

        if(!elide)
        {
            doc->setHtml(text);
        }
        else
        {
            QTextOption opt;
            opt.setWrapMode(QTextOption::NoWrap);
            doc->setDefaultTextOption(opt);
            doc->setPlainText(elidedText);
        }

        cursor = QTextCursor(doc);
    }

    doc->setTextWidth(width);
    doc->documentLayout()->update();

    if(doc->firstBlock().layout()->lineCount() > 0)
        vOffset = doc->firstBlock().layout()->lineAt(0).ascent();

    if(size != idealSize())
    {
        prepareGeometryChange();
        size = idealSize();
    }
}

void Text::freeResources()
{
    if(doc && !isVisible && !cursor.hasSelection())
    {
        delete doc;
        doc = nullptr;
        cursor = QTextCursor();
    }
}

QSizeF Text::idealSize()
{
    if(doc)
        return QSizeF(doc->idealWidth(), doc->size().height());

    return QSizeF();
}

int Text::cursorFromPos(QPointF scenePos) const
{
    if(doc)
        return doc->documentLayout()->hitTest(mapFromScene(scenePos), Qt::FuzzyHit);

    return -1;
}

void Text::detectAnchors()
{
    // detect urls
    QRegExp exp("(?:\\b)(www\\.|http[s]?:\\/\\/|ftp:\\/\\/|tox:\\/\\/|tox:)\\S+");
    int offset = 0;
    while ((offset = exp.indexIn(text, offset)) != -1)
    {
        QString url = exp.cap();

        // If there's a trailing " it's a HTML attribute, e.g. a smiley img's title=":tox:"
        if (url == "tox:\"")
        {
            offset += url.length();
            continue;
        }

        // add scheme if not specified
        if (exp.cap(1) == "www.")
            url.prepend("http://");

        QString htmledUrl = QString("<a href=\"%1\">%1</a>").arg(url);
        text.replace(offset, exp.cap().length(), htmledUrl);

        offset += htmledUrl.length();
    }
}

void Text::detectQuotes()
{
    // detect text quotes
    QStringList messageLines = text.split("\n");
    QString quotedText;
    for (int i=0;i<messageLines.size();++i)
    {
        if (QRegExp("^[ ]*&gt;.*").exactMatch(messageLines[i]))
            quotedText += "<span class=quote>" + messageLines[i] + "</span>";
        else
            quotedText += messageLines[i];

        if (i < messageLines.size() - 1)
            quotedText += "<br/>";
    }

    text = quotedText;
}

QString Text::toHtmlChars(const QString &str)
{
    static QList<QPair<QString, QString>> replaceList = {{"&","&amp;"}, {">","&gt;"}, {"<","&lt;"}};
    QString res = str;

    for (auto &it : replaceList)
        res = res.replace(it.first,it.second);

    return res;
}