#include "projectdialog.h"

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

#include "ui_projectdialog.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::SetComboBoxBlast(const QJsonArray &bapoYuanArray) {
    blasterArray = bapoYuanArray;

    // 创建标准项模型用于多选
    QStandardItemModel *blastModel = new QStandardItemModel(ui->comboBoxBlast);

    for (const QJsonValue &value : bapoYuanArray) {
        if (value.isObject()) {
            QJsonObject obj = value.toObject();
            if (obj.contains("nickName")) {
                QString name = obj["nickName"].toString();
                QStandardItem *item = new QStandardItem(name);
                item->setCheckable(true);
                item->setData(obj["identity"].toString(), Qt::UserRole);
                blastModel->appendRow(item);
            }
        }
    }

    ui->comboBoxBlast->setModel(blastModel);
    ui->comboBoxBlast->setView(new QListView(ui->comboBoxBlast));
    ui->comboBoxBlast->view()->setMinimumWidth(200);

    // 连接项变更信号
    connect(blastModel, &QStandardItemModel::itemChanged, this, &ProjectDialog::onBlasterItemChanged);
}

void ProjectDialog::SetComboBoxOperator(const QJsonArray &anQuanYuanArray) {
    operatorArray = anQuanYuanArray;

    // 创建标准项模型用于多选
    QStandardItemModel *operatorModel = new QStandardItemModel(ui->comboBoxOperator);

    for (const QJsonValue &value : anQuanYuanArray) {
        if (value.isObject()) {
            QJsonObject obj = value.toObject();
            if (obj.contains("nickName")) {
                QString name = obj["nickName"].toString();
                QStandardItem *item = new QStandardItem(name);
                item->setCheckable(true);
                item->setData(obj["identity"].toString(), Qt::UserRole);
                operatorModel->appendRow(item);
            }
        }
    }

    ui->comboBoxOperator->setModel(operatorModel);
    ui->comboBoxOperator->setView(new QListView(ui->comboBoxOperator));
    ui->comboBoxOperator->view()->setMinimumWidth(200);

    // 连接项变更信号
    connect(operatorModel, &QStandardItemModel::itemChanged, this, &ProjectDialog::onOperatorItemChanged);
}

void ProjectDialog::on_comboBoxOperator_currentIndexChanged(int index) {
    // 此槽函数保留以避免编译错误，但不再需要之前的功能
    // 多选框的变更由 onOperatorItemChanged 处理
}

void ProjectDialog::on_comboBoxBlast_currentIndexChanged(int index) {
    // 此槽函数保留以避免编译错误，但不再需要之前的功能
    // 多选框的变更由 onBlasterItemChanged 处理
}

void ProjectDialog::fillComboBox(QComboBox *comboBox, const QJsonArray &jsonArray) {
    for (const QJsonValue &value : jsonArray) {
        if (value.isObject()) {
            QJsonObject obj = value.toObject();
            if (obj.contains("nickName")) {
                QString name = obj["nickName"].toString();
                comboBox->addItem(name);
            }
        }
    }
}

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::extractNames(const QJsonArray &array, QStringList &names) {
    for (const auto &item : array) {
        if (item.isObject()) {
            QJsonObject obj = item.toObject();
            if (obj.contains("name")) {
                names.append(obj["name"].toString());
            }
            if (obj.contains("children") && obj["children"].isArray()) {
                extractNames(obj["children"].toArray(), names);
            }
        }
    }
}

void ProjectDialog::clearFormData() {
    QList<QLineEdit *> lineEdits = findChildren<QLineEdit *>();
    for (QLineEdit *lineEdit : lineEdits) {
        lineEdit->clear();
    }

    // 清除多选ComboBox
    if (ui->comboBoxBlast->model()) {
        QStandardItemModel *blastModel = qobject_cast<QStandardItemModel *>(ui->comboBoxBlast->model());
        if (blastModel) {
            for (int i = 0; i < blastModel->rowCount(); ++i) {
                QStandardItem *item = blastModel->item(i);
                if (item) {
                    item->setCheckState(Qt::Unchecked);
                }
            }
        }
        ui->comboBoxBlast->setEditText("请选择井下爆破员");
    }

    if (ui->comboBoxOperator->model()) {
        QStandardItemModel *operatorModel = qobject_cast<QStandardItemModel *>(ui->comboBoxOperator->model());
        if (operatorModel) {
            for (int i = 0; i < operatorModel->rowCount(); ++i) {
                QStandardItem *item = operatorModel->item(i);
                if (item) {
                    item->setCheckState(Qt::Unchecked);
                }
            }
        }
        ui->comboBoxOperator->setEditText("请选择井上爆破员");
    }

    // 清除普通ComboBox
    ui->comboBoxAddr->setCurrentIndex(-1);
    ui->comboBoxAddr_2->setCurrentIndex(-1);
    ui->comboBoxAddr_3->setCurrentIndex(-1);

    // 清除选中数据
    selectedBlasters.clear();
    selectedOperators.clear();
    blasterIds.clear();
    operatorIds.clear();

    // 显示确认按钮
    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() {
    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;
    }

    // 创建一个 QMap 集合，存储数据
    QMap<QString, QString> data;
    data["detNum"] = detNum;
    data["name"] = projectName;
    data["xmbh"] = xmbh;
    data["htid"] = htid;
    data["operatorName"] = selectedOperators.join(", ");
    data["blasterName"] = selectedBlasters.join(", ");

    // 选择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"] = blasterIds.join(",");
    data["operatorIdentity"] = operatorIds.join(",");
    data["blastCount"] = blastCount;

    if (operationStatus == 0) {
        emit validateDetNum(data);
    } else if (operationStatus == 1) {
        emit validateDetNumUpdate(data);
    }

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

int ProjectDialog::getOperationStatus() const { return operationStatus; }

void ProjectDialog::setOperationStatus(int newOperationStatus) { operationStatus = newOperationStatus; }

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());

        // 设置井下爆破员多选
        QStringList blasterNames = Project.getBlasterName().split(", ");
        QStringList blasterIdList = Project.getBlasterIdentity().split(",");
        QStandardItemModel *blastModel = qobject_cast<QStandardItemModel *>(ui->comboBoxBlast->model());
        if (blastModel) {
            for (int i = 0; i < blastModel->rowCount(); ++i) {
                QStandardItem *item = blastModel->item(i);
                if (item && blasterNames.contains(item->text())) {
                    item->setCheckState(Qt::Checked);
                }
            }
        }

        // 设置井上爆破员多选
        QStringList operatorNames = Project.getOperatorName().split(", ");
        QStringList operatorIdList = Project.getOperatorIdentity().split(",");
        QStandardItemModel *operatorModel = qobject_cast<QStandardItemModel *>(ui->comboBoxOperator->model());
        if (operatorModel) {
            for (int i = 0; i < operatorModel->rowCount(); ++i) {
                QStandardItem *item = operatorModel->item(i);
                if (item && operatorNames.contains(item->text())) {
                    item->setCheckState(Qt::Checked);
                }
            }
        }

        // 更新已选择列表
        selectedBlasters = blasterNames;
        blasterIds = blasterIdList;
        selectedOperators = operatorNames;
        operatorIds = operatorIdList;

        // 更新ComboBox显示文本
        ui->comboBoxBlast->setEditText(selectedBlasters.join(", "));
        ui->comboBoxOperator->setEditText(selectedOperators.join(", "));

        // 设置地址选择
        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);
            qDebug() << "已将选项 " << addressVariables[0] << " 设置为当前显示的选项。";
        } 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);
                qDebug() << "已将选项 " << addressVariables[1] << " 设置为当前显示的选项。";
            } 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);
                qDebug() << "已将选项 " << addressVariables[2] << " 设置为当前显示的选项。";
            } else {
                // 如果没找到，输出提示信息
                qDebug() << "未找到选项 " << addressVariables[2];
            }
        }
    } catch (const std::exception &e) {
        qDebug() << "发生异常: " << e.what();
    } catch (...) {
        qDebug() << "发生未知异常";
    }
}

void ProjectDialog::onBlasterItemChanged(QStandardItem *item) { updateSelectedBlasters(); }

void ProjectDialog::onOperatorItemChanged(QStandardItem *item) { updateSelectedOperators(); }

void ProjectDialog::updateSelectedBlasters() {
    selectedBlasters.clear();
    blasterIds.clear();
    QStandardItemModel *model = qobject_cast<QStandardItemModel *>(ui->comboBoxBlast->model());

    if (!model) return;

    for (int i = 0; i < model->rowCount(); ++i) {
        QStandardItem *item = model->item(i);
        if (item && item->checkState() == Qt::Checked) {
            selectedBlasters.append(item->text());
            blasterIds.append(item->data(Qt::UserRole).toString());
        }
    }

    // 更新显示的文本
    if (selectedBlasters.isEmpty()) {
        ui->comboBoxBlast->setEditText("请选择井下爆破员");
    } else {
        ui->comboBoxBlast->setEditText(selectedBlasters.join(", "));
    }
}

void ProjectDialog::updateSelectedOperators() {
    selectedOperators.clear();
    operatorIds.clear();
    QStandardItemModel *model = qobject_cast<QStandardItemModel *>(ui->comboBoxOperator->model());

    if (!model) return;

    for (int i = 0; i < model->rowCount(); ++i) {
        QStandardItem *item = model->item(i);
        if (item && item->checkState() == Qt::Checked) {
            selectedOperators.append(item->text());
            operatorIds.append(item->data(Qt::UserRole).toString());
        }
    }

    // 更新显示的文本
    if (selectedOperators.isEmpty()) {
        ui->comboBoxOperator->setEditText("请选择井上爆破员");
    } else {
        ui->comboBoxOperator->setEditText(selectedOperators.join(", "));
    }
}

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