/*

 * Hedgewars, a free turn based strategy game

 * Copyright (c) 2007 Igor Ulyanov <iulyanov@gmail.com>

 * Copyright (c) 2004-2014 Andrey Korotaev <unC0Rr@gmail.com>

 *

 * This program is free 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; version 2 of the License

 *

 * 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

 * GNU General Public License for more details.

 *

 * You should have received a copy of the GNU General Public License

 * along with this program; if not, write to the Free Software

 */

#include <QDesktopServices>

#include <QTextBrowser>

#include <QAction>

#include <QFile>

#include <QTextStream>

#include <QMenu>

#include <QCursor>

#include <QItemSelectionModel>

#include <QDateTime>

#include <QTime>

#include <QListView>

#include <QModelIndexList>

#include <QSortFilterProxyModel>

#include <QMenu>

#include <QScrollBar>

#include "DataManager.h"

#include "hwconsts.h"

#include "gameuiconfig.h"

#include "playerslistmodel.h"

#include "HWApplication.h"

#include "chatwidget.h"

QString * HWChatWidget::s_styleSheet = NULL;

QStringList * HWChatWidget::s_displayNone = NULL;

bool HWChatWidget::s_isTimeStamped = true;

QString HWChatWidget::s_tsFormat = ":mm:ss";

const QString & HWChatWidget::styleSheet()

{

    if (s_styleSheet != NULL)

        return *s_styleSheet;

    setStyleSheet();

    return *s_styleSheet;

}

void HWChatWidget::setStyleSheet(const QString & styleSheet)

{

    QString orgStyleSheet = styleSheet;

    QString style = QString(orgStyleSheet);

    // no stylesheet supplied, search for one or use default

    if (orgStyleSheet.isEmpty())

    {

        // load external stylesheet if there is any

        QFile extFile("physfs://css/chat.css");

        QFile resFile(":/res/css/chat.css");

        QFile & file = (extFile.exists()?extFile:resFile);

        if (file.open(QIODevice::ReadOnly | QIODevice::Text))

        {

            QTextStream in(&file);

            while (!in.atEnd())

            {

                style.append(in.readLine()+"\n");

            }

            orgStyleSheet = style;

            file.close();

        }

    }

    // let's parse display:none; ...

    // prepare for MAGIC :D

    // matches (multi-)whitespaces (for replacement with simple space)

    QRegExp ws("\\s+");

    // matches comments (for removal)

    QRegExp rem("/\\*([^*]|\\*(?!/))*\\*/");

    // strip comments and multi-whitespaces to compress the style-sheet a bit

    style = style.remove(rem).replace(ws," ");

    // now let's see what messages the user does not want to be displayed

    // by checking for display:none; (since QTextBrowser does not support it)

    // MOAR MAGIC :DDD

    // matches definitions lacking display:none; (for removal)

    QRegExp displayed(

        "([^{}]*\\{)(?!([^}]*;)* ?display ?: ?none ?(;[^}]*)?\\})[^}]*\\}");

    // matches all {...} and , (used as seperator for splitting into names)

    QRegExp split(" *(\\{[^}]*\\}|,) *");

    // matches class names that are referenced without hierachy

    QRegExp nohierarchy("^.[^ .]+$");

    QStringList victims = QString(style).

                          remove(displayed). // remove visible stuff

                          trimmed().

                          split(split). // get a list of the names

                          filter(nohierarchy). // only direct class names

                          replaceInStrings(QRegExp("^."),""); // crop .

    if (victims.contains("timestamp"))

    {

        s_isTimeStamped = false;

        victims.removeAll("timestamp");

    }

    else

    {

        s_isTimeStamped = true;

        s_tsFormat =

            ((victims.contains("timestamp:hours"))?"":"hh:") +

            QString("mm") +

            ((victims.contains("timestamp:seconds"))?"":":ss");

    }

    victims.removeAll("timestamp:hours");

    victims.removeAll("timestamp:seconds");

    victims.removeDuplicates();

    QStringList * oldDisplayNone = s_displayNone;

    QString * oldStyleSheet = s_styleSheet;

    s_displayNone = new QStringList(victims);

    s_styleSheet = new QString(orgStyleSheet);

    if (oldDisplayNone != NULL)

        delete oldDisplayNone;

    if (oldStyleSheet != NULL)

        delete oldStyleSheet;

}

void HWChatWidget::displayError(const QString & message)

{

    addLine("msg_Error", " !!! " + message);

}

void HWChatWidget::displayNotice(const QString & message)

{

    addLine("msg_Notice", " *** " + message);

}

void HWChatWidget::displayWarning(const QString & message)

{

    addLine("msg_Warning", " *!* " + message);

}

HWChatWidget::HWChatWidget(QWidget* parent, bool notify) :

    QWidget(parent),

    mainLayout(this)

{

    this->gameSettings = NULL;

    this->notify = notify;

    m_isAdmin = false;

    m_autoKickEnabled = false;

    QStringList vpList =

         QStringList() << "Classic" << "Default" << "Mobster" << "Russian";

    foreach (const QString & vp, vpList)

    {

        m_helloSounds.append(QString("/Sounds/voices/%1/Hello.ogg").arg(vp));

    }

    m_hilightSound = "/Sounds/beep.ogg";

    mainLayout.setMargin(0);

    QWidget * leftSideContainer = new QWidget();

    leftSideContainer->setObjectName("leftSideContainer");

    leftSideContainer->setStyleSheet("#leftSideContainer { border-width: 0px; background-color: #ffcc00; border-radius: 10px;} QTextBrowser, SmartLineEdit { background-color: rgb(13, 5, 68); }");

    QVBoxLayout * leftSide = new QVBoxLayout(leftSideContainer);

    leftSide->setSpacing(3);

    leftSide->setMargin(3);

    mainLayout.addWidget(leftSideContainer, 76);

    // Chat view

    chatText = new QTextBrowser(this);

    chatText->document()->setDefaultStyleSheet(styleSheet());

    chatText->setMinimumHeight(20);

    chatText->setMinimumWidth(10);

    chatText->setSizePolicy(QSizePolicy::Expanding, QSizePolicy::Expanding);

    chatText->setOpenLinks(false);

    chatText->setStyleSheet("QTextBrowser { background-color: rgb(23, 11, 54); border-width: 0px; }");

    connect(chatText, SIGNAL(anchorClicked(const QUrl&)),

            this, SLOT(linkClicked(const QUrl&)));

    leftSide->addWidget(chatText, 1);

    // Input box

    // Normal:  rgb(23, 11, 54)

    // Hover:   rgb(13, 5, 68)

    chatEditLine = new SmartLineEdit();

    chatEditLine->setMaxLength(300);

    chatEditLine->setStyleSheet("SmartLineEdit { background-color: rgb(23, 11, 54); padding: 2px 8px; border-width: 0px; border-radius: 7px; } SmartLineEdit:hover, SmartLineEdit:focus { background-color: rgb(13, 5, 68); }");

    chatEditLine->setFixedHeight(24);

    chatEditLine->setSizePolicy(QSizePolicy::Expanding, QSizePolicy::Fixed);

    connect(chatEditLine, SIGNAL(returnPressed()), this, SLOT(returnPressed()));

    leftSide->addWidget(chatEditLine, 0);

    // Nickname list

    chatNicks = new QListView(this);

    chatNicks->setIconSize(QSize(24, 16));

    chatNicks->setSelectionMode(QAbstractItemView::SingleSelection);

    chatNicks->setEditTriggers(QAbstractItemView::NoEditTriggers);

    chatNicks->setMinimumHeight(10);

    chatNicks->setMinimumWidth(10);

    chatNicks->setSizePolicy(QSizePolicy::Expanding, QSizePolicy::Expanding);

    chatNicks->setContextMenuPolicy(Qt::CustomContextMenu);

    connect(chatNicks, SIGNAL(doubleClicked(QModelIndex)),

            this, SLOT(chatNickDoubleClicked(QModelIndex)));

    connect(chatNicks, SIGNAL(customContextMenuRequested(QPoint)), this, SLOT(nicksContextMenuRequested(QPoint)));

    mainLayout.addSpacing(0);

    mainLayout.addWidget(chatNicks, 24);

    // the userData is used to flag things that are even available when user

    // is offline

    acInfo = new QAction(QAction::tr("Info"), chatNicks);

    acInfo->setIcon(QIcon(":/res/info.png"));

    acInfo->setData(QVariant(false));

    connect(acInfo, SIGNAL(triggered(bool)), this, SLOT(onInfo()));

    acKick = new QAction(QAction::tr("Kick"), chatNicks);

    acKick->setIcon(QIcon(":/res/kick.png"));

    acKick->setData(QVariant(false));

    connect(acKick, SIGNAL(triggered(bool)), this, SLOT(onKick()));

    acBan = new QAction(QAction::tr("Ban"), chatNicks);

    acBan->setIcon(QIcon(":/res/ban.png"));

    acBan->setData(QVariant(true));

    connect(acBan, SIGNAL(triggered(bool)), this, SLOT(onBan()));

    acFollow = new QAction(QAction::tr("Follow"), chatNicks);

    acFollow->setIcon(QIcon(":/res/follow.png"));

    acFollow->setData(QVariant(false));

    connect(acFollow, SIGNAL(triggered(bool)), this, SLOT(onFollow()));

    acIgnore = new QAction(QAction::tr("Ignore"), chatNicks);

    acIgnore->setIcon(QIcon(":/res/ignore.png"));

    acIgnore->setData(QVariant(true));

    connect(acIgnore, SIGNAL(triggered(bool)), this, SLOT(onIgnore()));

    acFriend = new QAction(QAction::tr("Add friend"), chatNicks);

    acFriend->setIcon(QIcon(":/res/addfriend.png"));

    acFriend->setData(QVariant(true));

    connect(acFriend, SIGNAL(triggered(bool)), this, SLOT(onFriend()));

    chatNicks->insertAction(0, acFriend);

    chatNicks->insertAction(0, acInfo);

    chatNicks->insertAction(0, acIgnore);

    setShowFollow(true);

    setAcceptDrops(true);

    m_nicksMenu = new QMenu(this);

    clear();

}

void HWChatWidget::setSettings(QSettings * settings)

{

    gameSettings = settings;

}

void HWChatWidget::linkClicked(const QUrl & link)

{

    if ((link.scheme() == "http") or (link.scheme() == "https"))

        QDesktopServices::openUrl(link);

    else if (link.scheme() == "hwnick")

    {

        // decode nick

        QString nick = QString::fromUtf8(QByteArray::fromBase64(link.encodedQuery()));

        QModelIndexList mil = chatNicks->model()->match(chatNicks->model()->index(0, 0), Qt::DisplayRole, nick);

        bool isOffline = (mil.size() < 1);

        if (isOffline)

        {

            m_clickedNick = nick;

            chatNicks->selectionModel()->clearSelection();

        }

        else

        {

            chatNicks->selectionModel()->select(mil[0], QItemSelectionModel::ClearAndSelect);

        }

        nicksContextMenuRequested(chatNicks->mapFromGlobal(QCursor::pos()));

    }

}

void HWChatWidget::setShowFollow(bool enabled)

{

    if (enabled)

    {

        if (!(chatNicks->actions().contains(acFollow)))

            chatNicks->insertAction(acFriend, acFollow);

    }

    else

    {

        if (chatNicks->actions().contains(acFollow))

            chatNicks->removeAction(acFollow);

    }

}

void HWChatWidget::setIgnoreListKick(bool enabled)

{

    m_autoKickEnabled = enabled;

}

void HWChatWidget::returnPressed()

{

    QStringList lines = chatEditLine->text().split('\n');

    chatEditLine->rememberCurrentText();

    foreach (const QString &line, lines)

    {

        // skip empty/whitespace lines

        if (line.trimmed().isEmpty())

            continue;

        if (!parseCommand(line))

            emit chatLine(line);

    }

    chatEditLine->clear();

}

// "link" nick, but before that encode it in base64 to make sure it can't

// intefere with html/url syntax the nick is put as querystring as putting

// it as host would convert it to it's lower case variant

QString HWChatWidget::linkedNick(const QString & nickname)

{

    if (nickname != m_userNick)

        return QString("<a href=\"hwnick://?%1\" class=\"nick\">%2</a>").arg(

                   QString(nickname.toUtf8().toBase64())).arg(Qt::escape(nickname));

    // unlinked nick (if own one)

    return QString("<span class=\"nick\">%1</span>").arg(Qt::escape(nickname));

}

const QRegExp HWChatWidget::URLREGEXP = QRegExp("(http(s)?://)?(www\\.)?((hedgewars\\.org|code\\.google\\.com|googlecode\\.com|hh\\.unit22\\.org)(/[^ ]*)?)");

bool HWChatWidget::containsHighlight(const QString & sender, const QString & message)

{

    if ((sender != m_userNick) && (!m_userNick.isEmpty()))

    {

        QString lcStr = message.toLower();

        foreach (const QRegExp & hl, m_highlights)

        {

            if (lcStr.contains(hl))

                return true;

        }

    }

    return false;

}

QString HWChatWidget::messageToHTML(const QString & message)

{

    QString formattedStr = Qt::escape(message);

    // link some urls

    formattedStr = formattedStr.replace(URLREGEXP, "<a href=\"http\\2://\\4\">\\4</a>");

    return formattedStr;

}

void HWChatWidget::onChatAction(const QString & nick, const QString & action)

{

    printChatString(nick, "* " + linkedNick(nick) + " " + messageToHTML(action), "Action", containsHighlight(nick, action));

}

void HWChatWidget::onChatMessage(const QString & nick, const QString & message)

{

    printChatString(nick, linkedNick(nick) + ": " + messageToHTML(message), "Chat", containsHighlight(nick, message));

}

void HWChatWidget::printChatString(

    const QString & nick, const QString & str, const QString & cssClassPart, bool highlight)

{

    QSortFilterProxyModel * playersSortFilterModel = qobject_cast<QSortFilterProxyModel *>(chatNicks->model());

    if(!playersSortFilterModel)

        return;

    PlayersListModel * players = qobject_cast<PlayersListModel *>(playersSortFilterModel->sourceModel());

    if(!players)

        return;

    // don't show chat lines that are from ignored nicks

    if (players->isFlagSet(nick, PlayersListModel::Ignore))

        return;

    bool isFriend = (!nick.isEmpty()) && players->isFlagSet(nick, PlayersListModel::Friend);

    QString cssClass = (isFriend ? "msg_Friend" : "msg_User") + cssClassPart;

    addLine(cssClass, str, highlight);

}

void HWChatWidget::addLine(const QString & cssClass, QString line, bool isHighlight)

{

    if (s_displayNone->contains(cssClass))

        return; // the css forbids us to display this line

    beforeContentAdd();

    if (chatStrings.size() > 250)

        chatStrings.removeFirst();

    if (s_isTimeStamped)

    {

        QString tsMarkUp = "<span class=\"timestamp\">[%1]</span> ";

        QTime now = QDateTime::currentDateTime().time();

        line = tsMarkUp.arg(now.toString(s_tsFormat)) + line;

    }

    line = QString("<span class=\"%1\">%2</span>").arg(cssClass).arg(line);

    if (isHighlight)

    {

        line = QString("<span class=\"highlight\">%1</span>").arg(line);

        SDLInteraction::instance().playSoundFile(m_hilightSound);

        HWApplication::alert(this, 800);

    }

    chatStrings.append(line);

    chatText->setHtml("<html><body>"+chatStrings.join("<br>")+"</body></html>");

    afterContentAdd();

}

void HWChatWidget::onServerMessage(const QString& str)

{

    beforeContentAdd();

    if (chatStrings.size() > 250)

        chatStrings.removeFirst();

    chatStrings.append("<hr>" + str + "<hr>");

    chatText->setHtml("<html><body>"+chatStrings.join("<br>")+"</body></html>");

    afterContentAdd();

}

void HWChatWidget::nickAdded(const QString & nick, bool notifyNick)

{

    QSortFilterProxyModel * playersSortFilterModel = qobject_cast<QSortFilterProxyModel *>(chatNicks->model());

    if(!playersSortFilterModel)

        return;