/*
 * Hedgewars, a free turn based strategy game
 * Copyright (c) 2007 Igor Ulyanov <iulyanov@gmail.com>
 * Copyright (c) 2009 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
 * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA
 */

#include <QTextBrowser>
#include <QLineEdit>
#include <QAction>
#include <QApplication>
#include <QTextDocument>
#include <QDir>
#include <QSettings>
#include <QFile>
#include <QTextStream>

#include "hwconsts.h"
#include "SDLs.h"
#include "gameuiconfig.h"
#include "chatwidget.h"

HWChatWidget::HWChatWidget(QWidget* parent, QSettings * gameSettings, SDLInteraction * sdli, bool notify) :
  QWidget(parent),
  mainLayout(this)
{
    this->gameSettings = gameSettings;
    this->sdli = sdli;
    this->notify = notify;
    if(notify && gameSettings->value("audio/frontendsound", true).toBool()) {
       QDir tmpdir;

       tmpdir.cd(datadir->absolutePath());
       tmpdir.cd("Sounds/voices");
       sdli->SDLMusicInit();
       sound[0] = Mix_LoadWAV(QString(tmpdir.absolutePath() + "/Classic/Hello.ogg").toLocal8Bit().constData());
       sound[1] = Mix_LoadWAV(QString(tmpdir.absolutePath() + "/Default/Hello.ogg").toLocal8Bit().constData());
       sound[2] = Mix_LoadWAV(QString(tmpdir.absolutePath() + "/Mobster/Hello.ogg").toLocal8Bit().constData());
       sound[3] = Mix_LoadWAV(QString(tmpdir.absolutePath() + "/Russian/Hello.ogg").toLocal8Bit().constData());
    }

	mainLayout.setSpacing(1);
	mainLayout.setMargin(1);
	mainLayout.setSizeConstraint(QLayout::SetMinimumSize);
	mainLayout.setColumnStretch(0, 75);
	mainLayout.setColumnStretch(1, 25);

	chatEditLine = new QLineEdit(this);
	chatEditLine->setMaxLength(300);
	connect(chatEditLine, SIGNAL(returnPressed()), this, SLOT(returnPressed()));

	mainLayout.addWidget(chatEditLine, 1, 0, 1, 2);

	chatText = new QTextBrowser(this);
	chatText->setMinimumHeight(20);
	chatText->setMinimumWidth(10);
	chatText->setSizePolicy(QSizePolicy::Expanding, QSizePolicy::Expanding);
	chatText->setOpenExternalLinks(true);
	mainLayout.addWidget(chatText, 0, 0);

	chatNicks = new QListWidget(this);
	chatNicks->setMinimumHeight(10);
	chatNicks->setMinimumWidth(10);
	chatNicks->setSortingEnabled(true);
	chatNicks->setSizePolicy(QSizePolicy::Expanding, QSizePolicy::Expanding);
	chatNicks->setContextMenuPolicy(Qt::ActionsContextMenu);
	connect(chatNicks, SIGNAL(itemDoubleClicked(QListWidgetItem *)),
		this, SLOT(chatNickDoubleClicked(QListWidgetItem *)));
	connect(chatNicks, SIGNAL(currentRowChanged(int)),
		this, SLOT(chatNickSelected(int)));

	mainLayout.addWidget(chatNicks, 0, 1);

	acInfo = new QAction(QAction::tr("Info"), chatNicks);
	connect(acInfo, SIGNAL(triggered(bool)), this, SLOT(onInfo()));
	acKick = new QAction(QAction::tr("Kick"), chatNicks);
	connect(acKick, SIGNAL(triggered(bool)), this, SLOT(onKick()));
	acBan = new QAction(QAction::tr("Ban"), chatNicks);
	connect(acBan, SIGNAL(triggered(bool)), this, SLOT(onBan()));
	acFollow = new QAction(QAction::tr("Follow"), chatNicks);
	connect(acFollow, SIGNAL(triggered(bool)), this, SLOT(onFollow()));
	acIgnore = new QAction(QAction::tr("Ignore"), chatNicks);
	connect(acIgnore, SIGNAL(triggered(bool)), this, SLOT(onIgnore()));
	acFriend = new QAction(QAction::tr("Add friend"), chatNicks);
	connect(acFriend, SIGNAL(triggered(bool)), this, SLOT(onFriend()));

	chatNicks->insertAction(0, acInfo);
	chatNicks->insertAction(0, acFollow);
	chatNicks->insertAction(0, acIgnore);
	chatNicks->insertAction(0, acFriend);
	
	showReady = false;
}

void HWChatWidget::loadList(QStringList & list, const QString & file)
{
	list.clear();
	QFile txt((cfgdir->absolutePath() + "/" + file).toLocal8Bit().constData());
	if(!txt.open(QIODevice::ReadOnly))
		return;
	QTextStream stream(&txt);
	stream.setCodec("UTF-8");

	while(!stream.atEnd())
	{
		QString str = stream.readLine();
		if(str.startsWith(";") || str.length() == 0)
			continue;
		list << str.trimmed();
	}
	list.removeDuplicates();
	txt.close();
}

void HWChatWidget::saveList(QStringList & list, const QString & file)
{
	QFile txt((cfgdir->absolutePath() + "/" + file).toLocal8Bit().constData());
	if(!txt.open(QIODevice::WriteOnly | QIODevice::Truncate))
		return;
	QTextStream stream(&txt);
	stream.setCodec("UTF-8");

	stream << "; this list is used by Hedgewars - do not edit it unless you know what you're doing!" << endl;
	for(int i = 0; i < list.size(); i++)
		stream << list[i] << endl;
	txt.close();
}

void HWChatWidget::updateIcon(QListWidgetItem *item)
{
	QString nick = item->text();

	if(ignoreList.contains(nick, Qt::CaseInsensitive))
	{
		item->setIcon(QIcon(showReady ? (item->data(Qt::UserRole).toBool() ? ":/res/chat_ignore_on" : ":/res/chat_ignore_off") : ":/res/chat_ignore.png"));
		item->setForeground(Qt::gray);
	}
	else if(friendsList.contains(nick, Qt::CaseInsensitive))
	{
		item->setIcon(QIcon(showReady ? (item->data(Qt::UserRole).toBool() ? ":/res/chat_friend_on" : ":/res/chat_friend_off") : ":/res/chat_friend.png"));
		item->setForeground(Qt::green);
	}
	else
	{
		item->setIcon(QIcon(showReady ? (item->data(Qt::UserRole).toBool() ? ":/res/chat_default_on" : ":/res/chat_default_off") : ":/res/chat_default.png"));
		item->setForeground(QBrush(QColor(0xff, 0xcc, 0x00)));
	}
}

void HWChatWidget::updateIcons()
{
	for(int i = 0; i < chatNicks->count(); i++)
		updateIcon(chatNicks->item(i));
}

void HWChatWidget::loadLists(const QString & nick)
{
	loadList(ignoreList, nick.toLower() + "_ignore.txt");
	loadList(friendsList, nick.toLower() + "_friends.txt");
	updateIcons();
}

void HWChatWidget::saveLists(const QString & nick)
{
	saveList(ignoreList, nick.toLower() + "_ignore.txt");
	saveList(friendsList, nick.toLower() + "_friends.txt");
}

void HWChatWidget::returnPressed()
{
	emit chatLine(chatEditLine->text());
	chatEditLine->clear();
}

void HWChatWidget::onChatString(const QString& str)
{
	if (chatStrings.size() > 250)
		chatStrings.removeFirst();

	QString formattedStr = Qt::escape(str.mid(1));
	QStringList parts = formattedStr.split(QRegExp("\\W+"), QString::SkipEmptyParts);

	if (!formattedStr.startsWith(" ***")) // don't ignore status messages
	{
		if (formattedStr.startsWith(" *")) // emote
			parts[0] = parts[1];
		if(parts.size() > 0 && ignoreList.contains(parts[0], Qt::CaseInsensitive))
			return;
	}

	QString color("");
	bool isFriend = friendsList.contains(parts[0], Qt::CaseInsensitive);
	
	if (str.startsWith("\x03"))
		color = QString("#c0c0c0");
	else if (str.startsWith("\x02"))
		color = QString(isFriend ? "#00ff00" : "#ff00ff");
	else if (isFriend)
		color = QString("#00c000");

	if(color.compare("") != 0)
		formattedStr = QString("<font color=\"%2\">%1</font>").arg(formattedStr).arg(color);

	chatStrings.append(formattedStr);

	chatText->setHtml(chatStrings.join("<br>"));

	chatText->moveCursor(QTextCursor::End);
}

void HWChatWidget::onServerMessage(const QString& str)
{
	if (chatStrings.size() > 250)
		chatStrings.removeFirst();

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

	chatText->setHtml(chatStrings.join("<br>"));

	chatText->moveCursor(QTextCursor::End);
}

void HWChatWidget::nickAdded(const QString& nick, bool notifyNick)
{
	QListWidgetItem * item = new QListWidgetItem(nick);
	updateIcon(item);
	chatNicks->addItem(item);

    if(notifyNick && notify && gameSettings->value("audio/frontendsound", true).toBool()) {
       Mix_PlayChannel(-1, sound[rand()%4], 0);
    }
}

void HWChatWidget::nickRemoved(const QString& nick)
{
	QList<QListWidgetItem *> items = chatNicks->findItems(nick, Qt::MatchExactly);
	for(QList<QListWidgetItem *>::iterator it=items.begin(); it!=items.end();) {
		chatNicks->takeItem(chatNicks->row(*it));
		++it;
	}
}

void HWChatWidget::clear()
{
	chatText->clear();
	chatStrings.clear();
	chatNicks->clear();
}

void HWChatWidget::onKick()
{
	QListWidgetItem * curritem = chatNicks->currentItem();
	if (curritem)
		emit kick(curritem->text());
}

void HWChatWidget::onBan()
{
	QListWidgetItem * curritem = chatNicks->currentItem();
	if (curritem)
		emit ban(curritem->text());
}

void HWChatWidget::onInfo()
{
	QListWidgetItem * curritem = chatNicks->currentItem();
	if (curritem)
		emit info(curritem->text());
}

void HWChatWidget::onFollow()
{
	QListWidgetItem * curritem = chatNicks->currentItem();
	if (curritem)
		emit follow(curritem->text());
}

void HWChatWidget::onIgnore()
{
	QListWidgetItem * curritem = chatNicks->currentItem();
	if(!curritem)
		return;

	if(ignoreList.contains(curritem->text(), Qt::CaseInsensitive)) // already on list - remove him
	{
		ignoreList.removeAll(curritem->text().toLower());
		onChatString(HWChatWidget::tr("%1 *** %2 has been removed from your ignore list").arg('\x03').arg(curritem->text()));
	}
	else // not on list - add
	{
		ignoreList << curritem->text().toLower();
		onChatString(HWChatWidget::tr("%1 *** %2 has been added to your ignore list").arg('\x03').arg(curritem->text()));
	}
	updateIcon(curritem); // update icon
	chatNickSelected(0); // update context menu
}

void HWChatWidget::onFriend()
{
	QListWidgetItem * curritem = chatNicks->currentItem();
	if(!curritem)
		return;

	if(friendsList.contains(curritem->text(), Qt::CaseInsensitive)) // already on list - remove him
	{
		friendsList.removeAll(curritem->text().toLower());
		onChatString(HWChatWidget::tr("%1 *** %2 has been removed from your friends list").arg('\x03').arg(curritem->text()));
	}
	else // not on list - add
	{
		friendsList << curritem->text().toLower();
		onChatString(HWChatWidget::tr("%1 *** %2 has been added to your friends list").arg('\x03').arg(curritem->text()));
	}
	updateIcon(curritem); // update icon
	chatNickSelected(0); // update context menu
}

void HWChatWidget::chatNickDoubleClicked(QListWidgetItem * item)
{
	if (item) onFollow();
}

void HWChatWidget::chatNickSelected(int index)
{
	QListWidgetItem* item = chatNicks->currentItem();
	if (!item)
		return;

	// update context menu labels according to possible action
	if(ignoreList.contains(item->text(), Qt::CaseInsensitive))
		acIgnore->setText(QAction::tr("Unignore"));
	else
		acIgnore->setText(QAction::tr("Ignore"));

	if(friendsList.contains(item->text(), Qt::CaseInsensitive))
		acFriend->setText(QAction::tr("Remove friend"));
	else
		acFriend->setText(QAction::tr("Add friend"));
}

void HWChatWidget::setShowReady(bool s)
{
	showReady = s;
}

void HWChatWidget::setReadyStatus(const QString & nick, bool isReady)
{
	QList<QListWidgetItem *> items = chatNicks->findItems(nick, Qt::MatchExactly);
	if (items.size() != 1)
	{
		qWarning("Bug: cannot find user in chat");
		return;
	}

	items[0]->setData(Qt::UserRole, isReady); // bulb status
	updateIcon(items[0]);

	// ensure we're still showing the status bulbs
	showReady = true;
}

void HWChatWidget::adminAccess(bool b)
{
	chatNicks->removeAction(acKick);
	chatNicks->removeAction(acBan);

	if(b)
	{
		chatNicks->insertAction(0, acKick);
//		chatNicks->insertAction(0, acBan);
	}
}