construct/matrix/user_rooms.cc

119 lines
2.2 KiB
C++

// Matrix Construct
//
// Copyright (C) Matrix Construct Developers, Authors & Contributors
// Copyright (C) 2016-2019 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.
bool
ircd::m::user::rooms::prefetch()
const
{
const m::events::state::tuple query
{
user,
"m.room.member"_sv,
m::room::id{},
-1L,
0UL
};
bool ret{false};
return m::events::state::for_each(query, [this, &ret]
(const auto &tuple)
{
const auto &[state_key, type, room_id, depth, event_idx]
{
tuple
};
assert(type == "m.room.member");
assert(state_key == user);
ret |= m::membership_prefetch(event_idx);
return true;
});
return ret;
}
size_t
ircd::m::user::rooms::count()
const
{
size_t ret{0};
for_each([&ret]
(const m::room &, const string_view &membership) noexcept
{
++ret;
});
return ret;
}
size_t
ircd::m::user::rooms::count(const string_view &membership)
const
{
size_t ret{0};
for_each(membership, [&ret]
(const m::room &, const string_view &membership) noexcept
{
++ret;
});
return ret;
}
bool
ircd::m::user::rooms::for_each(const closure_bool &closure)
const
{
return for_each(string_view{}, closure);
}
bool
ircd::m::user::rooms::for_each(const string_view &membership_,
const closure_bool &closure)
const
{
const m::events::state::tuple query
{
user,
"m.room.member"_sv,
m::room::id{},
-1L,
0UL
};
m::room::id::buf room_id_;
return m::events::state::for_each(query, [this, &membership_, &closure, &room_id_]
(const auto &tuple)
{
const auto &[state_key, type, room_id, depth, event_idx]
{
tuple
};
assert(type == "m.room.member");
assert(state_key == user);
if(room_id_ != room_id)
room_id_ = room_id;
else
return true;
char buf[m::room::MEMBERSHIP_MAX_SIZE];
const string_view &membership
{
m::membership(buf, event_idx)
};
if(likely(!membership_ || membership == membership_))
return closure(m::room::id(room_id), membership);
return true;
});
}