web3-proxy/web3_proxy/tests/common/rpc_key.rs

use std::time::Duration;

use crate::TestApp;
use serde::Deserialize;
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<String, RpcKey>,
}

#[derive(Debug, Deserialize)]
pub struct RpcKey {
    pub active: bool,
    pub allowed_ips: Option<serde_json::Value>,
    pub allowed_origins: Option<serde_json::Value>,
    pub allowed_referers: Option<serde_json::Value>,
    pub allowed_user_agents: Option<serde_json::Value>,
    pub description: Option<serde_json::Value>,
    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();
    info!(?rpc_key_response);

    let rpc_key_response = rpc_key_response.json::<serde_json::Value>().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<EthersHttpProvider> {
    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),
    )
}