From 22029a657c6c9f44ba26e4dad8c4e45d15da405f Mon Sep 17 00:00:00 2001
From: jdx <216188+jdx@users.noreply.github.com>
Date: Wed, 27 Nov 2024 14:33:13 -0600
Subject: [PATCH] fix: automatic reinstall of uvx tools during python upgrades

---
 docs/dev-tools/backends/pipx.md |  2 +
 e2e/backend/test_pipx_uvx       |  6 ++-
 settings.toml                   |  1 +
 src/backend/pipx.rs             | 73 +++++++++++++++++++++------------
 4 files changed, 54 insertions(+), 28 deletions(-)

diff --git a/docs/dev-tools/backends/pipx.md b/docs/dev-tools/backends/pipx.md
index eeb234964b..f57a77a092 100644
--- a/docs/dev-tools/backends/pipx.md
+++ b/docs/dev-tools/backends/pipx.md
@@ -54,6 +54,8 @@ Or you can reinstall all pipx packages with:
 mise install -f "pipx:*"
 ```
 
+mise _should_ do this automatically when using `mise up python`.
+
 ### Supported Pipx Syntax
 
 | Description                           | Usage                                                  |
diff --git a/e2e/backend/test_pipx_uvx b/e2e/backend/test_pipx_uvx
index 426afbce01..8ef33c57bb 100644
--- a/e2e/backend/test_pipx_uvx
+++ b/e2e/backend/test_pipx_uvx
@@ -20,8 +20,9 @@ export MISE_PIPX_UVX=1
 # Set up a 2-step installation: pipx@1.5.0 > pipx:mkdocs@1.6.0
 cat >.mise.toml <<EOF
 [tools]
+python = "3.12.3"
 pipx = "1.5.0"
-uv = "0.3.4"
+uv = "0.5.5"
 "pipx:mkdocs" = "1.6.0"
 EOF
 
@@ -31,3 +32,6 @@ mise install
 # Assert that mkdocs 1.6.0 has been installed with pipx
 # (mkdocs conveniently returns its installation path in with --version)
 assert_contains "mise x -- mkdocs --version" "/mise/installs/pipx-mkdocs/1.6.0/"
+
+assert "mise up --bump python"
+assert_contains "mise x -- mkdocs --version" "mkdocs, version 1.6.0"
diff --git a/settings.toml b/settings.toml
index 041ebada45..4643d8126c 100644
--- a/settings.toml
+++ b/settings.toml
@@ -487,6 +487,7 @@ passing `--fuzzy` on the command line.
 """
 
 [pipx.uvx]
+env = "MISE_PIPX_UVX"
 type = "Bool"
 description = "Use uvx instead of pipx if uv is installed and on PATH."
 docs = """
diff --git a/src/backend/pipx.rs b/src/backend/pipx.rs
index 613431ecde..ed3b402139 100644
--- a/src/backend/pipx.rs
+++ b/src/backend/pipx.rs
@@ -92,18 +92,14 @@ impl Backend for PIPXBackend {
             .pipx_request(&tv.version, &tv.request.options());
 
         if SETTINGS.pipx.uvx {
-            let mut cmd = CmdLineRunner::new("uv")
-                .arg("tool")
-                .arg("install")
-                .arg(pipx_request)
-                .with_pr(ctx.pr.as_ref())
-                .env("UV_TOOL_DIR", tv.install_path())
-                .env("UV_TOOL_BIN_DIR", tv.install_path().join("bin"))
-                .envs(ctx.ts.env_with_path(&config)?)
-                .prepend_path(ctx.ts.list_paths())?
-                // Prepend install path so pipx doesn't issue a warning about missing path
-                .prepend_path(vec![tv.install_path().join("bin")])?
-                .prepend_path(self.dependency_toolset()?.list_paths())?;
+            let mut cmd = Self::uvx_cmd(
+                &config,
+                &["tool", "install", &pipx_request],
+                self,
+                &tv,
+                ctx.ts,
+                &*ctx.pr,
+            )?;
             if let Some(args) = tv.request.options().get("uvx_args") {
                 cmd = cmd.args(shell_words::split(args)?);
             }
@@ -139,10 +135,6 @@ impl PIPXBackend {
     }
 
     pub fn reinstall_all() -> Result<()> {
-        if SETTINGS.pipx.uvx {
-            debug!("skipping pipx reinstall because uvx is enabled");
-            return Ok(());
-        }
         let config = Config::load()?;
         let ts = ToolsetBuilder::new().build(&config)?;
         let pipx_tools = ts
@@ -150,21 +142,48 @@ impl PIPXBackend {
             .into_iter()
             .filter(|(b, _tv)| b.ba().backend_type() == BackendType::Pipx)
             .collect_vec();
-        let pr = MultiProgressReport::get().add("reinstalling pipx tools");
-        for (b, tv) in pipx_tools {
-            Self::pipx_cmd(
-                &config,
-                &["reinstall", &tv.ba().tool_name],
-                &*b,
-                &tv,
-                &ts,
-                &*pr,
-            )?
-            .execute()?;
+        if SETTINGS.pipx.uvx {
+            let pr = MultiProgressReport::get().add("reinstalling pipx tools with uvx");
+            for (b, tv) in pipx_tools {
+                for (cmd, tool) in &[
+                    ("uninstall", tv.ba().tool_name.to_string()),
+                    ("install", format!("{}=={}", tv.ba().tool_name, tv.version)),
+                ] {
+                    let args = &["tool", cmd, tool];
+                    Self::uvx_cmd(&config, args, &*b, &tv, &ts, &*pr)?.execute()?;
+                }
+            }
+        } else {
+            let pr = MultiProgressReport::get().add("reinstalling pipx tools");
+            for (b, tv) in pipx_tools {
+                let args = &["reinstall", &tv.ba().tool_name];
+                Self::pipx_cmd(&config, args, &*b, &tv, &ts, &*pr)?.execute()?;
+            }
         }
         Ok(())
     }
 
+    fn uvx_cmd<'a>(
+        config: &Config,
+        args: &[&str],
+        b: &dyn Backend,
+        tv: &ToolVersion,
+        ts: &Toolset,
+        pr: &'a dyn SingleReport,
+    ) -> Result<CmdLineRunner<'a>> {
+        let mut cmd = CmdLineRunner::new("uv");
+        for arg in args {
+            cmd = cmd.arg(arg);
+        }
+        cmd.with_pr(pr)
+            .env("UV_TOOL_DIR", tv.install_path())
+            .env("UV_TOOL_BIN_DIR", tv.install_path().join("bin"))
+            .envs(ts.env_with_path(config)?)
+            .prepend_path(ts.list_paths())?
+            .prepend_path(vec![tv.install_path().join("bin")])?
+            .prepend_path(b.dependency_toolset()?.list_paths())
+    }
+
     fn pipx_cmd<'a>(
         config: &Config,
         args: &[&str],