diff options
Diffstat (limited to 'src/routes')
-rw-r--r-- | src/routes/device.rs | 4 | ||||
-rw-r--r-- | src/routes/start.rs | 76 |
2 files changed, 72 insertions, 8 deletions
diff --git a/src/routes/device.rs b/src/routes/device.rs index d01d9f0..bbc832d 100644 --- a/src/routes/device.rs +++ b/src/routes/device.rs | |||
@@ -20,7 +20,7 @@ use utoipa::ToSchema; | |||
20 | security(("api_key" = [])) | 20 | security(("api_key" = [])) |
21 | )] | 21 | )] |
22 | #[deprecated] | 22 | #[deprecated] |
23 | pub async fn get( | 23 | pub async fn get_payload( |
24 | State(state): State<Arc<crate::AppState>>, | 24 | State(state): State<Arc<crate::AppState>>, |
25 | Json(payload): Json<GetDevicePayload>, | 25 | Json(payload): Json<GetDevicePayload>, |
26 | ) -> Result<Json<Value>, Error> { | 26 | ) -> Result<Json<Value>, Error> { |
@@ -53,7 +53,7 @@ pub async fn get( | |||
53 | ), | 53 | ), |
54 | security(("api_key" = [])) | 54 | security(("api_key" = [])) |
55 | )] | 55 | )] |
56 | pub async fn get_path( | 56 | pub async fn get( |
57 | State(state): State<Arc<crate::AppState>>, | 57 | State(state): State<Arc<crate::AppState>>, |
58 | Path(path): Path<String>, | 58 | Path(path): Path<String>, |
59 | ) -> Result<Json<Value>, Error> { | 59 | ) -> Result<Json<Value>, Error> { |
diff --git a/src/routes/start.rs b/src/routes/start.rs index ef6e8f2..fa226d8 100644 --- a/src/routes/start.rs +++ b/src/routes/start.rs | |||
@@ -2,27 +2,28 @@ use crate::db::Device; | |||
2 | use crate::error::Error; | 2 | use crate::error::Error; |
3 | use crate::services::ping::Value as PingValue; | 3 | use crate::services::ping::Value as PingValue; |
4 | use crate::wol::{create_buffer, send_packet}; | 4 | use crate::wol::{create_buffer, send_packet}; |
5 | use axum::extract::State; | 5 | use axum::extract::{Path, State}; |
6 | use axum::Json; | 6 | use axum::Json; |
7 | use serde::{Deserialize, Serialize}; | 7 | use serde::{Deserialize, Serialize}; |
8 | use serde_json::{json, Value}; | 8 | use serde_json::{json, Value}; |
9 | use utoipa::ToSchema; | ||
10 | use std::sync::Arc; | 9 | use std::sync::Arc; |
11 | use tracing::{debug, info}; | 10 | use tracing::{debug, info}; |
11 | use utoipa::ToSchema; | ||
12 | use uuid::Uuid; | 12 | use uuid::Uuid; |
13 | 13 | ||
14 | #[utoipa::path( | 14 | #[utoipa::path( |
15 | post, | 15 | post, |
16 | path = "/start", | 16 | path = "/start", |
17 | request_body = Payload, | 17 | request_body = PayloadOld, |
18 | responses( | 18 | responses( |
19 | (status = 200, description = "List matching todos by query", body = [Response]) | 19 | (status = 200, description = "List matching todos by query", body = [Response]) |
20 | ), | 20 | ), |
21 | security(("api_key" = [])) | 21 | security(("api_key" = [])) |
22 | )] | 22 | )] |
23 | pub async fn start( | 23 | #[deprecated] |
24 | pub async fn start_payload( | ||
24 | State(state): State<Arc<crate::AppState>>, | 25 | State(state): State<Arc<crate::AppState>>, |
25 | Json(payload): Json<Payload>, | 26 | Json(payload): Json<PayloadOld>, |
26 | ) -> Result<Json<Value>, Error> { | 27 | ) -> Result<Json<Value>, Error> { |
27 | info!("POST request"); | 28 | info!("POST request"); |
28 | let device = sqlx::query_as!( | 29 | let device = sqlx::query_as!( |
@@ -59,6 +60,63 @@ pub async fn start( | |||
59 | }))) | 60 | }))) |
60 | } | 61 | } |
61 | 62 | ||
63 | #[utoipa::path( | ||
64 | post, | ||
65 | path = "/start/{id}", | ||
66 | request_body = Payload, | ||
67 | responses( | ||
68 | (status = 200, description = "Start the device with the given id", body = [Response]) | ||
69 | ), | ||
70 | params( | ||
71 | ("id" = String, Path, description = "Device id") | ||
72 | ), | ||
73 | security(("api_key" = [])) | ||
74 | )] | ||
75 | pub async fn start( | ||
76 | State(state): State<Arc<crate::AppState>>, | ||
77 | Path(id): Path<String>, | ||
78 | payload: Option<Json<Payload>>, | ||
79 | ) -> Result<Json<Value>, Error> { | ||
80 | info!("Start request for {id}"); | ||
81 | let device = sqlx::query_as!( | ||
82 | Device, | ||
83 | r#" | ||
84 | SELECT id, mac, broadcast_addr, ip, times | ||
85 | FROM devices | ||
86 | WHERE id = $1; | ||
87 | "#, | ||
88 | id | ||
89 | ) | ||
90 | .fetch_one(&state.db) | ||
91 | .await?; | ||
92 | |||
93 | info!("starting {}", device.id); | ||
94 | |||
95 | let bind_addr = "0.0.0.0:0"; | ||
96 | |||
97 | let _ = send_packet( | ||
98 | bind_addr, | ||
99 | &device.broadcast_addr, | ||
100 | &create_buffer(&device.mac.to_string())?, | ||
101 | )?; | ||
102 | let dev_id = device.id.clone(); | ||
103 | let uuid = if let Some(pl) = payload { | ||
104 | if pl.ping.is_some_and(|ping| ping) { | ||
105 | Some(setup_ping(state, device)) | ||
106 | } else { | ||
107 | None | ||
108 | } | ||
109 | } else { | ||
110 | None | ||
111 | }; | ||
112 | |||
113 | Ok(Json(json!(Response { | ||
114 | id: dev_id, | ||
115 | boot: true, | ||
116 | uuid | ||
117 | }))) | ||
118 | } | ||
119 | |||
62 | fn setup_ping(state: Arc<crate::AppState>, device: Device) -> String { | 120 | fn setup_ping(state: Arc<crate::AppState>, device: Device) -> String { |
63 | let mut uuid: Option<String> = None; | 121 | let mut uuid: Option<String> = None; |
64 | for (key, value) in state.ping_map.clone() { | 122 | for (key, value) in state.ping_map.clone() { |
@@ -99,11 +157,17 @@ fn setup_ping(state: Arc<crate::AppState>, device: Device) -> String { | |||
99 | } | 157 | } |
100 | 158 | ||
101 | #[derive(Deserialize, ToSchema)] | 159 | #[derive(Deserialize, ToSchema)] |
102 | pub struct Payload { | 160 | #[deprecated] |
161 | pub struct PayloadOld { | ||
103 | id: String, | 162 | id: String, |
104 | ping: Option<bool>, | 163 | ping: Option<bool>, |
105 | } | 164 | } |
106 | 165 | ||
166 | #[derive(Deserialize, ToSchema)] | ||
167 | pub struct Payload { | ||
168 | ping: Option<bool>, | ||
169 | } | ||
170 | |||
107 | #[derive(Serialize, ToSchema)] | 171 | #[derive(Serialize, ToSchema)] |
108 | pub struct Response { | 172 | pub struct Response { |
109 | id: String, | 173 | id: String, |