#include "countrymanagement.h"
#include "dbunit.h"
#include <QDebug>
#include <QGridLayout>
#include <QInputDialog>
#include <QPushButton>
#include <QSqlError>
#include <QSqlQuery>

using namespace ViewComp;

CountryManagement::CountryManagement(DBUnit* database, QWidget* parent)
    : QDialog(parent), db_host(database), keywords_input(new QLineEdit(this)),
      country_view(new QTableView(this)), country_model(new QStandardItemModel(this)),
      append(new QPushButton("添加", this)), remove(new QPushButton("移除", this))
{
    country_view->setModel(country_model);

    // ============================
    auto layout = new QGridLayout(this);
    layout->addWidget(keywords_input, 0, 0, 1, 4);
    layout->addWidget(country_view, 1, 0, 3, 4);
    layout->addWidget(append, 4, 0, 1, 2);
    layout->addWidget(remove, 4, 2, 1, 2);
    country_model->setHorizontalHeaderLabels(QStringList() << "国家名称"
                                                           << "简介");

    connect(append, &QPushButton::clicked, [this]() {
        QInputDialog in(this);
        in.setWindowTitle("输入国家名称");
        in.setLabelText("输入非重复国家名称：");
        if (in.exec() == QDialog::Rejected)
            return;

        auto q = db_host->getQuery();
        q.prepare("insert into country_information "
                  "(name, describe)values(:n, :d)");
        q.bindValue(":n", in.textValue());
        q.bindValue(":d", "请手动修改国家简述");
        if (!q.exec()) {
            qDebug() << q.lastError();
            return;
        }
        q.prepare("select max(cid) from country_information");
        if (!q.exec()) {
            qDebug() << q.lastError();
            return;
        }
        if (q.next()) {
            QList<QStandardItem*> row;
            row << new QStandardItem(in.textValue());
            row.last()->setData(q.value(0));
            row << new QStandardItem("请手动修改国家简述");
            country_model->appendRow(row);
        }
    });
    connect(remove, &QPushButton::clicked, [this]() {
        auto index = country_view->currentIndex();
        if (!index.isValid())
            return;

        auto inindex = index.sibling(index.row(), 0);
        auto q = db_host->getQuery();
        q.prepare("delete from country_information where cid = :id");
        q.bindValue(":id", inindex.data(Qt::UserRole + 1));
        if (!q.exec()) {
            qDebug() << q.lastError();
            return;
        }

        country_model->removeRow(index.row());
    });

    connect(keywords_input, &QLineEdit::textChanged, this, &CountryManagement::input_query);
}

void
CountryManagement::init_country_model(QStandardItemModel* model)
{
    model->removeRows(0, model->rowCount());
    auto q = db_host->getQuery();
    q.prepare("select cid, name, describe from country_information");
    if (!q.exec()) {
        qDebug() << q.lastError();
        return;
    }
    while (q.next()) {
        QList<QStandardItem*> row;
        row << new QStandardItem(q.value(1).toString());
        row.last()->setData(q.value(0));
        row << new QStandardItem(q.value(2).toString());
        country_model->appendRow(row);
    }
}

void
CountryManagement::input_query(const QString& keywords)
{
    country_model->removeRows(0, country_model->rowCount());
    if (keywords.isEmpty())
        return;

    QString str = "select cid, name, describe from country_information ";
    if (keywords != "*") {
        str += "where name like '%" + keywords + "%'";
    }
    auto q = db_host->getQuery();
    q.prepare(str);
    if (!q.exec()) {
        qDebug() << q.lastError();
        return;
    }
    while (q.next()) {
        QList<QStandardItem*> row;
        row << new QStandardItem(q.value(1).toString());
        row.last()->setData(q.value(0));
        row << new QStandardItem(q.value(2).toString());
        country_model->appendRow(row);
    }
}