Fixed redirection to paths not ending with a slash

This commit is contained in:
Eren AY 2024-06-12 11:55:30 +02:00
parent dcd3c10bdd
commit cf98738a0d
1 changed files with 17 additions and 3 deletions

View File

@ -13,7 +13,8 @@ use {
providers::{Env, Format as _, Toml}, providers::{Env, Format as _, Toml},
Profile, Profile,
}, },
response::Responder, http::uri::Origin,
response::{Redirect, Responder},
serde::Serialize, serde::Serialize,
State, State,
}, },
@ -29,6 +30,8 @@ enum FileView {
#[response(content_type = "application/octet-stream")] #[response(content_type = "application/octet-stream")]
File(Vec<u8>), File(Vec<u8>),
Redirect(Redirect),
} }
#[derive(Serialize)] #[derive(Serialize)]
@ -49,7 +52,11 @@ enum Error {
} }
#[rocket::get("/<path..>")] #[rocket::get("/<path..>")]
async fn index(path: PathBuf, state: &State<Settings>) -> Result<FileView, Error> { async fn index(
path: PathBuf,
uri: &Origin<'_>,
state: &State<Settings>,
) -> Result<FileView, Error> {
/* /*
The way things work in S3, the following holds for us: The way things work in S3, the following holds for us:
- we need to use a slash as separator - we need to use a slash as separator
@ -61,10 +68,17 @@ async fn index(path: PathBuf, state: &State<Settings>) -> Result<FileView, Error
We try first to retrieve list an object as a file. If we fail, We try first to retrieve list an object as a file. If we fail,
we fallback to retrieving the equivalent folder. we fallback to retrieving the equivalent folder.
*/ */
if let Ok(result) = s3_serve_file(&path, &state).await { if let Ok(result) = s3_serve_file(&path, &state).await {
Ok(result) Ok(result)
} else { } else {
// We need to redirect to a path ending with a slash as
// per comment above if we know this is not a file.
let mut uri = uri.to_string();
if !uri.ends_with('/') {
uri.push('/');
return Ok(FileView::Redirect(Redirect::permanent(uri)));
}
let objects = s3_fileview(&path, &state).await?; let objects = s3_fileview(&path, &state).await?;
let rendered = Template::render( let rendered = Template::render(
"index", "index",