|
|
|
|
@ -3,9 +3,9 @@ use std::sync::Arc;
|
|
|
|
|
|
|
|
|
|
pub use data::Data;
|
|
|
|
|
use ruma::{
|
|
|
|
|
api::client::relations::get_relating_events,
|
|
|
|
|
api::{client::relations::get_relating_events, Direction},
|
|
|
|
|
events::{relation::RelationType, TimelineEventType},
|
|
|
|
|
EventId, RoomId, UserId,
|
|
|
|
|
EventId, RoomId, UInt, UserId,
|
|
|
|
|
};
|
|
|
|
|
use serde::Deserialize;
|
|
|
|
|
|
|
|
|
|
@ -48,37 +48,57 @@ impl Service {
|
|
|
|
|
target: &EventId,
|
|
|
|
|
filter_event_type: Option<TimelineEventType>,
|
|
|
|
|
filter_rel_type: Option<RelationType>,
|
|
|
|
|
from: PduCount,
|
|
|
|
|
to: Option<PduCount>,
|
|
|
|
|
limit: usize,
|
|
|
|
|
from: Option<String>,
|
|
|
|
|
to: Option<String>,
|
|
|
|
|
limit: Option<UInt>,
|
|
|
|
|
recurse: bool,
|
|
|
|
|
dir: &Direction,
|
|
|
|
|
) -> Result<get_relating_events::v1::Response> {
|
|
|
|
|
let from = match from {
|
|
|
|
|
Some(from) => PduCount::try_from_string(&from)?,
|
|
|
|
|
None => match dir {
|
|
|
|
|
Direction::Forward => PduCount::min(),
|
|
|
|
|
Direction::Backward => PduCount::max(),
|
|
|
|
|
},
|
|
|
|
|
};
|
|
|
|
|
|
|
|
|
|
let to = to.as_ref().and_then(|t| PduCount::try_from_string(t).ok());
|
|
|
|
|
|
|
|
|
|
// Use limit or else 10, with maximum 100
|
|
|
|
|
let limit = limit
|
|
|
|
|
.and_then(|u| u32::try_from(u).ok())
|
|
|
|
|
.map_or(10_usize, |u| u as usize)
|
|
|
|
|
.min(100);
|
|
|
|
|
|
|
|
|
|
let next_token;
|
|
|
|
|
|
|
|
|
|
//TODO: Fix ruma: match body.dir {
|
|
|
|
|
match ruma::api::Direction::Backward {
|
|
|
|
|
ruma::api::Direction::Forward => {
|
|
|
|
|
let events_after: Vec<_> = services()
|
|
|
|
|
.rooms
|
|
|
|
|
.pdu_metadata
|
|
|
|
|
.relations_until(sender_user, room_id, target, from)? // TODO: should be relations_after
|
|
|
|
|
.filter(|r| {
|
|
|
|
|
r.as_ref().map_or(true, |(_, pdu)| {
|
|
|
|
|
filter_event_type.as_ref().map_or(true, |t| &pdu.kind == t)
|
|
|
|
|
&& if let Ok(content) =
|
|
|
|
|
serde_json::from_str::<ExtractRelatesToEventId>(
|
|
|
|
|
pdu.content.get(),
|
|
|
|
|
)
|
|
|
|
|
{
|
|
|
|
|
filter_rel_type
|
|
|
|
|
.as_ref()
|
|
|
|
|
.map_or(true, |r| &content.relates_to.rel_type == r)
|
|
|
|
|
} else {
|
|
|
|
|
false
|
|
|
|
|
}
|
|
|
|
|
})
|
|
|
|
|
// Spec (v1.10) recommends depth of at least 3
|
|
|
|
|
let depth: u8 = if recurse { 3 } else { 1 };
|
|
|
|
|
|
|
|
|
|
match dir {
|
|
|
|
|
Direction::Forward => {
|
|
|
|
|
let relations_until = &services().rooms.pdu_metadata.relations_until(
|
|
|
|
|
sender_user,
|
|
|
|
|
room_id,
|
|
|
|
|
target,
|
|
|
|
|
from,
|
|
|
|
|
depth,
|
|
|
|
|
)?;
|
|
|
|
|
let events_after: Vec<_> = relations_until // TODO: should be relations_after
|
|
|
|
|
.iter()
|
|
|
|
|
.filter(|(_, pdu)| {
|
|
|
|
|
filter_event_type.as_ref().map_or(true, |t| &pdu.kind == t)
|
|
|
|
|
&& if let Ok(content) =
|
|
|
|
|
serde_json::from_str::<ExtractRelatesToEventId>(pdu.content.get())
|
|
|
|
|
{
|
|
|
|
|
filter_rel_type
|
|
|
|
|
.as_ref()
|
|
|
|
|
.map_or(true, |r| &content.relates_to.rel_type == r)
|
|
|
|
|
} else {
|
|
|
|
|
false
|
|
|
|
|
}
|
|
|
|
|
})
|
|
|
|
|
.take(limit)
|
|
|
|
|
.filter_map(|r| r.ok()) // Filter out buggy events
|
|
|
|
|
.filter(|(_, pdu)| {
|
|
|
|
|
services()
|
|
|
|
|
.rooms
|
|
|
|
|
@ -86,7 +106,7 @@ impl Service {
|
|
|
|
|
.user_can_see_event(sender_user, room_id, &pdu.event_id)
|
|
|
|
|
.unwrap_or(false)
|
|
|
|
|
})
|
|
|
|
|
.take_while(|&(k, _)| Some(k) != to) // Stop at `to`
|
|
|
|
|
.take_while(|(k, _)| Some(k) != to.as_ref()) // Stop at `to`
|
|
|
|
|
.collect();
|
|
|
|
|
|
|
|
|
|
next_token = events_after.last().map(|(count, _)| count).copied();
|
|
|
|
|
@ -101,31 +121,32 @@ impl Service {
|
|
|
|
|
chunk: events_after,
|
|
|
|
|
next_batch: next_token.map(|t| t.stringify()),
|
|
|
|
|
prev_batch: Some(from.stringify()),
|
|
|
|
|
recursion_depth: if recurse { Some(depth.into()) } else { None },
|
|
|
|
|
})
|
|
|
|
|
}
|
|
|
|
|
ruma::api::Direction::Backward => {
|
|
|
|
|
let events_before: Vec<_> = services()
|
|
|
|
|
.rooms
|
|
|
|
|
.pdu_metadata
|
|
|
|
|
.relations_until(sender_user, room_id, target, from)?
|
|
|
|
|
.filter(|r| {
|
|
|
|
|
r.as_ref().map_or(true, |(_, pdu)| {
|
|
|
|
|
filter_event_type.as_ref().map_or(true, |t| &pdu.kind == t)
|
|
|
|
|
&& if let Ok(content) =
|
|
|
|
|
serde_json::from_str::<ExtractRelatesToEventId>(
|
|
|
|
|
pdu.content.get(),
|
|
|
|
|
)
|
|
|
|
|
{
|
|
|
|
|
filter_rel_type
|
|
|
|
|
.as_ref()
|
|
|
|
|
.map_or(true, |r| &content.relates_to.rel_type == r)
|
|
|
|
|
} else {
|
|
|
|
|
false
|
|
|
|
|
}
|
|
|
|
|
})
|
|
|
|
|
Direction::Backward => {
|
|
|
|
|
let relations_until = &services().rooms.pdu_metadata.relations_until(
|
|
|
|
|
sender_user,
|
|
|
|
|
room_id,
|
|
|
|
|
target,
|
|
|
|
|
from,
|
|
|
|
|
depth,
|
|
|
|
|
)?;
|
|
|
|
|
let events_before: Vec<_> = relations_until
|
|
|
|
|
.iter()
|
|
|
|
|
.filter(|(_, pdu)| {
|
|
|
|
|
filter_event_type.as_ref().map_or(true, |t| &pdu.kind == t)
|
|
|
|
|
&& if let Ok(content) =
|
|
|
|
|
serde_json::from_str::<ExtractRelatesToEventId>(pdu.content.get())
|
|
|
|
|
{
|
|
|
|
|
filter_rel_type
|
|
|
|
|
.as_ref()
|
|
|
|
|
.map_or(true, |r| &content.relates_to.rel_type == r)
|
|
|
|
|
} else {
|
|
|
|
|
false
|
|
|
|
|
}
|
|
|
|
|
})
|
|
|
|
|
.take(limit)
|
|
|
|
|
.filter_map(|r| r.ok()) // Filter out buggy events
|
|
|
|
|
.filter(|(_, pdu)| {
|
|
|
|
|
services()
|
|
|
|
|
.rooms
|
|
|
|
|
@ -133,7 +154,7 @@ impl Service {
|
|
|
|
|
.user_can_see_event(sender_user, room_id, &pdu.event_id)
|
|
|
|
|
.unwrap_or(false)
|
|
|
|
|
})
|
|
|
|
|
.take_while(|&(k, _)| Some(k) != to) // Stop at `to`
|
|
|
|
|
.take_while(|&(k, _)| Some(k) != to.as_ref()) // Stop at `to`
|
|
|
|
|
.collect();
|
|
|
|
|
|
|
|
|
|
next_token = events_before.last().map(|(count, _)| count).copied();
|
|
|
|
|
@ -147,6 +168,7 @@ impl Service {
|
|
|
|
|
chunk: events_before,
|
|
|
|
|
next_batch: next_token.map(|t| t.stringify()),
|
|
|
|
|
prev_batch: Some(from.stringify()),
|
|
|
|
|
recursion_depth: if recurse { Some(depth.into()) } else { None },
|
|
|
|
|
})
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
@ -158,14 +180,44 @@ impl Service {
|
|
|
|
|
room_id: &'a RoomId,
|
|
|
|
|
target: &'a EventId,
|
|
|
|
|
until: PduCount,
|
|
|
|
|
) -> Result<impl Iterator<Item = Result<(PduCount, PduEvent)>> + 'a> {
|
|
|
|
|
max_depth: u8,
|
|
|
|
|
) -> Result<Vec<(PduCount, PduEvent)>> {
|
|
|
|
|
let room_id = services().rooms.short.get_or_create_shortroomid(room_id)?;
|
|
|
|
|
let target = match services().rooms.timeline.get_pdu_count(target)? {
|
|
|
|
|
Some(PduCount::Normal(c)) => c,
|
|
|
|
|
// TODO: Support backfilled relations
|
|
|
|
|
_ => 0, // This will result in an empty iterator
|
|
|
|
|
};
|
|
|
|
|
self.db.relations_until(user_id, room_id, target, until)
|
|
|
|
|
|
|
|
|
|
self.db
|
|
|
|
|
.relations_until(user_id, room_id, target, until)
|
|
|
|
|
.map(|mut relations| {
|
|
|
|
|
let mut pdus: Vec<_> = (*relations).into_iter().filter_map(Result::ok).collect();
|
|
|
|
|
let mut stack: Vec<_> =
|
|
|
|
|
pdus.clone().iter().map(|pdu| (pdu.to_owned(), 1)).collect();
|
|
|
|
|
|
|
|
|
|
while let Some(stack_pdu) = stack.pop() {
|
|
|
|
|
let target = match stack_pdu.0 .0 {
|
|
|
|
|
PduCount::Normal(c) => c,
|
|
|
|
|
// TODO: Support backfilled relations
|
|
|
|
|
PduCount::Backfilled(_) => 0, // This will result in an empty iterator
|
|
|
|
|
};
|
|
|
|
|
|
|
|
|
|
if let Ok(relations) = self.db.relations_until(user_id, room_id, target, until)
|
|
|
|
|
{
|
|
|
|
|
for relation in relations.flatten() {
|
|
|
|
|
if stack_pdu.1 < max_depth {
|
|
|
|
|
stack.push((relation.clone(), stack_pdu.1 + 1));
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
pdus.push(relation);
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
pdus.sort_by(|a, b| a.0.cmp(&b.0));
|
|
|
|
|
pdus
|
|
|
|
|
})
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
#[tracing::instrument(skip(self, room_id, event_ids))]
|
|
|
|
|
|
Matthias Ahouansou
1 month ago