tetratto/crates/app/src/routes/pages/economy.rs

use axum::{
    extract::{Query, Path},
    response::{Html, IntoResponse},
    Extension,
};
use crate::cookie::CookieJar;
use tetratto_core::model::{economy::CoinTransferMethod, Error};
use crate::{assets::initial_context, get_lang, get_user_from_token, State};
use super::{render_error, PaginatedQuery};

/// `/wallet`
pub async fn wallet_request(
    jar: CookieJar,
    Extension(data): Extension<State>,
    Query(props): Query<PaginatedQuery>,
) -> impl IntoResponse {
    let data = data.read().await;
    let user = match get_user_from_token!(jar, data.0) {
        Some(ua) => ua,
        None => {
            return Err(Html(
                render_error(Error::NotAllowed, &jar, &data, &None).await,
            ));
        }
    };

    let list = match data.0.get_transfers_by_user(user.id, 12, props.page).await {
        Ok(x) => match data.0.fill_transfers(x).await {
            Ok(x) => x,
            Err(e) => return Err(Html(render_error(e, &jar, &data, &Some(user)).await)),
        },
        Err(e) => return Err(Html(render_error(e, &jar, &data, &Some(user)).await)),
    };

    let lang = get_lang!(jar, data.0);
    let mut context = initial_context(&data.0.0.0, lang, &Some(user)).await;

    context.insert("list", &list);
    context.insert("page", &props.page);

    // return
    Ok(Html(
        data.1.render("economy/wallet.html", &context).unwrap(),
    ))
}

/// `/products`
pub async fn products_request(
    jar: CookieJar,
    Extension(data): Extension<State>,
    Query(props): Query<PaginatedQuery>,
) -> impl IntoResponse {
    let data = data.read().await;
    let user = match get_user_from_token!(jar, data.0) {
        Some(ua) => ua,
        None => {
            return Err(Html(
                render_error(Error::NotAllowed, &jar, &data, &None).await,
            ));
        }
    };

    let list = match data.0.get_products_by_user(user.id, 12, props.page).await {
        Ok(x) => x,
        Err(e) => return Err(Html(render_error(e, &jar, &data, &Some(user)).await)),
    };

    let lang = get_lang!(jar, data.0);
    let mut context = initial_context(&data.0.0.0, lang, &Some(user)).await;

    context.insert("list", &list);
    context.insert("page", &props.page);

    // return
    Ok(Html(
        data.1.render("economy/products.html", &context).unwrap(),
    ))
}

/// `/product/{id}/edit`
pub async fn edit_product_request(
    jar: CookieJar,
    Extension(data): Extension<State>,
    Path(id): Path<usize>,
) -> impl IntoResponse {
    let data = data.read().await;
    let user = match get_user_from_token!(jar, data.0) {
        Some(ua) => ua,
        None => {
            return Err(Html(
                render_error(Error::NotAllowed, &jar, &data, &None).await,
            ));
        }
    };

    let product = match data.0.get_product_by_id(id).await {
        Ok(x) => x,
        Err(e) => return Err(Html(render_error(e, &jar, &data, &Some(user)).await)),
    };

    if user.id != product.owner {
        return Err(Html(
            render_error(Error::NotAllowed, &jar, &data, &None).await,
        ));
    }

    let lang = get_lang!(jar, data.0);
    let mut context = initial_context(&data.0.0.0, lang, &Some(user)).await;
    context.insert("product", &product);

    // return
    Ok(Html(data.1.render("economy/edit.html", &context).unwrap()))
}

/// `/product/{id}`
pub async fn product_request(
    jar: CookieJar,
    Extension(data): Extension<State>,
    Path(id): Path<usize>,
) -> impl IntoResponse {
    let data = data.read().await;
    let user = match get_user_from_token!(jar, data.0) {
        Some(ua) => ua,
        None => {
            return Err(Html(
                render_error(Error::NotAllowed, &jar, &data, &None).await,
            ));
        }
    };

    let product = match data.0.get_product_by_id(id).await {
        Ok(x) => x,
        Err(e) => return Err(Html(render_error(e, &jar, &data, &Some(user)).await)),
    };

    let owner = match data.0.get_user_by_id(product.owner).await {
        Ok(x) => x,
        Err(e) => return Err(Html(render_error(e, &jar, &data, &Some(user)).await)),
    };

    let already_purchased = if product.single_use {
        data.0
            .get_transfer_by_sender_method(user.id, CoinTransferMethod::Purchase(product.id))
            .await
            .is_ok()
    } else {
        false
    };

    let lang = get_lang!(jar, data.0);
    let mut context = initial_context(&data.0.0.0, lang, &Some(user)).await;

    context.insert("product", &product);
    context.insert("owner", &owner);
    context.insert("already_purchased", &already_purchased);

    // return
    Ok(Html(
        data.1.render("economy/product.html", &context).unwrap(),
    ))
}