add api to remove files

1 files changed, 16 insertions, 2 deletions

diff --git a/src/main.rs b/src/main.rs
index cfa6dcb..6c4ba49 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -233,10 +233,10 @@ async fn handle_data_list(info: web::Query<DataRequest>) -> HttpResponse {
     }
 }
 
-async fn handle_data_removedir(info: web::Query<DataRequest>) -> HttpResponse {
+async fn handle_data_remove(info: web::Query<DataRequest>) -> HttpResponse {
     let query = info.into_inner();
 
-    match fs::remove_dir_all(&query.path) {
+    match fs::remove_file(&query.path) {
         Ok(_) => HttpResponse::Ok()
             .content_type(ContentType::plaintext())
             .body(format!("successfully removed {}", query.path)),
@@ -246,6 +246,19 @@ async fn handle_data_removedir(info: web::Query<DataRequest>) -> HttpResponse {
     }
 }
 
+async fn handle_data_removedir(info: web::Query<DataRequest>) -> HttpResponse {
+    let query = info.into_inner();
+
+    match fs::remove_dir_all(&query.path) {
+        Ok(_) => HttpResponse::Ok()
+            .content_type(ContentType::plaintext())
+            .body(format!("successfully removed directory {}", query.path)),
+        Err(e) => HttpResponse::InternalServerError()
+            .content_type(ContentType::plaintext())
+            .body(format!("can't remove directory {}: {}", query.path, e)),
+    }
+}
+
 async fn handle_proc_top() -> HttpResponse {
     println!("monitor processes");
 
@@ -330,6 +343,7 @@ async fn start() -> Result<()> {
             .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("/data/remove").to(handle_data_remove))
             .service(web::resource("/data/removedir").to(handle_data_removedir))
             .service(web::resource("/proc/top").to(handle_proc_top))
             .service(web::resource("/proc/kill").to(handle_proc_kill))