wip

Author: Cole Miller

Cargo.lock

@@ -66,7 +66,7 @@ pub use environment::ProjectEnvironment;
 #[cfg(test)]
 use futures::future::join_all;
 use futures::{
-    StreamExt,
+    FutureExt as _, StreamExt,
     channel::mpsc::{self, UnboundedReceiver},
     future::{Shared, try_join_all},
 };
@@ -1894,12 +1894,32 @@ impl Project {
         })
     }
 
+    // FIXME migrate project environment stuff to use proto instead of bailing when not local
+
     pub fn directory_environment(
         &self,
         shell: &Shell,
         abs_path: Arc<Path>,
         cx: &mut App,
     ) -> Shared<Task<Option<HashMap<String, String>>>> {
+        if let Some(remote_client) = self.remote_client() {
+            let response =
+                remote_client
+                    .read(cx)
+                    .proto_client()
+                    .request(proto::GetDirectoryEnvironment {
+                        // FIXME
+                        project_id: REMOTE_SERVER_PROJECT_ID,
+                        shell: Some(shell.clone().to_proto()),
+                        directory: abs_path.to_string_lossy().to_string(),
+                    });
+            return cx
+                .spawn(async move |_| {
+                    let environment = response.await.log_err()?;
+                    Some(environment.environment.into_iter().collect())
+                })
+                .shared();
+        }
         self.environment.update(cx, |environment, cx| {
             environment.get_directory_environment_for_shell(shell, abs_path, cx)
         })
@@ -4963,21 +4983,6 @@ impl Project {
         Project::respond_to_open_buffer_request(this, buffer, peer_id, &mut cx)
     }
 
-    async fn handle_get_directory_environment(
-        this: Entity<Self>,
-        envelope: TypedEnvelope<proto::GetDirectoryEnvironment>,
-        mut cx: AsyncApp,
-    ) -> Result<proto::DirectoryEnvironment> {
-        let shell = Shell::from_proto(envelope.payload.shell.context("missing shell")?)?;
-        let directory = PathBuf::from(envelope.payload.directory);
-        let environment = this
-            .update(&mut cx, |this, cx| {
-                this.directory_environment(shell, directory.into(), cx)
-            })?
-            .await?;
-        Ok(proto::DirectoryEnvironment { environment })
-    }
-
     fn respond_to_open_buffer_request(
         this: Entity<Self>,
         buffer: Entity<Buffer>,

crates/project/src/project.rs

@@ -56,5 +56,5 @@ message GetDirectoryEnvironment {
 }
 
 message DirectoryEnvironment {
-    map<string, string> env = 1;
+    map<string, string> environment = 1;
 }

crates/proto/proto/task.proto

@@ -418,7 +418,10 @@ message Envelope {
 
         GitRenameBranch git_rename_branch = 380;
 
-        RemoteStarted remote_started = 381; // current max
+        RemoteStarted remote_started = 381;
+
+        GetDirectoryEnvironment get_directory_environment = 382;
+        DirectoryEnvironment directory_environment = 383; // current max
     }
 
     reserved 87 to 88;

crates/proto/proto/zed.proto

@@ -319,6 +319,8 @@ messages!(
     (GitClone, Background),
     (GitCloneResponse, Background),
     (ToggleLspLogs, Background),
+    (GetDirectoryEnvironment, Background),
+    (DirectoryEnvironment, Background),
     (GetAgentServerCommand, Background),
     (AgentServerCommand, Background),
     (ExternalAgentsUpdated, Background),
@@ -497,6 +499,7 @@ request_messages!(
     (GetDefaultBranch, GetDefaultBranchResponse),
     (GitClone, GitCloneResponse),
     (ToggleLspLogs, Ack),
+    (GetDirectoryEnvironment, DirectoryEnvironment),
     (GetProcesses, GetProcessesResponse),
     (GetAgentServerCommand, AgentServerCommand),
     (RemoteStarted, Ack),
@@ -634,6 +637,7 @@ entity_messages!(
     GitCheckoutFiles,
     SetIndexText,
     ToggleLspLogs,
+    GetDirectoryEnvironment,
 
     Push,
     Fetch,

crates/proto/src/proto.rs

@@ -60,6 +60,7 @@ settings.workspace = true
 shellexpand.workspace = true
 smol.workspace = true
 sysinfo.workspace = true
+task.workspace = true
 util.workspace = true
 watch.workspace = true
 worktree.workspace = true

crates/remote_server/Cargo.toml

@@ -50,6 +50,7 @@ pub struct HeadlessProject {
     pub languages: Arc<LanguageRegistry>,
     pub extensions: Entity<HeadlessExtensionStore>,
     pub git_store: Entity<GitStore>,
+    pub environment: Entity<ProjectEnvironment>,
     // Used mostly to keep alive the toolchain store for RPC handlers.
     // Local variant is used within LSP store, but that's a separate entity.
     pub _toolchain_store: Entity<ToolchainStore>,
@@ -196,7 +197,7 @@ impl HeadlessProject {
 
         let agent_server_store = cx.new(|cx| {
             let mut agent_server_store =
-                AgentServerStore::local(node_runtime.clone(), fs.clone(), environment, cx);
+                AgentServerStore::local(node_runtime.clone(), fs.clone(), environment.clone(), cx);
             agent_server_store.shared(REMOTE_SERVER_PROJECT_ID, session.clone(), cx);
             agent_server_store
         });
@@ -249,6 +250,7 @@ impl HeadlessProject {
         session.add_entity_request_handler(Self::handle_open_new_buffer);
         session.add_entity_request_handler(Self::handle_find_search_candidates);
         session.add_entity_request_handler(Self::handle_open_server_settings);
+        session.add_entity_request_handler(Self::handle_get_directory_environment);
         session.add_entity_message_handler(Self::handle_toggle_lsp_logs);
 
         session.add_entity_request_handler(BufferStore::handle_update_buffer);
@@ -289,6 +291,7 @@ impl HeadlessProject {
             languages,
             extensions,
             git_store,
+            environment,
             _toolchain_store: toolchain_store,
         }
     }
@@ -758,6 +761,26 @@ impl HeadlessProject {
 
         Ok(proto::GetProcessesResponse { processes })
     }
+
+    async fn handle_get_directory_environment(
+        this: Entity<Self>,
+        envelope: TypedEnvelope<proto::GetDirectoryEnvironment>,
+        mut cx: AsyncApp,
+    ) -> Result<proto::DirectoryEnvironment> {
+        let shell = task::Shell::from_proto(envelope.payload.shell.context("missing shell")?)?;
+        let directory = PathBuf::from(envelope.payload.directory);
+        let environment = this
+            .update(&mut cx, |this, cx| {
+                this.environment.update(cx, |environment, cx| {
+                    environment.get_directory_environment_for_shell(&shell, directory.into(), cx)
+                })
+            })?
+            .await
+            .context("failed to get directory environment")?
+            .into_iter()
+            .collect();
+        Ok(proto::DirectoryEnvironment { environment })
+    }
 }
 
 fn prompt_to_proto(