summaryrefslogtreecommitdiff
path: root/src/routes
diff options
context:
space:
mode:
authorFxQnLr <[email protected]>2024-02-11 21:17:58 +0100
committerFxQnLr <[email protected]>2024-02-11 21:17:58 +0100
commite4832b4cf36ba0eaed298ee458498eddd7176590 (patch)
tree98cb6e1e1d97f69be4f8379edc8c33f6eda4e621 /src/routes
parent6f94a825485b1d622d8ebe88ddf3e2f426d1c69c (diff)
downloadwebol-e4832b4cf36ba0eaed298ee458498eddd7176590.tar
webol-e4832b4cf36ba0eaed298ee458498eddd7176590.tar.gz
webol-e4832b4cf36ba0eaed298ee458498eddd7176590.zip
fix clippy
Diffstat (limited to 'src/routes')
-rw-r--r--src/routes/device.rs26
-rw-r--r--src/routes/start.rs72
-rw-r--r--src/routes/status.rs2
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};
7use tracing::{debug, info}; 7use tracing::{debug, info};
8use crate::auth::auth; 8use crate::auth::auth;
9use crate::db::Device; 9use crate::db::Device;
10use crate::error::WebolError; 10use crate::error::Error;
11 11
12pub async fn get_device(State(state): State<Arc<crate::AppState>>, headers: HeaderMap, Json(payload): Json<GetDevicePayload>) -> Result<Json<Value>, WebolError> { 12pub 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
39pub async fn put_device(State(state): State<Arc<crate::AppState>>, headers: HeaderMap, Json(payload): Json<PutDevicePayload>) -> Result<Json<Value>, WebolError> { 39pub 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
73pub async fn post_device(State(state): State<Arc<crate::AppState>>, headers: HeaderMap, Json(payload): Json<PostDevicePayload>) -> Result<Json<Value>, WebolError> { 73pub 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 @@
1use axum::Json; 1use crate::auth::auth;
2use crate::config::SETTINGS;
3use crate::db::Device;
4use crate::error::Error;
5use crate::services::ping::Value as PingValue;
6use crate::wol::{create_buffer, send_packet};
7use axum::extract::State;
2use axum::http::HeaderMap; 8use axum::http::HeaderMap;
9use axum::Json;
3use serde::{Deserialize, Serialize}; 10use serde::{Deserialize, Serialize};
4use std::sync::Arc;
5use axum::extract::State;
6use serde_json::{json, Value}; 11use serde_json::{json, Value};
12use std::sync::Arc;
7use tracing::{debug, info}; 13use tracing::{debug, info};
8use uuid::Uuid; 14use uuid::Uuid;
9use crate::auth::auth;
10use crate::config::SETTINGS;
11use crate::wol::{create_buffer, send_packet};
12use crate::db::Device;
13use crate::error::WebolError;
14use crate::services::ping::PingValue;
15 15
16#[axum_macros::debug_handler] 16#[axum_macros::debug_handler]
17pub async fn start(State(state): State<Arc<crate::AppState>>, headers: HeaderMap, Json(payload): Json<StartPayload>) -> Result<Json<Value>, WebolError> { 17pub 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)]
74pub struct StartPayload { 100pub 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)]
80struct StartResponse { 106struct 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]
8pub async fn status(State(state): State<Arc<AppState>>, ws: WebSocketUpgrade) -> Response { 8pub 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}