From 992860a46d768b6d615cbe39c48fe53f1e815fce Mon Sep 17 00:00:00 2001 From: yenicelik Date: Sun, 25 Jun 2023 16:18:09 -0400 Subject: [PATCH] forgot to allow aggregate global stats --- web3_proxy/src/stats/influxdb_queries.rs | 88 ++++++++++++------------ 1 file changed, 45 insertions(+), 43 deletions(-) diff --git a/web3_proxy/src/stats/influxdb_queries.rs b/web3_proxy/src/stats/influxdb_queries.rs index e1f432ee..3a592fce 100644 --- a/web3_proxy/src/stats/influxdb_queries.rs +++ b/web3_proxy/src/stats/influxdb_queries.rs @@ -57,55 +57,57 @@ pub async fn query_user_stats<'a>( // In any case, we don't allow stats if the target user does not have a balance // No subuser, we can check the balance directly - match balance::Entity::find() - .filter(balance::Column::UserId.eq(user_id)) - .one(db_replica.as_ref()) - .await? - { - // TODO: We should add the threshold that determines if a user is premium into app.config or so - Some(user_balance) => { - if user_balance.total_spent_outside_free_tier - user_balance.total_deposits - < Decimal::from(0) - { - trace!("User has 0 balance"); - return Err(Web3ProxyError::PaymentRequired); - } - // Otherwise make the user pass - } - None => { - trace!("User does not have a balance record, implying that he has no balance. Users must have a balance to access their stats dashboards"); - return Err(Web3ProxyError::PaymentRequired); - } - } - - // (Possible) subuser relation - // Check if the caller is a proper subuser (there is a subuser record) - // Check if the subuser has more than collaborator status - if user_id != caller_user_id { - // Find all rpc-keys related to the caller user - let user_rpc_keys: Vec = rpc_key::Entity::find() - .filter(rpc_key::Column::UserId.eq(user_id)) - .all(db_replica.as_ref()) - .await? - .into_iter() - .map(|x| x.id) - .collect::>(); - - match secondary_user::Entity::find() - .filter(secondary_user::Column::UserId.eq(caller_user_id)) - .filter(secondary_user::Column::RpcSecretKeyId.is_in(user_rpc_keys)) + if user_id != 0 { + match balance::Entity::find() + .filter(balance::Column::UserId.eq(user_id)) .one(db_replica.as_ref()) .await? { - Some(secondary_user_record) => { - if secondary_user_record.role == Role::Collaborator { - trace!("Subuser is only a collaborator, collaborators cannot see stats"); - return Err(Web3ProxyError::AccessDenied); + // TODO: We should add the threshold that determines if a user is premium into app.config or so + Some(user_balance) => { + if user_balance.total_spent_outside_free_tier - user_balance.total_deposits + < Decimal::from(0) + { + trace!("User has 0 balance"); + return Err(Web3ProxyError::PaymentRequired); } + // Otherwise make the user pass } None => { - // Then we must do an access denied - return Err(Web3ProxyError::AccessDeniedNoSubuser); + trace!("User does not have a balance record, implying that he has no balance. Users must have a balance to access their stats dashboards"); + return Err(Web3ProxyError::PaymentRequired); + } + } + + // (Possible) subuser relation + // Check if the caller is a proper subuser (there is a subuser record) + // Check if the subuser has more than collaborator status + if user_id != caller_user_id { + // Find all rpc-keys related to the caller user + let user_rpc_keys: Vec = rpc_key::Entity::find() + .filter(rpc_key::Column::UserId.eq(user_id)) + .all(db_replica.as_ref()) + .await? + .into_iter() + .map(|x| x.id) + .collect::>(); + + match secondary_user::Entity::find() + .filter(secondary_user::Column::UserId.eq(caller_user_id)) + .filter(secondary_user::Column::RpcSecretKeyId.is_in(user_rpc_keys)) + .one(db_replica.as_ref()) + .await? + { + Some(secondary_user_record) => { + if secondary_user_record.role == Role::Collaborator { + trace!("Subuser is only a collaborator, collaborators cannot see stats"); + return Err(Web3ProxyError::AccessDenied); + } + } + None => { + // Then we must do an access denied + return Err(Web3ProxyError::AccessDeniedNoSubuser); + } } } }