/*
 * Copyright (c) 2011-2014 Isode Limited.
 * All rights reserved.
 * See the COPYING file for more information.
 */



#include "QtAvatarWidget.h"

#include <QLabel>
#include <QVBoxLayout>
#include <QPixmap>
#include <QMenu>
#include <QAction>
#include <QMouseEvent>
#include <QFileDialog>
#include <QImageReader>
#include <QBuffer>
#include <QMessageBox>
#include <QPainter>

#include <QtSwiftUtil.h>
#include <Swiften/Base/Path.h>

namespace Swift {

QtAvatarWidget::QtAvatarWidget(QWidget* parent) : QWidget(parent), editable(false) {
	QVBoxLayout* layout = new QVBoxLayout(this);
	layout->setContentsMargins(0,0,0,0);

	QSizePolicy sp(QSizePolicy::Fixed, QSizePolicy::Fixed);
	sp.setHorizontalStretch(0);
	sp.setVerticalStretch(0);
	setSizePolicy(sp);
	setMinimumSize(QSize(96, 96));
	setMaximumSize(QSize(96, 96));

	label = new QLabel(this);
	label->setWordWrap(true);
	label->setSizePolicy(sp);
	label->setMinimumSize(QSize(96, 96));
	label->setMaximumSize(QSize(96, 96));
	label->setAlignment(Qt::AlignCenter);
	layout->addWidget(label);
}

void QtAvatarWidget::setAvatar(const ByteArray& data, const std::string& type) {
	this->data = data;
	this->type = type;

	QImage image;
	if (!data.empty()) {
		image.loadFromData(reinterpret_cast<const uchar*>(vecptr(data)), data.size());
	}

	if (image.isNull()) {
		image = QImage(":/icons/no-avatar.png");
		QPainter painter(&image);
		painter.setPen(Qt::gray);
		QFont font = painter.font();
		font.setPointSize(14);
		painter.setFont(font);
		painter.drawText(0, 0, image.height(), image.width(), Qt::AlignHCenter | Qt::AlignVCenter, tr("No picture"));
	}

	if (image.height() > label->height() || image.width() > label->width()) {
		image = image.scaled(label->size(), Qt::KeepAspectRatio, Qt::SmoothTransformation);
	}
	label->setPixmap(QPixmap::fromImage(image));
}

void QtAvatarWidget::mousePressEvent(QMouseEvent* event) {
	if (!editable) {
		return;
	}
	QMenu menu;

	QAction* selectPicture = new QAction(tr("Select picture ..."), this);
	menu.addAction(selectPicture);

	QAction* clearPicture = new QAction(tr("Clear picture"), this);
	menu.addAction(clearPicture);

	QAction* result = menu.exec(event->globalPos());
	if (result == selectPicture) {
		QString fileName = QFileDialog::getOpenFileName(this, tr("Select picture"), "", tr("Image Files (*.png *.jpg *.jpeg *.gif)"));
		if (!fileName.isEmpty()) {
			ByteArray data;
			readByteArrayFromFile(data, stringToPath(Q2PSTRING(fileName)));

			QBuffer buffer;
			buffer.setData(reinterpret_cast<const char*>(vecptr(data)), data.size());
			buffer.open(QIODevice::ReadOnly);
			QString type = QImageReader::imageFormat(&buffer).toLower();
			if (!type.isEmpty()) {
				type = "image/" + type;
				setAvatar(data, Q2PSTRING(type));
			}
			else {
				QMessageBox::critical(this, tr("Error"), tr("The selected picture is in an unrecognized format"));
			}
		}
	}
	else if (result == clearPicture) {
		setAvatar(ByteArray(), "");
	}
}



}