/****************************************************************
 * This file is distributed under the following license:
 *
 * Copyright (c) 2010, RetroShare Team
 *
 *  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 <rshare.h>
#include <retroshare/rsstatus.h>
#include <retroshare/rspeers.h>
#include <retroshare/rsmsgs.h>

#include "gui/notifyqt.h"
#include "gui/common/AvatarDefs.h"
#include "util/misc.h"

#include "AvatarWidget.h"
#include "ui_AvatarWidget.h"

#include <algorithm>

AvatarWidget::AvatarWidget(QWidget *parent) :
	QLabel(parent), ui(new Ui::AvatarWidget)
{
	ui->setupUi(this);

	mFlag.isOwnId = false;
//	mFlag.isGpg = false;
	defaultAvatar = ":/images/no_avatar_background.png";

	setFrameType(NO_FRAME);

	/* connect signals */
	connect(NotifyQt::getInstance(), SIGNAL(peerHasNewAvatar(const QString&)), this, SLOT(updateAvatar(const QString&)));
	connect(NotifyQt::getInstance(), SIGNAL(ownAvatarChanged()), this, SLOT(updateOwnAvatar()));
}

AvatarWidget::~AvatarWidget()
{
	delete ui;
}

QString AvatarWidget::frameState()
{
	switch (mFrameType)
	{
	case NO_FRAME:
		return "NOTHING";
	case NORMAL_FRAME:
		return "NORMAL";
	case STATUS_FRAME:
		switch (mPeerState)
		{
		case RS_STATUS_OFFLINE:
			return "OFFLINE";
		case RS_STATUS_INACTIVE:
			return "INACTIVE";
		case RS_STATUS_ONLINE:
			return "ONLINE";
		case RS_STATUS_AWAY:
			return "AWAY";
		case RS_STATUS_BUSY:
			return "BUSY";
		}
	}
	return "NOTHING";
}

void AvatarWidget::mouseReleaseEvent(QMouseEvent */*event*/)
{
	if (mFlag.isOwnId) {
		QByteArray ba;
		if (misc::getOpenAvatarPicture(this, ba)) {
			rsMsgs->setOwnAvatarData((unsigned char*)(ba.data()), ba.size()); // last char 0 included.
		}
	}
}

void AvatarWidget::setFrameType(FrameType type)
{
	mFrameType = type;

	switch (mFrameType) {
	case NO_FRAME:
		disconnect(NotifyQt::getInstance(), SIGNAL(peerStatusChanged(QString,int)), this, SLOT(updateStatus(const QString&, int)));
		break;
	case NORMAL_FRAME:
		disconnect(NotifyQt::getInstance(), SIGNAL(peerStatusChanged(QString,int)), this, SLOT(updateStatus(const QString&, int)));
		break;
	case STATUS_FRAME:
		connect(NotifyQt::getInstance(), SIGNAL(peerStatusChanged(QString,int)), this, SLOT(updateStatus(const QString&, int)));
		break;
	}

	refreshStatus();
    updateAvatar(QString::fromStdString(mId.toStdString()));
	Rshare::refreshStyleSheet(this, false);
}
void AvatarWidget::setId(const RsPeerId &id)
{
    mId = id;
//    mPgpId = rsPeers->getGPGId(id) ;
//    mFlag.isGpg = false ;

    if (mId == rsPeers->getOwnId()) {
        mFlag.isOwnId = true;
        setToolTip(tr("Click to change your avatar"));
    }

    setPixmap(QPixmap());

    if (id.isNull()) {
        setEnabled(false);
    }

    refreshStatus();
    updateAvatar(QString::fromStdString(mId.toStdString()));
}

void AvatarWidget::setOwnId()
{
    setId(rsPeers->getOwnId());
}

void AvatarWidget::setDefaultAvatar(const QString &avatar)
{
	defaultAvatar = avatar;
    updateAvatar(QString::fromStdString(mId.toStdString()));
}

void AvatarWidget::refreshStatus()
{
	switch (mFrameType) {
	case NO_FRAME:
	case NORMAL_FRAME:
	{
		Rshare::refreshStyleSheet(this, false);
		break;
	}
	case STATUS_FRAME:
	{
		StatusInfo statusInfo;

		if (mFlag.isOwnId) {
			rsStatus->getOwnStatus(statusInfo);
		} else {
			// No check of return value. Non existing status info is handled as offline.
			rsStatus->getStatus(mId, statusInfo);
		}
        updateStatus(QString::fromStdString(statusInfo.id.toStdString()), statusInfo.status);
		break;
	}
	}
}

void AvatarWidget::updateStatus(const QString peerId, int status)
{
	if (mFrameType != STATUS_FRAME) {
		return;
	}

    if (mId.isNull()) {
		mPeerState = status;
		Rshare::refreshStyleSheet(this, false);
	} else {
		/* set style for status */
        if (mId.toStdString() == peerId.toStdString()) {
			// the peers status has changed
			mPeerState = status;
			setEnabled(((uint32_t) status == RS_STATUS_OFFLINE) ? false : true);
			Rshare::refreshStyleSheet(this, false);
		}
	}
}

void AvatarWidget::updateAvatar(const QString &peerId)
{
    if (mId.isNull()) {
		QPixmap avatar(defaultAvatar);
		setPixmap(avatar);
		return;
	}

	if (mFlag.isOwnId) {
		QPixmap avatar;
		AvatarDefs::getOwnAvatar(avatar);
		setPixmap(avatar);
		return;
	}

    //if (mFlag.isGpg) {
    //	if (mId == peerId.toStdString()) {
    //		/* called from AvatarWidget with gpg id */
    //		QPixmap avatar;
    //		AvatarDefs::getAvatarFromGpgId(mId, avatar, defaultAvatar);
    //		setPixmap(avatar);
    //		return;
    //	}
    //
    //	/* Is this one of the ssl ids of the gpg id ? */
    //	std::list<std::string> sslIds;
    //	if (rsPeers->getAssociatedSSLIds(mId, sslIds) == false) {
    //		return;
    //	}
    //
    //	if (std::find(sslIds.begin(), sslIds.end(), peerId.toStdString()) != sslIds.end()) {
    //		/* One of the ssl ids of the gpg id */
    //		QPixmap avatar;
    //		AvatarDefs::getAvatarFromGpgId(mId, avatar, defaultAvatar);
    //		setPixmap(avatar);
    //	}
    //
    //	return;
    //}

    if (mId.toStdString() == peerId.toStdString()) {
		QPixmap avatar;
        AvatarDefs::getAvatarFromSslId(mId.toStdString(), avatar, defaultAvatar);
		setPixmap(avatar);
		return;
	}
}

void AvatarWidget::updateOwnAvatar()
{
	if (mFlag.isOwnId) {
        updateAvatar(QString::fromStdString(mId.toStdString()));
	}
}