#ifndef ATTACHMENTSMODEL_H
#define ATTACHMENTSMODEL_H

#include <QAbstractTableModel>

namespace TagParser {
class AbstractAttachment;
}

namespace QtGui {

class AttachmentItem {
public:
    AttachmentItem(TagParser::AbstractAttachment *attachment, bool activated = true, const QString &location = QString());

    TagParser::AbstractAttachment *attachment();
    const TagParser::AbstractAttachment *attachment() const;
    const QString &name() const;
    void setName(const QString &name);
    const QString &description() const;
    void setDescription(const QString &description);
    const QString &mimeType() const;
    void setMimeType(const QString &mimeType);
    const QString &size() const;
    const QString &location() const;
    bool isActivated() const;
    void setActivated(bool isActivated);

    void revert();
    void submit();

private:
    TagParser::AbstractAttachment *m_attachment;
    QString m_name;
    QString m_description;
    QString m_mimeType;
    QString m_size;
    QString m_location;
    bool m_activated;
};

class AttachmentsModel : public QAbstractTableModel {
    Q_OBJECT
public:
    AttachmentsModel(QObject *parent = nullptr);
    ~AttachmentsModel() override;

    QVariant data(const QModelIndex &index, int role) const override;
    bool setData(const QModelIndex &index, const QVariant &value, int role) override;
    Qt::ItemFlags flags(const QModelIndex &index) const override;
    QVariant headerData(int section, Qt::Orientation orientation, int role) const override;
    int rowCount(const QModelIndex &parent = QModelIndex()) const override;
    int columnCount(const QModelIndex &parent = QModelIndex()) const override;

    void revert() override;
    bool submit() override;
    void repealSelection();

    TagParser::AbstractAttachment *attachment(const QModelIndex &index);
    void addAttachment(int row, TagParser::AbstractAttachment *attachment, bool activated = true, const QString &location = QString());
    void setAttachments(const QList<TagParser::AbstractAttachment *> &attachments, bool activated = true, const QString &location = QString());
    void removeAttachments(int firstRow, int lastRow);

private:
    QList<AttachmentItem> m_attachments;
};

} // namespace QtGui

#endif // ATTACHMENTSMODEL_H