yk/http-server-cc
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

stage 7 attempt 4 (compatibility with previous tests)

Browse Source
This commit is contained in:
YK 2024-05-11 01:14:34 +03:00
parent 4a298c8b1e
commit f4b8eb502d
1 changed files with 19 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

28
src/main.rs
View File

@ -14,15 +14,19 @@ use utils::*;
#[derive(Debug, Clone)] #[derive(Debug, Clone)]
struct Args { struct Args {
pub directory: PathBuf, pub directory: Option<PathBuf>,
} }
type A = Arc<Args>; type A = Arc<Args>;
fn parse_args () -> Args { fn parse_args () -> Args {
let directory = std::env::args().position(|e| e == "--directory").unwrap() + 1; let directory = std::env::args()
let directory = PathBuf::from(std::env::args().nth(directory).unwrap()); .position(|e| e == "--directory")
.map(|e| e + 1)
.map(|d| std::env::args().nth(d))
.flatten()
.map(|d| PathBuf::from(d));
Args { Args {
directory directory
@ -68,13 +72,17 @@ async fn process (mut stream: TcpStream, args: A) -> Result<()> {
// p if let Some(echo) = p.strip_prefix("/echo/") => Response::TextPlain(echo), // a nicer way to do that, not available in stable yet // p if let Some(echo) = p.strip_prefix("/echo/") => Response::TextPlain(echo), // a nicer way to do that, not available in stable yet
p if p.starts_with("/echo/") => Response::TextPlain(p.trim_start_matches("/echo/").to_owned()), p if p.starts_with("/echo/") => Response::TextPlain(p.trim_start_matches("/echo/").to_owned()),
p if p.starts_with("/files/") => { p if p.starts_with("/files/") => {
let path = args.directory.join(p.trim_start_matches("/files/")); if let Some(path) = &args.directory {
let mut buf = vec![]; let path = path.join(p.trim_start_matches("/files/"));
if let Ok(mut f) = File::open(path).await { let mut buf = vec![];
let _ = f.read_to_end(&mut buf).await; if let Ok(mut f) = File::open(path).await {
Response::OctetStream(buf) let _ = f.read_to_end(&mut buf).await;
Response::OctetStream(buf)
} else {
Response::_404
}
} else { } else {
Response::_404 Response::_500
} }
}, },
_ => Response::_404, _ => Response::_404,
@ -113,6 +121,7 @@ impl Headers {
#[derive(Debug, Clone)] #[derive(Debug, Clone)]
enum Response { enum Response {
_404, _404,
_500,
Empty, Empty,
TextPlain (String), TextPlain (String),
OctetStream (Vec<u8>) OctetStream (Vec<u8>)
@ -127,6 +136,7 @@ impl Response {
let code = match self { let code = match self {
Self::_404 => "404 Not Found", Self::_404 => "404 Not Found",
Self::_500 => "500 Internal Server Error",
_ => "200 OK", _ => "200 OK",
}; };