2017-10-16 17:43:35 +00:00
|
|
|
//! Static files support.
|
|
|
|
|
//!
|
|
|
|
|
//! TODO: needs to re-implement actual files handling, current impl blocks
|
2017-10-16 08:19:23 +00:00
|
|
|
use std::io;
|
|
|
|
|
use std::io::Read;
|
|
|
|
|
use std::fmt::Write;
|
|
|
|
|
use std::fs::{File, DirEntry};
|
|
|
|
|
use std::path::PathBuf;
|
2017-10-15 05:52:38 +00:00
|
|
|
|
2017-10-16 08:19:23 +00:00
|
|
|
use mime_guess::get_mime_type;
|
2017-11-29 21:26:55 +00:00
|
|
|
use route::Handler;
|
2017-10-15 05:52:38 +00:00
|
|
|
use httprequest::HttpRequest;
|
2017-10-24 06:25:32 +00:00
|
|
|
use httpresponse::HttpResponse;
|
2017-12-02 22:58:22 +00:00
|
|
|
use httpcodes::{HTTPOk, HTTPNotFound};
|
2017-10-15 05:52:38 +00:00
|
|
|
|
2017-10-16 08:19:23 +00:00
|
|
|
/// Static files handling
|
|
|
|
|
///
|
|
|
|
|
/// Can be registered with `Application::route_handler()`.
|
|
|
|
|
///
|
|
|
|
|
/// ```rust
|
|
|
|
|
/// extern crate actix_web;
|
|
|
|
|
/// use actix_web::*;
|
|
|
|
|
///
|
|
|
|
|
/// fn main() {
|
2017-10-22 01:54:24 +00:00
|
|
|
/// let app = Application::default("/")
|
2017-11-29 21:53:52 +00:00
|
|
|
/// .route("/static", StaticFiles::new(".", true))
|
2017-10-16 08:19:23 +00:00
|
|
|
/// .finish();
|
|
|
|
|
/// }
|
|
|
|
|
/// ```
|
2017-10-15 05:52:38 +00:00
|
|
|
pub struct StaticFiles {
|
2017-10-16 08:19:23 +00:00
|
|
|
directory: PathBuf,
|
|
|
|
|
accessible: bool,
|
2017-11-25 18:52:43 +00:00
|
|
|
_show_index: bool,
|
|
|
|
|
_chunk_size: usize,
|
|
|
|
|
_follow_symlinks: bool,
|
2017-10-16 08:19:23 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
|
|
impl StaticFiles {
|
|
|
|
|
/// Create new `StaticFiles` instance
|
|
|
|
|
///
|
|
|
|
|
/// `dir` - base directory
|
|
|
|
|
/// `index` - show index for directory
|
|
|
|
|
pub fn new<D: Into<PathBuf>>(dir: D, index: bool) -> StaticFiles {
|
|
|
|
|
let dir = dir.into();
|
|
|
|
|
|
2017-10-22 05:59:09 +00:00
|
|
|
let (dir, access) = match dir.canonicalize() {
|
|
|
|
|
Ok(dir) => {
|
|
|
|
|
if dir.is_dir() {
|
|
|
|
|
(dir, true)
|
|
|
|
|
} else {
|
|
|
|
|
warn!("Is not directory `{:?}`", dir);
|
|
|
|
|
(dir, false)
|
|
|
|
|
}
|
|
|
|
|
},
|
|
|
|
|
Err(err) => {
|
|
|
|
|
warn!("Static files directory `{:?}` error: {}", dir, err);
|
2017-10-16 08:19:23 +00:00
|
|
|
(dir, false)
|
|
|
|
|
}
|
|
|
|
|
};
|
|
|
|
|
|
|
|
|
|
StaticFiles {
|
|
|
|
|
directory: dir,
|
|
|
|
|
accessible: access,
|
2017-11-25 18:52:43 +00:00
|
|
|
_show_index: index,
|
|
|
|
|
_chunk_size: 0,
|
|
|
|
|
_follow_symlinks: false,
|
2017-10-16 08:19:23 +00:00
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
2017-11-29 23:07:49 +00:00
|
|
|
fn index(&self, prefix: &str, relpath: &str, filename: &PathBuf) -> Result<HttpResponse, io::Error> {
|
|
|
|
|
let index_of = format!("Index of {}{}", prefix, relpath);
|
2017-10-16 08:19:23 +00:00
|
|
|
let mut body = String::new();
|
|
|
|
|
|
|
|
|
|
for entry in filename.read_dir()? {
|
|
|
|
|
if self.can_list(&entry) {
|
|
|
|
|
let entry = entry.unwrap();
|
|
|
|
|
// show file url as relative to static path
|
|
|
|
|
let file_url = format!(
|
2017-11-29 23:07:49 +00:00
|
|
|
"{}{}", prefix,
|
2017-10-16 08:19:23 +00:00
|
|
|
entry.path().strip_prefix(&self.directory).unwrap().to_string_lossy());
|
|
|
|
|
|
|
|
|
|
// if file is a directory, add '/' to the end of the name
|
2017-11-25 18:52:43 +00:00
|
|
|
if let Ok(metadata) = entry.metadata() {
|
2017-10-16 08:19:23 +00:00
|
|
|
if metadata.is_dir() {
|
|
|
|
|
//format!("<li><a href=\"{}\">{}</a></li>", file_url, file_name));
|
2017-11-25 18:52:43 +00:00
|
|
|
let _ = write!(body, "<li><a href=\"{}\">{}/</a></li>",
|
|
|
|
|
file_url, entry.file_name().to_string_lossy());
|
2017-10-16 08:19:23 +00:00
|
|
|
} else {
|
|
|
|
|
// write!(body, "{}/", entry.file_name())
|
2017-11-25 18:52:43 +00:00
|
|
|
let _ = write!(body, "<li><a href=\"{}\">{}</a></li>",
|
|
|
|
|
file_url, entry.file_name().to_string_lossy());
|
2017-10-16 08:19:23 +00:00
|
|
|
}
|
|
|
|
|
} else {
|
|
|
|
|
continue
|
2017-11-25 18:52:43 +00:00
|
|
|
}
|
2017-10-16 08:19:23 +00:00
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
let html = format!("<html>\
|
|
|
|
|
<head><title>{}</title></head>\
|
|
|
|
|
<body><h1>{}</h1>\
|
|
|
|
|
<ul>\
|
|
|
|
|
{}\
|
|
|
|
|
</ul></body>\n</html>", index_of, index_of, body);
|
|
|
|
|
Ok(
|
2017-11-27 06:31:29 +00:00
|
|
|
HTTPOk.build()
|
2017-10-16 08:19:23 +00:00
|
|
|
.content_type("text/html; charset=utf-8")
|
2017-10-24 06:25:32 +00:00
|
|
|
.body(html).unwrap()
|
2017-10-16 08:19:23 +00:00
|
|
|
)
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
fn can_list(&self, entry: &io::Result<DirEntry>) -> bool {
|
|
|
|
|
if let Ok(ref entry) = *entry {
|
|
|
|
|
if let Some(name) = entry.file_name().to_str() {
|
|
|
|
|
if name.starts_with('.') {
|
|
|
|
|
return false
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
if let Ok(ref md) = entry.metadata() {
|
|
|
|
|
let ft = md.file_type();
|
|
|
|
|
return ft.is_dir() || ft.is_file() || ft.is_symlink()
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
false
|
|
|
|
|
}
|
2017-10-15 05:52:38 +00:00
|
|
|
}
|
|
|
|
|
|
2017-11-29 21:26:55 +00:00
|
|
|
impl<S> Handler<S> for StaticFiles {
|
|
|
|
|
type Result = Result<HttpResponse, io::Error>;
|
2017-10-15 05:52:38 +00:00
|
|
|
|
2017-11-29 21:26:55 +00:00
|
|
|
fn handle(&self, req: HttpRequest<S>) -> Self::Result {
|
2017-10-16 08:19:23 +00:00
|
|
|
if !self.accessible {
|
2017-11-29 21:26:55 +00:00
|
|
|
Ok(HTTPNotFound.into())
|
2017-10-16 08:19:23 +00:00
|
|
|
} else {
|
|
|
|
|
let mut hidden = false;
|
2017-11-29 23:07:49 +00:00
|
|
|
let filepath = req.path()[req.prefix_len()..]
|
2017-10-16 08:19:23 +00:00
|
|
|
.split('/').filter(|s| {
|
|
|
|
|
if s.starts_with('.') {
|
|
|
|
|
hidden = true;
|
|
|
|
|
}
|
|
|
|
|
!s.is_empty()
|
|
|
|
|
})
|
|
|
|
|
.fold(String::new(), |s, i| {s + "/" + i});
|
|
|
|
|
|
|
|
|
|
// hidden file
|
|
|
|
|
if hidden {
|
2017-11-29 21:26:55 +00:00
|
|
|
return Ok(HTTPNotFound.into())
|
2017-10-16 08:19:23 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// full filepath
|
|
|
|
|
let idx = if filepath.starts_with('/') { 1 } else { 0 };
|
2017-12-02 22:58:22 +00:00
|
|
|
let filename = self.directory.join(&filepath[idx..]).canonicalize()?;
|
2017-10-16 08:19:23 +00:00
|
|
|
|
|
|
|
|
if filename.is_dir() {
|
2017-12-02 22:58:22 +00:00
|
|
|
self.index(&req.path()[..req.prefix_len()], &filepath[idx..], &filename)
|
2017-10-16 08:19:23 +00:00
|
|
|
} else {
|
2017-11-27 06:31:29 +00:00
|
|
|
let mut resp = HTTPOk.build();
|
2017-10-16 08:19:23 +00:00
|
|
|
if let Some(ext) = filename.extension() {
|
|
|
|
|
let mime = get_mime_type(&ext.to_string_lossy());
|
|
|
|
|
resp.content_type(format!("{}", mime).as_str());
|
|
|
|
|
}
|
|
|
|
|
match File::open(filename) {
|
|
|
|
|
Ok(mut file) => {
|
|
|
|
|
let mut data = Vec::new();
|
|
|
|
|
let _ = file.read_to_end(&mut data);
|
2017-11-29 21:26:55 +00:00
|
|
|
Ok(resp.body(data).unwrap())
|
2017-10-16 08:19:23 +00:00
|
|
|
},
|
2017-11-29 21:26:55 +00:00
|
|
|
Err(err) => Err(err),
|
2017-10-16 08:19:23 +00:00
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|
2017-10-15 05:52:38 +00:00
|
|
|
}
|
|
|
|
|
}
|
