//! Static files support. //! //! TODO: needs to re-implement actual files handling, current impl blocks use std::io; use std::io::Read; use std::fmt::Write; use std::fs::{File, DirEntry}; use std::path::PathBuf; use mime_guess::get_mime_type; use route::Handler; use httprequest::HttpRequest; use httpresponse::HttpResponse; use httpcodes::{HTTPOk, HTTPNotFound, HTTPForbidden}; /// Static files handling /// /// Can be registered with `Application::route_handler()`. /// /// ```rust /// extern crate actix_web; /// use actix_web::*; /// /// fn main() { /// let app = Application::default("/") /// .route("/static", StaticFiles::new(".", true)) /// .finish(); /// } /// ``` pub struct StaticFiles { directory: PathBuf, accessible: bool, _show_index: bool, _chunk_size: usize, _follow_symlinks: bool, prefix: String, } impl StaticFiles { /// Create new `StaticFiles` instance /// /// `dir` - base directory /// `index` - show index for directory pub fn new>(dir: D, index: bool) -> StaticFiles { let dir = dir.into(); 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); (dir, false) } }; StaticFiles { directory: dir, accessible: access, _show_index: index, _chunk_size: 0, _follow_symlinks: false, prefix: String::new(), } } fn index(&self, relpath: &str, filename: &PathBuf) -> Result { let index_of = format!("Index of {}/{}", self.prefix, relpath); 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!( "{}/{}", self.prefix, entry.path().strip_prefix(&self.directory).unwrap().to_string_lossy()); // if file is a directory, add '/' to the end of the name if let Ok(metadata) = entry.metadata() { if metadata.is_dir() { //format!("

", file_url, file_name)); let _ = write!(body, "

", file_url, entry.file_name().to_string_lossy()); } else { // write!(body, "{}/", entry.file_name()) let _ = write!(body, "

{}

", file_url, entry.file_name().to_string_lossy()); } } else { continue } } } let html = format!("\ {}\

{}

\ {}\ \n", index_of, index_of, body); Ok( HTTPOk.build() .content_type("text/html; charset=utf-8") .body(html).unwrap() ) } fn can_list(&self, entry: &io::Result) -> 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 } } impl Handler for StaticFiles { type Result = Result; fn set_prefix(&mut self, prefix: String) { if prefix != "/" { self.prefix += &prefix; } } fn handle(&self, req: HttpRequest) -> Self::Result { if !self.accessible { Ok(HTTPNotFound.into()) } else { let mut hidden = false; let filepath = req.path()[self.prefix.len()..] .split('/').filter(|s| { if s.starts_with('.') { hidden = true; } !s.is_empty() }) .fold(String::new(), |s, i| {s + "/" + i}); // hidden file if hidden { return Ok(HTTPNotFound.into()) } // full filepath let idx = if filepath.starts_with('/') { 1 } else { 0 }; let filename = match self.directory.join(&filepath[idx..]).canonicalize() { Ok(fname) => fname, Err(err) => return match err.kind() { io::ErrorKind::NotFound => Ok(HTTPNotFound.into()), io::ErrorKind::PermissionDenied => Ok(HTTPForbidden.into()), _ => Err(err), } }; if filename.is_dir() { match self.index(&filepath[idx..], &filename) { Ok(resp) => Ok(resp), Err(err) => match err.kind() { io::ErrorKind::NotFound => Ok(HTTPNotFound.into()), io::ErrorKind::PermissionDenied => Ok(HTTPForbidden.into()), _ => Err(err), } } } else { let mut resp = HTTPOk.build(); 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); Ok(resp.body(data).unwrap()) }, Err(err) => Err(err), } } } } }