[+] Simple file listing server

2023-02-16 20:35:48 -05:00
parent a7cf931742
commit b72f2ff701
1 changed files with 46 additions and 2 deletions
@@ -1,3 +1,47 @@
-fn main() {
-    println!("Hello, world!");
+use std::convert::Infallible;
+use std::{env, fs};
+use std::net::SocketAddr;
+use std::os::unix::fs::MetadataExt;
+use hyper::{Body, Request, Response, Server};
+use hyper::service::{make_service_fn, service_fn};
+use serde::{Deserialize, Serialize};
+
+#[tokio::main]
+async fn main() {
+    let cwd = env::current_dir().unwrap();
+    let addr = SocketAddr::from(([127, 0, 0, 1], 3029));
+    println!("Serving {} started on http://127.0.0.1:3029", cwd.display());
+
+    // A `Service` is needed for every connection, so this
+    // creates one from our `hello_world` function.
+    let make_svc = make_service_fn(|_conn| async {
+        // service_fn converts our function into a `Service`
+        Ok::<_, Infallible>(service_fn(hello_world))
+    });
+
+    let server = Server::bind(&addr).serve(make_svc);
+
+    // Run this server for... forever!
+    if let Err(e) = server.await {
+        eprintln!("server error: {}", e);
+    }
+}
+
+#[derive(Serialize, Deserialize)]
+struct ReturnPath {
+    name: String,
+    file_type: String,
+    mtime: i64
+}
+
+async fn hello_world(_req: Request<Body>) -> Result<Response<Body>, Infallible> {
+    let paths: Vec<ReturnPath> = fs::read_dir("./").unwrap()
+        .map(|x| x.unwrap())
+        .map(|x| ReturnPath {
+            name: x.file_name().to_str().unwrap().parse().unwrap(),
+            file_type: if x.path().is_file() { "file" } else { "directory" }.parse().unwrap(),
+            mtime: x.metadata().unwrap().mtime(),
+        }).collect();
+
+    Ok(Response::new(serde_json::to_string(&paths).unwrap().into()))
 }