add ignore_envs_groups parametr

Author: palandovalex

README.md

@@ -27,6 +27,26 @@ require('swenv.api').pick_venv()
 to pick an environment. Uses `vim.ui.select` so a tip is to use eg
 [dressing.nvim](https://github.com/stevearc/dressing.nvim).
 
+### Supported envs
+
+Swenv supports various env types:
+Conda, pixi, micromamba, pyenv... For some envs types, such as poetry, you can specify env path: 
+
+```lua
+require('swenv').setup({
+  venvs_path = vim.fn.expand('~/.cache/pypoetry/virtualenvs'),
+})
+```
+
+If you havent some envs group, you can disable they, to save startup time:
+
+```lua
+require('swenv').setup({
+  ignore_envs_groups = {'conda', 'pixi', 'micromamba', 'pyenv'}
+})
+```
+
+
 ### Get Environment
 
 To show the current venv in for example a status-line you can call

lua/swenv/api.lua

@@ -81,13 +81,14 @@ M.get_current_venv = function()
   return current_venv
 end
 
+
 local get_venvs_for = function(base_path, source, opts)
   local venvs = {}
   if base_path == nil then
     return venvs
   end
   local paths = scan_dir(base_path, vim.tbl_extend('force', { depth = 1, only_dirs = true, silent = true }, opts or {}))
-  for _, path in ipairs(paths) do
+  for _, path in pairs(paths) do
     table.insert(venvs, {
       name = Path:new(path):make_relative(base_path),
       path = path,
@@ -117,7 +118,7 @@ local get_conda_base_path = function()
   end
 end
 
-local get_conda_base_env = function ()
+local get_conda_base_env = function()
   local venvs = {}
   local path = os.getenv("CONDA_EXE")
   if path then
@@ -148,15 +149,33 @@ local get_pyenv_base_path = function()
   end
 end
 
+local to_set = function(some_list)
+  local set = {}
+  for _, key in ipair(some_list) do
+    set[key] = true
+  end
+  return set
+end
+
 M.get_venvs = function(venvs_path)
   local venvs = {}
   vim.list_extend(venvs, get_venvs_for(venvs_path, 'venv'))
-  vim.list_extend(venvs, get_venvs_for(get_pixi_base_path(), 'pixi'))
-  vim.list_extend(venvs, get_venvs_for(get_conda_base_path(), 'conda'))
-  vim.list_extend(venvs, get_conda_base_env())
-  vim.list_extend(venvs, get_venvs_for(get_micromamba_base_path(), 'micromamba'))
-  vim.list_extend(venvs, get_venvs_for(get_pyenv_base_path(), 'pyenv'))
-  vim.list_extend(venvs, get_venvs_for(get_pyenv_base_path(), 'pyenv', { only_dirs = false }))
+
+  local ignore_envs = to_set(settings.ignore_envs_groups)
+  if ignore_envs and not ignore_envs['conda'] then
+    vim.list_extend(venvs, get_venvs_for(get_conda_base_path(), 'conda'))
+    vim.list_extend(venvs, get_conda_base_env())
+  end
+  if ignore_envs and not ignore_envs['pixi'] then
+    vim.list_extend(venvs, get_venvs_for(get_pixi_base_path(), 'pixi'))
+  end
+  if ignore_envs and not ignore_envs['micromamba'] then
+    vim.list_extend(venvs, get_venvs_for(get_micromamba_base_path(), 'micromamba'))
+  end
+  if ignore_envs and not ignore_envs['pyenv'] then
+    vim.list_extend(venvs, get_venvs_for(get_pyenv_base_path(), 'pyenv'))
+    vim.list_extend(venvs, get_venvs_for(get_pyenv_base_path(), 'pyenv', { only_dirs = false }))
+  end
   return venvs
 end
 

lua/swenv/config.lua

@@ -10,6 +10,7 @@ M.settings = {
   -- Path passed to `get_venvs`.
   venvs_path = vim.fn.expand('~/venvs'),
   -- Something to do after setting an environment
+  ignore_envs_groups = {},
   post_set_venv = nil,
 }
 

lua/swenv/match.lua

@@ -21,13 +21,13 @@ end
 
 -- Function to find the best match based on Levenshtein distance
 M.best_match = function(items, query)
-  local min_distance = math.huge -- Initialize minimum distance as infinity
-  local match = nil -- Initialize match as nil to handle case if no match found
-  for _, item in ipairs(items) do -- Iterate over items
+  local min_distance = math.huge           -- Initialize minimum distance as infinity
+  local match = nil                        -- Initialize match as nil to handle case if no match found
+  for _, item in ipairs(items) do          -- Iterate over items
     local distance = lev(query, item.name) -- Compute Levenshtein distance to the query
-    if distance < min_distance then -- If this item is closer to query...
-      min_distance = distance -- Update minimum distance
-      match = item -- Update the match
+    if distance < min_distance then        -- If this item is closer to query...
+      min_distance = distance              -- Update minimum distance
+      match = item                         -- Update the match
     end
   end
   return match -- Return the best match

lua/swenv/project.lua

@@ -6,7 +6,7 @@ M.read_venv_name = function(project_dir)
   if not file then
     return nil
   end
-  local content = file:read('*a') -- *a or *all reads the whole file
+  local content = file:read('*a')      -- *a or *all reads the whole file
   file:close()
   return content:match('^%s*(.-)%s*$') -- Trim whitespace
 end