wip: refactor
This commit is contained in:
@@ -277,8 +277,8 @@ async fn main() {
|
|||||||
while let Ok(signal) = rx.recv().await {
|
while let Ok(signal) = rx.recv().await {
|
||||||
match signal {
|
match signal {
|
||||||
Signal::Eose => {
|
Signal::Eose => {
|
||||||
_ = async_cx.update_global::<ChatRegistry, _>(|state, cx| {
|
_ = async_cx.update_global::<ChatRegistry, _>(|chat, cx| {
|
||||||
state.init(cx);
|
chat.init(cx);
|
||||||
});
|
});
|
||||||
}
|
}
|
||||||
Signal::Metadata(public_key) => {
|
Signal::Metadata(public_key) => {
|
||||||
@@ -290,16 +290,18 @@ async fn main() {
|
|||||||
let metadata = async_cx
|
let metadata = async_cx
|
||||||
.background_executor()
|
.background_executor()
|
||||||
.spawn(async move {
|
.spawn(async move {
|
||||||
client
|
if let Ok(metadata) =
|
||||||
.database()
|
client.database().metadata(event.pubkey).await
|
||||||
.metadata(event.pubkey)
|
{
|
||||||
.await
|
metadata.unwrap_or_default()
|
||||||
.unwrap_or_default()
|
} else {
|
||||||
|
Metadata::new()
|
||||||
|
}
|
||||||
})
|
})
|
||||||
.await;
|
.await;
|
||||||
|
|
||||||
_ = async_cx.update_global::<ChatRegistry, _>(|state, cx| {
|
_ = async_cx.update_global::<ChatRegistry, _>(|chat, cx| {
|
||||||
state.new_message(event, metadata, cx)
|
chat.receive(event, metadata, cx)
|
||||||
});
|
});
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -1,199 +0,0 @@
|
|||||||
use crate::{get_client, utils::room_hash};
|
|
||||||
use gpui::{AppContext, Context, Global, Model, SharedString, WeakModel};
|
|
||||||
use itertools::Itertools;
|
|
||||||
use nostr_sdk::prelude::*;
|
|
||||||
use profile::cut_public_key;
|
|
||||||
use rnglib::{Language, RNG};
|
|
||||||
use serde::Deserialize;
|
|
||||||
use std::{
|
|
||||||
cmp::Reverse,
|
|
||||||
collections::HashMap,
|
|
||||||
sync::{Arc, RwLock},
|
|
||||||
};
|
|
||||||
|
|
||||||
#[derive(Clone, PartialEq, Eq, Deserialize)]
|
|
||||||
pub struct Member {
|
|
||||||
public_key: PublicKey,
|
|
||||||
metadata: Metadata,
|
|
||||||
}
|
|
||||||
|
|
||||||
impl Member {
|
|
||||||
pub fn new(public_key: PublicKey, metadata: Metadata) -> Self {
|
|
||||||
Self {
|
|
||||||
public_key,
|
|
||||||
metadata,
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
pub fn public_key(&self) -> PublicKey {
|
|
||||||
self.public_key
|
|
||||||
}
|
|
||||||
|
|
||||||
pub fn metadata(&self) -> Metadata {
|
|
||||||
self.metadata.clone()
|
|
||||||
}
|
|
||||||
|
|
||||||
pub fn name(&self) -> String {
|
|
||||||
if let Some(display_name) = &self.metadata.display_name {
|
|
||||||
if !display_name.is_empty() {
|
|
||||||
return display_name.clone();
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
if let Some(name) = &self.metadata.name {
|
|
||||||
if !name.is_empty() {
|
|
||||||
return name.clone();
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
cut_public_key(self.public_key)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
#[derive(Clone, PartialEq, Eq, Deserialize)]
|
|
||||||
pub struct Room {
|
|
||||||
pub id: SharedString,
|
|
||||||
pub owner: PublicKey,
|
|
||||||
pub members: Vec<Member>,
|
|
||||||
pub last_seen: Timestamp,
|
|
||||||
pub title: Option<SharedString>,
|
|
||||||
}
|
|
||||||
|
|
||||||
impl Room {
|
|
||||||
pub fn new(
|
|
||||||
id: SharedString,
|
|
||||||
owner: PublicKey,
|
|
||||||
last_seen: Timestamp,
|
|
||||||
title: Option<SharedString>,
|
|
||||||
members: Vec<Member>,
|
|
||||||
) -> Self {
|
|
||||||
let title = if title.is_none() {
|
|
||||||
let rng = RNG::from(&Language::Roman);
|
|
||||||
let name = rng.generate_names(2, true).join("-").to_lowercase();
|
|
||||||
|
|
||||||
Some(name.into())
|
|
||||||
} else {
|
|
||||||
title
|
|
||||||
};
|
|
||||||
|
|
||||||
Self {
|
|
||||||
id,
|
|
||||||
title,
|
|
||||||
members,
|
|
||||||
last_seen,
|
|
||||||
owner,
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
#[derive(Clone, Debug)]
|
|
||||||
pub struct Message {
|
|
||||||
pub event: Event,
|
|
||||||
pub metadata: Option<Metadata>,
|
|
||||||
}
|
|
||||||
|
|
||||||
impl Message {
|
|
||||||
pub fn new(event: Event, metadata: Option<Metadata>) -> Self {
|
|
||||||
// TODO: parse event's content
|
|
||||||
Self { event, metadata }
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
type Inbox = Vec<Event>;
|
|
||||||
type Messages = RwLock<HashMap<SharedString, Arc<RwLock<Vec<Message>>>>>;
|
|
||||||
|
|
||||||
pub struct ChatRegistry {
|
|
||||||
messages: Model<Messages>,
|
|
||||||
inbox: Model<Inbox>,
|
|
||||||
}
|
|
||||||
|
|
||||||
impl Global for ChatRegistry {}
|
|
||||||
|
|
||||||
impl ChatRegistry {
|
|
||||||
pub fn set_global(cx: &mut AppContext) {
|
|
||||||
let inbox = cx.new_model(|_| Vec::new());
|
|
||||||
let messages = cx.new_model(|_| RwLock::new(HashMap::new()));
|
|
||||||
|
|
||||||
cx.set_global(Self { inbox, messages });
|
|
||||||
}
|
|
||||||
|
|
||||||
pub fn init(&mut self, cx: &mut AppContext) {
|
|
||||||
let async_cx = cx.to_async();
|
|
||||||
// Get all current room's hashes
|
|
||||||
let hashes: Vec<u64> = self
|
|
||||||
.inbox
|
|
||||||
.read(cx)
|
|
||||||
.iter()
|
|
||||||
.map(|ev| room_hash(&ev.tags))
|
|
||||||
.collect();
|
|
||||||
|
|
||||||
cx.foreground_executor()
|
|
||||||
.spawn(async move {
|
|
||||||
let client = get_client();
|
|
||||||
let query: anyhow::Result<Vec<Event>, anyhow::Error> = async_cx
|
|
||||||
.background_executor()
|
|
||||||
.spawn(async move {
|
|
||||||
let signer = client.signer().await?;
|
|
||||||
let public_key = signer.get_public_key().await?;
|
|
||||||
|
|
||||||
let filter = Filter::new()
|
|
||||||
.kind(Kind::PrivateDirectMessage)
|
|
||||||
.author(public_key);
|
|
||||||
|
|
||||||
// Get all DM events from database
|
|
||||||
let events = client.database().query(vec![filter]).await?;
|
|
||||||
|
|
||||||
// Filter result
|
|
||||||
// 1. Only new rooms
|
|
||||||
// 2. Only unique rooms
|
|
||||||
// 3. Sorted by created_at
|
|
||||||
let result = events
|
|
||||||
.into_iter()
|
|
||||||
.filter(|ev| !hashes.iter().any(|h| h == &room_hash(&ev.tags)))
|
|
||||||
.unique_by(|ev| room_hash(&ev.tags))
|
|
||||||
.sorted_by_key(|ev| Reverse(ev.created_at))
|
|
||||||
.collect::<Vec<_>>();
|
|
||||||
|
|
||||||
Ok(result)
|
|
||||||
})
|
|
||||||
.await;
|
|
||||||
|
|
||||||
if let Ok(events) = query {
|
|
||||||
_ = async_cx.update_global::<Self, _>(|state, cx| {
|
|
||||||
state.inbox.update(cx, |model, cx| {
|
|
||||||
model.extend(events);
|
|
||||||
cx.notify();
|
|
||||||
});
|
|
||||||
});
|
|
||||||
}
|
|
||||||
})
|
|
||||||
.detach();
|
|
||||||
}
|
|
||||||
|
|
||||||
pub fn new_message(&mut self, event: Event, metadata: Option<Metadata>, cx: &mut AppContext) {
|
|
||||||
// Get room id
|
|
||||||
let room_id = SharedString::from(room_hash(&event.tags).to_string());
|
|
||||||
// Create message
|
|
||||||
let message = Message::new(event, metadata);
|
|
||||||
|
|
||||||
self.messages.update(cx, |this, cx| {
|
|
||||||
this.write()
|
|
||||||
.unwrap()
|
|
||||||
.entry(room_id)
|
|
||||||
.or_insert(Arc::new(RwLock::new(Vec::new())))
|
|
||||||
.write()
|
|
||||||
.unwrap()
|
|
||||||
.push(message);
|
|
||||||
|
|
||||||
cx.notify();
|
|
||||||
});
|
|
||||||
}
|
|
||||||
|
|
||||||
pub fn messages(&self) -> WeakModel<Messages> {
|
|
||||||
self.messages.downgrade()
|
|
||||||
}
|
|
||||||
|
|
||||||
pub fn inbox(&self) -> WeakModel<Inbox> {
|
|
||||||
self.inbox.downgrade()
|
|
||||||
}
|
|
||||||
}
|
|
||||||
173
crates/app/src/states/chat/mod.rs
Normal file
173
crates/app/src/states/chat/mod.rs
Normal file
@@ -0,0 +1,173 @@
|
|||||||
|
use crate::{get_client, utils::room_hash};
|
||||||
|
use gpui::{AppContext, Context, Global, Model, SharedString, WeakModel};
|
||||||
|
use itertools::Itertools;
|
||||||
|
use nostr_sdk::prelude::*;
|
||||||
|
use room::Room;
|
||||||
|
use std::{
|
||||||
|
cmp::Reverse,
|
||||||
|
collections::HashMap,
|
||||||
|
sync::{Arc, RwLock},
|
||||||
|
};
|
||||||
|
|
||||||
|
pub mod room;
|
||||||
|
|
||||||
|
#[derive(Clone, Debug)]
|
||||||
|
pub struct NewMessage {
|
||||||
|
pub event: Event,
|
||||||
|
pub metadata: Metadata,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl NewMessage {
|
||||||
|
pub fn new(event: Event, metadata: Metadata) -> Self {
|
||||||
|
// TODO: parse event's content
|
||||||
|
Self { event, metadata }
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
type NewMessages = RwLock<HashMap<SharedString, Arc<RwLock<Vec<NewMessage>>>>>;
|
||||||
|
|
||||||
|
pub struct ChatRegistry {
|
||||||
|
inbox: Model<Vec<Model<Room>>>,
|
||||||
|
new_messages: Model<NewMessages>,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl Global for ChatRegistry {}
|
||||||
|
|
||||||
|
impl ChatRegistry {
|
||||||
|
pub fn set_global(cx: &mut AppContext) {
|
||||||
|
let inbox = cx.new_model(|_| Vec::new());
|
||||||
|
let new_messages = cx.new_model(|_| RwLock::new(HashMap::new()));
|
||||||
|
|
||||||
|
cx.observe_new_models::<Room>(|this, cx| {
|
||||||
|
// Get all pubkeys to load metadata
|
||||||
|
let pubkeys: Vec<PublicKey> = this.members.iter().map(|m| m.public_key()).collect();
|
||||||
|
|
||||||
|
cx.spawn(|weak_model, mut async_cx| async move {
|
||||||
|
let query: Result<Vec<(PublicKey, Metadata)>, Error> = async_cx
|
||||||
|
.background_executor()
|
||||||
|
.spawn(async move {
|
||||||
|
let client = get_client();
|
||||||
|
let mut profiles = Vec::new();
|
||||||
|
|
||||||
|
for public_key in pubkeys.into_iter() {
|
||||||
|
let query = client.database().metadata(public_key).await?;
|
||||||
|
let metadata = query.unwrap_or_default();
|
||||||
|
|
||||||
|
profiles.push((public_key, metadata));
|
||||||
|
}
|
||||||
|
|
||||||
|
Ok(profiles)
|
||||||
|
})
|
||||||
|
.await;
|
||||||
|
|
||||||
|
if let Ok(profiles) = query {
|
||||||
|
if let Some(model) = weak_model.upgrade() {
|
||||||
|
_ = async_cx.update_model(&model, |model, cx| {
|
||||||
|
for profile in profiles.into_iter() {
|
||||||
|
model.set_metadata(profile.0, profile.1);
|
||||||
|
}
|
||||||
|
cx.notify();
|
||||||
|
});
|
||||||
|
}
|
||||||
|
}
|
||||||
|
})
|
||||||
|
.detach();
|
||||||
|
})
|
||||||
|
.detach();
|
||||||
|
|
||||||
|
cx.set_global(Self {
|
||||||
|
inbox,
|
||||||
|
new_messages,
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn init(&mut self, cx: &mut AppContext) {
|
||||||
|
let mut async_cx = cx.to_async();
|
||||||
|
let async_inbox = self.inbox.clone();
|
||||||
|
|
||||||
|
// Get all current room's id
|
||||||
|
let hashes: Vec<u64> = self
|
||||||
|
.inbox
|
||||||
|
.read(cx)
|
||||||
|
.iter()
|
||||||
|
.map(|room| room.read(cx).id)
|
||||||
|
.collect();
|
||||||
|
|
||||||
|
cx.foreground_executor()
|
||||||
|
.spawn(async move {
|
||||||
|
let client = get_client();
|
||||||
|
let query: anyhow::Result<Vec<Event>, anyhow::Error> = async_cx
|
||||||
|
.background_executor()
|
||||||
|
.spawn(async move {
|
||||||
|
let signer = client.signer().await?;
|
||||||
|
let public_key = signer.get_public_key().await?;
|
||||||
|
|
||||||
|
let filter = Filter::new()
|
||||||
|
.kind(Kind::PrivateDirectMessage)
|
||||||
|
.author(public_key);
|
||||||
|
|
||||||
|
// Get all DM events from database
|
||||||
|
let events = client.database().query(vec![filter]).await?;
|
||||||
|
|
||||||
|
// Filter result
|
||||||
|
// - Only unique rooms
|
||||||
|
// - Sorted by created_at
|
||||||
|
let result = events
|
||||||
|
.into_iter()
|
||||||
|
.filter(|ev| ev.tags.public_keys().peekable().peek().is_some())
|
||||||
|
.unique_by(|ev| room_hash(&ev.tags))
|
||||||
|
.sorted_by_key(|ev| Reverse(ev.created_at))
|
||||||
|
.collect::<Vec<_>>();
|
||||||
|
|
||||||
|
Ok(result)
|
||||||
|
})
|
||||||
|
.await;
|
||||||
|
|
||||||
|
if let Ok(events) = query {
|
||||||
|
_ = async_cx.update_model(&async_inbox, |model, cx| {
|
||||||
|
let items: Vec<Model<Room>> = events
|
||||||
|
.into_iter()
|
||||||
|
.filter_map(|ev| {
|
||||||
|
let id = room_hash(&ev.tags);
|
||||||
|
// Filter all seen events
|
||||||
|
if !hashes.iter().any(|h| h == &id) {
|
||||||
|
Some(cx.new_model(|_| Room::new(&ev)))
|
||||||
|
} else {
|
||||||
|
None
|
||||||
|
}
|
||||||
|
})
|
||||||
|
.collect();
|
||||||
|
|
||||||
|
model.extend(items);
|
||||||
|
cx.notify();
|
||||||
|
});
|
||||||
|
}
|
||||||
|
})
|
||||||
|
.detach();
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn inbox(&self) -> WeakModel<Vec<Model<Room>>> {
|
||||||
|
self.inbox.downgrade()
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn new_messages(&self) -> WeakModel<NewMessages> {
|
||||||
|
self.new_messages.downgrade()
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn receive(&mut self, event: Event, metadata: Metadata, cx: &mut AppContext) {
|
||||||
|
let entry = room_hash(&event.tags).to_string().into();
|
||||||
|
let message = NewMessage::new(event, metadata);
|
||||||
|
|
||||||
|
self.new_messages.update(cx, |this, cx| {
|
||||||
|
this.write()
|
||||||
|
.unwrap()
|
||||||
|
.entry(entry)
|
||||||
|
.or_insert(Arc::new(RwLock::new(Vec::new())))
|
||||||
|
.write()
|
||||||
|
.unwrap()
|
||||||
|
.push(message);
|
||||||
|
|
||||||
|
cx.notify();
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
||||||
97
crates/app/src/states/chat/room.rs
Normal file
97
crates/app/src/states/chat/room.rs
Normal file
@@ -0,0 +1,97 @@
|
|||||||
|
use crate::utils::{room_hash, shorted_public_key};
|
||||||
|
use gpui::SharedString;
|
||||||
|
use nostr_sdk::prelude::*;
|
||||||
|
use rnglib::{Language, RNG};
|
||||||
|
|
||||||
|
#[derive(Debug, Clone)]
|
||||||
|
pub struct Member {
|
||||||
|
public_key: PublicKey,
|
||||||
|
metadata: Metadata,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl Member {
|
||||||
|
pub fn new(public_key: PublicKey, metadata: Metadata) -> Self {
|
||||||
|
Self {
|
||||||
|
public_key,
|
||||||
|
metadata,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn public_key(&self) -> PublicKey {
|
||||||
|
self.public_key
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn metadata(&self) -> Metadata {
|
||||||
|
self.metadata.clone()
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn name(&self) -> String {
|
||||||
|
if let Some(display_name) = &self.metadata.display_name {
|
||||||
|
if !display_name.is_empty() {
|
||||||
|
return display_name.clone();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if let Some(name) = &self.metadata.name {
|
||||||
|
if !name.is_empty() {
|
||||||
|
return name.clone();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
shorted_public_key(self.public_key)
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn update(&mut self, metadata: &Metadata) {
|
||||||
|
self.metadata = metadata.clone()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Debug)]
|
||||||
|
pub struct Room {
|
||||||
|
pub id: u64,
|
||||||
|
pub title: Option<SharedString>,
|
||||||
|
pub members: Vec<Member>,
|
||||||
|
pub last_seen: Timestamp,
|
||||||
|
pub is_group: bool,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl Room {
|
||||||
|
pub fn new(event: &Event) -> Self {
|
||||||
|
let id = room_hash(&event.tags);
|
||||||
|
let last_seen = event.created_at;
|
||||||
|
|
||||||
|
let members: Vec<Member> = event
|
||||||
|
.tags
|
||||||
|
.public_keys()
|
||||||
|
.copied()
|
||||||
|
.map(|public_key| Member::new(public_key, Metadata::default()))
|
||||||
|
.collect();
|
||||||
|
|
||||||
|
let title = if let Some(tag) = event.tags.find(TagKind::Title) {
|
||||||
|
tag.content().map(|s| s.to_owned().into())
|
||||||
|
} else {
|
||||||
|
let rng = RNG::from(&Language::Roman);
|
||||||
|
let name = rng.generate_names(2, true).join("-").to_lowercase();
|
||||||
|
|
||||||
|
Some(name.into())
|
||||||
|
};
|
||||||
|
|
||||||
|
let is_group = members.len() > 1;
|
||||||
|
|
||||||
|
Self {
|
||||||
|
id,
|
||||||
|
members,
|
||||||
|
title,
|
||||||
|
last_seen,
|
||||||
|
is_group,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn set_metadata(&mut self, public_key: PublicKey, metadata: Metadata) {
|
||||||
|
for member in self.members.iter_mut() {
|
||||||
|
if member.public_key() == public_key {
|
||||||
|
member.update(&metadata);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -11,6 +11,11 @@ pub fn room_hash(tags: &Tags) -> u64 {
|
|||||||
hasher.finish()
|
hasher.finish()
|
||||||
}
|
}
|
||||||
|
|
||||||
|
pub fn shorted_public_key(public_key: PublicKey) -> String {
|
||||||
|
let pk = public_key.to_string();
|
||||||
|
format!("{}:{}", &pk[0..4], &pk[pk.len() - 4..])
|
||||||
|
}
|
||||||
|
|
||||||
pub fn show_npub(public_key: PublicKey, len: usize) -> String {
|
pub fn show_npub(public_key: PublicKey, len: usize) -> String {
|
||||||
let bech32 = public_key.to_bech32().unwrap_or_default();
|
let bech32 = public_key.to_bech32().unwrap_or_default();
|
||||||
let separator = " ... ";
|
let separator = " ... ";
|
||||||
|
|||||||
@@ -1,3 +1,4 @@
|
|||||||
|
use crate::{constants::IMAGE_SERVICE, get_client, states::app::AppRegistry};
|
||||||
use gpui::prelude::FluentBuilder;
|
use gpui::prelude::FluentBuilder;
|
||||||
use gpui::{
|
use gpui::{
|
||||||
actions, img, Context, IntoElement, Model, ObjectFit, ParentElement, Render, Styled,
|
actions, img, Context, IntoElement, Model, ObjectFit, ParentElement, Render, Styled,
|
||||||
@@ -10,9 +11,6 @@ use ui::{
|
|||||||
Icon, IconName, Sizable,
|
Icon, IconName, Sizable,
|
||||||
};
|
};
|
||||||
|
|
||||||
use crate::states::app::AppRegistry;
|
|
||||||
use crate::{constants::IMAGE_SERVICE, get_client};
|
|
||||||
|
|
||||||
actions!(account, [ToDo]);
|
actions!(account, [ToDo]);
|
||||||
|
|
||||||
pub struct Account {
|
pub struct Account {
|
||||||
|
|||||||
@@ -1,8 +1,5 @@
|
|||||||
use super::{
|
use super::{account::Account, onboarding::Onboarding, sidebar::Sidebar, welcome::WelcomePanel};
|
||||||
account::Account, chat::ChatPanel, onboarding::Onboarding, sidebar::Sidebar,
|
use crate::states::app::AppRegistry;
|
||||||
welcome::WelcomePanel,
|
|
||||||
};
|
|
||||||
use crate::states::{app::AppRegistry, chat::Room};
|
|
||||||
use gpui::{
|
use gpui::{
|
||||||
div, impl_actions, px, Axis, Context, Edges, InteractiveElement, IntoElement, Model,
|
div, impl_actions, px, Axis, Context, Edges, InteractiveElement, IntoElement, Model,
|
||||||
ParentElement, Render, Styled, View, ViewContext, VisualContext, WeakView, WindowContext,
|
ParentElement, Render, Styled, View, ViewContext, VisualContext, WeakView, WindowContext,
|
||||||
@@ -18,7 +15,7 @@ use ui::{
|
|||||||
|
|
||||||
#[derive(Clone, PartialEq, Eq, Deserialize)]
|
#[derive(Clone, PartialEq, Eq, Deserialize)]
|
||||||
pub enum PanelKind {
|
pub enum PanelKind {
|
||||||
Room(Arc<Room>),
|
Room(u64),
|
||||||
}
|
}
|
||||||
|
|
||||||
#[derive(Clone, PartialEq, Eq, Deserialize)]
|
#[derive(Clone, PartialEq, Eq, Deserialize)]
|
||||||
@@ -132,15 +129,17 @@ impl AppView {
|
|||||||
}
|
}
|
||||||
|
|
||||||
fn on_action_add_panel(&mut self, action: &AddPanel, cx: &mut ViewContext<Self>) {
|
fn on_action_add_panel(&mut self, action: &AddPanel, cx: &mut ViewContext<Self>) {
|
||||||
|
/*
|
||||||
match &action.panel {
|
match &action.panel {
|
||||||
PanelKind::Room(room) => {
|
PanelKind::Room(id) => {
|
||||||
let panel = Arc::new(ChatPanel::new(room, cx));
|
let panel = Arc::new(ChatPanel::new(id, cx));
|
||||||
|
|
||||||
self.dock.update(cx, |dock_area, cx| {
|
self.dock.update(cx, |dock_area, cx| {
|
||||||
dock_area.add_panel(panel, action.position, cx);
|
dock_area.add_panel(panel, action.position, cx);
|
||||||
});
|
});
|
||||||
}
|
}
|
||||||
};
|
};
|
||||||
|
*/
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
@@ -1,4 +1,4 @@
|
|||||||
use crate::{get_client, states::chat::Room, utils::room_hash};
|
use crate::{get_client, states::chat::room::Room};
|
||||||
use gpui::{
|
use gpui::{
|
||||||
div, list, px, AnyElement, AppContext, Context, EventEmitter, Flatten, FocusHandle,
|
div, list, px, AnyElement, AppContext, Context, EventEmitter, Flatten, FocusHandle,
|
||||||
FocusableView, IntoElement, ListAlignment, ListState, Model, ParentElement, PathPromptOptions,
|
FocusableView, IntoElement, ListAlignment, ListState, Model, ParentElement, PathPromptOptions,
|
||||||
@@ -40,8 +40,8 @@ pub struct ChatPanel {
|
|||||||
}
|
}
|
||||||
|
|
||||||
impl ChatPanel {
|
impl ChatPanel {
|
||||||
pub fn new(room: &Arc<Room>, cx: &mut WindowContext) -> View<Self> {
|
pub fn new(room_id: &u64, cx: &mut WindowContext) -> View<Self> {
|
||||||
let room = Arc::clone(room);
|
let room = Arc::new(room);
|
||||||
let id = room.id.clone();
|
let id = room.id.clone();
|
||||||
let name = room.title.clone().unwrap_or("Untitled".into());
|
let name = room.title.clone().unwrap_or("Untitled".into());
|
||||||
|
|
||||||
|
|||||||
@@ -1,5 +1,5 @@
|
|||||||
mod account;
|
mod account;
|
||||||
mod chat;
|
// mod chat;
|
||||||
mod onboarding;
|
mod onboarding;
|
||||||
mod sidebar;
|
mod sidebar;
|
||||||
mod welcome;
|
mod welcome;
|
||||||
|
|||||||
@@ -1,72 +1,35 @@
|
|||||||
use crate::{
|
use crate::{constants::IMAGE_SERVICE, states::chat::ChatRegistry, utils::ago};
|
||||||
constants::IMAGE_SERVICE,
|
|
||||||
get_client,
|
|
||||||
states::{
|
|
||||||
app::AppRegistry,
|
|
||||||
chat::{ChatRegistry, Member, Room},
|
|
||||||
},
|
|
||||||
utils::{ago, room_hash},
|
|
||||||
views::app::{AddPanel, PanelKind},
|
|
||||||
};
|
|
||||||
use gpui::prelude::FluentBuilder;
|
|
||||||
use gpui::{
|
use gpui::{
|
||||||
div, img, percentage, Context, InteractiveElement, IntoElement, Model, ParentElement, Render,
|
div, img, percentage, prelude::FluentBuilder, InteractiveElement, IntoElement, ParentElement,
|
||||||
SharedString, StatefulInteractiveElement, Styled, View, ViewContext, VisualContext,
|
Render, RenderOnce, SharedString, StatefulInteractiveElement, Styled, ViewContext,
|
||||||
WindowContext,
|
WindowContext,
|
||||||
};
|
};
|
||||||
use nostr_sdk::prelude::*;
|
|
||||||
use std::sync::Arc;
|
|
||||||
use ui::{skeleton::Skeleton, theme::ActiveTheme, v_flex, Collapsible, Icon, IconName, StyledExt};
|
use ui::{skeleton::Skeleton, theme::ActiveTheme, v_flex, Collapsible, Icon, IconName, StyledExt};
|
||||||
|
|
||||||
pub struct Inbox {
|
pub struct Inbox {
|
||||||
label: SharedString,
|
label: SharedString,
|
||||||
items: Model<Option<Vec<View<InboxListItem>>>>,
|
|
||||||
is_loading: bool,
|
|
||||||
is_collapsed: bool,
|
is_collapsed: bool,
|
||||||
}
|
}
|
||||||
|
|
||||||
impl Inbox {
|
impl Inbox {
|
||||||
pub fn new(cx: &mut ViewContext<'_, Self>) -> Self {
|
pub fn new(_cx: &mut ViewContext<'_, Self>) -> Self {
|
||||||
let items = cx.new_model(|_| None);
|
|
||||||
let inbox = cx.global::<ChatRegistry>().inbox();
|
|
||||||
|
|
||||||
if let Some(inbox) = inbox.upgrade() {
|
|
||||||
cx.observe(&inbox, |this, model, cx| {
|
|
||||||
this.load(model, cx);
|
|
||||||
})
|
|
||||||
.detach();
|
|
||||||
}
|
|
||||||
|
|
||||||
Self {
|
Self {
|
||||||
items,
|
|
||||||
label: "Inbox".into(),
|
label: "Inbox".into(),
|
||||||
is_loading: true,
|
|
||||||
is_collapsed: false,
|
is_collapsed: false,
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
pub fn load(&mut self, model: Model<Vec<Event>>, cx: &mut ViewContext<Self>) {
|
fn skeleton(&self, total: i32) -> impl IntoIterator<Item = impl IntoElement> {
|
||||||
let events = model.read(cx).clone();
|
(0..total).map(|_| {
|
||||||
let views: Vec<View<InboxListItem>> = events
|
div()
|
||||||
.into_iter()
|
.h_8()
|
||||||
.map(|event| {
|
.px_1()
|
||||||
cx.new_view(|cx| {
|
.flex()
|
||||||
let view = InboxListItem::new(event, cx);
|
.items_center()
|
||||||
// Initial metadata
|
.gap_2()
|
||||||
view.load_metadata(cx);
|
.child(Skeleton::new().flex_shrink_0().size_6().rounded_full())
|
||||||
|
.child(Skeleton::new().w_20().h_3().rounded_sm())
|
||||||
view
|
})
|
||||||
})
|
|
||||||
})
|
|
||||||
.collect();
|
|
||||||
|
|
||||||
self.items.update(cx, |model, cx| {
|
|
||||||
*model = Some(views);
|
|
||||||
cx.notify();
|
|
||||||
});
|
|
||||||
|
|
||||||
self.is_loading = false;
|
|
||||||
cx.notify();
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -84,22 +47,35 @@ impl Collapsible for Inbox {
|
|||||||
impl Render for Inbox {
|
impl Render for Inbox {
|
||||||
fn render(&mut self, cx: &mut ViewContext<Self>) -> impl IntoElement {
|
fn render(&mut self, cx: &mut ViewContext<Self>) -> impl IntoElement {
|
||||||
let mut content = div();
|
let mut content = div();
|
||||||
|
let weak_model = cx.global::<ChatRegistry>().inbox();
|
||||||
|
|
||||||
if self.is_loading {
|
if let Some(model) = weak_model.upgrade() {
|
||||||
content = content.children((0..5).map(|_| {
|
content = content.children(model.read(cx).iter().map(|model| {
|
||||||
div()
|
let room = model.read(cx);
|
||||||
.h_8()
|
let id = room.id.to_string().into();
|
||||||
.px_1()
|
let ago = ago(room.last_seen.as_u64()).into();
|
||||||
.flex()
|
// Get first member
|
||||||
.items_center()
|
let sender = room.members.first().unwrap();
|
||||||
.gap_2()
|
// Compute group name based on member' names
|
||||||
.child(Skeleton::new().flex_shrink_0().size_6().rounded_full())
|
let name: SharedString = room
|
||||||
.child(Skeleton::new().w_20().h_3().rounded_sm())
|
.members
|
||||||
|
.iter()
|
||||||
|
.map(|profile| profile.name())
|
||||||
|
.collect::<Vec<String>>()
|
||||||
|
.join(", ")
|
||||||
|
.into();
|
||||||
|
|
||||||
|
InboxListItem::new(
|
||||||
|
id,
|
||||||
|
ago,
|
||||||
|
room.is_group,
|
||||||
|
name,
|
||||||
|
sender.metadata().picture,
|
||||||
|
sender.name(),
|
||||||
|
)
|
||||||
}))
|
}))
|
||||||
} else if let Some(items) = self.items.read(cx).as_ref() {
|
|
||||||
content = content.children(items.clone())
|
|
||||||
} else {
|
} else {
|
||||||
// TODO: handle error
|
content = content.children(self.skeleton(5))
|
||||||
}
|
}
|
||||||
|
|
||||||
v_flex()
|
v_flex()
|
||||||
@@ -134,110 +110,38 @@ impl Render for Inbox {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[derive(Clone, IntoElement)]
|
||||||
struct InboxListItem {
|
struct InboxListItem {
|
||||||
id: SharedString,
|
id: SharedString,
|
||||||
created_at: Timestamp,
|
ago: SharedString,
|
||||||
owner: PublicKey,
|
|
||||||
pubkeys: Vec<PublicKey>,
|
|
||||||
members: Model<Vec<Member>>,
|
|
||||||
is_group: bool,
|
is_group: bool,
|
||||||
|
group_name: SharedString,
|
||||||
|
sender_avatar: Option<String>,
|
||||||
|
sender_name: String,
|
||||||
}
|
}
|
||||||
|
|
||||||
impl InboxListItem {
|
impl InboxListItem {
|
||||||
pub fn new(event: Event, cx: &mut ViewContext<'_, Self>) -> Self {
|
pub fn new(
|
||||||
let id = room_hash(&event.tags).to_string().into();
|
id: SharedString,
|
||||||
let created_at = event.created_at;
|
ago: SharedString,
|
||||||
let owner = event.pubkey;
|
is_group: bool,
|
||||||
|
group_name: SharedString,
|
||||||
let pubkeys: Vec<PublicKey> = event.tags.public_keys().copied().collect();
|
sender_avatar: Option<String>,
|
||||||
let is_group = pubkeys.len() > 1;
|
sender_name: String,
|
||||||
|
) -> Self {
|
||||||
let members = cx.new_model(|_| Vec::new());
|
|
||||||
let refreshs = cx.global_mut::<AppRegistry>().refreshs();
|
|
||||||
|
|
||||||
if let Some(refreshs) = refreshs.upgrade() {
|
|
||||||
cx.observe(&refreshs, |this, _, cx| {
|
|
||||||
this.load_metadata(cx);
|
|
||||||
})
|
|
||||||
.detach();
|
|
||||||
}
|
|
||||||
|
|
||||||
Self {
|
Self {
|
||||||
id,
|
id,
|
||||||
created_at,
|
ago,
|
||||||
owner,
|
|
||||||
pubkeys,
|
|
||||||
members,
|
|
||||||
is_group,
|
is_group,
|
||||||
|
group_name,
|
||||||
|
sender_avatar,
|
||||||
|
sender_name,
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
pub fn load_metadata(&self, cx: &mut ViewContext<Self>) {
|
|
||||||
let owner = self.owner;
|
|
||||||
let public_keys = self.pubkeys.clone();
|
|
||||||
let async_members = self.members.clone();
|
|
||||||
|
|
||||||
let mut async_cx = cx.to_async();
|
|
||||||
|
|
||||||
cx.foreground_executor()
|
|
||||||
.spawn({
|
|
||||||
let client = get_client();
|
|
||||||
|
|
||||||
async move {
|
|
||||||
let metadata: anyhow::Result<Vec<Member>, anyhow::Error> = async_cx
|
|
||||||
.background_executor()
|
|
||||||
.spawn(async move {
|
|
||||||
let mut result = Vec::new();
|
|
||||||
|
|
||||||
for public_key in public_keys.into_iter() {
|
|
||||||
let metadata = client.database().metadata(public_key).await?;
|
|
||||||
let profile = Member::new(public_key, metadata.unwrap_or_default());
|
|
||||||
|
|
||||||
result.push(profile);
|
|
||||||
}
|
|
||||||
|
|
||||||
let metadata = client.database().metadata(owner).await?;
|
|
||||||
let profile = Member::new(owner, metadata.unwrap_or_default());
|
|
||||||
|
|
||||||
result.push(profile);
|
|
||||||
|
|
||||||
Ok(result)
|
|
||||||
})
|
|
||||||
.await;
|
|
||||||
|
|
||||||
if let Ok(metadata) = metadata {
|
|
||||||
_ = async_cx.update_model(&async_members, |model, cx| {
|
|
||||||
*model = metadata;
|
|
||||||
cx.notify();
|
|
||||||
});
|
|
||||||
}
|
|
||||||
}
|
|
||||||
})
|
|
||||||
.detach();
|
|
||||||
}
|
|
||||||
|
|
||||||
pub fn action(&self, cx: &mut WindowContext<'_>) {
|
|
||||||
let members = self.members.read(cx).clone();
|
|
||||||
let room = Arc::new(Room::new(
|
|
||||||
self.id.clone(),
|
|
||||||
self.owner,
|
|
||||||
self.created_at,
|
|
||||||
None,
|
|
||||||
members,
|
|
||||||
));
|
|
||||||
|
|
||||||
cx.dispatch_action(Box::new(AddPanel {
|
|
||||||
panel: PanelKind::Room(room),
|
|
||||||
position: ui::dock::DockPlacement::Center,
|
|
||||||
}))
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
impl Render for InboxListItem {
|
impl RenderOnce for InboxListItem {
|
||||||
fn render(&mut self, cx: &mut ViewContext<Self>) -> impl IntoElement {
|
fn render(self, cx: &mut WindowContext) -> impl IntoElement {
|
||||||
let ago = ago(self.created_at.as_u64());
|
|
||||||
let members = self.members.read(cx);
|
|
||||||
|
|
||||||
let mut content = div()
|
let mut content = div()
|
||||||
.font_medium()
|
.font_medium()
|
||||||
.text_color(cx.theme().sidebar_accent_foreground);
|
.text_color(cx.theme().sidebar_accent_foreground);
|
||||||
@@ -248,58 +152,33 @@ impl Render for InboxListItem {
|
|||||||
.items_center()
|
.items_center()
|
||||||
.gap_2()
|
.gap_2()
|
||||||
.child(img("brand/avatar.png").size_6().rounded_full())
|
.child(img("brand/avatar.png").size_6().rounded_full())
|
||||||
.map(|this| {
|
.child(self.group_name)
|
||||||
let names: Vec<String> = members
|
|
||||||
.iter()
|
|
||||||
.filter_map(|m| {
|
|
||||||
if m.public_key() != self.owner {
|
|
||||||
Some(m.name())
|
|
||||||
} else {
|
|
||||||
None
|
|
||||||
}
|
|
||||||
})
|
|
||||||
.collect();
|
|
||||||
|
|
||||||
this.child(names.join(", "))
|
|
||||||
})
|
|
||||||
} else {
|
} else {
|
||||||
content = content.flex().items_center().gap_2().map(|this| {
|
content = content.flex().items_center().gap_2().map(|mut this| {
|
||||||
if let Some(member) = members.first() {
|
// Avatar
|
||||||
let mut child = this;
|
if let Some(picture) = self.sender_avatar {
|
||||||
|
this = this.child(
|
||||||
// Avatar
|
img(format!(
|
||||||
if let Some(picture) = member.metadata().picture.clone() {
|
"{}/?url={}&w=72&h=72&fit=cover&mask=circle&n=-1",
|
||||||
child = child.child(
|
IMAGE_SERVICE, picture
|
||||||
img(format!(
|
))
|
||||||
"{}/?url={}&w=72&h=72&fit=cover&mask=circle&n=-1",
|
.flex_shrink_0()
|
||||||
IMAGE_SERVICE, picture
|
.size_6()
|
||||||
))
|
.rounded_full(),
|
||||||
.flex_shrink_0()
|
);
|
||||||
.size_6()
|
|
||||||
.rounded_full(),
|
|
||||||
);
|
|
||||||
} else {
|
|
||||||
child = child.child(
|
|
||||||
img("brand/avatar.png")
|
|
||||||
.flex_shrink_0()
|
|
||||||
.size_6()
|
|
||||||
.rounded_full(),
|
|
||||||
);
|
|
||||||
}
|
|
||||||
|
|
||||||
// Display name
|
|
||||||
child = child.child(member.name());
|
|
||||||
|
|
||||||
child
|
|
||||||
} else {
|
} else {
|
||||||
this.child(
|
this = this.child(
|
||||||
img("brand/avatar.png")
|
img("brand/avatar.png")
|
||||||
.flex_shrink_0()
|
.flex_shrink_0()
|
||||||
.size_6()
|
.size_6()
|
||||||
.rounded_full(),
|
.rounded_full(),
|
||||||
)
|
);
|
||||||
.child("Unknown")
|
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Display name
|
||||||
|
this = this.child(self.sender_name);
|
||||||
|
|
||||||
|
this
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -319,11 +198,8 @@ impl Render for InboxListItem {
|
|||||||
.child(content)
|
.child(content)
|
||||||
.child(
|
.child(
|
||||||
div()
|
div()
|
||||||
.child(ago)
|
.child(self.ago)
|
||||||
.text_color(cx.theme().sidebar_accent_foreground.opacity(0.7)),
|
.text_color(cx.theme().sidebar_accent_foreground.opacity(0.7)),
|
||||||
)
|
)
|
||||||
.on_click(cx.listener(|this, _, cx| {
|
|
||||||
this.action(cx);
|
|
||||||
}))
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
Reference in New Issue
Block a user