/*
    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 "genericchatform.h"
#include "ui_mainwindow.h"
#include <QFileDialog>
#include <QHBoxLayout>
#include "src/misc/smileypack.h"
#include "src/widget/emoticonswidget.h"
#include "src/misc/style.h"
#include "src/widget/widget.h"
#include "src/misc/settings.h"
#include "src/widget/tool/chatactions/messageaction.h"
#include "src/widget/tool/chatactions/systemmessageaction.h"
#include "src/widget/tool/chatactions/actionaction.h"
#include "src/widget/tool/chatactions/alertaction.h"
#include "src/widget/chatareawidget.h"
#include "src/widget/tool/chattextedit.h"
#include "src/widget/maskablepixmapwidget.h"
#include "src/core.h"
#include "src/grouplist.h"
#include "src/group.h"
#include "src/friendlist.h"
#include "src/friend.h"

GenericChatForm::GenericChatForm(QWidget *parent) :
    QWidget(parent),
    earliestMessage(nullptr)
  , audioInputFlag(false)
  , audioOutputFlag(false)
{
    curRow = 0;

    headWidget = new QWidget();

    nameLabel = new CroppingLabel();
    nameLabel->setObjectName("nameLabel");
    nameLabel->setMinimumHeight(Style::getFont(Style::Medium).pixelSize());
    nameLabel->setEditable(true);

    avatar = new MaskablePixmapWidget(this, QSize(40,40), ":/img/avatar_mask.png");
    QHBoxLayout *headLayout = new QHBoxLayout(), *mainFootLayout = new QHBoxLayout();
    headTextLayout = new QVBoxLayout();
    QVBoxLayout *mainLayout = new QVBoxLayout();
    QVBoxLayout *footButtonsSmall = new QVBoxLayout(), *volMicLayout = new QVBoxLayout();

    chatWidget = new ChatAreaWidget();

    msgEdit = new ChatTextEdit();

    sendButton = new QPushButton();
    sendButton->setToolTip(tr("Send message"));
    emoteButton = new QPushButton();
    emoteButton->setToolTip(tr("Smileys"));

    // Setting the sizes in the CSS doesn't work (glitch with high DPIs)
    fileButton = new QPushButton();
    fileButton->setToolTip(tr("Send file(s)"));
    callButton = new QPushButton();
    callButton->setFixedSize(50,40);
    callButton->setToolTip(tr("Audio call: RED means you're on a call"));
    videoButton = new QPushButton();
    videoButton->setFixedSize(50,40);
    videoButton->setToolTip(tr("Video call: RED means you're on a call"));
    volButton = new QPushButton();
    volButton->setFixedSize(25,20);
    volButton->setToolTip(tr("Toggle speakers volume: RED is OFF"));
    micButton = new QPushButton();
    micButton->setFixedSize(25,20);
    micButton->setToolTip(tr("Toggle microphone: RED is OFF"));

    footButtonsSmall->setSpacing(2);

    msgEdit->setStyleSheet(Style::getStylesheet(":/ui/msgEdit/msgEdit.css"));
    msgEdit->setFixedHeight(50);
    msgEdit->setFrameStyle(QFrame::NoFrame);

    sendButton->setStyleSheet(Style::getStylesheet(":/ui/sendButton/sendButton.css"));
    fileButton->setStyleSheet(Style::getStylesheet(":/ui/fileButton/fileButton.css"));
    emoteButton->setStyleSheet(Style::getStylesheet(":/ui/emoteButton/emoteButton.css"));

    callButton->setObjectName("green");
    callButton->setStyleSheet(Style::getStylesheet(":/ui/callButton/callButton.css"));

    videoButton->setObjectName("green");
    videoButton->setStyleSheet(Style::getStylesheet(":/ui/videoButton/videoButton.css"));

    QString volButtonStylesheet = Style::getStylesheet(":/ui/volButton/volButton.css");
    volButton->setObjectName("green");
    volButton->setStyleSheet(volButtonStylesheet);

    QString micButtonStylesheet = Style::getStylesheet(":/ui/micButton/micButton.css");
    micButton->setObjectName("green");
    micButton->setStyleSheet(micButtonStylesheet);

    setLayout(mainLayout);
    mainLayout->addWidget(chatWidget);
    mainLayout->addLayout(mainFootLayout);
    mainLayout->setMargin(0);

    footButtonsSmall->addWidget(emoteButton);
    footButtonsSmall->addWidget(fileButton);

    mainFootLayout->addWidget(msgEdit);
    mainFootLayout->addLayout(footButtonsSmall);
    mainFootLayout->addSpacing(5);
    mainFootLayout->addWidget(sendButton);
    mainFootLayout->setSpacing(0);

    headWidget->setLayout(headLayout);
    headLayout->addWidget(avatar);
    headLayout->addSpacing(5);
    headLayout->addLayout(headTextLayout);
    headLayout->addLayout(volMicLayout);
    headLayout->addWidget(callButton);
    headLayout->addSpacing(3);
    headLayout->addWidget(videoButton);
    headLayout->setSpacing(0);

    volMicLayout->addStretch();
    volMicLayout->addWidget(micButton);
    volMicLayout->addWidget(volButton);
    volMicLayout->addStretch();

    headTextLayout->addStretch();
    headTextLayout->addWidget(nameLabel);

    //Fix for incorrect layouts on OS X as per
    //https://bugreports.qt-project.org/browse/QTBUG-14591
    sendButton->setAttribute(Qt::WA_LayoutUsesWidgetRect);
    fileButton->setAttribute(Qt::WA_LayoutUsesWidgetRect);
    emoteButton->setAttribute(Qt::WA_LayoutUsesWidgetRect);

    menu.addAction(tr("Save chat log"), this, SLOT(onSaveLogClicked()));
    menu.addAction(tr("Clear displayed messages"), this, SLOT(clearChatArea(bool)));
    menu.addSeparator();

    connect(emoteButton,  SIGNAL(clicked()), this, SLOT(onEmoteButtonClicked()));
    connect(chatWidget, SIGNAL(customContextMenuRequested(QPoint)), this, SLOT(onChatContextMenuRequested(QPoint)));
    connect(chatWidget, SIGNAL(onClick()), this, SLOT(onChatWidgetClicked()));

    chatWidget->document()->setDefaultStyleSheet(Style::getStylesheet(":ui/chatArea/innerStyle.css"));
    chatWidget->setStyleSheet(Style::getStylesheet(":/ui/chatArea/chatArea.css"));
    headWidget->setStyleSheet(Style::getStylesheet(":/ui/chatArea/chatHead.css"));

    ChatAction::setupFormat();
}

bool GenericChatForm::isEmpty()
{
    return chatWidget->isEmpty();
}

void GenericChatForm::setName(const QString &newName)
{
    nameLabel->setText(newName);
    nameLabel->setToolTip(newName); // for overlength names
}

void GenericChatForm::show(Ui::MainWindow &ui)
{
    ui.mainContent->layout()->addWidget(this);
    ui.mainHead->layout()->addWidget(headWidget);
    headWidget->show();
    QWidget::show();
}

void GenericChatForm::onChatContextMenuRequested(QPoint pos)
{
    QWidget* sender = (QWidget*)QObject::sender();
    pos = sender->mapToGlobal(pos);
    menu.exec(pos);
}

void GenericChatForm::onSaveLogClicked()
{
    QString path = QFileDialog::getSaveFileName(0, tr("Save chat log"));
    if (path.isEmpty())
        return;

    QFile file(path);
    if (!file.open(QIODevice::WriteOnly | QIODevice::Text))
        return;

    QString log;
    log = chatWidget->toPlainText();

    file.write(log.toUtf8());
    file.close();
}

MessageActionPtr GenericChatForm::addMessage(const ToxID& author, const QString &message, bool isAction,
                                             const QDateTime &datetime, bool isSent)
{
    MessageActionPtr ca = genMessageActionAction(author, message, isAction, datetime);
    if (isSent)
        ca->markAsSent();
    chatWidget->insertMessage(ca);
    return ca;
}

MessageActionPtr GenericChatForm::addSelfMessage(const QString &message, bool isAction, const QDateTime &datetime, bool isSent)
{
    MessageActionPtr ca = genSelfActionAction(message, isAction, datetime);
    if (isSent)
        ca->markAsSent();
    chatWidget->insertMessage(ca);
    return ca;
}

void GenericChatForm::addAlertMessage(const ToxID &author, QString message, QDateTime datetime)
{
    QString authorStr = resolveToxID(author);
    if (authorStr.isEmpty())
        authorStr = author.publicKey;

    QString date = datetime.toString(Settings::getInstance().getTimestampFormat());
    MessageActionPtr ca = MessageActionPtr(new AlertAction(getElidedName(authorStr), message, date));
    ca->markAsSent();
    chatWidget->insertMessage(ca);
    previousId = author;
}

void GenericChatForm::onEmoteButtonClicked()
{
    // don't show the smiley selection widget if there are no smileys available
    if (SmileyPack::getInstance().getEmoticons().empty())
        return;

    EmoticonsWidget widget;
    connect(&widget, SIGNAL(insertEmoticon(QString)), this, SLOT(onEmoteInsertRequested(QString)));

    QWidget* sender = qobject_cast<QWidget*>(QObject::sender());
    if (sender)
    {
        QPoint pos = -QPoint(widget.sizeHint().width() / 2, widget.sizeHint().height()) - QPoint(0, 10);
        widget.exec(sender->mapToGlobal(pos));
    }
}

void GenericChatForm::onChatWidgetClicked()
{
    msgEdit->setFocus();
}

void GenericChatForm::onEmoteInsertRequested(QString str)
{
    // insert the emoticon
    QWidget* sender = qobject_cast<QWidget*>(QObject::sender());
    if (sender)
        msgEdit->insertPlainText(str);

    msgEdit->setFocus(); // refocus so that we can continue typing
}

void GenericChatForm::focusInput()
{
    msgEdit->setFocus();
}

void GenericChatForm::addSystemInfoMessage(const QString &message, const QString &type, const QDateTime &datetime)
{
    ChatActionPtr ca = genSystemInfoAction(message, type, datetime);
    chatWidget->insertMessage(ca);
}

QString GenericChatForm::getElidedName(const QString& name)
{
    // update this whenever you change the font in innerStyle.css
    QFontMetrics fm(Style::getFont(Style::BigBold));

    return fm.elidedText(name, Qt::ElideRight, chatWidget->nameColWidth());
}

void GenericChatForm::clearChatArea(bool notinform)
{
    chatWidget->clearChatArea();
    previousId = ToxID();

    if (!notinform)
        addSystemInfoMessage(tr("Cleared"), "white", QDateTime::currentDateTime());

    if (earliestMessage)
    {
        delete earliestMessage;
        earliestMessage = nullptr;
    }

    emit chatAreaCleared();
}

MessageActionPtr GenericChatForm::genMessageActionAction(const ToxID& author, QString message, bool isAction, const QDateTime &datetime)
{
    if (earliestMessage == nullptr)
    {
        earliestMessage = new QDateTime(datetime);
    }

    const Core* core = Core::getInstance();

    QString date = datetime.toString(Settings::getInstance().getTimestampFormat());
    bool isMe = (author == core->getSelfId());
    QString authorStr;
    if (isMe)
        authorStr = core->getUsername();
    else {
        authorStr = resolveToxID(author);
    }

    if (authorStr.isEmpty()) // Fallback if we can't find a username
        authorStr = author.toString();

    if (!isAction && message.startsWith("/me "))
    { // always render actions regardless of what core thinks
        isAction = true;
        message = message.right(message.length()-4);
    }

    if (isAction)
    {
        previousId = ToxID(); // next msg has a name regardless
        return MessageActionPtr(new ActionAction (getElidedName(authorStr), message, date, isMe));
    }

    MessageActionPtr res;
    if (previousId == author)
        res = MessageActionPtr(new MessageAction(QString(), message, date, isMe));
    else
        res = MessageActionPtr(new MessageAction(getElidedName(authorStr), message, date, isMe));

    previousId = author;
    return res;
}

MessageActionPtr GenericChatForm::genSelfActionAction(QString message, bool isAction, const QDateTime &datetime)
{
    if (earliestMessage == nullptr)
    {
        earliestMessage = new QDateTime(datetime);
    }

    const Core* core = Core::getInstance();

    QString date = datetime.toString(Settings::getInstance().getTimestampFormat());
    QString author = core->getUsername();;

    if (!isAction && message.startsWith("/me "))
    { // always render actions regardless of what core thinks
        isAction = true;
        message = message.right(message.length()-4);
    }

    if (isAction)
    {
        previousId = ToxID(); // next msg has a name regardless
        return MessageActionPtr(new ActionAction (getElidedName(author), message, date, true));
    }

    MessageActionPtr res;
    if (previousId.isMine())
        res = MessageActionPtr(new MessageAction(QString(), message, date, true));
    else
        res = MessageActionPtr(new MessageAction(getElidedName(author), message, date, true));

    previousId = Core::getInstance()->getSelfId();
    return res;
}

ChatActionPtr GenericChatForm::genSystemInfoAction(const QString &message, const QString &type, const QDateTime &datetime)
{
    previousId = ToxID();
    QString date = datetime.toString(Settings::getInstance().getTimestampFormat());

    return ChatActionPtr(new SystemMessageAction(message, type, date));
}

QString GenericChatForm::resolveToxID(const ToxID &id)
{
    Friend *f = FriendList::findFriend(id);
    if (f)
    {
        return f->getDisplayedName();
    } else {
        for (auto it : GroupList::getAllGroups())
        {
            QString res = it->resolveToxID(id);
            if (res.size())
                return res;
        }
    }

    return QString();
}