catchup with main

This commit is contained in:
KCaverly 2023-10-06 16:30:31 +02:00
commit ecfece3ac4
72 changed files with 4218 additions and 1104 deletions

680
Cargo.lock generated

File diff suppressed because it is too large Load diff

View file

@ -64,6 +64,7 @@ members = [
"crates/sqlez", "crates/sqlez",
"crates/sqlez_macros", "crates/sqlez_macros",
"crates/feature_flags", "crates/feature_flags",
"crates/rich_text",
"crates/storybook", "crates/storybook",
"crates/sum_tree", "crates/sum_tree",
"crates/terminal", "crates/terminal",
@ -105,6 +106,7 @@ rand = { version = "0.8.5" }
refineable = { path = "./crates/refineable" } refineable = { path = "./crates/refineable" }
regex = { version = "1.5" } regex = { version = "1.5" }
rust-embed = { version = "8.0", features = ["include-exclude"] } rust-embed = { version = "8.0", features = ["include-exclude"] }
rusqlite = { version = "0.29.0", features = ["blob", "array", "modern_sqlite"] }
schemars = { version = "0.8" } schemars = { version = "0.8" }
serde = { version = "1.0", features = ["derive", "rc"] } serde = { version = "1.0", features = ["derive", "rc"] }
serde_derive = { version = "1.0", features = ["deserialize_in_place"] } serde_derive = { version = "1.0", features = ["deserialize_in_place"] }
@ -120,6 +122,7 @@ tree-sitter = "0.20"
unindent = { version = "0.1.7" } unindent = { version = "0.1.7" }
pretty_assertions = "1.3.0" pretty_assertions = "1.3.0"
git2 = { version = "0.15", default-features = false} git2 = { version = "0.15", default-features = false}
uuid = { version = "1.1.2", features = ["v4"] }
tree-sitter-bash = { git = "https://github.com/tree-sitter/tree-sitter-bash", rev = "1b0321ee85701d5036c334a6f04761cdc672e64c" } tree-sitter-bash = { git = "https://github.com/tree-sitter/tree-sitter-bash", rev = "1b0321ee85701d5036c334a6f04761cdc672e64c" }
tree-sitter-c = "0.20.1" tree-sitter-c = "0.20.1"

View file

@ -1,6 +1,6 @@
# syntax = docker/dockerfile:1.2 # syntax = docker/dockerfile:1.2
FROM rust:1.72-bullseye as builder FROM rust:1.73-bullseye as builder
WORKDIR app WORKDIR app
COPY . . COPY . .

View file

@ -27,7 +27,7 @@ log.workspace = true
parse_duration = "2.1.1" parse_duration = "2.1.1"
tiktoken-rs = "0.5.0" tiktoken-rs = "0.5.0"
matrixmultiply = "0.3.7" matrixmultiply = "0.3.7"
rusqlite = { version = "0.27.0", features = ["blob", "array", "modern_sqlite"] } rusqlite = { version = "0.29.0", features = ["blob", "array", "modern_sqlite"] }
bincode = "1.3.3" bincode = "1.3.3"
[dev-dependencies] [dev-dependencies]

View file

@ -21,11 +21,11 @@ search = { path = "../search" }
settings = { path = "../settings" } settings = { path = "../settings" }
theme = { path = "../theme" } theme = { path = "../theme" }
util = { path = "../util" } util = { path = "../util" }
uuid = { version = "1.1.2", features = ["v4"] }
workspace = { path = "../workspace" } workspace = { path = "../workspace" }
semantic_index = { path = "../semantic_index" } semantic_index = { path = "../semantic_index" }
project = { path = "../project" } project = { path = "../project" }
uuid.workspace = true
log.workspace = true log.workspace = true
anyhow.workspace = true anyhow.workspace = true
chrono = { version = "0.4", features = ["serde"] } chrono = { version = "0.4", features = ["serde"] }

View file

@ -291,10 +291,10 @@ impl ActiveCall {
&mut self, &mut self,
channel_id: u64, channel_id: u64,
cx: &mut ModelContext<Self>, cx: &mut ModelContext<Self>,
) -> Task<Result<()>> { ) -> Task<Result<ModelHandle<Room>>> {
if let Some(room) = self.room().cloned() { if let Some(room) = self.room().cloned() {
if room.read(cx).channel_id() == Some(channel_id) { if room.read(cx).channel_id() == Some(channel_id) {
return Task::ready(Ok(())); return Task::ready(Ok(room));
} else { } else {
room.update(cx, |room, cx| room.clear_state(cx)); room.update(cx, |room, cx| room.clear_state(cx));
} }
@ -309,7 +309,7 @@ impl ActiveCall {
this.update(&mut cx, |this, cx| { this.update(&mut cx, |this, cx| {
this.report_call_event("join channel", cx) this.report_call_event("join channel", cx)
}); });
Ok(()) Ok(room)
}) })
} }

View file

@ -104,6 +104,10 @@ impl Room {
self.channel_id self.channel_id
} }
pub fn is_sharing_project(&self) -> bool {
!self.shared_projects.is_empty()
}
#[cfg(any(test, feature = "test-support"))] #[cfg(any(test, feature = "test-support"))]
pub fn is_connected(&self) -> bool { pub fn is_connected(&self) -> bool {
if let Some(live_kit) = self.live_kit.as_ref() { if let Some(live_kit) = self.live_kit.as_ref() {
@ -594,6 +598,31 @@ impl Room {
.map_or(&[], |v| v.as_slice()) .map_or(&[], |v| v.as_slice())
} }
/// Returns the most 'active' projects, defined as most people in the project
pub fn most_active_project(&self) -> Option<(u64, u64)> {
let mut projects = HashMap::default();
let mut hosts = HashMap::default();
for participant in self.remote_participants.values() {
match participant.location {
ParticipantLocation::SharedProject { project_id } => {
*projects.entry(project_id).or_insert(0) += 1;
}
ParticipantLocation::External | ParticipantLocation::UnsharedProject => {}
}
for project in &participant.projects {
*projects.entry(project.id).or_insert(0) += 1;
hosts.insert(project.id, participant.user.id);
}
}
let mut pairs: Vec<(u64, usize)> = projects.into_iter().collect();
pairs.sort_by_key(|(_, count)| *count as i32);
pairs
.first()
.map(|(project_id, _)| (*project_id, hosts[&project_id]))
}
async fn handle_room_updated( async fn handle_room_updated(
this: ModelHandle<Self>, this: ModelHandle<Self>,
envelope: TypedEnvelope<proto::RoomUpdated>, envelope: TypedEnvelope<proto::RoomUpdated>,

View file

@ -23,6 +23,7 @@ language = { path = "../language" }
settings = { path = "../settings" } settings = { path = "../settings" }
feature_flags = { path = "../feature_flags" } feature_flags = { path = "../feature_flags" }
sum_tree = { path = "../sum_tree" } sum_tree = { path = "../sum_tree" }
clock = { path = "../clock" }
anyhow.workspace = true anyhow.workspace = true
futures.workspace = true futures.workspace = true
@ -38,7 +39,7 @@ smol.workspace = true
thiserror.workspace = true thiserror.workspace = true
time.workspace = true time.workspace = true
tiny_http = "0.8" tiny_http = "0.8"
uuid = { version = "1.1.2", features = ["v4"] } uuid.workspace = true
url = "2.2" url = "2.2"
serde.workspace = true serde.workspace = true
serde_derive.workspace = true serde_derive.workspace = true

View file

@ -2,7 +2,7 @@ mod channel_buffer;
mod channel_chat; mod channel_chat;
mod channel_store; mod channel_store;
pub use channel_buffer::{ChannelBuffer, ChannelBufferEvent}; pub use channel_buffer::{ChannelBuffer, ChannelBufferEvent, ACKNOWLEDGE_DEBOUNCE_INTERVAL};
pub use channel_chat::{ChannelChat, ChannelChatEvent, ChannelMessage, ChannelMessageId}; pub use channel_chat::{ChannelChat, ChannelChatEvent, ChannelMessage, ChannelMessageId};
pub use channel_store::{ pub use channel_store::{
Channel, ChannelData, ChannelEvent, ChannelId, ChannelMembership, ChannelPath, ChannelStore, Channel, ChannelData, ChannelEvent, ChannelId, ChannelMembership, ChannelPath, ChannelStore,

View file

@ -2,14 +2,17 @@ use crate::Channel;
use anyhow::Result; use anyhow::Result;
use client::{Client, Collaborator, UserStore}; use client::{Client, Collaborator, UserStore};
use collections::HashMap; use collections::HashMap;
use gpui::{AppContext, AsyncAppContext, Entity, ModelContext, ModelHandle}; use gpui::{AppContext, AsyncAppContext, Entity, ModelContext, ModelHandle, Task};
use language::proto::serialize_version;
use rpc::{ use rpc::{
proto::{self, PeerId}, proto::{self, PeerId},
TypedEnvelope, TypedEnvelope,
}; };
use std::sync::Arc; use std::{sync::Arc, time::Duration};
use util::ResultExt; use util::ResultExt;
pub const ACKNOWLEDGE_DEBOUNCE_INTERVAL: Duration = Duration::from_millis(250);
pub(crate) fn init(client: &Arc<Client>) { pub(crate) fn init(client: &Arc<Client>) {
client.add_model_message_handler(ChannelBuffer::handle_update_channel_buffer); client.add_model_message_handler(ChannelBuffer::handle_update_channel_buffer);
client.add_model_message_handler(ChannelBuffer::handle_update_channel_buffer_collaborators); client.add_model_message_handler(ChannelBuffer::handle_update_channel_buffer_collaborators);
@ -24,11 +27,13 @@ pub struct ChannelBuffer {
buffer_epoch: u64, buffer_epoch: u64,
client: Arc<Client>, client: Arc<Client>,
subscription: Option<client::Subscription>, subscription: Option<client::Subscription>,
acknowledge_task: Option<Task<Result<()>>>,
} }
pub enum ChannelBufferEvent { pub enum ChannelBufferEvent {
CollaboratorsChanged, CollaboratorsChanged,
Disconnected, Disconnected,
BufferEdited,
} }
impl Entity for ChannelBuffer { impl Entity for ChannelBuffer {
@ -36,6 +41,9 @@ impl Entity for ChannelBuffer {
fn release(&mut self, _: &mut AppContext) { fn release(&mut self, _: &mut AppContext) {
if self.connected { if self.connected {
if let Some(task) = self.acknowledge_task.take() {
task.detach();
}
self.client self.client
.send(proto::LeaveChannelBuffer { .send(proto::LeaveChannelBuffer {
channel_id: self.channel.id, channel_id: self.channel.id,
@ -81,6 +89,7 @@ impl ChannelBuffer {
client, client,
connected: true, connected: true,
collaborators: Default::default(), collaborators: Default::default(),
acknowledge_task: None,
channel, channel,
subscription: Some(subscription.set_model(&cx.handle(), &mut cx.to_async())), subscription: Some(subscription.set_model(&cx.handle(), &mut cx.to_async())),
user_store, user_store,
@ -159,19 +168,45 @@ impl ChannelBuffer {
&mut self, &mut self,
_: ModelHandle<language::Buffer>, _: ModelHandle<language::Buffer>,
event: &language::Event, event: &language::Event,
_: &mut ModelContext<Self>, cx: &mut ModelContext<Self>,
) { ) {
if let language::Event::Operation(operation) = event { match event {
let operation = language::proto::serialize_operation(operation); language::Event::Operation(operation) => {
self.client let operation = language::proto::serialize_operation(operation);
.send(proto::UpdateChannelBuffer { self.client
channel_id: self.channel.id, .send(proto::UpdateChannelBuffer {
operations: vec![operation], channel_id: self.channel.id,
}) operations: vec![operation],
.log_err(); })
.log_err();
}
language::Event::Edited => {
cx.emit(ChannelBufferEvent::BufferEdited);
}
_ => {}
} }
} }
pub fn acknowledge_buffer_version(&mut self, cx: &mut ModelContext<'_, ChannelBuffer>) {
let buffer = self.buffer.read(cx);
let version = buffer.version();
let buffer_id = buffer.remote_id();
let client = self.client.clone();
let epoch = self.epoch();
self.acknowledge_task = Some(cx.spawn_weak(|_, cx| async move {
cx.background().timer(ACKNOWLEDGE_DEBOUNCE_INTERVAL).await;
client
.send(proto::AckBufferOperation {
buffer_id,
epoch,
version: serialize_version(&version),
})
.ok();
Ok(())
}));
}
pub fn epoch(&self) -> u64 { pub fn epoch(&self) -> u64 {
self.buffer_epoch self.buffer_epoch
} }

View file

@ -1,4 +1,4 @@
use crate::Channel; use crate::{Channel, ChannelId, ChannelStore};
use anyhow::{anyhow, Result}; use anyhow::{anyhow, Result};
use client::{ use client::{
proto, proto,
@ -16,7 +16,9 @@ use util::{post_inc, ResultExt as _, TryFutureExt};
pub struct ChannelChat { pub struct ChannelChat {
channel: Arc<Channel>, channel: Arc<Channel>,
messages: SumTree<ChannelMessage>, messages: SumTree<ChannelMessage>,
channel_store: ModelHandle<ChannelStore>,
loaded_all_messages: bool, loaded_all_messages: bool,
last_acknowledged_id: Option<u64>,
next_pending_message_id: usize, next_pending_message_id: usize,
user_store: ModelHandle<UserStore>, user_store: ModelHandle<UserStore>,
rpc: Arc<Client>, rpc: Arc<Client>,
@ -34,7 +36,7 @@ pub struct ChannelMessage {
pub nonce: u128, pub nonce: u128,
} }
#[derive(Copy, Clone, Debug, PartialEq, Eq, PartialOrd, Ord)] #[derive(Copy, Clone, Debug, PartialEq, Eq, PartialOrd, Ord, Hash)]
pub enum ChannelMessageId { pub enum ChannelMessageId {
Saved(u64), Saved(u64),
Pending(usize), Pending(usize),
@ -55,6 +57,10 @@ pub enum ChannelChatEvent {
old_range: Range<usize>, old_range: Range<usize>,
new_count: usize, new_count: usize,
}, },
NewMessage {
channel_id: ChannelId,
message_id: u64,
},
} }
pub fn init(client: &Arc<Client>) { pub fn init(client: &Arc<Client>) {
@ -77,6 +83,7 @@ impl Entity for ChannelChat {
impl ChannelChat { impl ChannelChat {
pub async fn new( pub async fn new(
channel: Arc<Channel>, channel: Arc<Channel>,
channel_store: ModelHandle<ChannelStore>,
user_store: ModelHandle<UserStore>, user_store: ModelHandle<UserStore>,
client: Arc<Client>, client: Arc<Client>,
mut cx: AsyncAppContext, mut cx: AsyncAppContext,
@ -94,11 +101,13 @@ impl ChannelChat {
let mut this = Self { let mut this = Self {
channel, channel,
user_store, user_store,
channel_store,
rpc: client, rpc: client,
outgoing_messages_lock: Default::default(), outgoing_messages_lock: Default::default(),
messages: Default::default(), messages: Default::default(),
loaded_all_messages, loaded_all_messages,
next_pending_message_id: 0, next_pending_message_id: 0,
last_acknowledged_id: None,
rng: StdRng::from_entropy(), rng: StdRng::from_entropy(),
_subscription: subscription.set_model(&cx.handle(), &mut cx.to_async()), _subscription: subscription.set_model(&cx.handle(), &mut cx.to_async()),
}; };
@ -219,6 +228,26 @@ impl ChannelChat {
false false
} }
pub fn acknowledge_last_message(&mut self, cx: &mut ModelContext<Self>) {
if let ChannelMessageId::Saved(latest_message_id) = self.messages.summary().max_id {
if self
.last_acknowledged_id
.map_or(true, |acknowledged_id| acknowledged_id < latest_message_id)
{
self.rpc
.send(proto::AckChannelMessage {
channel_id: self.channel.id,
message_id: latest_message_id,
})
.ok();
self.last_acknowledged_id = Some(latest_message_id);
self.channel_store.update(cx, |store, cx| {
store.acknowledge_message_id(self.channel.id, latest_message_id, cx);
});
}
}
}
pub fn rejoin(&mut self, cx: &mut ModelContext<Self>) { pub fn rejoin(&mut self, cx: &mut ModelContext<Self>) {
let user_store = self.user_store.clone(); let user_store = self.user_store.clone();
let rpc = self.rpc.clone(); let rpc = self.rpc.clone();
@ -313,10 +342,15 @@ impl ChannelChat {
.payload .payload
.message .message
.ok_or_else(|| anyhow!("empty message"))?; .ok_or_else(|| anyhow!("empty message"))?;
let message_id = message.id;
let message = ChannelMessage::from_proto(message, &user_store, &mut cx).await?; let message = ChannelMessage::from_proto(message, &user_store, &mut cx).await?;
this.update(&mut cx, |this, cx| { this.update(&mut cx, |this, cx| {
this.insert_messages(SumTree::from_item(message, &()), cx) this.insert_messages(SumTree::from_item(message, &()), cx);
cx.emit(ChannelChatEvent::NewMessage {
channel_id: this.channel.id,
message_id,
})
}); });
Ok(()) Ok(())
@ -388,6 +422,7 @@ impl ChannelChat {
old_range: start_ix..end_ix, old_range: start_ix..end_ix,
new_count, new_count,
}); });
cx.notify(); cx.notify();
} }
} }

View file

@ -43,6 +43,8 @@ pub type ChannelData = (Channel, ChannelPath);
pub struct Channel { pub struct Channel {
pub id: ChannelId, pub id: ChannelId,
pub name: String, pub name: String,
pub unseen_note_version: Option<(u64, clock::Global)>,
pub unseen_message_id: Option<u64>,
} }
#[derive(Debug, PartialEq, Eq, PartialOrd, Ord, Hash, Clone, Serialize, Deserialize)] #[derive(Debug, PartialEq, Eq, PartialOrd, Ord, Hash, Clone, Serialize, Deserialize)]
@ -207,6 +209,64 @@ impl ChannelStore {
) )
} }
pub fn has_channel_buffer_changed(&self, channel_id: ChannelId) -> Option<bool> {
self.channel_index
.by_id()
.get(&channel_id)
.map(|channel| channel.unseen_note_version.is_some())
}
pub fn has_new_messages(&self, channel_id: ChannelId) -> Option<bool> {
self.channel_index
.by_id()
.get(&channel_id)
.map(|channel| channel.unseen_message_id.is_some())
}
pub fn notes_changed(
&mut self,
channel_id: ChannelId,
epoch: u64,
version: &clock::Global,
cx: &mut ModelContext<Self>,
) {
self.channel_index.note_changed(channel_id, epoch, version);
cx.notify();
}
pub fn new_message(
&mut self,
channel_id: ChannelId,
message_id: u64,
cx: &mut ModelContext<Self>,
) {
self.channel_index.new_message(channel_id, message_id);
cx.notify();
}
pub fn acknowledge_message_id(
&mut self,
channel_id: ChannelId,
message_id: u64,
cx: &mut ModelContext<Self>,
) {
self.channel_index
.acknowledge_message_id(channel_id, message_id);
cx.notify();
}
pub fn acknowledge_notes_version(
&mut self,
channel_id: ChannelId,
epoch: u64,
version: &clock::Global,
cx: &mut ModelContext<Self>,
) {
self.channel_index
.acknowledge_note_version(channel_id, epoch, version);
cx.notify();
}
pub fn open_channel_chat( pub fn open_channel_chat(
&mut self, &mut self,
channel_id: ChannelId, channel_id: ChannelId,
@ -214,10 +274,11 @@ impl ChannelStore {
) -> Task<Result<ModelHandle<ChannelChat>>> { ) -> Task<Result<ModelHandle<ChannelChat>>> {
let client = self.client.clone(); let client = self.client.clone();
let user_store = self.user_store.clone(); let user_store = self.user_store.clone();
let this = cx.handle();
self.open_channel_resource( self.open_channel_resource(
channel_id, channel_id,
|this| &mut this.opened_chats, |this| &mut this.opened_chats,
|channel, cx| ChannelChat::new(channel, user_store, client, cx), |channel, cx| ChannelChat::new(channel, this, user_store, client, cx),
cx, cx,
) )
} }
@ -779,6 +840,8 @@ impl ChannelStore {
Arc::new(Channel { Arc::new(Channel {
id: channel.id, id: channel.id,
name: channel.name, name: channel.name,
unseen_note_version: None,
unseen_message_id: None,
}), }),
), ),
} }
@ -787,7 +850,9 @@ impl ChannelStore {
let channels_changed = !payload.channels.is_empty() let channels_changed = !payload.channels.is_empty()
|| !payload.delete_channels.is_empty() || !payload.delete_channels.is_empty()
|| !payload.insert_edge.is_empty() || !payload.insert_edge.is_empty()
|| !payload.delete_edge.is_empty(); || !payload.delete_edge.is_empty()
|| !payload.unseen_channel_messages.is_empty()
|| !payload.unseen_channel_buffer_changes.is_empty();
if channels_changed { if channels_changed {
if !payload.delete_channels.is_empty() { if !payload.delete_channels.is_empty() {
@ -814,6 +879,22 @@ impl ChannelStore {
index.insert(channel) index.insert(channel)
} }
for unseen_buffer_change in payload.unseen_channel_buffer_changes {
let version = language::proto::deserialize_version(&unseen_buffer_change.version);
index.note_changed(
unseen_buffer_change.channel_id,
unseen_buffer_change.epoch,
&version,
);
}
for unseen_channel_message in payload.unseen_channel_messages {
index.new_messages(
unseen_channel_message.channel_id,
unseen_channel_message.message_id,
);
}
for edge in payload.insert_edge { for edge in payload.insert_edge {
index.insert_edge(edge.channel_id, edge.parent_id); index.insert_edge(edge.channel_id, edge.parent_id);
} }

View file

@ -38,6 +38,43 @@ impl ChannelIndex {
channels_by_id: &mut self.channels_by_id, channels_by_id: &mut self.channels_by_id,
} }
} }
pub fn acknowledge_note_version(
&mut self,
channel_id: ChannelId,
epoch: u64,
version: &clock::Global,
) {
if let Some(channel) = self.channels_by_id.get_mut(&channel_id) {
let channel = Arc::make_mut(channel);
if let Some((unseen_epoch, unseen_version)) = &channel.unseen_note_version {
if epoch > *unseen_epoch
|| epoch == *unseen_epoch && version.observed_all(unseen_version)
{
channel.unseen_note_version = None;
}
}
}
}
pub fn acknowledge_message_id(&mut self, channel_id: ChannelId, message_id: u64) {
if let Some(channel) = self.channels_by_id.get_mut(&channel_id) {
let channel = Arc::make_mut(channel);
if let Some(unseen_message_id) = channel.unseen_message_id {
if message_id >= unseen_message_id {
channel.unseen_message_id = None;
}
}
}
}
pub fn note_changed(&mut self, channel_id: ChannelId, epoch: u64, version: &clock::Global) {
insert_note_changed(&mut self.channels_by_id, channel_id, epoch, version);
}
pub fn new_message(&mut self, channel_id: ChannelId, message_id: u64) {
insert_new_message(&mut self.channels_by_id, channel_id, message_id)
}
} }
impl Deref for ChannelIndex { impl Deref for ChannelIndex {
@ -76,6 +113,14 @@ impl<'a> ChannelPathsInsertGuard<'a> {
} }
} }
pub fn note_changed(&mut self, channel_id: ChannelId, epoch: u64, version: &clock::Global) {
insert_note_changed(&mut self.channels_by_id, channel_id, epoch, &version);
}
pub fn new_messages(&mut self, channel_id: ChannelId, message_id: u64) {
insert_new_message(&mut self.channels_by_id, channel_id, message_id)
}
pub fn insert(&mut self, channel_proto: proto::Channel) { pub fn insert(&mut self, channel_proto: proto::Channel) {
if let Some(existing_channel) = self.channels_by_id.get_mut(&channel_proto.id) { if let Some(existing_channel) = self.channels_by_id.get_mut(&channel_proto.id) {
Arc::make_mut(existing_channel).name = channel_proto.name; Arc::make_mut(existing_channel).name = channel_proto.name;
@ -85,6 +130,8 @@ impl<'a> ChannelPathsInsertGuard<'a> {
Arc::new(Channel { Arc::new(Channel {
id: channel_proto.id, id: channel_proto.id,
name: channel_proto.name, name: channel_proto.name,
unseen_note_version: None,
unseen_message_id: None,
}), }),
); );
self.insert_root(channel_proto.id); self.insert_root(channel_proto.id);
@ -160,3 +207,32 @@ fn channel_path_sorting_key<'a>(
path.iter() path.iter()
.map(|id| Some(channels_by_id.get(id)?.name.as_str())) .map(|id| Some(channels_by_id.get(id)?.name.as_str()))
} }
fn insert_note_changed(
channels_by_id: &mut BTreeMap<ChannelId, Arc<Channel>>,
channel_id: u64,
epoch: u64,
version: &clock::Global,
) {
if let Some(channel) = channels_by_id.get_mut(&channel_id) {
let unseen_version = Arc::make_mut(channel)
.unseen_note_version
.get_or_insert((0, clock::Global::new()));
if epoch > unseen_version.0 {
*unseen_version = (epoch, version.clone());
} else {
unseen_version.1.join(&version);
}
}
}
fn insert_new_message(
channels_by_id: &mut BTreeMap<ChannelId, Arc<Channel>>,
channel_id: u64,
message_id: u64,
) {
if let Some(channel) = channels_by_id.get_mut(&channel_id) {
let unseen_message_id = Arc::make_mut(channel).unseen_message_id.get_or_insert(0);
*unseen_message_id = message_id.max(*unseen_message_id);
}
}

View file

@ -41,8 +41,8 @@ tempfile = "3"
thiserror.workspace = true thiserror.workspace = true
time.workspace = true time.workspace = true
tiny_http = "0.8" tiny_http = "0.8"
uuid.workspace = true
url = "2.2" url = "2.2"
uuid = { version = "1.1.2", features = ["v4"] }
[dev-dependencies] [dev-dependencies]
collections = { path = "../collections", features = ["test-support"] } collections = { path = "../collections", features = ["test-support"] }

View file

@ -34,7 +34,7 @@ use std::{
future::Future, future::Future,
marker::PhantomData, marker::PhantomData,
path::PathBuf, path::PathBuf,
sync::{Arc, Weak}, sync::{atomic::AtomicU64, Arc, Weak},
time::{Duration, Instant}, time::{Duration, Instant},
}; };
use telemetry::Telemetry; use telemetry::Telemetry;
@ -105,7 +105,7 @@ pub fn init(client: &Arc<Client>, cx: &mut AppContext) {
} }
pub struct Client { pub struct Client {
id: usize, id: AtomicU64,
peer: Arc<Peer>, peer: Arc<Peer>,
http: Arc<dyn HttpClient>, http: Arc<dyn HttpClient>,
telemetry: Arc<Telemetry>, telemetry: Arc<Telemetry>,
@ -374,7 +374,7 @@ impl settings::Setting for TelemetrySettings {
impl Client { impl Client {
pub fn new(http: Arc<dyn HttpClient>, cx: &AppContext) -> Arc<Self> { pub fn new(http: Arc<dyn HttpClient>, cx: &AppContext) -> Arc<Self> {
Arc::new(Self { Arc::new(Self {
id: 0, id: AtomicU64::new(0),
peer: Peer::new(0), peer: Peer::new(0),
telemetry: Telemetry::new(http.clone(), cx), telemetry: Telemetry::new(http.clone(), cx),
http, http,
@ -387,17 +387,16 @@ impl Client {
}) })
} }
pub fn id(&self) -> usize { pub fn id(&self) -> u64 {
self.id self.id.load(std::sync::atomic::Ordering::SeqCst)
} }
pub fn http_client(&self) -> Arc<dyn HttpClient> { pub fn http_client(&self) -> Arc<dyn HttpClient> {
self.http.clone() self.http.clone()
} }
#[cfg(any(test, feature = "test-support"))] pub fn set_id(&self, id: u64) -> &Self {
pub fn set_id(&mut self, id: usize) -> &Self { self.id.store(id, std::sync::atomic::Ordering::SeqCst);
self.id = id;
self self
} }
@ -454,7 +453,7 @@ impl Client {
} }
fn set_status(self: &Arc<Self>, status: Status, cx: &AsyncAppContext) { fn set_status(self: &Arc<Self>, status: Status, cx: &AsyncAppContext) {
log::info!("set status on client {}: {:?}", self.id, status); log::info!("set status on client {}: {:?}", self.id(), status);
let mut state = self.state.write(); let mut state = self.state.write();
*state.status.0.borrow_mut() = status; *state.status.0.borrow_mut() = status;
@ -805,6 +804,7 @@ impl Client {
} }
} }
let credentials = credentials.unwrap(); let credentials = credentials.unwrap();
self.set_id(credentials.user_id);
if was_disconnected { if was_disconnected {
self.set_status(Status::Connecting, cx); self.set_status(Status::Connecting, cx);
@ -1221,7 +1221,7 @@ impl Client {
} }
pub fn send<T: EnvelopedMessage>(&self, message: T) -> Result<()> { pub fn send<T: EnvelopedMessage>(&self, message: T) -> Result<()> {
log::debug!("rpc send. client_id:{}, name:{}", self.id, T::NAME); log::debug!("rpc send. client_id:{}, name:{}", self.id(), T::NAME);
self.peer.send(self.connection_id()?, message) self.peer.send(self.connection_id()?, message)
} }
@ -1237,7 +1237,7 @@ impl Client {
&self, &self,
request: T, request: T,
) -> impl Future<Output = Result<TypedEnvelope<T::Response>>> { ) -> impl Future<Output = Result<TypedEnvelope<T::Response>>> {
let client_id = self.id; let client_id = self.id();
log::debug!( log::debug!(
"rpc request start. client_id:{}. name:{}", "rpc request start. client_id:{}. name:{}",
client_id, client_id,
@ -1258,7 +1258,7 @@ impl Client {
} }
fn respond<T: RequestMessage>(&self, receipt: Receipt<T>, response: T::Response) -> Result<()> { fn respond<T: RequestMessage>(&self, receipt: Receipt<T>, response: T::Response) -> Result<()> {
log::debug!("rpc respond. client_id:{}. name:{}", self.id, T::NAME); log::debug!("rpc respond. client_id:{}. name:{}", self.id(), T::NAME);
self.peer.respond(receipt, response) self.peer.respond(receipt, response)
} }
@ -1267,7 +1267,7 @@ impl Client {
receipt: Receipt<T>, receipt: Receipt<T>,
error: proto::Error, error: proto::Error,
) -> Result<()> { ) -> Result<()> {
log::debug!("rpc respond. client_id:{}. name:{}", self.id, T::NAME); log::debug!("rpc respond. client_id:{}. name:{}", self.id(), T::NAME);
self.peer.respond_with_error(receipt, error) self.peer.respond_with_error(receipt, error)
} }
@ -1336,7 +1336,7 @@ impl Client {
if let Some(handler) = handler { if let Some(handler) = handler {
let future = handler(subscriber, message, &self, cx.clone()); let future = handler(subscriber, message, &self, cx.clone());
let client_id = self.id; let client_id = self.id();
log::debug!( log::debug!(
"rpc message received. client_id:{}, sender_id:{:?}, type:{}", "rpc message received. client_id:{}, sender_id:{:?}, type:{}",
client_id, client_id,

View file

@ -8,6 +8,7 @@ use sysinfo::{Pid, PidExt, ProcessExt, System, SystemExt};
use tempfile::NamedTempFile; use tempfile::NamedTempFile;
use util::http::HttpClient; use util::http::HttpClient;
use util::{channel::ReleaseChannel, TryFutureExt}; use util::{channel::ReleaseChannel, TryFutureExt};
use uuid::Uuid;
pub struct Telemetry { pub struct Telemetry {
http_client: Arc<dyn HttpClient>, http_client: Arc<dyn HttpClient>,
@ -18,7 +19,8 @@ pub struct Telemetry {
#[derive(Default)] #[derive(Default)]
struct TelemetryState { struct TelemetryState {
metrics_id: Option<Arc<str>>, // Per logged-in user metrics_id: Option<Arc<str>>, // Per logged-in user
installation_id: Option<Arc<str>>, // Per app installation installation_id: Option<Arc<str>>, // Per app installation (different for dev, preview, and stable)
session_id: String, // Per app launch
app_version: Option<Arc<str>>, app_version: Option<Arc<str>>,
release_channel: Option<&'static str>, release_channel: Option<&'static str>,
os_name: &'static str, os_name: &'static str,
@ -41,6 +43,7 @@ lazy_static! {
struct ClickhouseEventRequestBody { struct ClickhouseEventRequestBody {
token: &'static str, token: &'static str,
installation_id: Option<Arc<str>>, installation_id: Option<Arc<str>>,
session_id: String,
is_staff: Option<bool>, is_staff: Option<bool>,
app_version: Option<Arc<str>>, app_version: Option<Arc<str>>,
os_name: &'static str, os_name: &'static str,
@ -90,14 +93,12 @@ pub enum ClickhouseEvent {
model: &'static str, model: &'static str,
}, },
Cpu { Cpu {
usage_as_percent: f32, usage_as_percentage: f32,
core_count: u32, core_count: u32,
}, },
Memory { Memory {
memory_in_bytes: u64, memory_in_bytes: u64,
virtual_memory_in_bytes: u64, virtual_memory_in_bytes: u64,
start_time_in_seconds: u64,
run_time_in_seconds: u64,
}, },
} }
@ -133,6 +134,7 @@ impl Telemetry {
release_channel, release_channel,
installation_id: None, installation_id: None,
metrics_id: None, metrics_id: None,
session_id: Uuid::new_v4().to_string(),
clickhouse_events_queue: Default::default(), clickhouse_events_queue: Default::default(),
flush_clickhouse_events_task: Default::default(), flush_clickhouse_events_task: Default::default(),
log_file: None, log_file: None,
@ -168,8 +170,6 @@ impl Telemetry {
const DURATION_BETWEEN_SYSTEM_EVENTS: Duration = Duration::from_secs(60); const DURATION_BETWEEN_SYSTEM_EVENTS: Duration = Duration::from_secs(60);
smol::Timer::after(DURATION_BETWEEN_SYSTEM_EVENTS).await; smol::Timer::after(DURATION_BETWEEN_SYSTEM_EVENTS).await;
let telemetry_settings = cx.update(|cx| *settings::get::<TelemetrySettings>(cx));
system.refresh_memory(); system.refresh_memory();
system.refresh_processes(); system.refresh_processes();
@ -184,15 +184,15 @@ impl Telemetry {
let memory_event = ClickhouseEvent::Memory { let memory_event = ClickhouseEvent::Memory {
memory_in_bytes: process.memory(), memory_in_bytes: process.memory(),
virtual_memory_in_bytes: process.virtual_memory(), virtual_memory_in_bytes: process.virtual_memory(),
start_time_in_seconds: process.start_time(),
run_time_in_seconds: process.run_time(),
}; };
let cpu_event = ClickhouseEvent::Cpu { let cpu_event = ClickhouseEvent::Cpu {
usage_as_percent: process.cpu_usage(), usage_as_percentage: process.cpu_usage(),
core_count: system.cpus().len() as u32, core_count: system.cpus().len() as u32,
}; };
let telemetry_settings = cx.update(|cx| *settings::get::<TelemetrySettings>(cx));
this.report_clickhouse_event(memory_event, telemetry_settings); this.report_clickhouse_event(memory_event, telemetry_settings);
this.report_clickhouse_event(cpu_event, telemetry_settings); this.report_clickhouse_event(cpu_event, telemetry_settings);
} }
@ -289,6 +289,7 @@ impl Telemetry {
&ClickhouseEventRequestBody { &ClickhouseEventRequestBody {
token: ZED_SECRET_CLIENT_TOKEN, token: ZED_SECRET_CLIENT_TOKEN,
installation_id: state.installation_id.clone(), installation_id: state.installation_id.clone(),
session_id: state.session_id.clone(),
is_staff: state.is_staff.clone(), is_staff: state.is_staff.clone(),
app_version: state.app_version.clone(), app_version: state.app_version.clone(),
os_name: state.os_name, os_name: state.os_name,

View file

@ -3,7 +3,7 @@ authors = ["Nathan Sobo <nathan@zed.dev>"]
default-run = "collab" default-run = "collab"
edition = "2021" edition = "2021"
name = "collab" name = "collab"
version = "0.22.1" version = "0.23.2"
publish = false publish = false
[[bin]] [[bin]]
@ -42,14 +42,12 @@ rand.workspace = true
reqwest = { version = "0.11", features = ["json"], optional = true } reqwest = { version = "0.11", features = ["json"], optional = true }
scrypt = "0.7" scrypt = "0.7"
smallvec.workspace = true smallvec.workspace = true
# Remove fork dependency when a version with https://github.com/SeaQL/sea-orm/pull/1283 is released. sea-orm = { version = "0.12.x", features = ["sqlx-postgres", "postgres-array", "runtime-tokio-rustls", "with-uuid"] }
sea-orm = { git = "https://github.com/zed-industries/sea-orm", rev = "18f4c691085712ad014a51792af75a9044bacee6", features = ["sqlx-postgres", "postgres-array", "runtime-tokio-rustls"] }
sea-query = "0.27"
serde.workspace = true serde.workspace = true
serde_derive.workspace = true serde_derive.workspace = true
serde_json.workspace = true serde_json.workspace = true
sha-1 = "0.9" sha-1 = "0.9"
sqlx = { version = "0.6", features = ["runtime-tokio-rustls", "postgres", "json", "time", "uuid", "any"] } sqlx = { version = "0.7", features = ["runtime-tokio-rustls", "postgres", "json", "time", "uuid", "any"] }
time.workspace = true time.workspace = true
tokio = { version = "1", features = ["full"] } tokio = { version = "1", features = ["full"] }
tokio-tungstenite = "0.17" tokio-tungstenite = "0.17"
@ -59,6 +57,7 @@ toml.workspace = true
tracing = "0.1.34" tracing = "0.1.34"
tracing-log = "0.1.3" tracing-log = "0.1.3"
tracing-subscriber = { version = "0.3.11", features = ["env-filter", "json"] } tracing-subscriber = { version = "0.3.11", features = ["env-filter", "json"] }
uuid.workspace = true
[dev-dependencies] [dev-dependencies]
audio = { path = "../audio" } audio = { path = "../audio" }
@ -87,9 +86,9 @@ env_logger.workspace = true
indoc.workspace = true indoc.workspace = true
util = { path = "../util" } util = { path = "../util" }
lazy_static.workspace = true lazy_static.workspace = true
sea-orm = { git = "https://github.com/zed-industries/sea-orm", rev = "18f4c691085712ad014a51792af75a9044bacee6", features = ["sqlx-sqlite"] } sea-orm = { version = "0.12.x", features = ["sqlx-sqlite"] }
serde_json.workspace = true serde_json.workspace = true
sqlx = { version = "0.6", features = ["sqlite"] } sqlx = { version = "0.7", features = ["sqlite"] }
unindent.workspace = true unindent.workspace = true
[features] [features]

View file

@ -289,3 +289,24 @@ CREATE TABLE "user_features" (
CREATE UNIQUE INDEX "index_user_features_user_id_and_feature_id" ON "user_features" ("user_id", "feature_id"); CREATE UNIQUE INDEX "index_user_features_user_id_and_feature_id" ON "user_features" ("user_id", "feature_id");
CREATE INDEX "index_user_features_on_user_id" ON "user_features" ("user_id"); CREATE INDEX "index_user_features_on_user_id" ON "user_features" ("user_id");
CREATE INDEX "index_user_features_on_feature_id" ON "user_features" ("feature_id"); CREATE INDEX "index_user_features_on_feature_id" ON "user_features" ("feature_id");
CREATE TABLE "observed_buffer_edits" (
"user_id" INTEGER NOT NULL REFERENCES users (id) ON DELETE CASCADE,
"buffer_id" INTEGER NOT NULL REFERENCES buffers (id) ON DELETE CASCADE,
"epoch" INTEGER NOT NULL,
"lamport_timestamp" INTEGER NOT NULL,
"replica_id" INTEGER NOT NULL,
PRIMARY KEY (user_id, buffer_id)
);
CREATE UNIQUE INDEX "index_observed_buffers_user_and_buffer_id" ON "observed_buffer_edits" ("user_id", "buffer_id");
CREATE TABLE IF NOT EXISTS "observed_channel_messages" (
"user_id" INTEGER NOT NULL REFERENCES users (id) ON DELETE CASCADE,
"channel_id" INTEGER NOT NULL REFERENCES channels (id) ON DELETE CASCADE,
"channel_message_id" INTEGER NOT NULL,
PRIMARY KEY (user_id, channel_id)
);
CREATE UNIQUE INDEX "index_observed_channel_messages_user_and_channel_id" ON "observed_channel_messages" ("user_id", "channel_id");

View file

@ -0,0 +1,19 @@
CREATE TABLE IF NOT EXISTS "observed_buffer_edits" (
"user_id" INTEGER NOT NULL REFERENCES users (id) ON DELETE CASCADE,
"buffer_id" INTEGER NOT NULL REFERENCES buffers (id) ON DELETE CASCADE,
"epoch" INTEGER NOT NULL,
"lamport_timestamp" INTEGER NOT NULL,
"replica_id" INTEGER NOT NULL,
PRIMARY KEY (user_id, buffer_id)
);
CREATE UNIQUE INDEX "index_observed_buffer_user_and_buffer_id" ON "observed_buffer_edits" ("user_id", "buffer_id");
CREATE TABLE IF NOT EXISTS "observed_channel_messages" (
"user_id" INTEGER NOT NULL REFERENCES users (id) ON DELETE CASCADE,
"channel_id" INTEGER NOT NULL REFERENCES channels (id) ON DELETE CASCADE,
"channel_message_id" INTEGER NOT NULL,
PRIMARY KEY (user_id, channel_id)
);
CREATE UNIQUE INDEX "index_observed_channel_messages_user_and_channel_id" ON "observed_channel_messages" ("user_id", "channel_id");

View file

@ -19,11 +19,12 @@ use rpc::{
ConnectionId, ConnectionId,
}; };
use sea_orm::{ use sea_orm::{
entity::prelude::*, ActiveValue, Condition, ConnectionTrait, DatabaseConnection, entity::prelude::*,
DatabaseTransaction, DbErr, FromQueryResult, IntoActiveModel, IsolationLevel, JoinType, sea_query::{Alias, Expr, OnConflict, Query},
QueryOrder, QuerySelect, Statement, TransactionTrait, ActiveValue, Condition, ConnectionTrait, DatabaseConnection, DatabaseTransaction, DbErr,
FromQueryResult, IntoActiveModel, IsolationLevel, JoinType, QueryOrder, QuerySelect, Statement,
TransactionTrait,
}; };
use sea_query::{Alias, Expr, OnConflict, Query};
use serde::{Deserialize, Serialize}; use serde::{Deserialize, Serialize};
use sqlx::{ use sqlx::{
migrate::{Migrate, Migration, MigrationSource}, migrate::{Migrate, Migration, MigrationSource},
@ -62,6 +63,7 @@ pub struct Database {
// separate files in the `queries` folder. // separate files in the `queries` folder.
impl Database { impl Database {
pub async fn new(options: ConnectOptions, executor: Executor) -> Result<Self> { pub async fn new(options: ConnectOptions, executor: Executor) -> Result<Self> {
sqlx::any::install_default_drivers();
Ok(Self { Ok(Self {
options: options.clone(), options: options.clone(),
pool: sea_orm::Database::connect(options).await?, pool: sea_orm::Database::connect(options).await?,
@ -119,7 +121,7 @@ impl Database {
Ok(new_migrations) Ok(new_migrations)
} }
async fn transaction<F, Fut, T>(&self, f: F) -> Result<T> pub async fn transaction<F, Fut, T>(&self, f: F) -> Result<T>
where where
F: Send + Fn(TransactionHandle) -> Fut, F: Send + Fn(TransactionHandle) -> Fut,
Fut: Send + Future<Output = Result<T>>, Fut: Send + Future<Output = Result<T>>,
@ -321,7 +323,7 @@ fn is_serialization_error(error: &Error) -> bool {
} }
} }
struct TransactionHandle(Arc<Option<DatabaseTransaction>>); pub struct TransactionHandle(Arc<Option<DatabaseTransaction>>);
impl Deref for TransactionHandle { impl Deref for TransactionHandle {
type Target = DatabaseTransaction; type Target = DatabaseTransaction;
@ -437,6 +439,8 @@ pub struct ChannelsForUser {
pub channels: ChannelGraph, pub channels: ChannelGraph,
pub channel_participants: HashMap<ChannelId, Vec<UserId>>, pub channel_participants: HashMap<ChannelId, Vec<UserId>>,
pub channels_with_admin_privileges: HashSet<ChannelId>, pub channels_with_admin_privileges: HashSet<ChannelId>,
pub unseen_buffer_changes: Vec<proto::UnseenChannelBufferChange>,
pub channel_messages: Vec<proto::UnseenChannelMessage>,
} }
#[derive(Debug)] #[derive(Debug)]

View file

@ -1,6 +1,5 @@
use crate::Result; use crate::Result;
use sea_orm::DbErr; use sea_orm::{entity::prelude::*, DbErr};
use sea_query::{Value, ValueTypeErr};
use serde::{Deserialize, Serialize}; use serde::{Deserialize, Serialize};
macro_rules! id_type { macro_rules! id_type {
@ -17,6 +16,7 @@ macro_rules! id_type {
Hash, Hash,
Serialize, Serialize,
Deserialize, Deserialize,
DeriveValueType,
)] )]
#[serde(transparent)] #[serde(transparent)]
pub struct $name(pub i32); pub struct $name(pub i32);
@ -42,40 +42,6 @@ macro_rules! id_type {
} }
} }
impl From<$name> for sea_query::Value {
fn from(value: $name) -> Self {
sea_query::Value::Int(Some(value.0))
}
}
impl sea_orm::TryGetable for $name {
fn try_get(
res: &sea_orm::QueryResult,
pre: &str,
col: &str,
) -> Result<Self, sea_orm::TryGetError> {
Ok(Self(i32::try_get(res, pre, col)?))
}
}
impl sea_query::ValueType for $name {
fn try_from(v: Value) -> Result<Self, sea_query::ValueTypeErr> {
Ok(Self(value_to_integer(v)?))
}
fn type_name() -> String {
stringify!($name).into()
}
fn array_type() -> sea_query::ArrayType {
sea_query::ArrayType::Int
}
fn column_type() -> sea_query::ColumnType {
sea_query::ColumnType::Integer(None)
}
}
impl sea_orm::TryFromU64 for $name { impl sea_orm::TryFromU64 for $name {
fn try_from_u64(n: u64) -> Result<Self, DbErr> { fn try_from_u64(n: u64) -> Result<Self, DbErr> {
Ok(Self(n.try_into().map_err(|_| { Ok(Self(n.try_into().map_err(|_| {
@ -88,7 +54,7 @@ macro_rules! id_type {
} }
} }
impl sea_query::Nullable for $name { impl sea_orm::sea_query::Nullable for $name {
fn null() -> Value { fn null() -> Value {
Value::Int(None) Value::Int(None)
} }
@ -96,20 +62,6 @@ macro_rules! id_type {
}; };
} }
fn value_to_integer(v: Value) -> Result<i32, ValueTypeErr> {
match v {
Value::TinyInt(Some(int)) => int.try_into().map_err(|_| ValueTypeErr),
Value::SmallInt(Some(int)) => int.try_into().map_err(|_| ValueTypeErr),
Value::Int(Some(int)) => int.try_into().map_err(|_| ValueTypeErr),
Value::BigInt(Some(int)) => int.try_into().map_err(|_| ValueTypeErr),
Value::TinyUnsigned(Some(int)) => int.try_into().map_err(|_| ValueTypeErr),
Value::SmallUnsigned(Some(int)) => int.try_into().map_err(|_| ValueTypeErr),
Value::Unsigned(Some(int)) => int.try_into().map_err(|_| ValueTypeErr),
Value::BigUnsigned(Some(int)) => int.try_into().map_err(|_| ValueTypeErr),
_ => Err(ValueTypeErr),
}
}
id_type!(BufferId); id_type!(BufferId);
id_type!(AccessTokenId); id_type!(AccessTokenId);
id_type!(ChannelChatParticipantId); id_type!(ChannelChatParticipantId);

View file

@ -74,7 +74,32 @@ impl Database {
.await?; .await?;
collaborators.push(collaborator); collaborators.push(collaborator);
let (base_text, operations) = self.get_buffer_state(&buffer, &tx).await?; let (base_text, operations, max_operation) =
self.get_buffer_state(&buffer, &tx).await?;
// Save the last observed operation
if let Some(op) = max_operation {
observed_buffer_edits::Entity::insert(observed_buffer_edits::ActiveModel {
user_id: ActiveValue::Set(user_id),
buffer_id: ActiveValue::Set(buffer.id),
epoch: ActiveValue::Set(op.epoch),
lamport_timestamp: ActiveValue::Set(op.lamport_timestamp),
replica_id: ActiveValue::Set(op.replica_id),
})
.on_conflict(
OnConflict::columns([
observed_buffer_edits::Column::UserId,
observed_buffer_edits::Column::BufferId,
])
.update_columns([
observed_buffer_edits::Column::Epoch,
observed_buffer_edits::Column::LamportTimestamp,
])
.to_owned(),
)
.exec(&*tx)
.await?;
}
Ok(proto::JoinChannelBufferResponse { Ok(proto::JoinChannelBufferResponse {
buffer_id: buffer.id.to_proto(), buffer_id: buffer.id.to_proto(),
@ -373,33 +398,46 @@ impl Database {
channel_id: ChannelId, channel_id: ChannelId,
) -> Result<Vec<UserId>> { ) -> Result<Vec<UserId>> {
self.transaction(|tx| async move { self.transaction(|tx| async move {
#[derive(Debug, Clone, Copy, EnumIter, DeriveColumn)] self.get_channel_buffer_collaborators_internal(channel_id, &*tx)
enum QueryUserIds { .await
UserId,
}
let users: Vec<UserId> = channel_buffer_collaborator::Entity::find()
.select_only()
.column(channel_buffer_collaborator::Column::UserId)
.filter(
Condition::all()
.add(channel_buffer_collaborator::Column::ChannelId.eq(channel_id)),
)
.into_values::<_, QueryUserIds>()
.all(&*tx)
.await?;
Ok(users)
}) })
.await .await
} }
async fn get_channel_buffer_collaborators_internal(
&self,
channel_id: ChannelId,
tx: &DatabaseTransaction,
) -> Result<Vec<UserId>> {
#[derive(Debug, Clone, Copy, EnumIter, DeriveColumn)]
enum QueryUserIds {
UserId,
}
let users: Vec<UserId> = channel_buffer_collaborator::Entity::find()
.select_only()
.column(channel_buffer_collaborator::Column::UserId)
.filter(
Condition::all().add(channel_buffer_collaborator::Column::ChannelId.eq(channel_id)),
)
.into_values::<_, QueryUserIds>()
.all(&*tx)
.await?;
Ok(users)
}
pub async fn update_channel_buffer( pub async fn update_channel_buffer(
&self, &self,
channel_id: ChannelId, channel_id: ChannelId,
user: UserId, user: UserId,
operations: &[proto::Operation], operations: &[proto::Operation],
) -> Result<Vec<ConnectionId>> { ) -> Result<(
Vec<ConnectionId>,
Vec<UserId>,
i32,
Vec<proto::VectorClockEntry>,
)> {
self.transaction(move |tx| async move { self.transaction(move |tx| async move {
self.check_user_is_channel_member(channel_id, user, &*tx) self.check_user_is_channel_member(channel_id, user, &*tx)
.await?; .await?;
@ -418,7 +456,38 @@ impl Database {
.iter() .iter()
.filter_map(|op| operation_to_storage(op, &buffer, serialization_version)) .filter_map(|op| operation_to_storage(op, &buffer, serialization_version))
.collect::<Vec<_>>(); .collect::<Vec<_>>();
let mut channel_members;
let max_version;
if !operations.is_empty() { if !operations.is_empty() {
let max_operation = operations
.iter()
.max_by_key(|op| (op.lamport_timestamp.as_ref(), op.replica_id.as_ref()))
.unwrap();
max_version = vec![proto::VectorClockEntry {
replica_id: *max_operation.replica_id.as_ref() as u32,
timestamp: *max_operation.lamport_timestamp.as_ref() as u32,
}];
// get current channel participants and save the max operation above
self.save_max_operation(
user,
buffer.id,
buffer.epoch,
*max_operation.replica_id.as_ref(),
*max_operation.lamport_timestamp.as_ref(),
&*tx,
)
.await?;
channel_members = self.get_channel_members_internal(channel_id, &*tx).await?;
let collaborators = self
.get_channel_buffer_collaborators_internal(channel_id, &*tx)
.await?;
channel_members.retain(|member| !collaborators.contains(member));
buffer_operation::Entity::insert_many(operations) buffer_operation::Entity::insert_many(operations)
.on_conflict( .on_conflict(
OnConflict::columns([ OnConflict::columns([
@ -432,6 +501,9 @@ impl Database {
) )
.exec(&*tx) .exec(&*tx)
.await?; .await?;
} else {
channel_members = Vec::new();
max_version = Vec::new();
} }
let mut connections = Vec::new(); let mut connections = Vec::new();
@ -450,11 +522,53 @@ impl Database {
}); });
} }
Ok(connections) Ok((connections, channel_members, buffer.epoch, max_version))
}) })
.await .await
} }
async fn save_max_operation(
&self,
user_id: UserId,
buffer_id: BufferId,
epoch: i32,
replica_id: i32,
lamport_timestamp: i32,
tx: &DatabaseTransaction,
) -> Result<()> {
use observed_buffer_edits::Column;
observed_buffer_edits::Entity::insert(observed_buffer_edits::ActiveModel {
user_id: ActiveValue::Set(user_id),
buffer_id: ActiveValue::Set(buffer_id),
epoch: ActiveValue::Set(epoch),
replica_id: ActiveValue::Set(replica_id),
lamport_timestamp: ActiveValue::Set(lamport_timestamp),
})
.on_conflict(
OnConflict::columns([Column::UserId, Column::BufferId])
.update_columns([Column::Epoch, Column::LamportTimestamp, Column::ReplicaId])
.action_cond_where(
Condition::any().add(Column::Epoch.lt(epoch)).add(
Condition::all().add(Column::Epoch.eq(epoch)).add(
Condition::any()
.add(Column::LamportTimestamp.lt(lamport_timestamp))
.add(
Column::LamportTimestamp
.eq(lamport_timestamp)
.and(Column::ReplicaId.lt(replica_id)),
),
),
),
)
.to_owned(),
)
.exec_without_returning(tx)
.await?;
Ok(())
}
async fn get_buffer_operation_serialization_version( async fn get_buffer_operation_serialization_version(
&self, &self,
buffer_id: BufferId, buffer_id: BufferId,
@ -472,7 +586,7 @@ impl Database {
.ok_or_else(|| anyhow!("missing buffer snapshot"))?) .ok_or_else(|| anyhow!("missing buffer snapshot"))?)
} }
async fn get_channel_buffer( pub async fn get_channel_buffer(
&self, &self,
channel_id: ChannelId, channel_id: ChannelId,
tx: &DatabaseTransaction, tx: &DatabaseTransaction,
@ -491,7 +605,11 @@ impl Database {
&self, &self,
buffer: &buffer::Model, buffer: &buffer::Model,
tx: &DatabaseTransaction, tx: &DatabaseTransaction,
) -> Result<(String, Vec<proto::Operation>)> { ) -> Result<(
String,
Vec<proto::Operation>,
Option<buffer_operation::Model>,
)> {
let id = buffer.id; let id = buffer.id;
let (base_text, version) = if buffer.epoch > 0 { let (base_text, version) = if buffer.epoch > 0 {
let snapshot = buffer_snapshot::Entity::find() let snapshot = buffer_snapshot::Entity::find()
@ -516,16 +634,28 @@ impl Database {
.eq(id) .eq(id)
.and(buffer_operation::Column::Epoch.eq(buffer.epoch)), .and(buffer_operation::Column::Epoch.eq(buffer.epoch)),
) )
.order_by_asc(buffer_operation::Column::LamportTimestamp)
.order_by_asc(buffer_operation::Column::ReplicaId)
.stream(&*tx) .stream(&*tx)
.await?; .await?;
let mut operations = Vec::new(); let mut operations = Vec::new();
let mut last_row = None;
while let Some(row) = rows.next().await { while let Some(row) = rows.next().await {
let row = row?;
last_row = Some(buffer_operation::Model {
buffer_id: row.buffer_id,
epoch: row.epoch,
lamport_timestamp: row.lamport_timestamp,
replica_id: row.lamport_timestamp,
value: Default::default(),
});
operations.push(proto::Operation { operations.push(proto::Operation {
variant: Some(operation_from_storage(row?, version)?), variant: Some(operation_from_storage(row, version)?),
}) });
} }
Ok((base_text, operations)) Ok((base_text, operations, last_row))
} }
async fn snapshot_channel_buffer( async fn snapshot_channel_buffer(
@ -534,7 +664,7 @@ impl Database {
tx: &DatabaseTransaction, tx: &DatabaseTransaction,
) -> Result<()> { ) -> Result<()> {
let buffer = self.get_channel_buffer(channel_id, tx).await?; let buffer = self.get_channel_buffer(channel_id, tx).await?;
let (base_text, operations) = self.get_buffer_state(&buffer, tx).await?; let (base_text, operations, _) = self.get_buffer_state(&buffer, tx).await?;
if operations.is_empty() { if operations.is_empty() {
return Ok(()); return Ok(());
} }
@ -567,6 +697,150 @@ impl Database {
Ok(()) Ok(())
} }
pub async fn observe_buffer_version(
&self,
buffer_id: BufferId,
user_id: UserId,
epoch: i32,
version: &[proto::VectorClockEntry],
) -> Result<()> {
self.transaction(|tx| async move {
// For now, combine concurrent operations.
let Some(component) = version.iter().max_by_key(|version| version.timestamp) else {
return Ok(());
};
self.save_max_operation(
user_id,
buffer_id,
epoch,
component.replica_id as i32,
component.timestamp as i32,
&*tx,
)
.await?;
Ok(())
})
.await
}
pub async fn unseen_channel_buffer_changes(
&self,
user_id: UserId,
channel_ids: &[ChannelId],
tx: &DatabaseTransaction,
) -> Result<Vec<proto::UnseenChannelBufferChange>> {
#[derive(Debug, Clone, Copy, EnumIter, DeriveColumn)]
enum QueryIds {
ChannelId,
Id,
}
let mut channel_ids_by_buffer_id = HashMap::default();
let mut rows = buffer::Entity::find()
.filter(buffer::Column::ChannelId.is_in(channel_ids.iter().copied()))
.stream(&*tx)
.await?;
while let Some(row) = rows.next().await {
let row = row?;
channel_ids_by_buffer_id.insert(row.id, row.channel_id);
}
drop(rows);
let mut observed_edits_by_buffer_id = HashMap::default();
let mut rows = observed_buffer_edits::Entity::find()
.filter(observed_buffer_edits::Column::UserId.eq(user_id))
.filter(
observed_buffer_edits::Column::BufferId
.is_in(channel_ids_by_buffer_id.keys().copied()),
)
.stream(&*tx)
.await?;
while let Some(row) = rows.next().await {
let row = row?;
observed_edits_by_buffer_id.insert(row.buffer_id, row);
}
drop(rows);
let latest_operations = self
.get_latest_operations_for_buffers(channel_ids_by_buffer_id.keys().copied(), &*tx)
.await?;
let mut changes = Vec::default();
for latest in latest_operations {
if let Some(observed) = observed_edits_by_buffer_id.get(&latest.buffer_id) {
if (
observed.epoch,
observed.lamport_timestamp,
observed.replica_id,
) >= (latest.epoch, latest.lamport_timestamp, latest.replica_id)
{
continue;
}
}
if let Some(channel_id) = channel_ids_by_buffer_id.get(&latest.buffer_id) {
changes.push(proto::UnseenChannelBufferChange {
channel_id: channel_id.to_proto(),
epoch: latest.epoch as u64,
version: vec![proto::VectorClockEntry {
replica_id: latest.replica_id as u32,
timestamp: latest.lamport_timestamp as u32,
}],
});
}
}
Ok(changes)
}
pub async fn get_latest_operations_for_buffers(
&self,
buffer_ids: impl IntoIterator<Item = BufferId>,
tx: &DatabaseTransaction,
) -> Result<Vec<buffer_operation::Model>> {
let mut values = String::new();
for id in buffer_ids {
if !values.is_empty() {
values.push_str(", ");
}
write!(&mut values, "({})", id).unwrap();
}
if values.is_empty() {
return Ok(Vec::default());
}
let sql = format!(
r#"
SELECT
*
FROM
(
SELECT
*,
row_number() OVER (
PARTITION BY buffer_id
ORDER BY
epoch DESC,
lamport_timestamp DESC,
replica_id DESC
) as row_number
FROM buffer_operations
WHERE
buffer_id in ({values})
) AS last_operations
WHERE
row_number = 1
"#,
);
let stmt = Statement::from_string(self.pool.get_database_backend(), sql);
Ok(buffer_operation::Entity::find()
.from_raw_sql(stmt)
.all(&*tx)
.await?)
}
} }
fn operation_to_storage( fn operation_to_storage(

View file

@ -1,8 +1,7 @@
use super::*;
use rpc::proto::ChannelEdge; use rpc::proto::ChannelEdge;
use smallvec::SmallVec; use smallvec::SmallVec;
use super::*;
type ChannelDescendants = HashMap<ChannelId, SmallSet<ChannelId>>; type ChannelDescendants = HashMap<ChannelId, SmallSet<ChannelId>>;
impl Database { impl Database {
@ -391,7 +390,8 @@ impl Database {
.all(&*tx) .all(&*tx)
.await?; .await?;
self.get_user_channels(channel_memberships, &tx).await self.get_user_channels(user_id, channel_memberships, &tx)
.await
}) })
.await .await
} }
@ -414,13 +414,15 @@ impl Database {
.all(&*tx) .all(&*tx)
.await?; .await?;
self.get_user_channels(channel_membership, &tx).await self.get_user_channels(user_id, channel_membership, &tx)
.await
}) })
.await .await
} }
pub async fn get_user_channels( pub async fn get_user_channels(
&self, &self,
user_id: UserId,
channel_memberships: Vec<channel_member::Model>, channel_memberships: Vec<channel_member::Model>,
tx: &DatabaseTransaction, tx: &DatabaseTransaction,
) -> Result<ChannelsForUser> { ) -> Result<ChannelsForUser> {
@ -460,10 +462,21 @@ impl Database {
} }
} }
let channel_ids = graph.channels.iter().map(|c| c.id).collect::<Vec<_>>();
let channel_buffer_changes = self
.unseen_channel_buffer_changes(user_id, &channel_ids, &*tx)
.await?;
let unseen_messages = self
.unseen_channel_messages(user_id, &channel_ids, &*tx)
.await?;
Ok(ChannelsForUser { Ok(ChannelsForUser {
channels: graph, channels: graph,
channel_participants, channel_participants,
channels_with_admin_privileges, channels_with_admin_privileges,
unseen_buffer_changes: channel_buffer_changes,
channel_messages: unseen_messages,
}) })
} }
@ -645,7 +658,7 @@ impl Database {
) -> Result<Vec<ChannelId>> { ) -> Result<Vec<ChannelId>> {
let paths = channel_path::Entity::find() let paths = channel_path::Entity::find()
.filter(channel_path::Column::ChannelId.eq(channel_id)) .filter(channel_path::Column::ChannelId.eq(channel_id))
.order_by(channel_path::Column::IdPath, sea_query::Order::Desc) .order_by(channel_path::Column::IdPath, sea_orm::Order::Desc)
.all(tx) .all(tx)
.await?; .await?;
let mut channel_ids = Vec::new(); let mut channel_ids = Vec::new();

View file

@ -18,12 +18,12 @@ impl Database {
let user_b_participant = Alias::new("user_b_participant"); let user_b_participant = Alias::new("user_b_participant");
let mut db_contacts = contact::Entity::find() let mut db_contacts = contact::Entity::find()
.column_as( .column_as(
Expr::tbl(user_a_participant.clone(), room_participant::Column::Id) Expr::col((user_a_participant.clone(), room_participant::Column::Id))
.is_not_null(), .is_not_null(),
"user_a_busy", "user_a_busy",
) )
.column_as( .column_as(
Expr::tbl(user_b_participant.clone(), room_participant::Column::Id) Expr::col((user_b_participant.clone(), room_participant::Column::Id))
.is_not_null(), .is_not_null(),
"user_b_busy", "user_b_busy",
) )

View file

@ -89,6 +89,7 @@ impl Database {
let mut rows = channel_message::Entity::find() let mut rows = channel_message::Entity::find()
.filter(condition) .filter(condition)
.order_by_asc(channel_message::Column::Id)
.limit(count as u64) .limit(count as u64)
.stream(&*tx) .stream(&*tx)
.await?; .await?;
@ -108,7 +109,7 @@ impl Database {
}), }),
}); });
} }
drop(rows);
Ok(messages) Ok(messages)
}) })
.await .await
@ -121,7 +122,7 @@ impl Database {
body: &str, body: &str,
timestamp: OffsetDateTime, timestamp: OffsetDateTime,
nonce: u128, nonce: u128,
) -> Result<(MessageId, Vec<ConnectionId>)> { ) -> Result<(MessageId, Vec<ConnectionId>, Vec<UserId>)> {
self.transaction(|tx| async move { self.transaction(|tx| async move {
let mut rows = channel_chat_participant::Entity::find() let mut rows = channel_chat_participant::Entity::find()
.filter(channel_chat_participant::Column::ChannelId.eq(channel_id)) .filter(channel_chat_participant::Column::ChannelId.eq(channel_id))
@ -130,11 +131,13 @@ impl Database {
let mut is_participant = false; let mut is_participant = false;
let mut participant_connection_ids = Vec::new(); let mut participant_connection_ids = Vec::new();
let mut participant_user_ids = Vec::new();
while let Some(row) = rows.next().await { while let Some(row) = rows.next().await {
let row = row?; let row = row?;
if row.user_id == user_id { if row.user_id == user_id {
is_participant = true; is_participant = true;
} }
participant_user_ids.push(row.user_id);
participant_connection_ids.push(row.connection()); participant_connection_ids.push(row.connection());
} }
drop(rows); drop(rows);
@ -167,11 +170,141 @@ impl Database {
ConnectionId, ConnectionId,
} }
Ok((message.last_insert_id, participant_connection_ids)) // Observe this message for the sender
self.observe_channel_message_internal(
channel_id,
user_id,
message.last_insert_id,
&*tx,
)
.await?;
let mut channel_members = self.get_channel_members_internal(channel_id, &*tx).await?;
channel_members.retain(|member| !participant_user_ids.contains(member));
Ok((
message.last_insert_id,
participant_connection_ids,
channel_members,
))
}) })
.await .await
} }
pub async fn observe_channel_message(
&self,
channel_id: ChannelId,
user_id: UserId,
message_id: MessageId,
) -> Result<()> {
self.transaction(|tx| async move {
self.observe_channel_message_internal(channel_id, user_id, message_id, &*tx)
.await?;
Ok(())
})
.await
}
async fn observe_channel_message_internal(
&self,
channel_id: ChannelId,
user_id: UserId,
message_id: MessageId,
tx: &DatabaseTransaction,
) -> Result<()> {
observed_channel_messages::Entity::insert(observed_channel_messages::ActiveModel {
user_id: ActiveValue::Set(user_id),
channel_id: ActiveValue::Set(channel_id),
channel_message_id: ActiveValue::Set(message_id),
})
.on_conflict(
OnConflict::columns([
observed_channel_messages::Column::ChannelId,
observed_channel_messages::Column::UserId,
])
.update_column(observed_channel_messages::Column::ChannelMessageId)
.action_cond_where(observed_channel_messages::Column::ChannelMessageId.lt(message_id))
.to_owned(),
)
// TODO: Try to upgrade SeaORM so we don't have to do this hack around their bug
.exec_without_returning(&*tx)
.await?;
Ok(())
}
pub async fn unseen_channel_messages(
&self,
user_id: UserId,
channel_ids: &[ChannelId],
tx: &DatabaseTransaction,
) -> Result<Vec<proto::UnseenChannelMessage>> {
let mut observed_messages_by_channel_id = HashMap::default();
let mut rows = observed_channel_messages::Entity::find()
.filter(observed_channel_messages::Column::UserId.eq(user_id))
.filter(observed_channel_messages::Column::ChannelId.is_in(channel_ids.iter().copied()))
.stream(&*tx)
.await?;
while let Some(row) = rows.next().await {
let row = row?;
observed_messages_by_channel_id.insert(row.channel_id, row);
}
drop(rows);
let mut values = String::new();
for id in channel_ids {
if !values.is_empty() {
values.push_str(", ");
}
write!(&mut values, "({})", id).unwrap();
}
if values.is_empty() {
return Ok(Default::default());
}
let sql = format!(
r#"
SELECT
*
FROM (
SELECT
*,
row_number() OVER (
PARTITION BY channel_id
ORDER BY id DESC
) as row_number
FROM channel_messages
WHERE
channel_id in ({values})
) AS messages
WHERE
row_number = 1
"#,
);
let stmt = Statement::from_string(self.pool.get_database_backend(), sql);
let last_messages = channel_message::Model::find_by_statement(stmt)
.all(&*tx)
.await?;
let mut changes = Vec::new();
for last_message in last_messages {
if let Some(observed_message) =
observed_messages_by_channel_id.get(&last_message.channel_id)
{
if observed_message.channel_message_id == last_message.id {
continue;
}
}
changes.push(proto::UnseenChannelMessage {
channel_id: last_message.channel_id.to_proto(),
message_id: last_message.id.to_proto(),
});
}
Ok(changes)
}
pub async fn remove_channel_message( pub async fn remove_channel_message(
&self, &self,
channel_id: ChannelId, channel_id: ChannelId,

View file

@ -184,7 +184,7 @@ impl Database {
Ok(user::Entity::find() Ok(user::Entity::find()
.from_raw_sql(Statement::from_sql_and_values( .from_raw_sql(Statement::from_sql_and_values(
self.pool.get_database_backend(), self.pool.get_database_backend(),
query.into(), query,
vec![like_string.into(), name_query.into(), limit.into()], vec![like_string.into(), name_query.into(), limit.into()],
)) ))
.all(&*tx) .all(&*tx)

View file

@ -12,6 +12,8 @@ pub mod contact;
pub mod feature_flag; pub mod feature_flag;
pub mod follower; pub mod follower;
pub mod language_server; pub mod language_server;
pub mod observed_buffer_edits;
pub mod observed_channel_messages;
pub mod project; pub mod project;
pub mod project_collaborator; pub mod project_collaborator;
pub mod room; pub mod room;

View file

@ -0,0 +1,43 @@
use crate::db::{BufferId, UserId};
use sea_orm::entity::prelude::*;
#[derive(Clone, Debug, PartialEq, Eq, DeriveEntityModel)]
#[sea_orm(table_name = "observed_buffer_edits")]
pub struct Model {
#[sea_orm(primary_key)]
pub user_id: UserId,
pub buffer_id: BufferId,
pub epoch: i32,
pub lamport_timestamp: i32,
pub replica_id: i32,
}
#[derive(Copy, Clone, Debug, EnumIter, DeriveRelation)]
pub enum Relation {
#[sea_orm(
belongs_to = "super::buffer::Entity",
from = "Column::BufferId",
to = "super::buffer::Column::Id"
)]
Buffer,
#[sea_orm(
belongs_to = "super::user::Entity",
from = "Column::UserId",
to = "super::user::Column::Id"
)]
User,
}
impl Related<super::buffer::Entity> for Entity {
fn to() -> RelationDef {
Relation::Buffer.def()
}
}
impl Related<super::user::Entity> for Entity {
fn to() -> RelationDef {
Relation::User.def()
}
}
impl ActiveModelBehavior for ActiveModel {}

View file

@ -0,0 +1,41 @@
use crate::db::{ChannelId, MessageId, UserId};
use sea_orm::entity::prelude::*;
#[derive(Clone, Debug, PartialEq, Eq, DeriveEntityModel)]
#[sea_orm(table_name = "observed_channel_messages")]
pub struct Model {
#[sea_orm(primary_key)]
pub user_id: UserId,
pub channel_id: ChannelId,
pub channel_message_id: MessageId,
}
#[derive(Copy, Clone, Debug, EnumIter, DeriveRelation)]
pub enum Relation {
#[sea_orm(
belongs_to = "super::channel::Entity",
from = "Column::ChannelId",
to = "super::channel::Column::Id"
)]
Channel,
#[sea_orm(
belongs_to = "super::user::Entity",
from = "Column::UserId",
to = "super::user::Column::Id"
)]
User,
}
impl Related<super::channel::Entity> for Entity {
fn to() -> RelationDef {
Relation::Channel.def()
}
}
impl Related<super::user::Entity> for Entity {
fn to() -> RelationDef {
Relation::User.def()
}
}
impl ActiveModelBehavior for ActiveModel {}

View file

@ -39,7 +39,7 @@ impl TestDb {
db.pool db.pool
.execute(sea_orm::Statement::from_string( .execute(sea_orm::Statement::from_string(
db.pool.get_database_backend(), db.pool.get_database_backend(),
sql.into(), sql,
)) ))
.await .await
.unwrap(); .unwrap();
@ -134,7 +134,7 @@ impl Drop for TestDb {
db.pool db.pool
.execute(sea_orm::Statement::from_string( .execute(sea_orm::Statement::from_string(
db.pool.get_database_backend(), db.pool.get_database_backend(),
query.into(), query,
)) ))
.await .await
.log_err(); .log_err();

View file

@ -1,6 +1,6 @@
use super::*; use super::*;
use crate::test_both_dbs; use crate::test_both_dbs;
use language::proto; use language::proto::{self, serialize_version};
use text::Buffer; use text::Buffer;
test_both_dbs!( test_both_dbs!(
@ -163,3 +163,349 @@ async fn test_channel_buffers(db: &Arc<Database>) {
assert_eq!(buffer_response_b.base_text, "hello, cruel world"); assert_eq!(buffer_response_b.base_text, "hello, cruel world");
assert_eq!(buffer_response_b.operations, &[]); assert_eq!(buffer_response_b.operations, &[]);
} }
test_both_dbs!(
test_channel_buffers_last_operations,
test_channel_buffers_last_operations_postgres,
test_channel_buffers_last_operations_sqlite
);
async fn test_channel_buffers_last_operations(db: &Database) {
let user_id = db
.create_user(
"user_a@example.com",
false,
NewUserParams {
github_login: "user_a".into(),
github_user_id: 101,
invite_count: 0,
},
)
.await
.unwrap()
.user_id;
let observer_id = db
.create_user(
"user_b@example.com",
false,
NewUserParams {
github_login: "user_b".into(),
github_user_id: 102,
invite_count: 0,
},
)
.await
.unwrap()
.user_id;
let owner_id = db.create_server("production").await.unwrap().0 as u32;
let connection_id = ConnectionId {
owner_id,
id: user_id.0 as u32,
};
let mut buffers = Vec::new();
let mut text_buffers = Vec::new();
for i in 0..3 {
let channel = db
.create_root_channel(&format!("channel-{i}"), &format!("room-{i}"), user_id)
.await
.unwrap();
db.invite_channel_member(channel, observer_id, user_id, false)
.await
.unwrap();
db.respond_to_channel_invite(channel, observer_id, true)
.await
.unwrap();
db.join_channel_buffer(channel, user_id, connection_id)
.await
.unwrap();
buffers.push(
db.transaction(|tx| async move { db.get_channel_buffer(channel, &*tx).await })
.await
.unwrap(),
);
text_buffers.push(Buffer::new(0, 0, "".to_string()));
}
let operations = db
.transaction(|tx| {
let buffers = &buffers;
async move {
db.get_latest_operations_for_buffers([buffers[0].id, buffers[2].id], &*tx)
.await
}
})
.await
.unwrap();
assert!(operations.is_empty());
update_buffer(
buffers[0].channel_id,
user_id,
db,
vec![
text_buffers[0].edit([(0..0, "a")]),
text_buffers[0].edit([(0..0, "b")]),
text_buffers[0].edit([(0..0, "c")]),
],
)
.await;
update_buffer(
buffers[1].channel_id,
user_id,
db,
vec![
text_buffers[1].edit([(0..0, "d")]),
text_buffers[1].edit([(1..1, "e")]),
text_buffers[1].edit([(2..2, "f")]),
],
)
.await;
// cause buffer 1's epoch to increment.
db.leave_channel_buffer(buffers[1].channel_id, connection_id)
.await
.unwrap();
db.join_channel_buffer(buffers[1].channel_id, user_id, connection_id)
.await
.unwrap();
text_buffers[1] = Buffer::new(1, 0, "def".to_string());
update_buffer(
buffers[1].channel_id,
user_id,
db,
vec![
text_buffers[1].edit([(0..0, "g")]),
text_buffers[1].edit([(0..0, "h")]),
],
)
.await;
update_buffer(
buffers[2].channel_id,
user_id,
db,
vec![text_buffers[2].edit([(0..0, "i")])],
)
.await;
let operations = db
.transaction(|tx| {
let buffers = &buffers;
async move {
db.get_latest_operations_for_buffers([buffers[1].id, buffers[2].id], &*tx)
.await
}
})
.await
.unwrap();
assert_operations(
&operations,
&[
(buffers[1].id, 1, &text_buffers[1]),
(buffers[2].id, 0, &text_buffers[2]),
],
);
let operations = db
.transaction(|tx| {
let buffers = &buffers;
async move {
db.get_latest_operations_for_buffers([buffers[0].id, buffers[1].id], &*tx)
.await
}
})
.await
.unwrap();
assert_operations(
&operations,
&[
(buffers[0].id, 0, &text_buffers[0]),
(buffers[1].id, 1, &text_buffers[1]),
],
);
let buffer_changes = db
.transaction(|tx| {
let buffers = &buffers;
async move {
db.unseen_channel_buffer_changes(
observer_id,
&[
buffers[0].channel_id,
buffers[1].channel_id,
buffers[2].channel_id,
],
&*tx,
)
.await
}
})
.await
.unwrap();
pretty_assertions::assert_eq!(
buffer_changes,
[
rpc::proto::UnseenChannelBufferChange {
channel_id: buffers[0].channel_id.to_proto(),
epoch: 0,
version: serialize_version(&text_buffers[0].version()),
},
rpc::proto::UnseenChannelBufferChange {
channel_id: buffers[1].channel_id.to_proto(),
epoch: 1,
version: serialize_version(&text_buffers[1].version())
.into_iter()
.filter(|vector| vector.replica_id
== buffer_changes[1].version.first().unwrap().replica_id)
.collect::<Vec<_>>(),
},
rpc::proto::UnseenChannelBufferChange {
channel_id: buffers[2].channel_id.to_proto(),
epoch: 0,
version: serialize_version(&text_buffers[2].version()),
},
]
);
db.observe_buffer_version(
buffers[1].id,
observer_id,
1,
serialize_version(&text_buffers[1].version()).as_slice(),
)
.await
.unwrap();
let buffer_changes = db
.transaction(|tx| {
let buffers = &buffers;
async move {
db.unseen_channel_buffer_changes(
observer_id,
&[
buffers[0].channel_id,
buffers[1].channel_id,
buffers[2].channel_id,
],
&*tx,
)
.await
}
})
.await
.unwrap();
assert_eq!(
buffer_changes,
[
rpc::proto::UnseenChannelBufferChange {
channel_id: buffers[0].channel_id.to_proto(),
epoch: 0,
version: serialize_version(&text_buffers[0].version()),
},
rpc::proto::UnseenChannelBufferChange {
channel_id: buffers[2].channel_id.to_proto(),
epoch: 0,
version: serialize_version(&text_buffers[2].version()),
},
]
);
// Observe an earlier version of the buffer.
db.observe_buffer_version(
buffers[1].id,
observer_id,
1,
&[rpc::proto::VectorClockEntry {
replica_id: 0,
timestamp: 0,
}],
)
.await
.unwrap();
let buffer_changes = db
.transaction(|tx| {
let buffers = &buffers;
async move {
db.unseen_channel_buffer_changes(
observer_id,
&[
buffers[0].channel_id,
buffers[1].channel_id,
buffers[2].channel_id,
],
&*tx,
)
.await
}
})
.await
.unwrap();
assert_eq!(
buffer_changes,
[
rpc::proto::UnseenChannelBufferChange {
channel_id: buffers[0].channel_id.to_proto(),
epoch: 0,
version: serialize_version(&text_buffers[0].version()),
},
rpc::proto::UnseenChannelBufferChange {
channel_id: buffers[2].channel_id.to_proto(),
epoch: 0,
version: serialize_version(&text_buffers[2].version()),
},
]
);
}
async fn update_buffer(
channel_id: ChannelId,
user_id: UserId,
db: &Database,
operations: Vec<text::Operation>,
) {
let operations = operations
.into_iter()
.map(|op| proto::serialize_operation(&language::Operation::Buffer(op)))
.collect::<Vec<_>>();
db.update_channel_buffer(channel_id, user_id, &operations)
.await
.unwrap();
}
fn assert_operations(
operations: &[buffer_operation::Model],
expected: &[(BufferId, i32, &text::Buffer)],
) {
let actual = operations
.iter()
.map(|op| buffer_operation::Model {
buffer_id: op.buffer_id,
epoch: op.epoch,
lamport_timestamp: op.lamport_timestamp,
replica_id: op.replica_id,
value: vec![],
})
.collect::<Vec<_>>();
let expected = expected
.iter()
.map(|(buffer_id, epoch, buffer)| buffer_operation::Model {
buffer_id: *buffer_id,
epoch: *epoch,
lamport_timestamp: buffer.lamport_clock.value as i32 - 1,
replica_id: buffer.replica_id() as i32,
value: vec![],
})
.collect::<Vec<_>>();
assert_eq!(actual, expected, "unexpected operations")
}

View file

@ -57,3 +57,188 @@ async fn test_channel_message_nonces(db: &Arc<Database>) {
assert_eq!(msg1_id, msg3_id); assert_eq!(msg1_id, msg3_id);
assert_eq!(msg2_id, msg4_id); assert_eq!(msg2_id, msg4_id);
} }
test_both_dbs!(
test_channel_message_new_notification,
test_channel_message_new_notification_postgres,
test_channel_message_new_notification_sqlite
);
async fn test_channel_message_new_notification(db: &Arc<Database>) {
let user = db
.create_user(
"user_a@example.com",
false,
NewUserParams {
github_login: "user_a".into(),
github_user_id: 1,
invite_count: 0,
},
)
.await
.unwrap()
.user_id;
let observer = db
.create_user(
"user_b@example.com",
false,
NewUserParams {
github_login: "user_b".into(),
github_user_id: 1,
invite_count: 0,
},
)
.await
.unwrap()
.user_id;
let channel_1 = db
.create_channel("channel", None, "room", user)
.await
.unwrap();
let channel_2 = db
.create_channel("channel-2", None, "room", user)
.await
.unwrap();
db.invite_channel_member(channel_1, observer, user, false)
.await
.unwrap();
db.respond_to_channel_invite(channel_1, observer, true)
.await
.unwrap();
db.invite_channel_member(channel_2, observer, user, false)
.await
.unwrap();
db.respond_to_channel_invite(channel_2, observer, true)
.await
.unwrap();
let owner_id = db.create_server("test").await.unwrap().0 as u32;
let user_connection_id = rpc::ConnectionId { owner_id, id: 0 };
db.join_channel_chat(channel_1, user_connection_id, user)
.await
.unwrap();
let _ = db
.create_channel_message(channel_1, user, "1_1", OffsetDateTime::now_utc(), 1)
.await
.unwrap();
let (second_message, _, _) = db
.create_channel_message(channel_1, user, "1_2", OffsetDateTime::now_utc(), 2)
.await
.unwrap();
let (third_message, _, _) = db
.create_channel_message(channel_1, user, "1_3", OffsetDateTime::now_utc(), 3)
.await
.unwrap();
db.join_channel_chat(channel_2, user_connection_id, user)
.await
.unwrap();
let (fourth_message, _, _) = db
.create_channel_message(channel_2, user, "2_1", OffsetDateTime::now_utc(), 4)
.await
.unwrap();
// Check that observer has new messages
let unseen_messages = db
.transaction(|tx| async move {
db.unseen_channel_messages(observer, &[channel_1, channel_2], &*tx)
.await
})
.await
.unwrap();
assert_eq!(
unseen_messages,
[
rpc::proto::UnseenChannelMessage {
channel_id: channel_1.to_proto(),
message_id: third_message.to_proto(),
},
rpc::proto::UnseenChannelMessage {
channel_id: channel_2.to_proto(),
message_id: fourth_message.to_proto(),
},
]
);
// Observe the second message
db.observe_channel_message(channel_1, observer, second_message)
.await
.unwrap();
// Make sure the observer still has a new message
let unseen_messages = db
.transaction(|tx| async move {
db.unseen_channel_messages(observer, &[channel_1, channel_2], &*tx)
.await
})
.await
.unwrap();
assert_eq!(
unseen_messages,
[
rpc::proto::UnseenChannelMessage {
channel_id: channel_1.to_proto(),
message_id: third_message.to_proto(),
},
rpc::proto::UnseenChannelMessage {
channel_id: channel_2.to_proto(),
message_id: fourth_message.to_proto(),
},
]
);
// Observe the third message,
db.observe_channel_message(channel_1, observer, third_message)
.await
.unwrap();
// Make sure the observer does not have a new method
let unseen_messages = db
.transaction(|tx| async move {
db.unseen_channel_messages(observer, &[channel_1, channel_2], &*tx)
.await
})
.await
.unwrap();
assert_eq!(
unseen_messages,
[rpc::proto::UnseenChannelMessage {
channel_id: channel_2.to_proto(),
message_id: fourth_message.to_proto(),
}]
);
// Observe the second message again, should not regress our observed state
db.observe_channel_message(channel_1, observer, second_message)
.await
.unwrap();
// Make sure the observer does not have a new message
let unseen_messages = db
.transaction(|tx| async move {
db.unseen_channel_messages(observer, &[channel_1, channel_2], &*tx)
.await
})
.await
.unwrap();
assert_eq!(
unseen_messages,
[rpc::proto::UnseenChannelMessage {
channel_id: channel_2.to_proto(),
message_id: fourth_message.to_proto(),
}]
);
}

View file

@ -3,8 +3,8 @@ mod connection_pool;
use crate::{ use crate::{
auth, auth,
db::{ db::{
self, ChannelId, ChannelsForUser, Database, MessageId, ProjectId, RoomId, ServerId, User, self, BufferId, ChannelId, ChannelsForUser, Database, MessageId, ProjectId, RoomId,
UserId, ServerId, User, UserId,
}, },
executor::Executor, executor::Executor,
AppState, Result, AppState, Result,
@ -274,7 +274,9 @@ impl Server {
.add_message_handler(unfollow) .add_message_handler(unfollow)
.add_message_handler(update_followers) .add_message_handler(update_followers)
.add_message_handler(update_diff_base) .add_message_handler(update_diff_base)
.add_request_handler(get_private_user_info); .add_request_handler(get_private_user_info)
.add_message_handler(acknowledge_channel_message)
.add_message_handler(acknowledge_buffer_version);
Arc::new(server) Arc::new(server)
} }
@ -1904,13 +1906,10 @@ async fn follow(
.check_room_participants(room_id, leader_id, session.connection_id) .check_room_participants(room_id, leader_id, session.connection_id)
.await?; .await?;
let mut response_payload = session let response_payload = session
.peer .peer
.forward_request(session.connection_id, leader_id, request) .forward_request(session.connection_id, leader_id, request)
.await?; .await?;
response_payload
.views
.retain(|view| view.leader_id != Some(follower_id.into()));
response.send(response_payload)?; response.send(response_payload)?;
if let Some(project_id) = project_id { if let Some(project_id) = project_id {
@ -1971,14 +1970,17 @@ async fn update_followers(request: proto::UpdateFollowers, session: Session) ->
.await? .await?
}; };
let leader_id = request.variant.as_ref().and_then(|variant| match variant { // For now, don't send view update messages back to that view's current leader.
proto::update_followers::Variant::CreateView(payload) => payload.leader_id, let connection_id_to_omit = request.variant.as_ref().and_then(|variant| match variant {
proto::update_followers::Variant::UpdateView(payload) => payload.leader_id, proto::update_followers::Variant::UpdateView(payload) => payload.leader_id,
proto::update_followers::Variant::UpdateActiveView(payload) => payload.leader_id, _ => None,
}); });
for follower_peer_id in request.follower_ids.iter().copied() { for follower_peer_id in request.follower_ids.iter().copied() {
let follower_connection_id = follower_peer_id.into(); let follower_connection_id = follower_peer_id.into();
if Some(follower_peer_id) != leader_id && connection_ids.contains(&follower_connection_id) { if Some(follower_peer_id) != connection_id_to_omit
&& connection_ids.contains(&follower_connection_id)
{
session.peer.forward_send( session.peer.forward_send(
session.connection_id, session.connection_id,
follower_connection_id, follower_connection_id,
@ -2568,6 +2570,8 @@ async fn respond_to_channel_invite(
name: channel.name, name: channel.name,
}), }),
); );
update.unseen_channel_messages = result.channel_messages;
update.unseen_channel_buffer_changes = result.unseen_buffer_changes;
update.insert_edge = result.channels.edges; update.insert_edge = result.channels.edges;
update update
.channel_participants .channel_participants
@ -2691,7 +2695,7 @@ async fn update_channel_buffer(
let db = session.db().await; let db = session.db().await;
let channel_id = ChannelId::from_proto(request.channel_id); let channel_id = ChannelId::from_proto(request.channel_id);
let collaborators = db let (collaborators, non_collaborators, epoch, version) = db
.update_channel_buffer(channel_id, session.user_id, &request.operations) .update_channel_buffer(channel_id, session.user_id, &request.operations)
.await?; .await?;
@ -2704,6 +2708,29 @@ async fn update_channel_buffer(
}, },
&session.peer, &session.peer,
); );
let pool = &*session.connection_pool().await;
broadcast(
None,
non_collaborators
.iter()
.flat_map(|user_id| pool.user_connection_ids(*user_id)),
|peer_id| {
session.peer.send(
peer_id.into(),
proto::UpdateChannels {
unseen_channel_buffer_changes: vec![proto::UnseenChannelBufferChange {
channel_id: channel_id.to_proto(),
epoch: epoch as u64,
version: version.clone(),
}],
..Default::default()
},
)
},
);
Ok(()) Ok(())
} }
@ -2799,7 +2826,7 @@ async fn send_channel_message(
.ok_or_else(|| anyhow!("nonce can't be blank"))?; .ok_or_else(|| anyhow!("nonce can't be blank"))?;
let channel_id = ChannelId::from_proto(request.channel_id); let channel_id = ChannelId::from_proto(request.channel_id);
let (message_id, connection_ids) = session let (message_id, connection_ids, non_participants) = session
.db() .db()
.await .await
.create_channel_message( .create_channel_message(
@ -2829,6 +2856,27 @@ async fn send_channel_message(
response.send(proto::SendChannelMessageResponse { response.send(proto::SendChannelMessageResponse {
message: Some(message), message: Some(message),
})?; })?;
let pool = &*session.connection_pool().await;
broadcast(
None,
non_participants
.iter()
.flat_map(|user_id| pool.user_connection_ids(*user_id)),
|peer_id| {
session.peer.send(
peer_id.into(),
proto::UpdateChannels {
unseen_channel_messages: vec![proto::UnseenChannelMessage {
channel_id: channel_id.to_proto(),
message_id: message_id.to_proto(),
}],
..Default::default()
},
)
},
);
Ok(()) Ok(())
} }
@ -2851,6 +2899,38 @@ async fn remove_channel_message(
Ok(()) Ok(())
} }
async fn acknowledge_channel_message(
request: proto::AckChannelMessage,
session: Session,
) -> Result<()> {
let channel_id = ChannelId::from_proto(request.channel_id);
let message_id = MessageId::from_proto(request.message_id);
session
.db()
.await
.observe_channel_message(channel_id, session.user_id, message_id)
.await?;
Ok(())
}
async fn acknowledge_buffer_version(
request: proto::AckBufferOperation,
session: Session,
) -> Result<()> {
let buffer_id = BufferId::from_proto(request.buffer_id);
session
.db()
.await
.observe_buffer_version(
buffer_id,
session.user_id,
request.epoch as i32,
&request.version,
)
.await?;
Ok(())
}
async fn join_channel_chat( async fn join_channel_chat(
request: proto::JoinChannelChat, request: proto::JoinChannelChat,
response: Response<proto::JoinChannelChat>, response: Response<proto::JoinChannelChat>,
@ -2986,6 +3066,8 @@ fn build_initial_channels_update(
}); });
} }
update.unseen_channel_buffer_changes = channels.unseen_buffer_changes;
update.unseen_channel_messages = channels.channel_messages;
update.insert_edge = channels.channels.edges; update.insert_edge = channels.channels.edges;
for (channel_id, participants) in channels.channel_participants { for (channel_id, participants) in channels.channel_participants {

View file

@ -3,7 +3,7 @@ use crate::{
tests::TestServer, tests::TestServer,
}; };
use call::ActiveCall; use call::ActiveCall;
use channel::Channel; use channel::{Channel, ACKNOWLEDGE_DEBOUNCE_INTERVAL};
use client::ParticipantIndex; use client::ParticipantIndex;
use client::{Collaborator, UserId}; use client::{Collaborator, UserId};
use collab_ui::channel_view::ChannelView; use collab_ui::channel_view::ChannelView;
@ -410,10 +410,7 @@ async fn test_channel_buffer_disconnect(
channel_buffer_a.update(cx_a, |buffer, _| { channel_buffer_a.update(cx_a, |buffer, _| {
assert_eq!( assert_eq!(
buffer.channel().as_ref(), buffer.channel().as_ref(),
&Channel { &channel(channel_id, "the-channel")
id: channel_id,
name: "the-channel".to_string()
}
); );
assert!(!buffer.is_connected()); assert!(!buffer.is_connected());
}); });
@ -438,15 +435,21 @@ async fn test_channel_buffer_disconnect(
channel_buffer_b.update(cx_b, |buffer, _| { channel_buffer_b.update(cx_b, |buffer, _| {
assert_eq!( assert_eq!(
buffer.channel().as_ref(), buffer.channel().as_ref(),
&Channel { &channel(channel_id, "the-channel")
id: channel_id,
name: "the-channel".to_string()
}
); );
assert!(!buffer.is_connected()); assert!(!buffer.is_connected());
}); });
} }
fn channel(id: u64, name: &'static str) -> Channel {
Channel {
id,
name: name.to_string(),
unseen_note_version: None,
unseen_message_id: None,
}
}
#[gpui::test] #[gpui::test]
async fn test_rejoin_channel_buffer( async fn test_rejoin_channel_buffer(
deterministic: Arc<Deterministic>, deterministic: Arc<Deterministic>,
@ -627,6 +630,7 @@ async fn test_following_to_channel_notes_without_a_shared_project(
let mut server = TestServer::start(&deterministic).await; let mut server = TestServer::start(&deterministic).await;
let client_a = server.create_client(cx_a, "user_a").await; let client_a = server.create_client(cx_a, "user_a").await;
let client_b = server.create_client(cx_b, "user_b").await; let client_b = server.create_client(cx_b, "user_b").await;
let client_c = server.create_client(cx_c, "user_c").await; let client_c = server.create_client(cx_c, "user_c").await;
cx_a.update(editor::init); cx_a.update(editor::init);
@ -755,6 +759,126 @@ async fn test_following_to_channel_notes_without_a_shared_project(
}); });
} }
#[gpui::test]
async fn test_channel_buffer_changes(
deterministic: Arc<Deterministic>,
cx_a: &mut TestAppContext,
cx_b: &mut TestAppContext,
) {
deterministic.forbid_parking();
let mut server = TestServer::start(&deterministic).await;
let client_a = server.create_client(cx_a, "user_a").await;
let client_b = server.create_client(cx_b, "user_b").await;
let channel_id = server
.make_channel(
"the-channel",
None,
(&client_a, cx_a),
&mut [(&client_b, cx_b)],
)
.await;
let channel_buffer_a = client_a
.channel_store()
.update(cx_a, |store, cx| store.open_channel_buffer(channel_id, cx))
.await
.unwrap();
// Client A makes an edit, and client B should see that the note has changed.
channel_buffer_a.update(cx_a, |buffer, cx| {
buffer.buffer().update(cx, |buffer, cx| {
buffer.edit([(0..0, "1")], None, cx);
})
});
deterministic.run_until_parked();
let has_buffer_changed = cx_b.read(|cx| {
client_b
.channel_store()
.read(cx)
.has_channel_buffer_changed(channel_id)
.unwrap()
});
assert!(has_buffer_changed);
// Opening the buffer should clear the changed flag.
let project_b = client_b.build_empty_local_project(cx_b);
let workspace_b = client_b.build_workspace(&project_b, cx_b).root(cx_b);
let channel_view_b = cx_b
.update(|cx| ChannelView::open(channel_id, workspace_b.clone(), cx))
.await
.unwrap();
deterministic.run_until_parked();
let has_buffer_changed = cx_b.read(|cx| {
client_b
.channel_store()
.read(cx)
.has_channel_buffer_changed(channel_id)
.unwrap()
});
assert!(!has_buffer_changed);
// Editing the channel while the buffer is open should not show that the buffer has changed.
channel_buffer_a.update(cx_a, |buffer, cx| {
buffer.buffer().update(cx, |buffer, cx| {
buffer.edit([(0..0, "2")], None, cx);
})
});
deterministic.run_until_parked();
let has_buffer_changed = cx_b.read(|cx| {
client_b
.channel_store()
.read(cx)
.has_channel_buffer_changed(channel_id)
.unwrap()
});
assert!(!has_buffer_changed);
deterministic.advance_clock(ACKNOWLEDGE_DEBOUNCE_INTERVAL);
// Test that the server is tracking things correctly, and we retain our 'not changed'
// state across a disconnect
server.simulate_long_connection_interruption(client_b.peer_id().unwrap(), &deterministic);
let has_buffer_changed = cx_b.read(|cx| {
client_b
.channel_store()
.read(cx)
.has_channel_buffer_changed(channel_id)
.unwrap()
});
assert!(!has_buffer_changed);
// Closing the buffer should re-enable change tracking
cx_b.update(|cx| {
workspace_b.update(cx, |workspace, cx| {
workspace.close_all_items_and_panes(&Default::default(), cx)
});
drop(channel_view_b)
});
deterministic.run_until_parked();
channel_buffer_a.update(cx_a, |buffer, cx| {
buffer.buffer().update(cx, |buffer, cx| {
buffer.edit([(0..0, "3")], None, cx);
})
});
deterministic.run_until_parked();
let has_buffer_changed = cx_b.read(|cx| {
client_b
.channel_store()
.read(cx)
.has_channel_buffer_changed(channel_id)
.unwrap()
});
assert!(has_buffer_changed);
}
#[track_caller] #[track_caller]
fn assert_collaborators(collaborators: &HashMap<PeerId, Collaborator>, ids: &[Option<UserId>]) { fn assert_collaborators(collaborators: &HashMap<PeerId, Collaborator>, ids: &[Option<UserId>]) {
let mut user_ids = collaborators let mut user_ids = collaborators

View file

@ -1,7 +1,9 @@
use crate::{rpc::RECONNECT_TIMEOUT, tests::TestServer}; use crate::{rpc::RECONNECT_TIMEOUT, tests::TestServer};
use channel::{ChannelChat, ChannelMessageId}; use channel::{ChannelChat, ChannelMessageId};
use gpui::{executor::Deterministic, ModelHandle, TestAppContext}; use collab_ui::chat_panel::ChatPanel;
use gpui::{executor::Deterministic, BorrowAppContext, ModelHandle, TestAppContext};
use std::sync::Arc; use std::sync::Arc;
use workspace::dock::Panel;
#[gpui::test] #[gpui::test]
async fn test_basic_channel_messages( async fn test_basic_channel_messages(
@ -223,3 +225,136 @@ fn assert_messages(chat: &ModelHandle<ChannelChat>, messages: &[&str], cx: &mut
messages messages
); );
} }
#[gpui::test]
async fn test_channel_message_changes(
deterministic: Arc<Deterministic>,
cx_a: &mut TestAppContext,
cx_b: &mut TestAppContext,
) {
deterministic.forbid_parking();
let mut server = TestServer::start(&deterministic).await;
let client_a = server.create_client(cx_a, "user_a").await;
let client_b = server.create_client(cx_b, "user_b").await;
let channel_id = server
.make_channel(
"the-channel",
None,
(&client_a, cx_a),
&mut [(&client_b, cx_b)],
)
.await;
// Client A sends a message, client B should see that there is a new message.
let channel_chat_a = client_a
.channel_store()
.update(cx_a, |store, cx| store.open_channel_chat(channel_id, cx))
.await
.unwrap();
channel_chat_a
.update(cx_a, |c, cx| c.send_message("one".into(), cx).unwrap())
.await
.unwrap();
deterministic.run_until_parked();
let b_has_messages = cx_b.read_with(|cx| {
client_b
.channel_store()
.read(cx)
.has_new_messages(channel_id)
.unwrap()
});
assert!(b_has_messages);
// Opening the chat should clear the changed flag.
cx_b.update(|cx| {
collab_ui::init(&client_b.app_state, cx);
});
let project_b = client_b.build_empty_local_project(cx_b);
let workspace_b = client_b.build_workspace(&project_b, cx_b).root(cx_b);
let chat_panel_b = workspace_b.update(cx_b, |workspace, cx| ChatPanel::new(workspace, cx));
chat_panel_b
.update(cx_b, |chat_panel, cx| {
chat_panel.set_active(true, cx);
chat_panel.select_channel(channel_id, cx)
})
.await
.unwrap();
deterministic.run_until_parked();
let b_has_messages = cx_b.read_with(|cx| {
client_b
.channel_store()
.read(cx)
.has_new_messages(channel_id)
.unwrap()
});
assert!(!b_has_messages);
// Sending a message while the chat is open should not change the flag.
channel_chat_a
.update(cx_a, |c, cx| c.send_message("two".into(), cx).unwrap())
.await
.unwrap();
deterministic.run_until_parked();
let b_has_messages = cx_b.read_with(|cx| {
client_b
.channel_store()
.read(cx)
.has_new_messages(channel_id)
.unwrap()
});
assert!(!b_has_messages);
// Sending a message while the chat is closed should change the flag.
chat_panel_b.update(cx_b, |chat_panel, cx| {
chat_panel.set_active(false, cx);
});
// Sending a message while the chat is open should not change the flag.
channel_chat_a
.update(cx_a, |c, cx| c.send_message("three".into(), cx).unwrap())
.await
.unwrap();
deterministic.run_until_parked();
let b_has_messages = cx_b.read_with(|cx| {
client_b
.channel_store()
.read(cx)
.has_new_messages(channel_id)
.unwrap()
});
assert!(b_has_messages);
// Closing the chat should re-enable change tracking
cx_b.update(|_| drop(chat_panel_b));
channel_chat_a
.update(cx_a, |c, cx| c.send_message("four".into(), cx).unwrap())
.await
.unwrap();
deterministic.run_until_parked();
let b_has_messages = cx_b.read_with(|cx| {
client_b
.channel_store()
.read(cx)
.has_new_messages(channel_id)
.unwrap()
});
assert!(b_has_messages);
}

View file

@ -4,6 +4,7 @@ use collab_ui::project_shared_notification::ProjectSharedNotification;
use editor::{Editor, ExcerptRange, MultiBuffer}; use editor::{Editor, ExcerptRange, MultiBuffer};
use gpui::{executor::Deterministic, geometry::vector::vec2f, TestAppContext, ViewHandle}; use gpui::{executor::Deterministic, geometry::vector::vec2f, TestAppContext, ViewHandle};
use live_kit_client::MacOSDisplay; use live_kit_client::MacOSDisplay;
use rpc::proto::PeerId;
use serde_json::json; use serde_json::json;
use std::{borrow::Cow, sync::Arc}; use std::{borrow::Cow, sync::Arc};
use workspace::{ use workspace::{
@ -724,10 +725,9 @@ async fn test_peers_following_each_other(
.await .await
.unwrap(); .unwrap();
// Client A opens some editors. // Client A opens a file.
let workspace_a = client_a.build_workspace(&project_a, cx_a).root(cx_a); let workspace_a = client_a.build_workspace(&project_a, cx_a).root(cx_a);
let pane_a1 = workspace_a.read_with(cx_a, |workspace, _| workspace.active_pane().clone()); workspace_a
let _editor_a1 = workspace_a
.update(cx_a, |workspace, cx| { .update(cx_a, |workspace, cx| {
workspace.open_path((worktree_id, "1.txt"), None, true, cx) workspace.open_path((worktree_id, "1.txt"), None, true, cx)
}) })
@ -736,10 +736,9 @@ async fn test_peers_following_each_other(
.downcast::<Editor>() .downcast::<Editor>()
.unwrap(); .unwrap();
// Client B opens an editor. // Client B opens a different file.
let workspace_b = client_b.build_workspace(&project_b, cx_b).root(cx_b); let workspace_b = client_b.build_workspace(&project_b, cx_b).root(cx_b);
let pane_b1 = workspace_b.read_with(cx_b, |workspace, _| workspace.active_pane().clone()); workspace_b
let _editor_b1 = workspace_b
.update(cx_b, |workspace, cx| { .update(cx_b, |workspace, cx| {
workspace.open_path((worktree_id, "2.txt"), None, true, cx) workspace.open_path((worktree_id, "2.txt"), None, true, cx)
}) })
@ -754,9 +753,7 @@ async fn test_peers_following_each_other(
}); });
workspace_a workspace_a
.update(cx_a, |workspace, cx| { .update(cx_a, |workspace, cx| {
assert_ne!(*workspace.active_pane(), pane_a1); workspace.follow(client_b.peer_id().unwrap(), cx).unwrap()
let leader_id = *project_a.read(cx).collaborators().keys().next().unwrap();
workspace.follow(leader_id, cx).unwrap()
}) })
.await .await
.unwrap(); .unwrap();
@ -765,85 +762,443 @@ async fn test_peers_following_each_other(
}); });
workspace_b workspace_b
.update(cx_b, |workspace, cx| { .update(cx_b, |workspace, cx| {
assert_ne!(*workspace.active_pane(), pane_b1); workspace.follow(client_a.peer_id().unwrap(), cx).unwrap()
let leader_id = *project_b.read(cx).collaborators().keys().next().unwrap();
workspace.follow(leader_id, cx).unwrap()
}) })
.await .await
.unwrap(); .unwrap();
workspace_a.update(cx_a, |workspace, cx| { // Clients A and B return focus to the original files they had open
workspace.activate_next_pane(cx); workspace_a.update(cx_a, |workspace, cx| workspace.activate_next_pane(cx));
}); workspace_b.update(cx_b, |workspace, cx| workspace.activate_next_pane(cx));
// Wait for focus effects to be fully flushed deterministic.run_until_parked();
workspace_a.update(cx_a, |workspace, _| {
assert_eq!(*workspace.active_pane(), pane_a1);
});
// Both clients see the other client's focused file in their right pane.
assert_eq!(
pane_summaries(&workspace_a, cx_a),
&[
PaneSummary {
active: true,
leader: None,
items: vec![(true, "1.txt".into())]
},
PaneSummary {
active: false,
leader: client_b.peer_id(),
items: vec![(false, "1.txt".into()), (true, "2.txt".into())]
},
]
);
assert_eq!(
pane_summaries(&workspace_b, cx_b),
&[
PaneSummary {
active: true,
leader: None,
items: vec![(true, "2.txt".into())]
},
PaneSummary {
active: false,
leader: client_a.peer_id(),
items: vec![(false, "2.txt".into()), (true, "1.txt".into())]
},
]
);
// Clients A and B each open a new file.
workspace_a workspace_a
.update(cx_a, |workspace, cx| { .update(cx_a, |workspace, cx| {
workspace.open_path((worktree_id, "3.txt"), None, true, cx) workspace.open_path((worktree_id, "3.txt"), None, true, cx)
}) })
.await .await
.unwrap(); .unwrap();
workspace_b.update(cx_b, |workspace, cx| {
workspace.activate_next_pane(cx);
});
workspace_b workspace_b
.update(cx_b, |workspace, cx| { .update(cx_b, |workspace, cx| {
assert_eq!(*workspace.active_pane(), pane_b1);
workspace.open_path((worktree_id, "4.txt"), None, true, cx) workspace.open_path((worktree_id, "4.txt"), None, true, cx)
}) })
.await .await
.unwrap(); .unwrap();
cx_a.foreground().run_until_parked(); deterministic.run_until_parked();
// Ensure leader updates don't change the active pane of followers // Both client's see the other client open the new file, but keep their
workspace_a.read_with(cx_a, |workspace, _| { // focus on their own active pane.
assert_eq!(*workspace.active_pane(), pane_a1);
});
workspace_b.read_with(cx_b, |workspace, _| {
assert_eq!(*workspace.active_pane(), pane_b1);
});
// Ensure peers following each other doesn't cause an infinite loop.
assert_eq!( assert_eq!(
workspace_a.read_with(cx_a, |workspace, cx| workspace pane_summaries(&workspace_a, cx_a),
.active_item(cx) &[
.unwrap() PaneSummary {
.project_path(cx)), active: true,
Some((worktree_id, "3.txt").into()) leader: None,
items: vec![(false, "1.txt".into()), (true, "3.txt".into())]
},
PaneSummary {
active: false,
leader: client_b.peer_id(),
items: vec![
(false, "1.txt".into()),
(false, "2.txt".into()),
(true, "4.txt".into())
]
},
]
); );
workspace_a.update(cx_a, |workspace, cx| { assert_eq!(
assert_eq!( pane_summaries(&workspace_b, cx_b),
workspace.active_item(cx).unwrap().project_path(cx), &[
Some((worktree_id, "3.txt").into()) PaneSummary {
); active: true,
workspace.activate_next_pane(cx); leader: None,
items: vec![(false, "2.txt".into()), (true, "4.txt".into())]
},
PaneSummary {
active: false,
leader: client_a.peer_id(),
items: vec![
(false, "2.txt".into()),
(false, "1.txt".into()),
(true, "3.txt".into())
]
},
]
);
// Client A focuses their right pane, in which they're following client B.
workspace_a.update(cx_a, |workspace, cx| workspace.activate_next_pane(cx));
deterministic.run_until_parked();
// Client B sees that client A is now looking at the same file as them.
assert_eq!(
pane_summaries(&workspace_a, cx_a),
&[
PaneSummary {
active: false,
leader: None,
items: vec![(false, "1.txt".into()), (true, "3.txt".into())]
},
PaneSummary {
active: true,
leader: client_b.peer_id(),
items: vec![
(false, "1.txt".into()),
(false, "2.txt".into()),
(true, "4.txt".into())
]
},
]
);
assert_eq!(
pane_summaries(&workspace_b, cx_b),
&[
PaneSummary {
active: true,
leader: None,
items: vec![(false, "2.txt".into()), (true, "4.txt".into())]
},
PaneSummary {
active: false,
leader: client_a.peer_id(),
items: vec![
(false, "2.txt".into()),
(false, "1.txt".into()),
(false, "3.txt".into()),
(true, "4.txt".into())
]
},
]
);
// Client B focuses their right pane, in which they're following client A,
// who is following them.
workspace_b.update(cx_b, |workspace, cx| workspace.activate_next_pane(cx));
deterministic.run_until_parked();
// Client A sees that client B is now looking at the same file as them.
assert_eq!(
pane_summaries(&workspace_b, cx_b),
&[
PaneSummary {
active: false,
leader: None,
items: vec![(false, "2.txt".into()), (true, "4.txt".into())]
},
PaneSummary {
active: true,
leader: client_a.peer_id(),
items: vec![
(false, "2.txt".into()),
(false, "1.txt".into()),
(false, "3.txt".into()),
(true, "4.txt".into())
]
},
]
);
assert_eq!(
pane_summaries(&workspace_a, cx_a),
&[
PaneSummary {
active: false,
leader: None,
items: vec![(false, "1.txt".into()), (true, "3.txt".into())]
},
PaneSummary {
active: true,
leader: client_b.peer_id(),
items: vec![
(false, "1.txt".into()),
(false, "2.txt".into()),
(true, "4.txt".into())
]
},
]
);
// Client B focuses a file that they previously followed A to, breaking
// the follow.
workspace_b.update(cx_b, |workspace, cx| {
workspace.active_pane().update(cx, |pane, cx| {
pane.activate_prev_item(true, cx);
});
}); });
deterministic.run_until_parked();
// Both clients see that client B is looking at that previous file.
assert_eq!(
pane_summaries(&workspace_b, cx_b),
&[
PaneSummary {
active: false,
leader: None,
items: vec![(false, "2.txt".into()), (true, "4.txt".into())]
},
PaneSummary {
active: true,
leader: None,
items: vec![
(false, "2.txt".into()),
(false, "1.txt".into()),
(true, "3.txt".into()),
(false, "4.txt".into())
]
},
]
);
assert_eq!(
pane_summaries(&workspace_a, cx_a),
&[
PaneSummary {
active: false,
leader: None,
items: vec![(false, "1.txt".into()), (true, "3.txt".into())]
},
PaneSummary {
active: true,
leader: client_b.peer_id(),
items: vec![
(false, "1.txt".into()),
(false, "2.txt".into()),
(false, "4.txt".into()),
(true, "3.txt".into()),
]
},
]
);
// Client B closes tabs, some of which were originally opened by client A,
// and some of which were originally opened by client B.
workspace_b.update(cx_b, |workspace, cx| {
workspace.active_pane().update(cx, |pane, cx| {
pane.close_inactive_items(&Default::default(), cx)
.unwrap()
.detach();
});
});
deterministic.run_until_parked();
// Both clients see that Client B is looking at the previous tab.
assert_eq!(
pane_summaries(&workspace_b, cx_b),
&[
PaneSummary {
active: false,
leader: None,
items: vec![(false, "2.txt".into()), (true, "4.txt".into())]
},
PaneSummary {
active: true,
leader: None,
items: vec![(true, "3.txt".into()),]
},
]
);
assert_eq!(
pane_summaries(&workspace_a, cx_a),
&[
PaneSummary {
active: false,
leader: None,
items: vec![(false, "1.txt".into()), (true, "3.txt".into())]
},
PaneSummary {
active: true,
leader: client_b.peer_id(),
items: vec![
(false, "1.txt".into()),
(false, "2.txt".into()),
(false, "4.txt".into()),
(true, "3.txt".into()),
]
},
]
);
// Client B follows client A again.
workspace_b
.update(cx_b, |workspace, cx| {
workspace.follow(client_a.peer_id().unwrap(), cx).unwrap()
})
.await
.unwrap();
// Client A cycles through some tabs.
workspace_a.update(cx_a, |workspace, cx| {
workspace.active_pane().update(cx, |pane, cx| {
pane.activate_prev_item(true, cx);
});
});
deterministic.run_until_parked();
// Client B follows client A into those tabs.
assert_eq!(
pane_summaries(&workspace_a, cx_a),
&[
PaneSummary {
active: false,
leader: None,
items: vec![(false, "1.txt".into()), (true, "3.txt".into())]
},
PaneSummary {
active: true,
leader: None,
items: vec![
(false, "1.txt".into()),
(false, "2.txt".into()),
(true, "4.txt".into()),
(false, "3.txt".into()),
]
},
]
);
assert_eq!(
pane_summaries(&workspace_b, cx_b),
&[
PaneSummary {
active: false,
leader: None,
items: vec![(false, "2.txt".into()), (true, "4.txt".into())]
},
PaneSummary {
active: true,
leader: client_a.peer_id(),
items: vec![(false, "3.txt".into()), (true, "4.txt".into())]
},
]
);
workspace_a.update(cx_a, |workspace, cx| { workspace_a.update(cx_a, |workspace, cx| {
assert_eq!( workspace.active_pane().update(cx, |pane, cx| {
workspace.active_item(cx).unwrap().project_path(cx), pane.activate_prev_item(true, cx);
Some((worktree_id, "4.txt").into()) });
);
}); });
deterministic.run_until_parked();
workspace_b.update(cx_b, |workspace, cx| { assert_eq!(
assert_eq!( pane_summaries(&workspace_a, cx_a),
workspace.active_item(cx).unwrap().project_path(cx), &[
Some((worktree_id, "4.txt").into()) PaneSummary {
); active: false,
workspace.activate_next_pane(cx); leader: None,
}); items: vec![(false, "1.txt".into()), (true, "3.txt".into())]
},
PaneSummary {
active: true,
leader: None,
items: vec![
(false, "1.txt".into()),
(true, "2.txt".into()),
(false, "4.txt".into()),
(false, "3.txt".into()),
]
},
]
);
assert_eq!(
pane_summaries(&workspace_b, cx_b),
&[
PaneSummary {
active: false,
leader: None,
items: vec![(false, "2.txt".into()), (true, "4.txt".into())]
},
PaneSummary {
active: true,
leader: client_a.peer_id(),
items: vec![
(false, "3.txt".into()),
(false, "4.txt".into()),
(true, "2.txt".into())
]
},
]
);
workspace_b.update(cx_b, |workspace, cx| { workspace_a.update(cx_a, |workspace, cx| {
assert_eq!( workspace.active_pane().update(cx, |pane, cx| {
workspace.active_item(cx).unwrap().project_path(cx), pane.activate_prev_item(true, cx);
Some((worktree_id, "3.txt").into()) });
);
}); });
deterministic.run_until_parked();
assert_eq!(
pane_summaries(&workspace_a, cx_a),
&[
PaneSummary {
active: false,
leader: None,
items: vec![(false, "1.txt".into()), (true, "3.txt".into())]
},
PaneSummary {
active: true,
leader: None,
items: vec![
(true, "1.txt".into()),
(false, "2.txt".into()),
(false, "4.txt".into()),
(false, "3.txt".into()),
]
},
]
);
assert_eq!(
pane_summaries(&workspace_b, cx_b),
&[
PaneSummary {
active: false,
leader: None,
items: vec![(false, "2.txt".into()), (true, "4.txt".into())]
},
PaneSummary {
active: true,
leader: client_a.peer_id(),
items: vec![
(false, "3.txt".into()),
(false, "4.txt".into()),
(false, "2.txt".into()),
(true, "1.txt".into()),
]
},
]
);
} }
#[gpui::test(iterations = 10)] #[gpui::test(iterations = 10)]
@ -1074,24 +1429,6 @@ async fn test_peers_simultaneously_following_each_other(
}); });
} }
fn visible_push_notifications(
cx: &mut TestAppContext,
) -> Vec<gpui::ViewHandle<ProjectSharedNotification>> {
let mut ret = Vec::new();
for window in cx.windows() {
window.read_with(cx, |window| {
if let Some(handle) = window
.root_view()
.clone()
.downcast::<ProjectSharedNotification>()
{
ret.push(handle)
}
});
}
ret
}
#[gpui::test(iterations = 10)] #[gpui::test(iterations = 10)]
async fn test_following_across_workspaces( async fn test_following_across_workspaces(
deterministic: Arc<Deterministic>, deterministic: Arc<Deterministic>,
@ -1304,3 +1641,59 @@ async fn test_following_across_workspaces(
assert_eq!(item.tab_description(0, cx).unwrap(), Cow::Borrowed("y.rs")); assert_eq!(item.tab_description(0, cx).unwrap(), Cow::Borrowed("y.rs"));
}); });
} }
fn visible_push_notifications(
cx: &mut TestAppContext,
) -> Vec<gpui::ViewHandle<ProjectSharedNotification>> {
let mut ret = Vec::new();
for window in cx.windows() {
window.read_with(cx, |window| {
if let Some(handle) = window
.root_view()
.clone()
.downcast::<ProjectSharedNotification>()
{
ret.push(handle)
}
});
}
ret
}
#[derive(Debug, PartialEq, Eq)]
struct PaneSummary {
active: bool,
leader: Option<PeerId>,
items: Vec<(bool, String)>,
}
fn pane_summaries(workspace: &ViewHandle<Workspace>, cx: &mut TestAppContext) -> Vec<PaneSummary> {
workspace.read_with(cx, |workspace, cx| {
let active_pane = workspace.active_pane();
workspace
.panes()
.iter()
.map(|pane| {
let leader = workspace.leader_for_pane(pane);
let active = pane == active_pane;
let pane = pane.read(cx);
let active_ix = pane.active_item_index();
PaneSummary {
active,
leader,
items: pane
.items()
.enumerate()
.map(|(ix, item)| {
(
ix == active_ix,
item.tab_description(0, cx)
.map_or(String::new(), |s| s.to_string()),
)
})
.collect(),
}
})
.collect()
})
}

View file

@ -1,7 +1,7 @@
use crate::{ use crate::{
db::{tests::TestDb, NewUserParams, UserId}, db::{tests::TestDb, NewUserParams, UserId},
executor::Executor, executor::Executor,
rpc::{Server, CLEANUP_TIMEOUT}, rpc::{Server, CLEANUP_TIMEOUT, RECONNECT_TIMEOUT},
AppState, AppState,
}; };
use anyhow::anyhow; use anyhow::anyhow;
@ -17,6 +17,7 @@ use gpui::{executor::Deterministic, ModelHandle, Task, TestAppContext, WindowHan
use language::LanguageRegistry; use language::LanguageRegistry;
use parking_lot::Mutex; use parking_lot::Mutex;
use project::{Project, WorktreeId}; use project::{Project, WorktreeId};
use rpc::RECEIVE_TIMEOUT;
use settings::SettingsStore; use settings::SettingsStore;
use std::{ use std::{
cell::{Ref, RefCell, RefMut}, cell::{Ref, RefCell, RefMut},
@ -151,12 +152,12 @@ impl TestServer {
Arc::get_mut(&mut client) Arc::get_mut(&mut client)
.unwrap() .unwrap()
.set_id(user_id.0 as usize) .set_id(user_id.to_proto())
.override_authenticate(move |cx| { .override_authenticate(move |cx| {
cx.spawn(|_| async move { cx.spawn(|_| async move {
let access_token = "the-token".to_string(); let access_token = "the-token".to_string();
Ok(Credentials { Ok(Credentials {
user_id: user_id.0 as u64, user_id: user_id.to_proto(),
access_token, access_token,
}) })
}) })
@ -255,6 +256,19 @@ impl TestServer {
.store(true, SeqCst); .store(true, SeqCst);
} }
pub fn simulate_long_connection_interruption(
&self,
peer_id: PeerId,
deterministic: &Arc<Deterministic>,
) {
self.forbid_connections();
self.disconnect_client(peer_id);
deterministic.advance_clock(RECEIVE_TIMEOUT + RECONNECT_TIMEOUT);
self.allow_connections();
deterministic.advance_clock(RECEIVE_TIMEOUT + RECONNECT_TIMEOUT);
deterministic.run_until_parked();
}
pub fn forbid_connections(&self) { pub fn forbid_connections(&self) {
self.forbid_connections.store(true, SeqCst); self.forbid_connections.store(true, SeqCst);
} }

View file

@ -37,6 +37,7 @@ fuzzy = { path = "../fuzzy" }
gpui = { path = "../gpui" } gpui = { path = "../gpui" }
language = { path = "../language" } language = { path = "../language" }
menu = { path = "../menu" } menu = { path = "../menu" }
rich_text = { path = "../rich_text" }
picker = { path = "../picker" } picker = { path = "../picker" }
project = { path = "../project" } project = { path = "../project" }
recent_projects = {path = "../recent_projects"} recent_projects = {path = "../recent_projects"}

View file

@ -1,6 +1,6 @@
use anyhow::{anyhow, Result}; use anyhow::{anyhow, Result};
use call::report_call_event_for_channel; use call::report_call_event_for_channel;
use channel::{Channel, ChannelBuffer, ChannelBufferEvent, ChannelId}; use channel::{Channel, ChannelBuffer, ChannelBufferEvent, ChannelId, ChannelStore};
use client::{ use client::{
proto::{self, PeerId}, proto::{self, PeerId},
Collaborator, ParticipantIndex, Collaborator, ParticipantIndex,
@ -36,6 +36,7 @@ pub fn init(cx: &mut AppContext) {
pub struct ChannelView { pub struct ChannelView {
pub editor: ViewHandle<Editor>, pub editor: ViewHandle<Editor>,
project: ModelHandle<Project>, project: ModelHandle<Project>,
channel_store: ModelHandle<ChannelStore>,
channel_buffer: ModelHandle<ChannelBuffer>, channel_buffer: ModelHandle<ChannelBuffer>,
remote_id: Option<ViewId>, remote_id: Option<ViewId>,
_editor_event_subscription: Subscription, _editor_event_subscription: Subscription,
@ -94,7 +95,13 @@ impl ChannelView {
pane.update(&mut cx, |pane, cx| { pane.update(&mut cx, |pane, cx| {
pane.items_of_type::<Self>() pane.items_of_type::<Self>()
.find(|channel_view| channel_view.read(cx).channel_buffer == channel_buffer) .find(|channel_view| channel_view.read(cx).channel_buffer == channel_buffer)
.unwrap_or_else(|| cx.add_view(|cx| Self::new(project, channel_buffer, cx))) .unwrap_or_else(|| {
cx.add_view(|cx| {
let mut this = Self::new(project, channel_store, channel_buffer, cx);
this.acknowledge_buffer_version(cx);
this
})
})
}) })
.ok_or_else(|| anyhow!("pane was dropped")) .ok_or_else(|| anyhow!("pane was dropped"))
}) })
@ -102,6 +109,7 @@ impl ChannelView {
pub fn new( pub fn new(
project: ModelHandle<Project>, project: ModelHandle<Project>,
channel_store: ModelHandle<ChannelStore>,
channel_buffer: ModelHandle<ChannelBuffer>, channel_buffer: ModelHandle<ChannelBuffer>,
cx: &mut ViewContext<Self>, cx: &mut ViewContext<Self>,
) -> Self { ) -> Self {
@ -121,6 +129,7 @@ impl ChannelView {
Self { Self {
editor, editor,
project, project,
channel_store,
channel_buffer, channel_buffer,
remote_id: None, remote_id: None,
_editor_event_subscription, _editor_event_subscription,
@ -137,13 +146,44 @@ impl ChannelView {
event: &ChannelBufferEvent, event: &ChannelBufferEvent,
cx: &mut ViewContext<Self>, cx: &mut ViewContext<Self>,
) { ) {
if let ChannelBufferEvent::Disconnected = event { match event {
self.editor.update(cx, |editor, cx| { ChannelBufferEvent::Disconnected => self.editor.update(cx, |editor, cx| {
editor.set_read_only(true); editor.set_read_only(true);
cx.notify(); cx.notify();
}) }),
ChannelBufferEvent::BufferEdited => {
if cx.is_self_focused() || self.editor.is_focused(cx) {
self.acknowledge_buffer_version(cx);
} else {
self.channel_store.update(cx, |store, cx| {
let channel_buffer = self.channel_buffer.read(cx);
store.notes_changed(
channel_buffer.channel().id,
channel_buffer.epoch(),
&channel_buffer.buffer().read(cx).version(),
cx,
)
});
}
}
_ => {}
} }
} }
fn acknowledge_buffer_version(&mut self, cx: &mut ViewContext<'_, '_, ChannelView>) {
self.channel_store.update(cx, |store, cx| {
let channel_buffer = self.channel_buffer.read(cx);
store.acknowledge_notes_version(
channel_buffer.channel().id,
channel_buffer.epoch(),
&channel_buffer.buffer().read(cx).version(),
cx,
)
});
self.channel_buffer.update(cx, |buffer, cx| {
buffer.acknowledge_buffer_version(cx);
});
}
} }
impl Entity for ChannelView { impl Entity for ChannelView {
@ -161,6 +201,7 @@ impl View for ChannelView {
fn focus_in(&mut self, _: AnyViewHandle, cx: &mut ViewContext<Self>) { fn focus_in(&mut self, _: AnyViewHandle, cx: &mut ViewContext<Self>) {
if cx.is_self_focused() { if cx.is_self_focused() {
self.acknowledge_buffer_version(cx);
cx.focus(self.editor.as_any()) cx.focus(self.editor.as_any())
} }
} }
@ -200,6 +241,7 @@ impl Item for ChannelView {
fn clone_on_split(&self, _: WorkspaceId, cx: &mut ViewContext<Self>) -> Option<Self> { fn clone_on_split(&self, _: WorkspaceId, cx: &mut ViewContext<Self>) -> Option<Self> {
Some(Self::new( Some(Self::new(
self.project.clone(), self.project.clone(),
self.channel_store.clone(),
self.channel_buffer.clone(), self.channel_buffer.clone(),
cx, cx,
)) ))

View file

@ -3,6 +3,7 @@ use anyhow::Result;
use call::ActiveCall; use call::ActiveCall;
use channel::{ChannelChat, ChannelChatEvent, ChannelMessageId, ChannelStore}; use channel::{ChannelChat, ChannelChatEvent, ChannelMessageId, ChannelStore};
use client::Client; use client::Client;
use collections::HashMap;
use db::kvp::KEY_VALUE_STORE; use db::kvp::KEY_VALUE_STORE;
use editor::Editor; use editor::Editor;
use feature_flags::{ChannelsAlpha, FeatureFlagAppExt}; use feature_flags::{ChannelsAlpha, FeatureFlagAppExt};
@ -12,12 +13,13 @@ use gpui::{
platform::{CursorStyle, MouseButton}, platform::{CursorStyle, MouseButton},
serde_json, serde_json,
views::{ItemType, Select, SelectStyle}, views::{ItemType, Select, SelectStyle},
AnyViewHandle, AppContext, AsyncAppContext, Entity, ModelHandle, Subscription, Task, View, AnyViewHandle, AppContext, AsyncAppContext, Entity, ImageData, ModelHandle, Subscription, Task,
ViewContext, ViewHandle, WeakViewHandle, View, ViewContext, ViewHandle, WeakViewHandle,
}; };
use language::language_settings::SoftWrap; use language::{language_settings::SoftWrap, LanguageRegistry};
use menu::Confirm; use menu::Confirm;
use project::Fs; use project::Fs;
use rich_text::RichText;
use serde::{Deserialize, Serialize}; use serde::{Deserialize, Serialize};
use settings::SettingsStore; use settings::SettingsStore;
use std::sync::Arc; use std::sync::Arc;
@ -35,6 +37,7 @@ const CHAT_PANEL_KEY: &'static str = "ChatPanel";
pub struct ChatPanel { pub struct ChatPanel {
client: Arc<Client>, client: Arc<Client>,
channel_store: ModelHandle<ChannelStore>, channel_store: ModelHandle<ChannelStore>,
languages: Arc<LanguageRegistry>,
active_chat: Option<(ModelHandle<ChannelChat>, Subscription)>, active_chat: Option<(ModelHandle<ChannelChat>, Subscription)>,
message_list: ListState<ChatPanel>, message_list: ListState<ChatPanel>,
input_editor: ViewHandle<Editor>, input_editor: ViewHandle<Editor>,
@ -42,10 +45,12 @@ pub struct ChatPanel {
local_timezone: UtcOffset, local_timezone: UtcOffset,
fs: Arc<dyn Fs>, fs: Arc<dyn Fs>,
width: Option<f32>, width: Option<f32>,
active: bool,
pending_serialization: Task<Option<()>>, pending_serialization: Task<Option<()>>,
subscriptions: Vec<gpui::Subscription>, subscriptions: Vec<gpui::Subscription>,
workspace: WeakViewHandle<Workspace>, workspace: WeakViewHandle<Workspace>,
has_focus: bool, has_focus: bool,
markdown_data: HashMap<ChannelMessageId, RichText>,
} }
#[derive(Serialize, Deserialize)] #[derive(Serialize, Deserialize)]
@ -77,6 +82,7 @@ impl ChatPanel {
let fs = workspace.app_state().fs.clone(); let fs = workspace.app_state().fs.clone();
let client = workspace.app_state().client.clone(); let client = workspace.app_state().client.clone();
let channel_store = workspace.app_state().channel_store.clone(); let channel_store = workspace.app_state().channel_store.clone();
let languages = workspace.app_state().languages.clone();
let input_editor = cx.add_view(|cx| { let input_editor = cx.add_view(|cx| {
let mut editor = Editor::auto_height( let mut editor = Editor::auto_height(
@ -129,6 +135,8 @@ impl ChatPanel {
fs, fs,
client, client,
channel_store, channel_store,
languages,
active_chat: Default::default(), active_chat: Default::default(),
pending_serialization: Task::ready(None), pending_serialization: Task::ready(None),
message_list, message_list,
@ -138,7 +146,9 @@ impl ChatPanel {
has_focus: false, has_focus: false,
subscriptions: Vec::new(), subscriptions: Vec::new(),
workspace: workspace_handle, workspace: workspace_handle,
active: false,
width: None, width: None,
markdown_data: Default::default(),
}; };
let mut old_dock_position = this.position(cx); let mut old_dock_position = this.position(cx);
@ -154,9 +164,9 @@ impl ChatPanel {
}), }),
); );
this.init_active_channel(cx); this.update_channel_count(cx);
cx.observe(&this.channel_store, |this, _, cx| { cx.observe(&this.channel_store, |this, _, cx| {
this.init_active_channel(cx); this.update_channel_count(cx)
}) })
.detach(); .detach();
@ -175,10 +185,33 @@ impl ChatPanel {
}) })
.detach(); .detach();
let markdown = this.languages.language_for_name("Markdown");
cx.spawn(|this, mut cx| async move {
let markdown = markdown.await?;
this.update(&mut cx, |this, cx| {
this.input_editor.update(cx, |editor, cx| {
editor.buffer().update(cx, |multi_buffer, cx| {
multi_buffer
.as_singleton()
.unwrap()
.update(cx, |buffer, cx| buffer.set_language(Some(markdown), cx))
})
})
})?;
anyhow::Ok(())
})
.detach_and_log_err(cx);
this this
}) })
} }
pub fn active_chat(&self) -> Option<ModelHandle<ChannelChat>> {
self.active_chat.as_ref().map(|(chat, _)| chat.clone())
}
pub fn load( pub fn load(
workspace: WeakViewHandle<Workspace>, workspace: WeakViewHandle<Workspace>,
cx: AsyncAppContext, cx: AsyncAppContext,
@ -225,10 +258,8 @@ impl ChatPanel {
); );
} }
fn init_active_channel(&mut self, cx: &mut ViewContext<Self>) { fn update_channel_count(&mut self, cx: &mut ViewContext<Self>) {
let channel_count = self.channel_store.read(cx).channel_count(); let channel_count = self.channel_store.read(cx).channel_count();
self.message_list.reset(0);
self.active_chat = None;
self.channel_select.update(cx, |select, cx| { self.channel_select.update(cx, |select, cx| {
select.set_item_count(channel_count, cx); select.set_item_count(channel_count, cx);
}); });
@ -247,6 +278,7 @@ impl ChatPanel {
} }
let subscription = cx.subscribe(&chat, Self::channel_did_change); let subscription = cx.subscribe(&chat, Self::channel_did_change);
self.active_chat = Some((chat, subscription)); self.active_chat = Some((chat, subscription));
self.acknowledge_last_message(cx);
self.channel_select.update(cx, |select, cx| { self.channel_select.update(cx, |select, cx| {
if let Some(ix) = self.channel_store.read(cx).index_of_channel(id) { if let Some(ix) = self.channel_store.read(cx).index_of_channel(id) {
select.set_selected_index(ix, cx); select.set_selected_index(ix, cx);
@ -268,11 +300,34 @@ impl ChatPanel {
new_count, new_count,
} => { } => {
self.message_list.splice(old_range.clone(), *new_count); self.message_list.splice(old_range.clone(), *new_count);
if self.active {
self.acknowledge_last_message(cx);
}
}
ChannelChatEvent::NewMessage {
channel_id,
message_id,
} => {
if !self.active {
self.channel_store.update(cx, |store, cx| {
store.new_message(*channel_id, *message_id, cx)
})
}
} }
} }
cx.notify(); cx.notify();
} }
fn acknowledge_last_message(&mut self, cx: &mut ViewContext<'_, '_, ChatPanel>) {
if self.active {
if let Some((chat, _)) = &self.active_chat {
chat.update(cx, |chat, cx| {
chat.acknowledge_last_message(cx);
});
}
}
}
fn render_channel(&self, cx: &mut ViewContext<Self>) -> AnyElement<Self> { fn render_channel(&self, cx: &mut ViewContext<Self>) -> AnyElement<Self> {
let theme = theme::current(cx); let theme = theme::current(cx);
Flex::column() Flex::column()
@ -299,13 +354,33 @@ impl ChatPanel {
messages.flex(1., true).into_any() messages.flex(1., true).into_any()
} }
fn render_message(&self, ix: usize, cx: &mut ViewContext<Self>) -> AnyElement<Self> { fn render_message(&mut self, ix: usize, cx: &mut ViewContext<Self>) -> AnyElement<Self> {
let message = self.active_chat.as_ref().unwrap().0.read(cx).message(ix); let (message, is_continuation, is_last) = {
let active_chat = self.active_chat.as_ref().unwrap().0.read(cx);
let last_message = active_chat.message(ix.saturating_sub(1));
let this_message = active_chat.message(ix);
let is_continuation = last_message.id != this_message.id
&& this_message.sender.id == last_message.sender.id;
(
active_chat.message(ix).clone(),
is_continuation,
active_chat.message_count() == ix + 1,
)
};
let is_pending = message.is_pending();
let text = self
.markdown_data
.entry(message.id)
.or_insert_with(|| rich_text::render_markdown(message.body, &self.languages, None));
let now = OffsetDateTime::now_utc(); let now = OffsetDateTime::now_utc();
let theme = theme::current(cx); let theme = theme::current(cx);
let style = if message.is_pending() { let style = if is_pending {
&theme.chat_panel.pending_message &theme.chat_panel.pending_message
} else if is_continuation {
&theme.chat_panel.continuation_message
} else { } else {
&theme.chat_panel.message &theme.chat_panel.message
}; };
@ -318,52 +393,90 @@ impl ChatPanel {
None None
}; };
enum DeleteMessage {} enum MessageBackgroundHighlight {}
MouseEventHandler::new::<MessageBackgroundHighlight, _>(ix, cx, |state, cx| {
let body = message.body.clone(); let container = style.container.style_for(state);
Flex::column() if is_continuation {
.with_child(
Flex::row() Flex::row()
.with_child( .with_child(
Label::new( text.element(
message.sender.github_login.clone(), theme.editor.syntax.clone(),
style.sender.text.clone(), style.body.clone(),
theme.editor.document_highlight_read_background,
cx,
) )
.contained() .flex(1., true),
.with_style(style.sender.container), )
.with_child(render_remove(message_id_to_remove, cx, &theme))
.contained()
.with_style(*container)
.with_margin_bottom(if is_last {
theme.chat_panel.last_message_bottom_spacing
} else {
0.
})
.into_any()
} else {
Flex::column()
.with_child(
Flex::row()
.with_child(
Flex::row()
.with_child(render_avatar(
message.sender.avatar.clone(),
&theme,
))
.with_child(
Label::new(
message.sender.github_login.clone(),
style.sender.text.clone(),
)
.contained()
.with_style(style.sender.container),
)
.with_child(
Label::new(
format_timestamp(
message.timestamp,
now,
self.local_timezone,
),
style.timestamp.text.clone(),
)
.contained()
.with_style(style.timestamp.container),
)
.align_children_center()
.flex(1., true),
)
.with_child(render_remove(message_id_to_remove, cx, &theme))
.align_children_center(),
) )
.with_child( .with_child(
Label::new( Flex::row()
format_timestamp(message.timestamp, now, self.local_timezone), .with_child(
style.timestamp.text.clone(), text.element(
) theme.editor.syntax.clone(),
.contained() style.body.clone(),
.with_style(style.timestamp.container), theme.editor.document_highlight_read_background,
cx,
)
.flex(1., true),
)
// Add a spacer to make everything line up
.with_child(render_remove(None, cx, &theme)),
) )
.with_children(message_id_to_remove.map(|id| { .contained()
MouseEventHandler::new::<DeleteMessage, _>( .with_style(*container)
id as usize, .with_margin_bottom(if is_last {
cx, theme.chat_panel.last_message_bottom_spacing
|mouse_state, _| { } else {
let button_style = 0.
theme.chat_panel.icon_button.style_for(mouse_state); })
render_icon_button(button_style, "icons/x.svg") .into_any()
.aligned() }
.into_any() })
}, .into_any()
)
.with_padding(Padding::uniform(2.))
.with_cursor_style(CursorStyle::PointingHand)
.on_click(MouseButton::Left, move |_, this, cx| {
this.remove_message(id, cx);
})
.flex_float()
})),
)
.with_child(Text::new(body, style.body.clone()))
.contained()
.with_style(style.container)
.into_any()
} }
fn render_input_box(&self, theme: &Arc<Theme>, cx: &AppContext) -> AnyElement<Self> { fn render_input_box(&self, theme: &Arc<Theme>, cx: &AppContext) -> AnyElement<Self> {
@ -537,6 +650,7 @@ impl ChatPanel {
cx.spawn(|this, mut cx| async move { cx.spawn(|this, mut cx| async move {
let chat = open_chat.await?; let chat = open_chat.await?;
this.update(&mut cx, |this, cx| { this.update(&mut cx, |this, cx| {
this.markdown_data = Default::default();
this.set_active_chat(chat, cx); this.set_active_chat(chat, cx);
}) })
}) })
@ -561,6 +675,72 @@ impl ChatPanel {
} }
} }
fn render_avatar(avatar: Option<Arc<ImageData>>, theme: &Arc<Theme>) -> AnyElement<ChatPanel> {
let avatar_style = theme.chat_panel.avatar;
avatar
.map(|avatar| {
Image::from_data(avatar)
.with_style(avatar_style.image)
.aligned()
.contained()
.with_corner_radius(avatar_style.outer_corner_radius)
.constrained()
.with_width(avatar_style.outer_width)
.with_height(avatar_style.outer_width)
.into_any()
})
.unwrap_or_else(|| {
Empty::new()
.constrained()
.with_width(avatar_style.outer_width)
.into_any()
})
.contained()
.with_style(theme.chat_panel.avatar_container)
.into_any()
}
fn render_remove(
message_id_to_remove: Option<u64>,
cx: &mut ViewContext<'_, '_, ChatPanel>,
theme: &Arc<Theme>,
) -> AnyElement<ChatPanel> {
enum DeleteMessage {}
message_id_to_remove
.map(|id| {
MouseEventHandler::new::<DeleteMessage, _>(id as usize, cx, |mouse_state, _| {
let button_style = theme.chat_panel.icon_button.style_for(mouse_state);
render_icon_button(button_style, "icons/x.svg")
.aligned()
.into_any()
})
.with_padding(Padding::uniform(2.))
.with_cursor_style(CursorStyle::PointingHand)
.on_click(MouseButton::Left, move |_, this, cx| {
this.remove_message(id, cx);
})
.flex_float()
.into_any()
})
.unwrap_or_else(|| {
let style = theme.chat_panel.icon_button.default;
Empty::new()
.constrained()
.with_width(style.icon_width)
.aligned()
.constrained()
.with_width(style.button_width)
.with_height(style.button_width)
.contained()
.with_uniform_padding(2.)
.flex_float()
.into_any()
})
}
impl Entity for ChatPanel { impl Entity for ChatPanel {
type Event = Event; type Event = Event;
} }
@ -627,8 +807,12 @@ impl Panel for ChatPanel {
} }
fn set_active(&mut self, active: bool, cx: &mut ViewContext<Self>) { fn set_active(&mut self, active: bool, cx: &mut ViewContext<Self>) {
if active && !is_chat_feature_enabled(cx) { self.active = active;
cx.emit(Event::Dismissed); if active {
self.acknowledge_last_message(cx);
if !is_chat_feature_enabled(cx) {
cx.emit(Event::Dismissed);
}
} }
} }

View file

@ -95,6 +95,11 @@ pub struct JoinChannelCall {
pub channel_id: u64, pub channel_id: u64,
} }
#[derive(Debug, Clone, PartialEq, Eq, Hash, Serialize, Deserialize)]
pub struct JoinChannelChat {
pub channel_id: u64,
}
#[derive(Debug, Copy, Clone, PartialEq, Eq, Hash, Serialize, Deserialize)] #[derive(Debug, Copy, Clone, PartialEq, Eq, Hash, Serialize, Deserialize)]
struct StartMoveChannelFor { struct StartMoveChannelFor {
channel_id: ChannelId, channel_id: ChannelId,
@ -151,6 +156,7 @@ impl_actions!(
ToggleCollapse, ToggleCollapse,
OpenChannelNotes, OpenChannelNotes,
JoinChannelCall, JoinChannelCall,
JoinChannelChat,
LinkChannel, LinkChannel,
StartMoveChannelFor, StartMoveChannelFor,
StartLinkChannelFor, StartLinkChannelFor,
@ -198,6 +204,7 @@ pub fn init(cx: &mut AppContext) {
cx.add_action(CollabPanel::collapse_selected_channel); cx.add_action(CollabPanel::collapse_selected_channel);
cx.add_action(CollabPanel::expand_selected_channel); cx.add_action(CollabPanel::expand_selected_channel);
cx.add_action(CollabPanel::open_channel_notes); cx.add_action(CollabPanel::open_channel_notes);
cx.add_action(CollabPanel::join_channel_chat);
cx.add_action( cx.add_action(
|panel: &mut CollabPanel, action: &ToggleSelectedIx, cx: &mut ViewContext<CollabPanel>| { |panel: &mut CollabPanel, action: &ToggleSelectedIx, cx: &mut ViewContext<CollabPanel>| {
@ -471,6 +478,12 @@ impl CollabPanel {
.iter() .iter()
.position(|entry| !matches!(entry, ListEntry::Header(_))); .position(|entry| !matches!(entry, ListEntry::Header(_)));
} }
} else if let editor::Event::Blurred = event {
let query = this.filter_editor.read(cx).text(cx);
if query.is_empty() {
this.selection.take();
this.update_entries(true, cx);
}
} }
}) })
.detach(); .detach();
@ -555,7 +568,7 @@ impl CollabPanel {
&*channel, &*channel,
*depth, *depth,
path.to_owned(), path.to_owned(),
&theme.collab_panel, &theme,
is_selected, is_selected,
ix, ix,
cx, cx,
@ -579,6 +592,7 @@ impl CollabPanel {
*channel_id, *channel_id,
&theme.collab_panel, &theme.collab_panel,
is_selected, is_selected,
ix,
cx, cx,
), ),
ListEntry::ChannelInvite(channel) => Self::render_channel_invite( ListEntry::ChannelInvite(channel) => Self::render_channel_invite(
@ -608,7 +622,7 @@ impl CollabPanel {
contact, contact,
*calling, *calling,
&this.project, &this.project,
&theme.collab_panel, &theme,
is_selected, is_selected,
cx, cx,
), ),
@ -768,9 +782,16 @@ impl CollabPanel {
let prev_selected_entry = self.selection.and_then(|ix| self.entries.get(ix).cloned()); let prev_selected_entry = self.selection.and_then(|ix| self.entries.get(ix).cloned());
let old_entries = mem::take(&mut self.entries); let old_entries = mem::take(&mut self.entries);
let mut scroll_to_top = false;
if let Some(room) = ActiveCall::global(cx).read(cx).room() { if let Some(room) = ActiveCall::global(cx).read(cx).room() {
self.entries.push(ListEntry::Header(Section::ActiveCall)); self.entries.push(ListEntry::Header(Section::ActiveCall));
if !old_entries
.iter()
.any(|entry| matches!(entry, ListEntry::Header(Section::ActiveCall)))
{
scroll_to_top = true;
}
if !self.collapsed_sections.contains(&Section::ActiveCall) { if !self.collapsed_sections.contains(&Section::ActiveCall) {
let room = room.read(cx); let room = room.read(cx);
@ -1138,44 +1159,49 @@ impl CollabPanel {
} }
let old_scroll_top = self.list_state.logical_scroll_top(); let old_scroll_top = self.list_state.logical_scroll_top();
self.list_state.reset(self.entries.len()); self.list_state.reset(self.entries.len());
// Attempt to maintain the same scroll position. if scroll_to_top {
if let Some(old_top_entry) = old_entries.get(old_scroll_top.item_ix) { self.list_state.scroll_to(ListOffset::default());
let new_scroll_top = self } else {
.entries // Attempt to maintain the same scroll position.
.iter() if let Some(old_top_entry) = old_entries.get(old_scroll_top.item_ix) {
.position(|entry| entry == old_top_entry) let new_scroll_top = self
.map(|item_ix| ListOffset { .entries
item_ix, .iter()
offset_in_item: old_scroll_top.offset_in_item, .position(|entry| entry == old_top_entry)
}) .map(|item_ix| ListOffset {
.or_else(|| {
let entry_after_old_top = old_entries.get(old_scroll_top.item_ix + 1)?;
let item_ix = self
.entries
.iter()
.position(|entry| entry == entry_after_old_top)?;
Some(ListOffset {
item_ix, item_ix,
offset_in_item: 0., offset_in_item: old_scroll_top.offset_in_item,
}) })
}) .or_else(|| {
.or_else(|| { let entry_after_old_top = old_entries.get(old_scroll_top.item_ix + 1)?;
let entry_before_old_top = let item_ix = self
old_entries.get(old_scroll_top.item_ix.saturating_sub(1))?; .entries
let item_ix = self .iter()
.entries .position(|entry| entry == entry_after_old_top)?;
.iter() Some(ListOffset {
.position(|entry| entry == entry_before_old_top)?; item_ix,
Some(ListOffset { offset_in_item: 0.,
item_ix, })
offset_in_item: 0.,
}) })
}); .or_else(|| {
let entry_before_old_top =
old_entries.get(old_scroll_top.item_ix.saturating_sub(1))?;
let item_ix = self
.entries
.iter()
.position(|entry| entry == entry_before_old_top)?;
Some(ListOffset {
item_ix,
offset_in_item: 0.,
})
});
self.list_state self.list_state
.scroll_to(new_scroll_top.unwrap_or(old_scroll_top)); .scroll_to(new_scroll_top.unwrap_or(old_scroll_top));
}
} }
cx.notify(); cx.notify();
@ -1633,16 +1659,20 @@ impl CollabPanel {
contact: &Contact, contact: &Contact,
calling: bool, calling: bool,
project: &ModelHandle<Project>, project: &ModelHandle<Project>,
theme: &theme::CollabPanel, theme: &theme::Theme,
is_selected: bool, is_selected: bool,
cx: &mut ViewContext<Self>, cx: &mut ViewContext<Self>,
) -> AnyElement<Self> { ) -> AnyElement<Self> {
enum ContactTooltip {}
let collab_theme = &theme.collab_panel;
let online = contact.online; let online = contact.online;
let busy = contact.busy || calling; let busy = contact.busy || calling;
let user_id = contact.user.id; let user_id = contact.user.id;
let github_login = contact.user.github_login.clone(); let github_login = contact.user.github_login.clone();
let initial_project = project.clone(); let initial_project = project.clone();
let mut event_handler =
let event_handler =
MouseEventHandler::new::<Contact, _>(contact.user.id as usize, cx, |state, cx| { MouseEventHandler::new::<Contact, _>(contact.user.id as usize, cx, |state, cx| {
Flex::row() Flex::row()
.with_children(contact.user.avatar.clone().map(|avatar| { .with_children(contact.user.avatar.clone().map(|avatar| {
@ -1652,9 +1682,9 @@ impl CollabPanel {
.collapsed() .collapsed()
.contained() .contained()
.with_style(if busy { .with_style(if busy {
theme.contact_status_busy collab_theme.contact_status_busy
} else { } else {
theme.contact_status_free collab_theme.contact_status_free
}) })
.aligned(), .aligned(),
) )
@ -1664,7 +1694,7 @@ impl CollabPanel {
Stack::new() Stack::new()
.with_child( .with_child(
Image::from_data(avatar) Image::from_data(avatar)
.with_style(theme.contact_avatar) .with_style(collab_theme.contact_avatar)
.aligned() .aligned()
.left(), .left(),
) )
@ -1673,58 +1703,94 @@ impl CollabPanel {
.with_child( .with_child(
Label::new( Label::new(
contact.user.github_login.clone(), contact.user.github_login.clone(),
theme.contact_username.text.clone(), collab_theme.contact_username.text.clone(),
) )
.contained() .contained()
.with_style(theme.contact_username.container) .with_style(collab_theme.contact_username.container)
.aligned() .aligned()
.left() .left()
.flex(1., true), .flex(1., true),
) )
.with_child( .with_children(if state.hovered() {
MouseEventHandler::new::<Cancel, _>( Some(
contact.user.id as usize, MouseEventHandler::new::<Cancel, _>(
cx, contact.user.id as usize,
|mouse_state, _| { cx,
let button_style = theme.contact_button.style_for(mouse_state); |mouse_state, _| {
render_icon_button(button_style, "icons/x.svg") let button_style =
.aligned() collab_theme.contact_button.style_for(mouse_state);
.flex_float() render_icon_button(button_style, "icons/x.svg")
}, .aligned()
.flex_float()
},
)
.with_padding(Padding::uniform(2.))
.with_cursor_style(CursorStyle::PointingHand)
.on_click(MouseButton::Left, move |_, this, cx| {
this.remove_contact(user_id, &github_login, cx);
})
.flex_float(),
) )
.with_padding(Padding::uniform(2.)) } else {
.with_cursor_style(CursorStyle::PointingHand) None
.on_click(MouseButton::Left, move |_, this, cx| { })
this.remove_contact(user_id, &github_login, cx);
})
.flex_float(),
)
.with_children(if calling { .with_children(if calling {
Some( Some(
Label::new("Calling", theme.calling_indicator.text.clone()) Label::new("Calling", collab_theme.calling_indicator.text.clone())
.contained() .contained()
.with_style(theme.calling_indicator.container) .with_style(collab_theme.calling_indicator.container)
.aligned(), .aligned(),
) )
} else { } else {
None None
}) })
.constrained() .constrained()
.with_height(theme.row_height) .with_height(collab_theme.row_height)
.contained() .contained()
.with_style(*theme.contact_row.in_state(is_selected).style_for(state)) .with_style(
}) *collab_theme
.on_click(MouseButton::Left, move |_, this, cx| { .contact_row
if online && !busy { .in_state(is_selected)
this.call(user_id, Some(initial_project.clone()), cx); .style_for(state),
} )
}); });
if online { if online && !busy {
event_handler = event_handler.with_cursor_style(CursorStyle::PointingHand); let room = ActiveCall::global(cx).read(cx).room();
} let label = if room.is_some() {
format!("Invite {} to join call", contact.user.github_login)
} else {
format!("Call {}", contact.user.github_login)
};
event_handler.into_any() event_handler
.on_click(MouseButton::Left, move |_, this, cx| {
this.call(user_id, Some(initial_project.clone()), cx);
})
.with_cursor_style(CursorStyle::PointingHand)
.with_tooltip::<ContactTooltip>(
contact.user.id as usize,
label,
None,
theme.tooltip.clone(),
cx,
)
.into_any()
} else {
event_handler
.with_tooltip::<ContactTooltip>(
contact.user.id as usize,
format!(
"{} is {}",
contact.user.github_login,
if busy { "on a call" } else { "offline" }
),
None,
theme.tooltip.clone(),
cx,
)
.into_any()
}
} }
fn render_contact_placeholder( fn render_contact_placeholder(
@ -1827,12 +1893,13 @@ impl CollabPanel {
channel: &Channel, channel: &Channel,
depth: usize, depth: usize,
path: ChannelPath, path: ChannelPath,
theme: &theme::CollabPanel, theme: &theme::Theme,
is_selected: bool, is_selected: bool,
ix: usize, ix: usize,
cx: &mut ViewContext<Self>, cx: &mut ViewContext<Self>,
) -> AnyElement<Self> { ) -> AnyElement<Self> {
let channel_id = channel.id; let channel_id = channel.id;
let collab_theme = &theme.collab_panel;
let has_children = self.channel_store.read(cx).has_children(channel_id); let has_children = self.channel_store.read(cx).has_children(channel_id);
let other_selected = let other_selected =
self.selected_channel().map(|channel| channel.0.id) == Some(channel.id); self.selected_channel().map(|channel| channel.0.id) == Some(channel.id);
@ -1851,6 +1918,10 @@ impl CollabPanel {
const FACEPILE_LIMIT: usize = 3; const FACEPILE_LIMIT: usize = 3;
enum ChannelCall {} enum ChannelCall {}
enum ChannelNote {}
enum NotesTooltip {}
enum ChatTooltip {}
enum ChannelTooltip {}
let mut is_dragged_over = false; let mut is_dragged_over = false;
if cx if cx
@ -1866,6 +1937,8 @@ impl CollabPanel {
is_dragged_over = true; is_dragged_over = true;
} }
let has_messages_notification = channel.unseen_message_id.is_some();
MouseEventHandler::new::<Channel, _>(ix, cx, |state, cx| { MouseEventHandler::new::<Channel, _>(ix, cx, |state, cx| {
let row_hovered = state.hovered(); let row_hovered = state.hovered();
@ -1886,60 +1959,151 @@ impl CollabPanel {
Flex::<Self>::row() Flex::<Self>::row()
.with_child( .with_child(
Svg::new("icons/hash.svg") Svg::new("icons/hash.svg")
.with_color(theme.channel_hash.color) .with_color(collab_theme.channel_hash.color)
.constrained() .constrained()
.with_width(theme.channel_hash.width) .with_width(collab_theme.channel_hash.width)
.aligned() .aligned()
.left(), .left(),
) )
.with_child( .with_child({
Label::new(channel.name.clone(), theme.channel_name.text.clone()) let style = collab_theme.channel_name.inactive_state();
.contained() Flex::row()
.with_style(theme.channel_name.container) .with_child(
.aligned() Label::new(channel.name.clone(), style.text.clone())
.left() .contained()
.flex(1., true), .with_style(style.container)
) .aligned()
.with_child( .left()
MouseEventHandler::new::<ChannelCall, _>(ix, cx, move |_, cx| { .with_tooltip::<ChannelTooltip>(
let participants = ix,
self.channel_store.read(cx).channel_participants(channel_id); "Join channel",
if !participants.is_empty() { None,
let extra_count = participants.len().saturating_sub(FACEPILE_LIMIT); theme.tooltip.clone(),
cx,
),
)
.with_children({
let participants =
self.channel_store.read(cx).channel_participants(channel_id);
FacePile::new(theme.face_overlap) if !participants.is_empty() {
.with_children( let extra_count = participants.len().saturating_sub(FACEPILE_LIMIT);
participants
.iter() let result = FacePile::new(collab_theme.face_overlap)
.filter_map(|user| { .with_children(
Some( participants
Image::from_data(user.avatar.clone()?) .iter()
.with_style(theme.channel_avatar), .filter_map(|user| {
) Some(
}) Image::from_data(user.avatar.clone()?)
.take(FACEPILE_LIMIT), .with_style(collab_theme.channel_avatar),
) )
.with_children((extra_count > 0).then(|| { })
Label::new( .take(FACEPILE_LIMIT),
format!("+{}", extra_count),
theme.extra_participant_label.text.clone(),
) )
.contained() .with_children((extra_count > 0).then(|| {
.with_style(theme.extra_participant_label.container) Label::new(
})) format!("+{}", extra_count),
collab_theme.extra_participant_label.text.clone(),
)
.contained()
.with_style(collab_theme.extra_participant_label.container)
}));
Some(result)
} else {
None
}
})
.with_spacing(8.)
.align_children_center()
.flex(1., true)
})
.with_child(
MouseEventHandler::new::<ChannelNote, _>(ix, cx, move |mouse_state, _| {
let container_style = collab_theme
.disclosure
.button
.style_for(mouse_state)
.container;
if channel.unseen_message_id.is_some() {
Svg::new("icons/conversations.svg")
.with_color(collab_theme.channel_note_active_color)
.constrained()
.with_width(collab_theme.channel_hash.width)
.contained()
.with_style(container_style)
.with_uniform_padding(4.)
.into_any() .into_any()
} else if row_hovered { } else if row_hovered {
Svg::new("icons/speaker-loud.svg") Svg::new("icons/conversations.svg")
.with_color(theme.channel_hash.color) .with_color(collab_theme.channel_hash.color)
.constrained() .constrained()
.with_width(theme.channel_hash.width) .with_width(collab_theme.channel_hash.width)
.contained()
.with_style(container_style)
.with_uniform_padding(4.)
.into_any() .into_any()
} else { } else {
Empty::new().into_any() Empty::new().into_any()
} }
}) })
.on_click(MouseButton::Left, move |_, this, cx| { .on_click(MouseButton::Left, move |_, this, cx| {
this.join_channel_call(channel_id, cx); this.join_channel_chat(&JoinChannelChat { channel_id }, cx);
})
.with_tooltip::<ChatTooltip>(
ix,
"Open channel chat",
None,
theme.tooltip.clone(),
cx,
)
.contained()
.with_margin_right(4.),
)
.with_child(
MouseEventHandler::new::<ChannelCall, _>(ix, cx, move |mouse_state, cx| {
let container_style = collab_theme
.disclosure
.button
.style_for(mouse_state)
.container;
if row_hovered || channel.unseen_note_version.is_some() {
Svg::new("icons/file.svg")
.with_color(if channel.unseen_note_version.is_some() {
collab_theme.channel_note_active_color
} else {
collab_theme.channel_hash.color
})
.constrained()
.with_width(collab_theme.channel_hash.width)
.contained()
.with_style(container_style)
.with_uniform_padding(4.)
.with_margin_right(collab_theme.channel_hash.container.margin.left)
.with_tooltip::<NotesTooltip>(
ix as usize,
"Open channel notes",
None,
theme.tooltip.clone(),
cx,
)
.into_any()
} else if has_messages_notification {
Empty::new()
.constrained()
.with_width(collab_theme.channel_hash.width)
.contained()
.with_uniform_padding(4.)
.with_margin_right(collab_theme.channel_hash.container.margin.left)
.into_any()
} else {
Empty::new().into_any()
}
})
.on_click(MouseButton::Left, move |_, this, cx| {
this.open_channel_notes(&OpenChannelNotes { channel_id }, cx);
}), }),
) )
.align_children_center() .align_children_center()
@ -1951,24 +2115,28 @@ impl CollabPanel {
}), }),
) )
.with_id(ix) .with_id(ix)
.with_style(theme.disclosure.clone()) .with_style(collab_theme.disclosure.clone())
.element() .element()
.constrained() .constrained()
.with_height(theme.row_height) .with_height(collab_theme.row_height)
.contained() .contained()
.with_style(select_state( .with_style(select_state(
theme collab_theme
.channel_row .channel_row
.in_state(is_selected || is_active || is_dragged_over), .in_state(is_selected || is_active || is_dragged_over),
)) ))
.with_padding_left( .with_padding_left(
theme.channel_row.default_style().padding.left collab_theme.channel_row.default_style().padding.left
+ theme.channel_indent * depth as f32, + collab_theme.channel_indent * depth as f32,
) )
}) })
.on_click(MouseButton::Left, move |_, this, cx| { .on_click(MouseButton::Left, move |_, this, cx| {
if this.drag_target_channel.take().is_none() { if this.drag_target_channel.take().is_none() {
this.join_channel_chat(channel_id, cx); if is_active {
this.open_channel_notes(&OpenChannelNotes { channel_id }, cx)
} else {
this.join_channel(channel_id, cx)
}
} }
}) })
.on_click(MouseButton::Right, { .on_click(MouseButton::Right, {
@ -2082,6 +2250,7 @@ impl CollabPanel {
channel_id: ChannelId, channel_id: ChannelId,
theme: &theme::CollabPanel, theme: &theme::CollabPanel,
is_selected: bool, is_selected: bool,
ix: usize,
cx: &mut ViewContext<Self>, cx: &mut ViewContext<Self>,
) -> AnyElement<Self> { ) -> AnyElement<Self> {
enum ChannelNotes {} enum ChannelNotes {}
@ -2091,7 +2260,7 @@ impl CollabPanel {
.or(theme.contact_avatar.height) .or(theme.contact_avatar.height)
.unwrap_or(0.); .unwrap_or(0.);
MouseEventHandler::new::<ChannelNotes, _>(channel_id as usize, cx, |state, cx| { MouseEventHandler::new::<ChannelNotes, _>(ix as usize, cx, |state, cx| {
let tree_branch = *theme.tree_branch.in_state(is_selected).style_for(state); let tree_branch = *theme.tree_branch.in_state(is_selected).style_for(state);
let row = theme.project_row.in_state(is_selected).style_for(state); let row = theme.project_row.in_state(is_selected).style_for(state);
@ -2392,6 +2561,13 @@ impl CollabPanel {
}, },
)); ));
items.push(ContextMenuItem::action(
"Open Chat",
JoinChannelChat {
channel_id: path.channel_id(),
},
));
if self.channel_store.read(cx).is_user_admin(path.channel_id()) { if self.channel_store.read(cx).is_user_admin(path.channel_id()) {
let parent_id = path.parent_id(); let parent_id = path.parent_id();
@ -2588,7 +2764,26 @@ impl CollabPanel {
} }
} }
ListEntry::Channel { channel, .. } => { ListEntry::Channel { channel, .. } => {
self.join_channel_chat(channel.id, cx); let is_active = iife!({
let call_channel = ActiveCall::global(cx)
.read(cx)
.room()?
.read(cx)
.channel_id()?;
Some(call_channel == channel.id)
})
.unwrap_or(false);
if is_active {
self.open_channel_notes(
&OpenChannelNotes {
channel_id: channel.id,
},
cx,
)
} else {
self.join_channel(channel.id, cx)
}
} }
ListEntry::ContactPlaceholder => self.toggle_contact_finder(cx), ListEntry::ContactPlaceholder => self.toggle_contact_finder(cx),
_ => {} _ => {}
@ -2991,13 +3186,54 @@ impl CollabPanel {
.detach_and_log_err(cx); .detach_and_log_err(cx);
} }
fn join_channel_call(&self, channel: u64, cx: &mut ViewContext<Self>) { fn join_channel(&self, channel_id: u64, cx: &mut ViewContext<Self>) {
ActiveCall::global(cx) let workspace = self.workspace.clone();
.update(cx, |call, cx| call.join_channel(channel, cx)) let window = cx.window();
.detach_and_log_err(cx); let active_call = ActiveCall::global(cx);
cx.spawn(|_, mut cx| async move {
if active_call.read_with(&mut cx, |active_call, cx| {
if let Some(room) = active_call.room() {
let room = room.read(cx);
room.is_sharing_project() && room.remote_participants().len() > 0
} else {
false
}
}) {
let answer = window.prompt(
PromptLevel::Warning,
"Leaving this call will unshare your current project.\nDo you want to switch channels?",
&["Yes, Join Channel", "Cancel"],
&mut cx,
);
if let Some(mut answer) = answer {
if answer.next().await == Some(1) {
return anyhow::Ok(());
}
}
}
let room = active_call
.update(&mut cx, |call, cx| call.join_channel(channel_id, cx))
.await?;
let task = room.update(&mut cx, |room, cx| {
let workspace = workspace.upgrade(cx)?;
let (project, host) = room.most_active_project()?;
let app_state = workspace.read(cx).app_state().clone();
Some(workspace::join_remote_project(project, host, app_state, cx))
});
if let Some(task) = task {
task.await?;
}
anyhow::Ok(())
})
.detach_and_log_err(cx);
} }
fn join_channel_chat(&mut self, channel_id: u64, cx: &mut ViewContext<Self>) { fn join_channel_chat(&mut self, action: &JoinChannelChat, cx: &mut ViewContext<Self>) {
let channel_id = action.channel_id;
if let Some(workspace) = self.workspace.upgrade(cx) { if let Some(workspace) = self.workspace.upgrade(cx) {
cx.app_context().defer(move |cx| { cx.app_context().defer(move |cx| {
workspace.update(cx, |workspace, cx| { workspace.update(cx, |workspace, cx| {

View file

@ -36,6 +36,7 @@ language = { path = "../language" }
lsp = { path = "../lsp" } lsp = { path = "../lsp" }
project = { path = "../project" } project = { path = "../project" }
rpc = { path = "../rpc" } rpc = { path = "../rpc" }
rich_text = { path = "../rich_text" }
settings = { path = "../settings" } settings = { path = "../settings" }
snippet = { path = "../snippet" } snippet = { path = "../snippet" }
sum_tree = { path = "../sum_tree" } sum_tree = { path = "../sum_tree" }

View file

@ -2454,7 +2454,13 @@ impl Editor {
let snapshot = this.buffer.read(cx).read(cx); let snapshot = this.buffer.read(cx).read(cx);
let new_selections = resolve_multiple::<usize, _>(new_anchor_selections, &snapshot) let new_selections = resolve_multiple::<usize, _>(new_anchor_selections, &snapshot)
.zip(new_selection_deltas) .zip(new_selection_deltas)
.map(|(selection, delta)| selection.map(|e| e + delta)) .map(|(selection, delta)| Selection {
id: selection.id,
start: selection.start + delta,
end: selection.end + delta,
reversed: selection.reversed,
goal: SelectionGoal::None,
})
.collect::<Vec<_>>(); .collect::<Vec<_>>();
let mut i = 0; let mut i = 0;

View file

@ -8,12 +8,12 @@ use futures::FutureExt;
use gpui::{ use gpui::{
actions, actions,
elements::{Flex, MouseEventHandler, Padding, ParentElement, Text}, elements::{Flex, MouseEventHandler, Padding, ParentElement, Text},
fonts::{HighlightStyle, Underline, Weight},
platform::{CursorStyle, MouseButton}, platform::{CursorStyle, MouseButton},
AnyElement, AppContext, CursorRegion, Element, ModelHandle, MouseRegion, Task, ViewContext, AnyElement, AppContext, Element, ModelHandle, Task, ViewContext,
}; };
use language::{Bias, DiagnosticEntry, DiagnosticSeverity, Language, LanguageRegistry}; use language::{Bias, DiagnosticEntry, DiagnosticSeverity, Language, LanguageRegistry};
use project::{HoverBlock, HoverBlockKind, InlayHintLabelPart, Project}; use project::{HoverBlock, HoverBlockKind, InlayHintLabelPart, Project};
use rich_text::{new_paragraph, render_code, render_markdown_mut, RichText};
use std::{ops::Range, sync::Arc, time::Duration}; use std::{ops::Range, sync::Arc, time::Duration};
use util::TryFutureExt; use util::TryFutureExt;
@ -346,158 +346,25 @@ fn show_hover(
} }
fn render_blocks( fn render_blocks(
theme_id: usize,
blocks: &[HoverBlock], blocks: &[HoverBlock],
language_registry: &Arc<LanguageRegistry>, language_registry: &Arc<LanguageRegistry>,
language: Option<&Arc<Language>>, language: Option<&Arc<Language>>,
style: &EditorStyle, ) -> RichText {
) -> RenderedInfo { let mut data = RichText {
let mut text = String::new(); text: Default::default(),
let mut highlights = Vec::new(); highlights: Default::default(),
let mut region_ranges = Vec::new(); region_ranges: Default::default(),
let mut regions = Vec::new(); regions: Default::default(),
};
for block in blocks { for block in blocks {
match &block.kind { match &block.kind {
HoverBlockKind::PlainText => { HoverBlockKind::PlainText => {
new_paragraph(&mut text, &mut Vec::new()); new_paragraph(&mut data.text, &mut Vec::new());
text.push_str(&block.text); data.text.push_str(&block.text);
} }
HoverBlockKind::Markdown => { HoverBlockKind::Markdown => {
use pulldown_cmark::{CodeBlockKind, Event, Options, Parser, Tag}; render_markdown_mut(&block.text, language_registry, language, &mut data)
let mut bold_depth = 0;
let mut italic_depth = 0;
let mut link_url = None;
let mut current_language = None;
let mut list_stack = Vec::new();
for event in Parser::new_ext(&block.text, Options::all()) {
let prev_len = text.len();
match event {
Event::Text(t) => {
if let Some(language) = &current_language {
render_code(
&mut text,
&mut highlights,
t.as_ref(),
language,
style,
);
} else {
text.push_str(t.as_ref());
let mut style = HighlightStyle::default();
if bold_depth > 0 {
style.weight = Some(Weight::BOLD);
}
if italic_depth > 0 {
style.italic = Some(true);
}
if let Some(link_url) = link_url.clone() {
region_ranges.push(prev_len..text.len());
regions.push(RenderedRegion {
link_url: Some(link_url),
code: false,
});
style.underline = Some(Underline {
thickness: 1.0.into(),
..Default::default()
});
}
if style != HighlightStyle::default() {
let mut new_highlight = true;
if let Some((last_range, last_style)) = highlights.last_mut() {
if last_range.end == prev_len && last_style == &style {
last_range.end = text.len();
new_highlight = false;
}
}
if new_highlight {
highlights.push((prev_len..text.len(), style));
}
}
}
}
Event::Code(t) => {
text.push_str(t.as_ref());
region_ranges.push(prev_len..text.len());
if link_url.is_some() {
highlights.push((
prev_len..text.len(),
HighlightStyle {
underline: Some(Underline {
thickness: 1.0.into(),
..Default::default()
}),
..Default::default()
},
));
}
regions.push(RenderedRegion {
code: true,
link_url: link_url.clone(),
});
}
Event::Start(tag) => match tag {
Tag::Paragraph => new_paragraph(&mut text, &mut list_stack),
Tag::Heading(_, _, _) => {
new_paragraph(&mut text, &mut list_stack);
bold_depth += 1;
}
Tag::CodeBlock(kind) => {
new_paragraph(&mut text, &mut list_stack);
current_language = if let CodeBlockKind::Fenced(language) = kind {
language_registry
.language_for_name(language.as_ref())
.now_or_never()
.and_then(Result::ok)
} else {
language.cloned()
}
}
Tag::Emphasis => italic_depth += 1,
Tag::Strong => bold_depth += 1,
Tag::Link(_, url, _) => link_url = Some(url.to_string()),
Tag::List(number) => {
list_stack.push((number, false));
}
Tag::Item => {
let len = list_stack.len();
if let Some((list_number, has_content)) = list_stack.last_mut() {
*has_content = false;
if !text.is_empty() && !text.ends_with('\n') {
text.push('\n');
}
for _ in 0..len - 1 {
text.push_str(" ");
}
if let Some(number) = list_number {
text.push_str(&format!("{}. ", number));
*number += 1;
*has_content = false;
} else {
text.push_str("- ");
}
}
}
_ => {}
},
Event::End(tag) => match tag {
Tag::Heading(_, _, _) => bold_depth -= 1,
Tag::CodeBlock(_) => current_language = None,
Tag::Emphasis => italic_depth -= 1,
Tag::Strong => bold_depth -= 1,
Tag::Link(_, _, _) => link_url = None,
Tag::List(_) => drop(list_stack.pop()),
_ => {}
},
Event::HardBreak => text.push('\n'),
Event::SoftBreak => text.push(' '),
_ => {}
}
}
} }
HoverBlockKind::Code { language } => { HoverBlockKind::Code { language } => {
if let Some(language) = language_registry if let Some(language) = language_registry
@ -505,62 +372,17 @@ fn render_blocks(
.now_or_never() .now_or_never()
.and_then(Result::ok) .and_then(Result::ok)
{ {
render_code(&mut text, &mut highlights, &block.text, &language, style); render_code(&mut data.text, &mut data.highlights, &block.text, &language);
} else { } else {
text.push_str(&block.text); data.text.push_str(&block.text);
} }
} }
} }
} }
RenderedInfo { data.text = data.text.trim().to_string();
theme_id,
text: text.trim().to_string(),
highlights,
region_ranges,
regions,
}
}
fn render_code( data
text: &mut String,
highlights: &mut Vec<(Range<usize>, HighlightStyle)>,
content: &str,
language: &Arc<Language>,
style: &EditorStyle,
) {
let prev_len = text.len();
text.push_str(content);
for (range, highlight_id) in language.highlight_text(&content.into(), 0..content.len()) {
if let Some(style) = highlight_id.style(&style.syntax) {
highlights.push((prev_len + range.start..prev_len + range.end, style));
}
}
}
fn new_paragraph(text: &mut String, list_stack: &mut Vec<(Option<u64>, bool)>) {
let mut is_subsequent_paragraph_of_list = false;
if let Some((_, has_content)) = list_stack.last_mut() {
if *has_content {
is_subsequent_paragraph_of_list = true;
} else {
*has_content = true;
return;
}
}
if !text.is_empty() {
if !text.ends_with('\n') {
text.push('\n');
}
text.push('\n');
}
for _ in 0..list_stack.len().saturating_sub(1) {
text.push_str(" ");
}
if is_subsequent_paragraph_of_list {
text.push_str(" ");
}
} }
#[derive(Default)] #[derive(Default)]
@ -623,22 +445,7 @@ pub struct InfoPopover {
symbol_range: RangeInEditor, symbol_range: RangeInEditor,
pub blocks: Vec<HoverBlock>, pub blocks: Vec<HoverBlock>,
language: Option<Arc<Language>>, language: Option<Arc<Language>>,
rendered_content: Option<RenderedInfo>, rendered_content: Option<RichText>,
}
#[derive(Debug, Clone)]
struct RenderedInfo {
theme_id: usize,
text: String,
highlights: Vec<(Range<usize>, HighlightStyle)>,
region_ranges: Vec<Range<usize>>,
regions: Vec<RenderedRegion>,
}
#[derive(Debug, Clone)]
struct RenderedRegion {
code: bool,
link_url: Option<String>,
} }
impl InfoPopover { impl InfoPopover {
@ -647,63 +454,24 @@ impl InfoPopover {
style: &EditorStyle, style: &EditorStyle,
cx: &mut ViewContext<Editor>, cx: &mut ViewContext<Editor>,
) -> AnyElement<Editor> { ) -> AnyElement<Editor> {
if let Some(rendered) = &self.rendered_content {
if rendered.theme_id != style.theme_id {
self.rendered_content = None;
}
}
let rendered_content = self.rendered_content.get_or_insert_with(|| { let rendered_content = self.rendered_content.get_or_insert_with(|| {
render_blocks( render_blocks(
style.theme_id,
&self.blocks, &self.blocks,
self.project.read(cx).languages(), self.project.read(cx).languages(),
self.language.as_ref(), self.language.as_ref(),
style,
) )
}); });
MouseEventHandler::new::<InfoPopover, _>(0, cx, |_, cx| { MouseEventHandler::new::<InfoPopover, _>(0, cx, move |_, cx| {
let mut region_id = 0;
let view_id = cx.view_id();
let code_span_background_color = style.document_highlight_read_background; let code_span_background_color = style.document_highlight_read_background;
let regions = rendered_content.regions.clone();
Flex::column() Flex::column()
.scrollable::<HoverBlock>(1, None, cx) .scrollable::<HoverBlock>(1, None, cx)
.with_child( .with_child(rendered_content.element(
Text::new(rendered_content.text.clone(), style.text.clone()) style.syntax.clone(),
.with_highlights(rendered_content.highlights.clone()) style.text.clone(),
.with_custom_runs( code_span_background_color,
rendered_content.region_ranges.clone(), cx,
move |ix, bounds, cx| { ))
region_id += 1;
let region = regions[ix].clone();
if let Some(url) = region.link_url {
cx.scene().push_cursor_region(CursorRegion {
bounds,
style: CursorStyle::PointingHand,
});
cx.scene().push_mouse_region(
MouseRegion::new::<Self>(view_id, region_id, bounds)
.on_click::<Editor, _>(
MouseButton::Left,
move |_, _, cx| cx.platform().open_url(&url),
),
);
}
if region.code {
cx.scene().push_quad(gpui::Quad {
bounds,
background: Some(code_span_background_color),
border: Default::default(),
corner_radii: (2.0).into(),
});
}
},
)
.with_soft_wrap(true),
)
.contained() .contained()
.with_style(style.hover_popover.container) .with_style(style.hover_popover.container)
}) })
@ -799,11 +567,12 @@ mod tests {
InlayId, InlayId,
}; };
use collections::BTreeSet; use collections::BTreeSet;
use gpui::fonts::Weight; use gpui::fonts::{HighlightStyle, Underline, Weight};
use indoc::indoc; use indoc::indoc;
use language::{language_settings::InlayHintSettings, Diagnostic, DiagnosticSet}; use language::{language_settings::InlayHintSettings, Diagnostic, DiagnosticSet};
use lsp::LanguageServerId; use lsp::LanguageServerId;
use project::{HoverBlock, HoverBlockKind}; use project::{HoverBlock, HoverBlockKind};
use rich_text::Highlight;
use smol::stream::StreamExt; use smol::stream::StreamExt;
use unindent::Unindent; use unindent::Unindent;
use util::test::marked_text_ranges; use util::test::marked_text_ranges;
@ -1014,7 +783,7 @@ mod tests {
.await; .await;
cx.condition(|editor, _| editor.hover_state.visible()).await; cx.condition(|editor, _| editor.hover_state.visible()).await;
cx.editor(|editor, cx| { cx.editor(|editor, _| {
let blocks = editor.hover_state.info_popover.clone().unwrap().blocks; let blocks = editor.hover_state.info_popover.clone().unwrap().blocks;
assert_eq!( assert_eq!(
blocks, blocks,
@ -1024,8 +793,7 @@ mod tests {
}], }],
); );
let style = editor.style(cx); let rendered = render_blocks(&blocks, &Default::default(), None);
let rendered = render_blocks(0, &blocks, &Default::default(), None, &style);
assert_eq!( assert_eq!(
rendered.text, rendered.text,
code_str.trim(), code_str.trim(),
@ -1217,7 +985,7 @@ mod tests {
expected_styles, expected_styles,
} in &rows[0..] } in &rows[0..]
{ {
let rendered = render_blocks(0, &blocks, &Default::default(), None, &style); let rendered = render_blocks(&blocks, &Default::default(), None);
let (expected_text, ranges) = marked_text_ranges(expected_marked_text, false); let (expected_text, ranges) = marked_text_ranges(expected_marked_text, false);
let expected_highlights = ranges let expected_highlights = ranges
@ -1228,8 +996,21 @@ mod tests {
rendered.text, expected_text, rendered.text, expected_text,
"wrong text for input {blocks:?}" "wrong text for input {blocks:?}"
); );
let rendered_highlights: Vec<_> = rendered
.highlights
.iter()
.filter_map(|(range, highlight)| {
let style = match highlight {
Highlight::Id(id) => id.style(&style.syntax)?,
Highlight::Highlight(style) => style.clone(),
};
Some((range.clone(), style))
})
.collect();
assert_eq!( assert_eq!(
rendered.highlights, expected_highlights, rendered_highlights, expected_highlights,
"wrong highlights for input {blocks:?}" "wrong highlights for input {blocks:?}"
); );
} }

View file

@ -53,7 +53,7 @@ thiserror.workspace = true
time.workspace = true time.workspace = true
tiny-skia = "0.5" tiny-skia = "0.5"
usvg = { version = "0.14", features = [] } usvg = { version = "0.14", features = [] }
uuid = { version = "1.1.2", features = ["v4"] } uuid.workspace = true
waker-fn = "1.1.0" waker-fn = "1.1.0"
[build-dependencies] [build-dependencies]

View file

@ -1252,7 +1252,7 @@ impl AppContext {
result result
}) })
} else { } else {
panic!("circular model update"); panic!("circular model update for {}", std::any::type_name::<T>());
} }
} }

View file

@ -53,8 +53,10 @@ impl Select {
} }
pub fn set_item_count(&mut self, count: usize, cx: &mut ViewContext<Self>) { pub fn set_item_count(&mut self, count: usize, cx: &mut ViewContext<Self>) {
self.item_count = count; if count != self.item_count {
cx.notify(); self.item_count = count;
cx.notify();
}
} }
fn toggle(&mut self, cx: &mut ViewContext<Self>) { fn toggle(&mut self, cx: &mut ViewContext<Self>) {
@ -63,9 +65,11 @@ impl Select {
} }
pub fn set_selected_index(&mut self, ix: usize, cx: &mut ViewContext<Self>) { pub fn set_selected_index(&mut self, ix: usize, cx: &mut ViewContext<Self>) {
self.selected_item_ix = ix; if ix != self.selected_item_ix || self.is_open {
self.is_open = false; self.selected_item_ix = ix;
cx.notify(); self.is_open = false;
cx.notify();
}
} }
pub fn selected_index(&self) -> usize { pub fn selected_index(&self) -> usize {

View file

@ -660,12 +660,12 @@ impl Buffer {
file_changed = true; file_changed = true;
}; };
self.file = Some(new_file);
if file_changed { if file_changed {
self.file_update_count += 1; self.file_update_count += 1;
cx.emit(Event::FileHandleChanged); cx.emit(Event::FileHandleChanged);
cx.notify(); cx.notify();
} }
self.file = Some(new_file);
task task
} }

View file

@ -1427,7 +1427,7 @@ fn test_autoindent_block_mode_without_original_indent_columns(cx: &mut AppContex
// Insert the block at column zero. The entire block is indented // Insert the block at column zero. The entire block is indented
// so that the first line matches the previous line's indentation. // so that the first line matches the previous line's indentation.
buffer.edit( buffer.edit(
[(Point::new(2, 0)..Point::new(2, 0), inserted_text.clone())], [(Point::new(2, 0)..Point::new(2, 0), inserted_text)],
Some(AutoindentMode::Block { Some(AutoindentMode::Block {
original_indent_columns: original_indent_columns.clone(), original_indent_columns: original_indent_columns.clone(),
}), }),

View file

@ -1841,6 +1841,7 @@ impl Project {
Worktree::Remote(_) => panic!("cannot remote buffers as new files"), Worktree::Remote(_) => panic!("cannot remote buffers as new files"),
}) })
.await?; .await?;
this.update(&mut cx, |this, cx| { this.update(&mut cx, |this, cx| {
this.detect_language_for_buffer(&buffer, cx); this.detect_language_for_buffer(&buffer, cx);
this.register_buffer_with_language_servers(&buffer, cx); this.register_buffer_with_language_servers(&buffer, cx);
@ -2368,7 +2369,30 @@ impl Project {
} }
} }
} }
BufferEvent::FileHandleChanged => {
let Some(file) = File::from_dyn(buffer.read(cx).file()) else {
return None;
};
match self.local_buffer_ids_by_entry_id.get(&file.entry_id) {
Some(_) => {
return None;
}
None => {
let remote_id = buffer.read(cx).remote_id();
self.local_buffer_ids_by_entry_id
.insert(file.entry_id, remote_id);
self.local_buffer_ids_by_path.insert(
ProjectPath {
worktree_id: file.worktree_id(cx),
path: file.path.clone(),
},
remote_id,
);
}
}
}
_ => {} _ => {}
} }
@ -5906,7 +5930,9 @@ impl Project {
Some(&buffer_id) => buffer_id, Some(&buffer_id) => buffer_id,
None => match self.local_buffer_ids_by_path.get(&project_path) { None => match self.local_buffer_ids_by_path.get(&project_path) {
Some(&buffer_id) => buffer_id, Some(&buffer_id) => buffer_id,
None => continue, None => {
continue;
}
}, },
}; };

View file

@ -1737,7 +1737,7 @@ mod tests {
use settings::SettingsStore; use settings::SettingsStore;
use std::{ use std::{
collections::HashSet, collections::HashSet,
path::Path, path::{Path, PathBuf},
sync::atomic::{self, AtomicUsize}, sync::atomic::{self, AtomicUsize},
}; };
use workspace::{pane, AppState}; use workspace::{pane, AppState};
@ -2759,6 +2759,71 @@ mod tests {
); );
} }
#[gpui::test]
async fn test_new_file_move(cx: &mut gpui::TestAppContext) {
init_test(cx);
let fs = FakeFs::new(cx.background());
fs.as_fake().insert_tree("/root", json!({})).await;
let project = Project::test(fs, ["/root".as_ref()], cx).await;
let workspace = cx
.add_window(|cx| Workspace::test_new(project.clone(), cx))
.root(cx);
let panel = workspace.update(cx, |workspace, cx| ProjectPanel::new(workspace, cx));
// Make a new buffer with no backing file
workspace.update(cx, |workspace, cx| {
Editor::new_file(workspace, &Default::default(), cx)
});
// "Save as"" the buffer, creating a new backing file for it
let task = workspace.update(cx, |workspace, cx| {
workspace.save_active_item(workspace::SaveIntent::Save, cx)
});
cx.foreground().run_until_parked();
cx.simulate_new_path_selection(|_| Some(PathBuf::from("/root/new")));
task.await.unwrap();
// Rename the file
select_path(&panel, "root/new", cx);
assert_eq!(
visible_entries_as_strings(&panel, 0..10, cx),
&["v root", " new <== selected"]
);
panel.update(cx, |panel, cx| panel.rename(&Rename, cx));
panel.update(cx, |panel, cx| {
panel
.filename_editor
.update(cx, |editor, cx| editor.set_text("newer", cx));
});
panel
.update(cx, |panel, cx| panel.confirm(&Confirm, cx))
.unwrap()
.await
.unwrap();
cx.foreground().run_until_parked();
assert_eq!(
visible_entries_as_strings(&panel, 0..10, cx),
&["v root", " newer <== selected"]
);
workspace
.update(cx, |workspace, cx| {
workspace.save_active_item(workspace::SaveIntent::Save, cx)
})
.await
.unwrap();
cx.foreground().run_until_parked();
// assert that saving the file doesn't restore "new"
assert_eq!(
visible_entries_as_strings(&panel, 0..10, cx),
&["v root", " newer <== selected"]
);
}
fn toggle_expand_dir( fn toggle_expand_dir(
panel: &ViewHandle<ProjectPanel>, panel: &ViewHandle<ProjectPanel>,
path: impl AsRef<Path>, path: impl AsRef<Path>,
@ -2862,6 +2927,7 @@ mod tests {
editor::init_settings(cx); editor::init_settings(cx);
crate::init((), cx); crate::init((), cx);
workspace::init_settings(cx); workspace::init_settings(cx);
client::init_settings(cx);
Project::init_settings(cx); Project::init_settings(cx);
}); });
} }

View file

@ -0,0 +1,30 @@
[package]
name = "rich_text"
version = "0.1.0"
edition = "2021"
publish = false
[lib]
path = "src/rich_text.rs"
doctest = false
[features]
test-support = [
"gpui/test-support",
"util/test-support",
]
[dependencies]
collections = { path = "../collections" }
gpui = { path = "../gpui" }
sum_tree = { path = "../sum_tree" }
theme = { path = "../theme" }
language = { path = "../language" }
util = { path = "../util" }
anyhow.workspace = true
futures.workspace = true
lazy_static.workspace = true
pulldown-cmark = { version = "0.9.2", default-features = false }
smallvec.workspace = true
smol.workspace = true

View file

@ -0,0 +1,287 @@
use std::{ops::Range, sync::Arc};
use futures::FutureExt;
use gpui::{
color::Color,
elements::Text,
fonts::{HighlightStyle, TextStyle, Underline, Weight},
platform::{CursorStyle, MouseButton},
AnyElement, CursorRegion, Element, MouseRegion, ViewContext,
};
use language::{HighlightId, Language, LanguageRegistry};
use theme::SyntaxTheme;
#[derive(Debug, Clone, PartialEq, Eq)]
pub enum Highlight {
Id(HighlightId),
Highlight(HighlightStyle),
}
#[derive(Debug, Clone)]
pub struct RichText {
pub text: String,
pub highlights: Vec<(Range<usize>, Highlight)>,
pub region_ranges: Vec<Range<usize>>,
pub regions: Vec<RenderedRegion>,
}
#[derive(Debug, Clone)]
pub struct RenderedRegion {
code: bool,
link_url: Option<String>,
}
impl RichText {
pub fn element<V: 'static>(
&self,
syntax: Arc<SyntaxTheme>,
style: TextStyle,
code_span_background_color: Color,
cx: &mut ViewContext<V>,
) -> AnyElement<V> {
let mut region_id = 0;
let view_id = cx.view_id();
let regions = self.regions.clone();
enum Markdown {}
Text::new(self.text.clone(), style.clone())
.with_highlights(
self.highlights
.iter()
.filter_map(|(range, highlight)| {
let style = match highlight {
Highlight::Id(id) => id.style(&syntax)?,
Highlight::Highlight(style) => style.clone(),
};
Some((range.clone(), style))
})
.collect::<Vec<_>>(),
)
.with_custom_runs(self.region_ranges.clone(), move |ix, bounds, cx| {
region_id += 1;
let region = regions[ix].clone();
if let Some(url) = region.link_url {
cx.scene().push_cursor_region(CursorRegion {
bounds,
style: CursorStyle::PointingHand,
});
cx.scene().push_mouse_region(
MouseRegion::new::<Markdown>(view_id, region_id, bounds)
.on_click::<V, _>(MouseButton::Left, move |_, _, cx| {
cx.platform().open_url(&url)
}),
);
}
if region.code {
cx.scene().push_quad(gpui::Quad {
bounds,
background: Some(code_span_background_color),
border: Default::default(),
corner_radii: (2.0).into(),
});
}
})
.with_soft_wrap(true)
.into_any()
}
}
pub fn render_markdown_mut(
block: &str,
language_registry: &Arc<LanguageRegistry>,
language: Option<&Arc<Language>>,
data: &mut RichText,
) {
use pulldown_cmark::{CodeBlockKind, Event, Options, Parser, Tag};
let mut bold_depth = 0;
let mut italic_depth = 0;
let mut link_url = None;
let mut current_language = None;
let mut list_stack = Vec::new();
for event in Parser::new_ext(&block, Options::all()) {
let prev_len = data.text.len();
match event {
Event::Text(t) => {
if let Some(language) = &current_language {
render_code(&mut data.text, &mut data.highlights, t.as_ref(), language);
} else {
data.text.push_str(t.as_ref());
let mut style = HighlightStyle::default();
if bold_depth > 0 {
style.weight = Some(Weight::BOLD);
}
if italic_depth > 0 {
style.italic = Some(true);
}
if let Some(link_url) = link_url.clone() {
data.region_ranges.push(prev_len..data.text.len());
data.regions.push(RenderedRegion {
link_url: Some(link_url),
code: false,
});
style.underline = Some(Underline {
thickness: 1.0.into(),
..Default::default()
});
}
if style != HighlightStyle::default() {
let mut new_highlight = true;
if let Some((last_range, last_style)) = data.highlights.last_mut() {
if last_range.end == prev_len
&& last_style == &Highlight::Highlight(style)
{
last_range.end = data.text.len();
new_highlight = false;
}
}
if new_highlight {
data.highlights
.push((prev_len..data.text.len(), Highlight::Highlight(style)));
}
}
}
}
Event::Code(t) => {
data.text.push_str(t.as_ref());
data.region_ranges.push(prev_len..data.text.len());
if link_url.is_some() {
data.highlights.push((
prev_len..data.text.len(),
Highlight::Highlight(HighlightStyle {
underline: Some(Underline {
thickness: 1.0.into(),
..Default::default()
}),
..Default::default()
}),
));
}
data.regions.push(RenderedRegion {
code: true,
link_url: link_url.clone(),
});
}
Event::Start(tag) => match tag {
Tag::Paragraph => new_paragraph(&mut data.text, &mut list_stack),
Tag::Heading(_, _, _) => {
new_paragraph(&mut data.text, &mut list_stack);
bold_depth += 1;
}
Tag::CodeBlock(kind) => {
new_paragraph(&mut data.text, &mut list_stack);
current_language = if let CodeBlockKind::Fenced(language) = kind {
language_registry
.language_for_name(language.as_ref())
.now_or_never()
.and_then(Result::ok)
} else {
language.cloned()
}
}
Tag::Emphasis => italic_depth += 1,
Tag::Strong => bold_depth += 1,
Tag::Link(_, url, _) => link_url = Some(url.to_string()),
Tag::List(number) => {
list_stack.push((number, false));
}
Tag::Item => {
let len = list_stack.len();
if let Some((list_number, has_content)) = list_stack.last_mut() {
*has_content = false;
if !data.text.is_empty() && !data.text.ends_with('\n') {
data.text.push('\n');
}
for _ in 0..len - 1 {
data.text.push_str(" ");
}
if let Some(number) = list_number {
data.text.push_str(&format!("{}. ", number));
*number += 1;
*has_content = false;
} else {
data.text.push_str("- ");
}
}
}
_ => {}
},
Event::End(tag) => match tag {
Tag::Heading(_, _, _) => bold_depth -= 1,
Tag::CodeBlock(_) => current_language = None,
Tag::Emphasis => italic_depth -= 1,
Tag::Strong => bold_depth -= 1,
Tag::Link(_, _, _) => link_url = None,
Tag::List(_) => drop(list_stack.pop()),
_ => {}
},
Event::HardBreak => data.text.push('\n'),
Event::SoftBreak => data.text.push(' '),
_ => {}
}
}
}
pub fn render_markdown(
block: String,
language_registry: &Arc<LanguageRegistry>,
language: Option<&Arc<Language>>,
) -> RichText {
let mut data = RichText {
text: Default::default(),
highlights: Default::default(),
region_ranges: Default::default(),
regions: Default::default(),
};
render_markdown_mut(&block, language_registry, language, &mut data);
data.text = data.text.trim().to_string();
data
}
pub fn render_code(
text: &mut String,
highlights: &mut Vec<(Range<usize>, Highlight)>,
content: &str,
language: &Arc<Language>,
) {
let prev_len = text.len();
text.push_str(content);
for (range, highlight_id) in language.highlight_text(&content.into(), 0..content.len()) {
highlights.push((
prev_len + range.start..prev_len + range.end,
Highlight::Id(highlight_id),
));
}
}
pub fn new_paragraph(text: &mut String, list_stack: &mut Vec<(Option<u64>, bool)>) {
let mut is_subsequent_paragraph_of_list = false;
if let Some((_, has_content)) = list_stack.last_mut() {
if *has_content {
is_subsequent_paragraph_of_list = true;
} else {
*has_content = true;
return;
}
}
if !text.is_empty() {
if !text.ends_with('\n') {
text.push('\n');
}
text.push('\n');
}
for _ in 0..list_stack.len().saturating_sub(1) {
text.push_str(" ");
}
if is_subsequent_paragraph_of_list {
text.push_str(" ");
}
}

View file

@ -155,6 +155,7 @@ message Envelope {
UpdateChannelBufferCollaborators update_channel_buffer_collaborators = 128; UpdateChannelBufferCollaborators update_channel_buffer_collaborators = 128;
RejoinChannelBuffers rejoin_channel_buffers = 129; RejoinChannelBuffers rejoin_channel_buffers = 129;
RejoinChannelBuffersResponse rejoin_channel_buffers_response = 130; RejoinChannelBuffersResponse rejoin_channel_buffers_response = 130;
AckBufferOperation ack_buffer_operation = 143;
JoinChannelChat join_channel_chat = 131; JoinChannelChat join_channel_chat = 131;
JoinChannelChatResponse join_channel_chat_response = 132; JoinChannelChatResponse join_channel_chat_response = 132;
@ -165,10 +166,11 @@ message Envelope {
GetChannelMessages get_channel_messages = 137; GetChannelMessages get_channel_messages = 137;
GetChannelMessagesResponse get_channel_messages_response = 138; GetChannelMessagesResponse get_channel_messages_response = 138;
RemoveChannelMessage remove_channel_message = 139; RemoveChannelMessage remove_channel_message = 139;
AckChannelMessage ack_channel_message = 144;
LinkChannel link_channel = 140; LinkChannel link_channel = 140;
UnlinkChannel unlink_channel = 141; UnlinkChannel unlink_channel = 141;
MoveChannel move_channel = 142; MoveChannel move_channel = 142; // current max: 144
} }
} }
@ -955,6 +957,19 @@ message UpdateChannels {
repeated uint64 remove_channel_invitations = 6; repeated uint64 remove_channel_invitations = 6;
repeated ChannelParticipants channel_participants = 7; repeated ChannelParticipants channel_participants = 7;
repeated ChannelPermission channel_permissions = 8; repeated ChannelPermission channel_permissions = 8;
repeated UnseenChannelMessage unseen_channel_messages = 9;
repeated UnseenChannelBufferChange unseen_channel_buffer_changes = 10;
}
message UnseenChannelMessage {
uint64 channel_id = 1;
uint64 message_id = 2;
}
message UnseenChannelBufferChange {
uint64 channel_id = 1;
uint64 epoch = 2;
repeated VectorClockEntry version = 3;
} }
message ChannelEdge { message ChannelEdge {
@ -1060,6 +1075,11 @@ message RemoveChannelMessage {
uint64 message_id = 2; uint64 message_id = 2;
} }
message AckChannelMessage {
uint64 channel_id = 1;
uint64 message_id = 2;
}
message SendChannelMessageResponse { message SendChannelMessageResponse {
ChannelMessage message = 1; ChannelMessage message = 1;
} }
@ -1115,6 +1135,12 @@ message RejoinChannelBuffersResponse {
repeated RejoinedChannelBuffer buffers = 1; repeated RejoinedChannelBuffer buffers = 1;
} }
message AckBufferOperation {
uint64 buffer_id = 1;
uint64 epoch = 2;
repeated VectorClockEntry version = 3;
}
message JoinChannelBufferResponse { message JoinChannelBufferResponse {
uint64 buffer_id = 1; uint64 buffer_id = 1;
uint32 replica_id = 2; uint32 replica_id = 2;

View file

@ -271,6 +271,8 @@ messages!(
(LeaveChannelBuffer, Background), (LeaveChannelBuffer, Background),
(UpdateChannelBuffer, Foreground), (UpdateChannelBuffer, Foreground),
(UpdateChannelBufferCollaborators, Foreground), (UpdateChannelBufferCollaborators, Foreground),
(AckBufferOperation, Background),
(AckChannelMessage, Background),
); );
request_messages!( request_messages!(
@ -406,7 +408,7 @@ entity_messages!(
ChannelMessageSent, ChannelMessageSent,
UpdateChannelBuffer, UpdateChannelBuffer,
RemoveChannelMessage, RemoveChannelMessage,
UpdateChannelBufferCollaborators UpdateChannelBufferCollaborators,
); );
const KIB: usize = 1024; const KIB: usize = 1024;

View file

@ -26,7 +26,7 @@ postage.workspace = true
futures.workspace = true futures.workspace = true
ordered-float.workspace = true ordered-float.workspace = true
smol.workspace = true smol.workspace = true
rusqlite = { version = "0.27.0", features = ["blob", "array", "modern_sqlite"] } rusqlite.workspace = true
log.workspace = true log.workspace = true
tree-sitter.workspace = true tree-sitter.workspace = true
lazy_static.workspace = true lazy_static.workspace = true

View file

@ -7,10 +7,10 @@ publish = false
[dependencies] [dependencies]
anyhow.workspace = true anyhow.workspace = true
indoc.workspace = true indoc.workspace = true
libsqlite3-sys = { version = "0.24", features = ["bundled"] } libsqlite3-sys = { version = "0.26", features = ["bundled"] }
smol.workspace = true smol.workspace = true
thread_local = "1.1.4" thread_local = "1.1.4"
lazy_static.workspace = true lazy_static.workspace = true
parking_lot.workspace = true parking_lot.workspace = true
futures.workspace = true futures.workspace = true
uuid = { version = "1.1.2", features = ["v4"] } uuid.workspace = true

View file

@ -239,6 +239,7 @@ pub struct CollabPanel {
pub log_in_button: Interactive<ContainedText>, pub log_in_button: Interactive<ContainedText>,
pub channel_editor: ContainerStyle, pub channel_editor: ContainerStyle,
pub channel_hash: Icon, pub channel_hash: Icon,
pub channel_note_active_color: Color,
pub tabbed_modal: TabbedModal, pub tabbed_modal: TabbedModal,
pub contact_finder: ContactFinder, pub contact_finder: ContactFinder,
pub channel_modal: ChannelModal, pub channel_modal: ChannelModal,
@ -252,7 +253,7 @@ pub struct CollabPanel {
pub leave_call: Interactive<ContainedText>, pub leave_call: Interactive<ContainedText>,
pub contact_row: Toggleable<Interactive<ContainerStyle>>, pub contact_row: Toggleable<Interactive<ContainerStyle>>,
pub channel_row: Toggleable<Interactive<ContainerStyle>>, pub channel_row: Toggleable<Interactive<ContainerStyle>>,
pub channel_name: ContainedText, pub channel_name: Toggleable<ContainedText>,
pub row_height: f32, pub row_height: f32,
pub project_row: Toggleable<Interactive<ProjectRow>>, pub project_row: Toggleable<Interactive<ProjectRow>>,
pub tree_branch: Toggleable<Interactive<TreeBranch>>, pub tree_branch: Toggleable<Interactive<TreeBranch>>,
@ -633,7 +634,11 @@ pub struct ChatPanel {
pub list: ContainerStyle, pub list: ContainerStyle,
pub channel_select: ChannelSelect, pub channel_select: ChannelSelect,
pub input_editor: FieldEditor, pub input_editor: FieldEditor,
pub avatar: AvatarStyle,
pub avatar_container: ContainerStyle,
pub message: ChatMessage, pub message: ChatMessage,
pub continuation_message: ChatMessage,
pub last_message_bottom_spacing: f32,
pub pending_message: ChatMessage, pub pending_message: ChatMessage,
pub sign_in_prompt: Interactive<TextStyle>, pub sign_in_prompt: Interactive<TextStyle>,
pub icon_button: Interactive<IconButton>, pub icon_button: Interactive<IconButton>,
@ -642,7 +647,7 @@ pub struct ChatPanel {
#[derive(Deserialize, Default, JsonSchema)] #[derive(Deserialize, Default, JsonSchema)]
pub struct ChatMessage { pub struct ChatMessage {
#[serde(flatten)] #[serde(flatten)]
pub container: ContainerStyle, pub container: Interactive<ContainerStyle>,
pub body: TextStyle, pub body: TextStyle,
pub sender: ContainedText, pub sender: ContainedText,
pub timestamp: ContainedText, pub timestamp: ContainedText,

View file

@ -135,7 +135,7 @@ impl ListHeader {
.size(IconSize::Small) .size(IconSize::Small)
})) }))
.child( .child(
Label::new(self.label.clone()) Label::new(self.label)
.color(LabelColor::Muted) .color(LabelColor::Muted)
.size(LabelSize::Small), .size(LabelSize::Small),
), ),
@ -191,7 +191,7 @@ impl ListSubHeader {
.size(IconSize::Small) .size(IconSize::Small)
})) }))
.child( .child(
Label::new(self.label.clone()) Label::new(self.label)
.color(LabelColor::Muted) .color(LabelColor::Muted)
.size(LabelSize::Small), .size(LabelSize::Small),
), ),

View file

@ -27,7 +27,7 @@ impl Details {
.gap_0p5() .gap_0p5()
.text_xs() .text_xs()
.text_color(theme.lowest.base.default.foreground) .text_color(theme.lowest.base.default.foreground)
.child(self.text.clone()) .child(self.text)
.children(self.meta.map(|m| m)) .children(self.meta.map(|m| m))
} }
} }

View file

@ -78,10 +78,14 @@ fn increment(vim: &mut Vim, mut delta: i32, step: i32, cx: &mut WindowContext) {
2 => format!("{:b}", result), 2 => format!("{:b}", result),
_ => unreachable!(), _ => unreachable!(),
}; };
if selection.is_empty() { edits.push((range.clone(), replace));
new_anchors.push((false, snapshot.anchor_after(range.end))) }
} if selection.is_empty() {
edits.push((range, replace)); new_anchors.push((false, snapshot.anchor_after(range.end)))
}
} else {
if selection.is_empty() {
new_anchors.push((true, snapshot.anchor_after(start)))
} }
} }
} }
@ -226,6 +230,8 @@ mod test {
cx.assert_matches_neovim("(ˇ0b10f)", ["ctrl-a"], "(0b1ˇ1f)") cx.assert_matches_neovim("(ˇ0b10f)", ["ctrl-a"], "(0b1ˇ1f)")
.await; .await;
cx.assert_matches_neovim("ˇ-1", ["ctrl-a"], "ˇ0").await; cx.assert_matches_neovim("ˇ-1", ["ctrl-a"], "ˇ0").await;
cx.assert_matches_neovim("banˇana", ["ctrl-a"], "banˇana")
.await;
} }
#[gpui::test] #[gpui::test]

View file

@ -635,3 +635,20 @@ async fn test_zero(cx: &mut gpui::TestAppContext) {
the lazy dog"}) the lazy dog"})
.await; .await;
} }
#[gpui::test]
async fn test_selection_goal(cx: &mut gpui::TestAppContext) {
let mut cx = NeovimBackedTestContext::new(cx).await;
cx.set_shared_state(indoc! {"
;;ˇ;
Lorem Ipsum"})
.await;
cx.simulate_shared_keystrokes(["a", "down", "up", ";", "down", "up"])
.await;
cx.assert_shared_state(indoc! {"
;;;;ˇ
Lorem Ipsum"})
.await;
}

View file

@ -13,3 +13,6 @@
{"Put":{"state":"ˇ-1"}} {"Put":{"state":"ˇ-1"}}
{"Key":"ctrl-a"} {"Key":"ctrl-a"}
{"Get":{"state":"ˇ0","mode":"Normal"}} {"Get":{"state":"ˇ0","mode":"Normal"}}
{"Put":{"state":"banˇana"}}
{"Key":"ctrl-a"}
{"Get":{"state":"banˇana","mode":"Normal"}}

View file

@ -0,0 +1,8 @@
{"Put":{"state":";;ˇ;\nLorem Ipsum"}}
{"Key":"a"}
{"Key":"down"}
{"Key":"up"}
{"Key":";"}
{"Key":"down"}
{"Key":"up"}
{"Get":{"state":";;;;ˇ\nLorem Ipsum","mode":"Insert"}}

View file

@ -51,7 +51,7 @@ serde.workspace = true
serde_derive.workspace = true serde_derive.workspace = true
serde_json.workspace = true serde_json.workspace = true
smallvec.workspace = true smallvec.workspace = true
uuid = { version = "1.1.2", features = ["v4"] } uuid.workspace = true
[dev-dependencies] [dev-dependencies]
call = { path = "../call", features = ["test-support"] } call = { path = "../call", features = ["test-support"] }

View file

@ -79,7 +79,7 @@ use status_bar::StatusBar;
pub use status_bar::StatusItemView; pub use status_bar::StatusItemView;
use theme::{Theme, ThemeSettings}; use theme::{Theme, ThemeSettings};
pub use toolbar::{ToolbarItemLocation, ToolbarItemView}; pub use toolbar::{ToolbarItemLocation, ToolbarItemView};
use util::{async_iife, ResultExt}; use util::ResultExt;
pub use workspace_settings::{AutosaveSetting, GitGutterSetting, WorkspaceSettings}; pub use workspace_settings::{AutosaveSetting, GitGutterSetting, WorkspaceSettings};
lazy_static! { lazy_static! {
@ -573,6 +573,7 @@ pub struct Workspace {
panes_by_item: HashMap<usize, WeakViewHandle<Pane>>, panes_by_item: HashMap<usize, WeakViewHandle<Pane>>,
active_pane: ViewHandle<Pane>, active_pane: ViewHandle<Pane>,
last_active_center_pane: Option<WeakViewHandle<Pane>>, last_active_center_pane: Option<WeakViewHandle<Pane>>,
last_active_view_id: Option<proto::ViewId>,
status_bar: ViewHandle<StatusBar>, status_bar: ViewHandle<StatusBar>,
titlebar_item: Option<AnyViewHandle>, titlebar_item: Option<AnyViewHandle>,
notifications: Vec<(TypeId, usize, Box<dyn NotificationHandle>)>, notifications: Vec<(TypeId, usize, Box<dyn NotificationHandle>)>,
@ -786,6 +787,7 @@ impl Workspace {
panes_by_item: Default::default(), panes_by_item: Default::default(),
active_pane: center_pane.clone(), active_pane: center_pane.clone(),
last_active_center_pane: Some(center_pane.downgrade()), last_active_center_pane: Some(center_pane.downgrade()),
last_active_view_id: None,
status_bar, status_bar,
titlebar_item: None, titlebar_item: None,
notifications: Default::default(), notifications: Default::default(),
@ -934,7 +936,8 @@ impl Workspace {
app_state, app_state,
cx, cx,
) )
.await; .await
.unwrap_or_default();
(workspace, opened_items) (workspace, opened_items)
}) })
@ -2862,6 +2865,7 @@ impl Workspace {
cx.notify(); cx.notify();
self.last_active_view_id = active_view_id.clone();
proto::FollowResponse { proto::FollowResponse {
active_view_id, active_view_id,
views: self views: self
@ -3028,7 +3032,7 @@ impl Workspace {
Ok(()) Ok(())
} }
fn update_active_view_for_followers(&self, cx: &AppContext) { fn update_active_view_for_followers(&mut self, cx: &AppContext) {
let mut is_project_item = true; let mut is_project_item = true;
let mut update = proto::UpdateActiveView::default(); let mut update = proto::UpdateActiveView::default();
if self.active_pane.read(cx).has_focus() { if self.active_pane.read(cx).has_focus() {
@ -3046,11 +3050,14 @@ impl Workspace {
} }
} }
self.update_followers( if update.id != self.last_active_view_id {
is_project_item, self.last_active_view_id = update.id.clone();
proto::update_followers::Variant::UpdateActiveView(update), self.update_followers(
cx, is_project_item,
); proto::update_followers::Variant::UpdateActiveView(update),
cx,
);
}
} }
fn update_followers( fn update_followers(
@ -3394,140 +3401,124 @@ impl Workspace {
serialized_workspace: SerializedWorkspace, serialized_workspace: SerializedWorkspace,
paths_to_open: Vec<Option<ProjectPath>>, paths_to_open: Vec<Option<ProjectPath>>,
cx: &mut AppContext, cx: &mut AppContext,
) -> Task<Vec<Option<Result<Box<dyn ItemHandle>, anyhow::Error>>>> { ) -> Task<Result<Vec<Option<Box<dyn ItemHandle>>>>> {
cx.spawn(|mut cx| async move { cx.spawn(|mut cx| async move {
let result = async_iife! {{ let (project, old_center_pane) = workspace.read_with(&cx, |workspace, _| {
let (project, old_center_pane) = (
workspace.read_with(&cx, |workspace, _| { workspace.project().clone(),
( workspace.last_active_center_pane.clone(),
workspace.project().clone(), )
workspace.last_active_center_pane.clone(), })?;
)
})?;
let mut center_items = None; let mut center_group = None;
let mut center_group = None; let mut center_items = None;
// Traverse the splits tree and add to things // Traverse the splits tree and add to things
if let Some((group, active_pane, items)) = serialized_workspace if let Some((group, active_pane, items)) = serialized_workspace
.center_group .center_group
.deserialize(&project, serialized_workspace.id, &workspace, &mut cx) .deserialize(&project, serialized_workspace.id, &workspace, &mut cx)
.await { .await
center_items = Some(items); {
center_group = Some((group, active_pane)) center_items = Some(items);
center_group = Some((group, active_pane))
}
let mut items_by_project_path = cx.read(|cx| {
center_items
.unwrap_or_default()
.into_iter()
.filter_map(|item| {
let item = item?;
let project_path = item.project_path(cx)?;
Some((project_path, item))
})
.collect::<HashMap<_, _>>()
});
let opened_items = paths_to_open
.into_iter()
.map(|path_to_open| {
path_to_open
.and_then(|path_to_open| items_by_project_path.remove(&path_to_open))
})
.collect::<Vec<_>>();
// Remove old panes from workspace panes list
workspace.update(&mut cx, |workspace, cx| {
if let Some((center_group, active_pane)) = center_group {
workspace.remove_panes(workspace.center.root.clone(), cx);
// Swap workspace center group
workspace.center = PaneGroup::with_root(center_group);
// Change the focus to the workspace first so that we retrigger focus in on the pane.
cx.focus_self();
if let Some(active_pane) = active_pane {
cx.focus(&active_pane);
} else {
cx.focus(workspace.panes.last().unwrap());
}
} else {
let old_center_handle = old_center_pane.and_then(|weak| weak.upgrade(cx));
if let Some(old_center_handle) = old_center_handle {
cx.focus(&old_center_handle)
} else {
cx.focus_self()
}
} }
let resulting_list = cx.read(|cx| { let docks = serialized_workspace.docks;
let mut opened_items = center_items workspace.left_dock.update(cx, |dock, cx| {
.unwrap_or_default() dock.set_open(docks.left.visible, cx);
.into_iter() if let Some(active_panel) = docks.left.active_panel {
.filter_map(|item| { if let Some(ix) = dock.panel_index_for_ui_name(&active_panel, cx) {
let item = item?; dock.activate_panel(ix, cx);
let project_path = item.project_path(cx)?;
Some((project_path, item))
})
.collect::<HashMap<_, _>>();
paths_to_open
.into_iter()
.map(|path_to_open| {
path_to_open.map(|path_to_open| {
Ok(opened_items.remove(&path_to_open))
})
.transpose()
.map(|item| item.flatten())
.transpose()
})
.collect::<Vec<_>>()
});
// Remove old panes from workspace panes list
workspace.update(&mut cx, |workspace, cx| {
if let Some((center_group, active_pane)) = center_group {
workspace.remove_panes(workspace.center.root.clone(), cx);
// Swap workspace center group
workspace.center = PaneGroup::with_root(center_group);
// Change the focus to the workspace first so that we retrigger focus in on the pane.
cx.focus_self();
if let Some(active_pane) = active_pane {
cx.focus(&active_pane);
} else {
cx.focus(workspace.panes.last().unwrap());
} }
} else { }
let old_center_handle = old_center_pane.and_then(|weak| weak.upgrade(cx)); dock.active_panel()
if let Some(old_center_handle) = old_center_handle { .map(|panel| panel.set_zoomed(docks.left.zoom, cx));
cx.focus(&old_center_handle) if docks.left.visible && docks.left.zoom {
} else { cx.focus_self()
cx.focus_self() }
});
// TODO: I think the bug is that setting zoom or active undoes the bottom zoom or something
workspace.right_dock.update(cx, |dock, cx| {
dock.set_open(docks.right.visible, cx);
if let Some(active_panel) = docks.right.active_panel {
if let Some(ix) = dock.panel_index_for_ui_name(&active_panel, cx) {
dock.activate_panel(ix, cx);
}
}
dock.active_panel()
.map(|panel| panel.set_zoomed(docks.right.zoom, cx));
if docks.right.visible && docks.right.zoom {
cx.focus_self()
}
});
workspace.bottom_dock.update(cx, |dock, cx| {
dock.set_open(docks.bottom.visible, cx);
if let Some(active_panel) = docks.bottom.active_panel {
if let Some(ix) = dock.panel_index_for_ui_name(&active_panel, cx) {
dock.activate_panel(ix, cx);
} }
} }
let docks = serialized_workspace.docks; dock.active_panel()
workspace.left_dock.update(cx, |dock, cx| { .map(|panel| panel.set_zoomed(docks.bottom.zoom, cx));
dock.set_open(docks.left.visible, cx);
if let Some(active_panel) = docks.left.active_panel {
if let Some(ix) = dock.panel_index_for_ui_name(&active_panel, cx) {
dock.activate_panel(ix, cx);
}
}
dock.active_panel()
.map(|panel| {
panel.set_zoomed(docks.left.zoom, cx)
});
if docks.left.visible && docks.left.zoom {
cx.focus_self()
}
});
// TODO: I think the bug is that setting zoom or active undoes the bottom zoom or something
workspace.right_dock.update(cx, |dock, cx| {
dock.set_open(docks.right.visible, cx);
if let Some(active_panel) = docks.right.active_panel {
if let Some(ix) = dock.panel_index_for_ui_name(&active_panel, cx) {
dock.activate_panel(ix, cx);
} if docks.bottom.visible && docks.bottom.zoom {
} cx.focus_self()
dock.active_panel() }
.map(|panel| { });
panel.set_zoomed(docks.right.zoom, cx)
});
if docks.right.visible && docks.right.zoom { cx.notify();
cx.focus_self() })?;
}
});
workspace.bottom_dock.update(cx, |dock, cx| {
dock.set_open(docks.bottom.visible, cx);
if let Some(active_panel) = docks.bottom.active_panel {
if let Some(ix) = dock.panel_index_for_ui_name(&active_panel, cx) {
dock.activate_panel(ix, cx);
}
}
dock.active_panel() // Serialize ourself to make sure our timestamps and any pane / item changes are replicated
.map(|panel| { workspace.read_with(&cx, |workspace, cx| workspace.serialize_workspace(cx))?;
panel.set_zoomed(docks.bottom.zoom, cx)
});
if docks.bottom.visible && docks.bottom.zoom { Ok(opened_items)
cx.focus_self()
}
});
cx.notify();
})?;
// Serialize ourself to make sure our timestamps and any pane / item changes are replicated
workspace.read_with(&cx, |workspace, cx| workspace.serialize_workspace(cx))?;
Ok::<_, anyhow::Error>(resulting_list)
}};
result.await.unwrap_or_default()
}) })
} }
@ -3601,7 +3592,7 @@ async fn open_items(
mut project_paths_to_open: Vec<(PathBuf, Option<ProjectPath>)>, mut project_paths_to_open: Vec<(PathBuf, Option<ProjectPath>)>,
app_state: Arc<AppState>, app_state: Arc<AppState>,
mut cx: AsyncAppContext, mut cx: AsyncAppContext,
) -> Vec<Option<anyhow::Result<Box<dyn ItemHandle>>>> { ) -> Result<Vec<Option<Result<Box<dyn ItemHandle>>>>> {
let mut opened_items = Vec::with_capacity(project_paths_to_open.len()); let mut opened_items = Vec::with_capacity(project_paths_to_open.len());
if let Some(serialized_workspace) = serialized_workspace { if let Some(serialized_workspace) = serialized_workspace {
@ -3619,16 +3610,19 @@ async fn open_items(
cx, cx,
) )
}) })
.await; .await?;
let restored_project_paths = cx.read(|cx| { let restored_project_paths = cx.read(|cx| {
restored_items restored_items
.iter() .iter()
.filter_map(|item| item.as_ref()?.as_ref().ok()?.project_path(cx)) .filter_map(|item| item.as_ref()?.project_path(cx))
.collect::<HashSet<_>>() .collect::<HashSet<_>>()
}); });
opened_items = restored_items; for restored_item in restored_items {
opened_items.push(restored_item.map(Ok));
}
project_paths_to_open project_paths_to_open
.iter_mut() .iter_mut()
.for_each(|(_, project_path)| { .for_each(|(_, project_path)| {
@ -3681,7 +3675,7 @@ async fn open_items(
} }
} }
opened_items Ok(opened_items)
} }
fn notify_of_new_dock(workspace: &WeakViewHandle<Workspace>, cx: &mut AsyncAppContext) { fn notify_of_new_dock(workspace: &WeakViewHandle<Workspace>, cx: &mut AsyncAppContext) {

View file

@ -3,7 +3,7 @@ authors = ["Nathan Sobo <nathansobo@gmail.com>"]
description = "The fast, collaborative code editor." description = "The fast, collaborative code editor."
edition = "2021" edition = "2021"
name = "zed" name = "zed"
version = "0.107.0" version = "0.108.0"
publish = false publish = false
[lib] [lib]
@ -138,7 +138,7 @@ tree-sitter-nu.workspace = true
url = "2.2" url = "2.2"
urlencoding = "2.1.2" urlencoding = "2.1.2"
uuid = { version = "1.1.2", features = ["v4"] } uuid.workspace = true
[dev-dependencies] [dev-dependencies]
call = { path = "../call", features = ["test-support"] } call = { path = "../call", features = ["test-support"] }

View file

@ -165,17 +165,25 @@ impl LspAdapter for RustLspAdapter {
lazy_static! { lazy_static! {
static ref REGEX: Regex = Regex::new("\\(…?\\)").unwrap(); static ref REGEX: Regex = Regex::new("\\(…?\\)").unwrap();
} }
let detail = completion.detail.as_ref().unwrap(); let detail = completion.detail.as_ref().unwrap();
if detail.starts_with("fn(") { const FUNCTION_PREFIXES: [&'static str; 2] = ["async fn", "fn"];
let text = REGEX.replace(&completion.label, &detail[2..]).to_string(); let prefix = FUNCTION_PREFIXES
let source = Rope::from(format!("fn {} {{}}", text).as_str()); .iter()
let runs = language.highlight_text(&source, 3..3 + text.len()); .find_map(|prefix| detail.strip_prefix(*prefix).map(|suffix| (prefix, suffix)));
return Some(CodeLabel { // fn keyword should be followed by opening parenthesis.
filter_range: 0..completion.label.find('(').unwrap_or(text.len()), if let Some((prefix, suffix)) = prefix {
text, if suffix.starts_with('(') {
runs, let text = REGEX.replace(&completion.label, suffix).to_string();
}); let source = Rope::from(format!("{prefix} {} {{}}", text).as_str());
let run_start = prefix.len() + 1;
let runs =
language.highlight_text(&source, run_start..run_start + text.len());
return Some(CodeLabel {
filter_range: 0..completion.label.find('(').unwrap_or(text.len()),
text,
runs,
});
}
} }
} }
Some(kind) => { Some(kind) => {
@ -377,7 +385,28 @@ mod tests {
], ],
}) })
); );
assert_eq!(
language
.label_for_completion(&lsp::CompletionItem {
kind: Some(lsp::CompletionItemKind::FUNCTION),
label: "hello(…)".to_string(),
detail: Some("async fn(&mut Option<T>) -> Vec<T>".to_string()),
..Default::default()
})
.await,
Some(CodeLabel {
text: "hello(&mut Option<T>) -> Vec<T>".to_string(),
filter_range: 0..5,
runs: vec![
(0..5, highlight_function),
(7..10, highlight_keyword),
(11..17, highlight_type),
(18..19, highlight_type),
(25..28, highlight_type),
(29..30, highlight_type),
],
})
);
assert_eq!( assert_eq!(
language language
.label_for_completion(&lsp::CompletionItem { .label_for_completion(&lsp::CompletionItem {

View file

@ -1,4 +1,4 @@
[toolchain] [toolchain]
channel = "1.72.1" channel = "1.73"
components = [ "rustfmt" ] components = [ "rustfmt" ]
targets = [ "x86_64-apple-darwin", "aarch64-apple-darwin", "wasm32-wasi" ] targets = [ "x86_64-apple-darwin", "aarch64-apple-darwin", "wasm32-wasi" ]

View file

@ -5,6 +5,7 @@ import {
} from "./components" } from "./components"
import { icon_button } from "../component/icon_button" import { icon_button } from "../component/icon_button"
import { useTheme } from "../theme" import { useTheme } from "../theme"
import { interactive } from "../element"
export default function chat_panel(): any { export default function chat_panel(): any {
const theme = useTheme() const theme = useTheme()
@ -27,11 +28,23 @@ export default function chat_panel(): any {
return { return {
background: background(layer), background: background(layer),
list: { avatar: {
margin: { icon_width: 24,
left: SPACING, icon_height: 24,
right: SPACING, corner_radius: 4,
outer_width: 24,
outer_corner_radius: 16,
},
avatar_container: {
padding: {
right: 6,
left: 2,
top: 2,
bottom: 2,
} }
},
list: {
}, },
channel_select: { channel_select: {
header: { header: {
@ -79,6 +92,22 @@ export default function chat_panel(): any {
}, },
}, },
message: { message: {
...interactive({
base: {
margin: { top: SPACING },
padding: {
top: 4,
bottom: 4,
left: SPACING / 2,
right: SPACING / 3,
}
},
state: {
hovered: {
background: background(layer, "hovered"),
},
},
}),
body: text(layer, "sans", "base"), body: text(layer, "sans", "base"),
sender: { sender: {
margin: { margin: {
@ -87,7 +116,32 @@ export default function chat_panel(): any {
...text(layer, "sans", "base", { weight: "bold" }), ...text(layer, "sans", "base", { weight: "bold" }),
}, },
timestamp: text(layer, "sans", "base", "disabled"), timestamp: text(layer, "sans", "base", "disabled"),
margin: { bottom: SPACING } },
last_message_bottom_spacing: SPACING,
continuation_message: {
body: text(layer, "sans", "base"),
sender: {
margin: {
right: 8,
},
...text(layer, "sans", "base", { weight: "bold" }),
},
timestamp: text(layer, "sans", "base", "disabled"),
...interactive({
base: {
padding: {
top: 4,
bottom: 4,
left: SPACING / 2,
right: SPACING / 3,
}
},
state: {
hovered: {
background: background(layer, "hovered"),
},
},
}),
}, },
pending_message: { pending_message: {
body: text(layer, "sans", "base"), body: text(layer, "sans", "base"),
@ -98,6 +152,21 @@ export default function chat_panel(): any {
...text(layer, "sans", "base", "disabled"), ...text(layer, "sans", "base", "disabled"),
}, },
timestamp: text(layer, "sans", "base"), timestamp: text(layer, "sans", "base"),
...interactive({
base: {
padding: {
top: 4,
bottom: 4,
left: SPACING / 2,
right: SPACING / 3,
}
},
state: {
hovered: {
background: background(layer, "hovered"),
},
},
}),
}, },
sign_in_prompt: { sign_in_prompt: {
default: text(layer, "sans", "base"), default: text(layer, "sans", "base"),

View file

@ -194,6 +194,7 @@ export default function contacts_panel(): any {
}, },
user_query_editor: filter_input, user_query_editor: filter_input,
channel_hash: icon_style, channel_hash: icon_style,
channel_note_active_color: foreground(layer, "active"),
user_query_editor_height: 33, user_query_editor_height: 33,
add_contact_button: header_icon_button, add_contact_button: header_icon_button,
add_channel_button: header_icon_button, add_channel_button: header_icon_button,
@ -267,10 +268,18 @@ export default function contacts_panel(): any {
}), }),
channel_row: item_row, channel_row: item_row,
channel_name: { channel_name: {
...text(layer, "sans", { size: "sm" }), active: {
margin: { ...text(layer, "sans", { size: "sm", weight: "bold" }),
left: CHANNEL_SPACING, margin: {
left: CHANNEL_SPACING,
},
}, },
inactive: {
...text(layer, "sans", { size: "sm" }),
margin: {
left: CHANNEL_SPACING,
},
}
}, },
list_empty_label_container: { list_empty_label_container: {
margin: { margin: {

View file

@ -21,6 +21,7 @@ export default function contacts_panel(): any {
...text(theme.lowest, "sans", "base"), ...text(theme.lowest, "sans", "base"),
button: icon_button({ variant: "ghost" }), button: icon_button({ variant: "ghost" }),
spacing: 4, spacing: 4,
padding: 4,
}, },
} }
} }