154 lines
4.8 KiB
Rust
154 lines
4.8 KiB
Rust
use std::net::SocketAddr;
|
|
use std::sync::Arc;
|
|
|
|
use sqlx::SqlitePool;
|
|
|
|
use axum::{
|
|
extract::FromRef,
|
|
routing::{get, post},
|
|
Router,
|
|
};
|
|
use hyper::StatusCode;
|
|
use tera::Tera;
|
|
|
|
mod routes;
|
|
mod users;
|
|
|
|
const DB_URL: &str = "sqlite://sqlite.db";
|
|
|
|
#[tokio::main]
|
|
async fn main() {
|
|
let db = accounters::create_db(DB_URL).await.unwrap();
|
|
|
|
let mut tmpls = Tera::new("templates/*").unwrap();
|
|
tmpls.autoescape_on(vec!["html"]);
|
|
|
|
let state = AppState {
|
|
db: Arc::new(db),
|
|
tmpls: Arc::new(tmpls),
|
|
};
|
|
|
|
let exec_id: u32 = rand::random();
|
|
|
|
let app = Router::new()
|
|
.nest(
|
|
"/",
|
|
Router::new()
|
|
.route("/", get(routes::ui::index))
|
|
.route("/accounts/id/:id", get(routes::ui::account::show))
|
|
.route(
|
|
"/accounts/id/:id/transactions/add",
|
|
get(routes::ui::account::add_transactions_view),
|
|
)
|
|
.route(
|
|
"/accounts/id/:id/transactions/add",
|
|
post(routes::ui::account::add_transactions_action),
|
|
)
|
|
.route(
|
|
"/accounts/id/:id/transactions",
|
|
get(routes::ui::account::list_transactions),
|
|
)
|
|
.route("/transaction/:id", get(routes::ui::transaction::view))
|
|
.route("/transaction/:id", post(routes::ui::transaction::update))
|
|
.route(
|
|
"/classifiers",
|
|
get(routes::ui::classifier::view_classifiers),
|
|
)
|
|
.route(
|
|
"/classifiers/new_rule",
|
|
get(routes::ui::classifier::rules_new_view),
|
|
)
|
|
.route(
|
|
"/classifiers/new_rule",
|
|
post(routes::ui::classifier::rules_new_action),
|
|
)
|
|
.route(
|
|
"/classifiers/new_category",
|
|
get(routes::ui::classifier::category_new_view),
|
|
)
|
|
.route(
|
|
"/classifiers/new_category",
|
|
post(routes::ui::classifier::category_new_action),
|
|
)
|
|
.nest(
|
|
"/static",
|
|
Router::new()
|
|
.route("/styles.css", get(routes::static_routes::styles))
|
|
.route("/csv.js", get(routes::static_routes::csv)),
|
|
)
|
|
.route(
|
|
"/execution",
|
|
get(move || async move { format!("{exec_id}") }),
|
|
),
|
|
)
|
|
.nest(
|
|
"/api/v1",
|
|
Router::new()
|
|
.route("/user", post(routes::api::create_user))
|
|
.route("/login", post(routes::api::login))
|
|
.route("/accounts", post(routes::api::accounts::account_create))
|
|
.route("/accounts", get(routes::api::accounts::account_list))
|
|
.route("/accounts/id/:id", get(routes::api::accounts::account_get))
|
|
.route(
|
|
"/accounts/id/:id/transaction",
|
|
post(routes::api::transactions::create),
|
|
)
|
|
.route(
|
|
"/accounts/id/:id/transaction",
|
|
get(routes::api::transactions::list),
|
|
)
|
|
.route(
|
|
"/accounts/id/:id/recategorize",
|
|
post(routes::api::accounts::recategorize),
|
|
)
|
|
.route("/categories", post(routes::api::categories::create))
|
|
.route("/categories", get(routes::api::categories::list))
|
|
.route("/rules", post(routes::api::rules::create))
|
|
.route("/rules", get(routes::api::rules::list)),
|
|
)
|
|
.with_state(state);
|
|
|
|
let addr = SocketAddr::from(([0, 0, 0, 0], 3000));
|
|
let server = axum::Server::bind(&addr).serve(app.into_make_service());
|
|
|
|
let wv_task = tokio::task::spawn_blocking(|| {
|
|
web_view::builder()
|
|
.title("Test")
|
|
.content(web_view::Content::Url("http://localhost:3000"))
|
|
.user_data(())
|
|
.invoke_handler(|_wv, _arg| Ok(()))
|
|
.run()
|
|
.unwrap();
|
|
});
|
|
|
|
tokio::select! {
|
|
_ = wv_task => {
|
|
println!("WebView finished");
|
|
}
|
|
e = server => {
|
|
println!("Axum finished with result {e:?}");
|
|
}
|
|
}
|
|
}
|
|
|
|
#[derive(Clone)]
|
|
pub struct AppState {
|
|
db: Arc<SqlitePool>,
|
|
tmpls: Arc<Tera>,
|
|
}
|
|
|
|
impl FromRef<AppState> for Arc<SqlitePool> {
|
|
fn from_ref(state: &AppState) -> Arc<SqlitePool> {
|
|
state.db.clone()
|
|
}
|
|
}
|
|
|
|
impl FromRef<AppState> for Arc<Tera> {
|
|
fn from_ref(state: &AppState) -> Arc<Tera> {
|
|
state.tmpls.clone()
|
|
}
|
|
}
|
|
|
|
async fn index() -> (StatusCode, String) {
|
|
(StatusCode::OK, String::from("Hello, World!"))
|
|
}
|