Max Brunsfeld
489077befc
This is a ~small~ pure refactor that's a step toward SSH remoting. I've extracted the Project's buffer state management into a smaller, separate struct called `BufferStore`, currently in the same crate. I did this as a separate PR to reduce conflicts between main and `remoting-over-ssh`. The idea is to make use of this struct (and other smaller structs that make up `Project`) in a dedicated, simpler `HeadlessProject` type that we will use in the SSH server to model the remote end of a project. With this approach, as we develop the headless project, we can avoid adding more conditional logic to `Project` itself (which is already very complex), and actually make `Project` a bit smaller by extracting out helper objects. Release Notes: - N/A
719 lines
22 KiB
Rust
719 lines
22 KiB
Rust
#![allow(non_snake_case)]
|
|
|
|
pub mod error;
|
|
mod macros;
|
|
mod typed_envelope;
|
|
|
|
pub use error::*;
|
|
pub use typed_envelope::*;
|
|
|
|
use anyhow::anyhow;
|
|
use collections::HashMap;
|
|
use futures::{future::BoxFuture, Future};
|
|
pub use prost::{DecodeError, Message};
|
|
use serde::Serialize;
|
|
use std::{
|
|
any::{Any, TypeId},
|
|
cmp,
|
|
fmt::{self, Debug},
|
|
iter, mem,
|
|
sync::Arc,
|
|
time::{Duration, Instant, SystemTime, UNIX_EPOCH},
|
|
};
|
|
|
|
include!(concat!(env!("OUT_DIR"), "/zed.messages.rs"));
|
|
|
|
pub trait EnvelopedMessage: Clone + Debug + Serialize + Sized + Send + Sync + 'static {
|
|
const NAME: &'static str;
|
|
const PRIORITY: MessagePriority;
|
|
fn into_envelope(
|
|
self,
|
|
id: u32,
|
|
responding_to: Option<u32>,
|
|
original_sender_id: Option<PeerId>,
|
|
) -> Envelope;
|
|
fn from_envelope(envelope: Envelope) -> Option<Self>;
|
|
}
|
|
|
|
pub trait EntityMessage: EnvelopedMessage {
|
|
type Entity;
|
|
fn remote_entity_id(&self) -> u64;
|
|
}
|
|
|
|
pub trait RequestMessage: EnvelopedMessage {
|
|
type Response: EnvelopedMessage;
|
|
}
|
|
|
|
pub trait AnyTypedEnvelope: 'static + Send + Sync {
|
|
fn payload_type_id(&self) -> TypeId;
|
|
fn payload_type_name(&self) -> &'static str;
|
|
fn as_any(&self) -> &dyn Any;
|
|
fn into_any(self: Box<Self>) -> Box<dyn Any + Send + Sync>;
|
|
fn is_background(&self) -> bool;
|
|
fn original_sender_id(&self) -> Option<PeerId>;
|
|
fn sender_id(&self) -> PeerId;
|
|
fn message_id(&self) -> u32;
|
|
}
|
|
|
|
pub enum MessagePriority {
|
|
Foreground,
|
|
Background,
|
|
}
|
|
|
|
pub trait ProtoClient: Send + Sync {
|
|
fn request(
|
|
&self,
|
|
envelope: Envelope,
|
|
request_type: &'static str,
|
|
) -> BoxFuture<'static, anyhow::Result<Envelope>>;
|
|
|
|
fn send(&self, envelope: Envelope) -> anyhow::Result<()>;
|
|
}
|
|
|
|
#[derive(Clone)]
|
|
pub struct AnyProtoClient(Arc<dyn ProtoClient>);
|
|
|
|
impl<T> From<Arc<T>> for AnyProtoClient
|
|
where
|
|
T: ProtoClient + 'static,
|
|
{
|
|
fn from(client: Arc<T>) -> Self {
|
|
Self(client)
|
|
}
|
|
}
|
|
|
|
impl AnyProtoClient {
|
|
pub fn new<T: ProtoClient + 'static>(client: Arc<T>) -> Self {
|
|
Self(client)
|
|
}
|
|
|
|
pub fn request<T: RequestMessage>(
|
|
&self,
|
|
request: T,
|
|
) -> impl Future<Output = anyhow::Result<T::Response>> {
|
|
let envelope = request.into_envelope(0, None, None);
|
|
let response = self.0.request(envelope, T::NAME);
|
|
async move {
|
|
T::Response::from_envelope(response.await?)
|
|
.ok_or_else(|| anyhow!("received response of the wrong type"))
|
|
}
|
|
}
|
|
|
|
pub fn send<T: EnvelopedMessage>(&self, request: T) -> anyhow::Result<()> {
|
|
let envelope = request.into_envelope(0, None, None);
|
|
self.0.send(envelope)
|
|
}
|
|
}
|
|
|
|
impl<T: EnvelopedMessage> AnyTypedEnvelope for TypedEnvelope<T> {
|
|
fn payload_type_id(&self) -> TypeId {
|
|
TypeId::of::<T>()
|
|
}
|
|
|
|
fn payload_type_name(&self) -> &'static str {
|
|
T::NAME
|
|
}
|
|
|
|
fn as_any(&self) -> &dyn Any {
|
|
self
|
|
}
|
|
|
|
fn into_any(self: Box<Self>) -> Box<dyn Any + Send + Sync> {
|
|
self
|
|
}
|
|
|
|
fn is_background(&self) -> bool {
|
|
matches!(T::PRIORITY, MessagePriority::Background)
|
|
}
|
|
|
|
fn original_sender_id(&self) -> Option<PeerId> {
|
|
self.original_sender_id
|
|
}
|
|
|
|
fn sender_id(&self) -> PeerId {
|
|
self.sender_id
|
|
}
|
|
|
|
fn message_id(&self) -> u32 {
|
|
self.message_id
|
|
}
|
|
}
|
|
|
|
impl PeerId {
|
|
pub fn from_u64(peer_id: u64) -> Self {
|
|
let owner_id = (peer_id >> 32) as u32;
|
|
let id = peer_id as u32;
|
|
Self { owner_id, id }
|
|
}
|
|
|
|
pub fn as_u64(self) -> u64 {
|
|
((self.owner_id as u64) << 32) | (self.id as u64)
|
|
}
|
|
}
|
|
|
|
impl Copy for PeerId {}
|
|
|
|
impl Eq for PeerId {}
|
|
|
|
impl Ord for PeerId {
|
|
fn cmp(&self, other: &Self) -> cmp::Ordering {
|
|
self.owner_id
|
|
.cmp(&other.owner_id)
|
|
.then_with(|| self.id.cmp(&other.id))
|
|
}
|
|
}
|
|
|
|
impl PartialOrd for PeerId {
|
|
fn partial_cmp(&self, other: &Self) -> Option<cmp::Ordering> {
|
|
Some(self.cmp(other))
|
|
}
|
|
}
|
|
|
|
impl std::hash::Hash for PeerId {
|
|
fn hash<H: std::hash::Hasher>(&self, state: &mut H) {
|
|
self.owner_id.hash(state);
|
|
self.id.hash(state);
|
|
}
|
|
}
|
|
|
|
impl fmt::Display for PeerId {
|
|
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
|
|
write!(f, "{}/{}", self.owner_id, self.id)
|
|
}
|
|
}
|
|
|
|
messages!(
|
|
(Ack, Foreground),
|
|
(AckBufferOperation, Background),
|
|
(AckChannelMessage, Background),
|
|
(AddNotification, Foreground),
|
|
(AddProjectCollaborator, Foreground),
|
|
(ApplyCodeAction, Background),
|
|
(ApplyCodeActionResponse, Background),
|
|
(ApplyCompletionAdditionalEdits, Background),
|
|
(ApplyCompletionAdditionalEditsResponse, Background),
|
|
(BufferReloaded, Foreground),
|
|
(BufferSaved, Foreground),
|
|
(Call, Foreground),
|
|
(CallCanceled, Foreground),
|
|
(CancelCall, Foreground),
|
|
(ChannelMessageSent, Foreground),
|
|
(ChannelMessageUpdate, Foreground),
|
|
(CompleteWithLanguageModel, Background),
|
|
(ComputeEmbeddings, Background),
|
|
(ComputeEmbeddingsResponse, Background),
|
|
(CopyProjectEntry, Foreground),
|
|
(CountTokensWithLanguageModel, Background),
|
|
(CountTokensResponse, Background),
|
|
(CreateBufferForPeer, Foreground),
|
|
(CreateChannel, Foreground),
|
|
(CreateChannelResponse, Foreground),
|
|
(CreateProjectEntry, Foreground),
|
|
(CreateRoom, Foreground),
|
|
(CreateRoomResponse, Foreground),
|
|
(DeclineCall, Foreground),
|
|
(DeleteChannel, Foreground),
|
|
(DeleteNotification, Foreground),
|
|
(UpdateNotification, Foreground),
|
|
(DeleteProjectEntry, Foreground),
|
|
(EndStream, Foreground),
|
|
(Error, Foreground),
|
|
(ExpandProjectEntry, Foreground),
|
|
(ExpandProjectEntryResponse, Foreground),
|
|
(Follow, Foreground),
|
|
(FollowResponse, Foreground),
|
|
(FormatBuffers, Foreground),
|
|
(FormatBuffersResponse, Foreground),
|
|
(FuzzySearchUsers, Foreground),
|
|
(GetCachedEmbeddings, Background),
|
|
(GetCachedEmbeddingsResponse, Background),
|
|
(GetChannelMembers, Foreground),
|
|
(GetChannelMembersResponse, Foreground),
|
|
(GetChannelMessages, Background),
|
|
(GetChannelMessagesById, Background),
|
|
(GetChannelMessagesResponse, Background),
|
|
(GetCodeActions, Background),
|
|
(GetCodeActionsResponse, Background),
|
|
(GetCompletions, Background),
|
|
(GetCompletionsResponse, Background),
|
|
(GetDefinition, Background),
|
|
(GetDefinitionResponse, Background),
|
|
(GetDocumentHighlights, Background),
|
|
(GetDocumentHighlightsResponse, Background),
|
|
(GetHover, Background),
|
|
(GetHoverResponse, Background),
|
|
(GetNotifications, Foreground),
|
|
(GetNotificationsResponse, Foreground),
|
|
(GetPrivateUserInfo, Foreground),
|
|
(GetPrivateUserInfoResponse, Foreground),
|
|
(GetProjectSymbols, Background),
|
|
(GetProjectSymbolsResponse, Background),
|
|
(GetReferences, Background),
|
|
(GetReferencesResponse, Background),
|
|
(GetSignatureHelp, Background),
|
|
(GetSignatureHelpResponse, Background),
|
|
(GetSupermavenApiKey, Background),
|
|
(GetSupermavenApiKeyResponse, Background),
|
|
(GetTypeDefinition, Background),
|
|
(GetTypeDefinitionResponse, Background),
|
|
(GetImplementation, Background),
|
|
(GetImplementationResponse, Background),
|
|
(GetUsers, Foreground),
|
|
(Hello, Foreground),
|
|
(IncomingCall, Foreground),
|
|
(InlayHints, Background),
|
|
(InlayHintsResponse, Background),
|
|
(InviteChannelMember, Foreground),
|
|
(JoinChannel, Foreground),
|
|
(JoinChannelBuffer, Foreground),
|
|
(JoinChannelBufferResponse, Foreground),
|
|
(JoinChannelChat, Foreground),
|
|
(JoinChannelChatResponse, Foreground),
|
|
(JoinProject, Foreground),
|
|
(JoinHostedProject, Foreground),
|
|
(JoinProjectResponse, Foreground),
|
|
(JoinRoom, Foreground),
|
|
(JoinRoomResponse, Foreground),
|
|
(LanguageModelResponse, Background),
|
|
(LeaveChannelBuffer, Background),
|
|
(LeaveChannelChat, Foreground),
|
|
(LeaveProject, Foreground),
|
|
(LeaveRoom, Foreground),
|
|
(MarkNotificationRead, Foreground),
|
|
(MoveChannel, Foreground),
|
|
(OnTypeFormatting, Background),
|
|
(OnTypeFormattingResponse, Background),
|
|
(OpenBufferById, Background),
|
|
(OpenBufferByPath, Background),
|
|
(OpenBufferForSymbol, Background),
|
|
(OpenBufferForSymbolResponse, Background),
|
|
(OpenBufferResponse, Background),
|
|
(PerformRename, Background),
|
|
(PerformRenameResponse, Background),
|
|
(Ping, Foreground),
|
|
(PrepareRename, Background),
|
|
(PrepareRenameResponse, Background),
|
|
(ProjectEntryResponse, Foreground),
|
|
(RefreshInlayHints, Foreground),
|
|
(RejoinChannelBuffers, Foreground),
|
|
(RejoinChannelBuffersResponse, Foreground),
|
|
(RejoinRoom, Foreground),
|
|
(RejoinRoomResponse, Foreground),
|
|
(ReloadBuffers, Foreground),
|
|
(ReloadBuffersResponse, Foreground),
|
|
(RemoveChannelMember, Foreground),
|
|
(RemoveChannelMessage, Foreground),
|
|
(UpdateChannelMessage, Foreground),
|
|
(RemoveContact, Foreground),
|
|
(RemoveProjectCollaborator, Foreground),
|
|
(RenameChannel, Foreground),
|
|
(RenameChannelResponse, Foreground),
|
|
(RenameProjectEntry, Foreground),
|
|
(RequestContact, Foreground),
|
|
(ResolveCompletionDocumentation, Background),
|
|
(ResolveCompletionDocumentationResponse, Background),
|
|
(ResolveInlayHint, Background),
|
|
(ResolveInlayHintResponse, Background),
|
|
(RespondToChannelInvite, Foreground),
|
|
(RespondToContactRequest, Foreground),
|
|
(RoomUpdated, Foreground),
|
|
(SaveBuffer, Foreground),
|
|
(SetChannelMemberRole, Foreground),
|
|
(SetChannelVisibility, Foreground),
|
|
(SearchProject, Background),
|
|
(SearchProjectResponse, Background),
|
|
(SendChannelMessage, Background),
|
|
(SendChannelMessageResponse, Background),
|
|
(ShareProject, Foreground),
|
|
(ShareProjectResponse, Foreground),
|
|
(ShowContacts, Foreground),
|
|
(StartLanguageServer, Foreground),
|
|
(SubscribeToChannels, Foreground),
|
|
(SynchronizeBuffers, Foreground),
|
|
(SynchronizeBuffersResponse, Foreground),
|
|
(TaskContextForLocation, Background),
|
|
(TaskContext, Background),
|
|
(TaskTemplates, Background),
|
|
(TaskTemplatesResponse, Background),
|
|
(Test, Foreground),
|
|
(Unfollow, Foreground),
|
|
(UnshareProject, Foreground),
|
|
(UpdateBuffer, Foreground),
|
|
(UpdateBufferFile, Foreground),
|
|
(UpdateChannelBuffer, Foreground),
|
|
(UpdateChannelBufferCollaborators, Foreground),
|
|
(UpdateChannels, Foreground),
|
|
(UpdateUserChannels, Foreground),
|
|
(UpdateContacts, Foreground),
|
|
(UpdateDiagnosticSummary, Foreground),
|
|
(UpdateDiffBase, Foreground),
|
|
(UpdateFollowers, Foreground),
|
|
(UpdateInviteInfo, Foreground),
|
|
(UpdateLanguageServer, Foreground),
|
|
(UpdateParticipantLocation, Foreground),
|
|
(UpdateProject, Foreground),
|
|
(UpdateProjectCollaborator, Foreground),
|
|
(UpdateWorktree, Foreground),
|
|
(UpdateWorktreeSettings, Foreground),
|
|
(UsersResponse, Foreground),
|
|
(LspExtExpandMacro, Background),
|
|
(LspExtExpandMacroResponse, Background),
|
|
(SetRoomParticipantRole, Foreground),
|
|
(BlameBuffer, Foreground),
|
|
(BlameBufferResponse, Foreground),
|
|
(CreateDevServerProject, Background),
|
|
(CreateDevServerProjectResponse, Foreground),
|
|
(CreateDevServer, Foreground),
|
|
(CreateDevServerResponse, Foreground),
|
|
(DevServerInstructions, Foreground),
|
|
(ShutdownDevServer, Foreground),
|
|
(ReconnectDevServer, Foreground),
|
|
(ReconnectDevServerResponse, Foreground),
|
|
(ShareDevServerProject, Foreground),
|
|
(JoinDevServerProject, Foreground),
|
|
(RejoinRemoteProjects, Foreground),
|
|
(RejoinRemoteProjectsResponse, Foreground),
|
|
(MultiLspQuery, Background),
|
|
(MultiLspQueryResponse, Background),
|
|
(DevServerProjectsUpdate, Foreground),
|
|
(ValidateDevServerProjectRequest, Background),
|
|
(DeleteDevServer, Foreground),
|
|
(DeleteDevServerProject, Foreground),
|
|
(RegenerateDevServerToken, Foreground),
|
|
(RegenerateDevServerTokenResponse, Foreground),
|
|
(RenameDevServer, Foreground),
|
|
(OpenNewBuffer, Foreground),
|
|
(RestartLanguageServers, Foreground),
|
|
(LinkedEditingRange, Background),
|
|
(LinkedEditingRangeResponse, Background),
|
|
(AdvertiseContexts, Foreground),
|
|
(OpenContext, Foreground),
|
|
(OpenContextResponse, Foreground),
|
|
(UpdateContext, Foreground),
|
|
(SynchronizeContexts, Foreground),
|
|
(SynchronizeContextsResponse, Foreground),
|
|
);
|
|
|
|
request_messages!(
|
|
(ApplyCodeAction, ApplyCodeActionResponse),
|
|
(
|
|
ApplyCompletionAdditionalEdits,
|
|
ApplyCompletionAdditionalEditsResponse
|
|
),
|
|
(Call, Ack),
|
|
(CancelCall, Ack),
|
|
(CopyProjectEntry, ProjectEntryResponse),
|
|
(CompleteWithLanguageModel, LanguageModelResponse),
|
|
(ComputeEmbeddings, ComputeEmbeddingsResponse),
|
|
(CountTokensWithLanguageModel, CountTokensResponse),
|
|
(CreateChannel, CreateChannelResponse),
|
|
(CreateProjectEntry, ProjectEntryResponse),
|
|
(CreateRoom, CreateRoomResponse),
|
|
(DeclineCall, Ack),
|
|
(DeleteChannel, Ack),
|
|
(DeleteProjectEntry, ProjectEntryResponse),
|
|
(ExpandProjectEntry, ExpandProjectEntryResponse),
|
|
(Follow, FollowResponse),
|
|
(FormatBuffers, FormatBuffersResponse),
|
|
(FuzzySearchUsers, UsersResponse),
|
|
(GetCachedEmbeddings, GetCachedEmbeddingsResponse),
|
|
(GetChannelMembers, GetChannelMembersResponse),
|
|
(GetChannelMessages, GetChannelMessagesResponse),
|
|
(GetChannelMessagesById, GetChannelMessagesResponse),
|
|
(GetCodeActions, GetCodeActionsResponse),
|
|
(GetCompletions, GetCompletionsResponse),
|
|
(GetDefinition, GetDefinitionResponse),
|
|
(GetImplementation, GetImplementationResponse),
|
|
(GetDocumentHighlights, GetDocumentHighlightsResponse),
|
|
(GetHover, GetHoverResponse),
|
|
(GetNotifications, GetNotificationsResponse),
|
|
(GetPrivateUserInfo, GetPrivateUserInfoResponse),
|
|
(GetProjectSymbols, GetProjectSymbolsResponse),
|
|
(GetReferences, GetReferencesResponse),
|
|
(GetSignatureHelp, GetSignatureHelpResponse),
|
|
(GetSupermavenApiKey, GetSupermavenApiKeyResponse),
|
|
(GetTypeDefinition, GetTypeDefinitionResponse),
|
|
(LinkedEditingRange, LinkedEditingRangeResponse),
|
|
(GetUsers, UsersResponse),
|
|
(IncomingCall, Ack),
|
|
(InlayHints, InlayHintsResponse),
|
|
(InviteChannelMember, Ack),
|
|
(JoinChannel, JoinRoomResponse),
|
|
(JoinChannelBuffer, JoinChannelBufferResponse),
|
|
(JoinChannelChat, JoinChannelChatResponse),
|
|
(JoinHostedProject, JoinProjectResponse),
|
|
(JoinProject, JoinProjectResponse),
|
|
(JoinRoom, JoinRoomResponse),
|
|
(LeaveChannelBuffer, Ack),
|
|
(LeaveRoom, Ack),
|
|
(MarkNotificationRead, Ack),
|
|
(MoveChannel, Ack),
|
|
(OnTypeFormatting, OnTypeFormattingResponse),
|
|
(OpenBufferById, OpenBufferResponse),
|
|
(OpenBufferByPath, OpenBufferResponse),
|
|
(OpenBufferForSymbol, OpenBufferForSymbolResponse),
|
|
(OpenNewBuffer, OpenBufferResponse),
|
|
(PerformRename, PerformRenameResponse),
|
|
(Ping, Ack),
|
|
(PrepareRename, PrepareRenameResponse),
|
|
(RefreshInlayHints, Ack),
|
|
(RejoinChannelBuffers, RejoinChannelBuffersResponse),
|
|
(RejoinRoom, RejoinRoomResponse),
|
|
(ReloadBuffers, ReloadBuffersResponse),
|
|
(RemoveChannelMember, Ack),
|
|
(RemoveChannelMessage, Ack),
|
|
(UpdateChannelMessage, Ack),
|
|
(RemoveContact, Ack),
|
|
(RenameChannel, RenameChannelResponse),
|
|
(RenameProjectEntry, ProjectEntryResponse),
|
|
(RequestContact, Ack),
|
|
(
|
|
ResolveCompletionDocumentation,
|
|
ResolveCompletionDocumentationResponse
|
|
),
|
|
(ResolveInlayHint, ResolveInlayHintResponse),
|
|
(RespondToChannelInvite, Ack),
|
|
(RespondToContactRequest, Ack),
|
|
(SaveBuffer, BufferSaved),
|
|
(SearchProject, SearchProjectResponse),
|
|
(SendChannelMessage, SendChannelMessageResponse),
|
|
(SetChannelMemberRole, Ack),
|
|
(SetChannelVisibility, Ack),
|
|
(ShareProject, ShareProjectResponse),
|
|
(SynchronizeBuffers, SynchronizeBuffersResponse),
|
|
(TaskContextForLocation, TaskContext),
|
|
(TaskTemplates, TaskTemplatesResponse),
|
|
(Test, Test),
|
|
(UpdateBuffer, Ack),
|
|
(UpdateParticipantLocation, Ack),
|
|
(UpdateProject, Ack),
|
|
(UpdateWorktree, Ack),
|
|
(LspExtExpandMacro, LspExtExpandMacroResponse),
|
|
(SetRoomParticipantRole, Ack),
|
|
(BlameBuffer, BlameBufferResponse),
|
|
(CreateDevServerProject, CreateDevServerProjectResponse),
|
|
(CreateDevServer, CreateDevServerResponse),
|
|
(ShutdownDevServer, Ack),
|
|
(ShareDevServerProject, ShareProjectResponse),
|
|
(JoinDevServerProject, JoinProjectResponse),
|
|
(RejoinRemoteProjects, RejoinRemoteProjectsResponse),
|
|
(ReconnectDevServer, ReconnectDevServerResponse),
|
|
(ValidateDevServerProjectRequest, Ack),
|
|
(MultiLspQuery, MultiLspQueryResponse),
|
|
(DeleteDevServer, Ack),
|
|
(DeleteDevServerProject, Ack),
|
|
(RegenerateDevServerToken, RegenerateDevServerTokenResponse),
|
|
(RenameDevServer, Ack),
|
|
(RestartLanguageServers, Ack),
|
|
(OpenContext, OpenContextResponse),
|
|
(SynchronizeContexts, SynchronizeContextsResponse),
|
|
);
|
|
|
|
entity_messages!(
|
|
{project_id, ShareProject},
|
|
AddProjectCollaborator,
|
|
ApplyCodeAction,
|
|
ApplyCompletionAdditionalEdits,
|
|
BlameBuffer,
|
|
BufferReloaded,
|
|
BufferSaved,
|
|
CopyProjectEntry,
|
|
CreateBufferForPeer,
|
|
CreateProjectEntry,
|
|
DeleteProjectEntry,
|
|
ExpandProjectEntry,
|
|
FormatBuffers,
|
|
GetCodeActions,
|
|
GetCompletions,
|
|
GetDefinition,
|
|
GetImplementation,
|
|
GetDocumentHighlights,
|
|
GetHover,
|
|
GetProjectSymbols,
|
|
GetReferences,
|
|
GetSignatureHelp,
|
|
GetTypeDefinition,
|
|
InlayHints,
|
|
JoinProject,
|
|
LeaveProject,
|
|
LinkedEditingRange,
|
|
MultiLspQuery,
|
|
RestartLanguageServers,
|
|
OnTypeFormatting,
|
|
OpenNewBuffer,
|
|
OpenBufferById,
|
|
OpenBufferByPath,
|
|
OpenBufferForSymbol,
|
|
PerformRename,
|
|
PrepareRename,
|
|
RefreshInlayHints,
|
|
ReloadBuffers,
|
|
RemoveProjectCollaborator,
|
|
RenameProjectEntry,
|
|
ResolveCompletionDocumentation,
|
|
ResolveInlayHint,
|
|
SaveBuffer,
|
|
SearchProject,
|
|
StartLanguageServer,
|
|
SynchronizeBuffers,
|
|
TaskContextForLocation,
|
|
TaskTemplates,
|
|
UnshareProject,
|
|
UpdateBuffer,
|
|
UpdateBufferFile,
|
|
UpdateDiagnosticSummary,
|
|
UpdateDiffBase,
|
|
UpdateLanguageServer,
|
|
UpdateProject,
|
|
UpdateProjectCollaborator,
|
|
UpdateWorktree,
|
|
UpdateWorktreeSettings,
|
|
LspExtExpandMacro,
|
|
AdvertiseContexts,
|
|
OpenContext,
|
|
UpdateContext,
|
|
SynchronizeContexts,
|
|
);
|
|
|
|
entity_messages!(
|
|
{channel_id, Channel},
|
|
ChannelMessageSent,
|
|
ChannelMessageUpdate,
|
|
RemoveChannelMessage,
|
|
UpdateChannelMessage,
|
|
UpdateChannelBuffer,
|
|
UpdateChannelBufferCollaborators,
|
|
);
|
|
|
|
impl From<Timestamp> for SystemTime {
|
|
fn from(val: Timestamp) -> Self {
|
|
UNIX_EPOCH
|
|
.checked_add(Duration::new(val.seconds, val.nanos))
|
|
.unwrap()
|
|
}
|
|
}
|
|
|
|
impl From<SystemTime> for Timestamp {
|
|
fn from(time: SystemTime) -> Self {
|
|
let duration = time.duration_since(UNIX_EPOCH).unwrap();
|
|
Self {
|
|
seconds: duration.as_secs(),
|
|
nanos: duration.subsec_nanos(),
|
|
}
|
|
}
|
|
}
|
|
|
|
impl From<u128> for Nonce {
|
|
fn from(nonce: u128) -> Self {
|
|
let upper_half = (nonce >> 64) as u64;
|
|
let lower_half = nonce as u64;
|
|
Self {
|
|
upper_half,
|
|
lower_half,
|
|
}
|
|
}
|
|
}
|
|
|
|
impl From<Nonce> for u128 {
|
|
fn from(nonce: Nonce) -> Self {
|
|
let upper_half = (nonce.upper_half as u128) << 64;
|
|
let lower_half = nonce.lower_half as u128;
|
|
upper_half | lower_half
|
|
}
|
|
}
|
|
|
|
pub fn split_worktree_update(
|
|
mut message: UpdateWorktree,
|
|
max_chunk_size: usize,
|
|
) -> impl Iterator<Item = UpdateWorktree> {
|
|
let mut done_files = false;
|
|
|
|
let mut repository_map = message
|
|
.updated_repositories
|
|
.into_iter()
|
|
.map(|repo| (repo.work_directory_id, repo))
|
|
.collect::<HashMap<_, _>>();
|
|
|
|
iter::from_fn(move || {
|
|
if done_files {
|
|
return None;
|
|
}
|
|
|
|
let updated_entries_chunk_size = cmp::min(message.updated_entries.len(), max_chunk_size);
|
|
let updated_entries: Vec<_> = message
|
|
.updated_entries
|
|
.drain(..updated_entries_chunk_size)
|
|
.collect();
|
|
|
|
let removed_entries_chunk_size = cmp::min(message.removed_entries.len(), max_chunk_size);
|
|
let removed_entries = message
|
|
.removed_entries
|
|
.drain(..removed_entries_chunk_size)
|
|
.collect();
|
|
|
|
done_files = message.updated_entries.is_empty() && message.removed_entries.is_empty();
|
|
|
|
let mut updated_repositories = Vec::new();
|
|
|
|
if !repository_map.is_empty() {
|
|
for entry in &updated_entries {
|
|
if let Some(repo) = repository_map.remove(&entry.id) {
|
|
updated_repositories.push(repo)
|
|
}
|
|
}
|
|
}
|
|
|
|
let removed_repositories = if done_files {
|
|
mem::take(&mut message.removed_repositories)
|
|
} else {
|
|
Default::default()
|
|
};
|
|
|
|
if done_files {
|
|
updated_repositories.extend(mem::take(&mut repository_map).into_values());
|
|
}
|
|
|
|
Some(UpdateWorktree {
|
|
project_id: message.project_id,
|
|
worktree_id: message.worktree_id,
|
|
root_name: message.root_name.clone(),
|
|
abs_path: message.abs_path.clone(),
|
|
updated_entries,
|
|
removed_entries,
|
|
scan_id: message.scan_id,
|
|
is_last_update: done_files && message.is_last_update,
|
|
updated_repositories,
|
|
removed_repositories,
|
|
})
|
|
})
|
|
}
|
|
|
|
#[cfg(test)]
|
|
mod tests {
|
|
use super::*;
|
|
|
|
#[test]
|
|
fn test_converting_peer_id_from_and_to_u64() {
|
|
let peer_id = PeerId {
|
|
owner_id: 10,
|
|
id: 3,
|
|
};
|
|
assert_eq!(PeerId::from_u64(peer_id.as_u64()), peer_id);
|
|
let peer_id = PeerId {
|
|
owner_id: u32::MAX,
|
|
id: 3,
|
|
};
|
|
assert_eq!(PeerId::from_u64(peer_id.as_u64()), peer_id);
|
|
let peer_id = PeerId {
|
|
owner_id: 10,
|
|
id: u32::MAX,
|
|
};
|
|
assert_eq!(PeerId::from_u64(peer_id.as_u64()), peer_id);
|
|
let peer_id = PeerId {
|
|
owner_id: u32::MAX,
|
|
id: u32::MAX,
|
|
};
|
|
assert_eq!(PeerId::from_u64(peer_id.as_u64()), peer_id);
|
|
}
|
|
}
|