use std::io; use std::net; use std::path::PathBuf; use http_io::protocol::{HttpBody, HttpResponse, HttpStatus}; use http_io::server::{HttpRequestHandler, HttpServer}; struct FileHandler { file_root: PathBuf, } impl FileHandler { fn new>(file_root: P) -> Self { FileHandler { file_root: file_root.into(), } } } #[derive(Debug)] struct Error(io::Error); type Result = std::result::Result; impl From for Error { fn from(e: io::Error) -> Self { Self(e) } } impl From for HttpResponse> { fn from(e: Error) -> Self { HttpResponse::from_string(HttpStatus::InternalServerError, e.0.to_string()) } } impl HttpRequestHandler for FileHandler { type Error = Error; fn get(&mut self, uri: String) -> Result>> { let path = self.file_root.join(uri.trim_start_matches("/")); println!("Request for {:?}", path); let attrs = std::fs::metadata(&path)?; if attrs.is_dir() { let mut file_list = String::new(); for entry in std::fs::read_dir(&path)? { let entry = entry?; if let Some(name) = entry.file_name().to_str() { let link = name.to_owned() + if entry.metadata()?.is_dir() { "/" } else { "" }; file_list += &format!("
  • {}
    ", link, name,); } } let page = format!( r#" Directory listing for {0}

    Directory listing for {0}

      {1}
    "#, uri, &file_list ); Ok(HttpResponse::new( HttpStatus::OK, Box::new(io::Cursor::new(page)), )) } else { let mut res = HttpResponse::new( HttpStatus::OK, Box::new(std::fs::File::open(path)?) as Box, ); res.add_header("Content-Length", attrs.len().to_string()); Ok(res) } } fn put( &mut self, uri: String, mut stream: HttpBody<&mut I>, ) -> Result>> { let path = self.file_root.join(uri.trim_start_matches("/")); println!("Uploading to {:?}", path); let mut file = std::fs::File::create(path)?; io::copy(&mut stream, &mut file)?; Ok(HttpResponse::new(HttpStatus::OK, Box::new(io::empty()))) } } fn main() -> Result<()> { let handler = FileHandler::new(std::env::current_dir()?); let socket = net::TcpListener::bind("127.0.0.1:8080")?; let mut server = HttpServer::new(socket, handler); println!("Server started on port 8080"); server.serve_forever(); }