301 lines
11 KiB
Rust
301 lines
11 KiB
Rust
use asset::Assets;
|
|
use common::constants::{
|
|
ALL_MESSAGES_SUB_ID, APP_ID, APP_NAME, FAKE_SIG, KEYRING_SERVICE, METADATA_DELAY,
|
|
NEW_MESSAGE_SUB_ID,
|
|
};
|
|
use gpui::{
|
|
actions, point, px, size, App, AppContext, Application, Bounds, SharedString, TitlebarOptions,
|
|
WindowBounds, WindowKind, WindowOptions,
|
|
};
|
|
#[cfg(target_os = "linux")]
|
|
use gpui::{WindowBackgroundAppearance, WindowDecorations};
|
|
use log::warn;
|
|
use nostr_sdk::prelude::*;
|
|
use registry::{app::AppRegistry, chat::ChatRegistry, contact::Contact};
|
|
use state::{get_client, initialize_client};
|
|
use std::{collections::HashSet, str::FromStr, sync::Arc, time::Duration};
|
|
use tokio::{
|
|
sync::{mpsc, Mutex},
|
|
time::sleep,
|
|
};
|
|
use ui::Root;
|
|
use views::app::AppView;
|
|
|
|
mod asset;
|
|
mod views;
|
|
|
|
actions!(main_menu, [Quit]);
|
|
|
|
#[derive(Clone)]
|
|
pub enum Signal {
|
|
/// Receive event
|
|
Event(Event),
|
|
/// Receive EOSE
|
|
Eose,
|
|
}
|
|
|
|
#[tokio::main]
|
|
async fn main() {
|
|
tracing_subscriber::fmt::init();
|
|
|
|
// Initialize client
|
|
initialize_client();
|
|
|
|
// Get client
|
|
let client = get_client();
|
|
|
|
// Add some bootstrap relays
|
|
_ = client.add_relay("wss://relay.damus.io/").await;
|
|
_ = client.add_relay("wss://relay.primal.net/").await;
|
|
_ = client.add_relay("wss://nos.lol/").await;
|
|
|
|
_ = client.add_discovery_relay("wss://user.kindpag.es/").await;
|
|
_ = client.add_discovery_relay("wss://directory.yabu.me/").await;
|
|
|
|
// Connect to all relays
|
|
_ = client.connect().await;
|
|
|
|
// Signal
|
|
let (signal_tx, mut signal_rx) = mpsc::channel::<Signal>(4096);
|
|
let (mta_tx, mta_rx) = mpsc::channel::<PublicKey>(4096);
|
|
|
|
// Handle notifications from relays
|
|
// Send notify back to GPUI
|
|
tokio::spawn(async move {
|
|
let mut notifications = client.notifications();
|
|
let sig = Signature::from_str(FAKE_SIG).unwrap();
|
|
let new_message = SubscriptionId::new(NEW_MESSAGE_SUB_ID);
|
|
let all_messages = SubscriptionId::new(ALL_MESSAGES_SUB_ID);
|
|
|
|
while let Ok(notification) = notifications.recv().await {
|
|
if let RelayPoolNotification::Message { message, .. } = notification {
|
|
if let RelayMessage::Event {
|
|
event,
|
|
subscription_id,
|
|
} = message
|
|
{
|
|
match event.kind {
|
|
Kind::GiftWrap => {
|
|
match client.unwrap_gift_wrap(&event).await {
|
|
Ok(UnwrappedGift { mut rumor, sender }) => {
|
|
// Request metadata
|
|
if let Err(e) = mta_tx.send(sender).await {
|
|
warn!("Send error: {}", e)
|
|
};
|
|
|
|
// Compute event id if not exist
|
|
rumor.ensure_id();
|
|
|
|
if let Some(id) = rumor.id {
|
|
let ev = Event::new(
|
|
id,
|
|
rumor.pubkey,
|
|
rumor.created_at,
|
|
rumor.kind,
|
|
rumor.tags,
|
|
rumor.content,
|
|
sig,
|
|
);
|
|
|
|
// Save rumor to database to further query
|
|
if let Err(e) = client.database().save_event(&ev).await {
|
|
warn!("Save error: {}", e);
|
|
}
|
|
|
|
// Send event back to channel
|
|
if subscription_id == new_message {
|
|
if let Err(e) = signal_tx.send(Signal::Event(ev)).await
|
|
{
|
|
warn!("Send error: {}", e)
|
|
}
|
|
}
|
|
}
|
|
}
|
|
Err(e) => warn!("Unwrap error: {}", e),
|
|
}
|
|
}
|
|
Kind::ContactList => {
|
|
let public_keys: Vec<PublicKey> =
|
|
event.tags.public_keys().copied().collect();
|
|
|
|
for public_key in public_keys.into_iter() {
|
|
if let Err(e) = mta_tx.send(public_key).await {
|
|
warn!("Send error: {}", e)
|
|
};
|
|
}
|
|
}
|
|
_ => {}
|
|
}
|
|
} else if let RelayMessage::EndOfStoredEvents(subscription_id) = message {
|
|
if subscription_id == all_messages {
|
|
if let Err(e) = signal_tx.send(Signal::Eose).await {
|
|
warn!("Send error: {}", e)
|
|
};
|
|
}
|
|
}
|
|
}
|
|
}
|
|
});
|
|
|
|
// Handle metadata request
|
|
// Merge all requests into single subscription
|
|
tokio::spawn(async move { handle_metadata(client, mta_rx).await });
|
|
|
|
Application::new()
|
|
.with_assets(Assets)
|
|
.with_http_client(Arc::new(reqwest_client::ReqwestClient::new()))
|
|
.run(move |cx| {
|
|
// App state
|
|
AppRegistry::set_global(cx);
|
|
// Chat state
|
|
ChatRegistry::set_global(cx);
|
|
|
|
// Initialize components
|
|
ui::init(cx);
|
|
|
|
// Set quit action
|
|
cx.on_action(quit);
|
|
|
|
// Spawn a thread to handle Nostr events
|
|
cx.spawn(|async_cx| async move {
|
|
let (tx, rx) = smol::channel::bounded::<Signal>(4096);
|
|
|
|
async_cx
|
|
.background_executor()
|
|
.spawn(async move {
|
|
while let Some(signal) = signal_rx.recv().await {
|
|
if let Err(e) = tx.send(signal).await {
|
|
warn!("Send error: {}", e)
|
|
}
|
|
}
|
|
})
|
|
.detach();
|
|
|
|
while let Ok(signal) = rx.recv().await {
|
|
match signal {
|
|
Signal::Eose => {
|
|
_ = async_cx.update_global::<ChatRegistry, _>(|chat, cx| {
|
|
chat.load(cx);
|
|
});
|
|
}
|
|
Signal::Event(event) => {
|
|
_ = async_cx.update_global::<ChatRegistry, _>(|state, cx| {
|
|
state.receive(event, cx)
|
|
});
|
|
}
|
|
}
|
|
}
|
|
})
|
|
.detach();
|
|
|
|
// Spawn a thread to update Nostr signer
|
|
cx.spawn(|async_cx| {
|
|
let task = cx.read_credentials(KEYRING_SERVICE);
|
|
|
|
async move {
|
|
if let Ok(Some((npub, secret))) = task.await {
|
|
let public_key =
|
|
PublicKey::from_bech32(&npub).expect("Public Key isn't valid.");
|
|
|
|
let query: anyhow::Result<Metadata, anyhow::Error> = async_cx
|
|
.background_executor()
|
|
.spawn(async move {
|
|
let hex = String::from_utf8(secret)?;
|
|
let keys = Keys::parse(&hex)?;
|
|
|
|
// Update signer
|
|
_ = client.set_signer(keys).await;
|
|
|
|
// Get metadata
|
|
if let Some(metadata) =
|
|
client.database().metadata(public_key).await?
|
|
{
|
|
Ok(metadata)
|
|
} else {
|
|
Ok(Metadata::new())
|
|
}
|
|
})
|
|
.await;
|
|
|
|
if let Ok(metadata) = query {
|
|
_ = async_cx.update_global::<AppRegistry, _>(|state, cx| {
|
|
state.set_user(Contact::new(public_key, metadata), cx);
|
|
});
|
|
}
|
|
} else {
|
|
_ = async_cx.update_global::<AppRegistry, _>(|state, _| {
|
|
state.is_loading = false;
|
|
});
|
|
}
|
|
}
|
|
})
|
|
.detach();
|
|
|
|
let opts = WindowOptions {
|
|
#[cfg(not(target_os = "linux"))]
|
|
titlebar: Some(TitlebarOptions {
|
|
title: Some(SharedString::new_static(APP_NAME)),
|
|
traffic_light_position: Some(point(px(9.0), px(9.0))),
|
|
appears_transparent: true,
|
|
}),
|
|
window_bounds: Some(WindowBounds::Windowed(Bounds::centered(
|
|
None,
|
|
size(px(900.0), px(680.0)),
|
|
cx,
|
|
))),
|
|
#[cfg(target_os = "linux")]
|
|
window_background: WindowBackgroundAppearance::Transparent,
|
|
#[cfg(target_os = "linux")]
|
|
window_decorations: Some(WindowDecorations::Client),
|
|
kind: WindowKind::Normal,
|
|
..Default::default()
|
|
};
|
|
|
|
cx.open_window(opts, |window, cx| {
|
|
window.set_window_title(APP_NAME);
|
|
window.set_app_id(APP_ID);
|
|
cx.activate(true);
|
|
cx.new(|cx| Root::new(cx.new(|cx| AppView::new(window, cx)).into(), window, cx))
|
|
})
|
|
.expect("System error");
|
|
});
|
|
}
|
|
|
|
async fn handle_metadata(client: &'static Client, mut mta_rx: mpsc::Receiver<PublicKey>) {
|
|
let queue: Arc<Mutex<HashSet<PublicKey>>> = Arc::new(Mutex::new(HashSet::new()));
|
|
let queue_clone = Arc::clone(&queue);
|
|
|
|
let (tx, mut rx) = mpsc::channel::<PublicKey>(200);
|
|
|
|
tokio::spawn(async move {
|
|
while let Some(public_key) = mta_rx.recv().await {
|
|
queue_clone.lock().await.insert(public_key);
|
|
_ = tx.send(public_key).await;
|
|
}
|
|
});
|
|
|
|
tokio::spawn(async move {
|
|
while rx.recv().await.is_some() {
|
|
sleep(Duration::from_millis(METADATA_DELAY)).await;
|
|
|
|
let authors: Vec<PublicKey> = queue.lock().await.drain().collect();
|
|
let total = authors.len();
|
|
|
|
if total > 0 {
|
|
let filter = Filter::new()
|
|
.authors(authors)
|
|
.kind(Kind::Metadata)
|
|
.limit(total);
|
|
|
|
if let Err(e) = client.sync(filter, &SyncOptions::default()).await {
|
|
warn!("Error: {}", e);
|
|
}
|
|
}
|
|
}
|
|
});
|
|
}
|
|
|
|
fn quit(_: &Quit, cx: &mut App) {
|
|
cx.quit();
|
|
}
|