Merge pull request #28 from FxQnLr/0.3.4

0.3.4

1 files changed, 98 insertions, 8 deletions

diff --git a/src/routes/start.rs b/src/routes/start.rs
index ef6e8f2..ff3d1be 100644
--- a/src/routes/start.rs
+++ b/src/routes/start.rs
@@ -2,27 +2,28 @@ use crate::db::Device;
 use crate::error::Error;
 use crate::services::ping::Value as PingValue;
 use crate::wol::{create_buffer, send_packet};
-use axum::extract::State;
+use axum::extract::{Path, State};
 use axum::Json;
 use serde::{Deserialize, Serialize};
 use serde_json::{json, Value};
-use utoipa::ToSchema;
 use std::sync::Arc;
 use tracing::{debug, info};
+use utoipa::ToSchema;
 use uuid::Uuid;
 
 #[utoipa::path(
     post,
     path = "/start",
-    request_body = Payload,
+    request_body = PayloadOld,
     responses(
-        (status = 200, description = "List matching todos by query", body = [Response])
+        (status = 200, description = "DEP", body = [Response])
     ),
-    security(("api_key" = []))
+    security((), ("api_key" = []))
 )]
-pub async fn start(
+#[deprecated]
+pub async fn start_payload(
     State(state): State<Arc<crate::AppState>>,
-    Json(payload): Json<Payload>,
+    Json(payload): Json<PayloadOld>,
 ) -> Result<Json<Value>, Error> {
     info!("POST request");
     let device = sqlx::query_as!(
@@ -59,6 +60,89 @@ pub async fn start(
     })))
 }
 
+#[utoipa::path(
+    post,
+    path = "/start/{id}",
+    request_body = Option<Payload>,
+    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 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!(
+        Device,
+        r#"
+        SELECT id, mac, broadcast_addr, ip, times
+        FROM devices
+        WHERE id = $1;
+        "#,
+        id
+    )
+    .fetch_one(&state.db)
+    .await?;
+
+    info!("starting {}", device.id);
+
+    let bind_addr = "0.0.0.0:0";
+
+    let _ = send_packet(
+        bind_addr,
+        &device.broadcast_addr,
+        &create_buffer(&device.mac.to_string())?,
+    )?;
+    let dev_id = device.id.clone();
+    let uuid = if let Some(pl) = payload {
+        if pl.ping.is_some_and(|ping| ping) {
+            Some(setup_ping(state, device))
+        } else {
+            None
+        }
+    } else {
+        None
+    };
+
+    Ok(Json(json!(Response {
+        id: dev_id,
+        boot: true,
+        uuid
+    })))
+}
+
 fn setup_ping(state: Arc<crate::AppState>, device: Device) -> String {
     let mut uuid: Option<String> = None;
     for (key, value) in state.ping_map.clone() {
@@ -99,11 +183,17 @@ fn setup_ping(state: Arc<crate::AppState>, device: Device) -> String {
 }
 
 #[derive(Deserialize, ToSchema)]
-pub struct Payload {
+#[deprecated]
+pub struct PayloadOld {
     id: String,
     ping: Option<bool>,
 }
 
+#[derive(Deserialize, ToSchema)]
+pub struct Payload {
+    ping: Option<bool>,
+}
+
 #[derive(Serialize, ToSchema)]
 pub struct Response {
     id: String,