conduit/src/client_server/media.rs

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 = 32;

#[cfg_attr(feature = "conduit_bin", get("/_matrix/media/r0/config"))]
#[tracing::instrument(skip(db))]
pub async 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>")
)]
#[tracing::instrument(skip(db, body))]
pub async 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_deref(),
        &body.content_type.as_deref(),
        &body.file,
    )?;

    db.flush().await?;

    Ok(create_content::Response {
        content_uri: mxc.try_into().expect("Invalid mxc:// URI"),
        blurhash: None,
    }
    .into())
}

#[cfg_attr(
    feature = "conduit_bin",
    get("/_matrix/media/r0/download/<_>/<_>", data = "<body>")
)]
#[tracing::instrument(skip(db, body))]
pub async fn get_content_route(
    db: State<'_, Database>,
    body: Ruma<get_content::Request<'_>>,
) -> ConduitResult<get_content::Response> {
    let mxc = format!("mxc://{}/{}", body.server_name, body.media_id);

    if let Some(FileMeta {
        filename,
        content_type,
        file,
    }) = db.media.get(&mxc)?
    {
        Ok(get_content::Response {
            file,
            content_type,
            content_disposition: filename,
        }
        .into())
    } else if &*body.server_name != db.globals.server_name() && body.allow_remote {
        let get_content_response = db
            .sending
            .send_federation_request(
                &db.globals,
                &body.server_name,
                get_content::Request {
                    allow_remote: false,
                    server_name: &body.server_name,
                    media_id: &body.media_id,
                },
            )
            .await?;

        db.media.create(
            mxc,
            &get_content_response.content_disposition.as_deref(),
            &get_content_response.content_type.as_deref(),
            &get_content_response.file,
        )?;

        Ok(get_content_response.into())
    } else {
        Err(Error::BadRequest(ErrorKind::NotFound, "Media not found."))
    }
}

#[cfg_attr(
    feature = "conduit_bin",
    get("/_matrix/media/r0/thumbnail/<_>/<_>", data = "<body>")
)]
#[tracing::instrument(skip(db, body))]
pub async fn get_content_thumbnail_route(
    db: State<'_, Database>,
    body: Ruma<get_content_thumbnail::Request<'_>>,
) -> ConduitResult<get_content_thumbnail::Response> {
    let mxc = format!("mxc://{}/{}", body.server_name, body.media_id);

    if let Some(FileMeta {
        content_type, file, ..
    }) = db.media.get_thumbnail(
        mxc.clone(),
        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 if &*body.server_name != db.globals.server_name() && body.allow_remote {
        let get_thumbnail_response = db
            .sending
            .send_federation_request(
                &db.globals,
                &body.server_name,
                get_content_thumbnail::Request {
                    allow_remote: false,
                    height: body.height,
                    width: body.width,
                    method: body.method.clone(),
                    server_name: &body.server_name,
                    media_id: &body.media_id,
                },
            )
            .await?;

        db.media.upload_thumbnail(
            mxc,
            &None,
            &get_thumbnail_response.content_type,
            body.width.try_into().expect("all UInts are valid u32s"),
            body.height.try_into().expect("all UInts are valid u32s"),
            &get_thumbnail_response.file,
        )?;

        Ok(get_thumbnail_response.into())
    } else {
        Err(Error::BadRequest(ErrorKind::NotFound, "Media not found."))
    }
}
refactor: put endpoints into modules 2020-07-30 16:14:47 +00:00			`use super::State;`
feat: improved state store 2020-12-19 15:00:11 +00:00			`use crate::{database::media::FileMeta, utils, ConduitResult, Database, Error, Ruma};`
refactor: put endpoints into modules 2020-07-30 16:14:47 +00:00			`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;`

Reduce media ID length from 256 to 32 Most common filesystems limit paths to 255 bytes. This change brings down the media ID length to be similar to Synapse servers (25), and makes it possible for clients to download media with the ID included in the filename. 2020-09-25 18:18:36 +00:00			`const MXC_LENGTH: usize = 32;`
refactor: put endpoints into modules 2020-07-30 16:14:47 +00:00
			`#[cfg_attr(feature = "conduit_bin", get("/_matrix/media/r0/config"))]`
feat: opentelemetry/jaeger support 2021-02-28 11:41:03 +00:00			`#[tracing::instrument(skip(db))]`
improvement: flush after every request that manipulates the db 2020-10-21 19:28:02 +00:00			`pub async fn get_media_config_route(`
refactor: put endpoints into modules 2020-07-30 16:14:47 +00:00			`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>")`
			`)]`
feat: opentelemetry/jaeger support 2021-02-28 11:41:03 +00:00			`#[tracing::instrument(skip(db, body))]`
improvement: flush after every request that manipulates the db 2020-10-21 19:28:02 +00:00			`pub async fn create_content_route(`
refactor: put endpoints into modules 2020-07-30 16:14:47 +00:00			`db: State<'_, Database>,`
Update ruma version 2020-09-08 15:32:03 +00:00			`body: Ruma<create_content::Request<'_>>,`
refactor: put endpoints into modules 2020-07-30 16:14:47 +00:00			`) -> ConduitResult<create_content::Response> {`
			`let mxc = format!(`
			`"mxc://{}/{}",`
			`db.globals.server_name(),`
			`utils::random_string(MXC_LENGTH)`
			`);`
			`db.media.create(`
			`mxc.clone(),`
Update dependencies 2020-08-25 09:49:51 +00:00			`&body.filename.as_deref(),`
Address some review issues fmt, errors, comments 2020-11-15 21:48:43 +00:00			`&body.content_type.as_deref(),`
refactor: put endpoints into modules 2020-07-30 16:14:47 +00:00			`&body.file,`
			`)?;`

improvement: flush after every request that manipulates the db 2020-10-21 19:28:02 +00:00			`db.flush().await?;`

feat: improved state store 2020-12-19 15:00:11 +00:00			`Ok(create_content::Response {`
Upgrade ruma … and refactor push rule code along the way. 2021-04-05 19:25:10 +00:00			`content_uri: mxc.try_into().expect("Invalid mxc:// URI"),`
feat: improved state store 2020-12-19 15:00:11 +00:00			`blurhash: None,`
			`}`
			`.into())`
refactor: put endpoints into modules 2020-07-30 16:14:47 +00:00			`}`

			`#[cfg_attr(`
			`feature = "conduit_bin",`
fix: send to all servers and fix media store 2020-09-15 06:16:20 +00:00			`get("/_matrix/media/r0/download/<_>/<_>", data = "<body>")`
refactor: put endpoints into modules 2020-07-30 16:14:47 +00:00			`)]`
feat: opentelemetry/jaeger support 2021-02-28 11:41:03 +00:00			`#[tracing::instrument(skip(db, body))]`
feat: download media and thumbnails over federation 2020-09-14 12:20:38 +00:00			`pub async fn get_content_route(`
refactor: put endpoints into modules 2020-07-30 16:14:47 +00:00			`db: State<'_, Database>,`
Update ruma version 2020-09-08 15:32:03 +00:00			`body: Ruma<get_content::Request<'_>>,`
refactor: put endpoints into modules 2020-07-30 16:14:47 +00:00			`) -> ConduitResult<get_content::Response> {`
fix: send to all servers and fix media store 2020-09-15 06:16:20 +00:00			`let mxc = format!("mxc://{}/{}", body.server_name, body.media_id);`
feat: download media and thumbnails over federation 2020-09-14 12:20:38 +00:00
refactor: put endpoints into modules 2020-07-30 16:14:47 +00:00			`if let Some(FileMeta {`
			`filename,`
			`content_type,`
			`file,`
Update dependencies 2020-08-25 09:49:51 +00:00			`}) = db.media.get(&mxc)?`
refactor: put endpoints into modules 2020-07-30 16:14:47 +00:00			`{`
			`Ok(get_content::Response {`
			`file,`
Have Media db return optional content_type, conversion fixes 2020-11-18 13:36:12 +00:00			`content_type,`
Update ruma to latest, fix unstable origin feature in ruma 2020-10-27 23:10:09 +00:00			`content_disposition: filename,`
refactor: put endpoints into modules 2020-07-30 16:14:47 +00:00			`}`
			`.into())`
fix: send to all servers and fix media store 2020-09-15 06:16:20 +00:00			`} else if &*body.server_name != db.globals.server_name() && body.allow_remote {`
feat: improved state store 2020-12-19 15:00:11 +00:00			`let get_content_response = db`
			`.sending`
			`.send_federation_request(`
			`&db.globals,`
Convert uses of Box<ServerName> to a ref 2021-01-14 19:39:56 +00:00			`&body.server_name,`
feat: improved state store 2020-12-19 15:00:11 +00:00			`get_content::Request {`
			`allow_remote: false,`
			`server_name: &body.server_name,`
			`media_id: &body.media_id,`
			`},`
			`)`
			`.await?;`
feat: download media and thumbnails over federation 2020-09-14 12:20:38 +00:00
			`db.media.create(`
			`mxc,`
Update ruma to latest, fix unstable origin feature in ruma 2020-10-27 23:10:09 +00:00			`&get_content_response.content_disposition.as_deref(),`
Fix review issues, Remove EventHash's in prev/auth_events in StateEvent The latest state-res crate uses ruma's PduRoomV3 PDU's which don't have tuples of (EventId, EventHashs) like previous versions did (this was left from rebasing onto master). The Media DB now takes an optional content_type like the updated ruma structs. 2020-11-11 19:30:12 +00:00			`&get_content_response.content_type.as_deref(),`
feat: download media and thumbnails over federation 2020-09-14 12:20:38 +00:00			`&get_content_response.file,`
			`)?;`

			`Ok(get_content_response.into())`
refactor: put endpoints into modules 2020-07-30 16:14:47 +00:00			`} else {`
			`Err(Error::BadRequest(ErrorKind::NotFound, "Media not found."))`
			`}`
			`}`

			`#[cfg_attr(`
			`feature = "conduit_bin",`
fix: send to all servers and fix media store 2020-09-15 06:16:20 +00:00			`get("/_matrix/media/r0/thumbnail/<_>/<_>", data = "<body>")`
refactor: put endpoints into modules 2020-07-30 16:14:47 +00:00			`)]`
feat: opentelemetry/jaeger support 2021-02-28 11:41:03 +00:00			`#[tracing::instrument(skip(db, body))]`
feat: download media and thumbnails over federation 2020-09-14 12:20:38 +00:00			`pub async fn get_content_thumbnail_route(`
refactor: put endpoints into modules 2020-07-30 16:14:47 +00:00			`db: State<'_, Database>,`
Update ruma version 2020-09-08 15:32:03 +00:00			`body: Ruma<get_content_thumbnail::Request<'_>>,`
refactor: put endpoints into modules 2020-07-30 16:14:47 +00:00			`) -> ConduitResult<get_content_thumbnail::Response> {`
fix: send to all servers and fix media store 2020-09-15 06:16:20 +00:00			`let mxc = format!("mxc://{}/{}", body.server_name, body.media_id);`

refactor: put endpoints into modules 2020-07-30 16:14:47 +00:00			`if let Some(FileMeta {`
			`content_type, file, ..`
			`}) = db.media.get_thumbnail(`
fix: send to all servers and fix media store 2020-09-15 06:16:20 +00:00			`mxc.clone(),`
refactor: put endpoints into modules 2020-07-30 16:14:47 +00:00			`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."))?,`
			`)? {`
Have Media db return optional content_type, conversion fixes 2020-11-18 13:36:12 +00:00			`Ok(get_content_thumbnail::Response { file, content_type }.into())`
fix: send to all servers and fix media store 2020-09-15 06:16:20 +00:00			`} else if &*body.server_name != db.globals.server_name() && body.allow_remote {`
feat: improved state store 2020-12-19 15:00:11 +00:00			`let get_thumbnail_response = db`
			`.sending`
			`.send_federation_request(`
			`&db.globals,`
Convert uses of Box<ServerName> to a ref 2021-01-14 19:39:56 +00:00			`&body.server_name,`
feat: improved state store 2020-12-19 15:00:11 +00:00			`get_content_thumbnail::Request {`
			`allow_remote: false,`
			`height: body.height,`
			`width: body.width,`
Update state-res, use the new Event trait This also bumps ruma to latest and removes js_int infavor of the ruma re-export 2020-12-31 13:40:49 +00:00			`method: body.method.clone(),`
feat: improved state store 2020-12-19 15:00:11 +00:00			`server_name: &body.server_name,`
			`media_id: &body.media_id,`
			`},`
			`)`
			`.await?;`
feat: download media and thumbnails over federation 2020-09-14 12:20:38 +00:00
			`db.media.upload_thumbnail(`
			`mxc,`
			`&None,`
Fix review issues, Remove EventHash's in prev/auth_events in StateEvent The latest state-res crate uses ruma's PduRoomV3 PDU's which don't have tuples of (EventId, EventHashs) like previous versions did (this was left from rebasing onto master). The Media DB now takes an optional content_type like the updated ruma structs. 2020-11-11 19:30:12 +00:00			`&get_thumbnail_response.content_type,`
feat: download media and thumbnails over federation 2020-09-14 12:20:38 +00:00			`body.width.try_into().expect("all UInts are valid u32s"),`
			`body.height.try_into().expect("all UInts are valid u32s"),`
			`&get_thumbnail_response.file,`
			`)?;`

			`Ok(get_thumbnail_response.into())`
refactor: put endpoints into modules 2020-07-30 16:14:47 +00:00			`} else {`
			`Err(Error::BadRequest(ErrorKind::NotFound, "Media not found."))`
			`}`
			`}`