feat: add basic relay management in rust

src-tauri/src/nostr/relay.rs

@@ -2,11 +2,81 @@ use crate::Nostr;
 use nostr_sdk::prelude::*;
 use tauri::State;
 
+#[derive(serde::Serialize)]
+pub struct Relays {
+  connected: Vec<String>,
+  read: Option<Vec<String>>,
+  write: Option<Vec<String>>,
+  both: Option<Vec<String>>,
+}
+
+#[tauri::command]
+pub async fn get_relays(state: State<'_, Nostr>) -> Result<Relays, ()> {
+  let client = &state.client;
+
+  // Get connected relays
+  let list = client.relays().await;
+  let connected_relays: Vec<String> = list.into_iter().map(|(url, _)| url.to_string()).collect();
+
+  // Get NIP-65 relay list
+  let signer = client.signer().await.unwrap();
+  let public_key = signer.public_key().await.unwrap();
+  let filter = Filter::new()
+    .author(public_key)
+    .kind(Kind::RelayList)
+    .limit(1);
+
+  match client.get_events_of(vec![filter], None).await {
+    Ok(events) => {
+      if let Some(event) = events.first() {
+        let nip65_list = nip65::extract_relay_list(event);
+        let read: Vec<String> = nip65_list
+          .clone()
+          .into_iter()
+          .filter(|i| matches!(&i.1, Some(y) if *y == RelayMetadata::Read))
+          .map(|(url, _)| url.to_string())
+          .collect();
+        let write: Vec<String> = nip65_list
+          .clone()
+          .into_iter()
+          .filter(|i| matches!(&i.1, Some(y) if *y == RelayMetadata::Write))
+          .map(|(url, _)| url.to_string())
+          .collect();
+        let both: Vec<String> = nip65_list
+          .into_iter()
+          .filter(|i| i.1.is_none())
+          .map(|(url, _)| url.to_string())
+          .collect();
+
+        Ok(Relays {
+          connected: connected_relays,
+          read: Some(read),
+          write: Some(write),
+          both: Some(both),
+        })
+      } else {
+        Ok(Relays {
+          connected: connected_relays,
+          read: None,
+          write: None,
+          both: None,
+        })
+      }
+    }
+    Err(_) => Ok(Relays {
+      connected: connected_relays,
+      read: None,
+      write: None,
+      both: None,
+    }),
+  }
+}
+
 #[tauri::command]
 pub async fn list_connected_relays(state: State<'_, Nostr>) -> Result<Vec<Url>, ()> {
   let client = &state.client;
-  let relays = client.relays().await;
-  let list: Vec<Url> = relays.into_keys().collect();
+  let connected_relays = client.relays().await;
+  let list = connected_relays.into_keys().collect();
 
   Ok(list)
 }
@@ -15,6 +85,7 @@ pub async fn list_connected_relays(state: State<'_, Nostr>) -> Result<Vec<Url>,
 pub async fn connect_relay(relay: &str, state: State<'_, Nostr>) -> Result<bool, ()> {
   let client = &state.client;
   if let Ok(_) = client.add_relay(relay).await {
+    let _ = client.connect_relay(relay);
     Ok(true)
   } else {
     Ok(false)
@@ -25,6 +96,7 @@ pub async fn connect_relay(relay: &str, state: State<'_, Nostr>) -> Result<bool,
 pub async fn remove_relay(relay: &str, state: State<'_, Nostr>) -> Result<bool, ()> {
   let client = &state.client;
   if let Ok(_) = client.remove_relay(relay).await {
+    let _ = client.disconnect_relay(relay);
     Ok(true)
   } else {
     Ok(false)