DatabaseTree.h

// Author: Kang Lin <kl222@126.com>
 
#pragma once
#include <QIcon>
#include "Database.h"
 
class PLUGIN_EXPORT TreeItem {
public:
    enum TYPE {
        Node,
        Leaf
    };
 
    TreeItem(TYPE type);
    TreeItem(const TreeItem& item);
    TreeItem& operator = (const TreeItem& item);
 
    [[nodiscard]] bool IsNode() const;
    [[nodiscard]] bool IsLeaf() const;
 
    [[nodiscard]] TYPE GetType() const;
    void SetType(TYPE type);
    [[nodiscard]] int GetId() const;
    void SetId(int newId);
    [[nodiscard]] QString GetName() const;
    void SetName(const QString &newName);
    [[nodiscard]] QDateTime GetCreateTime() const;
    void SetCreateTime(const QDateTime &newCreateTime);
    [[nodiscard]] QDateTime GetModifyTime() const;
    void SetModifyTime(const QDateTime &newModifyTime);
    [[nodiscard]] QDateTime GetLastVisitTime() const;
    void SetLastVisitTime(const QDateTime &newLastVisitTime);
    [[nodiscard]] int GetParentId() const;
    void SetParentId(int newParentId);
    [[nodiscard]] int GetSortOrder() const;
    void SetSortOrder(int newSortOrder);
 
    [[nodiscard]] int GetKey() const;
    void SetKey(int newNKey);
 
private:
    struct _DATA{
        TYPE m_Type;
        int m_id;
        QString m_szName;
        int m_nKey;
        QDateTime m_CreateTime;
        QDateTime m_ModifyTime;
        QDateTime m_LastVisitTime;
        int m_ParentId;
        int m_SortOrder;
    };
    _DATA m_Data;
};

class PLUGIN_EXPORT CDatabaseNode : public CDatabase
{
    Q_OBJECT
 
public:
    explicit CDatabaseNode(const QString& szPrefix = QString(),
                             QObject *parent = nullptr);
 
    // 文件夹操作
    int AddNode(const QString &name, int parentId = 0);
    bool RenameNode(int id, const QString &newName);
    bool DeleteNode(
        int id,
        std::function<bool(int parentId)> cbDeleteLeaf = nullptr,
        bool checkReturn = true);
    bool MoveNode(int id, int newParentId);
    // 文件夹查询
    [[nodiscard]] TreeItem GetNode(int id);
    [[nodiscard]] QList<TreeItem> GetAllNodes();
    [[nodiscard]] QList<TreeItem> GetSubNodes(int parentId);
    [[nodiscard]] bool InSubNode(int parentId, int id);

    [[nodiscard]] int GetCount(int parentId = 0);
 
    [[nodiscard]] virtual bool ExportToJson(QJsonObject& obj) override;
    [[nodiscard]] virtual bool ImportFromJson(const QJsonObject& obj) override;
 
Q_SIGNALS:
    void sigAddNode(int id, int parentId);
 
protected:
    [[nodiscard]] virtual bool OnDeleteLeafs(int id);
    [[nodiscard]] virtual bool OnInitializeSqliteDatabase() override;
    [[nodiscard]] virtual bool OnInitializeMySqlDatabase() override;
    
private:
    QString m_szTableName;
};

class PLUGIN_EXPORT CDatabaseTree : public CDatabase
{
    Q_OBJECT
 
public:
    explicit CDatabaseTree(const QString& szPrefix = QString(),
                           QObject* parent = nullptr);
 
    // ====== Leaf operate ======
    virtual int Add(const TreeItem& item);
    virtual bool Update(const TreeItem& item);
    virtual bool Delete(int id, bool delKey = false);
    virtual bool Delete(QList<int> items, bool delKey = false);
    virtual bool DeleteChild(int parentId, bool delKey = false);
    virtual bool Move(int id, int newParent);
 
    [[nodiscard]] TreeItem GetLeaf(int id);
    [[nodiscard]] QList<TreeItem> GetLeaves(int nodeId);
    [[nodiscard]] QList<TreeItem> GetLeavesByKey(int key);
    [[nodiscard]] QList<TreeItem> GetLeavesByKey(QList<int> key);
    [[nodiscard]] int GetLeafCount(int parentId = 0);
 
    // ====== Node operate ======
    virtual int AddNode(const QString &name, int parentId = 0);
    virtual bool RenameNode(int id, const QString &newName);
    virtual bool DeleteNode(int id, bool delKey = false);
    virtual bool MoveNode(int id, int newParentId);
    [[nodiscard]] TreeItem GetNode(int id);
    [[nodiscard]] QList<TreeItem> GetAllNodes();
    [[nodiscard]] QList<TreeItem> GetSubNodes(int parentId);
    [[nodiscard]] int GetNodeCount(int nParentId = 0);

    [[nodiscard]] int GetCount(int parentId = 0);
 
    [[nodiscard]] virtual bool ExportToJson(QJsonObject& obj) override;
    [[nodiscard]] virtual bool ImportFromJson(const QJsonObject& obj) override;
 
Q_SIGNALS:
    void sigAddNode(int id, int parentId);
    void sigAdd(int id, int parentId);
 
protected:
    [[nodiscard]] virtual bool OnDeleteKey(int key);
    [[nodiscard]] virtual bool OnInitializeDatabase() override;
    [[nodiscard]] virtual bool OnInitializeSqliteDatabase() override;
    [[nodiscard]] virtual bool OnInitializeMySqlDatabase() override;
 
private:
    QString m_szTableName;
    CDatabaseNode m_FolderDB;
};