BookmarkDatabase.h

// Author: Kang Lin <kl222@126.com>
 
#pragma once
 
#include <QDomElement>
#include <QString>
#include <QByteArray>
#include <QDateTime>
#include <QIcon>
#include <QList>
#include <QDateTime>
#include "DatabaseTree.h"
#include "DatabaseUrl.h"
 
enum BookmarkType {
    BookmarkType_Bookmark,
    BookmarkType_Folder,
    BookmarkType_Separator
};
 
struct BookmarkItem {
    int id;
    QString url;
    QString title;
    QIcon icon;
    QDateTime createdTime;
    QDateTime modifiedTime;
    QDateTime lastVisitTime;
 
    int folderId;
    BookmarkType type;
 
    BookmarkItem(BookmarkType t)
        : id(0)
        , folderId(1) // 指向默认收藏夹
        , type(t)
    {}
 
    [[nodiscard]] bool isFolder() const { return type == BookmarkType_Folder; }
    [[nodiscard]] bool isSeparator() const { return type == BookmarkType_Separator; }
    [[nodiscard]] bool isBookmark() const { return type == BookmarkType_Bookmark; }
 
    [[nodiscard]] QIcon getIcon() const {
        if (!icon.isNull()) {
            return icon;
        }
 
        // 根据URL返回默认图标
        if (url.startsWith("https://")) {
            return QIcon::fromTheme("security-high");
        } else if (url.startsWith("http://")) {
            return QIcon::fromTheme("security-low");
        } else if (isFolder()) {
            return QIcon::fromTheme("folder");
        }
 
        return QIcon::fromTheme("user-bookmarks");
    }
 
};
 
class CBookmarkDatabase : public CDatabase
{
    Q_OBJECT
 
public:
    explicit CBookmarkDatabase(QObject *parent = nullptr);
    ~CBookmarkDatabase();
 
    // 书签操作
    int addBookmark(const BookmarkItem &item);
    bool updateBookmark(const BookmarkItem &item);
    bool deleteBookmark(int id);
    bool deleteBookmark(const QList<BookmarkItem>& items);
    bool moveBookmark(int id, int newFolderId);
 
    // 文件夹操作
    int addFolder(const QString &name, int parentId = 0);
    bool renameFolder(int folderId, const QString &newName);
    bool deleteFolder(int folderId);
    bool moveFolder(int folderId, int newParentId);
 
    // 查询操作
    [[nodiscard]] BookmarkItem getBookmark(int id);
    [[nodiscard]] QList<BookmarkItem> getBookmarkByUrl(const QString &url);
    [[nodiscard]] QList<BookmarkItem> getAllBookmarks(int folderId = 0);
    [[nodiscard]] QList<BookmarkItem> searchBookmarks(const QString &keyword);
 
    // 文件夹查询
    [[nodiscard]] QList<BookmarkItem> getAllFolders();
    [[nodiscard]] QList<BookmarkItem> getSubFolders(int folderId);
 
    [[nodiscard]] TreeItem BookmarkToTree(const BookmarkItem& tree, bool setKey = false);
    [[nodiscard]] BookmarkItem TreeToBookmark(const TreeItem& tree);
    [[nodiscard]] BookmarkItem TreeToBookmark(const TreeItem& item, const CDatabaseUrl::UrlItem& url);
 
signals:
    void bookmarksChanged();
    void bookmarkAdded(const BookmarkItem &item);
    void bookmarkUpdated(const BookmarkItem &item);
    void bookmarkDeleted(int id);
    void folderAdded(const BookmarkItem &folder);
    void folderUpdated(const BookmarkItem &folder);
    void folderDeleted(int folderId);
 
private:
    [[nodiscard]] bool OnInitializeDatabase() override;
 
    CDatabaseTree m_TreeDB;
    CDatabaseUrl m_UrlDB;
 
    virtual bool ExportToJson(QJsonObject &obj) override;
    virtual bool ImportFromJson(const QJsonObject &obj) override;
private:
    bool ExportToJson(int parentId, QJsonArray& obj);
    bool ImportFromJson(int parentId, const QJsonArray& obj);
};