feat(uv): add uv as a pip alternative for dealing with pypi

README.md

@@ -252,6 +252,10 @@ local DEFAULT_SETTINGS = {
         -- Whether to upgrade pip to the latest version in the virtual environment before installing packages.
         upgrade_pip = false,
 
+        ---@since 1.8.0
+        -- Whether to use uv to install packages instead of pip
+        use_uv      = false,
+
         ---@since 1.0.0
         -- These args will be added to `pip install` calls. Note that setting extra args might impact intended behavior
         -- and is not recommended.

doc/mason.txt

@@ -314,6 +314,10 @@ Example:
             -- Whether to upgrade pip to the latest version in the virtual environment before installing packages.
             upgrade_pip = false,
 
+            ---@since 1.8.0
+            -- Whether to use uv to install packages instead of pip
+            use_uv      = false,
+
             ---@since 1.0.0
             -- These args will be added to `pip install` calls. Note that setting extra args might impact intended behavior
             -- and is not recommended.

lua/mason-core/installer/managers/pypi.lua

@@ -7,23 +7,36 @@ local log = require "mason-core.log"
 local path = require "mason-core.path"
 local platform = require "mason-core.platform"
 local semver = require "mason-core.semver"
+local settings = require "mason.settings"
 local spawn = require "mason-core.spawn"
 
 local M = {}
 
+local use_uv = settings.current.pip.use_uv
 local VENV_DIR = "venv"
+if use_uv then
+    VENV_DIR = ".venv"
+end
 
 local is_executable = _.compose(_.equals(1), vim.fn.executable)
 
 ---@async
 ---@param candidates string[]
 local function resolve_python3(candidates)
     a.scheduler()
+    if use_uv then
+        candidates = { "uv" }
+    end
     local available_candidates = _.filter(is_executable, candidates)
     for __, candidate in ipairs(available_candidates) do
         ---@type string
         local version_output = spawn[candidate]({ "--version" }):map(_.prop "stdout"):get_or_else ""
-        local ok, version = pcall(semver.new, version_output:match "Python (3%.%d+.%d+)")
+        local ok, version
+        if use_uv then
+            ok, version = pcall(semver.new, version_output:match "uv (%d+.%d+.%d+)")
+        else
+            ok, version = pcall(semver.new, version_output:match "Python (3%.%d+.%d+)")
+        end
         if ok then
             return { executable = candidate, version = version }
         end
@@ -70,9 +83,14 @@ local function create_venv()
         )
         return Result.failure "Failed to find python3 installation."
     end
-    log.fmt_debug("Found python3 installation version=%s, executable=%s", target.version, target.executable)
     ctx.stdio_sink.stdout "Creating virtual environment…\n"
-    return ctx.spawn[target.executable] { "-m", "venv", VENV_DIR }
+    if use_uv then
+        log.fmt_debug("Found uv installation version=%s, executable=%s", target.version, target.executable)
+        return ctx.spawn[target.executable] { "venv", VENV_DIR }
+    else
+        log.fmt_debug("Found python3 installation version=%s, executable=%s", target.version, target.executable)
+        return ctx.spawn[target.executable] { "-m", "venv", VENV_DIR }
+    end
 end
 
 ---@param ctx InstallContext
@@ -107,15 +125,28 @@ end
 ---@param pkgs string[]
 ---@param extra_args? string[]
 local function pip_install(pkgs, extra_args)
-    return venv_python {
-        "-m",
-        "pip",
-        "--disable-pip-version-check",
-        "install",
-        "-U",
-        extra_args or vim.NIL,
-        pkgs,
-    }
+    if use_uv then
+        local ctx = installer.context()
+        local task = ctx.spawn["uv"] {
+            "pip",
+            "install",
+            "-U",
+            extra_args or vim.NIL,
+            pkgs,
+        }
+        -- vim.api.nvim_set_current_dir(curdir)
+        return task
+    else
+        return venv_python {
+            "-m",
+            "pip",
+            "--disable-pip-version-check",
+            "install",
+            "-U",
+            extra_args or vim.NIL,
+            pkgs,
+        }
+    end
 end
 
 ---@async
@@ -129,7 +160,7 @@ function M.init(opts)
         ctx:promote_cwd()
         try(create_venv())
 
-        if opts.upgrade_pip then
+        if opts.upgrade_pip and not use_uv then
             ctx.stdio_sink.stdout "Upgrading pip inside the virtual environment…\n"
             try(pip_install({ "pip" }, opts.install_extra_args))
         end

lua/mason-core/installer/registry/providers/pypi.lua

@@ -27,6 +27,7 @@ function M.parse(source, purl)
             pip = {
                 upgrade = settings.current.pip.upgrade_pip,
                 extra_args = settings.current.pip.install_args,
+                use_uv = settings.current.pip.use_uv,
             },
         }
 
@@ -44,11 +45,13 @@ function M.install(ctx, source)
         try(pypi.init {
             upgrade_pip = source.pip.upgrade,
             install_extra_args = source.pip.extra_args,
+            use_uv = source.pip.use_uv,
         })
         try(pypi.install(source.package, source.version, {
             extra = source.extra,
             extra_packages = source.extra_packages,
             install_extra_args = source.pip.extra_args,
+            use_uv = source.pip.use_uv,
         }))
     end)
 end

lua/mason/settings.lua

@@ -60,6 +60,10 @@ local DEFAULT_SETTINGS = {
         -- Whether to upgrade pip to the latest version in the virtual environment before installing packages.
         upgrade_pip = false,
 
+        ---@since 1.8.0
+        -- Whether to use uv to install packages instead of pip
+        use_uv = false,
+
         ---@since 1.0.0
         -- These args will be added to `pip install` calls. Note that setting extra args might impact intended behavior
         -- and is not recommended.

tests/mason-core/installer/registry/providers/pypi_spec.lua

@@ -31,6 +31,7 @@ describe("pypi provider :: parsing", function()
                 pip = {
                     upgrade = true,
                     extra_args = { "--proxy", "http://localghost" },
+                    use_uv = false,
                 },
             },
             pypi.parse({ extra_packages = { "extra" } }, purl())