python.nvim/lua/python/hatch/commands.lua at 90a7677d2ef8af867ebd1fb2db4a81d83583be51 · joshzcold/python.nvim · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
local ui = require("python.ui")
local PythonHatchCommands = {}

function PythonHatchCommands.check_hatch()
  if vim.fn.executable("hatch") == 0 then
    vim.notify_once(
      ("python.nvim: Program 'hatch' is required: %s"):format("https://hatch.pypa.io/latest/"),
      vim.log.levels.ERROR
    )
    return false
  end
  return true
end

--- Get list of python versions hatch has installed
---@return table list of available python versions from hatch
local function hatch_installed_versions()
  local output = {}
  vim
    .system({ "hatch", "python", "show", "--ascii" }, {}, function(obj)
      local found_installed = {}
      for line in vim.gsplit(obj.stdout, "\n", { plain = true }) do
        local available_line = string.match(line, "Available")
        if available_line then
          break
        end
        local matched = string.match(line, "(%d%.%d+%s*|%s*%d.%d+)")
        table.insert(found_installed, 1, matched)
      end

      for _, f in pairs(found_installed) do
        local parts = vim.split(f, "|", { trimempty = true })
        parts[1] = parts[1]:gsub("%s+", "")
        table.insert(output, 1, parts[1])
      end
    end)
    :wait()
  return output
end

--- Get list of python versions hatch can install
---@return table list of available python versions from hatch
local function hatch_available_versions()
  local output = {}
  vim
    .system({ "hatch", "python", "show", "--ascii" }, {}, function(obj)
      local found_available = {}
      local available_line_shown = false
      for line in vim.gsplit(obj.stdout, "\n", { plain = true }) do
        local available_line = string.match(line, "Available")
        if available_line then
          available_line_shown = true
        end

        if available_line_shown then
          local matched = string.match(line, "(%s+%d%.%d+%s+|%s+%d.%d+)")
          table.insert(found_available, 1, matched)
        end
      end

      for _, f in pairs(found_available) do
        local parts = vim.split(f, "|", { trimempty = true })
        parts[1] = parts[1]:gsub("%s+", "")
        table.insert(output, 1, parts[1])
      end
    end)
    :wait()
  return output
end

--- Install a python version using hatch
---@param version string Python version to install via hatch
local function hatch_install_version(version)
  vim.schedule(function()
    vim.system({ "hatch", "python", "install", version }, {
      stdout = ui.show_system_call_progress,
    }, function(obj2)
      vim.schedule(function()
        if obj2.code ~= 0 then
          vim.notify_once("python.nvim: " .. vim.inspect(obj2.stderr), vim.log.levels.ERROR)
          ui.deactivate_system_call_ui(10000)
        else
          ui.show_system_call_progress(obj2.stderr, obj2.stdout, true, function()
            ui.deactivate_system_call_ui()
          end)
        end
      end)
    end)
    vim.schedule(function()
      ui.activate_system_call_ui()
    end)
  end)
end

--- Delete a python version using hatch
---@param version string Python version to delete via hatch
local function hatch_delete_version(version)
  vim.schedule(function()
    vim.system({ "hatch", "python", "remove", version }, {
      stdout = ui.show_system_call_progress,
    }, function(obj2)
      vim.schedule(function()
        if obj2.code ~= 0 then
          vim.notify_once("python.nvim: " .. vim.inspect(obj2.stderr), vim.log.levels.ERROR)
          ui.deactivate_system_call_ui(10000)
        else
          ui.show_system_call_progress(obj2.stderr, obj2.stdout, true, function()
            ui.deactivate_system_call_ui()
          end)
        end
      end)
    end)
    vim.schedule(function()
      ui.activate_system_call_ui()
    end)
  end)
end

function PythonHatchCommands.hatch_install_python()
  local versions = hatch_available_versions()
  vim.ui.select(versions, { prompt = "Select a python version to install via hatch: " }, function(selection)
    if not selection then
      return
    end
    hatch_install_version(selection)
  end)
end

function PythonHatchCommands.hatch_list_python()
  local versions = hatch_installed_versions()
end

function PythonHatchCommands.hatch_delete_python()
  local versions = hatch_installed_versions()
  vim.ui.select(versions, { prompt = "Select a python version to delete in hatch: " }, function(selection)
    if not selection then
      return
    end
    hatch_delete_version(selection)
  end)
end

return PythonHatchCommands