timokoesters
5 years ago
committed by
Timo
7 changed files with 193 additions and 17 deletions
@ -0,0 +1,93 @@ |
|||||||
|
use super::State; |
||||||
|
use crate::{ConduitResult, Database, Error, Ruma}; |
||||||
|
use js_int::uint; |
||||||
|
use ruma::api::client::{error::ErrorKind, r0::search::search_events}; |
||||||
|
|
||||||
|
#[cfg(feature = "conduit_bin")] |
||||||
|
use rocket::post; |
||||||
|
use search_events::{ResultCategories, ResultRoomEvents, SearchResult}; |
||||||
|
use std::collections::BTreeMap; |
||||||
|
|
||||||
|
#[cfg_attr(
|
||||||
|
feature = "conduit_bin", |
||||||
|
post("/_matrix/client/r0/search", data = "<body>") |
||||||
|
)] |
||||||
|
pub fn search_events_route( |
||||||
|
db: State<'_, Database>, |
||||||
|
body: Ruma<search_events::Request>, |
||||||
|
) -> ConduitResult<search_events::Response> { |
||||||
|
let sender_id = body.sender_id.as_ref().expect("user is authenticated"); |
||||||
|
|
||||||
|
let search_criteria = body.search_categories.room_events.as_ref().unwrap(); |
||||||
|
let filter = search_criteria |
||||||
|
.filter |
||||||
|
.as_ref() |
||||||
|
.unwrap(); |
||||||
|
|
||||||
|
let room_id = filter.rooms |
||||||
|
.as_ref() |
||||||
|
.unwrap() |
||||||
|
.first() |
||||||
|
.unwrap(); |
||||||
|
|
||||||
|
let limit = filter.limit.map_or(10, |l| u64::from(l) as usize); |
||||||
|
|
||||||
|
if !db.rooms.is_joined(sender_id, &room_id)? { |
||||||
|
return Err(Error::BadRequest( |
||||||
|
ErrorKind::Forbidden, |
||||||
|
"You don't have permission to view this room.", |
||||||
|
)); |
||||||
|
} |
||||||
|
|
||||||
|
let skip = match body.next_batch.as_ref().map(|s| s.parse()) { |
||||||
|
Some(Ok(s)) => s, |
||||||
|
Some(Err(_)) => { |
||||||
|
return Err(Error::BadRequest( |
||||||
|
ErrorKind::InvalidParam, |
||||||
|
"Invalid next_batch token.", |
||||||
|
)) |
||||||
|
} |
||||||
|
None => 0, // Default to the start
|
||||||
|
}; |
||||||
|
|
||||||
|
let search = db |
||||||
|
.rooms |
||||||
|
.search_pdus(&room_id, &search_criteria.search_term)?; |
||||||
|
|
||||||
|
let results = search |
||||||
|
.0 |
||||||
|
.map(|result| { |
||||||
|
Ok::<_, Error>(SearchResult { |
||||||
|
context: None, |
||||||
|
rank: None, |
||||||
|
result: dbg!(db |
||||||
|
.rooms |
||||||
|
.get_pdu_from_id(dbg!(&result))? |
||||||
|
.map(|pdu| pdu.to_room_event())), |
||||||
|
}) |
||||||
|
}) |
||||||
|
.filter_map(|r| r.ok()) |
||||||
|
.skip(skip) |
||||||
|
.take(limit) |
||||||
|
.collect::<Vec<_>>(); |
||||||
|
|
||||||
|
let next_batch = if results.len() < limit as usize { |
||||||
|
None |
||||||
|
} else { |
||||||
|
Some((skip + limit).to_string()) |
||||||
|
}; |
||||||
|
|
||||||
|
Ok(search_events::Response { |
||||||
|
search_categories: ResultCategories { |
||||||
|
room_events: Some(ResultRoomEvents { |
||||||
|
count: uint!(0), // TODO
|
||||||
|
groups: BTreeMap::new(), // TODO
|
||||||
|
next_batch, |
||||||
|
results, |
||||||
|
state: BTreeMap::new(), // TODO
|
||||||
|
highlights: search.1, |
||||||
|
}), |
||||||
|
}, |
||||||
|
} |
||||||
|
.into()) |
||||||
|
} |
||||||
Loading…
Reference in new issue