2021-06-08 16:10:00 +00:00
|
|
|
use std::sync::Arc;
|
|
|
|
|
|
2020-07-30 16:14:47 +00:00
|
|
|
use super::State;
|
|
|
|
|
use crate::{ConduitResult, Database, Ruma};
|
|
|
|
|
use ruma::api::client::r0::user_directory::search_users;
|
|
|
|
|
|
|
|
|
|
#[cfg(feature = "conduit_bin")]
|
|
|
|
|
use rocket::post;
|
|
|
|
|
|
|
|
|
|
#[cfg_attr(
|
|
|
|
|
feature = "conduit_bin",
|
|
|
|
|
post("/_matrix/client/r0/user_directory/search", data = "<body>")
|
|
|
|
|
)]
|
2021-02-28 11:41:03 +00:00
|
|
|
#[tracing::instrument(skip(db, body))]
|
2020-10-21 19:28:02 +00:00
|
|
|
pub async fn search_users_route(
|
2021-06-08 16:10:00 +00:00
|
|
|
db: State<'_, Arc<Database>>,
|
2020-09-08 15:32:03 +00:00
|
|
|
body: Ruma<search_users::Request<'_>>,
|
2020-07-30 16:14:47 +00:00
|
|
|
) -> ConduitResult<search_users::Response> {
|
2020-08-12 21:32:39 +00:00
|
|
|
let limit = u64::from(body.limit) as usize;
|
2020-07-30 16:14:47 +00:00
|
|
|
|
|
|
|
|
let mut users = db.users.iter().filter_map(|user_id| {
|
|
|
|
|
// Filter out buggy users (they should not exist, but you never know...)
|
|
|
|
|
let user_id = user_id.ok()?;
|
|
|
|
|
if db.users.is_deactivated(&user_id).ok()? {
|
|
|
|
|
return None;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
let user = search_users::User {
|
|
|
|
|
user_id: user_id.clone(),
|
|
|
|
|
display_name: db.users.displayname(&user_id).ok()?,
|
|
|
|
|
avatar_url: db.users.avatar_url(&user_id).ok()?,
|
|
|
|
|
};
|
|
|
|
|
|
|
|
|
|
if !user.user_id.to_string().contains(&body.search_term)
|
|
|
|
|
&& user
|
|
|
|
|
.display_name
|
|
|
|
|
.as_ref()
|
|
|
|
|
.filter(|name| name.contains(&body.search_term))
|
|
|
|
|
.is_none()
|
|
|
|
|
{
|
|
|
|
|
return None;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
Some(user)
|
|
|
|
|
});
|
|
|
|
|
|
|
|
|
|
let results = users.by_ref().take(limit).collect();
|
|
|
|
|
let limited = users.next().is_some();
|
|
|
|
|
|
|
|
|
|
Ok(search_users::Response { results, limited }.into())
|
|
|
|
|
}
|
