Add optional payment

This commit is contained in:
eraden 2022-05-04 22:26:10 +02:00
parent 252f5373ff
commit b489b4435a
9 changed files with 156 additions and 76 deletions

View File

@ -33,6 +33,31 @@ macro_rules! db_async_handler {
}
}
};
($msg: ty, $async: ident, $res: ty, $inner_async: ident) => {
async fn $inner_async(msg: $msg, pool: sqlx::PgPool) -> Result<$res> {
let mut t = pool.begin().await?;
match $async(msg, &mut t).await {
Ok(res) => {
t.commit().await?;
Ok(res)
}
Err(e) => {
let _ = t.rollback().await;
Err(e)
}
}
}
impl actix::Handler<$msg> for Database {
type Result = actix::ResponseActFuture<Self, Result<$res>>;
fn handle(&mut self, msg: $msg, _ctx: &mut Self::Context) -> Self::Result {
use actix::WrapFuture;
let pool = self.pool.clone();
Box::pin(async { $inner_async(msg, pool).await }.into_actor(self))
}
}
};
}
#[macro_export]

View File

@ -62,14 +62,17 @@ pub struct CreateAccountOrder {
pub shopping_cart_id: ShoppingCartId,
}
db_async_handler!(CreateAccountOrder, create_account_order, AccountOrder);
db_async_handler!(
CreateAccountOrder,
create_account_order,
AccountOrder,
inner_create_account_order
);
pub(crate) async fn create_account_order(
msg: CreateAccountOrder,
db: PgPool,
t: &mut sqlx::Transaction<'_, sqlx::Postgres>,
) -> Result<AccountOrder> {
let mut t = db.begin().await?;
let order: AccountOrder = match sqlx::query_as(
r#"
INSERT INTO account_orders (buyer_id, status)
@ -79,13 +82,12 @@ RETURNING id, buyer_id, status, order_ext_id, service_order_id
)
.bind(msg.buyer_id)
.bind(OrderStatus::Confirmed)
.fetch_one(&mut t)
.fetch_one(&mut *t)
.await
{
Ok(order) => order,
Err(e) => {
log::error!("{e:?}");
t.rollback().await.ok();
return Err(super::Error::AccountOrder(Error::CantCreate));
}
};
@ -97,13 +99,12 @@ RETURNING id, buyer_id, status, order_ext_id, service_order_id
quantity: item.quantity,
quantity_unit: item.quantity_unit,
},
&mut t,
&mut *t,
)
.await
{
log::error!("{e:?}");
t.rollback().await.ok();
return Err(super::Error::AccountOrder(Error::CantCreate));
}
}
@ -113,18 +114,15 @@ RETURNING id, buyer_id, status, order_ext_id, service_order_id
id: msg.shopping_cart_id,
state: ShoppingCartState::Closed,
},
&mut t,
t,
)
.await
{
log::error!("{e:?}");
t.rollback().await.ok();
return Err(super::Error::AccountOrder(Error::CantCreate));
};
t.commit().await.ok();
Ok(order)
}

View File

@ -1,5 +1,4 @@
use actix::Message;
use sqlx::PgPool;
use super::Result;
use crate::database::Database;
@ -27,9 +26,12 @@ pub enum Error {
#[rtype(result = "Result<Vec<model::Product>>")]
pub struct AllProducts;
crate::db_async_handler!(AllProducts, all, Vec<Product>);
crate::db_async_handler!(AllProducts, all, Vec<Product>, inner_all);
pub(crate) async fn all(_msg: AllProducts, pool: PgPool) -> Result<Vec<model::Product>> {
pub(crate) async fn all<'e, E>(_msg: AllProducts, pool: E) -> Result<Vec<model::Product>>
where
E: sqlx::Executor<'e, Database = sqlx::Postgres>,
{
sqlx::query_as(
r#"
SELECT id,
@ -42,7 +44,7 @@ SELECT id,
FROM products
"#,
)
.fetch_all(&pool)
.fetch_all(pool)
.await
.map_err(|e| {
log::error!("{e:?}");
@ -61,9 +63,12 @@ pub struct CreateProduct {
pub deliver_days_flag: Days,
}
crate::db_async_handler!(CreateProduct, create_product, Product);
crate::db_async_handler!(CreateProduct, create_product, Product, inner_create_product);
pub(crate) async fn create_product(msg: CreateProduct, pool: PgPool) -> Result<model::Product> {
pub(crate) async fn create_product<'e, E>(msg: CreateProduct, pool: E) -> Result<model::Product>
where
E: sqlx::Executor<'e, Database = sqlx::Postgres>,
{
sqlx::query_as(
r#"
INSERT INTO products (name, short_description, long_description, category, price, deliver_days_flag)
@ -83,7 +88,7 @@ RETURNING id,
.bind(msg.category)
.bind(msg.price)
.bind(msg.deliver_days_flag)
.fetch_one(&pool)
.fetch_one(pool)
.await
.map_err(|e| {
log::error!("{e:?}");
@ -103,9 +108,12 @@ pub struct UpdateProduct {
pub deliver_days_flag: Days,
}
crate::db_async_handler!(UpdateProduct, update_product, Product);
crate::db_async_handler!(UpdateProduct, update_product, Product, inner_update_product);
pub(crate) async fn update_product(msg: UpdateProduct, pool: PgPool) -> Result<model::Product> {
pub(crate) async fn update_product<'e, E>(msg: UpdateProduct, pool: E) -> Result<model::Product>
where
E: sqlx::Executor<'e, Database = sqlx::Postgres>,
{
sqlx::query_as(
r#"
UPDATE products
@ -132,7 +140,7 @@ RETURNING id,
.bind(msg.category)
.bind(msg.price)
.bind(msg.deliver_days_flag)
.fetch_one(&pool)
.fetch_one(pool)
.await
.map_err(|e| {
log::error!("{e:?}");
@ -146,9 +154,17 @@ pub struct DeleteProduct {
pub product_id: ProductId,
}
crate::db_async_handler!(DeleteProduct, delete_product, Option<model::Product>);
crate::db_async_handler!(
DeleteProduct,
delete_product,
Option<model::Product>,
inner_delete_product
);
pub(crate) async fn delete_product(msg: DeleteProduct, pool: PgPool) -> Result<Option<Product>> {
pub(crate) async fn delete_product<'e, E>(msg: DeleteProduct, pool: E) -> Result<Option<Product>>
where
E: sqlx::Executor<'e, Database = sqlx::Postgres>,
{
sqlx::query_as(
r#"
DELETE FROM products
@ -163,7 +179,7 @@ RETURNING id,
"#,
)
.bind(msg.product_id)
.fetch_optional(&pool)
.fetch_optional(pool)
.await
.map_err(|e| {
log::error!("{e:?}");
@ -180,13 +196,17 @@ pub struct ShoppingCartProducts {
crate::db_async_handler!(
ShoppingCartProducts,
shopping_cart_products,
Vec<model::Product>
Vec<model::Product>,
inner_shopping_cart_products
);
pub(crate) async fn shopping_cart_products(
pub(crate) async fn shopping_cart_products<'e, E>(
msg: ShoppingCartProducts,
pool: PgPool,
) -> Result<Vec<Product>> {
pool: E,
) -> Result<Vec<Product>>
where
E: sqlx::Executor<'e, Database = sqlx::Postgres>,
{
sqlx::query_as(
r#"
SELECT products.id,
@ -202,7 +222,7 @@ WHERE shopping_cart_id = $1
"#,
)
.bind(msg.shopping_cart_id)
.fetch_all(&pool)
.fetch_all(pool)
.await
.map_err(|e| {
log::error!("{e:?}");

View File

@ -229,9 +229,17 @@ pub struct CartItems {
pub shopping_cart_id: ShoppingCartId,
}
db_async_handler!(CartItems, cart_items, Vec<ShoppingCartItem>);
db_async_handler!(
CartItems,
cart_items,
Vec<ShoppingCartItem>,
inner_cart_items
);
pub(crate) async fn cart_items(msg: CartItems, pool: PgPool) -> Result<Vec<ShoppingCartItem>> {
pub(crate) async fn cart_items<'e, E>(msg: CartItems, pool: E) -> Result<Vec<ShoppingCartItem>>
where
E: sqlx::Executor<'e, Database = sqlx::Postgres>,
{
let shopping_cart_id = msg.shopping_cart_id;
sqlx::query_as(
r#"
@ -241,7 +249,7 @@ WHERE shopping_cart_id = $1
"#,
)
.bind(msg.shopping_cart_id)
.fetch_all(&pool)
.fetch_all(pool)
.await
.map_err(|e| {
log::error!("{e:?}");

View File

@ -232,12 +232,13 @@ pub struct EnsureActiveShoppingCart {
db_async_handler!(
EnsureActiveShoppingCart,
ensure_active_shopping_cart,
ShoppingCart
ShoppingCart,
inner_ensure_active_shopping_cart
);
pub(crate) async fn ensure_active_shopping_cart(
msg: EnsureActiveShoppingCart,
pool: PgPool,
pool: &mut sqlx::Transaction<'_, sqlx::Postgres>,
) -> Result<ShoppingCart> {
if let Ok(Some(cart)) = sqlx::query_as(
r#"
@ -249,7 +250,7 @@ RETURNING id, buyer_id, payment_method, state;
"#,
)
.bind(msg.buyer_id)
.fetch_optional(&pool)
.fetch_optional(&mut *pool)
.await
.map_err(|e| {
log::error!("{e:?}");
@ -265,7 +266,7 @@ WHERE buyer_id = $1 AND state = 'active'
"#,
)
.bind(msg.buyer_id)
.fetch_one(&pool)
.fetch_one(pool)
.await
.map_err(|e| {
log::error!("{e:?}");

View File

@ -159,6 +159,9 @@ pub struct RequestPayment {
pub buyer: Buyer,
pub customer_ip: String,
pub buyer_id: AccountId,
/// False if customer is allowed to be charged on site.
/// Otherwise it should be true to use payment service for charging
pub charge_client: bool,
}
pay_async_handler!(RequestPayment, request_payment, CreatePaymentResult);
@ -177,6 +180,7 @@ pub(crate) async fn request_payment(
format!("{}/payment/success", w.host()),
)
};
let cart: model::ShoppingCart = query_db!(
db,
database::EnsureActiveShoppingCart {
@ -233,46 +237,55 @@ pub(crate) async fn request_payment(
Error::CreateOrder
);
let pay_u::res::CreateOrder {
status: _,
redirect_uri,
order_id,
ext_order_id: _,
} = {
client
.lock()
.create_order(
pay_u::req::OrderCreate::build(
msg.buyer.into(),
msg.customer_ip,
msg.currency,
format!("Order #{}", db_order.id),
)?
.with_products(cart_products.into_iter().map(|p| {
pay_u::Product::new(
p.name.to_string(),
**p.price,
items
.remove(&p.id)
.map(|(quantity, _)| **quantity as u32)
.unwrap_or_default(),
)
}))
.with_ext_order_id(db_order.order_ext_id.to_string())
.with_notify_url(notify_uri)
.with_continue_url(continue_uri),
)
.await?
let payment_required = {
let l = config.lock();
l.payment().optional_payment() != false
};
let redirect_uri = if msg.charge_client || payment_required {
let pay_u::res::CreateOrder {
status: _,
redirect_uri,
order_id,
ext_order_id: _,
} = {
client
.lock()
.create_order(
pay_u::req::OrderCreate::build(
msg.buyer.into(),
msg.customer_ip,
msg.currency,
format!("Order #{}", db_order.id),
)?
.with_products(cart_products.into_iter().map(|p| {
pay_u::Product::new(
p.name.to_string(),
**p.price,
items
.remove(&p.id)
.map(|(quantity, _)| **quantity as u32)
.unwrap_or_default(),
)
}))
.with_ext_order_id(db_order.order_ext_id.to_string())
.with_notify_url(notify_uri)
.with_continue_url(continue_uri),
)
.await?
};
query_db!(
db,
database::SetOrderServiceId {
service_order_id: order_id.0,
id: db_order.id,
},
Error::CreateOrder
);
query_db!(
db,
database::SetOrderServiceId {
service_order_id: order_id.0,
id: db_order.id,
},
Error::CreateOrder
);
redirect_uri
} else {
String::from("/pay-on-site")
};
let order_items = query_db!(
db,

View File

@ -38,6 +38,7 @@ pub struct PaymentConfig {
payu_client_id: Option<pay_u::ClientId>,
payu_client_secret: Option<pay_u::ClientSecret>,
payu_client_merchant_id: Option<pay_u::MerchantPosId>,
optional_payment: bool,
}
impl Example for PaymentConfig {
@ -51,11 +52,16 @@ impl Example for PaymentConfig {
)),
/// "Create payu account and copy here merchant id"
payu_client_merchant_id: Some(pay_u::MerchantPosId::from(0)),
optional_payment: true,
}
}
}
impl PaymentConfig {
pub fn optional_payment(&self) -> bool {
self.optional_payment
}
pub fn payu_client_id(&self) -> pay_u::ClientId {
self.payu_client_id
.as_ref()

View File

@ -40,10 +40,14 @@ pub enum Error {
#[get("/")]
async fn landing() -> HttpResponse {
HttpResponse::SeeOther().append_header((actix_web::http::header::LOCATION, ""));
HttpResponse::NotImplemented().body("")
}
#[get("/pay-on-site")]
async fn pay_on_site() -> HttpResponse {
HttpResponse::Ok().body("<h1>Pay on Site</h1>")
}
pub fn configure(config: &mut ServiceConfig) {
config.service(landing).configure(api_v1::configure);
}

View File

@ -200,6 +200,9 @@ pub struct CreateOrderInput {
pub last_name: String,
/// Required customer language
pub language: String,
/// False if customer is allowed to be charged on site.
/// Otherwise it should be true to use payment service for charging
pub charge_client: bool,
}
#[post("/order")]
@ -233,6 +236,7 @@ pub(crate) async fn create_order(
first_name,
last_name,
language,
charge_client,
} = payload;
let ip = match req.peer_addr() {
Some(ip) => ip,
@ -252,6 +256,7 @@ pub(crate) async fn create_order(
},
customer_ip: ip.to_string(),
buyer_id,
charge_client
},
routes::Error::Public(PublicError::DatabaseConnection)
);