/*
 * Copyright (C) 2020-2022 Roy Qu (royqh1979@gmail.com)
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see <https://www.gnu.org/licenses/>.
 */
#ifndef BOOKMARKMODEL_H
#define BOOKMARKMODEL_H

#include <QAbstractTableModel>
#include <memory>
#include <QDebug>

struct Bookmark {
    QString filename;
    int line;
    QString description;
    qint64 timestamp;
};

using PBookmark=std::shared_ptr<Bookmark>;

class BookmarkModel : public QAbstractTableModel
{
    Q_OBJECT
public:
    BookmarkModel(QObject* parent=nullptr);
    QSet<int> bookmarksInFile(const QString& filename, bool forProject);
    void addBookmark(const QString&filename, int line, const QString& description, bool forProject);
    PBookmark bookmark(int i, bool forProject);
    PBookmark bookmark(int i);
    PBookmark bookmark(const QString&filename, int line, bool forProject);
    PBookmark bookmark(const QString&filename, int line);
    bool removeBookmark(const QString&filename, int line, bool forProject);
    void removeBookmarks(const QString& filename, bool forProject);
    void clear(bool forProject);
    bool updateDescription(const QString&filename, int line, const QString& description, bool forProject);
    bool updateDescription(const QString&filename, int line, const QString& description);
    void saveBookmarks(const QString& filename);
    void loadBookmarks(const QString& filename);
    void saveProjectBookmarks(const QString& filename, const QString& projectFolder);
    void loadProjectBookmarks(const QString& filename, const QString& projectFolder);
    void removeBookmarkAt(int i, bool forProject);
    void removeBookmarkAt(int i);
public slots:
    void onFileDeleteLines(const QString& filename, int startLine, int count, bool forProject);
    void onFileInsertLines(const QString& filename, int startLine, int count, bool forProject);
private:
    bool isBookmarkExists(const QString&filename, int line, bool forProject);
    void save(const QString& filename, const QString& projectFolder);
    QList<PBookmark> load(const QString& filename, qint64 criteriaTimestamp, qint64* pFileTimestamp);

private:
    QList<PBookmark> mBookmarks;
    QList<PBookmark> mProjectBookmarks;
    qint64 mLastLoadBookmarksTimestamp;
    qint64 mLastLoadProjectBookmarksTimestamp;
    bool mIsForProject;

    // QAbstractItemModel interface
public:
    int rowCount(const QModelIndex &parent) const override;
    QVariant data(const QModelIndex &index, int role) const override;
    int columnCount(const QModelIndex &parent) const override;

    // QAbstractItemModel interface
public:
    QVariant headerData(int section, Qt::Orientation orientation, int role) const override;
    bool isForProject() const;
    void setIsForProject(bool newIsForProject);

    // QAbstractItemModel interface
public:
    void sort(int column, Qt::SortOrder order) override;
};

using PBookmarkModel = std::shared_ptr<BookmarkModel>;

#endif // BOOKMARKMODEL_H