From a91a2ca5c88403e905bf0f798393587fc4d900fa Mon Sep 17 00:00:00 2001
From: FxQnLr <felixquinn03@gmail.com>
Date: Mon, 8 Apr 2024 15:44:31 +0200
Subject: Closes #26. Addtional GET request for /start

---
 src/auth.rs         |  6 ++++--
 src/main.rs         |  6 ++++--
 src/routes/start.rs | 32 +++++++++++++++++++++++++++++---
 3 files changed, 37 insertions(+), 7 deletions(-)

(limited to 'src')

diff --git a/src/auth.rs b/src/auth.rs
index 1f4518a..74008b5 100644
--- a/src/auth.rs
+++ b/src/auth.rs
@@ -23,11 +23,13 @@ pub async fn auth(
     match auth.method {
         Methods::Key => {
             if let Some(secret) = headers.get("authorization") {
-                if !(auth.secret.as_str() == secret) { return Err(StatusCode::UNAUTHORIZED); };
+                if auth.secret.as_str() != secret {
+                    return Err(StatusCode::UNAUTHORIZED);
+                };
                 let response = next.run(request).await;
                 Ok(response)
             } else {
-                return Err(StatusCode::UNAUTHORIZED);
+                Err(StatusCode::UNAUTHORIZED)
             }
         }
         Methods::None => Ok(next.run(request).await),
diff --git a/src/main.rs b/src/main.rs
index 43957ff..a8acc5f 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -37,7 +37,8 @@ mod wol;
 #[derive(OpenApi)]
 #[openapi(
     paths(
-        start::start,
+        start::post,
+        start::get,
         start::start_payload,
         device::get,
         device::get_payload,
@@ -119,13 +120,14 @@ async fn main() -> color_eyre::eyre::Result<()> {
 
     let app = Router::new()
         .route("/start", post(start::start_payload))
-        .route("/start/:id", post(start::start))
+        .route("/start/:id", post(start::post).get(start::get))
         .route(
             "/device",
             post(device::post).get(device::get_payload).put(device::put),
         )
         .route("/device/:id", get(device::get))
         .route("/status", get(status::status))
+        // TODO: Don't load on `None` Auth
         .route_layer(from_fn_with_state(shared_state.clone(), auth::auth))
         .merge(SwaggerUi::new("/swagger-ui").url("/api-docs/openapi.json", ApiDoc::openapi()))
         .with_state(Arc::new(shared_state));
diff --git a/src/routes/start.rs b/src/routes/start.rs
index c61d5a3..e74a943 100644
--- a/src/routes/start.rs
+++ b/src/routes/start.rs
@@ -18,7 +18,7 @@ use uuid::Uuid;
     responses(
         (status = 200, description = "List matching todos by query", body = [Response])
     ),
-    security(("api_key" = []))
+    security((), ("api_key" = []))
 )]
 #[deprecated]
 pub async fn start_payload(
@@ -70,12 +70,38 @@ pub async fn start_payload(
     params(
         ("id" = String, Path, description = "Device id")
     ),
-    security(("api_key" = []))
+    security((), ("api_key" = []))
 )]
-pub async fn start(
+pub async fn post(
     State(state): State<Arc<crate::AppState>>,
     Path(id): Path<String>,
     payload: Option<Json<Payload>>,
+) -> Result<Json<Value>, Error> {
+    send_wol(state, &id, payload).await
+}
+
+#[utoipa::path(
+    get,
+    path = "/start/{id}",
+    responses(
+        (status = 200, description = "Start the device with the given id", body = [Response])
+    ),
+    params(
+        ("id" = String, Path, description = "Device id")
+    ),
+    security((), ("api_key" = []))
+)]
+pub async fn get(
+    State(state): State<Arc<crate::AppState>>,
+    Path(id): Path<String>,
+) -> Result<Json<Value>, Error> {
+    send_wol(state, &id, None).await
+}
+
+async fn send_wol(
+    state: Arc<crate::AppState>,
+    id: &str,
+    payload: Option<Json<Payload>>,
 ) -> Result<Json<Value>, Error> {
     info!("Start request for {id}");
     let device = sqlx::query_as!(
-- 
cgit v1.2.3