Chat_Client/MessageWModel.cpp at sprint_3 · DanSizov/Chat_Client · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
#include "MessageWModel.h"

struct messageIdLessThan
{
    /*bool operator()(const messageItemPtr& left, const messageItemPtr& right) const
    {
        return left->getMsgId() < right->getMsgId();
    }

    bool operator()(const uint64_t left_id, const messageItemPtr& right) const
    {
        return left_id < right->get_msg_id();
    }

    bool operator()(const messageItemPtr& left, const uint64_t right_id) const
    {
        return left->get_msg_id() < right_id;
    }*/
};

MessageWModel::MessageWModel(QObject* parent)
    : QAbstractListModel(parent)
{}

MessageWModel::~MessageWModel()
{
    clear();
}

void MessageWModel::clear()
{
    beginResetModel();
    model_messages.clear();
    //_id_to_msg.clear();
    endResetModel();
}

void MessageWModel::addMessage(const QVariant& new_msg_var)
{
    if (const auto msg = new_msg_var.value<messageItemPtr>(); msg)
    {
        beginInsertRows(QModelIndex(), static_cast<int>(model_messages.size()), static_cast<int>(model_messages.size()));
        model_messages.emplaceBack(msg);
        //_id_to_msg.insert(msg->getMesId(), msg);
        endInsertRows();
    }
}

void MessageWModel::addMessages(const QVariantList& new_msg_list)
{
    if (!new_msg_list.isEmpty())
    {
        beginInsertRows(QModelIndex(), static_cast<int>(model_messages.size()), static_cast<int>(model_messages.size() + new_msg_list.size() - 1));
        for (auto const& msg_var : new_msg_list)
        {
            if (const auto msg = msg_var.value<messageItemPtr>(); msg)
            {
                model_messages.emplaceBack(msg);
                //_id_to_msg.insert(msg->getMesId(), msg);
            }
        }
        endInsertRows();
    }
}

bool MessageWModel::hashMessage(const std::bitset<128>& hash) const
{
    return model_hash_to_msg.contains(hash);
}

messageItemPtr MessageWModel::getMessageFromId(const int id) const
{
    return model_id_to_msg.value(id);
}

void MessageWModel::onAnotheLikeChanged(const QVariant& like_) {

    const auto like = like_.value<likeItemPtr>();
    for (const messageItemPtr& msg_ : model_messages) {
        if (like->getLikeChatId() == msg_->getMesId()) {
            if (like->getLikeReaction() == Like_enum::LIKE) {
                msg_->changeMesLikes(1);
            }
            else {
                msg_->changeMesLikes(-1);
            }
            msg_->addNewLike(like);
        }
    }
}


int MessageWModel::rowCount(const QModelIndex& parent) const
{
    Q_UNUSED(parent)
        return static_cast<int>(model_messages.size());
}

QVariant MessageWModel::data(const QModelIndex& index, int role) const
{
    if (auto const msg = (index.isValid() && index.row() < rowCount()) ? model_messages.at(index.row()) : nullptr)
    {
        switch (role)
        {
        case Qt::DisplayRole:
            return msg->getMesText();

        case Qt::SizeHintRole:
            return QSize{ 200, 200 };

        case Qt::DecorationRole:
            return msg->getMesAvatar();

        case MessageRole:
            return QVariant::fromValue(msg);

        default:
            break;
        }
    }

    return {};
}

Qt::ItemFlags MessageWModel::flags(const QModelIndex& index) const
{
    return QAbstractListModel::flags(index);
}