diff options
Diffstat (limited to 'src/routes/start.rs')
-rw-r--r-- | src/routes/start.rs | 76 |
1 files changed, 50 insertions, 26 deletions
diff --git a/src/routes/start.rs b/src/routes/start.rs index 1555db3..ce95bf3 100644 --- a/src/routes/start.rs +++ b/src/routes/start.rs | |||
@@ -1,23 +1,26 @@ | |||
1 | use axum::headers::HeaderMap; | 1 | use crate::auth::auth; |
2 | use crate::db::Device; | ||
3 | use crate::error::Error; | ||
4 | use crate::services::ping::Value as PingValue; | ||
5 | use crate::wol::{create_buffer, send_packet}; | ||
6 | use axum::extract::State; | ||
7 | use axum::http::HeaderMap; | ||
2 | use axum::Json; | 8 | use axum::Json; |
3 | use serde::{Deserialize, Serialize}; | 9 | use serde::{Deserialize, Serialize}; |
4 | use std::sync::Arc; | ||
5 | use axum::extract::State; | ||
6 | use serde_json::{json, Value}; | 10 | use serde_json::{json, Value}; |
11 | use std::sync::Arc; | ||
7 | use tracing::{debug, info}; | 12 | use tracing::{debug, info}; |
8 | use uuid::Uuid; | 13 | use uuid::Uuid; |
9 | use crate::auth::auth; | ||
10 | use crate::config::SETTINGS; | ||
11 | use crate::wol::{create_buffer, send_packet}; | ||
12 | use crate::db::Device; | ||
13 | use crate::error::WebolError; | ||
14 | use crate::services::ping::PingValue; | ||
15 | 14 | ||
16 | #[axum_macros::debug_handler] | 15 | #[axum_macros::debug_handler] |
17 | pub async fn start(State(state): State<Arc<crate::AppState>>, headers: HeaderMap, Json(payload): Json<StartPayload>) -> Result<Json<Value>, WebolError> { | 16 | pub async fn start( |
17 | State(state): State<Arc<crate::AppState>>, | ||
18 | headers: HeaderMap, | ||
19 | Json(payload): Json<Payload>, | ||
20 | ) -> Result<Json<Value>, Error> { | ||
18 | info!("POST request"); | 21 | info!("POST request"); |
19 | let secret = headers.get("authorization"); | 22 | let secret = headers.get("authorization"); |
20 | let authorized = auth(secret).map_err(WebolError::Auth)?; | 23 | let authorized = auth(&state.config, secret).map_err(Error::Auth)?; |
21 | if authorized { | 24 | if authorized { |
22 | let device = sqlx::query_as!( | 25 | let device = sqlx::query_as!( |
23 | Device, | 26 | Device, |
@@ -27,18 +30,19 @@ pub async fn start(State(state): State<Arc<crate::AppState>>, headers: HeaderMap | |||
27 | WHERE id = $1; | 30 | WHERE id = $1; |
28 | "#, | 31 | "#, |
29 | payload.id | 32 | payload.id |
30 | ).fetch_one(&state.db).await.map_err(WebolError::DB)?; | 33 | ) |
34 | .fetch_one(&state.db) | ||
35 | .await | ||
36 | .map_err(Error::DB)?; | ||
31 | 37 | ||
32 | info!("starting {}", device.id); | 38 | info!("starting {}", device.id); |
33 | 39 | ||
34 | let bind_addr = SETTINGS | 40 | let bind_addr = "0.0.0.0:0"; |
35 | .get_string("bindaddr") | ||
36 | .unwrap_or("0.0.0.0:1111".to_string()); | ||
37 | 41 | ||
38 | let _ = send_packet( | 42 | let _ = send_packet( |
39 | &bind_addr.parse().map_err(WebolError::IpParse)?, | 43 | &bind_addr.parse().map_err(Error::IpParse)?, |
40 | &device.broadcast_addr.parse().map_err(WebolError::IpParse)?, | 44 | &device.broadcast_addr.parse().map_err(Error::IpParse)?, |
41 | create_buffer(&device.mac)? | 45 | &create_buffer(&device.mac)?, |
42 | )?; | 46 | )?; |
43 | let dev_id = device.id.clone(); | 47 | let dev_id = device.id.clone(); |
44 | let uuid = if payload.ping.is_some_and(|ping| ping) { | 48 | let uuid = if payload.ping.is_some_and(|ping| ping) { |
@@ -49,7 +53,7 @@ pub async fn start(State(state): State<Arc<crate::AppState>>, headers: HeaderMap | |||
49 | uuid = Some(key); | 53 | uuid = Some(key); |
50 | break; | 54 | break; |
51 | } | 55 | } |
52 | }; | 56 | } |
53 | let uuid_gen = match uuid { | 57 | let uuid_gen = match uuid { |
54 | Some(u) => u, | 58 | Some(u) => u, |
55 | None => Uuid::new_v4().to_string(), | 59 | None => Uuid::new_v4().to_string(), |
@@ -58,26 +62,46 @@ pub async fn start(State(state): State<Arc<crate::AppState>>, headers: HeaderMap | |||
58 | 62 | ||
59 | tokio::spawn(async move { | 63 | tokio::spawn(async move { |
60 | debug!("init ping service"); | 64 | debug!("init ping service"); |
61 | state.ping_map.insert(uuid_gen.clone(), PingValue { ip: device.ip.clone(), online: false }); | 65 | state.ping_map.insert( |
66 | uuid_gen.clone(), | ||
67 | PingValue { | ||
68 | ip: device.ip.clone(), | ||
69 | online: false, | ||
70 | }, | ||
71 | ); | ||
62 | 72 | ||
63 | crate::services::ping::spawn(state.ping_send.clone(), device, uuid_gen.clone(), &state.ping_map, &state.db).await | 73 | crate::services::ping::spawn( |
74 | state.ping_send.clone(), | ||
75 | &state.config, | ||
76 | device, | ||
77 | uuid_gen.clone(), | ||
78 | &state.ping_map, | ||
79 | &state.db, | ||
80 | ) | ||
81 | .await; | ||
64 | }); | 82 | }); |
65 | Some(uuid_genc) | 83 | Some(uuid_genc) |
66 | } else { None }; | 84 | } else { |
67 | Ok(Json(json!(StartResponse { id: dev_id, boot: true, uuid }))) | 85 | None |
86 | }; | ||
87 | Ok(Json(json!(Response { | ||
88 | id: dev_id, | ||
89 | boot: true, | ||
90 | uuid | ||
91 | }))) | ||
68 | } else { | 92 | } else { |
69 | Err(WebolError::Generic) | 93 | Err(Error::Generic) |
70 | } | 94 | } |
71 | } | 95 | } |
72 | 96 | ||
73 | #[derive(Deserialize)] | 97 | #[derive(Deserialize)] |
74 | pub struct StartPayload { | 98 | pub struct Payload { |
75 | id: String, | 99 | id: String, |
76 | ping: Option<bool>, | 100 | ping: Option<bool>, |
77 | } | 101 | } |
78 | 102 | ||
79 | #[derive(Serialize)] | 103 | #[derive(Serialize)] |
80 | struct StartResponse { | 104 | struct Response { |
81 | id: String, | 105 | id: String, |
82 | boot: bool, | 106 | boot: bool, |
83 | uuid: Option<String>, | 107 | uuid: Option<String>, |