This commit is contained in:
2023-05-10 00:42:42 +02:00
parent 3264af5c65
commit 563f1b3d89
4 changed files with 261 additions and 44 deletions

View File

@@ -1,17 +1,28 @@
#![allow(unused_imports)]
use axum::{
extract::Path,
http::StatusCode,
response::Html,
extract::State,
headers,
headers::Header,
http::{header::HeaderMap, StatusCode},
response::{Html, Redirect},
routing::{get, post},
Router,
Form, Router,
};
use sqlx::sqlite::SqlitePoolOptions;
use sqlx::{
error::DatabaseError,
query,
sqlite::{SqliteConnectOptions, SqliteError, SqlitePoolOptions},
Pool, Sqlite,
};
use serde::Deserialize;
use tracing_subscriber;
use futures::TryStreamExt;
use uuid::Uuid;
use uuid::{uuid, Uuid};
use std::net::SocketAddr;
@@ -21,13 +32,20 @@ mod models;
use crate::components::*;
use crate::models::*;
pub struct State {
#[derive(Clone)]
pub struct AppState {
database_pool: Pool<Sqlite>,
client_state: ClientState,
}
#[derive(Clone)]
pub struct ClientState {
pub active_category_id: Option<Uuid>,
}
impl State {
impl ClientState {
pub fn new() -> Self {
State {
ClientState {
active_category_id: None,
}
}
@@ -39,17 +57,34 @@ async fn main() -> Result<(), sqlx::Error> {
.with_max_level(tracing::Level::DEBUG)
.init();
let database_pool = SqlitePoolOptions::new()
.max_connections(5)
.connect_with(
SqliteConnectOptions::new()
.filename("/home/hannes-private/sync/items/items.sqlite")
.pragma("foreign_keys", "1"),
)
.await
.unwrap();
let state = AppState {
database_pool,
client_state: ClientState::new(),
};
// build our application with a route
let app = Router::new()
.route("/", get(root))
.route("/trips/", get(trips))
.route("/inventory/", get(inventory_inactive))
.route("/inventory/item/", post(inventory_item_create))
.route("/inventory/category/:id", get(inventory_active))
.route("/inventory/item/:id/delete", get(inventory_item_delete))
// .route(
// "/inventory/category/:id/items",
// post(htmx_inventory_category_items),
// );
;
.with_state(state);
let addr = SocketAddr::from(([127, 0, 0, 1], 3000));
tracing::debug!("listening on {}", addr);
@@ -70,33 +105,30 @@ async fn root() -> (StatusCode, Html<String>) {
async fn inventory_active(
Path(id): Path<String>,
State(state): State<AppState>,
) -> Result<(StatusCode, Html<String>), (StatusCode, Html<String>)> {
inventory(Some(id)).await
inventory(state, Some(id)).await
}
async fn inventory_inactive() -> Result<(StatusCode, Html<String>), (StatusCode, Html<String>)> {
inventory(None).await
async fn inventory_inactive(
State(state): State<AppState>,
) -> Result<(StatusCode, Html<String>), (StatusCode, Html<String>)> {
inventory(state, None).await
}
async fn inventory(
mut state: AppState,
active_id: Option<String>,
) -> Result<(StatusCode, Html<String>), (StatusCode, Html<String>)> {
let mut state: State = State::new();
let active_id = active_id
.map(|id| Uuid::try_parse(&id))
.transpose()
.map_err(|e| (StatusCode::BAD_REQUEST, Html::from(e.to_string())))?;
state.active_category_id = active_id;
state.client_state.active_category_id = active_id;
let pool = SqlitePoolOptions::new()
.max_connections(5)
.connect("sqlite:///home/hannes-private/sync/items/items.sqlite")
.await
.unwrap();
let mut categories = sqlx::query("SELECT id,name,description FROM inventoryitemcategories")
.fetch(&pool)
let mut categories = query("SELECT id,name,description FROM inventoryitemcategories")
.fetch(&state.database_pool)
.map_ok(|row| row.try_into())
.try_collect::<Vec<Result<Category, models::Error>>>()
.await
@@ -126,7 +158,7 @@ async fn inventory(
StatusCode::OK,
Html::from(
Root::build(
Inventory::build(state, categories)
Inventory::build(state.client_state, categories)
.await
.map_err(|e| (StatusCode::INTERNAL_SERVER_ERROR, Html::from(e.to_string())))?
.into(),
@@ -137,15 +169,11 @@ async fn inventory(
))
}
async fn trips() -> Result<(StatusCode, Html<String>), (StatusCode, Html<String>)> {
let pool = SqlitePoolOptions::new()
.max_connections(5)
.connect("sqlite:///home/hannes-private/sync/items/items.sqlite")
.await
.unwrap();
let trips = sqlx::query("SELECT * FROM trips")
.fetch(&pool)
async fn trips(
State(state): State<AppState>,
) -> Result<(StatusCode, Html<String>), (StatusCode, Html<String>)> {
let trips = query("SELECT * FROM trips")
.fetch(&state.database_pool)
.map_ok(|row| row.try_into())
.try_collect::<Vec<Result<Trip, models::Error>>>()
.await
@@ -164,6 +192,113 @@ async fn trips() -> Result<(StatusCode, Html<String>), (StatusCode, Html<String>
))
}
#[derive(Deserialize)]
struct NewItem {
#[serde(rename = "new-item-name")]
name: String,
#[serde(rename = "new-item-weight")]
weight: u32,
// damn i just love how serde is integrated everywhere, just add a feature to the uuid in
// cargo.toml and go
#[serde(rename = "new-item-category-id")]
category_id: Uuid,
}
async fn inventory_item_create(
State(state): State<AppState>,
Form(new_item): Form<NewItem>,
) -> Result<Redirect, (StatusCode, String)> {
query(
"INSERT INTO inventoryitems
(id, name, description, weight, category_id)
VALUES
(?, ?, ?, ?, ?)",
)
.bind(Uuid::new_v4().to_string())
.bind(&new_item.name)
.bind("")
.bind(new_item.weight)
.bind(new_item.category_id.to_string())
.execute(&state.database_pool)
.await
.map_err(|e| match e {
sqlx::Error::Database(ref error) => {
let sqlite_error = error.downcast_ref::<SqliteError>();
if let Some(code) = sqlite_error.code() {
match &*code {
"787" => {
// SQLITE_CONSTRAINT_FOREIGNKEY
(
StatusCode::BAD_REQUEST,
format!("category {id} not found", id = new_item.category_id),
)
}
"2067" => {
// SQLITE_CONSTRAINT_UNIQUE
(
StatusCode::BAD_REQUEST,
format!(
"item with name \"{name}\" already exists in category {id}",
name = new_item.name,
id = new_item.category_id
),
)
}
_ => (
StatusCode::INTERNAL_SERVER_ERROR,
format!("got error with unknown code: {}", sqlite_error.to_string()),
),
}
} else {
(
StatusCode::INTERNAL_SERVER_ERROR,
format!("got error without code: {}", sqlite_error.to_string()),
)
}
}
_ => (
StatusCode::INTERNAL_SERVER_ERROR,
format!("got unknown error: {}", e.to_string()),
),
})?;
Ok(Redirect::to(&format!(
"/inventory/category/{id}",
id = new_item.category_id
)))
}
async fn inventory_item_delete(
State(state): State<AppState>,
headers: HeaderMap,
Path(id): Path<Uuid>,
) -> Result<Redirect, (StatusCode, String)> {
query(
"DELETE FROM inventoryitems
WHERE id = ?",
)
.bind(id.to_string())
.execute(&state.database_pool)
.await
.map_err(|e| ((StatusCode::BAD_REQUEST, e.to_string())))?;
Ok(Redirect::to(
headers
.get("referer")
.ok_or((
StatusCode::BAD_REQUEST,
"no referer header found".to_string(),
))?
.to_str()
.map_err(|e| {
(
StatusCode::BAD_REQUEST,
format!("referer could not be converted: {}", e),
)
})?,
))
}
// async fn htmx_inventory_category_items(
// Path(id): Path<String>,
// ) -> Result<(StatusCode, Html<String>), (StatusCode, Html<String>)> {
@@ -173,7 +308,8 @@ async fn trips() -> Result<(StatusCode, Html<String>), (StatusCode, Html<String>
// .await
// .unwrap();
// let items = sqlx::query(&format!(
// let items = query(&format!(
// //TODO bind this stuff!!!!!!! no sql injection pls
// "SELECT
// i.id, i.name, i.description, i.weight, i.category_id
// FROM inventoryitemcategories AS c