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 <QFontMetrics>
#include <QPainter>
#include <QPalette>
#include <QDebug>
#include <QTextBlock>
#include <QAbstractTextDocumentLayout>
#include <QApplication>
#include <QGraphicsSceneMouseEvent>
#include <QDesktopServices>
#include <QTextFragment>

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

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

void Text::setText(const QString& txt)
{
    text = txt;
    dirty = true;

    regenerate();
}

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

    width = w;
    dirty = true;

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

    regenerate();
}

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

        QTextBlock block = cursor.block();
        for(QTextBlock::Iterator itr = block.begin(); itr!=block.end(); ++itr)
        {
            int pos = itr.fragment().position(); //fragment position -> position of the first character in the fragment

            if(itr.fragment().charFormat().isImageFormat())
            {
                QTextImageFormat imgFmt = itr.fragment().charFormat().toImageFormat();
                QString key = imgFmt.name(); //img key (eg. key::D for :D)
                QString rune = key.mid(4);

                if(pos >= cursor.selectionStart() && pos < cursor.selectionEnd())
                {
                    selectedText += rune;
                    pos++;
                }
            }
            else
            {
                for(QChar c : itr.fragment().text())
                {
                    if(pos >= cursor.selectionStart() && pos < cursor.selectionEnd())
                        selectedText += c;

                    pos++;
                }
            }
        }
    }

    update();
}

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

void Text::selectionCleared()
{
    cursor.clearSelection();
    selectedText.clear();
    selectedText.squeeze();

    update();
}

void Text::selectAll()
{
    regenerate();
    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;

    regenerate();
    update();
}

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

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

void Text::mouseReleaseEvent(QGraphicsSceneMouseEvent *event)
{
    if(!doc)
        return;

    QString anchor = doc->documentLayout()->anchorAt(event->pos());

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

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

void Text::regenerate()
{
    if(!doc)
    {
        doc = new CustomTextDocument();
        doc->setDefaultFont(defFont);
        dirty = true;
    }

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

        cursor = QTextCursor(doc);
        dirty = false;
    }

    // width & layout
    doc->setTextWidth(width);
    doc->documentLayout()->update();

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

    // let the scene know about our change in size
    if(size != idealSize())
        prepareGeometryChange();

    // get the new width and height
    size = idealSize();

    // if we are not visible -> free mem
    if(!isVisible && !cursor.hasSelection())
        freeResources();
}

void Text::freeResources()
{
    delete doc;
    doc = nullptr;
    cursor = QTextCursor();
}

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

    return size;
}

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

    return -1;
}