dolysis/conduwuit
3
0
Fork 0
mirror of https://github.com/girlbossceo/conduwuit.git synced 2024-12-01 07:16:42 +00:00
Code Issues Packages Projects Releases Wiki Activity

Reduce code duplication in media download route handlers

Browse source
This commit is contained in:
Andrej Kacian 2022-01-27 16:12:39 +01:00
parent 52873c88b7
commit c4317a7a96
1 changed files with 51 additions and 48 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

99
src/client_server/media.rs
View file

@ -74,6 +74,38 @@ pub async fn create_content_route(
.into()) .into())
} }
pub async fn get_remote_content(
db: &DatabaseGuard,
mxc: &str,
server_name: &ruma::ServerName,
media_id: &str
) -> ConduitResult<get_content::Response> {
let content_response = db
.sending
.send_federation_request(
&db.globals,
server_name,
get_content::Request {
allow_remote: false,
server_name,
media_id
},
)
.await?;
db.media
.create(
mxc.to_string(),
&db.globals,
&content_response.content_disposition.as_deref(),
&content_response.content_type.as_deref(),
&content_response.file,
)
.await?;
Ok(content_response.into())
}
/// # `GET /_matrix/media/r0/download/{serverName}/{mediaId}` /// # `GET /_matrix/media/r0/download/{serverName}/{mediaId}`
/// ///
/// Load media from our server or over federation. /// Load media from our server or over federation.
@ -103,30 +135,13 @@ pub async fn get_content_route(
} }
.into()) .into())
} else if &*body.server_name != db.globals.server_name() && body.allow_remote { } else if &*body.server_name != db.globals.server_name() && body.allow_remote {
let get_content_response = db let remote_content_response = get_remote_content(
.sending &db,
.send_federation_request( &mxc,
&db.globals, &body.server_name,
&body.server_name, &body.media_id
get_content::Request { ).await?;
allow_remote: false, Ok(remote_content_response)
server_name: &body.server_name,
media_id: &body.media_id,
},
)
.await?;
db.media
.create(
mxc,
&db.globals,
&get_content_response.content_disposition.as_deref(),
&get_content_response.content_type.as_deref(),
&get_content_response.file,
)
.await?;
Ok(get_content_response.into())
} else { } else {
Err(Error::BadRequest(ErrorKind::NotFound, "Media not found.")) Err(Error::BadRequest(ErrorKind::NotFound, "Media not found."))
} }
@ -161,31 +176,19 @@ pub async fn get_content_as_filename_route(
} }
.into()) .into())
} else if &*body.server_name != db.globals.server_name() && body.allow_remote { } else if &*body.server_name != db.globals.server_name() && body.allow_remote {
let get_content_response = db let remote_content_response = get_remote_content(
.sending &db,
.send_federation_request( &mxc,
&db.globals, &body.server_name,
&body.server_name, &body.media_id
get_content_as_filename::Request { ).await?;
allow_remote: false,
server_name: &body.server_name,
media_id: &body.media_id,
filename: &body.filename,
},
)
.await?;
db.media Ok(get_content_as_filename::Response {
.create( content_disposition: Some(format!("inline: filename={}", body.filename)),
mxc, content_type: remote_content_response.0.content_type,
&db.globals, file: remote_content_response.0.file
&get_content_response.content_disposition.as_deref(), }
&get_content_response.content_type.as_deref(), .into())
&get_content_response.file,
)
.await?;
Ok(get_content_response.into())
} else { } else {
Err(Error::BadRequest(ErrorKind::NotFound, "Media not found.")) Err(Error::BadRequest(ErrorKind::NotFound, "Media not found."))
} }