Qv2ray/src/ui/widgets/ConnectionItemWidget.hpp

#pragma once

#include "core/handler/ConnectionHandler.hpp"
#include "ui_ConnectionItemWidget.h"

#include <QWidget>

enum ITEM_TYPE
{
    GROUP_HEADER_ITEM,
    NODE_ITEM
};

class ConnectionItemWidget
    : public QWidget
    , private Ui::ConnectionWidget
{
    Q_OBJECT
  public:
    explicit ConnectionItemWidget(const ConnectionId &connecionId, QWidget *parent = nullptr);
    explicit ConnectionItemWidget(const GroupId &groupId, QWidget *parent = nullptr);
    //
    void BeginConnection();
    ~ConnectionItemWidget();
    //
    inline bool NameMatched(const QString &arg)
    {
        auto searchString = arg.toLower();
        auto headerMatched = ConnectionManager->GetDisplayName(groupId).toLower().contains(arg);

        if (itemType != NODE_ITEM) { return headerMatched; }
        else
        {
            return headerMatched || ConnectionManager->GetDisplayName(connectionId).toLower().contains(searchString);
        }
    }
    inline const tuple<GroupId, ConnectionId> Identifier() const
    {
        return make_tuple(groupId, connectionId);
    }
    inline bool IsConnection() const
    {
        return itemType == NODE_ITEM;
    }
  signals:
    void RequestWidgetFocus(const ConnectionItemWidget *me);
  private slots:
    void OnConnectionStatsArrived(const ConnectionId &id, const quint64 upSpeed, const quint64 downSpeed, const quint64 totalUp,
                                  const quint64 totalDown);
    void OnConnected(const ConnectionId &id);
    void OnDisConnected(const ConnectionId &id);
    void OnLatencyTestStart(const ConnectionId &id);
    void OnLatencyTestFinished(const ConnectionId &id, const uint average);

  private:
    QString originalConnectionName;
    explicit ConnectionItemWidget(QWidget *parent = nullptr);
    ITEM_TYPE itemType;
    ConnectionId connectionId;
    GroupId groupId;

    Q_DISABLE_COPY_MOVE(ConnectionItemWidget)
};