wip: refactor

This commit is contained in:
2025-01-10 10:39:51 +07:00
parent 80506b72d9
commit 38d0061796
6 changed files with 533 additions and 314 deletions

View File

@@ -15,7 +15,7 @@ pub struct RoomMessage {
#[allow(dead_code)]
author: PublicKey,
fallback: SharedString,
metadata: Option<Metadata>,
metadata: Metadata,
content: SharedString,
created_at: SharedString,
}
@@ -23,7 +23,7 @@ pub struct RoomMessage {
impl RoomMessage {
pub fn new(
author: PublicKey,
metadata: Option<Metadata>,
metadata: Metadata,
content: String,
created_at: Timestamp,
) -> Self {
@@ -55,18 +55,14 @@ impl RenderOnce for RoomMessage {
.text_color(cx.theme().muted_foreground)
})
.child(div().flex_shrink_0().map(|this| {
if let Some(metadata) = self.metadata.clone() {
if let Some(picture) = metadata.picture {
this.child(
img(format!(
"{}/?url={}&w=100&h=100&fit=cover&mask=circle&n=-1",
IMAGE_SERVICE, picture
))
.size_8(),
)
} else {
this.child(img("brand/avatar.png").size_8().rounded_full())
}
if let Some(picture) = self.metadata.picture {
this.child(
img(format!(
"{}/?url={}&w=100&h=100&fit=cover&mask=circle&n=-1",
IMAGE_SERVICE, picture
))
.size_8(),
)
} else {
this.child(img("brand/avatar.png").size_8().rounded_full())
}
@@ -84,12 +80,8 @@ impl RenderOnce for RoomMessage {
.gap_2()
.text_xs()
.child(div().font_semibold().map(|this| {
if let Some(metadata) = self.metadata {
if let Some(display_name) = metadata.display_name {
this.child(display_name)
} else {
this.child(self.fallback)
}
if let Some(display_name) = self.metadata.display_name {
this.child(display_name)
} else {
this.child(self.fallback)
}

View File

@@ -1,20 +1,29 @@
use crate::{get_client, states::chat::Room, utils::room_hash};
use gpui::{
div, AnyElement, AppContext, EventEmitter, FocusHandle, FocusableView, IntoElement,
ParentElement, Render, SharedString, Styled, View, VisualContext, WindowContext,
div, list, px, AnyElement, AppContext, Context, EventEmitter, Flatten, FocusHandle,
FocusableView, IntoElement, ListAlignment, ListState, Model, ParentElement, PathPromptOptions,
Pixels, Render, SharedString, Styled, View, ViewContext, VisualContext, WindowContext,
};
use itertools::Itertools;
use message::RoomMessage;
use nostr_sdk::prelude::*;
use room::RoomPanel;
use std::sync::Arc;
use ui::{
button::Button,
button::{Button, ButtonVariants},
dock::{Panel, PanelEvent, PanelState},
input::{InputEvent, TextInput},
popup_menu::PopupMenu,
theme::ActiveTheme,
v_flex, Icon, IconName,
};
use crate::states::chat::Room;
mod message;
mod room;
#[derive(Clone)]
pub struct State {
count: usize,
items: Vec<RoomMessage>,
}
pub struct ChatPanel {
// Panel
@@ -22,38 +31,148 @@ pub struct ChatPanel {
closeable: bool,
zoomable: bool,
focus_handle: FocusHandle,
// Room
// Chat Room
id: SharedString,
room: View<RoomPanel>,
metadata: Option<Metadata>,
room: Arc<Room>,
input: View<TextInput>,
list: ListState,
state: Model<State>,
}
impl ChatPanel {
pub fn new(room: &Arc<Room>, cx: &mut WindowContext) -> View<Self> {
let room = Arc::clone(room);
let id = room.id.clone();
let title = room.title.clone();
let metadata = room.metadata.clone();
let name = room.title.clone().unwrap_or("Untitled".into());
let room = cx.new_view(|cx| {
let view = RoomPanel::new(room, cx);
// Load messages
view.load(cx);
// Subscribe for new messages
view.subscribe(cx);
cx.observe_new_views::<Self>(|this, cx| {
this.load_messages(cx);
})
.detach();
view
});
cx.new_view(|cx| {
// Form
let input = cx.new_view(|cx| {
TextInput::new(cx)
.appearance(false)
.text_size(ui::Size::Small)
.placeholder("Message...")
.cleanable()
});
cx.new_view(|cx| Self {
name: title.unwrap_or("Untitled".into()),
closeable: true,
zoomable: true,
focus_handle: cx.focus_handle(),
id,
room,
metadata,
// Send message when user presses enter on form.
cx.subscribe(&input, move |this: &mut ChatPanel, _, input_event, cx| {
if let InputEvent::PressEnter = input_event {
this.send_message(cx);
}
})
.detach();
let state = cx.new_model(|_| State {
count: 0,
items: vec![],
});
cx.observe(&state, |this, model, cx| {
let items = model.read(cx).items.clone();
this.list = ListState::new(
items.len(),
ListAlignment::Bottom,
Pixels(256.),
move |idx, _cx| {
let item = items.get(idx).unwrap().clone();
div().child(item).into_any_element()
},
);
cx.notify();
})
.detach();
let list = ListState::new(0, ListAlignment::Bottom, Pixels(256.), move |_, _| {
div().into_any_element()
});
Self {
closeable: true,
zoomable: true,
focus_handle: cx.focus_handle(),
id,
name,
room,
input,
list,
state,
}
})
}
fn load_messages(&self, cx: &mut ViewContext<Self>) {
let members = self.room.members.clone();
let async_state = self.state.clone();
let id = self.room.id.to_string();
let client = get_client();
let mut async_cx = cx.to_async();
cx.foreground_executor()
.spawn(async move {
let events: anyhow::Result<Events, anyhow::Error> = async_cx
.background_executor()
.spawn({
let pubkeys = members.iter().map(|m| m.public_key()).collect::<Vec<_>>();
async move {
let signer = client.signer().await?;
let author = signer.get_public_key().await?;
let recv = Filter::new()
.kind(Kind::PrivateDirectMessage)
.author(author)
.pubkeys(pubkeys.clone());
let send = Filter::new()
.kind(Kind::PrivateDirectMessage)
.authors(pubkeys)
.pubkey(author);
// Get all DM events in database
let query = client.database().query(vec![recv, send]).await?;
Ok(query)
}
})
.await;
if let Ok(events) = events {
let items: Vec<RoomMessage> = events
.into_iter()
.sorted_by_key(|ev| ev.created_at)
.map(|ev| {
let metadata = members
.iter()
.find(|&m| m.public_key() == ev.pubkey)
.unwrap()
.metadata();
RoomMessage::new(ev.pubkey, metadata, ev.content, ev.created_at)
})
.collect();
let total = items.len();
_ = async_cx.update_model(&async_state, |a, b| {
a.items = items;
a.count = total;
b.notify();
});
}
})
.detach();
}
fn send_message(&mut self, cx: &mut ViewContext<Self>) {}
}
impl Panel for ChatPanel {
@@ -62,7 +181,7 @@ impl Panel for ChatPanel {
}
fn panel_metadata(&self) -> Option<Metadata> {
self.metadata.clone()
None
}
fn title(&self, _cx: &WindowContext) -> AnyElement {
@@ -99,7 +218,52 @@ impl FocusableView for ChatPanel {
}
impl Render for ChatPanel {
fn render(&mut self, _cx: &mut gpui::ViewContext<Self>) -> impl IntoElement {
div().size_full().child(self.room.clone())
fn render(&mut self, cx: &mut ViewContext<Self>) -> impl IntoElement {
v_flex()
.size_full()
.child(list(self.list.clone()).flex_1())
.child(
div()
.flex_shrink_0()
.w_full()
.h_12()
.flex()
.items_center()
.gap_2()
.px_2()
.child(
Button::new("upload")
.icon(Icon::new(IconName::Upload))
.ghost()
.on_click(|_, cx| {
let paths = cx.prompt_for_paths(PathPromptOptions {
files: true,
directories: false,
multiple: false,
});
cx.spawn(move |_async_cx| async move {
match Flatten::flatten(paths.await.map_err(|e| e.into())) {
Ok(Some(paths)) => {
// TODO: upload file to blossom server
println!("Paths: {:?}", paths)
}
Ok(None) => {}
Err(_) => {}
}
})
.detach();
}),
)
.child(
div()
.flex_1()
.flex()
.bg(cx.theme().muted)
.rounded(px(cx.theme().radius))
.px_2()
.child(self.input.clone()),
),
)
}
}

View File

@@ -15,7 +15,7 @@ use ui::{
use super::message::RoomMessage;
use crate::{
get_client,
states::chat::{ChatRegistry, Room},
states::chat::{ChatRegistry, Member, Room},
};
#[derive(Clone)]
@@ -27,7 +27,7 @@ pub struct Messages {
pub struct RoomPanel {
id: SharedString,
owner: PublicKey,
members: Arc<[PublicKey]>,
members: Vec<Member>,
// Form
input: View<TextInput>,
// Messages
@@ -38,7 +38,7 @@ pub struct RoomPanel {
impl RoomPanel {
pub fn new(room: &Arc<Room>, cx: &mut ViewContext<'_, Self>) -> Self {
let id = room.id.clone();
let members: Arc<[PublicKey]> = room.members.clone().into();
let members = room.members.clone();
let owner = room.owner;
// Form
@@ -98,20 +98,21 @@ impl RoomPanel {
let async_messages = self.messages.clone();
let mut async_cx = cx.to_async();
let public_keys: Vec<PublicKey> = self.members.iter().map(|m| m.public_key()).collect();
cx.foreground_executor()
.spawn({
let client = get_client();
let owner = self.owner;
let members = self.members.to_vec();
let recv = Filter::new()
.kind(Kind::PrivateDirectMessage)
.author(owner)
.pubkeys(members.clone());
.pubkeys(public_keys.clone());
let send = Filter::new()
.kind(Kind::PrivateDirectMessage)
.authors(members)
.authors(public_keys)
.pubkey(owner);
async move {
@@ -200,8 +201,9 @@ impl RoomPanel {
}
fn send_message(&mut self, cx: &mut ViewContext<Self>) {
let members = self.members.clone();
let members: Vec<PublicKey> = self.members.iter().map(|m| m.public_key()).collect();
let members2 = members.clone();
let content = self.input.read(cx).text().to_string();
let content2 = content.clone();
let content3 = content2.clone();
@@ -227,19 +229,21 @@ impl RoomPanel {
async_cx
.background_executor()
.spawn(async move {
for member in members.iter() {
let tags: Vec<Tag> = members
.iter()
.filter_map(|public_key| {
if public_key != member {
Some(Tag::public_key(*public_key))
} else {
None
}
})
.collect();
let extra_tags: Vec<Tag> = members
.iter()
.filter_map(|m| {
if m != &current_user {
Some(Tag::public_key(*m))
} else {
None
}
})
.collect();
_ = client.send_private_msg(*member, &content, tags).await;
for member in members.iter() {
_ = client
.send_private_msg(*member, &content, extra_tags.clone())
.await
}
})
.detach();
@@ -248,18 +252,20 @@ impl RoomPanel {
async_cx
.background_executor()
.spawn(async move {
let tags: Vec<Tag> = members2
let extra_tags: Vec<Tag> = members2
.iter()
.filter_map(|public_key| {
if public_key != &current_user {
Some(Tag::public_key(*public_key))
.filter_map(|m| {
if m != &current_user {
Some(Tag::public_key(*m))
} else {
None
}
})
.collect();
_ = client.send_private_msg(current_user, content2, tags).await;
_ = client
.send_private_msg(current_user, content2, extra_tags)
.await;
})
.detach();