refactor: improve error handling (#216)

This commit is contained in:
XIAO YU
2024-06-24 09:19:36 +09:00
committed by GitHub
parent 3fbd66dece
commit f0712e5961
2 changed files with 32 additions and 38 deletions

View File

@@ -9,8 +9,8 @@ use tauri_nspanel::{
base::{id, nil}, base::{id, nil},
foundation::{NSPoint, NSRect}, foundation::{NSPoint, NSRect},
}, },
ManagerExt, objc::{class, msg_send, runtime::NO, sel, sel_impl},
objc::{class, msg_send, runtime::NO, sel, sel_impl}, panel_delegate, WebviewWindowExt, panel_delegate, ManagerExt, WebviewWindowExt,
}; };
#[allow(non_upper_case_globals)] #[allow(non_upper_case_globals)]
@@ -28,11 +28,8 @@ pub fn swizzle_to_menubar_panel(app_handle: &tauri::AppHandle) {
let handle = app_handle.clone(); let handle = app_handle.clone();
panel_delegate.set_listener(Box::new(move |delegate_name: String| { panel_delegate.set_listener(Box::new(move |delegate_name: String| {
match delegate_name.as_str() { if delegate_name.as_str() == "window_did_resign_key" {
"window_did_resign_key" => { let _ = handle.emit("menubar_panel_did_resign_key", ());
let _ = handle.emit("menubar_panel_did_resign_key", ());
}
_ => (),
} }
})); }));

View File

@@ -19,7 +19,7 @@ pub struct Relays {
#[tauri::command] #[tauri::command]
#[specta::specta] #[specta::specta]
pub async fn get_relays(state: State<'_, Nostr>) -> Result<Relays, ()> { pub async fn get_relays(state: State<'_, Nostr>) -> Result<Relays, String> {
let client = &state.client; let client = &state.client;
// Get connected relays // Get connected relays
@@ -27,8 +27,8 @@ pub async fn get_relays(state: State<'_, Nostr>) -> Result<Relays, ()> {
let connected_relays: Vec<String> = list.into_keys().map(|url| url.to_string()).collect(); let connected_relays: Vec<String> = list.into_keys().map(|url| url.to_string()).collect();
// Get NIP-65 relay list // Get NIP-65 relay list
let signer = client.signer().await.unwrap(); let signer = client.signer().await.map_err(|e| e.to_string())?;
let public_key = signer.public_key().await.unwrap(); let public_key = signer.public_key().await.map_err(|e| e.to_string())?;
let filter = Filter::new() let filter = Filter::new()
.author(public_key) .author(public_key)
.kind(Kind::RelayList) .kind(Kind::RelayList)
@@ -71,53 +71,50 @@ pub async fn get_relays(state: State<'_, Nostr>) -> Result<Relays, ()> {
}) })
} }
} }
Err(_) => Ok(Relays { Err(e) => Err(e.to_string()),
connected: connected_relays,
read: None,
write: None,
both: None,
}),
} }
} }
#[tauri::command] #[tauri::command]
#[specta::specta] #[specta::specta]
pub async fn connect_relay(relay: &str, state: State<'_, Nostr>) -> Result<bool, ()> { pub async fn connect_relay(relay: &str, state: State<'_, Nostr>) -> Result<bool, String> {
let client = &state.client; let client = &state.client;
if let Ok(status) = client.add_relay(relay).await { match client.add_relay(relay).await {
if status { Ok(status) => {
println!("connecting to relay: {}", relay); if status {
let _ = client.connect_relay(relay).await; println!("connecting to relay: {}", relay);
Ok(true) let _ = client.connect_relay(relay).await;
} else { Ok(true)
Ok(false) } else {
Ok(false)
}
} }
} else { Err(e) => Err(e.to_string()),
Ok(false)
} }
} }
#[tauri::command] #[tauri::command]
#[specta::specta] #[specta::specta]
pub async fn remove_relay(relay: &str, state: State<'_, Nostr>) -> Result<bool, ()> { pub async fn remove_relay(relay: &str, state: State<'_, Nostr>) -> Result<bool, String> {
let client = &state.client; let client = &state.client;
if (client.remove_relay(relay).await).is_ok() { match client.remove_relay(relay).await {
let _ = client.disconnect_relay(relay).await; Ok(_) => {
Ok(true) let _ = client.disconnect_relay(relay).await;
} else { Ok(true)
Ok(false) }
Err(e) => Err(e.to_string()),
} }
} }
#[tauri::command] #[tauri::command]
#[specta::specta] #[specta::specta]
pub fn get_bootstrap_relays(app: tauri::AppHandle) -> Result<Vec<String>, ()> { pub fn get_bootstrap_relays(app: tauri::AppHandle) -> Result<Vec<String>, String> {
let relays_path = app let relays_path = app
.path() .path()
.resolve("resources/relays.txt", BaseDirectory::Resource) .resolve("resources/relays.txt", BaseDirectory::Resource)
.expect("Bootstrap relays not found."); .map_err(|e| e.to_string())?;
let file = std::fs::File::open(relays_path).unwrap(); let file = std::fs::File::open(relays_path).map_err(|e| e.to_string())?;
let lines = io::BufReader::new(file).lines(); let lines = io::BufReader::new(file).lines();
let mut relays = Vec::new(); let mut relays = Vec::new();
@@ -135,15 +132,15 @@ pub fn save_bootstrap_relays(relays: &str, app: tauri::AppHandle) -> Result<(),
let relays_path = app let relays_path = app
.path() .path()
.resolve("resources/relays.txt", BaseDirectory::Resource) .resolve("resources/relays.txt", BaseDirectory::Resource)
.expect("Bootstrap relays not found."); .map_err(|e| e.to_string())?;
let mut file = fs::OpenOptions::new() let mut file = fs::OpenOptions::new()
.write(true) .write(true)
.open(relays_path) .open(relays_path)
.unwrap(); .map_err(|e| e.to_string())?;
match file.write_all(relays.as_bytes()) { match file.write_all(relays.as_bytes()) {
Ok(_) => Ok(()), Ok(_) => Ok(()),
Err(_) => Err("Cannot save bootstrap relays, please try again later.".into()), Err(e) => Err(e.to_string()),
} }
} }