#include "projectdialog.h"

#include <QtCore/qlogging.h>

#include <QCheckBox>
#include <QListView>
#include <QPushButton>
#include <QStandardItemModel>

#include "../components/MultiSelectComboBox.h"
#include "ui_projectdialog.h"
#include "../logger.h"

ProjectDialog::ProjectDialog(QWidget *parent) : QDialog(parent), ui(new Ui::ProjectDialog) {
    ui->setupUi(this);
    connect(ui->buttonBox, &QDialogButtonBox::accepted, this, &ProjectDialog::validateAndSaveProject);
    connect(ui->buttonBox, &QDialogButtonBox::rejected, this, &ProjectDialog::clearFormData);
}

void ProjectDialog::initComboBoxBlaster(const QJsonArray &undergroundOperators) {
    MultiSelectComboBox *comboBox = new MultiSelectComboBox(this);

    for (const QJsonValue &value : undergroundOperators) {
        if (value.isObject()) {
            QJsonObject obj = value.toObject();
            if (obj.contains("nickName")) {
                QString name = obj["nickName"].toString();
                comboBox->addItem(obj["identity"].toString(), name);
            }
        }
    }
    QVBoxLayout *layout = new QVBoxLayout(ui->undergroundOperatorsWidget);
    layout->addWidget(comboBox);
}

void ProjectDialog::initComboBoxSafetyInspectors(const QJsonArray &safeInspectors) {
    MultiSelectComboBox *comboBox = new MultiSelectComboBox(this);

    for (const QJsonValue &value : safeInspectors) {
        if (value.isObject()) {
            QJsonObject obj = value.toObject();
            if (obj.contains("nickName")) {
                QString name = obj["nickName"].toString();
                comboBox->addItem(obj["identity"].toString(), name);
            }
        }
    }
    QVBoxLayout *layout = new QVBoxLayout(ui->safetyInspectorsWidget);
    layout->addWidget(comboBox);
}

void ProjectDialog::initComboBoxOperator(const QJsonArray &groundOperators) {
    MultiSelectComboBox *comboBox = new MultiSelectComboBox(this);

    for (const QJsonValue &value : groundOperators) {
        if (value.isObject()) {
            QJsonObject obj = value.toObject();
            if (obj.contains("nickName")) {
                QString name = obj["nickName"].toString();
                comboBox->addItem(obj["identity"].toString(), name);
            }
        }
    }
    QVBoxLayout *layout = new QVBoxLayout(ui->groundOperatorsWidget);
    layout->addWidget(comboBox);
}

void ProjectDialog::SetL1AddressOptions(const QJsonArray &Options) {
    parentOptions = Options;
    ui->comboBoxAddr->clear();

    for (const QJsonValue &option : Options) {
        ui->comboBoxAddr->addItem(option.toObject()["name"].toString(), option.toObject()["uuid"].toString());
    }
    ui->comboBoxAddr->setCurrentIndex(-1);
}

void ProjectDialog::updateL2AddressOptions() {
    ui->comboBoxAddr_2->clear();
    for (const QJsonValue &child : L2AddressOptions) {
        ui->comboBoxAddr_2->addItem(child.toObject()["name"].toString(), child.toObject()["uuid"].toString());
    }
    ui->comboBoxAddr_2->setCurrentIndex(-1);
}

void ProjectDialog::updateL3AddressOptions() {
    ui->comboBoxAddr_3->clear();
    for (const QJsonValue &child : L3AddressOptions) {
        ui->comboBoxAddr_3->addItem(child.toObject()["name"].toString(), child.toObject()["uuid"].toString());
    }
    ui->comboBoxAddr_3->setCurrentIndex(-1);
}

void ProjectDialog::setChildOptions(const QJsonArray &newDataOptions) { dataOptions = newDataOptions; }

void ProjectDialog::clearFormData() {
    QList<QLineEdit *> lineEdits = findChildren<QLineEdit *>();
    for (QLineEdit *lineEdit : lineEdits) {
        lineEdit->clear();
    }
    QList<MultiSelectComboBox *> multiSelectCombos = findChildren<MultiSelectComboBox *>();
    for (MultiSelectComboBox *combo : multiSelectCombos) {
        combo->setCheckedKeys(QStringList());
    }
    ui->comboBoxAddr->setCurrentIndex(-1);
    ui->comboBoxAddr_2->setCurrentIndex(-1);
    ui->comboBoxAddr_3->setCurrentIndex(-1);

    // 清除选中数据
    selectedBlasters.clear();
    selectedOperators.clear();
    selectedSafetyInspectors.clear();  // Add this line

    // 显示确认按钮
    ui->buttonBox->button(QDialogButtonBox::Ok)->setVisible(true);
}

void ProjectDialog::on_comboBoxAddr_currentIndexChanged(int index) {
    if (index >= 0 && index < parentOptions.size()) {
        L2AddressOptions = parentOptions[index].toObject()["children"].toArray();
        updateL2AddressOptions();
    }
}

void ProjectDialog::on_comboBoxAddr_2_currentIndexChanged(int index) {
    if (index >= 0 && index < L2AddressOptions.size()) {
        QJsonArray subOptions = L2AddressOptions[index].toObject()["children"].toArray();
        L3AddressOptions = subOptions;
        updateL3AddressOptions();
    }
}

void ProjectDialog::validateAndSaveProject() {
    for (const auto &pair :
         ui->undergroundOperatorsWidget->findChild<MultiSelectComboBox *>()->checkedItemsWithKeys()) {
        selectedBlasters.append(pair.first);
    }
    for (const auto &pair : ui->groundOperatorsWidget->findChild<MultiSelectComboBox *>()->checkedItemsWithKeys()) {
        selectedOperators.append(pair.first);
    }
    for (const auto &pair : ui->safetyInspectorsWidget->findChild<MultiSelectComboBox *>()->checkedItemsWithKeys()) {
        selectedSafetyInspectors.append(pair.first);
    }

    QString detNum = ui->editDetNum->text().trimmed();
    QString blastCount = ui->editRegCount->text().trimmed();
    QString projectName = ui->editName->text().trimmed();
    QString xmbh = ui->editXMBH->text().trimmed();
    QString htid = ui->editHTID->text().trimmed();

    if (detNum.isEmpty() || !ui->editDetNum->hasAcceptableInput()) {
        QMessageBox::warning(nullptr, "输入错误", "请输入0-10000的数字！");
        return;
    }
    if (selectedBlasters.isEmpty()) {
        QMessageBox::warning(nullptr, "选择错误", "请至少选择一名井下爆破员！");
        return;
    }
    if (selectedOperators.isEmpty()) {
        QMessageBox::warning(nullptr, "选择错误", "请至少选择一名井上爆破员！");
        return;
    }
    if (selectedSafetyInspectors.isEmpty()) {
        QMessageBox::warning(nullptr, "选择错误", "请至少选择一名安全员！");
        return;
    }

    // 创建一个 QMap 集合，存储数据
    QMap<QString, QString> data;
    data["detNum"] = detNum;
    data["name"] = projectName;
    data["xmbh"] = xmbh;
    data["htid"] = htid;

    // 选择level最低的uuid
    QString addressUuid = ui->comboBoxAddr->currentData().toString().trimmed();
    QString l2AddressUuid = ui->comboBoxAddr_2->currentData().toString().trimmed();
    QString l3AddressUuid = ui->comboBoxAddr_3->currentData().toString().trimmed();
    QString l1AddressName = ui->comboBoxAddr->currentText().trimmed();
    QString l2AddressName = ui->comboBoxAddr_2->currentText().trimmed();
    QString l3AddressName = ui->comboBoxAddr_3->currentText().trimmed();

    if (l3AddressUuid != "") {
        addressUuid = l3AddressUuid;
    } else if (l2AddressUuid != "") {
        addressUuid = l2AddressUuid;
    } else if (addressUuid == "") {
        QMessageBox::warning(nullptr, "输入错误", "请选择地址！");
        return;
    }

    data["addressPath"] = l1AddressName + "/" + l2AddressName + "/" + l3AddressName;
    data["addressUuid"] = addressUuid;
    data["blasterIdentity"] = selectedBlasters.join(",");
    data["operatorIdentity"] = selectedOperators.join(",");
    data["safetyInspectorIdentity"] = selectedSafetyInspectors.join(",");
    data["blastCount"] = blastCount;

    if (operationType == 0) {
        qDebug() << "Project data to create:" << data;
        emit createProject(data);
    } else if (operationType == 1) {
        qDebug() << "Project data to update:" << data;
        emit saveProject(data);
    }

    clearFormData();  // 清除表单数据
}

int ProjectDialog::getOperationType() const { return operationType; }

void ProjectDialog::setOperationType(int newOperationType) { operationType = newOperationType; }

void ProjectDialog::setFormData(const HProject &Project) {
    try {
        clearFormData();  // 清除表单数据
        // hide the accept button
        ui->buttonBox->button(QDialogButtonBox::Ok)->setVisible(false);

        ui->editName->setText(Project.getName());
        ui->editDetNum->setText(Project.getDetSum());
        ui->editHTID->setText(Project.getHtid());
        ui->editXMBH->setText(Project.getXmbh());
        ui->editRegCount->setText(Project.getBlastCount());
        ui->level4Address->setText(Project.getLevel4Address());

        ui->groundOperatorsWidget->findChild<MultiSelectComboBox *>()->setCheckedKeys(
            Project.getOperatorIdentity().split(","));
        ui->undergroundOperatorsWidget->findChild<MultiSelectComboBox *>()->setCheckedKeys(
            Project.getBlasterIdentity().split(","));
        ui->safetyInspectorsWidget->findChild<MultiSelectComboBox *>()->setCheckedKeys(
            Project.getSafetyInspectorIdentity().split(","));

        // 设置地址选择
        QStringList addressParts = Project.getAddressPath().split("/");
        int numAddresses = addressParts.size();
        QVector<QString> addressVariables;
        for (int i = 0; i < numAddresses; ++i) {
            QString part = addressParts[i];
            addressVariables.append(part);
        }
        // 查找目标文本对应的索引
        int indexAddr = ui->comboBoxAddr->findText(addressVariables[0]);
        if (indexAddr != -1) {
            // 如果找到了对应的索引，设置为当前索引
            ui->comboBoxAddr->setCurrentIndex(indexAddr);
        } else {
            qDebug() << "未找到选项 " << addressVariables[0];
        }
        if (addressVariables.size() >= 2) {
            int indexAddr_2 = ui->comboBoxAddr_2->findText(addressVariables[1]);

            if (indexAddr_2 != -1) {
                // 如果找到了对应的索引，设置为当前索引
                ui->comboBoxAddr_2->setCurrentIndex(indexAddr_2);
            } else {
                qDebug() << "未找到选项 " << addressVariables[1];
            }
        }

        if (addressVariables.size() >= 3) {
            int indexAddr_3 = ui->comboBoxAddr_3->findText(addressVariables[2]);

            if (indexAddr_3 != -1) {
                // 如果找到了对应的索引，设置为当前索引
                ui->comboBoxAddr_3->setCurrentIndex(indexAddr_3);
            } else {
                // 如果没找到，输出提示信息
                qDebug() << "未找到选项 " << addressVariables[2];
            }
        }
    } catch (const std::exception& e) {
        Logger::getInstance().error(QString("设置表单数据时发生异常: %1").arg(e.what()));
        QMessageBox::warning(this, "错误", "无法设置表单数据，请检查数据格式是否正确。");
    }
}

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