use ::proto::{FromProto, ToProto};use anyhow::{Context as _, Result, anyhow}; use extension::ExtensionHostProxy;use extension_host::headless_host::HeadlessExtensionStore;use fs::Fs;use gpui::{App, AppContext as _, AsyncApp, Context, Entity, PromptLevel};use http_client::HttpClient;use language::{Buffer, BufferEvent, LanguageRegistry, proto::serialize_operation};use node_runtime::NodeRuntime;use project::{    LspStore, LspStoreEvent, ManifestTree, PrettierStore, ProjectEnvironment, ProjectPath,    ToolchainStore, WorktreeId,    buffer_store::{BufferStore, BufferStoreEvent},    debugger::{breakpoint_store::BreakpointStore, dap_store::DapStore},    git_store::GitStore,    project_settings::SettingsObserver,    search::SearchQuery,    task_store::TaskStore,    worktree_store::WorktreeStore,};use remote::ssh_session::ChannelClient;use rpc::{    AnyProtoClient, TypedEnvelope,    proto::{self, SSH_PEER_ID, SSH_PROJECT_ID},}; use settings::initial_server_settings_content;use smol::stream::StreamExt;use std::{    path::{Path, PathBuf},    sync::{Arc, atomic::AtomicUsize},};use util::ResultExt;use worktree::Worktree; pub struct HeadlessProject {    pub fs: Arc<dyn Fs>,    pub session: AnyProtoClient,    pub worktree_store: Entity<WorktreeStore>,    pub buffer_store: Entity<BufferStore>,    pub lsp_store: Entity<LspStore>,    pub task_store: Entity<TaskStore>,    pub dap_store: Entity<DapStore>,    pub settings_observer: Entity<SettingsObserver>,    pub next_entry_id: Arc<AtomicUsize>,    pub languages: Arc<LanguageRegistry>,    pub extensions: Entity<HeadlessExtensionStore>,    pub git_store: Entity<GitStore>,} pub struct HeadlessAppState {    pub session: Arc<ChannelClient>,    pub fs: Arc<dyn Fs>,    pub http_client: Arc<dyn HttpClient>,    pub node_runtime: NodeRuntime,    pub languages: Arc<LanguageRegistry>,    pub extension_host_proxy: Arc<ExtensionHostProxy>,} impl HeadlessProject {    pub fn init(cx: &mut App) {        settings::init(cx);        language::init(cx);        project::Project::init_settings(cx);    }     pub fn new(        HeadlessAppState {            session,            fs,            http_client,            node_runtime,            languages,            extension_host_proxy: proxy,        }: HeadlessAppState,        cx: &mut Context<Self>,    ) -> Self {        debug_adapter_extension::init(proxy.clone(), cx);        language_extension::init(proxy.clone(), languages.clone());        languages::init(languages.clone(), node_runtime.clone(), cx);         let worktree_store = cx.new(|cx| {            let mut store = WorktreeStore::local(true, fs.clone());            store.shared(SSH_PROJECT_ID, session.clone().into(), cx);            store        });         let environment = cx.new(|_| ProjectEnvironment::new(None));        let manifest_tree = ManifestTree::new(worktree_store.clone(), cx);        let toolchain_store = cx.new(|cx| {            ToolchainStore::local(                languages.clone(),                worktree_store.clone(),                environment.clone(),                manifest_tree.clone(),                cx,            )        });         let buffer_store = cx.new(|cx| {            let mut buffer_store = BufferStore::local(worktree_store.clone(), cx);            buffer_store.shared(SSH_PROJECT_ID, session.clone().into(), cx);            buffer_store        });         let breakpoint_store =            cx.new(|_| BreakpointStore::local(worktree_store.clone(), buffer_store.clone()));         let dap_store = cx.new(|cx| {            let mut dap_store = DapStore::new_local(                http_client.clone(),                node_runtime.clone(),                fs.clone(),                environment.clone(),                toolchain_store.read(cx).as_language_toolchain_store(),                worktree_store.clone(),                breakpoint_store.clone(),                cx,            );            dap_store.shared(SSH_PROJECT_ID, session.clone().into(), cx);            dap_store        });         let git_store = cx.new(|cx| {            let mut store = GitStore::local(                &worktree_store,                buffer_store.clone(),                environment.clone(),                fs.clone(),                cx,            );            store.shared(SSH_PROJECT_ID, session.clone().into(), cx);            store        });         let prettier_store = cx.new(|cx| {            PrettierStore::new(                node_runtime.clone(),                fs.clone(),                languages.clone(),                worktree_store.clone(),                cx,            )        });         let task_store = cx.new(|cx| {            let mut task_store = TaskStore::local(                fs.clone(),                buffer_store.downgrade(),                worktree_store.clone(),                toolchain_store.read(cx).as_language_toolchain_store(),                environment.clone(),                cx,            );            task_store.shared(SSH_PROJECT_ID, session.clone().into(), cx);            task_store        });        let settings_observer = cx.new(|cx| {            let mut observer = SettingsObserver::new_local(                fs.clone(),                worktree_store.clone(),                task_store.clone(),                cx,            );            observer.shared(SSH_PROJECT_ID, session.clone().into(), cx);            observer        });         let lsp_store = cx.new(|cx| {            let mut lsp_store = LspStore::new_local(                buffer_store.clone(),                worktree_store.clone(),                prettier_store.clone(),                toolchain_store.clone(),                environment,                manifest_tree,                languages.clone(),                http_client.clone(),                fs.clone(),                cx,            );            lsp_store.shared(SSH_PROJECT_ID, session.clone().into(), cx);            lsp_store        });         cx.subscribe(&lsp_store, Self::on_lsp_store_event).detach();         cx.subscribe(            &buffer_store,            |_this, _buffer_store, event, cx| match event {                BufferStoreEvent::BufferAdded(buffer) => {                    cx.subscribe(buffer, Self::on_buffer_event).detach();                }                _ => {}            },        )        .detach();         let extensions = HeadlessExtensionStore::new(            fs.clone(),            http_client.clone(),            paths::remote_extensions_dir().to_path_buf(),            proxy,            node_runtime,            cx,        );         let client: AnyProtoClient = session.clone().into();         // local_machine -> ssh handlers        session.subscribe_to_entity(SSH_PROJECT_ID, &worktree_store);        session.subscribe_to_entity(SSH_PROJECT_ID, &buffer_store);        session.subscribe_to_entity(SSH_PROJECT_ID, &cx.entity());        session.subscribe_to_entity(SSH_PROJECT_ID, &lsp_store);        session.subscribe_to_entity(SSH_PROJECT_ID, &task_store);        session.subscribe_to_entity(SSH_PROJECT_ID, &toolchain_store);        session.subscribe_to_entity(SSH_PROJECT_ID, &dap_store);        session.subscribe_to_entity(SSH_PROJECT_ID, &settings_observer);        session.subscribe_to_entity(SSH_PROJECT_ID, &git_store);         client.add_request_handler(cx.weak_entity(), Self::handle_list_remote_directory);        client.add_request_handler(cx.weak_entity(), Self::handle_get_path_metadata);        client.add_request_handler(cx.weak_entity(), Self::handle_shutdown_remote_server);        client.add_request_handler(cx.weak_entity(), Self::handle_ping);         client.add_entity_request_handler(Self::handle_add_worktree);        client.add_request_handler(cx.weak_entity(), Self::handle_remove_worktree);         client.add_entity_request_handler(Self::handle_open_buffer_by_path);        client.add_entity_request_handler(Self::handle_open_new_buffer);        client.add_entity_request_handler(Self::handle_find_search_candidates);        client.add_entity_request_handler(Self::handle_open_server_settings);         client.add_entity_request_handler(BufferStore::handle_update_buffer);        client.add_entity_message_handler(BufferStore::handle_close_buffer);         client.add_request_handler(            extensions.clone().downgrade(),            HeadlessExtensionStore::handle_sync_extensions,        );        client.add_request_handler(            extensions.clone().downgrade(),            HeadlessExtensionStore::handle_install_extension,        );         BufferStore::init(&client);        WorktreeStore::init(&client);        SettingsObserver::init(&client);        LspStore::init(&client);        TaskStore::init(Some(&client));        ToolchainStore::init(&client);        DapStore::init(&client, cx);        // todo(debugger): Re init breakpoint store when we set it up for collab        // BreakpointStore::init(&client);        GitStore::init(&client);         HeadlessProject {            session: client,            settings_observer,            fs,            worktree_store,            buffer_store,            lsp_store,            task_store,            dap_store,            next_entry_id: Default::default(),            languages,            extensions,            git_store,        }    }     fn on_buffer_event(        &mut self,        buffer: Entity<Buffer>,        event: &BufferEvent,        cx: &mut Context<Self>,    ) {        match event {            BufferEvent::Operation {                operation,                is_local: true,            } => cx                .background_spawn(self.session.request(proto::UpdateBuffer {                    project_id: SSH_PROJECT_ID,                    buffer_id: buffer.read(cx).remote_id().to_proto(),                    operations: vec![serialize_operation(operation)],                }))                .detach(),            _ => {}        }    }     fn on_lsp_store_event(        &mut self,        _lsp_store: Entity<LspStore>,        event: &LspStoreEvent,        cx: &mut Context<Self>,    ) {        match event {            LspStoreEvent::LanguageServerUpdate {                language_server_id,                name,                message,            } => {                self.session                    .send(proto::UpdateLanguageServer {                        project_id: SSH_PROJECT_ID,                        server_name: name.as_ref().map(|name| name.to_string()),                        language_server_id: language_server_id.to_proto(),                        variant: Some(message.clone()),                    })                    .log_err();            }            LspStoreEvent::Notification(message) => {                self.session                    .send(proto::Toast {                        project_id: SSH_PROJECT_ID,                        notification_id: "lsp".to_string(),                        message: message.clone(),                    })                    .log_err();            }            LspStoreEvent::LanguageServerLog(language_server_id, log_type, message) => {                self.session                    .send(proto::LanguageServerLog {                        project_id: SSH_PROJECT_ID,                        language_server_id: language_server_id.to_proto(),                        message: message.clone(),                        log_type: Some(log_type.to_proto()),                    })                    .log_err();            }            LspStoreEvent::LanguageServerPrompt(prompt) => {                let request = self.session.request(proto::LanguageServerPromptRequest {                    project_id: SSH_PROJECT_ID,                    actions: prompt                        .actions                        .iter()                        .map(|action| action.title.to_string())                        .collect(),                    level: Some(prompt_to_proto(&prompt)),                    lsp_name: prompt.lsp_name.clone(),                    message: prompt.message.clone(),                });                let prompt = prompt.clone();                cx.background_spawn(async move {                    let response = request.await?;                    if let Some(action_response) = response.action_response {                        prompt.respond(action_response as usize).await;                    }                    anyhow::Ok(())                })                .detach();            }            _ => {}        }    }     pub async fn handle_add_worktree(        this: Entity<Self>,        message: TypedEnvelope<proto::AddWorktree>,        mut cx: AsyncApp,    ) -> Result<proto::AddWorktreeResponse> {        use client::ErrorCodeExt;        let fs = this.read_with(&mut cx, |this, _| this.fs.clone())?;        let path = PathBuf::from_proto(shellexpand::tilde(&message.payload.path).to_string());         let canonicalized = match fs.canonicalize(&path).await {            Ok(path) => path,            Err(e) => {                let mut parent = path                    .parent()                    .ok_or(e)                    .with_context(|| format!("{path:?} does not exist"))?;                if parent == Path::new("") {                    parent = util::paths::home_dir();                }                let parent = fs.canonicalize(parent).await.map_err(|_| {                    anyhow!(                        proto::ErrorCode::DevServerProjectPathDoesNotExist                            .with_tag("path", &path.to_string_lossy().as_ref())                    )                })?;                parent.join(path.file_name().unwrap())            }        };         let worktree = this            .read_with(&mut cx.clone(), |this, _| {                Worktree::local(                    Arc::from(canonicalized.as_path()),                    message.payload.visible,                    this.fs.clone(),                    this.next_entry_id.clone(),                    &mut cx,                )            })?            .await?;         let response = this.read_with(&mut cx, |_, cx| {            let worktree = worktree.read(cx);            proto::AddWorktreeResponse {                worktree_id: worktree.id().to_proto(),                canonicalized_path: canonicalized.to_proto(),            }        })?;         // We spawn this asynchronously, so that we can send the response back        // *before* `worktree_store.add()` can send out UpdateProject requests        // to the client about the new worktree.        //        // That lets the client manage the reference/handles of the newly-added        // worktree, before getting interrupted by an UpdateProject request.        //        // This fixes the problem of the client sending the AddWorktree request,        // headless project sending out a project update, client receiving it        // and immediately dropping the reference of the new client, causing it        // to be dropped on the headless project, and the client only then        // receiving a response to AddWorktree.        cx.spawn(async move |cx| {            this.update(cx, |this, cx| {                this.worktree_store.update(cx, |worktree_store, cx| {                    worktree_store.add(&worktree, cx);                });            })            .log_err();        })        .detach();         Ok(response)    }     pub async fn handle_remove_worktree(        this: Entity<Self>,        envelope: TypedEnvelope<proto::RemoveWorktree>,        mut cx: AsyncApp,    ) -> Result<proto::Ack> {        let worktree_id = WorktreeId::from_proto(envelope.payload.worktree_id);        this.update(&mut cx, |this, cx| {            this.worktree_store.update(cx, |worktree_store, cx| {                worktree_store.remove_worktree(worktree_id, cx);            });        })?;        Ok(proto::Ack {})    }     pub async fn handle_open_buffer_by_path(        this: Entity<Self>,        message: TypedEnvelope<proto::OpenBufferByPath>,        mut cx: AsyncApp,    ) -> Result<proto::OpenBufferResponse> {        let worktree_id = WorktreeId::from_proto(message.payload.worktree_id);        let (buffer_store, buffer) = this.update(&mut cx, |this, cx| {            let buffer_store = this.buffer_store.clone();            let buffer = this.buffer_store.update(cx, |buffer_store, cx| {                buffer_store.open_buffer(                    ProjectPath {                        worktree_id,                        path: Arc::<Path>::from_proto(message.payload.path),                    },                    cx,                )            });            anyhow::Ok((buffer_store, buffer))        })??;         let buffer = buffer.await?;        let buffer_id = buffer.read_with(&cx, |b, _| b.remote_id())?;        buffer_store.update(&mut cx, |buffer_store, cx| {            buffer_store                .create_buffer_for_peer(&buffer, SSH_PEER_ID, cx)                .detach_and_log_err(cx);        })?;         Ok(proto::OpenBufferResponse {            buffer_id: buffer_id.to_proto(),        })    }     pub async fn handle_open_new_buffer(        this: Entity<Self>,        _message: TypedEnvelope<proto::OpenNewBuffer>,        mut cx: AsyncApp,    ) -> Result<proto::OpenBufferResponse> {        let (buffer_store, buffer) = this.update(&mut cx, |this, cx| {            let buffer_store = this.buffer_store.clone();            let buffer = this                .buffer_store                .update(cx, |buffer_store, cx| buffer_store.create_buffer(cx));            anyhow::Ok((buffer_store, buffer))        })??;         let buffer = buffer.await?;        let buffer_id = buffer.read_with(&cx, |b, _| b.remote_id())?;        buffer_store.update(&mut cx, |buffer_store, cx| {            buffer_store                .create_buffer_for_peer(&buffer, SSH_PEER_ID, cx)                .detach_and_log_err(cx);

Lines 1–500 of 685. Load more