refactor: put endpoints into modules

2020-07-30 18:14:47 +02:00 · 2020-07-30 18:14:47 +02:00 · c1c62b7eb4
commit c1c62b7eb4
parent 485a24398b
33 changed files with 4153 additions and 3848 deletions
--- a/src/client_server/media.rs
+++ b/src/client_server/media.rs
@ -0,0 +1,107 @@
+use super::State;
+use crate::{database::media::FileMeta, utils, ConduitResult, Database, Error, Ruma};
+use ruma::api::client::{
+    error::ErrorKind,
+    r0::media::{create_content, get_content, get_content_thumbnail, get_media_config},
+};
+
+#[cfg(feature = "conduit_bin")]
+use rocket::{get, post};
+use std::convert::TryInto;
+
+const MXC_LENGTH: usize = 256;
+
+#[cfg_attr(feature = "conduit_bin", get("/_matrix/media/r0/config"))]
+pub fn get_media_config_route(
+    db: State<'_, Database>,
+) -> ConduitResult<get_media_config::Response> {
+    Ok(get_media_config::Response {
+        upload_size: db.globals.max_request_size().into(),
+    }
+    .into())
+}
+
+#[cfg_attr(
+    feature = "conduit_bin",
+    post("/_matrix/media/r0/upload", data = "<body>")
+)]
+pub fn create_content_route(
+    db: State<'_, Database>,
+    body: Ruma<create_content::Request>,
+) -> ConduitResult<create_content::Response> {
+    let mxc = format!(
+        "mxc://{}/{}",
+        db.globals.server_name(),
+        utils::random_string(MXC_LENGTH)
+    );
+    db.media.create(
+        mxc.clone(),
+        body.filename.as_ref(),
+        &body.content_type,
+        &body.file,
+    )?;
+
+    Ok(create_content::Response { content_uri: mxc }.into())
+}
+
+#[cfg_attr(
+    feature = "conduit_bin",
+    get(
+        "/_matrix/media/r0/download/<_server_name>/<_media_id>",
+        data = "<body>"
+    )
+)]
+pub fn get_content_route(
+    db: State<'_, Database>,
+    body: Ruma<get_content::Request>,
+    _server_name: String,
+    _media_id: String,
+) -> ConduitResult<get_content::Response> {
+    if let Some(FileMeta {
+        filename,
+        content_type,
+        file,
+    }) = db
+        .media
+        .get(format!("mxc://{}/{}", body.server_name, body.media_id))?
+    {
+        Ok(get_content::Response {
+            file,
+            content_type,
+            content_disposition: filename.unwrap_or_default(), // TODO: Spec says this should be optional
+        }
+        .into())
+    } else {
+        Err(Error::BadRequest(ErrorKind::NotFound, "Media not found."))
+    }
+}
+
+#[cfg_attr(
+    feature = "conduit_bin",
+    get(
+        "/_matrix/media/r0/thumbnail/<_server_name>/<_media_id>",
+        data = "<body>"
+    )
+)]
+pub fn get_content_thumbnail_route(
+    db: State<'_, Database>,
+    body: Ruma<get_content_thumbnail::Request>,
+    _server_name: String,
+    _media_id: String,
+) -> ConduitResult<get_content_thumbnail::Response> {
+    if let Some(FileMeta {
+        content_type, file, ..
+    }) = db.media.get_thumbnail(
+        format!("mxc://{}/{}", body.server_name, body.media_id),
+        body.width
+            .try_into()
+            .map_err(|_| Error::BadRequest(ErrorKind::InvalidParam, "Width is invalid."))?,
+        body.height
+            .try_into()
+            .map_err(|_| Error::BadRequest(ErrorKind::InvalidParam, "Width is invalid."))?,
+    )? {
+        Ok(get_content_thumbnail::Response { file, content_type }.into())
+    } else {
+        Err(Error::BadRequest(ErrorKind::NotFound, "Media not found."))
+    }
+}