Show rooms you share with someone

2 files changed, 74 insertions, 0 deletions

diff --git a/src/ui/UserProfile.cpp b/src/ui/UserProfile.cpp

index 66a68bb8..80def409 100644
--- a/src/ui/UserProfile.cpp
+++ b/src/ui/UserProfile.cpp
@@ -58,6 +58,12 @@ UserProfile::UserProfile(const QString &roomid,
           emit verificationStatiChanged();
       });
     fetchDeviceList(this->userid_);
+
+    if (userid != utils::localUser())
+        sharedRooms_ =
+          new RoomInfoModel(cache::client()->getCommonRooms(userid.toStdString()), this);
+    else
+        sharedRooms_ = new RoomInfoModel({}, this);
 }
 
 QHash<int, QByteArray>
@@ -102,12 +108,53 @@ DeviceInfoModel::reset(const std::vector<DeviceInfo> &deviceList)
     endResetModel();
 }
 
+RoomInfoModel::RoomInfoModel(const std::map<std::string, RoomInfo> &raw, QObject *parent)
+  : QAbstractListModel(parent)
+{
+    for (const auto &e : raw)
+        roomInfos_.push_back(e);
+}
+
+QHash<int, QByteArray>
+RoomInfoModel::roleNames() const
+{
+    return {
+      {RoomId, "roomId"},
+      {RoomName, "roomName"},
+      {AvatarUrl, "avatarUrl"},
+    };
+}
+
+QVariant
+RoomInfoModel::data(const QModelIndex &index, int role) const
+{
+    if (!index.isValid() || index.row() >= (int)roomInfos_.size() || index.row() < 0)
+        return {};
+
+    switch (role) {
+    case RoomId:
+        return QString::fromStdString(roomInfos_[index.row()].first);
+    case RoomName:
+        return QString::fromStdString(roomInfos_[index.row()].second.name);
+    case AvatarUrl:
+        return QString::fromStdString(roomInfos_[index.row()].second.avatar_url);
+    default:
+        return {};
+    }
+}
+
 DeviceInfoModel *
 UserProfile::deviceList()
 {
     return &this->deviceList_;
 }
 
+RoomInfoModel *
+UserProfile::sharedRooms()
+{
+    return this->sharedRooms_;
+}
+
 QString
 UserProfile::userid()
 {
diff --git a/src/ui/UserProfile.h b/src/ui/UserProfile.h

index d8423ffa..a880f320 100644
--- a/src/ui/UserProfile.h
+++ b/src/ui/UserProfile.h
@@ -119,6 +119,30 @@ private:
     friend class UserProfile;
 };
 
+class RoomInfoModel final : public QAbstractListModel
+{
+    Q_OBJECT
+public:
+    enum Roles
+    {
+        RoomId,
+        RoomName,
+        AvatarUrl,
+    };
+
+    explicit RoomInfoModel(const std::map<std::string, RoomInfo> &, QObject *parent = nullptr);
+    QHash<int, QByteArray> roleNames() const override;
+    int rowCount(const QModelIndex &parent = QModelIndex()) const override
+    {
+        (void)parent;
+        return (int)roomInfos_.size();
+    }
+    QVariant data(const QModelIndex &index, int role = Qt::DisplayRole) const override;
+
+private:
+    std::vector<std::pair<std::string, RoomInfo>> roomInfos_;
+};
+
 class UserProfile final : public QObject
 {
     Q_OBJECT
@@ -126,6 +150,7 @@ class UserProfile final : public QObject
     Q_PROPERTY(QString userid READ userid CONSTANT)
     Q_PROPERTY(QString avatarUrl READ avatarUrl NOTIFY avatarUrlChanged)
     Q_PROPERTY(DeviceInfoModel *deviceList READ deviceList NOTIFY devicesChanged)
+    Q_PROPERTY(RoomInfoModel *sharedRooms READ sharedRooms CONSTANT)
     Q_PROPERTY(bool isGlobalUserProfile READ isGlobalUserProfile CONSTANT)
     Q_PROPERTY(int userVerified READ getUserStatus NOTIFY userStatusChanged)
     Q_PROPERTY(bool isLoading READ isLoading NOTIFY loadingChanged)
@@ -139,6 +164,7 @@ public:
                 TimelineModel *parent = nullptr);
 
     DeviceInfoModel *deviceList();
+    RoomInfoModel *sharedRooms();
 
     QString userid();
     QString displayName();
@@ -198,4 +224,5 @@ private:
     bool isLoading_              = false;
     TimelineViewManager *manager;
     TimelineModel *model;
+    RoomInfoModel *sharedRooms_ = nullptr;
 };