#include "SelectItemsPopup.h"
#include "ui_SelectItemsPopup.h"

#include <QPushButton>

SelectItemsPopup::SelectItemsPopup(const std::vector<std::string>& available, const std::vector<std::string>& selected, QWidget* parent) :
    QDialog(parent),
    ui(new Ui::SelectItemsPopup)
{
    ui->setupUi(this);
    setWindowFlags(Qt::Popup);

    // Load initial items
    for(const auto& s : available)
    {
        if(std::find(selected.begin(), selected.end(), s) == selected.end())
            new QListWidgetItem(QString::fromStdString(s), ui->listAvailable);
    }
    for(const auto& s : selected)
        new QListWidgetItem(QString::fromStdString(s), ui->listSelected);
}

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

std::vector<std::string> SelectItemsPopup::selectedItems() const
{
    std::vector<std::string> result;
    for(int i=0;i<ui->listSelected->count();i++)
        result.push_back(ui->listSelected->item(i)->text().toStdString());
    return result;
}

void SelectItemsPopup::selectItem(const QModelIndex& idx)
{
    // Get currently selected iitem if none was provided
    QListWidgetItem* item;
    if(idx.isValid())
        item = ui->listAvailable->item(idx.row());
    else
        item = ui->listAvailable->currentItem();

    if(!item)
        return;

    // Add it to the selected items list
    new QListWidgetItem(item->text(), ui->listSelected);

    // Remove it from available items list
    delete item;
}

void SelectItemsPopup::unselectItem(const QModelIndex& idx)
{
    // Get currently selected iitem if none was provided
    QListWidgetItem* item;
    if(idx.isValid())
        item = ui->listSelected->item(idx.row());
    else
        item = ui->listSelected->currentItem();

    if(!item)
        return;

    // Add it to the available items list
    new QListWidgetItem(item->text(), ui->listAvailable);

    // Remove it from selected items list
    delete item;
}

void SelectItemsPopup::resizeEvent(QResizeEvent*)
{
    // We modify the shape of the dialog to add an arrow shaped edge. See the ascii art image below for details. The edges
    // are numbered, their order is the same as in the polygon definition.

    /*
            /3\
           /   \
      1---2     4--------5
      |                  |
      |                  |
      7------------------6
    */

    const int arrow_height = ui->spacer->geometry().height();
    const int arrow_width = arrow_height * 3;
    const int arrow_position_div = 5;

    QPolygon poly;
    poly << QPoint(rect().x(), rect().y() + arrow_height)
         << QPoint(rect().x() + rect().width() / arrow_position_div - arrow_width / 2, rect().y() + arrow_height)
         << QPoint(rect().x() + rect().width() / arrow_position_div, rect().y())
         << QPoint(rect().x() + rect().width() / arrow_position_div + arrow_width / 2, rect().y() + arrow_height)
         << QPoint(rect().x() + rect().width(), rect().y() + arrow_height)
         << QPoint(rect().x() + rect().width(), rect().y() + rect().height())
         << QPoint(rect().x(), rect().y() + rect().height());
    setMask(QRegion(poly));
}

void SelectItemsPopup::buttonBoxClicked(QAbstractButton* button)
{
    if(button == ui->buttonBox->button(QDialogButtonBox::Apply))
        accept();
}

void SelectItemsPopup::moveItemUp()
{
    moveCurrentItem(false);
}

void SelectItemsPopup::moveItemDown()
{
    moveCurrentItem(true);
}

void SelectItemsPopup::moveCurrentItem(bool down)
{
    // Get current row number and calculate row number after the movement. Check the values
    int currentRow = ui->listSelected->currentRow();
    if(currentRow == -1)
        return;
    int newRow = currentRow + (down ? 1 : -1);
    if(newRow < 0)
        return;
    if(newRow >= ui->listSelected->count())
        return;

    // Swap items
    ui->listSelected->insertItem(newRow, ui->listSelected->takeItem(currentRow));

    // Select old item at new position
    ui->listSelected->setCurrentRow(newRow);
}