From 02e96821eb69549e16a9ded70669dab284079fa5 Mon Sep 17 00:00:00 2001
From: bestgopher <84328409@qq.com>
Date: Tue, 3 Sep 2024 21:56:47 +0800
Subject: [PATCH] extension: Delete working directory on uninstall (#17127)

Closes #17126

Release Notes:

- N/A
---
 crates/extension/src/extension_store.rs | 10 ++++++++++
 1 file changed, 10 insertions(+)

diff --git a/crates/extension/src/extension_store.rs b/crates/extension/src/extension_store.rs
index 35ee7747d9..66d29a26f2 100644
--- a/crates/extension/src/extension_store.rs
+++ b/crates/extension/src/extension_store.rs
@@ -772,6 +772,7 @@ impl ExtensionStore {
 
     pub fn uninstall_extension(&mut self, extension_id: Arc<str>, cx: &mut ModelContext<Self>) {
         let extension_dir = self.installed_dir.join(extension_id.as_ref());
+        let work_dir = self.wasm_host.work_dir.join(extension_id.as_ref());
         let fs = self.fs.clone();
 
         match self.outstanding_operations.entry(extension_id.clone()) {
@@ -793,6 +794,15 @@ impl ExtensionStore {
                 }
             });
 
+            fs.remove_dir(
+                &work_dir,
+                RemoveOptions {
+                    recursive: true,
+                    ignore_if_not_exists: true,
+                },
+            )
+            .await?;
+
             fs.remove_dir(
                 &extension_dir,
                 RemoveOptions {