/****************************************************************
 *  RetroShare is distributed under the following license:
 *
 *  Copyright (C) 2008 Robert Fernie
 *
 *  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; either version 2
 *  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
 *  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., 51 Franklin Street, Fifth Floor,
 *  Boston, MA  02110-1301, USA.
 ****************************************************************/
#include <QtGui>

#include <iostream>
#include <algorithm>

#include "rsiface/rschannels.h"

#include "ChannelFeed.h"
#include "gui/feeds/ChanGroupItem.h"
#include "gui/feeds/ChanMenuItem.h"
#include "gui/feeds/ChanMsgItem.h"

#include "gui/forums/CreateForum.h"

#include "gui/ChanGroupDelegate.h"

#include "GeneralMsgDialog.h"

/****
 * #define CHAN_DEBUG
 ***/

/** Constructor */
ChannelFeed::ChannelFeed(QWidget *parent)
: MainPage (parent)
{
  	/* Invoke the Qt Designer generated object setup routine */
  	setupUi(this);

  	connect(chanButton, SIGNAL(clicked()), this, SLOT(createChannel()));
  	connect(postButton, SIGNAL(clicked()), this, SLOT(sendMsg()));
  	connect(subscribeButton, SIGNAL( clicked( void ) ), this, SLOT( subscribeChannel ( void ) ) );
  	connect(unsubscribeButton, SIGNAL( clicked( void ) ), this, SLOT( unsubscribeChannel ( void ) ) );

	/*************** Setup Left Hand Side (List of Channels) ****************/

//	mGroupLayout = new QVBoxLayout;
//	mGroupLayout->setSpacing(0);
//	mGroupLayout->setMargin(0);
//	mGroupLayout->setContentsMargins(0,0,0,0);
//
//	mGroupOwn = new ChanGroupItem("Own Channels");
//	mGroupSub = new ChanGroupItem("Subscribed Channels");
//	mGroupPop = new ChanGroupItem("Popular Channels");
//	mGroupOther = new ChanGroupItem("Other Channels");
//
//	mGroupLayout->addWidget(mGroupOwn);
//	mGroupLayout->addWidget(mGroupSub);
//	mGroupLayout->addWidget(mGroupPop);
//	mGroupLayout->addWidget(mGroupOther);
//
//
//	QWidget *middleWidget = new QWidget();
//	//middleWidget->setSizePolicy( QSizePolicy::Policy::Maximum, QSizePolicy::Policy::Minimum);
//	middleWidget->setSizePolicy(QSizePolicy::Maximum, QSizePolicy::Minimum);
//	middleWidget->setLayout(mGroupLayout);
//
//     	QScrollArea *scrollArea = new QScrollArea;
//        //scrollArea->setBackgroundRole(QPalette::Dark);
//	scrollArea->setWidget(middleWidget);
//	scrollArea->setWidgetResizable(true);
//	scrollArea->setHorizontalScrollBarPolicy( Qt::ScrollBarAlwaysOff );
//
//	QVBoxLayout *layout2 = new QVBoxLayout;
//	layout2->addWidget(scrollArea);
//	layout2->setSpacing(0);
//	layout2->setMargin(0);
//	layout2->setContentsMargins(0,0,0,0);
//
//
//     	chanFrame->setLayout(layout2);
//
	/*************** Setup Right Hand Side (List of Messages) ****************/

	mMsgLayout = new QVBoxLayout;
	mMsgLayout->setSpacing(0);
	mMsgLayout->setMargin(0);
	mMsgLayout->setContentsMargins(0,0,0,0);

	QWidget *middleWidget2 = new QWidget();
	middleWidget2->setSizePolicy(QSizePolicy::Maximum, QSizePolicy::Minimum);
	middleWidget2->setLayout(mMsgLayout);

	QScrollArea *scrollArea2 = new QScrollArea;
	//scrollArea2->setBackgroundRole(QPalette::Dark);
	scrollArea2->setWidget(middleWidget2);
	scrollArea2->setWidgetResizable(true);
	scrollArea2->setHorizontalScrollBarPolicy( Qt::ScrollBarAlwaysOff );

	QVBoxLayout *layout3 = new QVBoxLayout;
	layout3->addWidget(scrollArea2);
	layout3->setSpacing(0);
	layout3->setMargin(0);
	layout3->setContentsMargins(0,0,0,0);


    msgFrame->setLayout(layout3);

//	mChannelId = "OWNID";

//	updateChannelList();
//
//	QTimer *timer = new QTimer(this);
//	timer->connect(timer, SIGNAL(timeout()), this, SLOT(checkUpdate()));
//	timer->start(1000);

  	mChannelId = "";
	model = new QStandardItemModel(0, 2, this);
	model->setHeaderData(0, Qt::Horizontal, tr("Name"), Qt::DisplayRole);
	model->setHeaderData(1, Qt::Horizontal, tr("ID"), Qt::DisplayRole);

	treeView->setModel(model);
	treeView->setEditTriggers(QAbstractItemView::NoEditTriggers);

	treeView->setItemDelegate(new ChanGroupDelegate());
	treeView->setRootIsDecorated(false);

	// hide header and id column
	treeView->setHeaderHidden(true);
	treeView->hideColumn(1);

	QStandardItem *item1 = new QStandardItem("Own Channels");
	QStandardItem *item2 = new QStandardItem("Subscribed Channels");
	QStandardItem *item3 = new QStandardItem("Popular Channels");
	QStandardItem *item4 = new QStandardItem("Other Channels");

	model->appendRow(item1);
	model->appendRow(item2);
	model->appendRow(item3);
	model->appendRow(item4);

	connect(treeView, SIGNAL(activated(const QModelIndex &)), this, SLOT(selectChannel(const QModelIndex &)));
	connect(treeView, SIGNAL(activated(const QModelIndex &)), this, SLOT(toggleSelection(const QModelIndex &)));

	//added from ahead
	updateChannelList();

	mChannelFont = QFont("MS SANS SERIF", 24);
    nameLabel->setFont(mChannelFont);

	QTimer *timer = new QTimer(this);
	timer->connect(timer, SIGNAL(timeout()), this, SLOT(checkUpdate()));
	timer->start(1000);
}


void ChannelFeed::createChannel()
{
	CreateForum *cf = new CreateForum(NULL, false);

	cf->setWindowTitle(tr("Create a new Channel"));
	cf->ui.labelicon->setPixmap(QPixmap(":/images/add_channel64.png"));
	QString titleStr("<span style=\"font-size:16pt; font-weight:500;"
                               "color:#32cd32;\">%1</span>");
	cf->ui.textlabelcreatforums->setText( titleStr.arg( tr("Create a new Channel") ) ) ;
	cf->show();
}

void ChannelFeed::channelSelection()
{
	/* which item was selected? */


	/* update mChannelId */

	updateChannelMsgs();
}

void ChannelFeed::sendMsg()
{
#ifdef CHAN_DEBUG
	std::cerr << "ChannelFeed::sendMsg()";
	std::cerr << std::endl;
#endif

	if (mChannelId != "")
	{
		openMsg(FEEDHOLDER_MSG_CHANNEL, mChannelId, "");
	}
	else
	{
#ifdef CHAN_DEBUG
		std::cerr << "ChannelFeed::sendMsg() no Channel Selected!";
		std::cerr << std::endl;
#endif
	}

}


/*************************************************************************************/
/*************************************************************************************/
/*************************************************************************************/

void ChannelFeed::deleteFeedItem(QWidget *item, uint32_t type)
{
	return;
}

void ChannelFeed::openChat(std::string peerId)
{
	return;
}


void ChannelFeed::openMsg(uint32_t type, std::string grpId, std::string inReplyTo)
{
#ifdef CHAN_DEBUG
	std::cerr << "ChannelFeed::openMsg()";
	std::cerr << std::endl;
#endif
	GeneralMsgDialog *msgDialog = new GeneralMsgDialog(NULL);


	msgDialog->addDestination(type, grpId, inReplyTo);

	msgDialog->show();
	return;
}

void ChannelFeed::selectChannel( std::string cId)
{
	mChannelId = cId;

	updateChannelMsgs();
}

void ChannelFeed::selectChannel(const QModelIndex &index)
{
	int row = index.row();
	int col = index.column();
	if (col != 1) {
		QModelIndex sibling = index.sibling(row, 1);
		if (sibling.isValid())
			mChannelId = sibling.data().toString().toStdString();
	} else
		mChannelId = index.data().toString().toStdString();
	updateChannelMsgs();
}

void ChannelFeed::checkUpdate()
{
	std::list<std::string> chanIds;
	std::list<std::string>::iterator it;
	if (!rsChannels)
		return;

	if (rsChannels->channelsChanged(chanIds))
	{
		/* update Forums List */
		updateChannelList();

		it = std::find(chanIds.begin(), chanIds.end(), mChannelId);
		if (it != chanIds.end())
		{
			updateChannelMsgs();
		}
	}
}


void ChannelFeed::updateChannelList()
{

	std::list<ChannelInfo> channelList;
	std::list<ChannelInfo>::iterator it;
	if (!rsChannels)
	{
		return;
	}

	rsChannels->getChannelList(channelList);

	/* get the ids for our lists */
	std::list<std::string> adminIds;
	std::list<std::string> subIds;
	std::list<std::string> popIds;
	std::list<std::string> otherIds;
	std::multimap<uint32_t, std::string> popMap;

	for(it = channelList.begin(); it != channelList.end(); it++)
	{
		/* sort it into Publish (Own), Subscribed, Popular and Other */
		uint32_t flags = it->channelFlags;

		if (flags & RS_DISTRIB_ADMIN)
		{
			adminIds.push_back(it->channelId);
		}
		else if (flags & RS_DISTRIB_SUBSCRIBED)
		{
			subIds.push_back(it->channelId);
		}
		else
		{
			/* rate the others by popularity */
			popMap.insert(std::make_pair(it->pop, it->channelId));
		}
	}

	/* iterate backwards through popMap - take the top 5 or 10% of list */
	uint32_t popCount = 5;
	if (popCount < popMap.size() / 10)
	{
		popCount = popMap.size() / 10;
	}

	uint32_t i = 0;
	uint32_t popLimit = 0;
	std::multimap<uint32_t, std::string>::reverse_iterator rit;
	for(rit = popMap.rbegin(); ((rit != popMap.rend()) && (i < popCount)); rit++, i++)
	{
		popIds.push_back(rit->second);
	}

	if (rit != popMap.rend())
	{
		popLimit = rit->first;
	}

	for(it = channelList.begin(); it != channelList.end(); it++)
	{
		/* ignore the ones we've done already */
		uint32_t flags = it->channelFlags;

		if (flags & RS_DISTRIB_ADMIN)
		{
			continue;
		}
		else if (flags & RS_DISTRIB_SUBSCRIBED)
		{
			continue;
		}
		else
		{
			if (it->pop < popLimit)
			{
				otherIds.push_back(it->channelId);
			}
		}
	}

	/* now we have our lists ---> update entries */

	updateChannelListOwn(adminIds);
	updateChannelListSub(subIds);
	updateChannelListPop(popIds);
	updateChannelListOther(otherIds);
}

void ChannelFeed::updateChannelListOwn(std::list<std::string> &ids)
{
//	std::list<ChanMenuItem *>::iterator it;
	std::list<std::string>::iterator iit;

//	/* TEMP just replace all of them */
//	for(it = mChannelListOwn.begin(); it != mChannelListOwn.end(); it++)
//	{
//		delete (*it);
//	}
//	mChannelListOwn.clear();
//
//	int topIndex = mGroupLayout->indexOf(mGroupOwn);
//	int index = topIndex + 1;
//	for (iit = ids.begin(); iit != ids.end(); iit++, index++)
//	{
//#ifdef CHAN_DEBUG
//		std::cerr << "ChannelFeed::updateChannelListOwn(): " << *iit << " at: " << index;
//		std::cerr << std::endl;
//#endif
//
//		ChanMenuItem *cmi = new ChanMenuItem(*iit);
//		mChannelListOwn.push_back(cmi);
//		mGroupLayout->insertWidget(index, cmi);
//
//		connect(cmi, SIGNAL( selectMe( std::string )), this, SLOT( selectChannel( std::string )));
//	}


	/* remove rows with groups before adding new ones */
	model->item(OWN)->removeRows(0, model->item(OWN)->rowCount());

	for (iit = ids.begin(); iit != ids.end(); iit ++) {
#ifdef CHAN_DEBUG
		std::cerr << "ChannelFeed::updateChannelListOwn(): " << *iit << std::endl;
#endif
		QStandardItem *ownGroup = model->item(OWN);
		QList<QStandardItem *> channel;
		QStandardItem *item1 = new QStandardItem();
		QStandardItem *item2 = new QStandardItem();

		ChannelInfo ci;
		if (rsChannels && rsChannels->getChannelInfo(*iit, ci)) {
			item1->setData(QVariant(QString::fromStdWString(ci.channelName)), Qt::DisplayRole);
			item2->setData(QVariant(QString::fromStdString(ci.channelId)), Qt::DisplayRole);
			item1->setToolTip(tr("Popularity: %1\nFetches: %2\nAvailable: %3"
					).arg(QString::number(ci.pop)).arg(9999).arg(9999));
		} else {
			item1->setData(QVariant(QString("Unknown Channel")), Qt::DisplayRole);
			item2->setData(QVariant(QString::fromStdString(*iit)), Qt::DisplayRole);
			item1->setToolTip("Unknown Channel\nNo Description");
		}

		channel.append(item1);
		channel.append(item2);
		ownGroup->appendRow(channel);
	}
}

void ChannelFeed::updateChannelListSub(std::list<std::string> &ids)
{
//	std::list<ChanMenuItem *>::iterator it;
	std::list<std::string>::iterator iit;

//	/* TEMP just replace all of them */
//	for(it = mChannelListSub.begin(); it != mChannelListSub.end(); it++)
//	{
//		delete (*it);
//	}
//	mChannelListSub.clear();
//
//	int topIndex = mGroupLayout->indexOf(mGroupSub);
//	int index = topIndex + 1;
//	for (iit = ids.begin(); iit != ids.end(); iit++, index++)
//	{
//#ifdef CHAN_DEBUG
//		std::cerr << "ChannelFeed::updateChannelListSub(): " << *iit << " at: " << index;
//		std::cerr << std::endl;
//#endif
//
//		ChanMenuItem *cmi = new ChanMenuItem(*iit);
//		mChannelListSub.push_back(cmi);
//		mGroupLayout->insertWidget(index, cmi);
//		connect(cmi, SIGNAL( selectMe( std::string )), this, SLOT( selectChannel( std::string )));
//	}

	/* remove rows with groups before adding new ones */
	model->item(SUBSCRIBED)->removeRows(0, model->item(SUBSCRIBED)->rowCount());

	for (iit = ids.begin(); iit != ids.end(); iit ++) {
#ifdef CHAN_DEBUG
		std::cerr << "ChannelFeed::updateChannelListSub(): " << *iit << std::endl;
#endif
		QStandardItem *ownGroup = model->item(SUBSCRIBED);
		QList<QStandardItem *> channel;
		QStandardItem *item1 = new QStandardItem();
		QStandardItem *item2 = new QStandardItem();

		ChannelInfo ci;
		if (rsChannels && rsChannels->getChannelInfo(*iit, ci)) {
			item1->setData(QVariant(QString::fromStdWString(ci.channelName)), Qt::DisplayRole);
			item2->setData(QVariant(QString::fromStdString(ci.channelId)), Qt::DisplayRole);
			item1->setToolTip(tr("Popularity: %1\nFetches: %2\nAvailable: %3"
					).arg(QString::number(ci.pop)).arg(9999).arg(9999));
		} else {
			item1->setData(QVariant(QString("Unknown Channel")), Qt::DisplayRole);
			item2->setData(QVariant(QString::fromStdString(*iit)), Qt::DisplayRole);
			item1->setToolTip("Unknown Channel\nNo Description");
		}

		channel.append(item1);
		channel.append(item2);
		ownGroup->appendRow(channel);
	}

}

void ChannelFeed::updateChannelListPop(std::list<std::string> &ids)
{
//	std::list<ChanMenuItem *>::iterator it;
	std::list<std::string>::iterator iit;

//	/* TEMP just replace all of them */
//	for(it = mChannelListPop.begin(); it != mChannelListPop.end(); it++)
//	{
//		delete (*it);
//	}
//	mChannelListPop.clear();
//
//	int topIndex = mGroupLayout->indexOf(mGroupPop);
//	int index = topIndex + 1;
//	for (iit = ids.begin(); iit != ids.end(); iit++, index++)
//	{
//#ifdef CHAN_DEBUG
//		std::cerr << "ChannelFeed::updateChannelListPop(): " << *iit << " at: " << index;
//		std::cerr << std::endl;
//#endif
//
//		ChanMenuItem *cmi = new ChanMenuItem(*iit);
//		mChannelListPop.push_back(cmi);
//		mGroupLayout->insertWidget(index, cmi);
//		connect(cmi, SIGNAL( selectMe( std::string )), this, SLOT( selectChannel( std::string )));
//	}

	/* remove rows with groups before adding new ones */
	model->item(POPULAR)->removeRows(0, model->item(POPULAR)->rowCount());

	for (iit = ids.begin(); iit != ids.end(); iit ++) {
#ifdef CHAN_DEBUG
		std::cerr << "ChannelFeed::updateChannelListPop(): " << *iit << std::endl;
#endif
		QStandardItem *ownGroup = model->item(POPULAR);
		QList<QStandardItem *> channel;
		QStandardItem *item1 = new QStandardItem();
		QStandardItem *item2 = new QStandardItem();

		ChannelInfo ci;
		if (rsChannels && rsChannels->getChannelInfo(*iit, ci)) {
			item1->setData(QVariant(QString::fromStdWString(ci.channelName)), Qt::DisplayRole);
			item2->setData(QVariant(QString::fromStdString(ci.channelId)), Qt::DisplayRole);
			item1->setToolTip(tr("Popularity: %1\nFetches: %2\nAvailable: %3"
					).arg(QString::number(ci.pop)).arg(9999).arg(9999));
		} else {
			item1->setData(QVariant(QString("Unknown Channel")), Qt::DisplayRole);
			item2->setData(QVariant(QString::fromStdString(*iit)), Qt::DisplayRole);
			item1->setToolTip("Unknown Channel\nNo Description");
		}

		channel.append(item1);
		channel.append(item2);
		ownGroup->appendRow(channel);
	}
}

void ChannelFeed::updateChannelListOther(std::list<std::string> &ids)
{
//	std::list<ChanMenuItem *>::iterator it;
	std::list<std::string>::iterator iit;

//	/* TEMP just replace all of them */
//	for(it = mChannelListOther.begin(); it != mChannelListOther.end(); it++)
//	{
//		delete (*it);
//	}
//	mChannelListOther.clear();
//
//	int topIndex = mGroupLayout->indexOf(mGroupOther);
//	int index = topIndex + 1;
//	for (iit = ids.begin(); iit != ids.end(); iit++, index++)
//	{
//#ifdef CHAN_DEBUG
//		std::cerr << "ChannelFeed::updateChannelListOther(): " << *iit << " at: " << index;
//		std::cerr << std::endl;
//#endif
//
//		ChanMenuItem *cmi = new ChanMenuItem(*iit);
//		mChannelListOther.push_back(cmi);
//		mGroupLayout->insertWidget(index, cmi);
//		connect(cmi, SIGNAL( selectMe( std::string )), this, SLOT( selectChannel( std::string )));
//	}

	/* remove rows with groups before adding new ones */
	model->item(OTHER)->removeRows(0, model->item(OTHER)->rowCount());

	for (iit = ids.begin(); iit != ids.end(); iit ++) {
#ifdef CHAN_DEBUG
		std::cerr << "ChannelFeed::updateChannelListOther(): " << *iit << std::endl;
#endif
		QStandardItem *ownGroup = model->item(OTHER);
		QList<QStandardItem *> channel;
		QStandardItem *item1 = new QStandardItem();
		QStandardItem *item2 = new QStandardItem();

		ChannelInfo ci;
		if (rsChannels && rsChannels->getChannelInfo(*iit, ci)) {
			item1->setData(QVariant(QString::fromStdWString(ci.channelName)), Qt::DisplayRole);
			item2->setData(QVariant(QString::fromStdString(ci.channelId)), Qt::DisplayRole);
			item1->setToolTip(tr("Popularity: %1\nFetches: %2\nAvailable: %3"
					).arg(QString::number(ci.pop)).arg(9999).arg(9999));
		} else {
			item1->setData(QVariant(QString("Unknown Channel")), Qt::DisplayRole);
			item2->setData(QVariant(QString::fromStdString(*iit)), Qt::DisplayRole);
			item1->setToolTip("Unknown Channel\nNo Description");
		}

		channel.append(item1);
		channel.append(item2);
		ownGroup->appendRow(channel);
	}
}

void ChannelFeed::updateChannelMsgs()
{
	if (!rsChannels)
		return;

	ChannelInfo ci;
	if (!rsChannels->getChannelInfo(mChannelId, ci))
	{
		postButton->setEnabled(false);
		subscribeButton->setEnabled(false);
		unsubscribeButton->setEnabled(false);
		nameLabel->setText("No Channel Selected");
		return;
	}
	/* set channel name */
	nameLabel->setText(QString::fromStdWString(ci.channelName));

	/* do buttons */
	if (ci.channelFlags & RS_DISTRIB_SUBSCRIBED)
	{
		subscribeButton->setEnabled(false);
		unsubscribeButton->setEnabled(true);
	}
	else
	{
		subscribeButton->setEnabled(true);
		unsubscribeButton->setEnabled(false);
	}

	if (ci.channelFlags & RS_DISTRIB_PUBLISH)
	{
		postButton->setEnabled(true);
	}
	else
	{
		postButton->setEnabled(false);
	}

	/* replace all the messages with new ones */
	std::list<ChanMsgItem *>::iterator mit;
	for(mit = mChanMsgItems.begin(); mit != mChanMsgItems.end(); mit++)
	{
		delete (*mit);
	}
	mChanMsgItems.clear();

	std::list<ChannelMsgSummary> msgs;
	std::list<ChannelMsgSummary>::iterator it;

	rsChannels->getChannelMsgList(mChannelId, msgs);

	for(it = msgs.begin(); it != msgs.end(); it++)
	{
		ChanMsgItem *cmi = new ChanMsgItem(this, 0, mChannelId, it->msgId, true);

		mChanMsgItems.push_back(cmi);
		mMsgLayout->addWidget(cmi);
	}
}


void ChannelFeed::unsubscribeChannel()
{
#ifdef CHAN_DEBUG
	std::cerr << "ChannelFeed::unsubscribeChannel()";
	std::cerr << std::endl;
#endif
	if (rsChannels)
	{
		rsChannels->channelSubscribe(mChannelId, false);
	}
	updateChannelMsgs();
}


void ChannelFeed::subscribeChannel()
{
#ifdef CHAN_DEBUG
	std::cerr << "ChannelFeed::subscribeChannel()";
	std::cerr << std::endl;
#endif
	if (rsChannels)
	{
		rsChannels->channelSubscribe(mChannelId, true);
	}
	updateChannelMsgs();
}


void ChannelFeed::toggleSelection(const QModelIndex &index)
{
	QItemSelectionModel *selectionModel = treeView->selectionModel();
	if (index.child(0, 0).isValid())
		selectionModel->select(index, QItemSelectionModel::Toggle);
}