diff options
author | FxQnLr <[email protected]> | 2024-02-11 21:17:58 +0100 |
---|---|---|
committer | FxQnLr <[email protected]> | 2024-02-11 21:17:58 +0100 |
commit | e4832b4cf36ba0eaed298ee458498eddd7176590 (patch) | |
tree | 98cb6e1e1d97f69be4f8379edc8c33f6eda4e621 /src/routes | |
parent | 6f94a825485b1d622d8ebe88ddf3e2f426d1c69c (diff) | |
download | webol-e4832b4cf36ba0eaed298ee458498eddd7176590.tar webol-e4832b4cf36ba0eaed298ee458498eddd7176590.tar.gz webol-e4832b4cf36ba0eaed298ee458498eddd7176590.zip |
fix clippy
Diffstat (limited to 'src/routes')
-rw-r--r-- | src/routes/device.rs | 26 | ||||
-rw-r--r-- | src/routes/start.rs | 72 | ||||
-rw-r--r-- | src/routes/status.rs | 2 |
3 files changed, 63 insertions, 37 deletions
diff --git a/src/routes/device.rs b/src/routes/device.rs index a3308d4..b80cb85 100644 --- a/src/routes/device.rs +++ b/src/routes/device.rs | |||
@@ -7,12 +7,12 @@ use serde_json::{json, Value}; | |||
7 | use tracing::{debug, info}; | 7 | use tracing::{debug, info}; |
8 | use crate::auth::auth; | 8 | use crate::auth::auth; |
9 | use crate::db::Device; | 9 | use crate::db::Device; |
10 | use crate::error::WebolError; | 10 | use crate::error::Error; |
11 | 11 | ||
12 | pub async fn get_device(State(state): State<Arc<crate::AppState>>, headers: HeaderMap, Json(payload): Json<GetDevicePayload>) -> Result<Json<Value>, WebolError> { | 12 | pub async fn get(State(state): State<Arc<crate::AppState>>, headers: HeaderMap, Json(payload): Json<GetDevicePayload>) -> Result<Json<Value>, Error> { |
13 | info!("add device {}", payload.id); | 13 | info!("add device {}", payload.id); |
14 | let secret = headers.get("authorization"); | 14 | let secret = headers.get("authorization"); |
15 | if auth(secret).map_err(WebolError::Auth)? { | 15 | if auth(secret).map_err(Error::Auth)? { |
16 | let device = sqlx::query_as!( | 16 | let device = sqlx::query_as!( |
17 | Device, | 17 | Device, |
18 | r#" | 18 | r#" |
@@ -21,13 +21,13 @@ pub async fn get_device(State(state): State<Arc<crate::AppState>>, headers: Head | |||
21 | WHERE id = $1; | 21 | WHERE id = $1; |
22 | "#, | 22 | "#, |
23 | payload.id | 23 | payload.id |
24 | ).fetch_one(&state.db).await.map_err(WebolError::DB)?; | 24 | ).fetch_one(&state.db).await.map_err(Error::DB)?; |
25 | 25 | ||
26 | debug!("got device {:?}", device); | 26 | debug!("got device {:?}", device); |
27 | 27 | ||
28 | Ok(Json(json!(device))) | 28 | Ok(Json(json!(device))) |
29 | } else { | 29 | } else { |
30 | Err(WebolError::Generic) | 30 | Err(Error::Generic) |
31 | } | 31 | } |
32 | } | 32 | } |
33 | 33 | ||
@@ -36,10 +36,10 @@ pub struct GetDevicePayload { | |||
36 | id: String, | 36 | id: String, |
37 | } | 37 | } |
38 | 38 | ||
39 | pub async fn put_device(State(state): State<Arc<crate::AppState>>, headers: HeaderMap, Json(payload): Json<PutDevicePayload>) -> Result<Json<Value>, WebolError> { | 39 | pub async fn put(State(state): State<Arc<crate::AppState>>, headers: HeaderMap, Json(payload): Json<PutDevicePayload>) -> Result<Json<Value>, Error> { |
40 | info!("add device {} ({}, {}, {})", payload.id, payload.mac, payload.broadcast_addr, payload.ip); | 40 | info!("add device {} ({}, {}, {})", payload.id, payload.mac, payload.broadcast_addr, payload.ip); |
41 | let secret = headers.get("authorization"); | 41 | let secret = headers.get("authorization"); |
42 | if auth(secret).map_err(WebolError::Auth)? { | 42 | if auth(secret).map_err(Error::Auth)? { |
43 | sqlx::query!( | 43 | sqlx::query!( |
44 | r#" | 44 | r#" |
45 | INSERT INTO devices (id, mac, broadcast_addr, ip) | 45 | INSERT INTO devices (id, mac, broadcast_addr, ip) |
@@ -49,11 +49,11 @@ pub async fn put_device(State(state): State<Arc<crate::AppState>>, headers: Head | |||
49 | payload.mac, | 49 | payload.mac, |
50 | payload.broadcast_addr, | 50 | payload.broadcast_addr, |
51 | payload.ip | 51 | payload.ip |
52 | ).execute(&state.db).await.map_err(WebolError::DB)?; | 52 | ).execute(&state.db).await.map_err(Error::DB)?; |
53 | 53 | ||
54 | Ok(Json(json!(PutDeviceResponse { success: true }))) | 54 | Ok(Json(json!(PutDeviceResponse { success: true }))) |
55 | } else { | 55 | } else { |
56 | Err(WebolError::Generic) | 56 | Err(Error::Generic) |
57 | } | 57 | } |
58 | } | 58 | } |
59 | 59 | ||
@@ -70,10 +70,10 @@ pub struct PutDeviceResponse { | |||
70 | success: bool | 70 | success: bool |
71 | } | 71 | } |
72 | 72 | ||
73 | pub async fn post_device(State(state): State<Arc<crate::AppState>>, headers: HeaderMap, Json(payload): Json<PostDevicePayload>) -> Result<Json<Value>, WebolError> { | 73 | pub async fn post(State(state): State<Arc<crate::AppState>>, headers: HeaderMap, Json(payload): Json<PostDevicePayload>) -> Result<Json<Value>, Error> { |
74 | info!("edit device {} ({}, {}, {})", payload.id, payload.mac, payload.broadcast_addr, payload.ip); | 74 | info!("edit device {} ({}, {}, {})", payload.id, payload.mac, payload.broadcast_addr, payload.ip); |
75 | let secret = headers.get("authorization"); | 75 | let secret = headers.get("authorization"); |
76 | if auth(secret).map_err(WebolError::Auth)? { | 76 | if auth(secret).map_err(Error::Auth)? { |
77 | let device = sqlx::query_as!( | 77 | let device = sqlx::query_as!( |
78 | Device, | 78 | Device, |
79 | r#" | 79 | r#" |
@@ -85,11 +85,11 @@ pub async fn post_device(State(state): State<Arc<crate::AppState>>, headers: Hea | |||
85 | payload.broadcast_addr, | 85 | payload.broadcast_addr, |
86 | payload.ip, | 86 | payload.ip, |
87 | payload.id | 87 | payload.id |
88 | ).fetch_one(&state.db).await.map_err(WebolError::DB)?; | 88 | ).fetch_one(&state.db).await.map_err(Error::DB)?; |
89 | 89 | ||
90 | Ok(Json(json!(device))) | 90 | Ok(Json(json!(device))) |
91 | } else { | 91 | } else { |
92 | Err(WebolError::Generic) | 92 | Err(Error::Generic) |
93 | } | 93 | } |
94 | } | 94 | } |
95 | 95 | ||
diff --git a/src/routes/start.rs b/src/routes/start.rs index a206cbd..4264588 100644 --- a/src/routes/start.rs +++ b/src/routes/start.rs | |||
@@ -1,23 +1,27 @@ | |||
1 | use axum::Json; | 1 | use crate::auth::auth; |
2 | use crate::config::SETTINGS; | ||
3 | use crate::db::Device; | ||
4 | use crate::error::Error; | ||
5 | use crate::services::ping::Value as PingValue; | ||
6 | use crate::wol::{create_buffer, send_packet}; | ||
7 | use axum::extract::State; | ||
2 | use axum::http::HeaderMap; | 8 | use axum::http::HeaderMap; |
9 | use axum::Json; | ||
3 | use serde::{Deserialize, Serialize}; | 10 | use serde::{Deserialize, Serialize}; |
4 | use std::sync::Arc; | ||
5 | use axum::extract::State; | ||
6 | use serde_json::{json, Value}; | 11 | use serde_json::{json, Value}; |
12 | use std::sync::Arc; | ||
7 | use tracing::{debug, info}; | 13 | use tracing::{debug, info}; |
8 | use uuid::Uuid; | 14 | 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 | 15 | ||
16 | #[axum_macros::debug_handler] | 16 | #[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> { | 17 | pub async fn start( |
18 | State(state): State<Arc<crate::AppState>>, | ||
19 | headers: HeaderMap, | ||
20 | Json(payload): Json<Payload>, | ||
21 | ) -> Result<Json<Value>, Error> { | ||
18 | info!("POST request"); | 22 | info!("POST request"); |
19 | let secret = headers.get("authorization"); | 23 | let secret = headers.get("authorization"); |
20 | let authorized = auth(secret).map_err(WebolError::Auth)?; | 24 | let authorized = auth(secret).map_err(Error::Auth)?; |
21 | if authorized { | 25 | if authorized { |
22 | let device = sqlx::query_as!( | 26 | let device = sqlx::query_as!( |
23 | Device, | 27 | Device, |
@@ -27,7 +31,10 @@ pub async fn start(State(state): State<Arc<crate::AppState>>, headers: HeaderMap | |||
27 | WHERE id = $1; | 31 | WHERE id = $1; |
28 | "#, | 32 | "#, |
29 | payload.id | 33 | payload.id |
30 | ).fetch_one(&state.db).await.map_err(WebolError::DB)?; | 34 | ) |
35 | .fetch_one(&state.db) | ||
36 | .await | ||
37 | .map_err(Error::DB)?; | ||
31 | 38 | ||
32 | info!("starting {}", device.id); | 39 | info!("starting {}", device.id); |
33 | 40 | ||
@@ -36,9 +43,9 @@ pub async fn start(State(state): State<Arc<crate::AppState>>, headers: HeaderMap | |||
36 | .unwrap_or("0.0.0.0:1111".to_string()); | 43 | .unwrap_or("0.0.0.0:1111".to_string()); |
37 | 44 | ||
38 | let _ = send_packet( | 45 | let _ = send_packet( |
39 | &bind_addr.parse().map_err(WebolError::IpParse)?, | 46 | &bind_addr.parse().map_err(Error::IpParse)?, |
40 | &device.broadcast_addr.parse().map_err(WebolError::IpParse)?, | 47 | &device.broadcast_addr.parse().map_err(Error::IpParse)?, |
41 | create_buffer(&device.mac)? | 48 | &create_buffer(&device.mac)?, |
42 | )?; | 49 | )?; |
43 | let dev_id = device.id.clone(); | 50 | let dev_id = device.id.clone(); |
44 | let uuid = if payload.ping.is_some_and(|ping| ping) { | 51 | let uuid = if payload.ping.is_some_and(|ping| ping) { |
@@ -49,7 +56,7 @@ pub async fn start(State(state): State<Arc<crate::AppState>>, headers: HeaderMap | |||
49 | uuid = Some(key); | 56 | uuid = Some(key); |
50 | break; | 57 | break; |
51 | } | 58 | } |
52 | }; | 59 | } |
53 | let uuid_gen = match uuid { | 60 | let uuid_gen = match uuid { |
54 | Some(u) => u, | 61 | Some(u) => u, |
55 | None => Uuid::new_v4().to_string(), | 62 | None => Uuid::new_v4().to_string(), |
@@ -58,26 +65,45 @@ pub async fn start(State(state): State<Arc<crate::AppState>>, headers: HeaderMap | |||
58 | 65 | ||
59 | tokio::spawn(async move { | 66 | tokio::spawn(async move { |
60 | debug!("init ping service"); | 67 | debug!("init ping service"); |
61 | state.ping_map.insert(uuid_gen.clone(), PingValue { ip: device.ip.clone(), online: false }); | 68 | state.ping_map.insert( |
69 | uuid_gen.clone(), | ||
70 | PingValue { | ||
71 | ip: device.ip.clone(), | ||
72 | online: false, | ||
73 | }, | ||
74 | ); | ||
62 | 75 | ||
63 | crate::services::ping::spawn(state.ping_send.clone(), device, uuid_gen.clone(), &state.ping_map, &state.db).await; | 76 | crate::services::ping::spawn( |
77 | state.ping_send.clone(), | ||
78 | device, | ||
79 | uuid_gen.clone(), | ||
80 | &state.ping_map, | ||
81 | &state.db, | ||
82 | ) | ||
83 | .await; | ||
64 | }); | 84 | }); |
65 | Some(uuid_genc) | 85 | Some(uuid_genc) |
66 | } else { None }; | 86 | } else { |
67 | Ok(Json(json!(StartResponse { id: dev_id, boot: true, uuid }))) | 87 | None |
88 | }; | ||
89 | Ok(Json(json!(Response { | ||
90 | id: dev_id, | ||
91 | boot: true, | ||
92 | uuid | ||
93 | }))) | ||
68 | } else { | 94 | } else { |
69 | Err(WebolError::Generic) | 95 | Err(Error::Generic) |
70 | } | 96 | } |
71 | } | 97 | } |
72 | 98 | ||
73 | #[derive(Deserialize)] | 99 | #[derive(Deserialize)] |
74 | pub struct StartPayload { | 100 | pub struct Payload { |
75 | id: String, | 101 | id: String, |
76 | ping: Option<bool>, | 102 | ping: Option<bool>, |
77 | } | 103 | } |
78 | 104 | ||
79 | #[derive(Serialize)] | 105 | #[derive(Serialize)] |
80 | struct StartResponse { | 106 | struct Response { |
81 | id: String, | 107 | id: String, |
82 | boot: bool, | 108 | boot: bool, |
83 | uuid: Option<String>, | 109 | uuid: Option<String>, |
diff --git a/src/routes/status.rs b/src/routes/status.rs index 45f3e51..31ef996 100644 --- a/src/routes/status.rs +++ b/src/routes/status.rs | |||
@@ -7,4 +7,4 @@ use crate::services::ping::status_websocket; | |||
7 | #[axum_macros::debug_handler] | 7 | #[axum_macros::debug_handler] |
8 | pub async fn status(State(state): State<Arc<AppState>>, ws: WebSocketUpgrade) -> Response { | 8 | pub async fn status(State(state): State<Arc<AppState>>, ws: WebSocketUpgrade) -> Response { |
9 | ws.on_upgrade(move |socket| status_websocket(socket, state)) | 9 | ws.on_upgrade(move |socket| status_websocket(socket, state)) |
10 | } \ No newline at end of file | 10 | } |