From c2f1f541705156f6de9cac385c0c0535402ef2db Mon Sep 17 00:00:00 2001
From: xeruf <27jf@pm.me>
Date: Fri, 26 Jul 2024 12:58:09 +0300
Subject: [PATCH] feat(main): dynamically specify relays

---
 .gitignore  |  1 +
 src/main.rs | 23 ++++++++++++++++++++---
 2 files changed, 21 insertions(+), 3 deletions(-)

diff --git a/.gitignore b/.gitignore
index cc4a062..7140c09 100644
--- a/.gitignore
+++ b/.gitignore
@@ -1,5 +1,6 @@
 /target
 
+relays
 keys
 *.html
 /src/bin
diff --git a/src/main.rs b/src/main.rs
index 60652a0..a198cf3 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -1,7 +1,8 @@
 use std::env::args;
 use std::fmt::Display;
 use std::fs;
-use std::io::{stdin, stdout, Write};
+use std::fs::File;
+use std::io::{BufRead, BufReader, stdin, stdout, Write};
 use std::net::{Ipv4Addr, SocketAddr, SocketAddrV4};
 use std::ops::Deref;
 use std::str::FromStr;
@@ -66,9 +67,25 @@ async fn main() {
     let proxy = Some(SocketAddr::V4(SocketAddrV4::new(Ipv4Addr::LOCALHOST, 9050)));
 
     let client = Client::new(MY_KEYS.deref());
-    client.add_relay("ws://localhost:4736").await;
     println!("My public key: {}", MY_KEYS.public_key());
-    //client.add_relay("wss://relay.damus.io").await;
+    match File::open("relays").map(|f| BufReader::new(f).lines().flatten()) {
+        Ok(lines) => {
+            for line in lines {
+                or_print(client.add_relay(line).await);
+            }
+        }
+        Err(e) => {
+            eprintln!("Could not read relays file: {}", e);
+            print!("Relay? ");
+            stdout().flush().unwrap();
+            match stdin().lines().next() {
+                Some(Ok(line)) => {
+                    or_print(client.add_relay(if line.contains("://") { line } else { "wss://".to_string() + &line }).await);
+                }
+                _ => {}
+            };
+        }
+    }
     //client
     //    .add_relay_with_opts(
     //        "wss://relay.nostr.info",