From 276255f5a20fa05a005e765287239ca5ffe439e4 Mon Sep 17 00:00:00 2001
From: Cara Salter <cara@devcara.com>
Date: Thu, 9 Jun 2022 10:28:42 -0400
Subject: planets: State manipulation

---
 src/handlers/planets.rs | 52 +++++++++++++++++++++++++++++++++++++++++++++++--
 1 file changed, 50 insertions(+), 2 deletions(-)

(limited to 'src/handlers')

diff --git a/src/handlers/planets.rs b/src/handlers/planets.rs
index e80e7fe..a57f605 100644
--- a/src/handlers/planets.rs
+++ b/src/handlers/planets.rs
@@ -1,11 +1,11 @@
-use axum::{response::IntoResponse, Json};
+use axum::{response::IntoResponse, Json, extract::Path};
 
 use tracing::{error, instrument};
 
 use solarlib::star::Star;
 use solarlib::planet::Planet;
 
-use crate::errors::*;
+use crate::{errors::*, get_star};
 
 pub async fn list() -> JsonResult<Json<Vec<Planet>>> {
     let con_url = std::env::var("QEMU_URL").unwrap_or("qemu:///system".to_string());
@@ -16,3 +16,51 @@ pub async fn list() -> JsonResult<Json<Vec<Planet>>> {
     Ok(Json(inhabitants))
 }
 
+pub async fn get(Path(uuid): Path<String>) -> JsonResult<Json<Planet>> {
+
+    let con_url = std::env::var("QEMU_URL").unwrap_or("qemu:///system".to_string());
+    let mut star = Star::new(con_url)?;
+
+    if let Ok(p) = star.find_planet(uuid) {
+        return Ok(Json(p));
+    } else {
+        return Err(ServiceError::NotFound);
+    }
+}
+
+pub async fn shutdown(Path(uuid): Path<String>) -> NoneResult {
+    let con_url = std::env::var("QEMU_URL").unwrap_or("qemu:///system".to_string());
+    let mut star = Star::new(con_url)?;
+
+    if let Ok(p) = star.find_planet(uuid) {
+        p.shutdown()?;
+    } else {
+        return Err(ServiceError::NotFound);
+    }
+
+    Ok(())
+}
+
+pub async fn start(Path(uuid): Path<String>) -> NoneResult {
+    let mut s = get_star()?;
+
+    if let Ok(p) = s.find_planet(uuid) {
+        p.start()?;
+    } else {
+        return Err(ServiceError::NotFound);
+    }
+
+    Ok(())
+}
+
+pub async fn pause(Path(uuid): Path<String>) -> NoneResult {
+    let mut s = get_star()?;
+
+    if let Ok(p) = s.find_planet(uuid) {
+        p.pause()?;
+    } else {
+        return Err(ServiceError::NotFound);
+    }
+
+    Ok(())
+}
-- 
cgit v1.2.3