From 5590238b1f8c765f60290a63f46d704020c66af4 Mon Sep 17 00:00:00 2001
From: Jason Volk <jason@zemos.net>
Date: Sun, 13 May 2018 19:16:08 -0700
Subject: [PATCH] modules/client/rooms: Add a state reindexer routine.

---
 modules/client/rooms/state.cc | 40 +++++++++++++++++++++++++++++++++++
 1 file changed, 40 insertions(+)

diff --git a/modules/client/rooms/state.cc b/modules/client/rooms/state.cc
index 720eba7ba..a738861b5 100644
--- a/modules/client/rooms/state.cc
+++ b/modules/client/rooms/state.cc
@@ -215,3 +215,43 @@ state__iov(const room &room,
 
 	return commit(room, event, content);
 }
+
+extern "C" size_t
+state__rebuild_present(const m::room &room)
+{
+	size_t ret{0};
+	const auto create_id
+	{
+		room::state{room}.get("m.room.create")
+	};
+
+	room::messages it
+	{
+		room, create_id
+	};
+
+	if(!it)
+		return ret;
+
+	db::txn txn{*dbs::events};
+	for(; it; ++it)
+	{
+		const event &event{*it};
+		if(!defined(json::get<"state_key"_>(event)))
+			continue;
+
+		dbs::write_opts opts;
+		opts.idx = it.event_idx();
+		opts.present = true;
+		opts.history = false;
+		opts.head = false;
+		opts.refs = false;
+
+		dbs::_index__room_state(txn, event, opts);
+		dbs::_index__room_joined(txn, event, opts);
+		++ret;
+	}
+
+	txn();
+	return ret;
+}