No OneTemporary
Actions

Size

75 KB

Referenced Files

None

Subscribers

None

View Options

	diff --git a/src/client/room/room-model.cpp b/src/client/room/room-model.cpp
	index 2097d99..f112c41 100644
	--- a/src/client/room/room-model.cpp
	+++ b/src/client/room/room-model.cpp
	@@ -1,462 +1,479 @@
	/*
	* This file is part of libkazv.
	* SPDX-FileCopyrightText: 2020-2023 tusooa
	* SPDX-License-Identifier: AGPL-3.0-or-later
	*/

	#include <libkazv-config.hpp>


	#include <lager/util.hpp>
	#include <zug/sequence.hpp>
	#include <zug/transducer/map.hpp>
	#include <zug/transducer/filter.hpp>

	#include "debug.hpp"

	#include "room-model.hpp"
	#include "cursorutil.hpp"
	#include "immer-utils.hpp"

	namespace Kazv
	{
	RoomModel RoomModel::update(RoomModel r, Action a)
	{
	return lager::match(std::move(a))(
	[&](AddStateEventsAction a) {
	r.stateEvents = merge(std::move(r.stateEvents), a.stateEvents, keyOfState);

	// If m.room.encryption state event appears,
	// configure the room to use encryption.
	if (r.stateEvents.find(KeyOfState{"m.room.encryption", ""})) {
	auto newRoom = update(std::move(r), SetRoomEncryptionAction{});
	r = std::move(newRoom);
	}
	return r;
	},
	[&](AppendTimelineAction a) {
	auto eventIds = intoImmer(immer::flex_vector<std::string>(),
	zug::map(keyOfTimeline), a.events);
	r.timeline = r.timeline + eventIds;
	r.messages = merge(std::move(r.messages), a.events, keyOfTimeline);
	return r;
	},
	[&](PrependTimelineAction a) {
	auto eventIds = intoImmer(immer::flex_vector<std::string>(),
	zug::map(keyOfTimeline), a.events);
	r.timeline = eventIds + r.timeline;
	r.messages = merge(std::move(r.messages), a.events, keyOfTimeline);
	r.paginateBackToken = a.paginateBackToken;
	// if there are no more events we should not allow further paginating
	r.canPaginateBack = a.events.size() != 0;
	return r;
	},
	[&](AddToTimelineAction a) {
	auto eventIds = intoImmer(immer::flex_vector<std::string>(),
	zug::map(keyOfTimeline), a.events);

	auto oldMessages = r.messages;
	r.messages = merge(std::move(r.messages), a.events, keyOfTimeline);
	auto exists =
	[=](auto eventId) -> bool {
	return !! oldMessages.find(eventId);
	};
	auto key =
	[=](auto eventId) {
	// sort first by timestamp, then by id
	return std::make_tuple(r.messages[eventId].originServerTs(), eventId);
	};

	auto handleRedaction =
	[&r](const auto &event) {
	if (event.type() == "m.room.redaction") {
	auto origJson = event.originalJson().get();
	if (origJson.contains("redacts") && origJson.at("redacts").is_string()) {
	auto redactedEventId = origJson.at("redacts").template get<std::string>();
	if (r.messages.find(redactedEventId)) {
	r.messages = std::move(r.messages).update(redactedEventId, [&origJson](const auto &eventToBeRedacted) {
	auto newJson = eventToBeRedacted.originalJson().get();
	newJson.merge_patch(json{
	{"unsigned", {{"redacted_because", std::move(origJson)}}},
	});
	newJson["content"] = json::object();
	return Event(newJson);
	});
	}
	}
	}
	return event;
	};

	immer::for_each(a.events, handleRedaction);

	r.timeline = sortedUniqueMerge(r.timeline, eventIds, exists, key);

	// TODO need other way to determine whether it is limited
	// in a pagination request (/messages does not have that field)
	if ((! a.limited.has_value() \|\| a.limited.value())
	&& a.prevBatch.has_value()) {
	// this sync is limited, add a Gap here
	if (!eventIds.empty()) {
	r.timelineGaps = std::move(r.timelineGaps).set(eventIds[0], a.prevBatch.value());
	}
	}

	// remove the original Gap, as it is resolved
	if (a.gapEventId.has_value()) {
	r.timelineGaps = std::move(r.timelineGaps).erase(a.gapEventId.value());
	}

	// remove all Gaps between the gapped event and the first event in this batch
	if (!eventIds.empty() && a.gapEventId.has_value()) {
	auto cmp = [=](auto a, auto b) {
	return key(a) < key(b);
	};
	auto thisBatchStart = std::equal_range(r.timeline.begin(), r.timeline.end(), eventIds[0], cmp).first;
	auto origBatchStart = std::equal_range(thisBatchStart, r.timeline.end(), a.gapEventId.value(), cmp).first;

	// Safety assert: we do not want to execute the for_each if the range is empty,
	// or it will go out of bounds.
	if (thisBatchStart.index() < origBatchStart.index()) {
	std::for_each(thisBatchStart + 1, origBatchStart,
	[&](auto eventId) {
	r.timelineGaps = std::move(r.timelineGaps).erase(eventId);
	});
	}
	}

	// remove all local echoes that are received
	for (const auto &e : a.events) {
	const auto &json = e.originalJson().get();
	if (json.contains("unsigned")
	&& json["unsigned"].contains("transaction_id")
	&& json["unsigned"]["transaction_id"].is_string()) {
	r = update(std::move(r), RemoveLocalEchoAction{json["unsigned"]["transaction_id"].template get<std::string>()});
	}
	}

	// calculate event relationships
	r.generateRelationships(a.events);

	return r;
	},
	[&](AddAccountDataAction a) {
	r.accountData = merge(std::move(r.accountData), a.events, keyOfAccountData);
	return r;
	},
	[&](ChangeMembershipAction a) {
	r.membership = a.membership;
	return r;
	},
	[&](ChangeInviteStateAction a) {
	r.inviteState = merge(immer::map<KeyOfState, Event>{}, a.events, keyOfState);
	return r;
	},
	[&](AddEphemeralAction a) {
	r.ephemeral = merge(std::move(r.ephemeral), a.events, keyOfEphemeral);
	return r;
	},
	[&](SetLocalDraftAction a) {
	r.localDraft = a.localDraft;
	return r;
	},
	[&](SetRoomEncryptionAction) {
	r.encrypted = true;
	return r;
	},
	[&](MarkMembersFullyLoadedAction) {
	r.membersFullyLoaded = true;
	return r;
	},
	[&](SetHeroIdsAction a) {
	r.heroIds = a.heroIds;
	return r;
	},
	[&](AddLocalEchoAction a) {
	auto it = std::find_if(r.localEchoes.begin(), r.localEchoes.end(), [a](const auto &desc) {
	return desc.txnId == a.localEcho.txnId;
	});
	if (it == r.localEchoes.end()) {
	r.localEchoes = std::move(r.localEchoes).push_back(a.localEcho);
	} else {
	r.localEchoes = std::move(r.localEchoes).set(it.index(), a.localEcho);
	}
	return r;
	},
	[&](RemoveLocalEchoAction a) {
	auto it = std::find_if(r.localEchoes.begin(), r.localEchoes.end(), [a](const auto &desc) {
	return desc.txnId == a.txnId;
	});
	if (it != r.localEchoes.end()) {
	r.localEchoes = std::move(r.localEchoes).erase(it.index());
	}
	return r;
	},
	[&](AddPendingRoomKeyAction a) {
	auto it = std::find_if(r.pendingRoomKeyEvents.begin(), r.pendingRoomKeyEvents.end(), [a](const auto &p) {
	return p.txnId == a.pendingRoomKeyEvent.txnId;
	});
	if (it == r.pendingRoomKeyEvents.end()) {
	r.pendingRoomKeyEvents = std::move(r.pendingRoomKeyEvents).push_back(a.pendingRoomKeyEvent);
	} else {
	r.pendingRoomKeyEvents = std::move(r.pendingRoomKeyEvents).set(it.index(), a.pendingRoomKeyEvent);
	}
	return r;
	},
	[&](RemovePendingRoomKeyAction a) {
	auto it = std::find_if(r.pendingRoomKeyEvents.begin(), r.pendingRoomKeyEvents.end(), [a](const auto &desc) {
	return desc.txnId == a.txnId;
	});
	if (it != r.pendingRoomKeyEvents.end()) {
	r.pendingRoomKeyEvents = std::move(r.pendingRoomKeyEvents).erase(it.index());
	}
	return r;
	},
	[&](UpdateJoinedMemberCountAction a) {
	r.joinedMemberCount = a.joinedMemberCount;
	return r;
	},
	[&](UpdateInvitedMemberCountAction a) {
	r.invitedMemberCount = a.invitedMemberCount;
	return r;
	}
	);
	}

	RoomListModel RoomListModel::update(RoomListModel l, Action a)
	{
	return lager::match(std::move(a))(
	[&](UpdateRoomAction a) {
	l.rooms = std::move(l.rooms)
	.update(a.roomId,
	[=](RoomModel oldRoom) {
	oldRoom.roomId = a.roomId; // in case it is a new room
	return RoomModel::update(std::move(oldRoom), a.roomAction);
	});
	return l;
	}
	);
	}

	static auto membershipTransducer(const std::string &membership)
	{
	return zug::filter([](auto val) {
	auto [k, v] = val;
	auto [type, stateKey] = k;
	return type == "m.room.member"s;
	})
	\| zug::map([](auto val) {
	auto [k, v] = val;
	auto [type, stateKey] = k;
	return std::pair<std::string, Kazv::Event>{stateKey, v};
	})
	\| zug::filter([&membership](auto val) {
	auto [stateKey, ev] = val;
	return ev.content().get()
	.at("membership"s) == membership;
	});
	}

	static auto memberIdsByMembership(immer::map<KeyOfState, Event> stateEvents, const std::string &membership)
	{
	return intoImmer(
	immer::flex_vector<std::string>{},
	membershipTransducer(membership)
	\| zug::map([](auto val) {
	auto [stateKey, ev] = val;
	return stateKey;
	}),
	stateEvents);
	}

	auto memberEventsByMembership(immer::map<KeyOfState, Event> stateEvents, const std::string &membership)
	{
	return intoImmer(
	EventList{},
	membershipTransducer(membership)
	\| zug::map([](auto val) {
	auto [stateKey, ev] = val;
	return ev;
	}),
	stateEvents);
	}

	immer::flex_vector<std::string> RoomModel::joinedMemberIds() const
	{
	return memberIdsByMembership(stateEvents, "join"s);
	}

	immer::flex_vector<std::string> RoomModel::invitedMemberIds() const
	{
	return memberIdsByMembership(stateEvents, "invite"s);
	}

	immer::flex_vector<std::string> RoomModel::knockedMemberIds() const
	{
	return memberIdsByMembership(stateEvents, "knock"s);
	}

	immer::flex_vector<std::string> RoomModel::leftMemberIds() const
	{
	return memberIdsByMembership(stateEvents, "leave"s);
	}

	immer::flex_vector<std::string> RoomModel::bannedMemberIds() const
	{
	return memberIdsByMembership(stateEvents, "ban"s);
	}

	EventList RoomModel::joinedMemberEvents() const
	{
	return memberEventsByMembership(stateEvents, "join"s);
	}

	EventList RoomModel::invitedMemberEvents() const
	{
	return memberEventsByMembership(stateEvents, "invite"s);
	}

	EventList RoomModel::knockedMemberEvents() const
	{
	return memberEventsByMembership(stateEvents, "knock"s);
	}

	EventList RoomModel::leftMemberEvents() const
	{
	return memberEventsByMembership(stateEvents, "leave"s);
	}

	EventList RoomModel::bannedMemberEvents() const
	{
	return memberEventsByMembership(stateEvents, "ban"s);
	}

	+ EventList RoomModel::heroMemberEvents() const
	+ {
	+ return intoImmer(
	+ EventList{},
	+ zug::filter([heroIds=heroIds](auto val) {
	+ auto [k, ev] = val;
	+ auto [type, stateKey] = k;
	+ return type == "m.room.member"s &&
	+ std::find(heroIds.begin(), heroIds.end(), stateKey) != heroIds.end();
	+ })
	+ \| zug::map([](auto val) {
	+ auto [_, ev] = val;
	+ return ev;
	+ }),
	+ stateEvents);
	+ }
	+
	static Timestamp defaultRotateMs = 604800000;
	static int defaultRotateMsgs = 100;

	MegOlmSessionRotateDesc RoomModel::sessionRotateDesc() const
	{
	auto k = KeyOfState{"m.room.encryption", ""};
	auto content = stateEvents[k].content().get();
	auto ms = content.contains("rotation_period_ms")
	? content["rotation_period_ms"].get<Timestamp>()
	: defaultRotateMs;
	auto msgs = content.contains("rotation_period_msgs")
	? content["rotation_period_msgs"].get<int>()
	: defaultRotateMsgs;
	return MegOlmSessionRotateDesc{ ms, msgs };
	}

	bool RoomModel::hasUser(std::string userId) const
	{
	try {
	auto ev = stateEvents.at(KeyOfState{"m.room.member", userId});
	if (ev.content().get().at("membership") == "join") {
	return true;
	}
	} catch (const std::exception &) {
	return false;
	}
	return false;
	}

	std::optional<LocalEchoDesc> RoomModel::getLocalEchoByTxnId(std::string txnId) const
	{
	auto it = std::find_if(localEchoes.begin(), localEchoes.end(), [txnId](const auto &desc) {
	return txnId == desc.txnId;
	});
	if (it != localEchoes.end()) {
	return *it;
	} else {
	return std::nullopt;
	}
	}

	std::optional<PendingRoomKeyEvent> RoomModel::getPendingRoomKeyEventByTxnId(std::string txnId) const
	{
	auto it = std::find_if(pendingRoomKeyEvents.begin(), pendingRoomKeyEvents.end(), [txnId](const auto &desc) {
	return txnId == desc.txnId;
	});
	if (it != pendingRoomKeyEvents.end()) {
	return *it;
	} else {
	return std::nullopt;
	}
	}

	static double getTagOrder(const json &tag)
	{
	// https://spec.matrix.org/v1.7/client-server-api/#events-12
	// If a room has a tag without an order key then it should appear after the rooms with that tag that have an order key.
	return tag.contains("order") && tag["order"].is_number()
	? tag["order"].template get<double>()
	: ROOM_TAG_DEFAULT_ORDER;
	}

	immer::map<std::string, double> RoomModel::tags() const
	{
	auto content = accountData["m.tag"].content().get();
	if (!content.contains("tags") \|\| !content["tags"].is_object()) {
	return {};
	}

	auto tagsObject = content["tags"];
	auto tagsItems = tagsObject.items();
	return std::accumulate(tagsItems.begin(), tagsItems.end(), immer::map<std::string, double>(),
	[=](auto acc, const auto &cur) {
	auto [id, tag] = cur;
	return std::move(acc).set(id, getTagOrder(tag));
	}
	);
	}

	static auto normalizeTagEventJson(Event e)
	{
	auto content = e.content().get();
	if (!content.contains("tags") \|\| !content["tags"].is_object()) {
	content["tags"] = json::object();
	}
	return json{
	{"content", content},
	{"type", "m.tag"},
	};
	}

	Event RoomModel::makeAddTagEvent(std::string tagId, std::optional<double> order) const
	{
	auto eventJson = normalizeTagEventJson(accountData["m.tag"]);
	auto tag = json::object();
	if (order.has_value()) {
	tag["order"] = order.value();
	}

	eventJson["content"]["tags"][tagId] = tag;
	return Event(eventJson);
	}

	Event RoomModel::makeRemoveTagEvent(std::string tagId) const
	{
	auto eventJson = normalizeTagEventJson(accountData["m.tag"]);
	eventJson["content"]["tags"].erase(tagId);
	return Event(eventJson);
	}

	void RoomModel::generateRelationships(EventList newEvents)
	{
	for (const auto &event: newEvents) {
	auto [relType, eventId] = event.relationship();
	if (!relType.empty()) {
	reverseEventRelationships = updateIn(std::move(reverseEventRelationships), [event](auto &&evs) {
	return evs.push_back(event.id());
	}, eventId, relType);
	}
	}
	}

	void RoomModel::regenerateRelationships()
	{
	generateRelationships(intoImmer(EventList{}, zug::map([](const auto &kv) {
	return kv.second;
	}), messages));
	}
	}
	diff --git a/src/client/room/room-model.hpp b/src/client/room/room-model.hpp
	index e270c7f..9497e33 100644
	--- a/src/client/room/room-model.hpp
	+++ b/src/client/room/room-model.hpp
	@@ -1,363 +1,364 @@
	/*
	* This file is part of libkazv.
	* SPDX-FileCopyrightText: 2021-2023 tusooa <tusooa@kazv.moe>
	* SPDX-License-Identifier: AGPL-3.0-or-later
	*/


	#pragma once
	#include <libkazv-config.hpp>

	#include <string>
	#include <variant>
	#include <immer/flex_vector.hpp>
	#include <immer/map.hpp>

	#include <serialization/immer-flex-vector.hpp>
	#include <serialization/immer-box.hpp>
	#include <serialization/immer-map.hpp>
	#include <serialization/immer-array.hpp>

	#include <csapi/sync.hpp>
	#include <event.hpp>

	#include <crypto.hpp>

	#include "local-echo.hpp"
	#include "clientutil.hpp"

	namespace Kazv
	{
	struct PendingRoomKeyEvent
	{
	std::string txnId;
	Event event;
	immer::map<std::string, immer::flex_vector<std::string>> devices;
	};

	struct AddStateEventsAction
	{
	immer::flex_vector<Event> stateEvents;
	};

	struct AppendTimelineAction
	{
	immer::flex_vector<Event> events;
	};

	struct PrependTimelineAction
	{
	immer::flex_vector<Event> events;
	std::string paginateBackToken;
	};

	struct AddToTimelineAction
	{
	/// Events from oldest to latest
	immer::flex_vector<Event> events;
	std::optional<std::string> prevBatch;
	std::optional<bool> limited;
	std::optional<std::string> gapEventId;
	};

	struct AddAccountDataAction
	{
	immer::flex_vector<Event> events;
	};

	struct ChangeMembershipAction
	{
	RoomMembership membership;
	};

	struct ChangeInviteStateAction
	{
	immer::flex_vector<Event> events;
	};

	struct AddEphemeralAction
	{
	EventList events;
	};

	struct SetLocalDraftAction
	{
	std::string localDraft;
	};

	struct SetRoomEncryptionAction
	{
	};

	struct MarkMembersFullyLoadedAction
	{
	};

	struct SetHeroIdsAction
	{
	immer::flex_vector<std::string> heroIds;
	};

	struct AddLocalEchoAction
	{
	LocalEchoDesc localEcho;
	};

	struct RemoveLocalEchoAction
	{
	std::string txnId;
	};

	struct AddPendingRoomKeyAction
	{
	PendingRoomKeyEvent pendingRoomKeyEvent;
	};

	struct RemovePendingRoomKeyAction
	{
	std::string txnId;
	};

	struct UpdateJoinedMemberCountAction
	{
	std::size_t joinedMemberCount;
	};

	struct UpdateInvitedMemberCountAction
	{
	std::size_t invitedMemberCount;
	};

	inline bool operator==(const PendingRoomKeyEvent &a, const PendingRoomKeyEvent &b)
	{
	return a.txnId == b.txnId && a.event == b.event && a.devices == b.devices;
	}

	inline bool operator!=(const PendingRoomKeyEvent &a, const PendingRoomKeyEvent &b)
	{
	return !(a == b);
	}

	inline const double ROOM_TAG_DEFAULT_ORDER = 2;

	template<class Archive>
	void serialize(Archive &ar, PendingRoomKeyEvent &e, std::uint32_t const version)
	{
	ar & e.txnId & e.event & e.devices;
	}

	struct RoomModel
	{
	using Membership = RoomMembership;
	using ReverseEventRelationshipMap = immer::map<
	std::string /* related event id */,
	immer::map<std::string /* relation type /, immer::flex_vector<std::string / relater event id */>>>;

	std::string roomId;
	immer::map<KeyOfState, Event> stateEvents;
	immer::map<KeyOfState, Event> inviteState;
	// Smaller indices mean earlier events
	// (oldest) 0 --------> n (latest)
	immer::flex_vector<std::string> timeline;
	immer::map<std::string, Event> messages;
	immer::map<std::string, Event> accountData;
	Membership membership{};
	std::string paginateBackToken;
	/// whether this room has earlier events to be fetched
	bool canPaginateBack{true};

	immer::map<std::string /* eventId /, std::string / prevBatch */> timelineGaps;

	immer::map<std::string, Event> ephemeral;

	std::string localDraft;

	bool encrypted{false};
	/// a marker to indicate whether we need to rotate
	/// the session key earlier than it expires
	/// (e.g. when a user in the room's device list changed
	/// or when someone joins or leaves)
	bool shouldRotateSessionKey{true};

	bool membersFullyLoaded{false};
	immer::flex_vector<std::string> heroIds;

	immer::flex_vector<LocalEchoDesc> localEchoes;

	immer::flex_vector<PendingRoomKeyEvent> pendingRoomKeyEvents;

	ReverseEventRelationshipMap reverseEventRelationships;

	std::size_t joinedMemberCount{0};
	std::size_t invitedMemberCount{0};

	immer::flex_vector<std::string> joinedMemberIds() const;
	immer::flex_vector<std::string> invitedMemberIds() const;
	immer::flex_vector<std::string> knockedMemberIds() const;
	immer::flex_vector<std::string> leftMemberIds() const;
	immer::flex_vector<std::string> bannedMemberIds() const;

	EventList joinedMemberEvents() const;
	EventList invitedMemberEvents() const;
	EventList knockedMemberEvents() const;
	EventList leftMemberEvents() const;
	EventList bannedMemberEvents() const;
	+ EventList heroMemberEvents() const;

	MegOlmSessionRotateDesc sessionRotateDesc() const;

	bool hasUser(std::string userId) const;

	std::optional<LocalEchoDesc> getLocalEchoByTxnId(std::string txnId) const;
	std::optional<PendingRoomKeyEvent> getPendingRoomKeyEventByTxnId(std::string txnId) const;

	immer::map<std::string, double> tags() const;

	Event makeAddTagEvent(std::string tagId, std::optional<double> order) const;
	Event makeRemoveTagEvent(std::string tagId) const;

	/**
	* Fill in reverseEventRelationships by gathering
	* the relationships specified in `newEvents`
	*
	* @param newEvents The events that just came in after last time event relationships
	* are gathered.
	*/
	void generateRelationships(EventList newEvents);

	void regenerateRelationships();

	using Action = std::variant<
	AddStateEventsAction,
	AppendTimelineAction,
	PrependTimelineAction,
	AddToTimelineAction,
	AddAccountDataAction,
	ChangeMembershipAction,
	ChangeInviteStateAction,
	AddEphemeralAction,
	SetLocalDraftAction,
	SetRoomEncryptionAction,
	MarkMembersFullyLoadedAction,
	SetHeroIdsAction,
	AddLocalEchoAction,
	RemoveLocalEchoAction,
	AddPendingRoomKeyAction,
	RemovePendingRoomKeyAction,
	UpdateJoinedMemberCountAction,
	UpdateInvitedMemberCountAction
	>;

	static RoomModel update(RoomModel r, Action a);
	};

	using RoomAction = RoomModel::Action;

	inline bool operator==(const RoomModel &a, const RoomModel &b)
	{
	return a.roomId == b.roomId
	&& a.stateEvents == b.stateEvents
	&& a.inviteState == b.inviteState
	&& a.timeline == b.timeline
	&& a.messages == b.messages
	&& a.accountData == b.accountData
	&& a.membership == b.membership
	&& a.paginateBackToken == b.paginateBackToken
	&& a.canPaginateBack == b.canPaginateBack
	&& a.timelineGaps == b.timelineGaps
	&& a.ephemeral == b.ephemeral
	&& a.localDraft == b.localDraft
	&& a.encrypted == b.encrypted
	&& a.shouldRotateSessionKey == b.shouldRotateSessionKey
	&& a.membersFullyLoaded == b.membersFullyLoaded
	&& a.heroIds == b.heroIds
	&& a.localEchoes == b.localEchoes
	&& a.pendingRoomKeyEvents == b.pendingRoomKeyEvents
	&& a.reverseEventRelationships == b.reverseEventRelationships;
	}

	struct UpdateRoomAction
	{
	std::string roomId;
	RoomAction roomAction;
	};

	struct RoomListModel
	{
	immer::map<std::string, RoomModel> rooms;

	inline auto at(std::string id) const { return rooms.at(id); }
	inline auto operator[](std::string id) const { return rooms[id]; }
	inline bool has(std::string id) const { return rooms.find(id); }

	using Action = std::variant<
	UpdateRoomAction
	>;
	static RoomListModel update(RoomListModel l, Action a);
	};

	using RoomListAction = RoomListModel::Action;

	inline bool operator==(const RoomListModel &a, const RoomListModel &b)
	{
	return a.rooms == b.rooms;
	}

	template<class Archive>
	void serialize(Archive &ar, RoomModel &r, std::uint32_t const version)
	{
	ar
	& r.roomId
	& r.stateEvents
	& r.inviteState

	& r.timeline
	& r.messages
	& r.accountData
	& r.membership
	& r.paginateBackToken
	& r.canPaginateBack

	& r.timelineGaps
	& r.ephemeral

	& r.localDraft

	& r.encrypted
	& r.shouldRotateSessionKey

	& r.membersFullyLoaded
	;

	if (version >= 1) {
	ar
	& r.heroIds
	;
	}
	if (version >= 2) {
	ar & r.localEchoes;
	}
	if (version >= 3) {
	ar & r.pendingRoomKeyEvents;
	}
	if (version >= 4) {
	ar & r.reverseEventRelationships;
	} else { // must be reading from an older version
	if constexpr (typename Archive::is_loading()) {
	r.regenerateRelationships();
	}
	}
	if (version >= 5) {
	ar & r.joinedMemberCount & r.invitedMemberCount;
	}
	}

	template<class Archive>
	void serialize(Archive &ar, RoomListModel &l, std::uint32_t const /version/)
	{
	ar & l.rooms;
	}
	}

	BOOST_CLASS_VERSION(Kazv::PendingRoomKeyEvent, 0)
	BOOST_CLASS_VERSION(Kazv::RoomModel, 5)
	BOOST_CLASS_VERSION(Kazv::RoomListModel, 0)
	diff --git a/src/client/room/room.cpp b/src/client/room/room.cpp
	index 57d4105..101f5f2 100644
	--- a/src/client/room/room.cpp
	+++ b/src/client/room/room.cpp
	@@ -1,871 +1,860 @@
	/*
	* This file is part of libkazv.
	* SPDX-FileCopyrightText: 2021-2023 tusooa <tusooa@kazv.moe>
	* SPDX-License-Identifier: AGPL-3.0-or-later
	*/

	#include <libkazv-config.hpp>

	#include <zug/into_vector.hpp>
	#include <immer/algorithm.hpp>
	#include <debug.hpp>

	#include "room.hpp"

	namespace Kazv
	{
	Room::Room(lager::reader<SdkModel> sdk,
	lager::reader<std::string> roomId,
	Context<ClientAction> ctx,
	DepsT deps)
	: m_sdk(sdk)
	, m_roomId(roomId)
	, m_ctx(ctx)
	, m_deps(deps)
	, m_roomCursor(makeRoomCursor())
	{
	}

	Room::Room(lager::reader<SdkModel> sdk,
	lager::reader<std::string> roomId,
	Context<ClientAction> ctx)
	: m_sdk(sdk)
	, m_roomId(roomId)
	, m_ctx(ctx)
	, m_deps(std::nullopt)
	, m_roomCursor(makeRoomCursor())
	{
	}

	Room::Room(InEventLoopTag, std::string roomId, ContextT ctx, DepsT deps)
	: m_sdk(std::nullopt)
	, m_roomId(roomId)
	, m_ctx(ctx)
	, m_deps(deps)
	, m_roomCursor(makeRoomCursor())
	#ifdef KAZV_USE_THREAD_SAFETY_HELPER
	, KAZV_ON_EVENT_LOOP_VAR(true)
	#endif
	{
	}

	Room Room::toEventLoop() const
	{
	assert(m_deps.has_value());
	return Room(InEventLoopTag{}, currentRoomId(), m_ctx, m_deps.value());
	}

	const lager::reader<SdkModel> &Room::sdkCursor() const
	{
	if (m_sdk.has_value()) { return m_sdk.value(); }

	assert(m_deps.has_value());

	return *lager::get<SdkModelCursorKey>(m_deps.value());
	}

	const lager::reader<RoomModel> &Room::roomCursor() const
	{
	return m_roomCursor;
	}

	lager::reader<RoomModel> Room::makeRoomCursor() const
	{
	return lager::match(m_roomId)(
	[&](const lager::reader<std::string> &roomId) -> lager::reader<RoomModel> {
	return lager::with(sdkCursor().map(&SdkModel::c)[&ClientModel::roomList], roomId)
	.map([](auto rooms, auto id) {
	return rooms[id];
	}).make();
	},
	[&](const std::string &roomId) -> lager::reader<RoomModel> {
	return sdkCursor().map(&SdkModel::c)[&ClientModel::roomList].map([roomId](auto rooms) { return rooms[roomId]; }).make();
	});
	}

	std::string Room::currentRoomId() const
	{
	return lager::match(m_roomId)(
	[&](const lager::reader<std::string> &roomId) {
	return roomId.get();
	},
	[&](const std::string &roomId) {
	return roomId;
	});
	}

	auto Room::message(lager::reader<std::string> eventId) const -> lager::reader<Event>
	{
	return lager::with(roomCursor()[&RoomModel::messages], eventId)
	.map([](const auto &msgs, const auto &id) {
	return msgs[id];
	});
	}

	auto Room::members() const -> lager::reader<immer::flex_vector<std::string>>
	{
	return roomCursor().map([](auto room) {
	return room.joinedMemberIds();
	});
	}

	auto Room::invitedMembers() const -> lager::reader<immer::flex_vector<std::string>>
	{
	return roomCursor().map([](auto room) {
	return room.invitedMemberIds();
	});
	}

	auto Room::knockedMembers() const -> lager::reader<immer::flex_vector<std::string>>
	{
	return roomCursor().map([](auto room) {
	return room.knockedMemberIds();
	});
	}

	auto Room::leftMembers() const -> lager::reader<immer::flex_vector<std::string>>
	{
	return roomCursor().map([](auto room) {
	return room.leftMemberIds();
	});
	}

	auto Room::bannedMembers() const -> lager::reader<immer::flex_vector<std::string>>
	{
	return roomCursor().map([](auto room) {
	return room.bannedMemberIds();
	});
	}

	auto Room::joinedMemberEvents() const -> lager::reader<EventList>
	{
	return roomCursor().map([](auto room) {
	return room.joinedMemberEvents();
	});
	}

	auto Room::invitedMemberEvents() const -> lager::reader<EventList>
	{
	return roomCursor().map([](auto room) {
	return room.invitedMemberEvents();
	});
	}

	auto Room::knockedMemberEvents() const -> lager::reader<EventList>
	{
	return roomCursor().map([](auto room) {
	return room.knockedMemberEvents();
	});
	}

	auto Room::leftMemberEvents() const -> lager::reader<EventList>
	{
	return roomCursor().map([](auto room) {
	return room.leftMemberEvents();
	});
	}

	auto Room::bannedMemberEvents() const -> lager::reader<EventList>
	{
	return roomCursor().map([](auto room) {
	return room.bannedMemberEvents();
	});
	}

	auto Room::memberEventByCursor(lager::reader<std::string> userId) const -> lager::reader<Event>
	{
	return inviteStateOrStateEvent(userId.map([](auto id) {
	return KeyOfState{"m.room.member", id};
	}));
	}

	auto Room::memberEventFor(std::string userId) const -> lager::reader<Event>
	{
	return memberEventByCursor(lager::make_constant(userId));
	}

	lager::reader<immer::map<KeyOfState, Event>> Room::inviteStateOrState() const
	{
	return lager::with(stateEvents(), inviteState(), membership())
	.map([](const auto &stateEv, const auto &inviteSt, const auto &mem) {
	if (mem == RoomMembership::Invite) {
	return inviteSt;
	} else {
	return stateEv;
	}
	});
	}

	lager::reader<Event> Room::inviteStateOrStateEvent(lager::reader<KeyOfState> key) const
	{
	return lager::with(stateEvents(), inviteState(), membership(), key)
	.map([](const auto &stateEv, const auto &inviteSt, const auto &mem, const auto &k) {
	if (mem == RoomMembership::Invite) {
	auto maybePtr = inviteSt.find(k);
	return maybePtr ? *maybePtr : stateEv[k];
	} else {
	return stateEv[k];
	}
	});
	}

	auto Room::inviteState() const -> lager::reader<immer::map<KeyOfState, Event>>
	{
	return roomCursor()[&RoomModel::inviteState];
	}

	auto Room::heroMemberEvents() const -> lager::reader<immer::flex_vector<Event>>
	{
	using namespace lager::lenses;
	- auto idsCursor = heroIds();
	- lager::reader<immer::map<KeyOfState, Event>> statesCursor = stateEvents();
	-
	- return lager::with(idsCursor, statesCursor)
	- .xform(zug::map([](const auto &ids, const auto &states) {
	- return intoImmer(
	- immer::flex_vector<Event>{},
	- zug::map([states](const auto &id) {
	- return states[{"m.room.member", id}];
	- }),
	- ids);
	- }));
	+ return roomCursor().map(&RoomModel::heroMemberEvents);
	}

	auto Room::heroDisplayNames() const
	-> lager::reader<immer::flex_vector<std::string>>
	{
	return heroMemberEvents()
	.xform(zug::map([](const auto &events) {
	return intoImmer(
	immer::flex_vector<std::string>{},
	zug::map([](const auto &event) {
	auto content = event.content();
	return content.get().contains("displayname")
	? content.get()["displayname"].template get<std::string>()
	: std::string();
	}),
	events);
	}));
	}

	auto Room::nameOpt() const -> lager::reader<std::optional<std::string>>
	{
	using namespace lager::lenses;
	return inviteStateOrState()
	[KeyOfState{"m.room.name", ""}]
	[or_default]
	.xform(eventContent)
	.xform(zug::map([](const JsonWrap &content) {
	return content.get().contains("name")
	? std::optional<std::string>(content.get()["name"])
	: std::nullopt;
	}));
	}

	auto Room::name() const -> lager::reader<std::string>
	{
	using namespace lager::lenses;
	return nameOpt()[value_or("<no name>")];
	}

	lager::reader<bool> Room::encrypted() const
	{
	return roomCursor()[&RoomModel::encrypted];
	}

	auto Room::heroIds() const
	-> lager::reader<immer::flex_vector<std::string>>
	{
	return roomCursor().map([](const auto &room) {
	return room.heroIds;
	});
	}

	auto Room::joinedMemberCount() const -> lager::reader<std::size_t>
	{
	return roomCursor()[&RoomModel::joinedMemberCount];
	}

	auto Room::invitedMemberCount() const -> lager::reader<std::size_t>
	{
	return roomCursor()[&RoomModel::invitedMemberCount];
	}

	auto Room::avatarMxcUri() const -> lager::reader<std::string>
	{
	using namespace lager::lenses;
	return inviteStateOrState()
	[KeyOfState{"m.room.avatar", ""}]
	[or_default]
	.map([](Event ev) {
	auto content = ev.content().get();
	return content.contains("url")
	? std::string(content["url"])
	: "";
	});
	}

	auto Room::setLocalDraft(std::string localDraft) const
	-> PromiseT
	{
	using namespace CursorOp;
	return m_ctx.dispatch(UpdateRoomAction{+roomId(), SetLocalDraftAction{localDraft}});
	}

	auto Room::sendMessage(Event msg) const
	-> PromiseT
	{
	using namespace CursorOp;
	auto hasCrypto = ~sdkCursor().map([](const auto &sdk) -> bool {
	return sdk.c().crypto.has_value();
	});
	auto roomEncrypted = ~roomCursor()[&RoomModel::encrypted];
	auto noFullMembers = ~roomCursor()[&RoomModel::membersFullyLoaded]
	.map([](auto b) { return !b; });

	auto rid = +roomId();

	// Don't use m_ctx directly in the callbacks
	// as `this` may have been destroyed when
	// the callbacks are called.
	auto ctx = m_ctx;

	auto promise = ctx.createResolvedPromise(true);

	// If we do not need encryption just send it as-is
	if (! +allCursors(hasCrypto, roomEncrypted)) {
	return promise
	.then([ctx, rid, msg](auto succ) {
	if (! succ) {
	return ctx.createResolvedPromise(false);
	}
	return ctx.dispatch(SendMessageAction{rid, msg});
	});
	}

	if (! m_deps) {
	return ctx.createResolvedPromise({false, json{{"error", "missing-deps"}}});
	}

	auto deps = m_deps.value();

	promise = promise.then([ctx, rid, msg](auto) {
	return ctx.dispatch(SaveLocalEchoAction{rid, msg});
	});

	auto saveLocalEchoPromise = promise;

	// If the room member list is not complete, load it fully first.
	if (+allCursors(hasCrypto, roomEncrypted, noFullMembers)) {
	kzo.client.dbg() << "The members of " << rid
	<< " are not fully loaded." << std::endl;

	promise = promise
	.then([ctx, rid](auto) { // SaveLocalEchoAction can't fail
	return ctx.dispatch(GetRoomStatesAction{rid});
	})
	.then([ctx, rid](auto succ) {
	if (! succ) {
	kzo.client.warn() << "Loading members of " << rid
	<< " failed." << std::endl;
	return ctx.createResolvedPromise(false);
	} else {
	// XXX remove the hard-coded initialSync parameter
	return ctx.dispatch(QueryKeysAction{true})
	.then([](auto succ) {
	if (! succ) {
	kzo.client.warn() << "Query keys failed" << std::endl;
	}
	return succ;
	});
	}
	});
	}

	auto encryptEvent = [ctx, rid, msg, deps](auto &&status) {
	if (! status) { return ctx.createResolvedPromise(status); }

	kzo.client.dbg() << "encrypting megolm" << std::endl;

	auto &rg = lager::get<RandomInterface &>(deps);

	return ctx.dispatch(EncryptMegOlmEventAction{
	rid,
	msg,
	currentTimeMs(),
	rg.generateRange<RandomData>(EncryptMegOlmEventAction::maxRandomSize())
	});
	};

	auto saveEncryptedLocalEcho = [saveLocalEchoPromise, msg, rid, ctx](auto &&status) mutable {
	if (! status) { return ctx.createResolvedPromise(status); }

	return saveLocalEchoPromise
	.then([status, msg, rid, ctx](auto &&st) {
	auto txnId = st.dataStr("txnId");
	kzo.client.dbg() << "saving encrypted local echo with txn id " << txnId << std::endl;

	auto encrypted = status.dataJson("encrypted");
	auto decrypted = msg.originalJson();
	auto event = Event(encrypted).setDecryptedJson(decrypted, Event::Decrypted);

	return ctx.dispatch(SaveLocalEchoAction{rid, event, txnId});
	})
	.then([status](auto &&) {
	return status;
	});
	};

	auto maybeSendKeys = [ctx, rid, r=toEventLoop(), deps](auto status) {
	if (! status) { return ctx.createResolvedPromise(status); }

	auto encryptedEvent = status.dataJson("encrypted");
	auto content = encryptedEvent.at("content");

	auto ret = ctx.createResolvedPromise({});
	if (status.data().get().contains("key")) {
	kzo.client.dbg() << "megolm session rotated, sending session key" << std::endl;

	auto key = status.dataStr("key");
	ret = ret
	.then([rid, r, key, ctx, deps, sessionId=content.at("session_id")](auto &&) {
	auto members = (+r.roomCursor()).joinedMemberIds();
	auto client = (+r.sdkCursor()).c();
	using DeviceMapT = immer::map<std::string, immer::flex_vector<std::string>>;

	auto devicesToSend = accumulate(
	members, DeviceMapT{}, [client](auto map, auto uid) {
	return std::move(map)
	.set(uid, client.devicesToSendKeys(uid));
	});
	auto &rg = lager::get<RandomInterface &>(deps);

	return ctx.dispatch(ClaimKeysAction{
	rid, sessionId, key, devicesToSend,
	rg.generateRange<RandomData>(ClaimKeysAction::randomSize(devicesToSend))
	})
	.then([rid, ctx, deps, devicesToSend](auto status) {
	if (! status) { return ctx.createResolvedPromise({}); }

	kzo.client.dbg() << "olm-encrypting key event" << std::endl;

	auto keyEv = status.dataJson("keyEvent");
	auto &rg = lager::get<RandomInterface &>(deps);

	return ctx.dispatch(PrepareForSharingRoomKeyAction{
	rid,
	devicesToSend, keyEv,
	rg.generateRange<RandomData>(EncryptOlmEventAction::randomSize(devicesToSend))
	});
	})
	.then([ctx, r, devicesToSend, rid](auto status) {
	if (! status) { return ctx.createResolvedPromise({}); }
	auto txnId = status.dataStr("txnId");
	return r.sendPendingKeyEvent(txnId);
	});
	});
	}
	return ret
	.then([ctx, prevStatus=status](auto status) {
	if (! status) { return status; }
	return prevStatus;
	});
	};

	auto sendEncryptedEvent = [ctx, rid, saveLocalEchoPromise, msg](auto status) mutable {
	if (! status) { return ctx.createResolvedPromise(status); }
	return saveLocalEchoPromise.then([ctx, rid, status, msg](auto st) {
	auto txnId = st.dataStr("txnId");
	kzo.client.dbg() << "sending encrypted message with txn id " << txnId << std::endl;
	auto encrypted = status.dataJson("encrypted");
	auto decrypted = msg.originalJson();
	auto event = Event(encrypted).setDecryptedJson(decrypted, Event::Decrypted);

	return ctx.dispatch(SendMessageAction{rid, event, txnId});
	});
	};

	auto maybeSetStatusToFailed = [ctx, saveLocalEchoPromise, rid](const auto &st) mutable {
	if (st.success()) {
	return ctx.createResolvedPromise(st);
	}

	return saveLocalEchoPromise
	.then([rid, ctx](const auto &saveLocalEchoStatus) {
	auto txnId = saveLocalEchoStatus.dataStr("txnId");
	return ctx.dispatch(UpdateLocalEchoStatusAction{rid, txnId, LocalEchoDesc::Failed});
	})
	.then([st](const auto &) { // can't fail
	return st;
	});
	};

	return promise
	.then(encryptEvent)
	.then(saveEncryptedLocalEcho)
	.then(maybeSendKeys)
	.then(sendEncryptedEvent)
	.then(maybeSetStatusToFailed);
	}

	auto Room::sendTextMessage(std::string text) const
	-> PromiseT
	{
	json j{
	{"type", "m.room.message"},
	{"content", {
	{"msgtype", "m.text"},
	{"body", text}
	}
	}
	};
	Event e{j};
	return sendMessage(e);
	}

	auto Room::resendMessage(std::string txnId) const
	-> PromiseT
	{
	return m_ctx.createResolvedPromise({})
	.then([that=toEventLoop()](const auto &) {
	kzo.client.dbg() << "resending all pending key events" << std::endl;
	return that.sendAllPendingKeyEvents();
	})
	.then([that=toEventLoop(), txnId](const auto &sendKeyStatus) {
	if (!sendKeyStatus.success()) {
	kzo.client.warn() << "resending all pending key events failed" << std::endl;
	return that.m_ctx.createResolvedPromise(sendKeyStatus);
	}
	auto maybeLocalEcho = that.roomCursor().map([txnId](const auto &room) {
	return room.getLocalEchoByTxnId(txnId);
	}).make();
	auto roomEncrypted = that.roomCursor()[&RoomModel::encrypted].make();

	if (!maybeLocalEcho.get()) {
	return that.m_ctx.createResolvedPromise({false, json{
	{"errorCode", "MOE_KAZV_MXC_NO_SUCH_TXNID"},
	{"error", "No such txn id"}
	}});
	}
	auto localEcho = maybeLocalEcho.get().value();
	auto event = localEcho.event;
	auto rid = that.roomId().make().get();
	if ((roomEncrypted.get() && event.encrypted()) \|\| !roomEncrypted.get()) {
	kzo.client.info() << "resending message with txn id " << localEcho.txnId << std::endl;
	return that.m_ctx.dispatch(SendMessageAction{rid, event, localEcho.txnId});
	} else {
	kzo.client.info() << "room encrypted but event isn't, just resend everything" << std::endl;
	return that.m_ctx.dispatch(RoomListAction{UpdateRoomAction{rid, RemoveLocalEchoAction{localEcho.txnId}}})
	.then([localEcho, that](auto) { // Can't fail
	return that.sendMessage(localEcho.event);
	});
	}
	});
	}

	auto Room::redactEvent(std::string eventId, std::optional<std::string> reason) const -> PromiseT
	{
	return m_ctx.dispatch(RedactEventAction{
	roomId().make().get(),
	eventId,
	reason
	});
	}

	auto Room::sendPendingKeyEvent(std::string txnId) const -> PromiseT
	{
	return m_ctx.createResolvedPromise({})
	.then([txnId, r=toEventLoop()](auto &&) {
	auto ctx = r.m_ctx;
	auto rid = r.roomId().make().get();
	kzo.client.dbg() << "sending key event as to-device message" << std::endl;

	kzo.client.dbg() << "txnId of key event: " << txnId << std::endl;
	auto maybePending = r.roomCursor().make().get().getPendingRoomKeyEventByTxnId(txnId);
	if (!maybePending.has_value()) {
	kzo.client.warn() << "No such pending room key event";
	return ctx.createResolvedPromise(EffectStatus(/* succ = */ false, json::object(
	{{"errorCode", "MOE_KAZV_MXC_NO_SUCH_PENDING_ROOM_KEY_EVENT"},
	{"error", "No such pending room key event"}}
	)));
	}
	auto pending = maybePending.value();
	auto event = pending.event;
	return ctx.dispatch(SendToDeviceMessageAction{event, pending.devices})
	.then([ctx, rid, txnId](const auto &sendToDeviceStatus) {
	if (!sendToDeviceStatus.success()) {
	return ctx.createResolvedPromise(sendToDeviceStatus);
	} else {
	return ctx.dispatch(UpdateRoomAction{rid, RemovePendingRoomKeyAction{txnId}});
	}
	});
	});
	}

	auto Room::sendAllPendingKeyEvents() const -> PromiseT
	{
	return m_ctx.createResolvedPromise({})
	.then([r=toEventLoop()](auto &&) {
	auto pendingEvents = r.pendingRoomKeyEvents().make().get();
	if (pendingEvents.empty()) {
	return r.m_ctx.createResolvedPromise(EffectStatus(/* succ = */ true));
	} else {
	auto txnId = pendingEvents[0].txnId;
	return r.sendPendingKeyEvent(txnId)
	.then([r, txnId](const auto &stat) {
	if (!stat.success()) {
	kzo.client.warn() << "Can't send pending key event of txnId " << txnId << std::endl;
	return r.m_ctx.createResolvedPromise(stat);
	}
	return r.sendAllPendingKeyEvents();
	});
	}
	});
	}

	auto Room::refreshRoomState() const
	-> PromiseT
	{
	using namespace CursorOp;
	return m_ctx.dispatch(GetRoomStatesAction{+roomId()});
	}

	auto Room::getStateEvent(std::string type, std::string stateKey) const
	-> PromiseT
	{
	using namespace CursorOp;
	return m_ctx.dispatch(GetStateEventAction{+roomId(), type, stateKey});
	}

	auto Room::sendStateEvent(Event state) const
	-> PromiseT
	{
	using namespace CursorOp;
	return m_ctx.dispatch(SendStateEventAction{+roomId(), state});
	}

	auto Room::setName(std::string name) const
	-> PromiseT
	{
	json j{
	{"type", "m.room.name"},
	{"content", {
	{"name", name}
	}
	}
	};
	Event e{j};
	return sendStateEvent(e);
	}

	auto Room::setTopic(std::string topic) const
	-> PromiseT
	{
	json j{
	{"type", "m.room.topic"},
	{"content", {
	{"topic", topic}
	}
	}
	};
	Event e{j};
	return sendStateEvent(e);
	}

	auto Room::invite(std::string userId) const
	-> PromiseT
	{
	using namespace CursorOp;
	return m_ctx.dispatch(InviteToRoomAction{+roomId(), userId});
	}

	auto Room::typingUsers() const -> lager::reader<immer::flex_vector<std::string>>
	{
	using namespace lager::lenses;
	return ephemeral("m.typing")
	.xform(eventContent
	\| jsonAtOr("user_ids", immer::flex_vector<std::string>{}));
	}

	auto Room::typingMemberEvents() const -> lager::reader<EventList>
	{
	return lager::with(typingUsers(), roomCursor()[&RoomModel::stateEvents])
	.map([](const auto &userIds, const auto &events) {
	return intoImmer(EventList{}, zug::map([events](const auto &id) {
	return events[KeyOfState{"m.room.member", id}];
	}), userIds);
	});
	}

	auto Room::setTyping(bool typing, std::optional<int> timeoutMs) const
	-> PromiseT
	{
	using namespace CursorOp;
	return m_ctx.dispatch(SetTypingAction{+roomId(), typing, timeoutMs});
	}

	auto Room::leave() const
	-> PromiseT
	{
	using namespace CursorOp;
	return m_ctx.dispatch(LeaveRoomAction{+roomId()});
	}

	auto Room::forget() const
	-> PromiseT
	{
	using namespace CursorOp;
	return m_ctx.dispatch(ForgetRoomAction{+roomId()});
	}

	auto Room::kick(std::string userId, std::optional<std::string> reason) const
	-> PromiseT
	{
	using namespace CursorOp;
	return m_ctx.dispatch(KickAction{+roomId(), userId, reason});
	}

	auto Room::ban(std::string userId, std::optional<std::string> reason) const
	-> PromiseT
	{
	using namespace CursorOp;
	return m_ctx.dispatch(BanAction{+roomId(), userId, reason});
	}

	auto Room::unban(std::string userId/, std::optional<std::string> reason/) const
	-> PromiseT
	{
	using namespace CursorOp;
	return m_ctx.dispatch(UnbanAction{+roomId(), userId});
	}

	auto Room::setAccountData(Event accountDataEvent) const -> PromiseT
	{
	using namespace CursorOp;
	return m_ctx.dispatch(SetAccountDataPerRoomAction{+roomId(), accountDataEvent});
	}

	auto Room::tags() const -> lager::reader<immer::map<std::string, double>>
	{
	return roomCursor().map(&RoomModel::tags);
	}

	auto Room::addOrSetTag(std::string tagId, std::optional<double> order) const -> PromiseT
	{
	using namespace CursorOp;
	return m_ctx.createResolvedPromise({})
	.then([that=toEventLoop(), tagId, order](auto) {
	return that.setAccountData((+that.roomCursor()).makeAddTagEvent(tagId, order));
	});
	}

	auto Room::removeTag(std::string tagId) const -> PromiseT
	{
	using namespace CursorOp;
	return m_ctx.createResolvedPromise({})
	.then([that=toEventLoop(), tagId](auto) {
	return that.setAccountData((+that.roomCursor()).makeRemoveTagEvent(tagId));
	});
	}

	auto Room::setPinnedEvents(immer::flex_vector<std::string> eventIds) const
	-> PromiseT
	{
	json j{
	{"type", "m.room.pinned_events"},
	{"content", {
	{"pinned", eventIds}
	}
	}
	};
	Event e{j};
	return sendStateEvent(e);
	}

	auto Room::timelineEventIds() const -> lager::reader<immer::flex_vector<std::string>>
	{
	return roomCursor()[&RoomModel::timeline];
	}

	auto Room::messagesMap() const -> lager::reader<immer::map<std::string, Event>>
	{
	return roomCursor()[&RoomModel::messages];
	}

	auto Room::timelineEvents() const -> lager::reader<immer::flex_vector<Event>>
	{
	return roomCursor()
	.xform(zug::map([](auto r) {
	auto messages = r.messages;
	auto timeline = r.timeline;
	return intoImmer(
	immer::flex_vector<Event>{},
	zug::map([=](auto eventId) {
	return messages[eventId];
	}),
	timeline);
	}));
	}

	auto Room::timelineGaps() const
	-> lager::reader<immer::map<std::string /* eventId */,
	std::string /* prevBatch */>>
	{
	return roomCursor()[&RoomModel::timelineGaps];
	}

	auto Room::paginateBackFromEvent(std::string eventId) const
	-> PromiseT
	{
	using namespace CursorOp;
	return m_ctx.dispatch(PaginateTimelineAction{
	+roomId(), eventId, std::nullopt});
	}

	auto Room::localEchoes() const -> lager::reader<immer::flex_vector<LocalEchoDesc>>
	{
	return roomCursor()[&RoomModel::localEchoes];
	}

	auto Room::removeLocalEcho(std::string txnId) const -> PromiseT
	{
	return m_ctx.dispatch(RoomListAction{UpdateRoomAction{
	currentRoomId(),
	RemoveLocalEchoAction{txnId},
	}});
	}

	auto Room::pendingRoomKeyEvents() const -> lager::reader<immer::flex_vector<PendingRoomKeyEvent>>
	{
	return roomCursor()[&RoomModel::pendingRoomKeyEvents];
	}

	auto Room::powerLevels() const -> lager::reader<PowerLevelsDesc>
	{
	return state({"m.room.power_levels", ""})
	.map([](const Event &event) {
	return PowerLevelsDesc(event);
	});
	}

	auto Room::relatedEvents(lager::reader<std::string> eventId, std::string relType) const -> lager::reader<EventList>
	{
	return lager::with(
	eventId,
	roomCursor()[&RoomModel::reverseEventRelationships],
	roomCursor()[&RoomModel::messages]
	).map([relType](const auto &eid, const auto &rels, const auto &msgs) {
	auto relatedEvents = zug::into_vector(
	zug::map([msgs](const auto &id) {
	return msgs[id];
	}),
	rels[eid][relType]
	);
	std::sort(relatedEvents.begin(), relatedEvents.end(), [](const Event &a, const Event &b) {
	return std::make_tuple(a.originServerTs(), a.id()) < std::make_tuple(b.originServerTs(), b.id());
	});

	return EventList(relatedEvents.begin(), relatedEvents.end());
	}).make();
	}
	}
	diff --git a/src/tests/client/room-test.cpp b/src/tests/client/room-test.cpp
	index 2f21adf..d4aabfd 100644
	--- a/src/tests/client/room-test.cpp
	+++ b/src/tests/client/room-test.cpp
	@@ -1,435 +1,448 @@
	/*
	* This file is part of libkazv.
	* SPDX-FileCopyrightText: 2021-2023 tusooa <tusooa@kazv.moe>
	* SPDX-License-Identifier: AGPL-3.0-or-later
	*/

	#include <libkazv-config.hpp>

	#include <catch2/catch_all.hpp>

	#include <lager/event_loop/boost_asio.hpp>
	#include <boost/asio.hpp>

	#include <sdk.hpp>
	#include <cprjobhandler.hpp>
	#include <lagerstoreeventemitter.hpp>
	#include <asio-promise-handler.hpp>

	#include "client-test-util.hpp"

	#include "factory.hpp"

	using namespace Kazv;
	using namespace Kazv::Factory;

	TEST_CASE("Result of Room::toEventLoop() should not vary with the original cursor", "[client][room]")
	{
	auto io = boost::asio::io_context{};
	auto jh = Kazv::CprJobHandler{io.get_executor()};
	auto ee = Kazv::LagerStoreEventEmitter(lager::with_boost_asio_event_loop{io.get_executor()});

	auto roomWithId = [](std::string roomId) {
	auto r = RoomModel{};
	r.roomId = roomId;
	return r;
	};

	auto initModel = Kazv::SdkModel{};
	initModel.client.roomList.rooms = std::move(initModel.client.roomList.rooms)
	.set("!foo:example.org", roomWithId("!foo:example.org"))
	.set("!bar:example.org", roomWithId("!bar:example.org"));
	auto sdk = Kazv::makeSdk(
	initModel,
	jh,
	ee,
	Kazv::AsioPromiseHandler{io.get_executor()},
	zug::identity
	);

	auto ctx = sdk.context();

	auto client = sdk.client();

	auto roomId = lager::make_state(std::string("!foo:example.org"), lager::automatic_tag{});

	auto room = client.roomByCursor(roomId);

	using namespace Kazv::CursorOp;

	REQUIRE(+room.roomId() == "!foo:example.org");

	auto roomInEventLoop = room.toEventLoop();

	REQUIRE(+roomInEventLoop.roomId() == "!foo:example.org");

	roomId.set("!bar:example.org");

	REQUIRE(+room.roomId() == "!bar:example.org");
	REQUIRE(+roomInEventLoop.roomId() == "!foo:example.org");
	}

	static const std::string exampleRoomId = "!example:example.org";

	static auto memberFoo = Event(R"(
	{
	"content": {
	"displayname": "foo's name",
	"membership": "join"
	},
	"event_id": "some id2",
	"room_id": "!example:example.org",
	"type": "m.room.member",
	"state_key": "@foo:example.org"
	})"_json);

	static auto memberBar = Event(R"(
	{
	"content": {
	"displayname": "bar's name",
	"membership": "join"
	},
	"event_id": "some id3",
	"room_id": "!example:example.org",
	"type": "m.room.member",
	"state_key": "@foo:example.org"
	})"_json);

	static auto exampleRoomWithoutNameEvent()
	{
	RoomModel model;
	model.membership = RoomMembership::Join;
	model.roomId = "!example:example.org";
	model.stateEvents = model.stateEvents
	.set({"m.room.member", "@foo:example.org"}, memberFoo)
	.set({"m.room.member", "@bar:example.org"}, memberBar);
	model.heroIds = {"@foo:example.org", "@bar:example.org"};
	return model;
	}

	static auto roomNameEvent = Event(R"(
	{
	"content": {
	"name": "some name"
	},
	"event_id": "some id",
	"room_id": "!example:example.org",
	"type": "m.room.name",
	"state_key": ""
	})"_json);

	// https://spec.matrix.org/v1.7/client-server-api/#mroomavatar
	static Event avatarEvent = R"({
	"content": {
	"info": {
	"h": 398,
	"mimetype": "image/jpeg",
	"size": 31037,
	"w": 394
	},
	"url": "mxc://example.org/JWEIFJgwEIhweiWJE"
	},
	"event_id": "$143273582443PhrSn:example.org",
	"origin_server_ts": 1432735824653,
	"room_id": "!jEsUZKDJdhlrceRyVU:example.org",
	"sender": "@example:example.org",
	"state_key": "",
	"type": "m.room.avatar",
	"unsigned": {
	"age": 1234
	}
	})"_json;

	static auto exampleRoomWithNameEvent()
	{
	auto model = exampleRoomWithoutNameEvent();
	model.stateEvents = model.stateEvents
	.set({"m.room.name", ""}, roomNameEvent);
	return model;
	};

	static auto sdkWith(RoomModel room)
	{
	SdkModel model;
	model.client.roomList.rooms = model.client.roomList.rooms.set(room.roomId, room);
	return model;
	}

	static auto makeRoomWithDumbContext(RoomModel room)
	{
	auto cursor = lager::make_constant(sdkWith(room));
	auto nameCursor = lager::make_constant(room.roomId);
	return Room(cursor, nameCursor, dumbContext());
	}

	TEST_CASE("Room::nameOpt()", "[client][room][getter]")
	{
	WHEN("the room has a name") {
	auto room = makeRoomWithDumbContext(exampleRoomWithNameEvent());
	THEN("it should give out the name") {
	auto val = room.nameOpt().make().get();
	REQUIRE(val.has_value());
	REQUIRE(val.value() == "some name");
	}
	}

	WHEN("the room has no name") {
	auto room = makeRoomWithDumbContext(exampleRoomWithoutNameEvent());
	THEN("it should give out nullopt") {
	auto val = room.nameOpt().make().get();
	REQUIRE(!val.has_value());
	}
	}
	}

	TEST_CASE("Room::heroMemberEvents()", "[client][room][getter]")
	{
	- auto room = makeRoomWithDumbContext(exampleRoomWithNameEvent());
	+ auto r = exampleRoomWithNameEvent();
	+ {
	+ auto val = r.heroMemberEvents();
	+ REQUIRE(val.size() == 2);
	+ REQUIRE(std::any_of(val.begin(), val.end(),
	+ [](const auto &v) {
	+ return v.id() == "some id2";
	+ }));
	+ REQUIRE(std::any_of(val.begin(), val.end(),
	+ [](const auto &v) {
	+ return v.id() == "some id3";
	+ }));
	+ }
	+ auto room = makeRoomWithDumbContext(r);
	THEN("it should give out the events") {
	auto val = room.heroMemberEvents().make().get();
	REQUIRE(val.size() == 2);
	REQUIRE(std::any_of(val.begin(), val.end(),
	[](const auto &v) {
	return v.id() == "some id2";
	}));
	REQUIRE(std::any_of(val.begin(), val.end(),
	[](const auto &v) {
	return v.id() == "some id3";
	}));
	}
	}

	TEST_CASE("Room::heroDisplayNames()", "[client][room][getter]")
	{
	auto room = makeRoomWithDumbContext(exampleRoomWithNameEvent());
	THEN("it should give out the names") {
	auto val = room.heroDisplayNames().make().get();
	REQUIRE(val.size() == 2);
	REQUIRE(std::any_of(val.begin(), val.end(),
	[](const auto &v) {
	return v == "foo's name";
	}));
	REQUIRE(std::any_of(val.begin(), val.end(),
	[](const auto &v) {
	return v == "bar's name";
	}));
	}
	}

	TEST_CASE("Room::encrypted()", "[client][room][getter]")
	{
	auto roomModel = exampleRoomWithNameEvent();
	WHEN("the room is encrypted") {
	roomModel.encrypted = true;
	auto room = makeRoomWithDumbContext(roomModel);
	THEN("it should give out true") {
	auto val = room.encrypted().make().get();
	REQUIRE(val);
	}
	}

	WHEN("the room is not encrypted") {
	roomModel.encrypted = false;
	auto room = makeRoomWithDumbContext(roomModel);
	THEN("it should give out false") {
	auto val = room.encrypted().make().get();
	REQUIRE(!val);
	}
	}
	}

	TEST_CASE("Room::avatarMxcUri()", "[client][room][getter]")
	{
	auto roomModel = exampleRoomWithoutNameEvent();
	roomModel.stateEvents = roomModel.stateEvents.set(KeyOfState{"m.room.avatar", ""}, avatarEvent);

	auto room = makeRoomWithDumbContext(roomModel);
	REQUIRE(room.avatarMxcUri().make().get() == "mxc://example.org/JWEIFJgwEIhweiWJE");
	}

	TEST_CASE("Room::typingMemberEvents()", "[client][room][getter][ephemeral]")
	{
	auto roomModel = exampleRoomWithoutNameEvent();
	auto typingJson = R"({
	"content": {
	"user_ids": ["@foo:example.org", "@bar:example.org"]
	},
	"type": "m.typing"
	})"_json;
	roomModel.ephemeral = roomModel.ephemeral.set("m.typing", Event{typingJson});

	auto room = makeRoomWithDumbContext(roomModel);
	REQUIRE(room.typingMemberEvents().make().get() == EventList{
	room.memberEventFor("@foo:example.org").make().get(),
	room.memberEventFor("@bar:example.org").make().get(),
	});
	}

	TEST_CASE("Room::message()", "[client][room][getter][timeline]")
	{
	auto roomModel = exampleRoomWithoutNameEvent();
	auto messageEvent = Event(R"({
	"content": {
	"body": "This is an example text message",
	"msgtype": "m.text",
	"format": "org.matrix.custom.html",
	"formatted_body": "<b>This is an example text message</b>"
	},
	"type": "m.room.message",
	"event_id": "$anothermessageevent:example.org",
	"room_id": "!726s6s6q:example.com",
	"sender": "@example:example.org",
	"origin_server_ts": 1432735824653,
	"unsigned": {
	"age": 1234
	}
	})"_json);
	roomModel.messages = roomModel.messages.set(messageEvent.id(), messageEvent);

	auto room = makeRoomWithDumbContext(roomModel);
	REQUIRE(room.message(lager::make_constant(messageEvent.id())).make().get() == messageEvent);
	}

	TEST_CASE("Room::memberEventFor(), Room::memberEventByCursor", "[client][room][getter][state]")
	{
	auto roomModel = exampleRoomWithoutNameEvent();
	auto memberFooJson = memberFoo.raw().get();
	memberFooJson["content"]["displayname"] = "some other name";
	auto newMemberFooEvent = Event(memberFooJson);
	roomModel.inviteState = roomModel.inviteState
	.set({"m.room.member", "@foo:example.org"}, newMemberFooEvent);

	WHEN("membership is Invite")
	{
	roomModel.membership = RoomMembership::Invite;
	auto room = makeRoomWithDumbContext(roomModel);
	THEN("should prefer inviteState")
	{
	REQUIRE(room.memberEventFor("@foo:example.org").make().get() == newMemberFooEvent);
	REQUIRE(room.memberEventFor("@bar:example.org").make().get() == memberBar);
	}
	}

	WHEN("membership is not Invite")
	{
	roomModel.membership = RoomMembership::Leave;
	auto room = makeRoomWithDumbContext(roomModel);
	THEN("should ignore inviteState")
	{
	REQUIRE(room.memberEventFor("@foo:example.org").make().get() == memberFoo);
	REQUIRE(room.memberEventFor("@bar:example.org").make().get() == memberBar);
	}
	}
	}

	TEST_CASE("Room::name(), Room::nameOpt()", "[client][room][getter][state]")
	{
	auto roomModel = exampleRoomWithoutNameEvent();
	WHEN("membership is invite and invite state contains name event")
	{
	roomModel.membership = RoomMembership::Invite;
	roomModel.inviteState = roomModel.inviteState.set({"m.room.name", ""}, roomNameEvent);
	auto room = makeRoomWithDumbContext(roomModel);
	THEN("it should give that name")
	{
	auto expectedName = roomNameEvent.content().get()["name"].template get<std::string>();
	REQUIRE(room.nameOpt().make().get().value() == expectedName);
	REQUIRE(room.name().make().get() == expectedName);
	}
	}

	WHEN("membership is not invite and invite state contains the name event")
	{
	roomModel.membership = RoomMembership::Join;
	roomModel.inviteState = roomModel.inviteState.set({"m.room.name", ""}, roomNameEvent);
	auto room = makeRoomWithDumbContext(roomModel);
	THEN("it should ignore that name")
	{
	REQUIRE(!room.nameOpt().make().get().has_value());
	REQUIRE(room.name().make().get() == "<no name>");
	}
	}
	}

	TEST_CASE("Room::avatarMxcUri()", "[client][room][getter][state]")
	{
	auto roomModel = exampleRoomWithoutNameEvent();
	roomModel.inviteState = roomModel.inviteState.set({"m.room.avatar", ""}, avatarEvent);
	WHEN("membership is invite and invite state contains name event")
	{
	roomModel.membership = RoomMembership::Invite;
	auto room = makeRoomWithDumbContext(roomModel);
	THEN("it should give that avatar")
	{
	auto expectedAvatar = avatarEvent.content().get()["url"].template get<std::string>();
	REQUIRE(room.avatarMxcUri().make().get() == expectedAvatar);
	}
	}

	WHEN("membership is not invite and invite state contains the name event")
	{
	roomModel.membership = RoomMembership::Join;
	auto room = makeRoomWithDumbContext(roomModel);
	THEN("it should ignore that avatar")
	{
	REQUIRE(room.avatarMxcUri().make().get() == "");
	}
	}
	}

	TEST_CASE("Room::joinedMemberEvents()", "[client][room][getter][state]")
	{
	auto members = immer::flex_vector<Event>{makeMemberEvent(), makeMemberEvent(), makeMemberEvent()};
	auto roomModel = makeRoom(
	withRoomState(members)
	\| withRoomState({
	makeMemberEvent(withMembership("leave")),
	makeMemberEvent(withMembership("invite")),
	makeMemberEvent(withMembership("ban")),
	})
	);

	auto room = makeRoomWithDumbContext(roomModel);
	THEN("it returns the joined members only")
	{
	REQUIRE_THAT(room.joinedMemberEvents().make().get(), Catch::Matchers::UnorderedRangeEquals(members));
	}
	}

	TEST_CASE("Room::powerLevels()", "[client][room][state][power-levels]")
	{
	auto content = json{
	{"users", {{"@foo:example.com", 40}}},
	};

	auto powerLevelsEvent = makeEvent(
	withEventType("m.room.power_levels")
	\| withEventContent(content)
	);
	auto roomModel = makeRoom(withRoomState({powerLevelsEvent}));
	auto room = makeRoomWithDumbContext(roomModel);
	REQUIRE(room.powerLevels().make().get().originalEvent() == powerLevelsEvent);
	}

	TEST_CASE("Room::{timelineEvents,timelineEventIds,messagesMap}()", "[client][room][getter][timeline]")
	{
	auto timelineEvents = immer::flex_vector<Event>{makeEvent(), makeEvent(), makeEvent()};
	auto timelineEventIds = intoImmer(
	immer::flex_vector<std::string>{},
	zug::map([](const auto &ev) { return ev.id(); }),
	timelineEvents
	);
	auto messagesMap = intoImmer(
	immer::map<std::string, Event>{},
	zug::map([](const auto &ev) { return std::make_pair(ev.id(), ev); }),
	timelineEvents
	);
	auto roomModel = makeRoom(
	withRoomTimeline(timelineEvents)
	);

	auto room = makeRoomWithDumbContext(roomModel);
	REQUIRE(room.timelineEvents().make().get() == timelineEvents);
	REQUIRE(room.timelineEventIds().make().get() == timelineEventIds);
	REQUIRE(room.messagesMap().make().get() == messagesMap);
	}

File Metadata

Mime Type: text/x-diff
Expires: Sun, Jan 19, 8:53 PM (1 d, 20 h)
Storage Engine: blob
Storage Format: Raw Data
Storage Handle: 55444
Default Alt Text: (75 KB)

No OneTemporaryActions

View Options

File Metadata

Event Timeline

No OneTemporary
Actions