mirror of
https://github.com/matrix-construct/construct
synced 2024-09-13 12:08:55 +02:00
383 lines
8.5 KiB
C++
383 lines
8.5 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 room_state_append(data &, json::stack::array &, const m::event &, const m::event::idx &);
|
|
|
|
static bool room_state_phased_member_events(data &, json::stack::array &);
|
|
static bool room_state_phased_events(data &);
|
|
static bool room_state_polylog_events(data &);
|
|
static bool _room_state_polylog(data &);
|
|
static bool room_state_polylog(data &);
|
|
static bool room_invite_state_polylog(data &);
|
|
|
|
static bool room_state_linear_events(data &);
|
|
static bool room_invite_state_linear(data &);
|
|
static bool room_state_linear(data &);
|
|
|
|
extern conf::item<int64_t> state_exposure_depth; //TODO: XXX
|
|
extern const event::keys::include _default_keys;
|
|
extern event::fetch::opts _default_fopts;
|
|
|
|
extern item room_invite_state;
|
|
extern item room_state;
|
|
}
|
|
|
|
ircd::mapi::header
|
|
IRCD_MODULE
|
|
{
|
|
"Client Sync :Room State", []
|
|
{
|
|
ircd::m::sync::_default_fopts.query_json_force = true;
|
|
}
|
|
};
|
|
|
|
decltype(ircd::m::sync::room_state)
|
|
ircd::m::sync::room_state
|
|
{
|
|
"rooms.state",
|
|
room_state_polylog,
|
|
room_state_linear,
|
|
{
|
|
{ "phased", true },
|
|
}
|
|
};
|
|
|
|
decltype(ircd::m::sync::room_invite_state)
|
|
ircd::m::sync::room_invite_state
|
|
{
|
|
"rooms.invite_state",
|
|
room_invite_state_polylog,
|
|
room_invite_state_linear,
|
|
{
|
|
{ "phased", true },
|
|
}
|
|
};
|
|
|
|
decltype(ircd::m::sync::_default_keys)
|
|
ircd::m::sync::_default_keys
|
|
{
|
|
"content",
|
|
"depth",
|
|
"event_id",
|
|
"origin_server_ts",
|
|
"redacts",
|
|
"room_id",
|
|
"sender",
|
|
"state_key",
|
|
"type",
|
|
};
|
|
|
|
decltype(ircd::m::sync::_default_fopts)
|
|
ircd::m::sync::_default_fopts
|
|
{
|
|
_default_keys
|
|
};
|
|
|
|
bool
|
|
ircd::m::sync::room_state_linear(data &data)
|
|
{
|
|
if(data.membership == "invite")
|
|
return false;
|
|
|
|
return room_state_linear_events(data);
|
|
}
|
|
|
|
bool
|
|
ircd::m::sync::room_invite_state_linear(data &data)
|
|
{
|
|
if(data.membership != "invite")
|
|
return false;
|
|
|
|
return room_state_linear_events(data);
|
|
}
|
|
|
|
//TODO: This has to be merged into the timeline conf items
|
|
decltype(ircd::m::sync::state_exposure_depth)
|
|
ircd::m::sync::state_exposure_depth
|
|
{
|
|
{ "name", "ircd.client.sync.rooms.state.exposure.depth" },
|
|
{ "default", 20L },
|
|
};
|
|
|
|
bool
|
|
ircd::m::sync::room_state_linear_events(data &data)
|
|
{
|
|
if(!data.event_idx)
|
|
return false;
|
|
|
|
if(!data.room)
|
|
return false;
|
|
|
|
if(!data.membership)
|
|
return false;
|
|
|
|
assert(data.event);
|
|
if(!json::get<"state_key"_>(*data.event))
|
|
return false;
|
|
|
|
// Figure out whether the event was included in the timeline or whether
|
|
// to include it here in the state, which comes before the timeline.
|
|
// Since linear-sync is already distinct from polylog-sync, the
|
|
// overwhelming majority of state events coming through linear-sync will
|
|
// use the timeline. We make an exception for past state events the server
|
|
// only recently obtained, to hide them from the timeline.
|
|
if(int64_t(state_exposure_depth) > -1)
|
|
if(data.membership != "invite")
|
|
if(json::get<"depth"_>(*data.event) + int64_t(state_exposure_depth) >= data.room_depth)
|
|
return false;
|
|
|
|
json::stack::object rooms
|
|
{
|
|
*data.out, "rooms"
|
|
};
|
|
|
|
json::stack::object membership_
|
|
{
|
|
*data.out, data.membership
|
|
};
|
|
|
|
json::stack::object room_
|
|
{
|
|
*data.out, data.room->room_id
|
|
};
|
|
|
|
const auto &state_member_name
|
|
{
|
|
data.membership == "invite"?
|
|
"invite_state": // "invite_state"_sv:
|
|
"state"
|
|
};
|
|
|
|
json::stack::object state
|
|
{
|
|
*data.out, state_member_name
|
|
};
|
|
|
|
json::stack::array array
|
|
{
|
|
*data.out, "events"
|
|
};
|
|
|
|
// Branch for supplying state to the client after its user's invite
|
|
// is processed. At this point the client has not received prior room
|
|
// state in /sync.
|
|
if(data.membership == "invite" &&
|
|
json::get<"type"_>(*data.event) == "m.room.member" &&
|
|
json::get<"state_key"_>(*data.event) == data.user.user_id)
|
|
{
|
|
const auto append{[&]
|
|
(const m::event &event)
|
|
{
|
|
room_state_append(data, array, event, index(event));
|
|
}};
|
|
|
|
const m::room::state state{*data.room};
|
|
state.get(std::nothrow, "m.room.create", "", append);
|
|
state.get(std::nothrow, "m.room.join_rules", "", append);
|
|
state.get(std::nothrow, "m.room.history_visibility", "", append);
|
|
state.get(std::nothrow, "m.room.avatar", "", append);
|
|
|
|
const auto &sender(json::get<"sender"_>(*data.event));
|
|
state.get(std::nothrow, "m.room.member", sender, append);
|
|
}
|
|
|
|
room_state_append(data, array, *data.event, data.event_idx);
|
|
return true;
|
|
}
|
|
|
|
bool
|
|
ircd::m::sync::room_state_polylog(data &data)
|
|
{
|
|
if(data.membership == "invite")
|
|
return false;
|
|
|
|
return _room_state_polylog(data);
|
|
}
|
|
|
|
bool
|
|
ircd::m::sync::room_invite_state_polylog(data &data)
|
|
{
|
|
if(data.membership != "invite")
|
|
return false;
|
|
|
|
return _room_state_polylog(data);
|
|
}
|
|
|
|
bool
|
|
ircd::m::sync::_room_state_polylog(data &data)
|
|
{
|
|
if(!apropos(data, data.room_head))
|
|
return false;
|
|
|
|
return room_state_polylog_events(data);
|
|
}
|
|
|
|
bool
|
|
ircd::m::sync::room_state_polylog_events(data &data)
|
|
{
|
|
if(data.phased && data.range.first == 0)
|
|
return room_state_phased_events(data);
|
|
|
|
bool ret{false};
|
|
ctx::mutex mutex;
|
|
json::stack::array array
|
|
{
|
|
*data.out, "events"
|
|
};
|
|
|
|
sync::pool.min(64); //TODO: XXX
|
|
ctx::concurrent<event::idx> concurrent
|
|
{
|
|
sync::pool, [&](const event::idx &event_idx)
|
|
{
|
|
const m::event::fetch event
|
|
{
|
|
event_idx, std::nothrow, _default_fopts
|
|
};
|
|
|
|
if(unlikely(!event.valid))
|
|
{
|
|
log::error
|
|
{
|
|
log, "Failed to fetch event idx:%lu in room %s state.",
|
|
event_idx,
|
|
string_view{data.room->room_id},
|
|
};
|
|
|
|
return;
|
|
}
|
|
|
|
const std::lock_guard lock{mutex};
|
|
ret |= room_state_append(data, array, event, event_idx);
|
|
}
|
|
};
|
|
|
|
const room::state state{*data.room};
|
|
state.for_each([&data, &concurrent]
|
|
(const event::idx &event_idx)
|
|
{
|
|
if(!apropos(data, event_idx))
|
|
return;
|
|
|
|
concurrent(event_idx);
|
|
});
|
|
|
|
concurrent.wait();
|
|
return ret;
|
|
}
|
|
|
|
bool
|
|
ircd::m::sync::room_state_phased_events(data &data)
|
|
{
|
|
bool ret{false};
|
|
ctx::mutex mutex;
|
|
json::stack::array array
|
|
{
|
|
*data.out, "events"
|
|
};
|
|
|
|
static const std::pair<string_view, string_view> keys[]
|
|
{
|
|
{ "m.room.create", "" },
|
|
{ "m.room.canonical_alias", "" },
|
|
{ "m.room.name", "" },
|
|
{ "m.room.avatar", "" },
|
|
{ "m.room.aliases", data.user.user_id.host() },
|
|
{ "m.room.member", data.user.user_id },
|
|
};
|
|
|
|
const auto append
|
|
{
|
|
[&data, &array, &ret, &mutex](const m::event &event)
|
|
{
|
|
const auto event_idx(m::index(event));
|
|
const std::lock_guard lock{mutex};
|
|
ret |= room_state_append(data, array, event, event_idx);
|
|
}
|
|
};
|
|
|
|
sync::pool.min(6);
|
|
ctx::concurrent_for_each<const std::pair<string_view, string_view>>
|
|
{
|
|
sync::pool, keys, [&data, &append](const auto &key)
|
|
{
|
|
data.room->get(std::nothrow, key.first, key.second, append);
|
|
}
|
|
};
|
|
|
|
ret |= room_state_phased_member_events(data, array);
|
|
return ret;
|
|
}
|
|
|
|
bool
|
|
ircd::m::sync::room_state_phased_member_events(data &data,
|
|
json::stack::array &array)
|
|
{
|
|
static const auto count{10}, bufsz{48}, limit{10};
|
|
std::array<char[bufsz], count> buf;
|
|
std::array<string_view, count> last;
|
|
size_t i(0), ret(0);
|
|
m::room::messages it
|
|
{
|
|
*data.room
|
|
};
|
|
|
|
const auto already{[&last, &ret]
|
|
(const string_view &sender) -> bool
|
|
{
|
|
return std::any_of(begin(last), begin(last)+ret, [&sender]
|
|
(const auto &last)
|
|
{
|
|
return startswith(last, sender);
|
|
});
|
|
}};
|
|
|
|
for(; it && ret < count && i < limit; --it, ++i)
|
|
{
|
|
const auto &event_idx(it.event_idx());
|
|
m::get(std::nothrow, event_idx, "sender", [&]
|
|
(const auto &sender)
|
|
{
|
|
if(already(sender))
|
|
return;
|
|
|
|
const m::event::fetch event
|
|
{
|
|
event_idx, std::nothrow
|
|
};
|
|
|
|
if(!event.valid)
|
|
return;
|
|
|
|
last.at(ret) = strlcpy(buf.at(ret), sender);
|
|
room_state_append(data, array, event, event_idx);
|
|
++ret;
|
|
});
|
|
};
|
|
|
|
return ret;
|
|
}
|
|
|
|
bool
|
|
ircd::m::sync::room_state_append(data &data,
|
|
json::stack::array &events,
|
|
const m::event &event,
|
|
const m::event::idx &event_idx)
|
|
{
|
|
m::event_append_opts opts;
|
|
opts.event_idx = &event_idx;
|
|
opts.user_id = &data.user.user_id;
|
|
opts.user_room = &data.user_room;
|
|
opts.query_txnid = false;
|
|
opts.room_depth = &data.room_depth;
|
|
return m::append(events, event, opts);
|
|
}
|