add directory listing api

1 files changed, 22 insertions, 0 deletions

diff --git a/src/main.rs b/src/main.rs
index 6ca60d2..e185658 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -189,6 +189,27 @@ async fn handle_data_write(info: web::Query<DataRequest>, data: web::Bytes) -> H
     }
 }
 
+async fn handle_data_list(info: web::Query<DataRequest>) -> HttpResponse {
+    let query = info.into_inner();
+
+    match fs::read_dir(&query.path) {
+        Ok(ls) => HttpResponse::Ok()
+            .content_type(ContentType::plaintext())
+            .body(
+                ls.map(|result| {
+                    result
+                        .map(|entry| format!("{}", entry.path().display()))
+                        .unwrap_or_else(|e| format!("{}", e))
+                })
+                .reduce(|acc, entry| acc + "\n" + &entry)
+                .unwrap_or(String::new()),
+            ),
+        Err(e) => HttpResponse::InternalServerError()
+            .content_type(ContentType::plaintext())
+            .body(format!("can't read dir at {}: {}", query.path, e)),
+    }
+}
+
 async fn handle_proc_top() -> HttpResponse {
     println!("monitor processes");
 
@@ -272,6 +293,7 @@ async fn start() -> Result<()> {
             .service(web::resource("/switch").to(handle_switch))
             .service(web::resource("/data/read").to(handle_data_read))
             .service(web::resource("/data/write").to(handle_data_write))
+            .service(web::resource("/data/list").to(handle_data_list))
             .service(web::resource("/proc/top").to(handle_proc_top))
             .service(web::resource("/proc/kill").to(handle_proc_kill))
     })