timokoesters
5 years ago
committed by
Timo
7 changed files with 193 additions and 17 deletions
@ -0,0 +1,93 @@
@@ -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