aboutsummaryrefslogtreecommitdiff
path: root/src/web/web_server.rs
diff options
context:
space:
mode:
Diffstat (limited to 'src/web/web_server.rs')
-rw-r--r--src/web/web_server.rs40
1 files changed, 36 insertions, 4 deletions
diff --git a/src/web/web_server.rs b/src/web/web_server.rs
index cbb2aaac..16b27cef 100644
--- a/src/web/web_server.rs
+++ b/src/web/web_server.rs
@@ -1,3 +1,4 @@
+use std::borrow::Cow;
use std::net::SocketAddr;
use std::sync::Arc;
@@ -55,17 +56,18 @@ async fn handler(
// Get path
let path = req.uri().path().to_string();
- let key = percent_encoding::percent_decode_str(&path).decode_utf8()?;
+ let index = &garage.config.s3_web.index;
+ let key = path_to_key(&path, &index)?;
- // Get bucket descriptor
+ info!("Selected bucket: \"{}\", selected key: \"{}\"", bucket, key);
+
+ // Get bucket descriptor
let object = garage
.object_table
.get(&bucket.to_string(), &key.to_string())
.await?
.ok_or(Error::NotFound)?;
- info!("Selected bucket: \"{}\", selected key: \"{}\"", bucket, key);
-
Ok(Response::new(Body::from("hello world\n")))
}
@@ -121,6 +123,27 @@ fn host_to_bucket<'a>(host: &'a str, root: &str) -> &'a str {
&host[..cursor]
}
+/// Path to key
+///
+/// Convert the provided path to the internal key
+/// When a path ends with "/", we append the index name to match traditional web server behavior
+/// which is also AWS S3 behavior.
+fn path_to_key<'a>(path: &'a str, index: &str) -> Result<Cow<'a, str>, Error> {
+ let path_utf8 = percent_encoding::percent_decode_str(&path).decode_utf8()?;
+ match path_utf8.chars().last() {
+ None => Err(Error::BadRequest(format!(
+ "Path must have at least a character"
+ ))),
+ Some('/') => {
+ let mut key = String::with_capacity(path_utf8.len() + index.len());
+ key.push_str(&path_utf8);
+ key.push_str(index);
+ Ok(key.into())
+ }
+ Some(_) => Ok(path_utf8.into()),
+ }
+}
+
#[cfg(test)]
mod tests {
use super::*;
@@ -170,4 +193,13 @@ mod tests {
assert_eq!(host_to_bucket("garage.tld", ".garage.tld"), "garage.tld");
}
+
+ #[test]
+ fn path_to_key_test() -> Result<(), Error> {
+ assert_eq!(path_to_key("/file%20.jpg", "index.html")?, "/file .jpg");
+ assert_eq!(path_to_key("/%20t/", "index.html")?, "/ t/index.html");
+ assert_eq!(path_to_key("/", "index.html")?, "/index.html");
+ assert!(path_to_key("", "index.html").is_err());
+ Ok(())
+ }
}