#include <QGuiApplication>
#include <QQmlApplicationEngine>
#include <QQmlContext>
#include <QTimer>
#include <QAbstractListModel>
#include <QColor>

// Message data structure
struct MessageData {
    QString text;
    bool isMine;
    bool upGroup;
    bool downGroup;
};

// Custom List Model
class MessageModel : public QAbstractListModel {
    Q_OBJECT
public:
    enum MessageRoles {
        TextRole = Qt::UserRole + 1,
        IsMineRole,
        UpGroupRole,
        DownGroupRole
    };

    explicit MessageModel(QObject *parent = nullptr) : QAbstractListModel(parent) {}

    int rowCount(const QModelIndex &parent = QModelIndex()) const override {
        Q_UNUSED(parent);
        return m_messages.size();
    }

    QVariant data(const QModelIndex &index, int role = Qt::DisplayRole) const override {
        if (!index.isValid() || index.row() >= m_messages.size())
            return QVariant();

        const MessageData &msg = m_messages.at(index.row());
       
        switch (role) {
        case TextRole:
            return msg.text;
        case IsMineRole:
            return msg.isMine;
        case UpGroupRole:
            return msg.upGroup;
        case DownGroupRole:
            return msg.downGroup;
        default:
            return QVariant();
        }
    }

    QHash<int, QByteArray> roleNames() const override {
        QHash<int, QByteArray> roles;
        roles[TextRole] = "messageText";
        roles[IsMineRole] = "isMine";
        roles[UpGroupRole] = "upGroup";
        roles[DownGroupRole] = "downGroup";
        return roles;
    }

    Q_INVOKABLE void addMessage(const QString &text, bool isMine, bool upGroup, bool downGroup) {
        beginInsertRows(QModelIndex(), rowCount(), rowCount());
        m_messages.append({text, isMine, upGroup, downGroup});
        endInsertRows();
    }

    Q_INVOKABLE void clearMessages() {
        beginResetModel();
        m_messages.clear();
        endResetModel();
    }

private:
    QList<MessageData> m_messages;
};

#include "main.moc"

int main(int argc, char *argv[])
{
    QGuiApplication app(argc, argv);

    // Create and expose the model
    MessageModel messageModel;

    // Add some initial messages
    messageModel.addMessage("sfsfdsdf", false, false, true);
    messageModel.addMessage("sdf", false, true, false);
    messageModel.addMessage("fjklkj", true, false, false);

    QQmlApplicationEngine engine;
   
    // Expose the model to QML
    engine.rootContext()->setContextProperty("messageModel", &messageModel);

    QObject::connect(
        &engine,
        &QQmlApplicationEngine::objectCreationFailed,
        &app,
        []() { QCoreApplication::exit(-1); },
        Qt::QueuedConnection);

    engine.loadFromModule("SuperMatrixClient", "Main");

    // Example: Add messages dynamically from C++
    QTimer::singleShot(1000, [&messageModel]() {
        messageModel.addMessage("Hello from C++ after 1 second!", true, false, true);
    });

    QTimer::singleShot(3000, [&messageModel]() {
        messageModel.addMessage("Another message after 3 seconds", false, true, false);
    });

    // Example: Add messages periodically
    QTimer *periodicTimer = new QTimer(&app);
    QObject::connect(periodicTimer, &QTimer::timeout, [&messageModel]() {
        static int counter = 1;
        messageModel.addMessage(QString("Periodic message #%1").arg(counter++),
                               counter % 2 == 0, false, false);
    });
    periodicTimer->start(5000); // Add a message every 5 seconds

    return app.exec();
}