Resolve name and avatar on 1-on-1 rooms

1 files changed, 128 insertions, 8 deletions

diff --git a/src/RoomState.cc b/src/RoomState.cc

index 98f418e3..3eaff452 100644
--- a/src/RoomState.cc
+++ b/src/RoomState.cc
@@ -15,18 +15,138 @@
  * along with this program.  If not, see <http://www.gnu.org/licenses/>.
  */
 
+#include <QDebug>
+#include <QSettings>
+
 #include "RoomState.h"
 
-QString RoomState::resolveName() const
+namespace events = matrix::events;
+
+void RoomState::resolveName()
 {
-	if (!name.content().name().isEmpty())
-		return name.content().name().simplified();
+	name_ = "Empty Room";
+	userAvatar_.clear();
+
+	if (!name.content().name().isEmpty()) {
+		name_ = name.content().name().simplified();
+		return;
+	}
+
+	if (!canonical_alias.content().alias().isEmpty()) {
+		name_ = canonical_alias.content().alias().simplified();
+		return;
+	}
+
+	// FIXME: Doesn't follow the spec guidelines.
+	if (aliases.content().aliases().size() != 0) {
+		name_ = aliases.content().aliases()[0].simplified();
+		return;
+	}
+
+	QSettings settings;
+	auto user_id = settings.value("auth/user_id");
+
+	// TODO: Display names should be sorted alphabetically.
+	for (const auto membership : memberships) {
+		if (membership.stateKey() == user_id)
+			continue;
+
+		if (membership.content().membershipState() == events::Membership::Join) {
+			userAvatar_ = membership.stateKey();
+
+			if (membership.content().displayName().isEmpty())
+				name_ = membership.stateKey();
+			else
+				name_ = membership.content().displayName();
+
+			break;
+		}
+	}
+
+	// TODO: pluralization
+	if (memberships.size() > 2)
+		name_ = QString("%1 and %2 others").arg(name_).arg(memberships.size());
+}
+
+void RoomState::resolveAvatar()
+{
+	if (userAvatar_.isEmpty()) {
+		avatar_ = avatar.content().url();
+		return;
+	}
+
+	if (memberships.contains(userAvatar_)) {
+		avatar_ = memberships[userAvatar_].content().avatarUrl();
+	} else {
+		qWarning() << "Setting room avatar from unknown user id" << userAvatar_;
+	}
+}
+
+// Should be used only after initial sync.
+void RoomState::removeLeaveMemberships()
+{
+	for (auto it = memberships.begin(); it != memberships.end();) {
+		if (it.value().content().membershipState() == events::Membership::Leave) {
+			it = memberships.erase(it);
+		} else {
+			++it;
+		}
+	}
+}
+
+void RoomState::update(const RoomState &state)
+{
+	bool needsNameCalculation = false;
+	bool needsAvatarCalculation = false;
+
+	if (aliases.eventId() != state.aliases.eventId()) {
+		aliases = state.aliases;
+	}
+
+	if (avatar.eventId() != state.avatar.eventId()) {
+		avatar = state.avatar;
+		needsAvatarCalculation = true;
+	}
+
+	if (canonical_alias.eventId() != state.canonical_alias.eventId()) {
+		canonical_alias = state.canonical_alias;
+		needsNameCalculation = true;
+	}
+
+	if (create.eventId() != state.create.eventId())
+		create = state.create;
+	if (history_visibility.eventId() != state.history_visibility.eventId())
+		history_visibility = state.history_visibility;
+	if (join_rules.eventId() != state.join_rules.eventId())
+		join_rules = state.join_rules;
+
+	if (name.eventId() != state.name.eventId()) {
+		name = state.name;
+		needsNameCalculation = true;
+	}
+
+	if (power_levels.eventId() != state.power_levels.eventId())
+		power_levels = state.power_levels;
+	if (topic.eventId() != state.topic.eventId())
+		topic = state.topic;
+
+	for (auto it = state.memberships.constBegin(); it != state.memberships.constEnd(); ++it) {
+		auto membershipState = it.value().content().membershipState();
+
+		if (it.key() == userAvatar_) {
+			needsNameCalculation = true;
+			needsAvatarCalculation = true;
+		}
 
-	if (!canonical_alias.content().alias().isEmpty())
-		return canonical_alias.content().alias().simplified();
+		if (membershipState == events::Membership::Leave)
+			this->memberships.remove(it.key());
+		else
+			this->memberships.insert(it.key(), it.value());
+	}
 
-	if (aliases.content().aliases().size() != 0)
-		return aliases.content().aliases()[0].simplified();
+	if (needsNameCalculation)
+		resolveName();
 
-	return "Unknown Room Name";
+	if (needsAvatarCalculation)
+		resolveAvatar();
 }