summaryrefslogtreecommitdiff
path: root/src/handlers
diff options
context:
space:
mode:
Diffstat (limited to 'src/handlers')
-rw-r--r--src/handlers/ships.rs75
1 files changed, 53 insertions, 22 deletions
diff --git a/src/handlers/ships.rs b/src/handlers/ships.rs
index 9d985ea..91944fc 100644
--- a/src/handlers/ships.rs
+++ b/src/handlers/ships.rs
@@ -1,56 +1,87 @@
use std::sync::Arc;
-use axum::{Json, extract::Path, Extension};
-use hyper::StatusCode;
-use solarlib::ship::{Ship, DbShip, Sha256};
-use sqlx::{query_as, query, Error as SqlxError};
+use axum::{extract::Path, Extension, Json};
use axum_auth::AuthBearer;
+use hyper::StatusCode;
+use solarlib::ship::{DbShip, Sha256, Ship};
+use sqlx::{query, query_as, Error as SqlxError};
use tracing::log::warn;
-use crate::{errors::{JsonResult, StringResult, ServiceError}, State};
-
+use crate::{
+ errors::{JsonResult, ServiceError, StringResult},
+ State,
+};
pub async fn list(state: Extension<Arc<State>>) -> JsonResult<Json<Vec<Ship>>> {
let mut conn = state.conn.acquire().await?;
- let db_ships = query_as!(DbShip, "SELECT * FROM ships").fetch_all(&mut conn).await?;
+ let db_ships = query_as!(DbShip, "SELECT * FROM ships")
+ .fetch_all(&mut conn)
+ .await?;
- let ships = db_ships.into_iter().map(|d| d.into()).collect::<Vec<Ship>>();
+ let ships = db_ships
+ .into_iter()
+ .map(|d| d.into())
+ .collect::<Vec<Ship>>();
Ok(Json(ships))
}
-pub async fn new(Json(new_ship): Json<Ship>, state: Extension<Arc<State>>, AuthBearer(token): AuthBearer) -> StringResult {
+pub async fn new(
+ Json(new_ship): Json<Ship>,
+ state: Extension<Arc<State>>,
+ AuthBearer(token): AuthBearer,
+) -> StringResult {
check_bearer(token)?;
let mut conn = state.conn.acquire().await?;
- query!("INSERT INTO ships (name, shasum, download_url, version) VALUES ($1, $2, $3, $4)", new_ship.name, new_ship.shasum.to_string(), new_ship.download_url, new_ship.version).execute(&mut conn).await?;
+ query!(
+ "INSERT INTO ships (name, shasum, download_url, version) VALUES ($1, $2, $3, $4)",
+ new_ship.name,
+ new_ship.shasum.to_string(),
+ new_ship.download_url,
+ new_ship.version
+ )
+ .execute(&mut conn)
+ .await?;
Ok("OK")
}
-pub async fn delete(Path(shasum): Path<Sha256>, state: Extension<Arc<State>>, AuthBearer(token): AuthBearer) -> StringResult {
+pub async fn delete(
+ Path(shasum): Path<Sha256>,
+ state: Extension<Arc<State>>,
+ AuthBearer(token): AuthBearer,
+) -> StringResult {
check_bearer(token)?;
let mut conn = state.conn.acquire().await?;
- query!("DELETE FROM ships WHERE shasum=$1", shasum.to_string()).execute(&mut conn).await?;
+ query!("DELETE FROM ships WHERE shasum=$1", shasum.to_string())
+ .execute(&mut conn)
+ .await?;
Ok("OK")
}
-pub async fn get(Path(shasum): Path<Sha256>, state: Extension<Arc<State>>) -> JsonResult<Json<DbShip>> {
+pub async fn get(
+ Path(shasum): Path<Sha256>,
+ state: Extension<Arc<State>>,
+) -> JsonResult<Json<DbShip>> {
let mut conn = state.conn.acquire().await?;
- let db_ship = match query_as!(DbShip, "SELECT * FROM ships WHERE shasum=$1", shasum.to_string()).fetch_one(&mut conn).await {
+ let db_ship = match query_as!(
+ DbShip,
+ "SELECT * FROM ships WHERE shasum=$1",
+ shasum.to_string()
+ )
+ .fetch_one(&mut conn)
+ .await
+ {
Ok(d) => d,
- Err(e) => {
- match e {
- SqlxError::RowNotFound => {
- return Err(ServiceError::NotFound)
- },
- _ => {
- return Err(e.into());
- }
+ Err(e) => match e {
+ SqlxError::RowNotFound => return Err(ServiceError::NotFound),
+ _ => {
+ return Err(e.into());
}
},
};