mirror of
https://github.com/matrix-construct/construct
synced 2024-10-30 02:18:58 +01:00
314 lines
5.3 KiB
C++
314 lines
5.3 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.
|
|
|
|
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, int64_t &phase);
|
|
static bool rooms_polylog(data &);
|
|
|
|
static bool _rooms_linear(data &, const string_view &membership);
|
|
static bool rooms_linear(data &);
|
|
|
|
extern item rooms;
|
|
}
|
|
|
|
ircd::mapi::header
|
|
IRCD_MODULE
|
|
{
|
|
"Client Sync :Rooms"
|
|
};
|
|
|
|
decltype(ircd::m::sync::rooms)
|
|
ircd::m::sync::rooms
|
|
{
|
|
"rooms", rooms_polylog, rooms_linear,
|
|
{
|
|
{ "phased", true },
|
|
{ "prefetch", true },
|
|
}
|
|
};
|
|
|
|
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[room::MEMBERSHIP_MAX_SIZE];
|
|
const string_view &membership
|
|
{
|
|
data.room?
|
|
m::membership(membuf, room, data.user):
|
|
string_view{}
|
|
};
|
|
|
|
const scope_restore their_membership
|
|
{
|
|
data.membership, membership
|
|
};
|
|
|
|
if(should_ignore(data))
|
|
return false;
|
|
|
|
const auto room_head
|
|
{
|
|
data.event_idx && room != data.user_room?
|
|
m::head_idx(std::nothrow, room):
|
|
0UL
|
|
};
|
|
|
|
const scope_restore their_head
|
|
{
|
|
data.room_head, room_head
|
|
};
|
|
|
|
const auto room_depth
|
|
{
|
|
data.event_idx && room != data.user_room?
|
|
m::depth(std::nothrow, room):
|
|
-1L
|
|
};
|
|
|
|
const scope_restore their_depth
|
|
{
|
|
data.room_depth, room_depth
|
|
};
|
|
|
|
bool ret{false};
|
|
m::sync::for_each("rooms", [&data, &ret]
|
|
(item &item)
|
|
{
|
|
json::stack::checkpoint checkpoint
|
|
{
|
|
*data.out
|
|
};
|
|
|
|
if(item.linear(data))
|
|
ret = true;
|
|
else
|
|
checkpoint.rollback();
|
|
|
|
return true;
|
|
});
|
|
|
|
return ret;
|
|
}
|
|
|
|
bool
|
|
ircd::m::sync::rooms_polylog(data &data)
|
|
{
|
|
bool ret{false};
|
|
int64_t phase(0);
|
|
|
|
ret |= _rooms_polylog(data, "join", phase);
|
|
if(data.phased && ret)
|
|
return ret;
|
|
|
|
ret |= _rooms_polylog(data, "invite", phase);
|
|
if(data.phased && ret)
|
|
return ret;
|
|
|
|
ret |= _rooms_polylog(data, "leave", phase);
|
|
if(data.phased && ret)
|
|
return ret;
|
|
|
|
ret |= _rooms_polylog(data, "ban", phase);
|
|
if(data.phased && ret)
|
|
return ret;
|
|
|
|
return ret;
|
|
}
|
|
|
|
bool
|
|
ircd::m::sync::_rooms_polylog(data &data,
|
|
const string_view &membership,
|
|
int64_t &phase)
|
|
{
|
|
const scope_restore theirs
|
|
{
|
|
data.membership, membership
|
|
};
|
|
|
|
json::stack::object object
|
|
{
|
|
*data.out, membership
|
|
};
|
|
|
|
bool ret{false};
|
|
const user::rooms::closure_bool closure{[&data, &ret, &phase]
|
|
(const m::room &room, const string_view &membership_)
|
|
{
|
|
if(data.phased)
|
|
{
|
|
if(phase < int64_t(data.range.first) && ret)
|
|
return false;
|
|
|
|
if(int64_t(data.range.first) < 0L)
|
|
--phase;
|
|
|
|
if(phase > int64_t(data.range.first))
|
|
return true;
|
|
}
|
|
|
|
#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 %s in %s",
|
|
loghead(data),
|
|
membership_,
|
|
string_view{room.room_id},
|
|
ircd::pretty(tmbuf, stats.timer.at<milliseconds>(), true)
|
|
};
|
|
#endif
|
|
|
|
return true;
|
|
}};
|
|
|
|
const bool done
|
|
{
|
|
data.user_rooms.for_each(membership, closure)
|
|
};
|
|
|
|
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 auto &[top_event_id, top_depth, top_event_idx]
|
|
{
|
|
m::top(std::nothrow, room)
|
|
};
|
|
|
|
const scope_restore their_head
|
|
{
|
|
data.room_head, top_event_idx
|
|
};
|
|
|
|
const scope_restore their_depth
|
|
{
|
|
data.room_depth, top_depth
|
|
};
|
|
|
|
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;
|
|
data.out->invalidate_checkpoints();
|
|
}
|
|
else checkpoint.committing(false);
|
|
|
|
return true;
|
|
});
|
|
|
|
if(!ret)
|
|
checkpoint.committing(false);
|
|
|
|
return ret;
|
|
}
|
|
|
|
bool
|
|
ircd::m::sync::should_ignore(const data &data)
|
|
{
|
|
if(data.prefetch)
|
|
return false;
|
|
|
|
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;
|
|
}
|