feat: search pdus

2020-08-18 12:15:27 +02:00 · 2020-08-18 12:15:27 +02:00 · e457e19088
commit e457e19088
parent 27d35f5ab4
7 changed files with 193 additions and 17 deletions
--- a/src/client_server/search.rs
+++ b/src/client_server/search.rs
@ -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())
+}