mainwindow.cpp

/**
 * @file mainwindow.cpp
 * @brief Implementation file for the main window.
 * @author howdy213
 * @date 2026-05-04
 * @version 2.0.0
 *
 * @copyright Copyright 2025-2026 howdy213
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
#include "mainwindow.h"
#include "WECore/plugin/wpluginmanager.h"
#include "aboutwindow.h"
#include "ui_mainwindow.h"
#include "Shared/flowlayout.h"

#include "WECore/def/wedef.h"
#include "WECore/we/we.h"
#include "WECore/we/webase.h"
#include "WECore/file/wpath.h"
#include "WECore/file/wshellexecute.h"
#include "WECore/widget/wwidgetmanager.h"
#include "WECore/plugin/wplugindata.h"

#include <QButtonGroup>
#include <QFile>
#include <QLockFile>
#include <QPluginLoader>
#include <QToolBar>

using namespace we::Consts;
using namespace we;

/**
 * @class MainWindowPrivate
 * @brief Private implementation class for MainWindow.
 */
class MainWindowPrivate {
public:
    Ui::MainWindow *ui = nullptr;
    AboutWindow *aboutWnd = nullptr;
};

/**
 * @brief Constructs the main window.
 * @param parent The parent widget.
 */
MainWindow::MainWindow(QWidget *parent) : QMainWindow(parent) {
    d = new MainWindowPrivate;
    d->ui = new Ui::MainWindow;
    d->ui->setupUi(this);
}

/**
 * @brief Destroys the main window.
 */
MainWindow::~MainWindow() {
    delete d->ui;
    d->ui = nullptr;
    delete d->aboutWnd;
    d->aboutWnd = nullptr;
}

/**
 * @brief Initializes the main window after construction.
 */
void MainWindow::init()
{
    initWindow();
    initPlugin();
    initList();
    initTable();
    initWidgetTable();
    initMenu();
    PClass->widgetManager()->initWidget();
}

/**
 * @brief Adds an action to the plugin toolbar.
 * @param action The action to add.
 */
void MainWindow::addToolBarAction(QAction *action) {
    if (!action) return;
    QToolBar *toolbar = findChild<QToolBar*>();
    if (!toolbar) {
        toolbar = addToolBar("插件工具");
    }
    toolbar->addAction(action);
}

/**
 * @brief Adds a dock widget as an extension.
 * @param dock The dock widget to add.
 */
void MainWindow::addExtensionDock(QDockWidget *dock) {
    if (!dock) return;
    addDockWidget(Qt::RightDockWidgetArea, dock);
}

/**
 * @brief Returns a pointer to the UI object.
 * @return Pointer to the main window's UI.
 */
Ui::MainWindow *MainWindow::getUiPointer()
{
    return d->ui;
}

/**
 * @brief Initializes window properties (size, background, etc.).
 */
void MainWindow::initWindow() {
    this->resize(1200, 800);
    this->setMinimumSize(800, 566);

    QPalette pal(this->palette());
    pal.setColor(QPalette::ColorRole::Window, QColor(255, 255, 255));
    this->setAutoFillBackground(true);
    this->setPalette(pal);
}

/**
 * @brief Loads and initializes all plugins.
 */
void MainWindow::initPlugin() {
    auto man = PClass->pluginManager();
    auto insts = man->allPluginsInst();
    foreach (auto inst, insts) {
        if (!man->loadPlugin(inst))
            continue;
        man->initPlugin(inst);
    }
}

/**
 * @brief Initializes the list of buttons from the link file.
 */
void MainWindow::initList()
{
    QStringList links = ReadLinkFile();
    QWidget *btnContainer = d->ui->btnWidget;
    if (!btnContainer) {
        qWarning() << "btnWidget is null!";
        return;
    }

    QLayout *oldLayout = btnContainer->layout();
    if (oldLayout) {
        QLayoutItem *child;
        while ((child = oldLayout->takeAt(0)) != nullptr) {
            if (child->widget())
                delete child->widget();
            delete child;
        }
        delete oldLayout;
    } else {
        QList<QPushButton*> btns = btnContainer->findChildren<QPushButton*>();
        for (QPushButton *btn : std::as_const(btns))
            delete btn;
    }

    FlowLayout *flowLayout = new FlowLayout(btnContainer, 10, 10, 10);
    btnContainer->setLayout(flowLayout);

    for (const QString &link : std::as_const(links)) {
        if (link.trimmed().isEmpty())
            continue;

        QPushButton *btn = new QPushButton(link, btnContainer);
        btn->setMaximumWidth(200);
        btn->setMinimumHeight(100);
        btn->setSizePolicy(QSizePolicy::Minimum, QSizePolicy::Fixed);
        btn->setToolTip(link);
        connect(btn, &QPushButton::clicked, this, [this,link]() {
            this->d->ui->editCmd->setText(link);
        });

        flowLayout->addWidget(btn);
    }
}

/**
 * @brief Initializes the plugin information table.
 */
void MainWindow::initTable() {
    d->ui->tablePlugin->setEditTriggers(QAbstractItemView::NoEditTriggers);
    d->ui->tablePlugin->setSelectionMode(QAbstractItemView::NoSelection);

    QFont font;
    font.setPointSize(10);
    font.setFamily("黑体");

    QColor color(0, 0, 0);

    int colCount = 4;

    d->ui->tablePlugin->setColumnCount(colCount);

    createCol(0, "插件名", font, color);
    createCol(1, "版本", font, color);
    createCol(2, "作者", font, color);
    createCol(3, "路径", font, color);

    auto list = WE::inst()->getWEClass()->pluginManager()->allPluginsInst();
    d->ui->tablePlugin->setRowCount(list.length());
    auto it = list.begin();
    for (int i = 0; i <= list.length() - 1; i++) {
        createRow(i, *it);
        it++;
    }
}

/**
 * @brief Initializes the widget catalog table.
 */
void MainWindow::initWidgetTable() {
    d->ui->tableWidget->setEditTriggers(QAbstractItemView::NoEditTriggers);
    d->ui->tableWidget->setSelectionMode(QAbstractItemView::NoSelection);

    QFont font;
    font.setPointSize(10);
    font.setFamily("黑体");
    QColor color(0, 0, 0);

    const int colCount = 2;
    d->ui->tableWidget->setColumnCount(colCount);

    QTableWidgetItem *headerName = new QTableWidgetItem("名称");
    headerName->setFont(font);
    headerName->setForeground(QBrush(color));
    headerName->setTextAlignment(Qt::AlignHCenter | Qt::AlignVCenter);
    d->ui->tableWidget->setHorizontalHeaderItem(0, headerName);
    d->ui->tableWidget->horizontalHeader()->setSectionResizeMode(
        0, QHeaderView::ResizeToContents);

    QTableWidgetItem *headerDesc = new QTableWidgetItem("描述");
    headerDesc->setFont(font);
    headerDesc->setForeground(QBrush(color));
    headerDesc->setTextAlignment(Qt::AlignHCenter | Qt::AlignVCenter);
    d->ui->tableWidget->setHorizontalHeaderItem(1, headerDesc);
    d->ui->tableWidget->horizontalHeader()->setSectionResizeMode(
        1, QHeaderView::ResizeToContents);

    auto widgets = PClass->widgetManager()->getWidgets();
    d->ui->tableWidget->setRowCount(widgets.size());

    for (int i = 0; i < widgets.size(); ++i) {
        QObject *widget = widgets[i];
        QString name =
            PClass->widgetManager()->getAttr(widget, Widget::Name).toString();
        QString desc =
            PClass->widgetManager()->getAttr(widget, Widget::Desc).toString();

        QTableWidgetItem *itemName = new QTableWidgetItem(name);
        itemName->setTextAlignment(Qt::AlignHCenter | Qt::AlignVCenter);
        d->ui->tableWidget->setItem(i, 0, itemName);

        QTableWidgetItem *itemDesc = new QTableWidgetItem(desc);
        itemDesc->setTextAlignment(Qt::AlignHCenter | Qt::AlignVCenter);
        d->ui->tableWidget->setItem(i, 1, itemDesc);
    }
}

/**
 * @brief Handles double-click on the widget table.
 * @param row Row index.
 * @param column Column index.
 */
void MainWindow::on_tableWidget_cellDoubleClicked(int row, int column) {
    QTableWidgetItem *item = d->ui->tableWidget->item(row, column);
    if (item) {
        d->ui->editCmd->setText(item->text());
    }
}

/**
 * @brief Initializes the menu bar.
 */
void MainWindow::initMenu() {
    QMenuBar *menuBar = this->menuBar();
    QMenu *menuPlugin = new QMenu("插件");
    QAction *actNew = new QAction("创建新插件");
    QAction *actImport = new QAction("导入现有插件");
    QAction *actExport = new QAction("导出");
    QMenu *menuOption = new QMenu("选项");
    QAction *actSetting = new QAction("设置");
    QAction *actReset = new QAction("重启");
    QMenu *menuAbout = new QMenu("关于");
    QAction *actAbout = new QAction("关于");

    menuBar->addMenu(menuPlugin);
    menuPlugin->addAction(actNew);
    menuPlugin->addAction(actImport);
    menuPlugin->addAction(actExport);
    menuBar->addMenu(menuOption);
    menuOption->addAction(actSetting);
    menuOption->addAction(actReset);
    menuBar->addMenu(menuAbout);
    menuAbout->addAction(actAbout);

    connect(actAbout, &QAction::triggered, this, &MainWindow::about);
    connect(actReset, &QAction::triggered, this, &MainWindow::restart);
    connect(actSetting, &QAction::triggered, this, []() {
        WShellExecute::syncExecute(WPath().getModuleFolder() + Config::ConfigPath);
    });
}

/**
 * @brief Creates a column header for the plugin table.
 * @param col Column index.
 * @param title Header text.
 * @param font Font for the header.
 * @param color Color for the header text.
 */
void MainWindow::createCol(int col, QString title, QFont font, QColor color) {
    QTableWidgetItem *item = new QTableWidgetItem(title);
    item->setFont(font);
    item->setForeground(QBrush(color));
    item->setTextAlignment(Qt::AlignHCenter | Qt::AlignVCenter);
    d->ui->tablePlugin->setHorizontalHeaderItem(col, item);
    d->ui->tablePlugin->horizontalHeader()->setSectionResizeMode(
        col, QHeaderView::ResizeToContents);
}

/**
 * @brief Creates a row in the plugin table for a given plugin.
 * @param row Row index.
 * @param info Pointer to the plugin object.
 */
void MainWindow::createRow(int row, WPlugin *info) {
    if (row >= d->ui->tablePlugin->rowCount())
        return;
    QTableWidgetItem *item = nullptr;
    QString str;
    QStringList list = {Plugin::Name, Plugin::Version, Plugin::Author,
                        Plugin::Path};
    for (int i = 0; i <= list.length() - 1; i++) {
        str = qvariant_cast<QString>(info->getMetaData(list[i]));
        item = new QTableWidgetItem(str);
        item->setTextAlignment(Qt::AlignHCenter | Qt::AlignVCenter);
        d->ui->tablePlugin->setItem(row, i, item);
    }
}

/**
 * @brief Reads the link file and returns a list of links.
 * @return QStringList containing the links.
 */
QStringList MainWindow::ReadLinkFile() {
    QFile file(WPath().getModuleFolder() + Plugins::ConfigFolder + "link.txt");
    QStringList list;
    if (file.open(QIODevice::ReadOnly | QIODevice::Text)) {
        QTextStream in(&file);
        QString str = in.readAll();
        list = str.split('\n');
        file.close();
    }
    return list;
}

/**
 * @brief Handles the close event (hides the window instead of closing).
 * @param event The close event.
 */
void MainWindow::closeEvent(QCloseEvent *event) {
    event->ignore();
    setVisible(false);
}

/**
 * @brief Handles system tray icon activation.
 * @param reason The activation reason.
 */
void MainWindow::tray(QSystemTrayIcon::ActivationReason reason) {
    switch (reason) {
    case QSystemTrayIcon::DoubleClick:
        if (!this->isVisible())
            this->showNormal();
        else
            this->activateWindow();
        break;
    default:
        break;
    }
}

/**
 * @brief Shows the main window (used from tray context menu).
 */
void MainWindow::showPanel() {
    this->show();
    this->raise();
}

/**
 * @brief Opens the About dialog.
 */
void MainWindow::about() {
    if (d->aboutWnd == nullptr)
        d->aboutWnd = new AboutWindow;
    if (d->aboutWnd->isHidden())
        d->aboutWnd->show();
    d->aboutWnd->activateWindow();
}

/**
 * @brief Restarts the application via launcher.
 */
void MainWindow::restart() {
    WShellExecute::asyncExecute(WPath().getModuleFolder()+"tools/WELauncher.exe","open","-t 500");
    QApplication::exit(0);
}

/**
 * @brief Handles double-click on plugin table: sets clicked text to command line.
 * @param row Row index.
 * @param column Column index.
 */
void MainWindow::on_tablePlugin_cellDoubleClicked(int row, int column) {
    auto item = d->ui->tablePlugin->item(row, column);
    d->ui->editCmd->setText(item->text());
}

/**
 * @brief Executes the command entered in the edit field.
 */
void MainWindow::on_btnCmd_clicked() {
    QString str = d->ui->editCmd->text();
    QString topic;
    QString param;

    str = str.trimmed();
    if (str.isEmpty()) {
        return;
    }

    QChar firstChar = str[0];
    if (firstChar == '"') {
        int endIdx = -1;
        for (int i = 1; i < str.length(); ++i) {
            if (str[i] == firstChar && (i == 0 || str[i-1] != '\\')) {
                endIdx = i;
                break;
            }
        }
        if (endIdx != -1) {
            topic = str.mid(1, endIdx - 1);
            QString remaining = str.mid(endIdx + 1).trimmed();
            param = remaining;
        } else {
            int spaceIdx = str.indexOf(' ');
            if (spaceIdx != -1) {
                topic = str.left(spaceIdx);
                param = str.mid(spaceIdx + 1);
            } else {
                topic = str;
                param = QString();
            }
        }
    } else {
        int spaceIdx = str.indexOf(' ');
        if (spaceIdx != -1) {
            topic = str.left(spaceIdx);
            param = str.mid(spaceIdx + 1);
        } else {
            topic = str;
            param = QString();
        }
    }

    WMessage msg;
    msg.from = WESender;
    msg.command = param;

    WEvent event;
    event.topic = topic;
    event.msg = msg;

    WApp->getWEClass()->widgetManager()->publish(event);
}

/**
 * @brief Clears the command line edit field.
 */
void MainWindow::on_btnClear_clicked() { d->ui->editCmd->clear(); }

/**
 * @brief Refreshes the table content when the tab bar is clicked.
 * @param index Index of the clicked tab.
 */
void MainWindow::on_tabWidget_tabBarClicked(int index)
{
    switch (index) {
    case 0:
    {
        d->ui->tablePlugin->clearContents();
        auto pluginList = PClass->pluginManager()->allPluginsInst();
        d->ui->tablePlugin->setRowCount(pluginList.size());
        for (int i = 0; i < pluginList.size(); ++i) {
            createRow(i, pluginList[i]);
        }
        break;
    }
    case 1:
    {
        d->ui->tableWidget->clearContents();
        auto widgets = PClass->widgetManager()->getWidgets();
        d->ui->tableWidget->setRowCount(widgets.size());
        for (int i = 0; i < widgets.size(); ++i) {
            QString name = PClass->widgetManager()->getAttr(widgets[i], Widget::Name).toString();
            QString desc = PClass->widgetManager()->getAttr(widgets[i], Widget::Desc).toString();
            d->ui->tableWidget->setItem(i, 0, new QTableWidgetItem(name));
            d->ui->tableWidget->setItem(i, 1, new QTableWidgetItem(desc));
        }
        break;
    }
    default:
        break;
    }
}