accounters/webserver/src/main.rs

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!"))
}