From f64edeb631bfdefcd757483d7dfbe204daabf1ab Mon Sep 17 00:00:00 2001
From: Cara Salter <cara@devcara.com>
Date: Thu, 9 Jun 2022 15:20:17 -0400
Subject: state: Force reboot/shutdown

Closes #1
---
 src/handlers/planets.rs | 24 ++++++++++++++++++++++++
 src/main.rs             |  2 ++
 2 files changed, 26 insertions(+)

(limited to 'src')

diff --git a/src/handlers/planets.rs b/src/handlers/planets.rs
index ede7b2b..06773bc 100644
--- a/src/handlers/planets.rs
+++ b/src/handlers/planets.rs
@@ -76,3 +76,27 @@ pub async fn reboot(Path(uuid): Path<String>) -> NoneResult {
     
     Ok(())
 }
+
+pub async fn force_reboot(Path(uuid): Path<String>) -> NoneResult {
+    let mut s = get_star()?;
+
+    if let Ok(p) = s.find_planet(uuid) {
+        p.hard_reboot()?;
+    } else {
+        return Err(ServiceError::NotFound);
+    }
+
+    Ok(())
+}
+
+pub async fn force_shutdown(Path(uuid): Path<String>) -> NoneResult {
+    let mut s = get_star()?;
+
+    if let Ok(p) = s.find_planet(uuid) {
+        p.hard_shutdown()?;
+    } else {
+        return Err(ServiceError::NotFound);
+    }
+
+    Ok(())
+}
diff --git a/src/main.rs b/src/main.rs
index b962764..c235d41 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -37,9 +37,11 @@ async fn main() {
         .route("/planets/list", get(handlers::planets::list))
         .route("/planets/:uuid", get(handlers::planets::get))
         .route("/planets/:uuid/shutdown", post(handlers::planets::shutdown))
+        .route("/planets/:uuid/shutdown/hard", post(handlers::planets::force_shutdown))
         .route("/planets/:uuid/start", post(handlers::planets::start))
         .route("/planets/:uuid/pause", post(handlers::planets::pause))
         .route("/planets/:uuid/reboot", post(handlers::planets::reboot))
+        .route("/planets/:uuid/reboot/hard", post(handlers::planets::force_reboot))
         .layer(            ServiceBuilder::new()
                 .layer(HandleErrorLayer::new(|error: BoxError| async move {
                     if error.is::<tower::timeout::error::Elapsed>() {
-- 
cgit v1.2.3