From c7163b63dbf67af145437520d582e5849d5b1484 Mon Sep 17 00:00:00 2001
From: Gideon <87426140+GideonBear@users.noreply.github.com>
Date: Tue, 10 Jun 2025 12:10:18 +0200
Subject: [PATCH] Run `uv cache prune` in `uv` step when `cleanup` is enabled
 (#1165)

Add uv cache prune
---
 src/steps/generic.rs | 12 +++++++++++-
 1 file changed, 11 insertions(+), 1 deletion(-)

diff --git a/src/steps/generic.rs b/src/steps/generic.rs
index 4e7e9796..15bbd991 100644
--- a/src/steps/generic.rs
+++ b/src/steps/generic.rs
@@ -1439,7 +1439,17 @@ pub fn run_uv(ctx: &ExecutionContext) -> Result<()> {
     ctx.run_type()
         .execute(&uv_exec)
         .args(["tool", "upgrade", "--all"])
-        .status_checked()
+        .status_checked()?;
+
+    if ctx.config().cleanup() {
+        // 3. Prune cache
+        ctx.run_type()
+            .execute(&uv_exec)
+            .args(["cache", "prune"])
+            .status_checked()?;
+    }
+
+    Ok(())
 }
 
 /// Involve `zvm upgrade` to update ZVM