0
0
Fork 0
mirror of https://github.com/matrix-construct/construct synced 2024-10-31 10:58:54 +01:00
construct/modules/client/sync/rooms.cc

238 lines
4.1 KiB
C++

// 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.
ircd::mapi::header
IRCD_MODULE
{
"Client Sync :Rooms"
};
namespace ircd::m::sync
{
static bool should_ignore(const data &);
static bool _rooms_polylog_room(data &, const m::room &);
static bool _rooms_polylog(data &, const string_view &membership);
static bool rooms_polylog(data &);
static bool _rooms_linear(data &, const string_view &membership);
static bool rooms_linear(data &);
extern item rooms;
}
decltype(ircd::m::sync::rooms)
ircd::m::sync::rooms
{
"rooms",
rooms_polylog,
rooms_linear
};
bool
ircd::m::sync::rooms_linear(data &data)
{
assert(data.event);
const m::room room
{
json::get<"room_id"_>(*data.event)?
m::room::id{json::get<"room_id"_>(*data.event)}:
m::room::id{}
};
const scope_restore their_room
{
data.room, &room
};
char membuf[32];
const string_view &membership
{
data.room?
room.membership(membuf, data.user):
string_view{}
};
const scope_restore their_membership
{
data.membership, membership
};
if(should_ignore(data))
return false;
return !m::sync::for_each("rooms", [&data]
(item &item)
{
json::stack::checkpoint checkpoint
{
*data.out
};
if(item.linear(data))
return false;
checkpoint.rollback();
return true;
});
}
bool
ircd::m::sync::rooms_polylog(data &data)
{
bool ret{false};
ret |= _rooms_polylog(data, "invite");
ret |= _rooms_polylog(data, "join");
ret |= _rooms_polylog(data, "leave");
ret |= _rooms_polylog(data, "ban");
return ret;
}
bool
ircd::m::sync::_rooms_polylog(data &data,
const string_view &membership)
{
const scope_restore theirs
{
data.membership, membership
};
json::stack::object object
{
*data.out, membership
};
bool ret{false};
data.user_rooms.for_each(membership, [&data, &ret]
(const m::room &room, const string_view &membership_)
{
#if defined(RB_DEBUG)
sync::stats stats
{
data.stats && rooms.stats_debug?
*data.stats:
sync::stats{}
};
if(data.stats)
stats.timer = timer{};
#endif
ret |= _rooms_polylog_room(data, room);
#if defined(RB_DEBUG)
thread_local char tmbuf[32];
if(data.stats && rooms.stats_debug) log::debug
{
log, "polylog %s %s in %s",
loghead(data),
string_view{room.room_id},
ircd::pretty(tmbuf, stats.timer.at<milliseconds>(), true)
};
#endif
});
return ret;
}
bool
ircd::m::sync::_rooms_polylog_room(data &data,
const m::room &room)
{
const scope_restore theirs
{
data.room, &room
};
if(should_ignore(data))
return false;
json::stack::checkpoint checkpoint
{
*data.out
};
json::stack::object object
{
*data.out, room.room_id
};
const event::idx room_head
{
head_idx(std::nothrow, room)
};
const scope_restore their_head
{
data.room_head, room_head
};
bool ret{false};
m::sync::for_each("rooms", [&data, &ret]
(item &item)
{
json::stack::checkpoint checkpoint
{
*data.out
};
json::stack::object object
{
*data.out, item.member_name()
};
if(item.polylog(data))
ret = true;
else
checkpoint.rollback();
return true;
});
if(!ret)
checkpoint.rollback();
return ret;
}
bool
ircd::m::sync::should_ignore(const data &data)
{
if(data.membership != "invite")
return false;
if(!m::user::ignores::enforce("invites"))
return false;
assert(data.room);
const m::room::state state
{
*data.room
};
const m::event::idx &event_idx
{
state.get(std::nothrow, "m.room.member", data.user.user_id)
};
const m::user::ignores ignores
{
data.user.user_id
};
bool ret{false};
m::get(std::nothrow, event_idx, "sender", [&ignores, &ret]
(const m::user::id &sender)
{
ret = ignores.has(sender);
});
return ret;
}