forked from matteo/serves3
Fix display of folders
This commit is contained in:
parent
eccd4ef797
commit
053854ce72
103
src/main.rs
103
src/main.rs
|
|
@ -1,9 +1,10 @@
|
||||||
// SPDX-FileCopyrightText: © Matteo Settenvini <matteo.settenvini@montecristosoftware.eu>
|
// SPDX-FileCopyrightText: © Matteo Settenvini <matteo.settenvini@montecristosoftware.eu>
|
||||||
// SPDX-License-Identifier: EUPL-1.2
|
// SPDX-License-Identifier: EUPL-1.2
|
||||||
|
|
||||||
|
use s3::serde_types::ListBucketResult;
|
||||||
|
|
||||||
use {
|
use {
|
||||||
lazy_static::lazy_static,
|
lazy_static::lazy_static,
|
||||||
rocket::http::ContentType,
|
|
||||||
rocket::response::Responder,
|
rocket::response::Responder,
|
||||||
rocket_dyn_templates::{context, Template},
|
rocket_dyn_templates::{context, Template},
|
||||||
std::path::PathBuf,
|
std::path::PathBuf,
|
||||||
|
|
@ -63,44 +64,98 @@ lazy_static! {
|
||||||
}
|
}
|
||||||
|
|
||||||
#[derive(Responder)]
|
#[derive(Responder)]
|
||||||
enum Error {
|
enum FileView {
|
||||||
#[response(status = 422)]
|
#[response(content_type = "text/html")]
|
||||||
InvalidArgument(String),
|
Folder(Template),
|
||||||
|
|
||||||
|
#[response(content_type = "application/octet-stream")]
|
||||||
|
File(Vec<u8>),
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Responder, Debug)]
|
||||||
|
enum Error {
|
||||||
#[response(status = 404)]
|
#[response(status = 404)]
|
||||||
NotFound(String),
|
NotFound(String),
|
||||||
}
|
}
|
||||||
|
|
||||||
#[rocket::get("/<path..>")]
|
#[rocket::get("/<path..>")]
|
||||||
async fn index(path: PathBuf) -> Result<(ContentType, Template), Error> {
|
async fn index(path: PathBuf) -> Result<FileView, Error> {
|
||||||
let path: String = path
|
let parent = path.parent();
|
||||||
.to_str()
|
|
||||||
.ok_or_else(|| Error::InvalidArgument(format!("Invalid path '{}'", path.display())))?
|
|
||||||
.into();
|
|
||||||
|
|
||||||
let s3_objects = BUCKET
|
let s3_path = format!(
|
||||||
.list(path, Some("/".into()))
|
"{}{}",
|
||||||
.await
|
path.display(),
|
||||||
.map_err(|_| Error::NotFound("Object not found".into()))?;
|
match parent {
|
||||||
|
Some(_) => "/",
|
||||||
|
None => "",
|
||||||
|
}
|
||||||
|
);
|
||||||
|
|
||||||
let objects: Vec<String> = s3_objects
|
/*
|
||||||
.into_iter()
|
The way things work in S3, the following holds for us:
|
||||||
.flat_map(|list| -> Vec<String> {
|
- we need to use a slash as separator
|
||||||
if let Some(common_prefixes) = list.common_prefixes {
|
- folders need to be queried ending with a slash
|
||||||
common_prefixes.into_iter().map(|dir| dir.prefix).collect()
|
|
||||||
} else {
|
|
||||||
list.contents.into_iter().map(|obj| obj.key).collect()
|
|
||||||
}
|
|
||||||
})
|
|
||||||
.collect();
|
|
||||||
|
|
||||||
|
We try first to retrieve list an object as a folder. If we fail,
|
||||||
|
we fallback to retrieving the object itself.
|
||||||
|
*/
|
||||||
|
let s3_objects = BUCKET.list(s3_path, Some("/".into())).await;
|
||||||
|
|
||||||
|
let s3_objects = match s3_objects {
|
||||||
|
Ok(s3_objects) => s3_objects,
|
||||||
|
Err(_) => {
|
||||||
|
// TODO: this can be big, we should use streaming,
|
||||||
|
// not loading in memory.
|
||||||
|
let data = BUCKET
|
||||||
|
.get_object(format!("{}", path.display()))
|
||||||
|
.await
|
||||||
|
.map_err(|_| Error::NotFound("Object not found".into()))?
|
||||||
|
.bytes()
|
||||||
|
.to_vec();
|
||||||
|
return Ok(FileView::File(data));
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
let objects = s3_fileview(&s3_objects);
|
||||||
let rendered = Template::render(
|
let rendered = Template::render(
|
||||||
"index",
|
"index",
|
||||||
context! {
|
context! {
|
||||||
|
path: format!("{}/", path.display()),
|
||||||
|
has_parent: !path.as_os_str().is_empty(),
|
||||||
objects
|
objects
|
||||||
},
|
},
|
||||||
);
|
);
|
||||||
Ok((ContentType::HTML, rendered))
|
Ok(FileView::Folder(rendered))
|
||||||
|
}
|
||||||
|
|
||||||
|
fn s3_fileview(s3_objects: &Vec<ListBucketResult>) -> Vec<&str> {
|
||||||
|
/*
|
||||||
|
if listing a folder:
|
||||||
|
- folders will be under 'common_prefixes'
|
||||||
|
- files will be under the 'contents' property
|
||||||
|
*/
|
||||||
|
s3_objects
|
||||||
|
.iter()
|
||||||
|
.flat_map(|list| -> Vec<Option<&str>> {
|
||||||
|
let prefix = if let Some(p) = &list.prefix {
|
||||||
|
p.as_str()
|
||||||
|
} else {
|
||||||
|
""
|
||||||
|
};
|
||||||
|
|
||||||
|
let folders = list
|
||||||
|
.common_prefixes
|
||||||
|
.iter()
|
||||||
|
.flatten()
|
||||||
|
.map(|dir| dir.prefix.strip_prefix(&prefix));
|
||||||
|
let files = list
|
||||||
|
.contents
|
||||||
|
.iter()
|
||||||
|
.map(|obj| obj.key.strip_prefix(&prefix));
|
||||||
|
folders.chain(files).collect()
|
||||||
|
})
|
||||||
|
.flatten()
|
||||||
|
.collect()
|
||||||
}
|
}
|
||||||
|
|
||||||
#[rocket::launch]
|
#[rocket::launch]
|
||||||
|
|
|
||||||
|
|
@ -6,7 +6,12 @@
|
||||||
<html>
|
<html>
|
||||||
<head></head>
|
<head></head>
|
||||||
<body>
|
<body>
|
||||||
|
<h1>{{ path }}</h1>
|
||||||
<ul>
|
<ul>
|
||||||
|
{% if has_parent %}
|
||||||
|
<li><a href="../">..</a></li>
|
||||||
|
{% endif %}
|
||||||
|
|
||||||
{% for object in objects %}
|
{% for object in objects %}
|
||||||
<li><a href="{{ object }}">{{ object }}</a></li>
|
<li><a href="{{ object }}">{{ object }}</a></li>
|
||||||
{% endfor %}
|
{% endfor %}
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue