Merge pull request #27 from tinydancer-io/custom_ws

fix: support custom websocket rpc
tinydancer-io · Mar 30, 2023 · 039d3fe · 039d3fe
2 parents f6f5ec8 + 75f4fdb
commit 039d3fe
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/tinydancer/src/macros.rs b/tinydancer/src/macros.rs
@@ -3,6 +3,8 @@ macro_rules! convert_to_websocket {
     ($test:expr) => {
         if $test.contains("https") {
             $test.replace("https://", "wss://")
+        } else if $test.contains("http") {
+            $test.replace("http://", "ws://")
         } else {
             String::from("ws://0.0.0.0:8900")
         }

diff --git a/tinydancer/src/main.rs b/tinydancer/src/main.rs
@@ -318,7 +318,7 @@ pub fn get_endpoint(cluster: String) -> String {
         "Mainnet" => "https://api.mainnet-beta.solana.com".to_owned(),
         "Devnet" => "https://api.devnet.solana.com".to_owned(),
         "Localnet" => "http://0.0.0.0:8899".to_owned(),
-        _ => "http://0.0.0.0:8899".to_owned(),
+        _ => cluster.to_owned(),
     }
 }