sendtextedit.cpp

#include "sendtextedit.h"
#include <QDragEnterEvent>
#include <QDropEvent>
#include <QMimeData>
#include <QFileInfo>
#include <QDebug>
#include <QDateTime>
#include <QDir>
#include <QImage>
#include <QImageReader>
#include <QMenu>
#include <QPixmap>
#include <QStandardPaths>
#include <QTextBlock>
#include <QTextFragment>
#include <QTextImageFormat>
#include "emoji.h"

SendTextEdit::SendTextEdit(QWidget *parent)
    :QTextEdit(parent)
{
    setAcceptDrops(true);
    installEventFilter(this);
}

void SendTextEdit::newLine()
{
    append("");
}

void SendTextEdit::insertEmoji(const QString& emoji)
{
    auto index = emojiIndexOf(emoji);
    if (index < 0)
    {
        insertPlainText(emoji);
        return;
    }

    QTextImageFormat imageFormat;
    imageFormat.setName(emojiResourceName(index));
    textCursor().insertImage(imageFormat);
}

QString SendTextEdit::textWithEmojiCode() const
{
    QString text;
    auto doc = document();
    for (auto block = doc->begin(); block != doc->end(); block = block.next())
    {
        if (block != doc->begin())
            text.append("\n");

        for (auto it = block.begin(); !it.atEnd(); ++it)
        {
            auto fragment = it.fragment();
            if (!fragment.isValid())
                continue;

            auto format = fragment.charFormat();
            if (format.isImageFormat())
            {
                auto imageName = format.toImageFormat().name();
                for (auto i = 0; i < EMOJI_LEN; i++)
                {
                    if (imageName == emojiResourceName(i))
                    {
                        text.append(g_emojis[i]);
                        break;
                    }
                }
            }
            else
            {
                text.append(fragment.text());
            }
        }
    }

    return text;
}

void SendTextEdit::contextMenuEvent(QContextMenuEvent *e)
{
    QMenu menu(this);
    auto undoAction = menu.addAction("撤销");
    auto redoAction = menu.addAction("重做");
    menu.addSeparator();
    auto cutAction = menu.addAction("剪切");
    auto copyAction = menu.addAction("复制");
    auto pasteAction = menu.addAction("粘贴");
    auto deleteAction = menu.addAction("删除");
    menu.addSeparator();
    auto selectAllAction = menu.addAction("全选");

    undoAction->setEnabled(document()->isUndoAvailable());
    redoAction->setEnabled(document()->isRedoAvailable());
    cutAction->setEnabled(textCursor().hasSelection());
    copyAction->setEnabled(textCursor().hasSelection());
    deleteAction->setEnabled(textCursor().hasSelection());
    selectAllAction->setEnabled(!document()->isEmpty());

    auto selected = menu.exec(e->globalPos());
    if (selected == undoAction)
        undo();
    else if (selected == redoAction)
        redo();
    else if (selected == cutAction)
        cut();
    else if (selected == copyAction)
        copy();
    else if (selected == pasteAction)
        paste();
    else if (selected == deleteAction)
        textCursor().removeSelectedText();
    else if (selected == selectAllAction)
        selectAll();
}

bool SendTextEdit::canInsertFromMimeData(const QMimeData *source) const
{
    if (source->hasImage())
        return true;

    if (source->hasUrls())
    {
        for (auto url : source->urls())
        {
            if (isImageFile(url.toLocalFile()))
                return true;
        }
    }

    return QTextEdit::canInsertFromMimeData(source);
}

void SendTextEdit::insertFromMimeData(const QMimeData *source)
{
    QList<QFileInfo> images;

    if (source->hasImage())
    {
        auto data = source->imageData();
        QImage image;
        if (data.canConvert<QImage>())
            image = qvariant_cast<QImage>(data);
        else if (data.canConvert<QPixmap>())
            image = qvariant_cast<QPixmap>(data).toImage();

        auto path = saveClipboardImage(image);
        if (!path.isEmpty())
            images.append(QFileInfo(path));
    }
    else if (source->hasUrls())
    {
        for (auto url : source->urls())
        {
            auto path = url.toLocalFile();
            if (isImageFile(path))
                images.append(QFileInfo(path));
        }
    }

    if (!images.isEmpty())
    {
        emit acceptPasteImages(images);
        return;
    }

    QTextEdit::insertFromMimeData(source);
}

QString SendTextEdit::saveClipboardImage(const QImage &image) const
{
    if (image.isNull())
        return "";

    auto dirPath = QStandardPaths::writableLocation(QStandardPaths::CacheLocation);
    if (dirPath.isEmpty())
        dirPath = QDir::home().filePath(".feiq_cache");
    dirPath = QDir(dirPath).filePath("pasted_images");
    QDir().mkpath(dirPath);

    auto fileName = QString("paste_%1.png").arg(QDateTime::currentMSecsSinceEpoch());
    auto path = QDir(dirPath).filePath(fileName);
    if (!image.save(path, "PNG"))
        return "";

    return path;
}

bool SendTextEdit::isImageFile(const QString &path) const
{
    if (path.isEmpty() || !QFileInfo(path).isFile())
        return false;

    return QImageReader(path).canRead();
}

void SendTextEdit::dragEnterEvent(QDragEnterEvent *e)
{
    if (e->mimeData()->hasUrls())
    {
        auto urls = e->mimeData()->urls();
        for (auto url : urls)
        {
            if (QFileInfo(url.toLocalFile()).isFile())
            {
                e->accept();
                return;
            }
        }
    }
    else
    {
        QTextEdit::dragEnterEvent(e);
    }
}

void SendTextEdit::dropEvent(QDropEvent *e)
{
    if (e->mimeData()->hasUrls())
    {
        auto urls = e->mimeData()->urls();
        QList<QFileInfo> files;
        for (auto url : urls)
        {
            files.append(QFileInfo(url.toLocalFile()));
            e->accept();
        }
        emit acceptDropFiles(files);
    }   
    else
    {
        QTextEdit::dropEvent(e);
    }
}

bool SendTextEdit::eventFilter(QObject *, QEvent * e)
{
    if (e->type() == QEvent::KeyPress)
    {
        auto keyEvent = static_cast<QKeyEvent*>(e);
        auto enter = keyEvent->key() == Qt::Key_Return || keyEvent->key() == Qt::Key_Enter;
        auto ctrl  = keyEvent->modifiers() & Qt::ControlModifier;
        if (enter && ctrl)
        {
            emit ctrlEnterPressed();
            return true;
        }
        else if (enter)
        {
            emit enterPressed();
            return true;
        }
    }
    return false;
}