PowerModeler/source/mainwindow.cpp

#include "mainwindow.h"
#include "./ui_mainwindow.h"

#include "dbManager.h"
#include "dbBrowser.h"
#include "connectionDialog.h"
#include "dbStructureView.h"
#include "dbStructureModel.h"
#include "logger.h"
#include "sqlQueryExecutor.h"
#include "modelInfoEditDialog.h"

#include <QKeyEvent>
#include <QScreen>

MainWindow::MainWindow(QWidget *parent)
    : QMainWindow(parent)
    , ui(new Ui::MainWindow)
    , m_dbManager(nullptr)
    , m_dbBrowser(nullptr)
    , m_pMessageDialog(nullptr)
    , m_pConnectionDialog(nullptr)
    , m_pModelInfoDialog(nullptr)
{
    ui->setupUi(this);

    initialize();
}

MainWindow::~MainWindow()
{
    delete ui;
    LOG_INFO("Application", "------------------------------------------------Application exit------------------------------------------------");
}

bool MainWindow::eventFilter(QObject* obj, QEvent* event)
{
    QDialog *pDialog = qobject_cast<QDialog*>(obj);
    if(pDialog)
    {
        if(event->type() == QEvent::KeyPress)
        {
            QKeyEvent* pKeyEvent = static_cast<QKeyEvent*>(event);
            if (pKeyEvent->key() == Qt::Key_Escape)
            {
                return true;
            }
        }
    }
    return QObject::eventFilter(obj, event);
}

void MainWindow::initialize()
{    
    connect(ui->connectAction, &QAction::triggered, this, &MainWindow::onActionTrigger_connect);
    connect(ui->disconnectAction, &QAction::triggered, this, &MainWindow::onActionTrigger_disconnect);

    connect(ui->createTableAction, &QAction::triggered, this, &MainWindow::onActionTrigger_addModel);
    connect(ui->deleteTableAction, &QAction::triggered, this, &MainWindow::onActionTrigger_removeModel);

    connect(&SqlQueryExecutor::instance(), &SqlQueryExecutor::errorOccurred, this, &MainWindow::onSIG_errorFormSQLExecutor);

    m_dbManager = new DatabaseManager(this);
    connect(m_dbManager, &DatabaseManager::errorOccurred, this, &MainWindow::onSIG_errorFromDBManger);
    connect(m_dbManager, &DatabaseManager::connectionStatusChanged, this, &MainWindow::onSIG_connectionStatusChanged);

    m_dbBrowser = new DatabaseBrowser(this);
    m_dbBrowser->setMainWindow(this);
    ui->layoutAttributeBrowser->addWidget(m_dbBrowser);

    m_pDBStrutureView = new DBStructureView(m_dbManager, this);
    m_pDBStrutureView->setMainWindow(this);
    connect(m_pDBStrutureView, &DBStructureView::actionTrigger_addModel, this, &MainWindow::onActionTrigger_addModel);
    connect(m_pDBStrutureView, &DBStructureView::openAttributeInfo, this, &MainWindow::onSIG_openAttributeInfo);
    ui->layoutDBStructure->addWidget(m_pDBStrutureView);
    m_pDBStrutureModel = new DBStructureModel(this);
    m_pDBStrutureModel->setMainWindow(this);
    m_pDBStrutureView->setModel(m_pDBStrutureModel);

    QScreen* screen = this->screen();
    int nWidth = screen->size().width() * 0.8;
    int nHeight = screen->size().height() * 0.8;
    this->resize(nWidth, nHeight);
    //在子部件内含有复杂布局和嵌套空间的时候，setStretchFactor会失效，固采用setSizes的方法来设置初始大小
    QVector<double> factor{1.0, 6.0};
    int nDBStructureWidth = nWidth * (factor.at(0) / (factor.at(0) + factor.at(1)));
    int nDBBrowserWidth = nWidth - nDBStructureWidth;
    ui->splitter->setSizes({nDBStructureWidth, nDBBrowserWidth});
    // ui->splitter->setStretchFactor(0, 1);
    // ui->splitter->setStretchFactor(1, 7);
}

void MainWindow::showMessageDialog(MessageDialogType type,const QString& strTitle,const QString& strContent)
{
    if(m_pMessageDialog == nullptr)
    {
        m_pMessageDialog = new MessageDialog(this);
        m_pMessageDialog->installEventFilter(this);
    }

    m_pMessageDialog->setMessage(type, strTitle, strContent);
    int nX = this->geometry().x() + (this->width() - m_pMessageDialog->width()) * 0.5;
    int nY = this->geometry().y() + (this->height() - m_pMessageDialog->height()) * 0.5;
    m_pMessageDialog->move(nX, nY);
    //m_pMessageDialog->raise();
    // if(type == type_question)
    //     m_pMessageDialog->exec();
    // else
    //     m_pMessageDialog->show();
    m_pMessageDialog->exec();
}
void MainWindow::hideMessageDialog()
{
    if(m_pMessageDialog && m_pMessageDialog->isVisible())
        m_pMessageDialog->close();
}

const QString MainWindow::getCurConnection()
{
    if(m_pDBStrutureView)
        return m_pDBStrutureView->curConnection();
    else
        return QString("");
}

void MainWindow::onActionTrigger_connect()
{
    if(m_pConnectionDialog && m_pConnectionDialog->isVisible())
        return;

    // if(m_dbManager->conncetions().count() >=1 ) //暂时只支持一个链接
    // {
    //     showMessageDialog(type_information, QString::fromWCharArray(L"提示"), QString::fromWCharArray(L"若要开启新的图模库链接，请先断开当前链接"));
    //     return;
    // }

    if(m_pConnectionDialog == nullptr)
    {
        m_pConnectionDialog = new ConnectionDialog(this);
        m_pConnectionDialog->setMainWindow(this);
        m_pConnectionDialog->installEventFilter(this);
        connect(m_pConnectionDialog, &ConnectionDialog::addConnection, this, &MainWindow::onSIG_addConnection);
    }

    int nX = this->geometry().x() + (this->width() - m_pConnectionDialog->width()) * 0.5;
    int nY = this->geometry().y() + (this->height() - m_pConnectionDialog->height()) * 0.5;
    m_pConnectionDialog->move(nX, nY);
    // QPoint centerPos = this->mapToGlobal(this->rect().center());
    // centerPos -= QPoint(m_pConnectionDialog->width()/2, m_pConnectionDialog->height()/2);
    // m_pConnectionDialog->move(centerPos);
    m_pConnectionDialog->show();
}
void MainWindow::onActionTrigger_disconnect()
{
    if(m_pDBStrutureView)
        m_pDBStrutureView->disconnectCurConnection();
}

void MainWindow::onActionTrigger_addModel()
{
    if(m_pModelInfoDialog && m_pModelInfoDialog->isVisible())
        return;

    if(m_pModelInfoDialog == nullptr)
    {
        m_pModelInfoDialog = new ModelInfoEditDialog(this);
        m_pModelInfoDialog->setMainWindow(this);
        m_pModelInfoDialog->installEventFilter(this);
        connect(m_pModelInfoDialog, &ModelInfoEditDialog::addModel, this, &MainWindow::onSIG_addModel);
    }

    int nX = this->geometry().x() + (this->width() - m_pModelInfoDialog->width()) * 0.5;
    int nY = this->geometry().y() + (this->height() - m_pModelInfoDialog->height()) * 0.5;
    m_pModelInfoDialog->move(nX, nY);
    m_pModelInfoDialog->setState(DS_New);
    m_pModelInfoDialog->exec();
}
void MainWindow::onActionTrigger_removeModel()
{
    if(m_pDBStrutureView)
        m_pDBStrutureView->onActionTrigger_removeModel();
}

void MainWindow::onSIG_addConnection(DatabaseConfig& config)
{
    bool result = m_dbManager->addDatabase(config);
    if(!result)
    {
        //m_pConnectionDialog->setErrorInfo(m_lastSqlError.databaseText());
        return;
    }

    m_pConnectionDialog->close();
    m_pDBStrutureModel->addConnection(config.strConnectionName, config.strDBType);
}

void MainWindow::onSIG_errorFromDBManger(const QString& strConnectionName, const QString& error)
{
    //m_lastSqlError = error;
    showMessageDialog(type_warning, QString::fromWCharArray(L"错误"),error);
}

void MainWindow::onSIG_errorFormSQLExecutor(const QString& error)
{
    showMessageDialog(type_warning, QString::fromWCharArray(L"错误"),error);
}

void MainWindow::onSIG_connectionStatusChanged(const QString& strConnectionName, bool bConnected)
{
    ui->createTableAction->setEnabled(bConnected);
    ui->deleteTableAction->setEnabled(bConnected);
    ui->importAciton->setEnabled(bConnected);
    ui->exportAction->setEnabled(bConnected);
    ui->refreshAction->setEnabled(bConnected);
    ui->saveAction->setEnabled(bConnected);
}

void MainWindow::onSIG_addModel(Model& model)
{
    QString connection = m_pDBStrutureView->curConnection();
    m_pDBStrutureModel->addDataModel(connection, model);
}

void MainWindow::onSIG_openAttributeInfo(const QString& connection, ModelAttributeGroup& attributeGroup)
{
    if(m_dbBrowser)
        m_dbBrowser->addTab_attribute(connection, attributeGroup);
}