/*

 * Hedgewars, a free turn based strategy game

 * Copyright (c) 2004-2012 Andrey Korotaev <unC0Rr@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; version 2 of the License

 *

 * 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., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA

 */

#include <QDialog>

#include <QGridLayout>

#include <QHBoxLayout>

#include <QScrollArea>

#include <QPushButton>

#include <QToolButton>

#include <QWidgetItem>

#include <QModelIndex>

#include <QListView>

#include <QLineEdit>

#include <QLabel>

#include <QSortFilterProxyModel>

#include <QDebug>

#include "DataManager.h"

#include "lineeditcursor.h"

#include "HatModel.h"

#include "hatprompt.h"

HatPrompt::HatPrompt(int currentIndex, QWidget* parent) : QDialog(parent)

{

    setModal(true);

    setWindowFlags(Qt::Sheet);

    setWindowModality(Qt::WindowModal);

    setMinimumSize(550, 430);

    resize(550, 430);

    setSizePolicy(QSizePolicy::Minimum, QSizePolicy::Minimum);

    setStyleSheet("QPushButton { padding: 5px; margin-top: 10px; }");

    // Hat model, and a model for setting a filter

    HatModel * hatModel = DataManager::instance().hatModel();

    filterModel = new QSortFilterProxyModel();

    filterModel->setSourceModel(hatModel);

    filterModel->setFilterCaseSensitivity(Qt::CaseInsensitive);

    // Grid

    QGridLayout * dialogLayout = new QGridLayout(this);

    dialogLayout->setSpacing(0);

    dialogLayout->setColumnStretch(1, 1);

    QHBoxLayout * topLayout = new QHBoxLayout();

    // Help/prompt message at top

    QLabel * lblDesc = new QLabel(tr("Select a hat"));

    lblDesc->setObjectName("lblDesc");

    lblDesc->setStyleSheet("#lblDesc { color: #130F2A; background: #F6CB1C; border: solid 4px #F6CB1C; border-top-left-radius: 10px; padding: 4px 10px;}");

    lblDesc->setSizePolicy(QSizePolicy::Expanding, QSizePolicy::Fixed);

    lblDesc->setFixedHeight(24);

    lblDesc->setMinimumWidth(0);

    // Filter text box

    QWidget * filterContainer = new QWidget();

    filterContainer->setFixedHeight(24);

    filterContainer->setObjectName("filterContainer");

    filterContainer->setStyleSheet("#filterContainer { background: #F6CB1C; border-top-right-radius: 10px; padding: 3px; }");

    filterContainer->setFixedWidth(250);

    txtFilter = new LineEditCursor(filterContainer);

    txtFilter->setFixedWidth(250);

    txtFilter->setFocus();

    txtFilter->setFixedHeight(22);

    connect(txtFilter, SIGNAL(textChanged(const QString &)), this, SLOT(filterChanged(const QString &)));

    connect(txtFilter, SIGNAL(moveUp()), this, SLOT(moveUp()));

    connect(txtFilter, SIGNAL(moveDown()), this, SLOT(moveDown()));

    connect(txtFilter, SIGNAL(moveLeft()), this, SLOT(moveLeft()));

    connect(txtFilter, SIGNAL(moveRight()), this, SLOT(moveRight()));

    // Filter label

    QLabel * lblFilter = new QLabel(tr("Filter: "), txtFilter);

    // Corner widget

    QLabel * corner = new QLabel();

    corner->setPixmap(QPixmap(QString::fromUtf8(":/res/inverse-corner-bl.png")));

    corner->setFixedSize(10, 10);

    // Add widgets to top layout

    topLayout->addWidget(lblDesc);

    topLayout->addWidget(filterContainer);

    topLayout->addWidget(corner, 0, Qt::AlignBottom);

    topLayout->addStretch(1);

    // Cancel button (closes dialog)

    QPushButton * btnCancel = new QPushButton(tr("Cancel"));

    connect(btnCancel, SIGNAL(clicked()), this, SLOT(reject()));

    // Select button

    QPushButton * btnSelect = new QPushButton(tr("Use selected hat"));

    btnSelect->setDefault(true);

    connect(btnSelect, SIGNAL(clicked()), this, SLOT(onAccepted()));

    // Add hats

    list = new HatListView();

    list->setModel(filterModel);

    list->setViewMode(QListView::IconMode);

    list->setResizeMode(QListView::Adjust);

    list->setMovement(QListView::Static);

    list->setEditTriggers(QAbstractItemView::NoEditTriggers);

    list->setSpacing(8);

    list->setWordWrap(true);

    list->setSelectionMode(QAbstractItemView::SingleSelection);

    list->setObjectName("hatList");

    list->setCurrentIndex(filterModel->index(currentIndex, 0));

    connect(list, SIGNAL(activated(const QModelIndex &)), this, SLOT(hatChosen(const QModelIndex &)));

    connect(list, SIGNAL(clicked(const QModelIndex &)), this, SLOT(hatChosen(const QModelIndex &)));

    // Add elements to layouts

    dialogLayout->addLayout(topLayout, 0, 0, 1, 3);

    dialogLayout->addWidget(list, 1, 0, 1, 3);

    dialogLayout->addWidget(btnCancel, 2, 0, 1, 1, Qt::AlignLeft);

    dialogLayout->addWidget(btnSelect, 2, 2, 1, 1, Qt::AlignRight);

}

void HatPrompt::moveUp()

{

    list->setCurrentIndex(list->moveCursor(QAbstractItemView::MoveUp, Qt::NoModifier));

}

void HatPrompt::moveDown()

{

    list->setCurrentIndex(list->moveCursor(QAbstractItemView::MoveDown, Qt::NoModifier));

}

void HatPrompt::moveLeft()

{

    list->setCurrentIndex(list->moveCursor(QAbstractItemView::MoveLeft, Qt::NoModifier));

}

void HatPrompt::moveRight()

{

    list->setCurrentIndex(list->moveCursor(QAbstractItemView::MoveRight, Qt::NoModifier));

}

void HatPrompt::onAccepted()

{

    hatChosen(list->currentIndex());

}

// When a hat is selected

void HatPrompt::hatChosen(const QModelIndex & index)

{

    done(filterModel->mapToSource(index).row() + 1); // Since returning 0 means canceled

}

// When the text in the filter text box is changed

void HatPrompt::filterChanged(const QString & text)

{

    filterModel->setFilterFixedString(text);

    list->setCurrentIndex(filterModel->index(0, 0));

}