traffic_tab.h

Go to the documentation of this file.

 
#ifndef TRAFFIC_TAB_H
#define TRAFFIC_TAB_H
 
#include "config.h"
 
#include <ui/qt/models/atap_data_model.h>
#include <ui/qt/filter_action.h>
#include <ui/qt/widgets/traffic_tree.h>
#include <ui/qt/widgets/detachable_tabwidget.h>
#include <ui/qt/widgets/traffic_types_list.h>
 
#include <QTabWidget>
#include <QTreeView>
#include <QFile>
#include <QUrl>
#include <QAbstractItemDelegate>
#include <QSortFilterProxyModel>
 
typedef ATapDataModel * (*ATapModelCallback)(int protoId, QString filter);
 
typedef QAbstractItemDelegate * (*ATapCreateDelegate)(QWidget * parent);
 
class TabData
{
public:
    TabData();
    TabData(const TabData &) = default;
    TabData &operator=(const TabData &) = default;
 
    TabData(QString name, int proto);
 
    QString name() const;
    int protoId() const;
 
private:
    QString _name;
    int _protoId;
};
 
Q_DECLARE_METATYPE(TabData)
 
 
class TrafficTab : public DetachableTabWidget
{
    Q_OBJECT
 
public:
    TrafficTab(QWidget *parent = nullptr);
    virtual ~TrafficTab();
 
    void setProtocolInfo(QString tableName, TrafficTypesList * trafficList, GList ** recentList, GList ** recentColumnList, ATapModelCallback createModel);
 
    void setDelegate(ATapCreateDelegate createDelegate);
 
    void setFilter(QString filter = QString());
 
    void setNameResolution(bool checked);
 
    void disableTap();
 
    QMenu * createCopyMenu(QWidget * parent = nullptr);
 
    bool hasNameResolution(int tabIdx = -1);
 
#ifdef HAVE_MAXMINDDB
    bool hasGeoIPData(int tabIdx = -1);
 
    QUrl createGeoIPMap(bool onlyJSON, int tabIdx = -1);
#endif
 
    QVariant currentItemData(int role = Qt::DisplayRole);
 
    void useNanosecondTimestamps(bool useNSTime);
    ATapDataModel * dataModelForTabIndex(int tabIdx = -1);
 
public slots:
 
    void useAbsoluteTime(bool absolute);
 
    void setOpenTabs(QList<int> protocols);
 
signals:
    void filterAction(QString filter, FilterAction::Action action, FilterAction::ActionType type);
    void tabDataChanged(int idx);
    void retapRequired();
    void disablingTaps();
    void tabsChanged(QList<int> protocols);
    void columnsHaveChanged(QList<int> columns);
 
protected slots:
 
    virtual void detachTab(int idx, QPoint pos) override;
    virtual void attachTab(QWidget * content, QString name) override;
 
private:
    QList<int> _allProtocols;
    QMap<int, int> _tabs;
    ATapModelCallback _createModel;
    ATapCreateDelegate _createDelegate;
    GList ** _recentList;
    GList ** _recentColumnList;
 
    bool _disableTaps;
    bool _nameResolution;
 
    QTreeView * createTree(int protoId);
    TrafficDataFilterProxy * modelForTabIndex(int tabIdx = -1);
    TrafficDataFilterProxy * modelForWidget(QWidget * widget);
    ATapDataModel * dataModelForWidget(QWidget * widget);
 
    void insertProtoTab(int protoId, bool emitSignals = true);
    void removeProtoTab(int protoId, bool emitSignals = true);
 
#ifdef HAVE_MAXMINDDB
    bool writeGeoIPMapFile(QFile * fp, bool json_only, TrafficDataFilterProxy * model);
#endif
 
private slots:
    void modelReset();
 
    void doCurrentIndexChange(const QModelIndex & cur, const QModelIndex & prev);
};
 
#endif // TRAFFIC_TAB_H