mirror of
https://gitlab.com/famedly/conduit.git
synced 2025-01-10 16:24:45 +00:00
94 lines
2.5 KiB
Rust
94 lines
2.5 KiB
Rust
|
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())
|
||
|
|
}
|