#ifndef FACE_LIST_MODEL_H
#define FACE_LIST_MODEL_H

#include <QAbstractListModel>
#include <QByteArray>
#include <QHash>
#include <QModelIndex>
#include <QString>
#include <QVariant>
#include <QVector>

class FaceListModel : public QAbstractListModel
{
    Q_OBJECT

public:
    enum Roles {
        FaceIdRole = Qt::UserRole + 1,
        StudentIdRole,
        HasPhotoRole,
        PhotoUrlRole
    };

    explicit FaceListModel(QObject *parent = nullptr);

    int rowCount(const QModelIndex &parent = QModelIndex()) const override;
    QVariant data(const QModelIndex &index, int role = Qt::DisplayRole) const override;
    bool setData(const QModelIndex &index, const QVariant &value, int role) override;
    Qt::ItemFlags flags(const QModelIndex &index) const override;
    QHash<int, QByteArray> roleNames() const override;

    struct Item {
        QString faceId;
        QString studentId;
        bool hasPhoto = false;
        QString photoUrl;
    };

    void setItems(const QVector<Item> &items);
    QVector<Item> items() const;
    Q_INVOKABLE void updateStudentId(int index, const QString &studentId);

private:
    QVector<Item> m_items;
};

#endif