/*
 * Copyright 2008-2009 Ben Boeckel <MathStuf@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, either version 3 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, see <http://www.gnu.org/licenses/>.
 */

// Header include
#include "MapTrainerUI.h"

// Sigmodr widget includes
#include "ScriptWidget.h"

// Sigmod includes
#include <sigmod/Game.h>
#include <sigmod/MapTrainer.h>
#include <sigmod/MapTrainerTeamMember.h>
#include <sigmod/Rules.h>
#include <sigmod/Species.h>
#include <sigmod/Trainer.h>

// KDE includes
#include <KActionSelector>
#include <KComboBox>
#include <KIntNumInput>
#include <KLineEdit>

// Qt includes
#include <QtCore/QFile>
#include <QtGui/QListWidget>
#include <QtGui/QListWidgetItem>
#include <QtUiTools/QUiLoader>

using namespace Sigcore;
using namespace Sigmod;
using namespace Sigmodr::Widgets;

MapTrainerUI::MapTrainerUI(MapTrainer* trainer, QWidget* parent) :
        ObjectUI(parent)
{
    setObjects(trainer, new MapTrainer(*trainer));
}

void MapTrainerUI::initGui()
{
    QFile file(":/gui/maptrainer.ui");
    file.open(QFile::ReadOnly);
    QWidget *formWidget = QUiLoader().load(&file, this);
    file.close();
    ui_name = formWidget->findChild<KLineEdit*>("varName");
    ui_trainerClass = formWidget->findChild<KComboBox*>("varTrainerClass");
    ui_numberFight = formWidget->findChild<KIntNumInput*>("varNumberFight");
    ui_script = formWidget->findChild<ScriptWidget*>("varScript");
    ui_leadTeamMembers = formWidget->findChild<KActionSelector*>("varLeadTeamMembers");
    connect(ui_name, SIGNAL(textChanged(QString)), this, SLOT(nameChanged(QString)));
    connect(ui_trainerClass, SIGNAL(currentIndexChanged(int)), this, SLOT(trainerClassChanged(int)));
    connect(ui_numberFight, SIGNAL(valueChanged(int)), this, SLOT(numberFightChanged(int)));
    connect(ui_script, SIGNAL(valueChanged(Sigcore::Script)), this, SLOT(scriptChanged(Sigcore::Script)));
    connect(ui_leadTeamMembers, SIGNAL(added(QListWidgetItem*)), this, SLOT(leadTeamMemberAdded(QListWidgetItem*)));
    connect(ui_leadTeamMembers, SIGNAL(removed(QListWidgetItem*)), this, SLOT(leadTeamMemberRemoved(QListWidgetItem*)));
}

void MapTrainerUI::refreshGui()
{
    const bool blockedTrainerClass = ui_trainerClass->blockSignals(true);
    ui_trainerClass->clear();
    for (int i = 0; i < game()->trainerCount(); ++i)
        ui_trainerClass->addItem(game()->trainer(i)->name());
    ui_trainerClass->blockSignals(blockedTrainerClass);
    ui_numberFight->setMaximum(game()->rules()->maxFight());
    const bool blockedLeadTeamMember = ui_leadTeamMembers->blockSignals(true);
    ui_leadTeamMembers->availableListWidget()->clear();
    ui_leadTeamMembers->selectedListWidget()->clear();
    for (int i = 0; i < qobject_cast<MapTrainer*>(original())->teamMemberCount(); ++i)
    {
        QString speciesName;
        const MapTrainerTeamMember* teamMember = qobject_cast<MapTrainer*>(original())->teamMember(i);
        const Species* species = game()->species(teamMember->species());
        speciesName = species ? species->name() : "(Invalid)";
        QListWidgetItem* widgetItem = new QListWidgetItem(QString("%1 level %2 (%3)").arg(speciesName, teamMember->level(), i), ui_leadTeamMembers->availableListWidget());
        widgetItem->setData(Qt::UserRole, teamMember->id());
    }
    ui_leadTeamMembers->blockSignals(blockedLeadTeamMember);
    ui_leadTeamMembers->setButtonsEnabled();
}

void MapTrainerUI::setGui()
{
    ui_name->setText(qobject_cast<MapTrainer*>(modified())->name());
    ui_trainerClass->setCurrentIndex(game()->trainerIndex(qobject_cast<MapTrainer*>(modified())->trainerClass()));
    ui_numberFight->setValue(qobject_cast<MapTrainer*>(modified())->numberFight());
    ui_script->setValue(qobject_cast<MapTrainer*>(modified())->script());
    for (int i = 0; i < ui_leadTeamMembers->availableListWidget()->count(); ++i)
    {
        QListWidgetItem* widgetItem = ui_leadTeamMembers->availableListWidget()->item(i);
        if (qobject_cast<MapTrainer*>(modified())->leadTeamMember(widgetItem->data(Qt::UserRole).toInt()))
            ui_leadTeamMembers->selectedListWidget()->addItem(ui_leadTeamMembers->availableListWidget()->takeItem(i--));
    }
    for (int i = 0; i < ui_leadTeamMembers->selectedListWidget()->count(); ++i)
    {
        QListWidgetItem* widgetItem = ui_leadTeamMembers->selectedListWidget()->item(i);
        if (!qobject_cast<MapTrainer*>(modified())->leadTeamMember(widgetItem->data(Qt::UserRole).toInt()))
            ui_leadTeamMembers->availableListWidget()->addItem(ui_leadTeamMembers->selectedListWidget()->takeItem(i--));
    }
    ui_leadTeamMembers->setButtonsEnabled();
}

void MapTrainerUI::apply()
{
    *qobject_cast<MapTrainer*>(original()) = *qobject_cast<MapTrainer*>(modified());
    emit(changed(false));
}

void MapTrainerUI::discard()
{
    *qobject_cast<MapTrainer*>(modified()) = *qobject_cast<MapTrainer*>(original());
    setGui();
    emit(changed(false));
}

void MapTrainerUI::nameChanged(const QString& name)
{
    const int cursor = ui_name->cursorPosition();
    qobject_cast<MapTrainer*>(modified())->setName(name);
    ui_name->setCursorPosition(cursor);
}

void MapTrainerUI::trainerClassChanged(const int trainerClass)
{
    qobject_cast<MapTrainer*>(modified())->setTrainerClass(game()->trainer(trainerClass)->id());
}

void MapTrainerUI::numberFightChanged(const int numberFight)
{
    qobject_cast<MapTrainer*>(modified())->setNumberFight(numberFight);
}

void MapTrainerUI::scriptChanged(const Script& script)
{
    qobject_cast<MapTrainer*>(modified())->setScript(script);
}

void MapTrainerUI::leadTeamMemberAdded(QListWidgetItem* item)
{
    qobject_cast<MapTrainer*>(modified())->setLeadTeamMember(item->data(Qt::UserRole).toInt(), true);
    setGui();
}

void MapTrainerUI::leadTeamMemberRemoved(QListWidgetItem* item)
{
    qobject_cast<MapTrainer*>(modified())->setLeadTeamMember(item->data(Qt::UserRole).toInt(), false);
    setGui();
}