context_tree_model.h

Go to the documentation of this file.

// MIT License
// 
// Copyright (c) 2019 Ruhr University Bochum, Chair for Embedded Security. All Rights reserved.
// Copyright (c) 2019 Marc Fyrbiak, Sebastian Wallat, Max Hoffmann ("ORIGINAL AUTHORS"). All rights reserved.
// Copyright (c) 2021 Max Planck Institute for Security and Privacy. All Rights reserved.
// Copyright (c) 2021 Jörn Langheinrich, Julian Speith, Nils Albartus, René Walendy, Simon Klix ("ORIGINAL AUTHORS"). All Rights reserved.
// 
// Permission is hereby granted, free of charge, to any person obtaining a copy
// of this software and associated documentation files (the "Software"), to deal
// in the Software without restriction, including without limitation the rights
// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
// copies of the Software, and to permit persons to whom the Software is
// furnished to do so, subject to the following conditions:
// 
// The above copyright notice and this permission notice shall be included in all
// copies or substantial portions of the Software.
// 
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
// SOFTWARE.
 
#pragma once
 
#include "hal_core/defines.h"
#include "gui/graph_widget/contexts/graph_context.h"
#include "gui/basic_tree_model/base_tree_model.h"
 
#include <QMimeData>
 
namespace hal
{
 
    class ContextDirectory
    {
        private:
            u32 mId;
            QString mName;
            u32 mParentId;
 
        public:
            ContextDirectory(QString name, u32 parentId, u32 id):mName(name){
                mId = id;
                mParentId = parentId;
            }
 
            void writeToFile(QJsonObject& json);
 
            QString name() const { return mName; }
            u32 id() const { return mId; }
            void setId(u32 id_) { mId = id_; }
            void setName(QString name_) { mName = name_; }
    };
 
    class ContextTreeItem : public BaseTreeItem
    {
        private:
            GraphContext* mContext;
            ContextDirectory* mDirectory;
        public:
 
            ContextTreeItem(GraphContext* context);
            ContextTreeItem(ContextDirectory* directory);
            QVariant getData(int column) const override;
            void setData(QList<QVariant> data) override;
            void setDataAtColumn(int column, QVariant& data) override;
            void appendData(QVariant data) override;
            int getColumnCount() const override;
            int row() const;
            bool isDirectory() const;
            bool isContext() const;
            u32 getId() const;
            QString getName() const;
            QDateTime getTimestamp() const;
            GraphContext* context() const;
            ContextDirectory* directory() const;
    };
 
    class ContextTreeModel : public BaseTreeModel
    {
        Q_OBJECT
 
    public:
        ContextTreeModel(QObject* parent = nullptr);
 
        //int rowCount(const QModelIndex& parent = QModelIndex()) const override;
        QVariant data(const QModelIndex& inddex, int role = Qt::DisplayRole) const override;
 
        BaseTreeItem* getDirectory(u32 directoryId) const;
 
        BaseTreeItem* getContext(u32 contextId) const;
 
        ContextDirectory* addDirectory(QString name, BaseTreeItem* parentItem, u32 id);
 
        void addContext(GraphContext* context, BaseTreeItem* parent = nullptr);
 
        void removeContext(GraphContext* context);
 
        void removeDirectory(ContextDirectory* directory);
 
        GraphContext* getContext(const QModelIndex& index) const;
 
        QModelIndex getIndexFromContext(GraphContext* context) const;
 
        void clear() override;
 
        Qt::ItemFlags flags(const QModelIndex& index) const override;
        QStringList mimeTypes() const override;
 
        const QVector<GraphContext*>& list();
 
        const QVector<ContextDirectory*>& directoryList();
 
        void setCurrentDirectory(ContextTreeItem* currentItem);
 
        ContextTreeItem* getCurrentDirectory();
 
        bool moveItem(ContextTreeItem* itemToMove, BaseTreeItem* newParent, int row = -1);
 
        std::vector<u32> getChildDirectoriesOf(u32 directoryId);
 
        std::vector<u32> getChildContextsOf(u32 directoryId);
 
    Q_SIGNALS:
        void directoryCreatedSignal(ContextTreeItem* item);
 
    private:
        BaseTreeItem* getItemInternal(BaseTreeItem* parentItem, u32 id, bool isDirectory) const;
        ContextTreeItem *mCurrentDirectory;
        std::map<GraphContext *, ContextTreeItem *> mContextMap;
        QVector<GraphContext*> mContextList;
        QVector<ContextDirectory*> mDirectoryList;
 
        void dumpRecursion(ContextTreeItem* parent = nullptr, int level = 0) const;
    };
}    // namespace hal