mainwindow.cpp

#include "mainwindow.h"
#include <QVBoxLayout>
#include <QWidget>
#include <QMessageBox>
#include <QLabel>
#include <QPushButton>
#include "global.h"
#include "util.h"
#include "jwt-cpp/jwt.h"
#include "views/settingview.h"
#include "views/organizationview.h"
#include "views/createdomainview.h"
#include "views/domaininstanceview.h"
#include "views/instantiatedomainview.h"
#include "views/domaininfoview.h"
#include "style.h"
#include "dscc/core/db/domain.h"
#include "dscc/core/db/domain_instance.h"
#include "components/treeitemwidget.h"
#include <QApplication>

MainWindow::MainWindow(const QString &token, QWidget *parent)
    : QMainWindow(parent), username(token) {
    setWindowTitle("数据安全柜 - 欢迎使用");
    //resize(800, 600);
    setFixedSize(1080, 720);
    setupUI();
    loadDomain();
    setupConnections();
}

MainWindow::~MainWindow() = default;

void MainWindow::onLogoutButtonClicked() {
    if (QMessageBox::question(this, "确认退出", "确定要退出登录吗？") == QMessageBox::Yes) {
        emit logoutRequested();
    }
}

void MainWindow::setupUI(){
    // 创建主分割器
    QSplitter *mainSplitter = new QSplitter(Qt::Horizontal, this);
    setCentralWidget(mainSplitter);

    // 左侧面板 - 垂直布局
    QWidget *leftPanel = new QWidget(mainSplitter);
    makeLeftPanel(*leftPanel);
    QVBoxLayout *leftLayout = new QVBoxLayout(leftPanel);
    leftLayout->setContentsMargins(10, 10, 10, 10);
    leftLayout->setSpacing(24);

    // 组织名称（可点击的标签）
    auto & g = global::get_mutable_instance();

    auto value = parse_json(g.current_organization()->info);
    if(value.empty()){
        QMessageBox::critical(this, "Invalid data", "空，无法解析JSON");
    }
    QString org = QString::fromStdString(value["orgName"].asString());

    organizationButton = new QPushButton(org, leftPanel);
    organizationButton->setStyleSheet("font-size: 24pt; font-weight: bold; color: #1890FF;");

    leftLayout->addWidget(organizationButton);

    leftLayout->setSpacing(24);
    createDomain = new QPushButton("+创建安全域", leftPanel);
    leftLayout->addWidget(createDomain, 0, Qt::AlignHCenter);
    leftLayout->setSpacing(10);
    makeDefaultButton(*createDomain);
    createDomain->setMaximumWidth(160);

    // 树形控件
    treeWidget = new QTreeWidget(leftPanel);
    treeWidget->setColumnCount(1);
    treeWidget->setMinimumWidth(200);
    treeWidget->setHeaderHidden(true);
    makeLeftTree(*treeWidget);
    leftLayout->addWidget(treeWidget, 1); // 占主要空间

    leftLayout->setSpacing(24);
    auto decoded = jwt::decode(g.current_organization()->access_token);

    auto payload = decoded.get_payload();
    auto r = parse_json(payload);
    auto name = r["name"].asString();
    auto avatar = r["avatar"].asString();


    userInfoButton = new QPushButton(QString::fromStdString(name), leftPanel);
    userInfoButton->setStyleSheet("font-size: 16pt; font-weight: bold; color: #0066cc;");

    QIcon icon(":/logo/resources/avator.svg");
    userInfoButton->setIconSize(QSize(userInfoButton->height(), userInfoButton->height()));
    userInfoButton->setIcon(icon);

    leftLayout->addWidget(userInfoButton);

    leftLayout->setSpacing(24);
    // 设置按钮
    settingsButton = new QPushButton("设置", leftPanel);
    leftLayout->addWidget(settingsButton, 0, Qt::AlignHCenter);
    leftLayout->setSpacing(24);
    settingsButton->setFixedWidth(160);
    makeHelpButton(*settingsButton);


    helpButton = new QPushButton("帮助中心", leftPanel);
    //organizationLabel->setStyleSheet("font-size: 16pt; font-weight: bold; color: #0066cc; text-decoration: underline;");
    leftLayout->addWidget(helpButton, 0, Qt::AlignHCenter);
    makeHelpButton(*helpButton);
    helpButton->setFixedWidth(160);
    leftLayout->addSpacing(24);


    // 右侧面板 - 使用堆叠窗口管理不同视图
    rightPanel = new QWidget(mainSplitter);
    rightLayout = new QVBoxLayout(rightPanel);
    rightLayout->setContentsMargins(0, 0, 0, 0);

    stackedWidget = new QStackedWidget(rightPanel);
    rightLayout->addWidget(stackedWidget);

    // 创建各个视图并添加到堆叠窗口
    organizationView = new OrganizationView(rightPanel);

    createDomainView = new CreateDomainView(rightPanel);
    domainInfoView = new DomainInfoView(rightPanel);
    instantiateDomainView = new InstantiateDomainView(rightPanel);
    domainInstanceView = new DomainInstanceView(rightPanel);

    settingView = new SettingView(rightPanel);

    stackedWidget->addWidget(organizationView);
    stackedWidget->addWidget(domainInfoView);
    stackedWidget->addWidget(createDomainView);
    stackedWidget->addWidget(instantiateDomainView);
    stackedWidget->addWidget(domainInstanceView);

    stackedWidget->addWidget(settingView);
    /*
    organizationView = new OrganizationView(rightPanel);
    departmentView = new DepartmentView(rightPanel);
    employeeView = new EmployeeView(rightPanel);


    stackedWidget->addWidget(organizationView);
    stackedWidget->addWidget(departmentView);
    stackedWidget->addWidget(employeeView);
    stackedWidget->addWidget(settingsView);


    */
    // 初始显示组织视图
    stackedWidget->setCurrentWidget(organizationView);

    // 设置分割器初始大小
    mainSplitter->setSizes({250, 750});
}
void MainWindow::setupConnections(){

    connect(settingsButton, &QPushButton::clicked, [this](){
        setMainWindowSubtitle("设置");
        this->treeWidget->clearSelection();
        this->stackedWidget->setCurrentWidget(settingView);
    });

    connect(settingView, &SettingView::logoutRequested,
            [this](){
                emit logoutRequested();
    });
    connect(createDomain, &QPushButton::clicked, [this](){
        setMainWindowSubtitle("创建安全域");
        this->treeWidget->clearSelection();
        this->stackedWidget->setCurrentWidget(createDomainView);
    });
    connect(organizationButton, &QPushButton::clicked, [this]{
        setMainWindowSubtitle("组织信息");
        this->treeWidget->clearSelection();
        this->stackedWidget->setCurrentWidget(organizationView);
    });
    connect(treeWidget, &QTreeWidget::itemClicked, this, &MainWindow::onTreeItemClicked);
    connect(createDomainView, &CreateDomainView::newDomain, this, &MainWindow::onNewDomain);
}


void MainWindow::setMainWindowSubtitle(const QString & sub){
    setWindowTitle("数据安全柜 - " + sub);
}

enum TreeItemTypeRoles{
    type = Qt::UserRole + 1,
    id,
    pub_key,
};
enum TreeTopItemType{
    domain_top,
    instance_top,
    domain_item,
    instance_item
};

void MainWindow::loadDomain(){
    domainTreeItem = new QTreeWidgetItem(treeWidget);
    domainTreeItem->setData(0, TreeItemTypeRoles::type, TreeTopItemType::domain_top);
    TreeItemWidget * domainWidget = new TreeItemWidget();
    domainWidget->setText(QString{"安全域"});
    domainWidget->setIcon(QApplication::style()->standardIcon(QStyle::SP_DirIcon));
    domainWidget->setBadgeCount(22);
    treeWidget->setItemWidget(domainTreeItem, 0, domainWidget);

    instanceTreeItem = new QTreeWidgetItem(treeWidget);
    instanceTreeItem->setData(0, TreeItemTypeRoles::type, TreeTopItemType::instance_top);
    TreeItemWidget * instanceWidget = new TreeItemWidget();
    instanceWidget->setText(QString{"安全域实例"});
    treeWidget->setItemWidget(instanceTreeItem, 0, instanceWidget);

    treeWidget->addTopLevelItem(domainTreeItem);
    treeWidget->addTopLevelItem(instanceTreeItem);

    auto & g = global::get_mutable_instance();

    WCDB::Database database(g.current_organization()->local_db_file);

    std::unordered_map<std::string, dscc::db::domain> pub_key_to_domain;
    auto ds_objects = database.getAllObjects<dscc::db::domain>(DB_DOMAIN);
    auto & ds = *ds_objects;
    auto ds1_objects = database.getAllObjects<dscc::db::domain_instance>(DB_DOMAIN_INSTANCE);
    auto & dis = *ds1_objects;

    for(int i = 0; i < ds.size(); i++){
        auto & d = ds[i];
        QTreeWidgetItem * item = new QTreeWidgetItem(QStringList{QString::fromStdString( d.name)});
        item->setData(0, TreeItemTypeRoles::type, TreeTopItemType::domain_item);
        item->setData(0, TreeItemTypeRoles::pub_key, QString::fromStdString(d.pub_key));
        domainTreeItem->addChild(item);
        pub_key_to_domain.insert(std::make_pair(d.pub_key, d));
    }
    for(int i = 0; i < dis.size(); i++){
        auto & d = dis[i];
        auto & name = pub_key_to_domain[d.pub_key].name;
        QTreeWidgetItem * item = new QTreeWidgetItem(QStringList{QString::fromStdString(name)});
        item->setData(0, TreeItemTypeRoles::type, TreeTopItemType::instance_item);
        item->setData(0, TreeItemTypeRoles::pub_key, QString::fromStdString(d.pub_key));
        item->setData(0, TreeItemTypeRoles::id, d.id);
        instanceTreeItem->addChild(item);
    }
}

void MainWindow::onTreeItemClicked(QTreeWidgetItem *item, int column){
    QTreeWidgetItem * parent = item->parent();
    auto type = item->data(0, TreeItemTypeRoles::type).toInt();
    if(parent == nullptr){
        item->setExpanded(!item->isExpanded());
        if(item->childCount() == 0){
            if(type == TreeTopItemType::domain_top){
                QMessageBox::information(nullptr, "安全域", "您还没有创建任何安全域，也没有对您可见的安全域!");
            }else if(type == TreeTopItemType::instance_top){
                QMessageBox::information(nullptr, "安全域实例", "您还没有实例化任何安全域");
            }
        }
        return ;
    }

    if(type == TreeTopItemType::domain_item){
        setMainWindowSubtitle("安全域详情");
        auto  item_pub_key = item->data(0, TreeItemTypeRoles::pub_key).toString().toStdString();
        domainInfoView->renderDomain(item_pub_key);
        stackedWidget->setCurrentWidget(domainInfoView);

    }else if(type == TreeTopItemType::instance_item){
        setMainWindowSubtitle("安全域实例详情");
        stackedWidget->setCurrentWidget(domainInstanceView);
    }

}

void MainWindow::onNewDomain(const std::string & pub_key){
    WCDB::Database database(global::get_mutable_instance().current_organization()->local_db_file);
    auto objects = database.getAllObjects<dscc::db::domain>(DB_DOMAIN, WCDB_FIELD(dscc::db::domain::pub_key) ==  pub_key);
    if(objects->size() == 0){
        QMessageBox::warning(nullptr, "错误", "找不到安全域");
        return ;
    }
    auto & d = (*objects)[0];
    QTreeWidgetItem * item = new QTreeWidgetItem(QStringList{QString::fromStdString( d.name)});
    item->setData(0, TreeItemTypeRoles::type, TreeTopItemType::domain_item);
    item->setData(0, TreeItemTypeRoles::pub_key, QString::fromStdString(d.pub_key));
    domainTreeItem->addChild(item);

    onTreeItemClicked(item, 0);
}