use actix_web::{error, fs, App, HttpRequest, HttpResponse, Responder, middleware}; use actix_web::dev::FromParam; use futures::Stream; use percent_encoding::{utf8_percent_encode, DEFAULT_ENCODE_SET}; use htmlescape::encode_minimal as escape_html_entity; use channel; use std::fmt::Write; use std::path::PathBuf; use std; use bytes; pub fn create_app(directory: &PathBuf) -> App { let root = directory.to_path_buf(); let static_files = fs::StaticFiles::new(&root).unwrap() .show_files_listing() .files_listing_renderer(handle_directory); App::with_state(root) .middleware(middleware::Logger::new(r#"%a "%r" %s %b "%{Referer}i" "%{User-Agent}i" %T"#)) .resource(r"/{tail:.*}.tar", |r| r.get().f(handle_tar)) .resource(r"/favicon.ico", |r| r.get().f(favicon_ico)) .handler("/", static_files) } fn handle_directory<'a, 'b>( dir: &'a fs::Directory, req: &'b HttpRequest, ) -> std::io::Result { let mut paths: Vec<_> = std::fs::read_dir(&dir.path).unwrap() .filter_map(|entry| if dir.is_visible(&entry) { entry.ok() } else {None}) .collect(); paths.sort_by_key(|r| (!r.metadata().unwrap().file_type().is_dir(), r.file_name())); let dir_tar_path = String::from(req.path().trim_right_matches('/')) + ".tar"; let tar_url = utf8_percent_encode(&dir_tar_path, DEFAULT_ENCODE_SET).to_string(); let mut body = String::new(); writeln!(body, "

Index of {}

", req.path()).unwrap(); writeln!(body, r#"[.tar of whole directory]"#, tar_url).unwrap(); writeln!(body, "").unwrap(); writeln!(body, "").unwrap(); for entry in paths { let meta = entry.metadata()?; let file_url = utf8_percent_encode(&entry.file_name().to_string_lossy(), DEFAULT_ENCODE_SET).to_string(); let file_name = escape_html_entity(&entry.file_name().to_string_lossy()); let size = meta.len(); write!(body, "").unwrap(); if meta.file_type().is_dir() { writeln!(body, r#""#, file_url, file_name).unwrap(); write!(body, r#" "#, file_url).unwrap(); } else { writeln!(body, r#""#, file_url, file_name).unwrap(); write!(body, " ", size).unwrap(); } writeln!(body, "").unwrap(); } writeln!(body, "

📁 ../	Size
📂 {}/	[.tar]	🗎 {}	{}

").unwrap(); let mut html = String::new(); writeln!(html, "").unwrap(); writeln!(html, "").unwrap(); writeln!(html, "Index of {}", req.path()).unwrap(); writeln!(html, "", include_str!("style.css")).unwrap(); writeln!(html, "").unwrap(); writeln!(html, "\n{}", body).unwrap(); writeln!(html, "").unwrap(); Ok(HttpResponse::Ok().content_type("text/html; charset=utf-8").body(html)) } fn handle_tar(req: &HttpRequest) -> impl Responder { let root = req.state(); let tail: String = req.match_info().query("tail")?; let relpath = PathBuf::from_param(tail.trim_left_matches('/'))?; let fullpath = root.join(&relpath).canonicalize()?; if !(fullpath.is_dir()) { return Err(error::ErrorBadRequest("not a directory")); } let stream = channel::stream_tar_in_thread(fullpath); let resp = HttpResponse::Ok() .content_type("application/x-tar") .streaming(stream.map_err(|_e| error::ErrorBadRequest("stream error"))); Ok(resp) } fn favicon_ico(_req: &HttpRequest) -> impl Responder { bytes::Bytes::from_static(include_bytes!("favicon.png")) }