From 36b43b860750ca12d9d9d5881e284aa000ad4947 Mon Sep 17 00:00:00 2001
From: Jason Volk <jason@zemos.net>
Date: Fri, 4 May 2018 20:10:35 -0700
Subject: [PATCH] ircd::m::dbs: Add indexing sequence for _room_head.

---
 include/ircd/m/dbs.h |  2 ++
 ircd/m/dbs.cc        | 61 ++++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 63 insertions(+)

diff --git a/include/ircd/m/dbs.h b/include/ircd/m/dbs.h
index 4ae7d9f8b..c908de28e 100644
--- a/include/ircd/m/dbs.h
+++ b/include/ircd/m/dbs.h
@@ -77,6 +77,8 @@ struct ircd::m::dbs::write_opts
 	bool present {true};
 	bool history {true};
 	bool indexer {true};
+	bool head {true};
+	bool refs {true};
 };
 
 /// Database Schema Descriptors
diff --git a/ircd/m/dbs.cc b/ircd/m/dbs.cc
index 7a40b9416..b99d29caf 100644
--- a/ircd/m/dbs.cc
+++ b/ircd/m/dbs.cc
@@ -103,6 +103,7 @@ namespace ircd::m::dbs
 	static void _index__room_state(db::txn &,  const event &, const write_opts &);
 	static void _index__room_events(db::txn &,  const event &, const write_opts &, const string_view &);
 	static void _index__room_joined(db::txn &, const event &, const write_opts &);
+	static void _index__room_head(db::txn &, const event &, const write_opts &);
 	static string_view _index_state(db::txn &, const event &, const write_opts &);
 	static string_view _index_redact(db::txn &, const event &, const write_opts &);
 	static string_view _index_ephem(db::txn &, const event &, const write_opts &);
@@ -152,6 +153,9 @@ ircd::m::dbs::write(db::txn &txn,
 		txn, byte_view<string_view>(opts.idx), event, event_column, opts.op
 	};
 
+	if(opts.head || opts.refs)
+		_index__room_head(txn, event, opts);
+
 	if(defined(json::get<"state_key"_>(event)))
 		return _index_state(txn, event, opts);
 
@@ -256,6 +260,63 @@ catch(const std::exception &e)
 	throw;
 }
 
+void
+ircd::m::dbs::_index__room_head(db::txn &txn,
+                                const event &event,
+                                const write_opts &opts)
+{
+	thread_local char buf[ROOM_HEAD_KEY_MAX_SIZE];
+	const ctx::critical_assertion ca;
+
+	if(opts.head)
+	{
+		const string_view &key
+		{
+			room_head_key(buf, at<"room_id"_>(event), at<"event_id"_>(event))
+		};
+
+		db::txn::append
+		{
+			txn, room_head,
+			{
+				opts.op,
+				key,
+				byte_view<string_view>{opts.idx}
+			}
+		};
+	}
+
+	//TODO: If op is DELETE and we are deleting this event and thereby
+	//TODO: potentially creating a gap in the reference graph (just for us
+	//TODO: though) can we *re-add* the prev_events to the head?
+
+	if(opts.refs && opts.op == db::op::SET)
+	{
+		const m::event::prev &prev{event};
+		for(const json::array &p : json::get<"prev_events"_>(prev))
+		{
+			const auto &event_id
+			{
+				unquote(p.at(0))
+			};
+
+			const string_view &key
+			{
+				room_head_key(buf, at<"room_id"_>(event), event_id)
+			};
+
+			db::txn::append
+			{
+				txn, room_head,
+				{
+					db::op::DELETE,
+					key,
+				}
+			};
+		}
+	}
+}
+
 /// Adds the entry for the room_events column into the txn.
 /// You need find/create the right state_root before this.
 void