pc-system/blastProject/blastprojectpage.cpp

#include "blastprojectpage.h"

#include <QJsonArray>
#include <QJsonObject>
#include <QUrl>

#include "../components/buttondelegate.h"
#include "../components/loadingWidget.h"
#include "../utils/backendapimanager.h"
#include "../utils/databasemanager.h"
#include "../utils/global.h"
#include "ui_blastprojectpage.h"

#define BLASTER_ROLE_ID 4
#define OPERATOR_ROLE_ID 7

BlastProjectPage::BlastProjectPage(QWidget *parent)
    : QWidget(parent), ui(new Ui::BlastProjectPage), dao(DatabaseManager::getInstance().getDatabase()) {
    ui->setupUi(this);
    pageWidget = new PageWidget;
    connect(pageWidget, &PageWidget::currentPageChanged, this, &BlastProjectPage::PageChanged);
    connect(pageWidget->getComboBox(), QOverload<int>::of(&QComboBox::currentIndexChanged), this,
            &BlastProjectPage::onComboBoxIndexChanged);
    ui->verticalLayout_4->addWidget(pageWidget);
    pageSize = 10;
    currentPage = 1;

    m_dialog = new ProjectDialog(this);
    QJsonArray pcs = backendAPIManager::getHAddresses();
    addressOptions(pcs);

    QJsonArray allUsers = backendAPIManager::getSysUsers();

    for (const QJsonValue &value : allUsers) {
        if (value.isObject()) {
            QJsonObject obj = value.toObject();
            m_userNameById.insert(obj["identity"].toString(), obj["nickName"].toString());
        }
    }
    initUserAndDialogOptions(allUsers);

    connect(m_dialog, &ProjectDialog::createProject, this, &BlastProjectPage::createProject);
    connect(m_dialog, &ProjectDialog::saveProject, this, &BlastProjectPage::updateProject);

    refreshData();
}

void BlastProjectPage::refreshData() {
    loadDataFromSource(currentPage, pageSize);
    publishAllPendingBlastProjects();
}

void BlastProjectPage::loadDataFromSource(int currentPage, int pageSize) {
    LoadingWidget::showLoading(this, "正在加载数据...");
    QJsonObject extParams({{"blast_status", QJsonArray({"2", "1"})}});
    QJsonObject result = backendAPIManager::getHProjects(currentPage, pageSize, extParams);

    QList<QSharedPointer<HProject>> projectList =
        dao.getHProjectsFromJsonArray(result["data"].toObject()["list"].toArray());

    totalCount = result["data"].toObject()["count"].toInt();
    pageWidget->setMaxPage(ceil(static_cast<double>(totalCount) / pageSize));
    QStandardItemModel *model;
    model = new QStandardItemModel(this);

    // 定义表头信息
    QList<HeaderInfo> headers = {
        {"工程名称", "name"},
        {"井上爆破员", "operatorIdentity"},
        {"井下爆破员", "blasterIdentity"},
        {"井下地址", "addressPath"},
        {"雷管数量", "detSum"},
        {"起爆器数量", "blastCount"},
        {"起爆状态", "blastStatus"},
        {"操作", ""},
    };

    int headerCount = headers.size();
    QStringList headerLabels;
    QMap<int, QString> propMap;
    for (int i = 0; i < headers.size(); ++i) {
        headerLabels << headers[i].label;
        propMap[i] = headers[i].prop;
    }
    model->setHorizontalHeaderLabels(headerLabels);

    for (int row = 0; row < projectList.size(); ++row) {
        HProject &hProject = *projectList.at(row).data();
        for (int col = 0; col < headers.size(); ++col) {
            QString prop = propMap[col];
            QStandardItem *item = nullptr;

            if (!prop.isEmpty()) {
                QMetaProperty metaProp =
                    hProject.metaObject()->property(hProject.metaObject()->indexOfProperty(prop.toUtf8()));
                QVariant value = metaProp.read(&hProject);
                if (prop == "blastStatus") {
                    QString statusText;
                    if (value.toString() == BlastStatus::Created) {
                        statusText = "未注册";
                        item = new QStandardItem(statusText);
                        item->setForeground(QColor("#e7c66b"));
                    } else if (value.toString() == BlastStatus::Registered) {
                        statusText = "已注册";
                        item = new QStandardItem(statusText);
                        item->setForeground(QColor("#f3a361"));
                    } else if (value.toString() == BlastStatus::Blasted) {
                        statusText = "起爆完成";
                        item = new QStandardItem(statusText);
                        item->setForeground(QColor("#90d543"));
                    } else {
                        item = new QStandardItem(value.toString());
                    }
                } else if (prop == "operatorIdentity" || prop == "blasterIdentity") {
                    QJsonArray userInfoArray = getUserInfoByIds(value.toString().split(","));
                    QStringList userNames;
                    for (const QJsonValue &value : userInfoArray) {
                        if (value.isObject()) {
                            QJsonObject obj = value.toObject();
                            userNames.append(obj["name"].toString());
                        }
                    }
                    item = new QStandardItem(userNames.join(","));
                } else {
                    item = new QStandardItem(value.toString());
                }
            } else {
                item = new QStandardItem("操作");
                item->setTextAlignment(Qt::AlignCenter);
                item->setData(QSize(30, 30), Qt::SizeHintRole);
            }
            if (item) {
                item->setTextAlignment(Qt::AlignCenter);
                model->setItem(row, col, item);
            }
        }
    }
    ui->tableView->setModel(model);
    ui->tableView->horizontalHeader()->setSectionResizeMode(QHeaderView::Stretch);
    ui->tableView->setEditTriggers(QAbstractItemView::NoEditTriggers);
    ui->tableView->setAlternatingRowColors(true);
    // 创建并设置自定义委托
    ButtonDelegate *delegate = new ButtonDelegate(headerCount - 1, this);
    QObject::connect(delegate, &ButtonDelegate::buttonClicked, [=](int row, int column) {
        // 获取 HPlan 引用
        HProject &HPlanDeleteOrUpdate = *projectList.at(row).data();
        updateId = HPlanDeleteOrUpdate.getId();
        if (column == 1) {
            deleteButtonClicked(HPlanDeleteOrUpdate);
        }
        if (column == 2) {
            updateButtonClicked(HPlanDeleteOrUpdate);
        }
    });
    ui->tableView->setItemDelegateForColumn(headerCount - 1, delegate);
    LoadingWidget::hideLoading();
}

void BlastProjectPage::PageChanged(int page) {
    currentPage = page;
    loadDataFromSource(currentPage, pageSize);
}

void BlastProjectPage::onComboBoxIndexChanged(int index) {
    QVariant variant = pageWidget->getComboBox()->itemData(index);
    int value = variant.toInt();
    pageSize = value;
    currentPage = 1;
    loadDataFromSource(currentPage, pageSize);
}

void BlastProjectPage::initUserAndDialogOptions(const QJsonArray &dataArray) {
    QJsonArray blasterDeployerArray;
    QJsonArray operatorArray;
    for (const QJsonValue &value : dataArray) {
        if (value.isObject()) {
            QJsonObject obj = value.toObject();
            if (obj.contains("roleIds")) {
                QJsonArray roleIdsArray = obj["roleIds"].toArray();
                // Note: 一个用户可能存在多个角色
                if (roleIdsArray.contains(BLASTER_ROLE_ID)) {  // 井下爆破员
                    blasterDeployerArray.append(obj);
                }
                if (roleIdsArray.contains(OPERATOR_ROLE_ID)) {  // 井上爆破员
                    operatorArray.append(obj);
                }
            }
        }
    }

    m_dialog->initComboBoxBlaster(blasterDeployerArray);
    m_dialog->initComboBoxOperator(operatorArray);
}

void BlastProjectPage::addressOptions(const QJsonArray &dataArrayAddress) {
    m_dialog->SetL1AddressOptions(dataArrayAddress);
    // dialog->setChildOptions(dataArrayAddress);
}

void BlastProjectPage::parseJsonString(const QJsonDocument &jsonDoc) {
    QJsonObject jsonObj = jsonDoc.object();
    QJsonObject dataObj = jsonObj["data"].toObject();
    dataArray = dataObj["list"].toArray();
}

void BlastProjectPage::on_pushButton_clicked() {
    // m_dialog->setModal(false);
    // m_dialog->clearFormData();
    // m_dialog->setOperationType(0);
    // int ref = m_dialog->exec();
    // if (ref == QDialog::Accepted) {};

    refreshData();
}

void BlastProjectPage::createProject(const QMap<QString, QString> &data) {
    HProject project;
    project.setName(data.value("name"));
    project.setDetSum(data.value("detNum"));
    project.setAddressUuid(data.value("addressUuid"));
    project.setAddressPath(data.value("addressPath"));
    project.setBlasterIdentity(data.value("blasterIdentity"));
    project.setOperatorIdentity(data.value("operatorIdentity"));
    project.setLevel4Address(data.value("level4Address"));
    project.setBlastCount(data.value("blastCount"));
    project.setBlastStatus("1");
    project.setCompanyCode(COMPANY_CODE);
    project.setXmbh(data.value("xmbh"));
    project.setHtid(data.value("htid"));
    project.setUuid(QUuid::createUuid().toString(QUuid::WithoutBraces).remove(QRegularExpression("-")));
    project.setCreatedAt(QDateTime::currentDateTime());
    if (dao.addHProject(project)) {
        refreshData();
        publishAllPendingBlastProjects();
    }
}

QJsonArray BlastProjectPage::getUserInfoByIds(const QStringList &userIds) {
    QJsonArray userInfoArray;
    for (const QString &userId : userIds) {
        if (m_userNameById.contains(userId)) {
            QJsonObject userInfo;
            userInfo["name"] = m_userNameById[userId];
            userInfo["identity"] = userId;
            userInfoArray.append(userInfo);
        } else {
            // TODO
            continue;
        }
    }
    return userInfoArray;
}

void BlastProjectPage::publishAllPendingBlastProjects() {
    QJsonObject extParams{{"blast_status", QJsonArray({"1", "2"})}};
    QJsonObject result = backendAPIManager::getHProjects(currentPage, pageSize, extParams);

    QJsonArray jsonArray = result["data"].toObject()["list"].toArray();
    for (int i = 0; i < jsonArray.size(); ++i) {
        QJsonObject projectObj = jsonArray[i].toObject();
        QStringList blasterIds = projectObj["blasterIdentity"].toString().split(",");
        QStringList operatorIds = projectObj["operatorIdentity"].toString().split(",");
        // remove blasterIdentity and operatorIdentity from projectObj
        projectObj.remove("blasterIdentity");
        projectObj.remove("operatorIdentity");
        projectObj.remove("blasterName");
        projectObj.remove("operatorName");

        QJsonArray blasterInfo = getUserInfoByIds(blasterIds);
        QJsonArray operatorInfo = getUserInfoByIds(operatorIds);

        projectObj["underground_blasters"] = blasterInfo;
        projectObj["ground_blasters"] = operatorInfo;

        jsonArray[i] = QJsonValue(projectObj);
    }
    QJsonDocument jsonDoc(jsonArray);
    QByteArray jsonData = jsonDoc.toJson(QJsonDocument::Indented);

    mainMqttClient->sendMessage(MQTT_TOPIC_COMPANY_PROJECTS_PUBLISH, jsonData, 2, true);
};

void BlastProjectPage::updateButtonClicked(const HProject &project) {
    m_dialog->setModal(false);
    m_dialog->setOperationType(1);
    m_dialog->setFormData(project);
    int ref = m_dialog->exec();
    if (ref == QDialog::Accepted) {};
}

void BlastProjectPage::updateProject(const QMap<QString, QString> &data) {
    HProject project;
    project.setId(updateId);
    project.setName(data.value("name"));
    project.setDetSum(data.value("detNum"));
    project.setAddressUuid(data.value("addressUuid"));
    project.setBlasterIdentity(data.value("blasterIdentity"));
    project.setOperatorIdentity(data.value("operatorIdentity"));
    project.setUpdatedAt(QDateTime::currentDateTime());
    if (dao.updateHProject(project)) {
        refreshData();
        publishAllPendingBlastProjects();
    }
}
void BlastProjectPage::deleteButtonClicked(const HProject &project) {
    if (dao.deleteHProject(project)) {
        refreshData();
        publishAllPendingBlastProjects();
    }
}

void BlastProjectPage::messageAndTopicReceived(const QByteArray &message, const QMqttTopicName &topic) {
    QJsonDocument jsonDoc = QJsonDocument::fromJson(message);
    if (!jsonDoc.isNull() && jsonDoc.isObject()) {
        QJsonObject jsonObj = jsonDoc.object();
        if (jsonObj.contains("uuid") && jsonObj.contains("status")) {
            QJsonValue uuidValue = jsonObj["uuid"];
            QJsonValue statusValue = jsonObj["status"];
            if (statusValue.isString() && statusValue.toString() == "1") {
                if (uuidValue.isNull()) {
                    qDebug() << "uuid 的值为 null";
                } else {
                    QString uuid = uuidValue.toString();
                    updateProjectUUId(uuid);
                }
            }
        }
    }
}

void BlastProjectPage::updateProjectUUId(QString uuid) { dao.updateBlastStatusByUuid(uuid, "2"); }

QString BlastProjectPage::getUserNameByIds(const QStringList &userIds) {
    QStringList userNames;
    for (const QString &userId : userIds) {
        if (m_userNameById.contains(userId)) {
            userNames.append(m_userNameById[userId]);
        }
    }
    return userNames.join(",");
}

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