Qv2ray/src/ui/w_MainWindow.hpp

#pragma once

#include "common/HTTPRequestHelper.hpp"
#include "common/LogHighlighter.hpp"
#include "components/pac/QvPACHandler.hpp"
#include "components/speedchart/speedwidget.hpp"
#include "core/handler/ConfigHandler.hpp"
#include "ui/messaging/QvMessageBus.hpp"
#include "ui_w_MainWindow.h"

#include <QMainWindow>
#include <QMenu>
#include <QScrollBar>
#include <QSystemTrayIcon>

// ==========================================================================================
#include "ui/widgets/ConnectionInfoWidget.hpp"
#include "ui/widgets/ConnectionItemWidget.hpp"

enum MW_ITEM_COL
{
    MW_ITEM_COL_NAME = 1,
    MW_ITEM_COL_PING = 2,
    MW_ITEM_COL_IMPORTTIME = 3,
    MW_ITEM_COL_LASTCONNETED = 4,
    MW_ITEM_COL_DATA = 5
};

class MainWindow
    : public QMainWindow
    , Ui::MainWindow
{
    Q_OBJECT
  public:
    explicit MainWindow(QWidget *parent = nullptr);
    ~MainWindow() override;
    static MainWindow *mwInstance;
  signals:
    void StartConnection() const;
    void StopConnection() const;
    void RestartConnection() const;

  private:
    QvMessageBusSlotDecl;
  private slots:
#ifndef DISABLE_AUTO_UPDATE
    void VersionUpdate(QByteArray &data);
#endif
    void on_activatedTray(QSystemTrayIcon::ActivationReason reason);
    void on_preferencesBtn_clicked();
    void on_clearlogButton_clicked();
    void on_connectionListWidget_customContextMenuRequested(const QPoint &pos);
    void on_importConfigButton_clicked();
    void on_subsButton_clicked();
    //
    void on_connectionListWidget_itemDoubleClicked(QTreeWidgetItem *item, int column);
    void on_connectionFilterTxt_textEdited(const QString &arg1);
    void on_connectionListWidget_itemClicked(QTreeWidgetItem *item, int column);
    void on_locateBtn_clicked();
    //
    void on_chartVisibilityBtn_clicked();
    void on_logVisibilityBtn_clicked();
    void on_clearChartBtn_clicked();
    void on_connectionListWidget_currentItemChanged(QTreeWidgetItem *current, QTreeWidgetItem *previous);
    void on_masterLogBrowser_textChanged();

  private:
    void on_actionExit_triggered();
    void on_action_StartThis_triggered();
    void on_action_RCM_SetAutoConnection_triggered();
    void on_action_RCM_EditThis_triggered();
    void on_action_RCM_EditAsJson_triggered();
    void on_action_RCM_EditAsComplex_triggered();
    void on_action_RCM_RenameThis_triggered();
    void on_action_RCM_DeleteThese_triggered();
    void on_action_RCM_DuplicateThese_triggered();
    void on_action_RCM_ClearUsage_triggered();
    //
    void on_action_RCM_tovCoreLog_triggered();
    void on_action_RCM_toQvLog_triggered();
    //
    void OnConnectionWidgetFocusRequested(const ConnectionItemWidget *widget);
    //
    void ToggleVisibility();
    void OnEditRequested(const ConnectionId &id);
    void OnEditJsonRequested(const ConnectionId &id);
    void OnConnected(const ConnectionId &id);
    void OnDisconnected(const ConnectionId &id);
    //
    void OnStatsAvailable(const ConnectionId &id, const quint64 upS, const quint64 downS, const quint64 upD, const quint64 downD);
    void OnVCoreLogAvailable(const ConnectionId &id, const QString &log);
    //
    void OnConnectionCreated(const ConnectionId &id, const QString &displayName);
    void OnConnectionDeleted(const ConnectionId &id, const GroupId &groupId);
    void OnConnectionGroupChanged(const ConnectionId &id, const GroupId &originalGroup, const GroupId &newGroup);
    //
    void OnGroupCreated(const GroupId &id, const QString &displayName);
    void OnGroupDeleted(const GroupId &id, const QList<ConnectionId> &connections);
    //
    void SortConnectionList(MW_ITEM_COL byCol, bool asending);

    void on_pluginsBtn_clicked();

  protected:
    void timerEvent(QTimerEvent *event) override;
    void keyPressEvent(QKeyEvent *e) override;
    void keyReleaseEvent(QKeyEvent *e) override;
    void closeEvent(QCloseEvent *) override;

  private:
    QHash<GroupId, shared_ptr<QTreeWidgetItem>> groupNodes;
    QHash<ConnectionId, shared_ptr<QTreeWidgetItem>> connectionNodes;
    // Charts
    SpeedWidget *speedChartWidget;
#ifndef DISABLE_AUTO_UPDATE
    QvHttpRequestHelper *requestHelper;
#endif
    QSystemTrayIcon hTray;
    PACServer *pacServer;
    SyntaxHighlighter *vCoreLogHighlighter;
    ConnectionInfoWidget *infoWidget;
    //
    // Actions in the system tray menu
    QMenu *tray_RootMenu = new QMenu(this);
    QMenu *tray_SystemProxyMenu = new QMenu(tr("System Proxy"), this);
    //
    QAction *tray_action_ShowHide = new QAction(tr("Hide"), this);
    QAction *tray_action_ShowPreferencesWindow = new QAction(tr("Preferences"), this);
    QAction *tray_action_Quit = new QAction(tr("Quit"), this);
    QAction *tray_action_Start = new QAction(tr("Connect"), this);
    QAction *tray_action_Restart = new QAction(tr("Reconnect"), this);
    QAction *tray_action_Stop = new QAction(tr("Disconnect"), this);
    QAction *tray_action_SetSystemProxy = new QAction(tr("Enable System Proxy"), this);
    QAction *tray_action_ClearSystemProxy = new QAction(tr("Disable System Proxy"), this);
    //
    QMenu *connectionListRCM_Menu = new QMenu(this);
    QAction *action_RCM_Start = new QAction(tr("Connect to this"), this);
    QAction *action_RCM_SetAutoConnection = new QAction(tr("Set as automatically connected"), this);
    QAction *action_RCM_Edit = new QAction(tr("Edit"), this);
    QAction *action_RCM_EditJson = new QAction(tr("Edit as JSON"), this);
    QAction *action_RCM_EditComplex = new QAction(tr("Edit as Complex Config"), this);
    QAction *action_RCM_Rename = new QAction(tr("Rename"), this);
    QAction *action_RCM_Duplicate = new QAction(tr("Duplicate to the Same Group"), this);
    QAction *action_RCM_ClearUsage = new QAction(tr("Clear Usage Data"), this);
    QAction *action_RCM_Delete = new QAction(tr("Delete Connection"), this);
    //
    QMenu *sortMenu = new QMenu(tr("Sort connection list."), this);
    QAction *sortAction_SortByName_Asc = new QAction(tr("By connection name, A-Z"));
    QAction *sortAction_SortByName_Dsc = new QAction(tr("By connection name, Z-A"));
    QAction *sortAction_SortByPing_Asc = new QAction(tr("By latency, Ascending"));
    QAction *sortAction_SortByPing_Dsc = new QAction(tr("By latency, Descending"));
    QAction *sortAction_SortByData_Asc = new QAction(tr("By data, Ascending"));
    QAction *sortAction_SortByData_Dsc = new QAction(tr("By data, Descending"));
    //
    QTextDocument *vCoreLogDocument = new QTextDocument(this);
    QTextDocument *qvLogDocument = new QTextDocument(this);
    //
    int qvLogTimerId = -1;
    QMenu *logRCM_Menu = new QMenu(this);
    QAction *action_RCM_tovCoreLog = new QAction(tr("Switch to vCore log"), this);
    QAction *action_RCM_toQvLog = new QAction(tr("Switch to Qv2ray log"), this);
    //
    ConnectionId lastConnectedId;
    void MWSetSystemProxy();
    void CheckSubscriptionsUpdate();
    //
    void UpdateColorScheme();
    //
    void MWAddConnectionItem_p(const ConnectionId &connection, const GroupId &groupId);
    void MWAddGroupItem_p(const GroupId &groupId);
};