nero/tsuki
1
0
Fork 0
Code Issues 2 Pull requests Projects Releases Packages Wiki Activity Actions

fix: stop /messages at to

Browse source 
Fixes #150
This commit is contained in:
timokoesters 2020-07-26 17:34:12 +02:00
parent 20ab19d828
commit b7df8fe83c
No known key found for this signature in database
GPG key ID: 24DA7517711A2BA4
2 changed files with 46 additions and 38 deletions
Download patch file Download diff file Expand all files Collapse all files

64
src/client_server.rs
View file

@ -2763,7 +2763,7 @@ pub fn get_context_route(
.filter_map(|r| r.ok()) // Remove buggy events
.collect::<Vec<_>>();
let start_token = events_before.last().map_or(Ok(None), |e| {
let start_token = events_before.last().map_or(Ok(None), |(_, e)| {
Ok::<_, Error>(Some(
db.rooms
.get_pdu_count(&e.event_id)?
@ -2774,7 +2774,7 @@ pub fn get_context_route(
let events_before = events_before
.into_iter()
.map(|pdu| pdu.to_room_event())
.map(|(_, pdu)| pdu.to_room_event())
.collect::<Vec<_>>();
let events_after = db
@ -2789,18 +2789,19 @@ pub fn get_context_route(
.filter_map(|r| r.ok()) // Remove buggy events
.collect::<Vec<_>>();
let end_token = events_after.last().map_or(Ok(None), |e| {
Ok::<_, Error>(Some(
db.rooms
.get_pdu_count(&e.event_id)?
.ok_or_else(|| Error::bad_database("Can't find count from event in db."))?
let end_token = if let Some(last_event) = events_after.last() {
Some(
utils::u64_from_bytes(&last_event.0)
.map_err(|_| Error::bad_database("Invalid pdu id in db."))?
.to_string(),
))
})?;
)
} else {
None
};
let events_after = events_after
.into_iter()
.map(|pdu| pdu.to_room_event())
.map(|(_, pdu)| pdu.to_room_event())
.collect::<Vec<_>>();
Ok(get_context::Response {
@ -2839,32 +2840,38 @@ pub fn get_message_events_route(
.clone()
.parse()
.map_err(|_| Error::BadRequest(ErrorKind::InvalidParam, "Invalid `from` value."))?;
let to = body.to.as_ref().map(|t| t.as_bytes());
// Use limit or else 10
let limit = body
.limit
.try_into()
.map_or(Ok::<_, Error>(10_usize), |l: u32| Ok(l as usize))?;
match body.dir {
get_message_events::Direction::Forward => {
let events_after = db
.rooms
.pdus_after(&user_id, &body.room_id, from)
// Use limit or else 10
.take(limit)
.filter_map(|r| r.ok()) // Filter out buggy events
.take_while(|(k, _)| Some(&**k) != to) // Stop at `to`
.collect::<Vec<_>>();
let end_token = events_after.last().map_or(Ok::<_, Error>(None), |e| {
Ok(Some(
db.rooms
.get_pdu_count(&e.event_id)?
.ok_or_else(|| Error::bad_database("Can't find count from event in db."))?
let end_token = if let Some(last_event) = events_after.last() {
Some(
utils::u64_from_bytes(&last_event.0)
.map_err(|_| Error::bad_database("Invalid pdu id in db."))?
.to_string(),
))
})?;
)
} else {
None
};
let events_after = events_after
.into_iter()
.map(|pdu| pdu.to_room_event())
.map(|(_, pdu)| pdu.to_room_event())
.collect::<Vec<_>>();
Ok(get_message_events::Response {
@ -2879,23 +2886,24 @@ pub fn get_message_events_route(
let events_before = db
.rooms
.pdus_until(&user_id, &body.room_id, from)
// Use limit or else 10
.take(limit)
.filter_map(|r| r.ok()) // Filter out buggy events
.take_while(|(k, _)| Some(&**k) != to) // Stop at `to`
.collect::<Vec<_>>();
let start_token = events_before.last().map_or(Ok::<_, Error>(None), |e| {
Ok(Some(
db.rooms
.get_pdu_count(&e.event_id)?
.ok_or_else(|| Error::bad_database("Can't find count from event in db."))?
let start_token = if let Some(last_event) = events_before.last() {
Some(
utils::u64_from_bytes(&last_event.0)
.map_err(|_| Error::bad_database("Invalid pdu id in db."))?
.to_string(),
))
})?;
)
} else {
None
};
let events_before = events_before
.into_iter()
.map(|pdu| pdu.to_room_event())
.map(|(_, pdu)| pdu.to_room_event())
.collect::<Vec<_>>();
Ok(get_message_events::Response {