3 files changed, 107 insertions, 40 deletions

diff --git a/src/routes/device.rs b/src/routes/device.rs
new file mode 100644
index 0000000..d5d7144
--- /dev/null
+++ b/src/routes/device.rs
@@ -0,0 +1,92 @@
+use std::sync::Arc;
+use axum::extract::State;
+use axum::headers::HeaderMap;
+use axum::Json;
+use serde::{Deserialize, Serialize};
+use serde_json::{json, Value};
+use crate::auth::auth;
+use crate::db::Device;
+use crate::error::WebolError;
+
+pub async fn get_device(State(state): State<Arc<crate::AppState>>, headers: HeaderMap, Json(payload): Json<GetDevicePayload>) -> Result<Json<Value>, WebolError> {
+    let secret = headers.get("authorization");
+    if auth(secret).map_err(WebolError::Auth)? {
+        let device = sqlx::query_as!(
+            Device,
+            r#"
+            SELECT id, mac, broadcast_addr
+            FROM devices
+            WHERE id = $1;
+            "#,
+            payload.id
+        ).fetch_one(&state.db).await.map_err(|err| WebolError::Server(Box::new(err)))?;
+
+        Ok(Json(json!(device)))
+    } else {
+        Err(WebolError::Generic)
+    }
+}
+
+#[derive(Deserialize)]
+pub struct GetDevicePayload {
+    id: String,
+}
+
+pub async fn put_device(State(state): State<Arc<crate::AppState>>, headers: HeaderMap, Json(payload): Json<PutDevicePayload>) -> Result<Json<Value>, WebolError> {
+    let secret = headers.get("authorization");
+    if auth(secret).map_err(WebolError::Auth)? {
+        sqlx::query!(
+            r#"
+            INSERT INTO devices (id, mac, broadcast_addr)
+            VALUES ($1, $2, $3);
+            "#,
+            payload.id,
+            payload.mac,
+            payload.broadcast_addr
+        ).execute(&state.db).await.map_err(|err| WebolError::Server(Box::new(err)))?;
+
+        Ok(Json(json!(PutDeviceResponse { success: true })))
+    } else {
+        Err(WebolError::Generic)
+    }
+}
+
+#[derive(Deserialize)]
+pub struct PutDevicePayload {
+    id: String,
+    mac: String,
+    broadcast_addr: String,
+}
+
+#[derive(Serialize)]
+pub struct PutDeviceResponse {
+    success: bool
+}
+
+pub async fn post_device(State(state): State<Arc<crate::AppState>>, headers: HeaderMap, Json(payload): Json<PostDevicePayload>) -> Result<Json<Value>, WebolError> {
+    let secret = headers.get("authorization");
+    if auth(secret).map_err(WebolError::Auth)? {
+        let device = sqlx::query_as!(
+            Device,
+            r#"
+            UPDATE devices
+            SET mac = $1, broadcast_addr = $2 WHERE id = $3
+            RETURNING id, mac, broadcast_addr;
+            "#,
+            payload.mac,
+            payload.broadcast_addr,
+            payload.id
+        ).fetch_one(&state.db).await.map_err(|err| WebolError::Server(Box::new(err)))?;
+
+        Ok(Json(json!(device)))
+    } else {
+        Err(WebolError::Generic)
+    }
+}
+
+#[derive(Deserialize)]
+pub struct PostDevicePayload {
+    id: String,
+    mac: String,
+    broadcast_addr: String,
+}
\ No newline at end of file
diff --git a/src/routes/mod.rs b/src/routes/mod.rs
index 78d4375..12fbfab 100644
--- a/src/routes/mod.rs
+++ b/src/routes/mod.rs
@@ -1 +1,2 @@
-pub mod start;
\ No newline at end of file
+pub mod start;
+pub mod device;
\ No newline at end of file
diff --git a/src/routes/start.rs b/src/routes/start.rs
index 2d505fc..d16ea4e 100644
--- a/src/routes/start.rs
+++ b/src/routes/start.rs
@@ -1,45 +1,43 @@
 use axum::headers::HeaderMap;
-use axum::http::StatusCode;
 use axum::Json;
-use axum::response::{IntoResponse, Response};
 use serde::{Deserialize, Serialize};
-use std::error::Error;
 use std::sync::Arc;
 use axum::extract::State;
 use serde_json::{json, Value};
-use tracing::{error, info};
-use crate::auth::{auth, AuthError};
+use tracing::info;
+use crate::auth::auth;
 use crate::config::SETTINGS;
 use crate::wol::{create_buffer, send_packet};
 use crate::db::Device;
+use crate::error::WebolError;
 
-pub async fn start(State(state): State<Arc<crate::AppState>>, headers: HeaderMap, Json(payload): Json<StartPayload>) -> Result<Json<Value>, StartError> {
+pub async fn start(State(state): State<Arc<crate::AppState>>, headers: HeaderMap, Json(payload): Json<StartPayload>) -> Result<Json<Value>, WebolError> {
     let secret = headers.get("authorization");
-    if auth(secret).map_err(StartError::Auth)? {
+    if auth(secret).map_err(WebolError::Auth)? {
         let device = sqlx::query_as!(
             Device,
             r#"
             SELECT id, mac, broadcast_addr
             FROM devices
-            WHERE id = ?1;
+            WHERE id = $1;
             "#,
             payload.id
-        ).fetch_one(&state.db).await.map_err(|err| StartError::Server(Box::new(err)))?;
+        ).fetch_one(&state.db).await.map_err(|err| WebolError::Server(Box::new(err)))?;
 
         info!("starting {}", device.id);
 
         let bind_addr = SETTINGS
             .get_string("bindaddr")
-            .map_err(|err| StartError::Server(Box::new(err)))?;
+            .map_err(|err| WebolError::Server(Box::new(err)))?;
 
         let _ = send_packet(
-            &bind_addr.parse().map_err(|err| StartError::Server(Box::new(err)))?,
-            &device.broadcast_addr.parse().map_err(|err| StartError::Server(Box::new(err)))?,
-            create_buffer(&device.mac).map_err(|err| StartError::Server(Box::new(err)))?
-        ).map_err(|err| StartError::Server(Box::new(err)));
+            &bind_addr.parse().map_err(|err| WebolError::Server(Box::new(err)))?,
+            &device.broadcast_addr.parse().map_err(|err| WebolError::Server(Box::new(err)))?,
+            create_buffer(&device.mac).map_err(|err| WebolError::Server(Box::new(err)))?
+        ).map_err(|err| WebolError::Server(Box::new(err)));
         Ok(Json(json!(StartResponse { id: device.id, boot: true })))
     } else {
-        Err(StartError::Generic)
+        Err(WebolError::Generic)
     }
 }
 
@@ -53,28 +51,4 @@ pub struct StartPayload {
 struct StartResponse {
     id: String,
     boot: bool,
-}
-
-pub enum StartError {
-    Auth(AuthError),
-    Generic,
-    Server(Box<dyn Error>),
-}
-
-impl IntoResponse for StartError {
-    fn into_response(self) -> Response {
-        let (status, error_message) = match self {
-            StartError::Auth(err) => err.get(),
-            StartError::Generic => (StatusCode::INTERNAL_SERVER_ERROR, ""),
-            StartError::Server(err) => {
-                error!("server error: {}", err.to_string());
-                (StatusCode::INTERNAL_SERVER_ERROR, "Server Error")
-            },
-
-        };
-        let body = Json(json!({
-            "error": error_message,
-        }));
-        (status, body).into_response()
-    }
 }
\ No newline at end of file