feat(cast): support websockets

Author: bernard-wagner

Cargo.lock

@@ -9,12 +9,12 @@ use foundry_cli::{
     opts::{EthereumOpts, TransactionOpts},
     utils::{self, handle_traces, parse_ether_value, TraceResult},
 };
+use foundry_common::runtime_client::RuntimeClient;
 use foundry_config::{find_project_root_path, Config};
 use foundry_evm::{executor::opts::EvmOpts, trace::TracingExecutor};
 use std::str::FromStr;
 
-type Provider =
-    ethers::providers::Provider<ethers::providers::RetryClient<ethers::providers::Http>>;
+type Provider = ethers::providers::Provider<RuntimeClient>;
 
 /// CLI arguments for `cast call`.
 #[derive(Debug, Parser)]

crates/cast/bin/cmd/call.rs

@@ -33,6 +33,7 @@ tempfile = "3"
 
 #  misc
 auto_impl = "1.1.0"
+async-trait = "0.1"
 serde = "1"
 serde_json = "1"
 thiserror = "1"
@@ -43,6 +44,8 @@ once_cell = "1"
 dunce = "1"
 regex = "1"
 globset = "0.4"
+tokio = "1"
+url = "2"
 
 [dev-dependencies]
 tokio = { version = "1", features = ["rt-multi-thread", "macros"] }

crates/common/Cargo.toml

@@ -14,6 +14,7 @@ pub mod fmt;
 pub mod fs;
 pub mod glob;
 pub mod provider;
+pub mod runtime_client;
 pub mod selectors;
 pub mod shell;
 pub mod term;

crates/common/src/lib.rs

@@ -1,18 +1,16 @@
 //! Commonly used helpers to construct `Provider`s
 
-use crate::{ALCHEMY_FREE_TIER_CUPS, REQUEST_TIMEOUT};
+use crate::{runtime_client::RuntimeClient, ALCHEMY_FREE_TIER_CUPS, REQUEST_TIMEOUT};
 use ethers_core::types::{Chain, U256};
 use ethers_middleware::gas_oracle::{GasCategory, GasOracle, Polygon};
-use ethers_providers::{
-    is_local_endpoint, Http, HttpRateLimitRetryPolicy, Middleware, Provider, RetryClient,
-    RetryClientBuilder, DEFAULT_LOCAL_POLL_INTERVAL,
-};
+use ethers_providers::{is_local_endpoint, Middleware, Provider, DEFAULT_LOCAL_POLL_INTERVAL};
 use eyre::WrapErr;
 use reqwest::{IntoUrl, Url};
 use std::{borrow::Cow, time::Duration};
+use url::ParseError;
 
 /// Helper type alias for a retry provider
-pub type RetryProvider = Provider<RetryClient<Http>>;
+pub type RetryProvider = Provider<RuntimeClient>;
 
 /// Helper type alias for a rpc url
 pub type RpcUrl = String;
@@ -67,8 +65,15 @@ impl ProviderBuilder {
             return Self::new(format!("http://{url_str}"))
         }
         let err = format!("Invalid provider url: {url_str}");
+        let url = Url::parse(url_str)
+            .and_then(|url| match url.scheme() {
+                "http" | "https" | "wss" | "ws" | "file" => Ok(url),
+                _ => Err(ParseError::EmptyHost),
+            })
+            .wrap_err(err);
+
         Self {
-            url: url.into_url().wrap_err(err),
+            url,
             chain: Chain::Mainnet,
             max_retry: 100,
             timeout_retry: 5,
@@ -166,20 +171,16 @@ impl ProviderBuilder {
         } = self;
         let url = url?;
 
-        let client = reqwest::Client::builder().timeout(timeout).build()?;
-        let is_local = is_local_endpoint(url.as_str());
-
-        let provider = Http::new_with_client(url, client);
+        let mut provider = Provider::new(RuntimeClient::new(
+            url.clone(),
+            max_retry,
+            timeout_retry,
+            initial_backoff,
+            timeout,
+            compute_units_per_second,
+        ));
 
-        #[allow(clippy::box_default)]
-        let mut provider = Provider::new(
-            RetryClientBuilder::default()
-                .initial_backoff(Duration::from_millis(initial_backoff))
-                .rate_limit_retries(max_retry)
-                .timeout_retries(timeout_retry)
-                .compute_units_per_second(compute_units_per_second)
-                .build(provider, Box::new(HttpRateLimitRetryPolicy)),
-        );
+        let is_local = is_local_endpoint(url.as_str());
 
         if is_local {
             provider = provider.interval(DEFAULT_LOCAL_POLL_INTERVAL);