tornadocontrib/web3-proxy
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

mark as premium if no downgrade tier id

Browse Source
This commit is contained in:
Bryan Stitt 2023-07-20 14:54:42 -07:00
parent ae8a12047e
commit 3334dc5c0b
1 changed files with 30 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

34
web3_proxy/src/balance.rs
View File

@ -1,11 +1,12 @@
use crate::errors::{Web3ProxyErrorContext, Web3ProxyResult}; use crate::errors::{Web3ProxyError, Web3ProxyErrorContext, Web3ProxyResult};
use crate::premium::get_user_and_tier_from_id;
use entities::{ use entities::{
admin_increase_balance_receipt, increase_on_chain_balance_receipt, referee, referrer, admin_increase_balance_receipt, increase_on_chain_balance_receipt, referee, referrer,
rpc_accounting_v2, rpc_key, stripe_increase_balance_receipt, rpc_accounting_v2, rpc_key, stripe_increase_balance_receipt,
}; };
use migration::sea_orm::prelude::Decimal; use migration::sea_orm::prelude::Decimal;
use migration::sea_orm::DbConn;
use migration::sea_orm::{ColumnTrait, EntityTrait, QueryFilter, QuerySelect}; use migration::sea_orm::{ColumnTrait, EntityTrait, QueryFilter, QuerySelect};
use migration::sea_orm::{DbConn, TransactionTrait};
use migration::{Func, SimpleExpr}; use migration::{Func, SimpleExpr};
use serde::ser::SerializeStruct; use serde::ser::SerializeStruct;
use serde::{Deserialize, Serialize}; use serde::{Deserialize, Serialize};
@ -27,6 +28,8 @@ pub struct Balance {
pub total_spent: Decimal, pub total_spent: Decimal,
pub total_spent_paid_credits: Decimal, pub total_spent_paid_credits: Decimal,
pub user_id: u64, pub user_id: u64,
pub user_tier_id: u64,
pub downgrade_tier_id: Option<u64>,
} }
impl Serialize for Balance { impl Serialize for Balance {
@ -46,6 +49,8 @@ impl Serialize for Balance {
state.serialize_field("total_spent", &self.total_spent)?; state.serialize_field("total_spent", &self.total_spent)?;
state.serialize_field("total_spent_paid_credits", &self.total_spent_paid_credits)?; state.serialize_field("total_spent_paid_credits", &self.total_spent_paid_credits)?;
state.serialize_field("user_id", &self.user_id)?; state.serialize_field("user_id", &self.user_id)?;
state.serialize_field("user_tier_id", &self.user_tier_id)?;
state.serialize_field("downgrade_tier_id", &self.downgrade_tier_id)?;
state.serialize_field("active_premium", &self.active_premium())?; state.serialize_field("active_premium", &self.active_premium())?;
state.serialize_field("was_ever_premium", &self.was_ever_premium())?; state.serialize_field("was_ever_premium", &self.was_ever_premium())?;
@ -58,13 +63,16 @@ impl Serialize for Balance {
impl Balance { impl Balance {
pub fn active_premium(&self) -> bool { pub fn active_premium(&self) -> bool {
self.was_ever_premium() && self.total_deposits() > self.total_spent_paid_credits self.was_ever_premium()
&& (self.total_deposits() > self.total_spent_paid_credits
|| self.downgrade_tier_id.is_none())
} }
pub fn was_ever_premium(&self) -> bool { pub fn was_ever_premium(&self) -> bool {
// TODO: technically we should also check that user_tier.downgrade_tier_id.is_some() // TODO: technically we should also check that user_tier.downgrade_tier_id.is_some()
// but now we set premium automatically on deposit, so its fine for now // but now we set premium automatically on deposit, so its fine for now
self.user_id != 0 && self.total_deposits() >= Decimal::from(10) self.user_id != 0
&& (self.total_deposits() >= Decimal::from(10) || self.downgrade_tier_id.is_none())
} }
pub fn remaining(&self) -> Decimal { pub fn remaining(&self) -> Decimal {
@ -86,6 +94,22 @@ impl Balance {
return Ok(None); return Ok(None);
} }
let txn = db_conn.begin().await?;
// TODO: we probably already have the user and their tier, but this is easy
let (_, user_tier_entry) =
get_user_and_tier_from_id(user_id, &txn)
.await?
.ok_or(Web3ProxyError::BadRequest(
format!("No user found with id {}", user_id).into(),
))?;
txn.commit().await?;
let user_tier_entry = user_tier_entry.ok_or(Web3ProxyError::BadRequest(
format!("No user tier found for user id {}", user_id).into(),
))?;
let (admin_deposits,) = admin_increase_balance_receipt::Entity::find() let (admin_deposits,) = admin_increase_balance_receipt::Entity::find()
.select_only() .select_only()
.column_as( .column_as(
@ -218,6 +242,8 @@ impl Balance {
total_spent, total_spent,
total_spent_paid_credits, total_spent_paid_credits,
user_id, user_id,
user_tier_id: user_tier_entry.id,
downgrade_tier_id: user_tier_entry.downgrade_tier_id,
}; };
trace!("balance: {:#}", json!(&balance)); trace!("balance: {:#}", json!(&balance));