construct/modules/m_user.cc

// Matrix Construct
//
// Copyright (C) Matrix Construct Developers, Authors & Contributors
// Copyright (C) 2016-2018 Jason Volk <jason@zemos.net>
//
// Permission to use, copy, modify, and/or distribute this software for any
// purpose with or without fee is hereby granted, provided that the above
// copyright notice and this permission notice is present in all copies. The
// full license for this software is available in the LICENSE file.

using namespace ircd::m;
using namespace ircd;

mapi::header
IRCD_MODULE
{
	"Matrix user library; modular components."
};

extern "C" m::user
user_create(const m::user::id &user_id,
            const json::members &contents)
{
	const m::user user
	{
		user_id
	};

	const m::room::id::buf user_room_id
	{
		user.room_id()
	};

	//TODO: ABA
	//TODO: TXN
	m::room user_room
	{
		m::create(user_room_id, m::me.user_id, "user")
	};

	//TODO: ABA
	//TODO: TXN
	send(user.users, m::me.user_id, "ircd.user", user.user_id, contents);
	return user;
}

///////////////////////////////////////////////////////////////////////////////
//
// m/user/breadcrumb_rooms.h
//

ircd::m::event::id::buf
ircd::m::user::breadcrumb_rooms::set(const json::array &rooms)
const
{
	const json::strung object
	{
		json::members
		{
			{ "rooms", rooms }
		}
	};

	return account_data.set("im.vector.riot.breadcrumb_rooms", object);
}

bool
ircd::m::user::breadcrumb_rooms::for_each(const closure_bool &closure)
const
{
	bool ret{true};
	get(std::nothrow, [&closure, &ret]
	(const json::array &rooms)
	{
		for(const json::string &room : rooms)
			if(!closure(room))
			{
				ret = false;
				break;
			}
	});

	return ret;
}

void
ircd::m::user::breadcrumb_rooms::get(const closure &closure)
const
{
	if(!get(std::nothrow, closure))
		throw m::NOT_FOUND
		{
			"User has no breadcrumb_rooms set in their account_data."
		};
}

bool
ircd::m::user::breadcrumb_rooms::get(std::nothrow_t,
                                     const closure &closure)
const
{
	return account_data.get(std::nothrow, "im.vector.riot.breadcrumb_rooms", [&closure]
	(const string_view &key, const json::object &object)
	{
		const json::array &rooms
		{
			object["rooms"]
		};

		closure(rooms);
	});
}

///////////////////////////////////////////////////////////////////////////////
//
// m/user/highlight.h
//

decltype(ircd::m::user::highlight::enable_count)
ircd::m::user::highlight::enable_count
{
	{ "name",    "ircd.m.user.highlight.enable.count" },
	{ "default", true                                 },
};

decltype(ircd::m::user::highlight::match_mxid_full)
ircd::m::user::highlight::match_mxid_full
{
	{ "name",    "ircd.m.user.highlight.match.mxid.full" },
	{ "default", true                                    },
};

decltype(ircd::m::user::highlight::match_mxid_local_cs)
ircd::m::user::highlight::match_mxid_local_cs
{
	{ "name",    "ircd.m.user.highlight.match.mxid.local.cs" },
	{ "default", true                                        },
};

decltype(ircd::m::user::highlight::match_mxid_local_cs)
ircd::m::user::highlight::match_mxid_local_ci
{
	{ "name",    "ircd.m.user.highlight.match.mxid.local.ci" },
	{ "default", false                                       },
};

size_t
IRCD_MODULE_EXPORT
ircd::m::user::highlight::count()
const
{
	const m::user::rooms rooms
	{
		user
	};

	size_t ret(0);
	rooms.for_each("join", m::user::rooms::closure_bool{[this, &ret]
	(const m::room &room, const string_view &)
	{
		ret += this->count(room);
		return true;
	}});

	return ret;
}

size_t
IRCD_MODULE_EXPORT
ircd::m::user::highlight::count(const m::room &room)
const
{
	const auto &current
	{
		head_idx(room)
	};

	return count_to(room, current);
}

size_t
IRCD_MODULE_EXPORT
ircd::m::user::highlight::count_to(const m::room &room,
                                   const event::idx &current)
const
{
	event::id::buf last_read_buf;
	const event::id last_read
	{
		receipt::read(last_read_buf, room, user)
	};

	if(!last_read)
		return 0;

	const auto &a
	{
		index(last_read)
	};

	const event::idx_range range
	{
		a, current
	};

	return count_between(room, range);
}

size_t
IRCD_MODULE_EXPORT
ircd::m::user::highlight::count_between(const m::room &room,
                                        const event::idx_range &range)
const
{
	static const event::fetch::opts fopts{[]
	{
		event::fetch::opts ret;
		ret.keys = event::keys::include {"type", "content"};
		ret.query_json_force = true;
		return ret;
	}()};

	m::room::messages it
	{
		room, &fopts
	};

	assert(range.first <= range.second);
	it.seek_idx(range.first);

	if(!it && !exists(room))
		throw m::NOT_FOUND
		{
			"Cannot find room '%s' to count highlights for '%s'",
			string_view{room.room_id},
			string_view{user.user_id}
		};
	else if(!it)
		throw m::NOT_FOUND
		{
			"Event @ idx:%lu or idx:%lu not found in '%s' to count highlights for '%s'",
			range.first,
			range.second,
			string_view{room.room_id},
			string_view{user.user_id},
		};

	size_t ret{0};
	for(++it; it && it.event_idx() < range.second; ++it)
		ret += cached(it.event_idx(), fopts)?
			has(*it):
			has(it.event_idx());

	return ret;
}

bool
IRCD_MODULE_EXPORT
ircd::m::user::highlight::has(const event::idx &event_idx)
const
{
	char typebuf[event::TYPE_MAX_SIZE];
	const string_view type
	{
		m::get(std::nothrow, event_idx, "type", typebuf)
	};

	bool ret{false};
	if(type != "m.room.message")
		return ret;

	m::get(std::nothrow, event_idx, "content", [this, &type, &ret]
	(const json::object &content)
	{
		m::event event;
		json::get<"content"_>(event) = content;
		json::get<"type"_>(event) = type;
		ret = has(event);
	});

	return ret;
}

bool
IRCD_MODULE_EXPORT
ircd::m::user::highlight::has(const event &event)
const
{
	if(json::get<"type"_>(event) != "m.room.message")
		return false;

	const json::object &content
	{
		json::get<"content"_>(event)
	};

	const string_view &formatted_body
	{
		content.get("formatted_body")
	};

	if(match(formatted_body))
		return true;

	const string_view &body
	{
		content.get("body")
	};

	if(match(body))
		return true;

	return false;
}

bool
IRCD_MODULE_EXPORT
ircd::m::user::highlight::match(const string_view &text)
const
{
	// Case insensitive and case-sensitive are exlusive; if both
	// are true only one branch is taken.
	if(match_mxid_local_ci)
	{
		if(ircd::ihas(text, user.user_id.localname()))
			return true;
	}
	else if(match_mxid_local_cs)
	{
		if(ircd::has(text, user.user_id.localname()))
			return true;
	}

	if(match_mxid_full)
		if(ircd::has(text, user.user_id))
			return true;

	return false;
}