pilotclient/src/blackgui/textmessagetextedit.cpp

/* Copyright (C) 2013
 * swift project Community / Contributors
 *
 * This file is part of swift project. It is subject to the license terms in the LICENSE file found in the top-level
 * directory of this distribution and at http://www.swift-project.org/license.html. No part of swift project,
 * including this file, may be copied, modified, propagated, or distributed except according to the terms
 * contained in the LICENSE file.
 */

#include "blackgui/guiapplication.h"
#include "blackgui/textmessagetextedit.h"
#include "blackmisc/aviation/callsign.h"
#include "blackmisc/network/textmessage.h"

#include <QAction>
#include <QMenu>
#include <QScopedPointer>
#include <QTextDocument>
#include <QTextOption>
#include <Qt>
#include <QtGlobal>
#include <QStringBuilder>

using namespace BlackMisc;
using namespace BlackMisc::Network;

namespace BlackGui
{
    CTextMessageTextEdit::CTextMessageTextEdit(QWidget *parent) : QTextEdit(parent)
    {
        this->setDocument(&m_textDocument);
        this->setReadOnly(true);
        this->setWordWrap(true);

        // menu
        this->setContextMenuPolicy(Qt::CustomContextMenu);
        m_actionClearTextEdit = new QAction(CIcon::iconByIndex(CIcons::StandardIconDelete16).toQIcon(), "Clear", this);
        m_actionLast10 = new QAction("Keep last 10 messages", this);
        m_actionLast25 = new QAction("Keep last 25 messages", this);
        m_actionAll = new QAction("Keep all messages", this);
        m_actionWithRecipient = new QAction("With recipient", this);
        m_actionWithSender = new QAction("With sender", this);
        m_actionWordWrap = new QAction("Word wrap", this);
        m_actionWithRecipient->setCheckable(true);
        m_actionWithSender->setCheckable(true);
        m_actionWordWrap->setCheckable(true);

        connect(m_actionClearTextEdit, &QAction::triggered, this, &CTextMessageTextEdit::clear);
        connect(m_actionAll, &QAction::triggered, this, &CTextMessageTextEdit::keepLastNMessages);
        connect(m_actionLast10, &QAction::triggered, this, &CTextMessageTextEdit::keepLastNMessages);
        connect(m_actionLast25, &QAction::triggered, this, &CTextMessageTextEdit::keepLastNMessages);
        connect(m_actionWithSender, &QAction::triggered, this, &CTextMessageTextEdit::setVisibleFields);
        connect(m_actionWithRecipient, &QAction::triggered, this, &CTextMessageTextEdit::setVisibleFields);
        connect(m_actionWordWrap, &QAction::triggered, this, &CTextMessageTextEdit::setWordWrap);

        connect(this, &QTextEdit::customContextMenuRequested, this, &CTextMessageTextEdit::showContextMenuForTextEdit);
    }

    CTextMessageTextEdit::~CTextMessageTextEdit()
    { }

    void CTextMessageTextEdit::insertTextMessage(const CTextMessage &textMessage, int maxMessages)
    {
        if (maxMessages < 0 && m_keepMaxMessages >= 0) { maxMessages = m_keepMaxMessages; }
        if (maxMessages >= 0)
        {
            m_messages.push_frontMaxElements(textMessage, maxMessages);
        }
        else
        {
            m_messages.push_front(textMessage);
        }
        const QString html(toHtml(m_messages, m_withSender, m_withRecipient));
        m_textDocument.setHtml(html);
    }

    int CTextMessageTextEdit::count() const
    {
        return m_messages.size();
    }

    void CTextMessageTextEdit::clear()
    {
        m_messages.clear();
        QTextEdit::clear();
    }

    void CTextMessageTextEdit::redrawHtml()
    {
        const QString html(toHtml(m_messages, m_withSender, m_withRecipient));
        m_textDocument.setHtml(html);
    }

    void CTextMessageTextEdit::setStyleSheetForContent(const QString &styleSheet)
    {
        m_textDocument.setDefaultStyleSheet(styleSheet);
        this->redrawHtml();
    }

    QString CTextMessageTextEdit::toHtml(const CTextMessageList &messages, bool withFrom, bool withTo)
    {
        if (messages.isEmpty()) { return ""; }
        QString html("<table>");
        for (const CTextMessage &msg : messages)
        {
            html += toHtml(msg, withFrom, withTo);
        }
        html += "</table>";
        return html;
    }

    QString CTextMessageTextEdit::toHtml(const CTextMessage &message, bool withFrom, bool withTo)
    {
        QString html;
        QString rowClass;
        if (message.isSupervisorMessage()) { rowClass += " supervisor"; }
        if (message.wasSent()) { rowClass += " sent"; }
        else { rowClass += " received"; }

        if (rowClass.isEmpty())
        {
            html += "<tr>";
        }
        else
        {
            html += "<tr class=\"";
            html += rowClass.trimmed();
            html += "\">";
        }
        html += "<td class=\"timestamp\">";
        html += message.getFormattedUtcTimestampHms();
        html += "</td>";

        if (withFrom)
        {
            html += "<td class=\"sender\">";
            html += message.getSenderCallsign().asString();
            html += "</td>";
        }

        if (withTo)
        {
            html += "<td class=\"recipient\">";
            html += message.getRecipientCallsignOrFrequency();
            html += "</td>";
        }

        html += "<td class=\"message\">";
        html += message.getMessage();
        html += "</td>";

        html += "</tr>";
        return html;
    }

    void CTextMessageTextEdit::showContextMenuForTextEdit(const QPoint &pt)
    {
        m_actionWithRecipient->setChecked(m_withRecipient);
        m_actionWithSender->setChecked(m_withSender);
        m_actionWordWrap->setChecked(m_wordWrap);

        QScopedPointer<QMenu> menu(this->createStandardContextMenu());
        menu->setObjectName(this->objectName().append("_contextMenu"));
        menu->addSeparator();
        menu->addAction(m_actionWordWrap);
        QMenu *subMenu = menu->addMenu("Max.messages");
        subMenu->addAction(m_actionLast10);
        subMenu->addAction(m_actionLast25);
        subMenu->addAction(m_actionAll);
        subMenu = menu->addMenu("Fields");
        subMenu->addAction(m_actionWithSender);
        subMenu->addAction(m_actionWithRecipient);
        menu->addAction(m_actionClearTextEdit);
        menu->exec(this->mapToGlobal(pt));
    }

    void CTextMessageTextEdit::keepLastNMessages()
    {
        QObject *sender = QObject::sender();
        if (sender == m_actionAll)
        {
            m_keepMaxMessages = -1;
        }
        else if (sender == m_actionLast10)
        {
            m_keepMaxMessages = 10;
        }
        else if (sender == m_actionLast25)
        {
            m_keepMaxMessages = 25;
        }
    }

    void CTextMessageTextEdit::setVisibleFields()
    {
        QObject *sender = QObject::sender();
        if (sender == m_actionWithRecipient)
        {
            m_withRecipient = m_actionWithRecipient->isChecked();
        }
        else if (sender == m_actionWithSender)
        {
            m_withSender = m_actionWithSender->isChecked();
        }
    }

    void CTextMessageTextEdit::setWordWrap(bool wordWrap)
    {
        m_wordWrap = wordWrap;
        if (m_wordWrap)
        {
            this->setWordWrapMode(QTextOption::WrapAtWordBoundaryOrAnywhere);
        }
        else
        {
            this->setWordWrapMode(QTextOption::NoWrap);
        }
    }
} // namespace