use super::TestApp; use serde::Deserialize; use std::time::Duration; use tracing::info; use ulid::Ulid; use web3_proxy::{ errors::Web3ProxyResult, frontend::users::authentication::LoginPostResponse, rpcs::provider::{connect_http, EthersHttpProvider}, }; #[derive(Debug, Deserialize)] pub struct RpcKeyResponse { pub user_id: u64, pub user_rpc_keys: std::collections::HashMap, } #[derive(Debug, Deserialize)] pub struct RpcKey { pub active: bool, pub allowed_ips: Option, pub allowed_origins: Option, pub allowed_referers: Option, pub allowed_user_agents: Option, pub description: Option, pub id: u64, pub log_revert_chance: f64, pub private_txs: bool, pub role: String, pub secret_key: Ulid, pub user_id: u64, } /// Helper function to get the user's balance #[allow(unused)] pub async fn user_get_first_rpc_key( x: &TestApp, r: &reqwest::Client, login_response: &LoginPostResponse, ) -> RpcKey { // TODO: refactor to use login_response? or compare? let get_keys = format!("{}user/keys", x.proxy_provider.url()); info!("Get balance"); let rpc_key_response = r .get(get_keys) .bearer_auth(login_response.bearer_token) .send() .await .unwrap() .error_for_status() .unwrap(); info!(?rpc_key_response); let rpc_key_response = rpc_key_response.json::().await.unwrap(); info!(?rpc_key_response); info!("Parsing rpc key as json"); let rpc_key: RpcKeyResponse = serde_json::from_value(rpc_key_response).unwrap(); info!(?rpc_key); rpc_key.user_rpc_keys.into_iter().next().unwrap().1 } #[allow(unused)] pub async fn user_get_provider( x: &TestApp, r: &reqwest::Client, login_response: &LoginPostResponse, ) -> Web3ProxyResult { let first_key = login_response.rpc_keys.iter().next().unwrap().1; let rpc_url = format!( "{}rpc/{}", x.proxy_provider.url(), Ulid::from(first_key.secret_key) ); connect_http( rpc_url.parse().unwrap(), Some(r.clone()), Duration::from_secs(1), ) }