/*
 * Copyright (c) 2010 Kevin Smith
 * Licensed under the GNU General Public License v3.
 * See Documentation/Licenses/GPLv3.txt for more information.
 */

#include "Swift/QtUI/UserSearch/QtUserSearchWindow.h"

#include <QItemDelegate>
#include <QModelIndex>
#include <QWizardPage>
#include <QMovie>
#include <boost/smart_ptr/make_shared.hpp>

#include <Swiften/Base/foreach.h>
#include "Swift/Controllers/UIEvents/UIEventStream.h"
#include "Swift/Controllers/UIEvents/RequestChatUIEvent.h"
#include "Swift/Controllers/UIEvents/AddContactUIEvent.h"
#include "Swift/QtUI/UserSearch/UserSearchModel.h"
#include "Swift/QtUI/UserSearch/UserSearchDelegate.h"
#include "Swift/QtUI/QtSwiftUtil.h"
#include "Swift/QtUI/QtFormResultItemModel.h"
#include "QtUserSearchFirstPage.h"
#include "QtUserSearchFieldsPage.h"
#include "QtUserSearchResultsPage.h"
#include "QtUserSearchDetailsPage.h"

namespace Swift {

QtUserSearchWindow::QtUserSearchWindow(UIEventStream* eventStream, UserSearchWindow::Type type, const std::set<std::string>& groups) : eventStream_(eventStream), type_(type), model_(NULL) {
	setupUi(this);
#ifndef Q_WS_MAC
	setWindowIcon(QIcon(":/logo-icon-16.png"));
#endif
	QString title(type == UserSearchWindow::AddContact ? tr("Add Contact") : tr("Chat to User"));
	setWindowTitle(title);

	delegate_ = new UserSearchDelegate();

	firstPage_ = new QtUserSearchFirstPage(type, title);
	connect(firstPage_->byJID_, SIGNAL(toggled(bool)), this, SLOT(handleFirstPageRadioChange()));
	connect(firstPage_->byLocalSearch_, SIGNAL(toggled(bool)), this, SLOT(handleFirstPageRadioChange()));
	connect(firstPage_->byRemoteSearch_, SIGNAL(toggled(bool)), this, SLOT(handleFirstPageRadioChange()));
#if QT_VERSION >= 0x040700
		firstPage_->jid_->setPlaceholderText(tr("alice@wonderland.lit"));
#endif
	firstPage_->service_->setEnabled(false);
	setPage(1, firstPage_);

	fieldsPage_ = new QtUserSearchFieldsPage();
	fieldsPage_->fetchingThrobber_->setMovie(new QMovie(":/icons/throbber.gif", QByteArray(), this));
	fieldsPage_->fetchingThrobber_->movie()->stop();
	setPage(2, fieldsPage_);

	resultsPage_ = new QtUserSearchResultsPage();

#ifdef SWIFT_PLATFORM_MACOSX
	resultsPage_->results_->setAlternatingRowColors(true);
#endif
	if (type == AddContact) {
		connect(resultsPage_, SIGNAL(onUserTriggersContinue()), this, SLOT(next()));
	}
	else {
		connect(resultsPage_, SIGNAL(onUserTriggersContinue()), this, SLOT(accept()));
	}
	setPage(3, resultsPage_);

	detailsPage_ = new QtUserSearchDetailsPage(groups);
	setPage(4, detailsPage_);

	connect(this, SIGNAL(currentIdChanged(int)), this, SLOT(handleCurrentChanged(int)));
	connect(this, SIGNAL(accepted()), this, SLOT(handleAccepted()));
	clear();
}

QtUserSearchWindow::~QtUserSearchWindow() {
	delete model_;
}

void QtUserSearchWindow::handleCurrentChanged(int page) {
	resultsPage_->emitCompletenessCheck();
	if (page == 2 && lastPage_ == 1) {
		setError("");
		/* next won't be called if JID is selected */
		JID server = getServerToSearch();
		clearForm();
		onFormRequested(server);
	}
	else if (page == 3 && lastPage_ == 2) {
		handleSearch();
	}
	else if (page == 4) {
		detailsPage_->clear();
		detailsPage_->setJID(getContactJID());
		onNameSuggestionRequested(getContactJID());
	}
	lastPage_ = page;
}

JID QtUserSearchWindow::getServerToSearch() {
	return firstPage_->byRemoteSearch_->isChecked() ? JID(Q2PSTRING(firstPage_->service_->currentText().trimmed())) : myServer_;
}

void QtUserSearchWindow::handleAccepted() {
	JID jid = getContactJID();

	if (type_ == AddContact) {
		eventStream_->send(boost::make_shared<AddContactUIEvent>(jid, detailsPage_->getName(), detailsPage_->getSelectedGroups()));
	}
	else {
		boost::shared_ptr<UIEvent> event(new RequestChatUIEvent(jid));
		eventStream_->send(event);
	}
}

int QtUserSearchWindow::nextId() const {
	switch (currentId()) {
		case 1: return firstPage_->byJID_->isChecked() ? (type_ == AddContact ? 4 : -1) : 2;
		case 2: return 3;
		case 3: return type_ == AddContact ? 4 : -1;
		case 4: return -1;
		default: return -1;
	}
}

void QtUserSearchWindow::handleFirstPageRadioChange() {
	if (firstPage_->byJID_->isChecked()) {
		firstPage_->jid_->setText("");
		firstPage_->jid_->setEnabled(true);
		firstPage_->service_->setEnabled(false);
		restart();
	}
	else if (firstPage_->byRemoteSearch_->isChecked()) {
		firstPage_->service_->setEditText("");
		firstPage_->jid_->setEnabled(false);
		firstPage_->service_->setEnabled(true);
		//firstPage_->jid_->setText("");
		restart();
	}
	else {
		firstPage_->jid_->setEnabled(false);
		firstPage_->service_->setEnabled(false);
		restart();
	}
}

void QtUserSearchWindow::handleSearch() {
	boost::shared_ptr<SearchPayload> search(new SearchPayload());
	if (fieldsPage_->getFormWidget()) {
		search->setForm(fieldsPage_->getFormWidget()->getCompletedForm());
	} else {
		if (fieldsPage_->nickInput_->isEnabled()) {
			search->setNick(Q2PSTRING(fieldsPage_->nickInput_->text()));
		}
		if (fieldsPage_->firstInput_->isEnabled()) {
			search->setFirst(Q2PSTRING(fieldsPage_->firstInput_->text()));
		}
		if (fieldsPage_->lastInput_->isEnabled()) {
			search->setLast(Q2PSTRING(fieldsPage_->lastInput_->text()));
		}
		if (fieldsPage_->emailInput_->isEnabled()) {
			search->setEMail(Q2PSTRING(fieldsPage_->emailInput_->text()));
		}
	}
	onSearchRequested(search, getServerToSearch());
}

JID QtUserSearchWindow::getContactJID() const {
	JID jid;
	if (!firstPage_->byJID_->isChecked()) {
		if (dynamic_cast<UserSearchModel*>(model_)) {
			UserSearchResult* userItem = static_cast<UserSearchResult*>(resultsPage_->results_->currentIndex().internalPointer());
			if (userItem) { /* Remember to leave this if we change to dynamic cast */
				jid = userItem->getJID();
			}
		} else {
			int row = resultsPage_->results_->currentIndex().row();

			Form::FormItem item = dynamic_cast<QtFormResultItemModel*>(model_)->getForm()->getItems().at(row);
			JID fallbackJid;
			foreach(FormField::ref field, item) {
				if (boost::dynamic_pointer_cast<JIDSingleFormField>(field)) {
					jid = JID(field->getRawValues().at(0));
					break;
				}
				if (field->getName() == "jid") {
					fallbackJid = field->getRawValues().at(0);
				}
			}
			if (!jid.isValid()) {
				jid = fallbackJid;
			}
		}
	}
	else {
		jid = JID(Q2PSTRING(firstPage_->jid_->text().trimmed()));
	}
	return jid;
}

void QtUserSearchWindow::show() {
	clear();
	QWidget::show();
}

void QtUserSearchWindow::addSavedServices(const std::vector<JID>& services) {
	firstPage_->service_->clear();
	foreach (JID jid, services) {
		firstPage_->service_->addItem(P2QSTRING(jid.toString()));
	}
	firstPage_->service_->clearEditText();
}

void QtUserSearchWindow::setSearchFields(boost::shared_ptr<SearchPayload> fields) {
	fieldsPage_->fetchingThrobber_->hide();
	fieldsPage_->fetchingThrobber_->movie()->stop();
	fieldsPage_->fetchingLabel_->hide();

	fieldsPage_->instructionsLabel_->setText(fields->getInstructions() ? P2QSTRING(fields->getInstructions().get()) : "Enter search terms");
	if (fields->getForm()) {
		fieldsPage_->setFormWidget(new QtFormWidget(fields->getForm(), fieldsPage_));
	} else {
		fieldsPage_->setFormWidget(NULL);
		bool enabled[8] = {fields->getNick(), fields->getNick(), fields->getFirst(), fields->getFirst(), fields->getLast(), fields->getLast(), fields->getEMail(), fields->getEMail()};
		QWidget* legacySearchWidgets[8] = {fieldsPage_->nickInputLabel_, fieldsPage_->nickInput_, fieldsPage_->firstInputLabel_, fieldsPage_->firstInput_, fieldsPage_->lastInputLabel_, fieldsPage_->lastInput_, fieldsPage_->emailInputLabel_, fieldsPage_->emailInput_};
		for (int i = 0; i < 8; i++) {
			legacySearchWidgets[i]->setVisible(enabled[i]);
			legacySearchWidgets[i]->setEnabled(enabled[i]);
		}
	}
	fieldsPage_->emitCompletenessCheck();
}

void QtUserSearchWindow::setNameSuggestions(const std::vector<std::string>& suggestions) {
	if (detailsPage_) {
		detailsPage_->setNameSuggestions(suggestions);
	}
}

void QtUserSearchWindow::prepopulateJIDAndName(const JID& jid, const std::string& name) {
	firstPage_->jid_->setText(P2QSTRING(jid.toBare().toString()));
	detailsPage_->setJID(jid);
	lastPage_ = 1;
	restart();
	next();
	detailsPage_->setName(name);
}

void QtUserSearchWindow::setResults(const std::vector<UserSearchResult>& results) {
	UserSearchModel *newModel = new UserSearchModel();
	newModel->setResults(results);
	resultsPage_->results_->setModel(newModel);
	resultsPage_->results_->setItemDelegate(delegate_);
	resultsPage_->results_->setHeaderHidden(true);
	delete model_;
	model_ = newModel;
	resultsPage_->setNoResults(model_->rowCount() == 0);
	resultsPage_->emitCompletenessCheck();
}

void QtUserSearchWindow::setResultsForm(Form::ref results) {
	QtFormResultItemModel *newModel = new QtFormResultItemModel(this);
	newModel->setForm(results);
	resultsPage_->results_->setModel(newModel);
	resultsPage_->results_->setItemDelegate(new QItemDelegate());
	resultsPage_->results_->setHeaderHidden(false);
	resultsPage_->results_->header()->setResizeMode(QHeaderView::ResizeToContents);
	delete model_;
	model_ = newModel;
	resultsPage_->setNoResults(model_->rowCount() == 0);
	resultsPage_->emitCompletenessCheck();
}

void QtUserSearchWindow::setSelectedService(const JID& jid) {
	myServer_ = jid;
}

void QtUserSearchWindow::clearForm() {
	fieldsPage_->fetchingThrobber_->show();
	fieldsPage_->fetchingThrobber_->movie()->start();
	fieldsPage_->fetchingLabel_->show();
	QWidget* legacySearchWidgets[8] = {fieldsPage_->nickInputLabel_, fieldsPage_->nickInput_, fieldsPage_->firstInputLabel_, fieldsPage_->firstInput_, fieldsPage_->lastInputLabel_, fieldsPage_->lastInput_, fieldsPage_->emailInputLabel_, fieldsPage_->emailInput_};
	for (int i = 0; i < 8; i++) {
		legacySearchWidgets[i]->hide();
		if (QLineEdit* edit = qobject_cast<QLineEdit*>(legacySearchWidgets[i])) {
			edit->clear();
		}
	}
	fieldsPage_->emitCompletenessCheck();
}

void QtUserSearchWindow::clear() {
	firstPage_->errorLabel_->setVisible(false);
	QString howText;
	if (type_ == AddContact) {
		howText = QString(tr("How would you like to find the user to add?"));
	}
	else {
		howText = QString(tr("How would you like to find the user to chat to?"));
	}
	firstPage_->howLabel_->setText(howText);
	firstPage_->byJID_->setChecked(true);
	clearForm();
	resultsPage_->results_->setModel(NULL);
	delete model_;
	model_ = NULL;
	handleFirstPageRadioChange();
	restart();
	lastPage_ = 1;
}

void QtUserSearchWindow::setError(const QString& error) {
	if (error.isEmpty()) {
		firstPage_->errorLabel_->hide();
	}
	else {
		firstPage_->errorLabel_->setText(QString("<font color='red'>%1</font>").arg(error));
		firstPage_->errorLabel_->show();
		restart();
		lastPage_ = 1;
	}
}

void QtUserSearchWindow::setSearchError(bool error) {
	if (error) {
		setError(tr("Error while searching"));
	}
}

void QtUserSearchWindow::setServerSupportsSearch(bool support) {
	if (!support) {
		setError(tr("This server doesn't support searching for users."));
	}
}

}