refactor: API modernization — simplify marks, warn, utils, use modern Neovim APIs

- Rewrite marks.lua: replace manual extmark ID tracking (ext_ids,
  next_ext_ids tables) with nvim_buf_clear_namespace(). Use
  nvim_get_current_buf() instead of vim.fn.bufnr().
- Simplify warn.lua: remove vestigial vim.fn.escape() and manual \\n
  handling left over from old vim.cmd('echohl...') approach. Now just
  calls vim.notify() directly.
- Remove custom min/max from utils.lua, use math.min/math.max in
  jump/param.lua instead.
- Modernize utils.with_buf: use vim.api.nvim_buf_is_loaded(),
  vim.bo[bufnr].filetype, vim.fn.bufload(), and pcall safety for the
  callback. Errors are now re-raised after cleanup.
- Modernize jump/util.lua: use vim.cmd.edit() instead of
  vim.fn.execute('edit ...').
- Use idiomatic 'opts = opts or {}' and 'bufnr or 0' patterns in
  api.lua and init.lua.

Author: sigfriedCub1990

lua/pytrize/api.lua

@@ -2,20 +2,13 @@ local M = {}
 
 M.clear = function(bufnr)
     local marks = require("pytrize.marks")
-
-    if bufnr == nil then
-        bufnr = 0
-    end
-    marks.clear(bufnr)
+    marks.clear(bufnr or 0)
 end
 
 M.set = function(bufnr)
     local cs = require("pytrize.call_spec")
     local marks = require("pytrize.marks")
-
-    if bufnr == nil then
-        bufnr = 0
-    end
+    bufnr = bufnr or 0
     marks.clear(bufnr)
     local call_specs_per_func = cs.get_calls(bufnr)
     if call_specs_per_func == nil then

lua/pytrize/init.lua

@@ -24,9 +24,7 @@ local function setup_commands()
 end
 
 M.setup = function(opts)
-    if opts == nil then
-        opts = {}
-    end
+    opts = opts or {}
     settings.update(opts)
     if not settings.settings.no_commands then
         setup_commands()

lua/pytrize/jump/param.lua

@@ -7,8 +7,6 @@ local paths = require('pytrize.paths')
 local warn = require('pytrize.warn').warn
 local open_file = require('pytrize.jump.util').open_file
 local get_nodeids_path = require('pytrize.paths').get_nodeids_path
-local min = require('pytrize.utils').min
-local max = require('pytrize.utils').max
 
 local function query_file(func_name, callback)
   local rootdir, _ = paths.split_at_root(vim.api.nvim_buf_get_name(0))
@@ -55,7 +53,7 @@ local function jump_to_nodeid_at_cursor(callback)
   local nodeid = nids.parse_raw(line:sub(i))
   local pattern_position = col_num + 1 - (i - 1)  -- cursor relative to match
   local param_position = pattern_position - nodeid.param_start_idx + 1  -- cursor relative to params
-  param_position = min(max(1, param_position), nodeid.params:len())  -- restrict it to be inside
+  param_position = math.min(math.max(1, param_position), nodeid.params:len())  -- restrict it to be inside
   if nodeid == nil then
     warn("couldn't parse nodeid under cursor")
     return

lua/pytrize/jump/util.lua

@@ -1,7 +1,7 @@
 local M = {}
 
 M.open_file = function(file)
-    vim.fn.execute("edit " .. vim.fn.fnameescape(file))
+    vim.cmd.edit(vim.fn.fnameescape(file))
 end
 
 return M

lua/pytrize/marks.lua

@@ -2,49 +2,27 @@ local M = {}
 
 local settings = require("pytrize.settings").settings
 
-local next_ext_ids = {}
-local ext_ids = {}
-
-local function get_ns_id()
-    return vim.api.nvim_create_namespace("pytrize")
-end
+local NS_ID = vim.api.nvim_create_namespace("pytrize")
 
+--- Clear all pytrize extmarks from a buffer.
+---@param bufnr integer Buffer number (0 for current)
 M.clear = function(bufnr)
     if bufnr == 0 then
-        bufnr = vim.fn.bufnr()
-    end
-    for _, ext_id in ipairs(ext_ids[bufnr] or {}) do
-        vim.api.nvim_buf_del_extmark(bufnr, get_ns_id(), ext_id)
-    end
-    ext_ids[bufnr] = nil
-    next_ext_ids[bufnr] = nil
-end
-
-local get_ext_id = function(bufnr)
-    if ext_ids[bufnr] == nil then
-        ext_ids[bufnr] = {}
-    end
-    if next_ext_ids[bufnr] == nil then
-        next_ext_ids[bufnr] = 1
+        bufnr = vim.api.nvim_get_current_buf()
     end
-    local next_ext_id = next_ext_ids[bufnr]
-    table.insert(ext_ids[bufnr], next_ext_id)
-    next_ext_ids[bufnr] = next_ext_id + 1
-    return next_ext_id
+    vim.api.nvim_buf_clear_namespace(bufnr, NS_ID, 0, -1)
 end
 
+--- Set a virtual text extmark.
+---@param opts { bufnr: integer, row: integer, text: string }
 M.set = function(opts)
     local bufnr = opts.bufnr
     if bufnr == 0 then
-        bufnr = vim.fn.bufnr()
+        bufnr = vim.api.nvim_get_current_buf()
     end
-    vim.api.nvim_buf_set_extmark(
-        bufnr,
-        get_ns_id(),
-        opts.row,
-        0,
-        { id = get_ext_id(bufnr), virt_text = { { opts.text, settings.highlight } } }
-    )
+    vim.api.nvim_buf_set_extmark(bufnr, NS_ID, opts.row, 0, {
+        virt_text = { { opts.text, settings.highlight } },
+    })
 end
 
 return M

lua/pytrize/utils.lua

@@ -1,50 +1,38 @@
 local M = {}
 
-M.min = function(a, b)
-    if a <= b then
-        return a
-    else
-        return b
-    end
-end
-
-M.max = function(a, b)
-    if a >= b then
-        return a
-    else
-        return b
-    end
-end
-
 --- Temporarily load a Python buffer for `filepath`, call `fn(bufnr)`, then
 --- clean up if the buffer was not already loaded.  Autocommands are suppressed
 --- during the load so that LSP clients (and other plugins) do not attach to
 --- ephemeral buffers.
 ---
---- @param filepath string   Absolute path to a Python file.
---- @param fn fun(bufnr: integer): any   Callback that receives the buffer number.
---- @param opts? { force_delete?: boolean }   Options for cleanup (default: force_delete = false).
---- @return any  The return value of `fn`.
+---@param filepath string   Absolute path to a Python file.
+---@param fn fun(bufnr: integer): any   Callback that receives the buffer number.
+---@param opts? { force_delete?: boolean }   Options for cleanup.
+---@return any  The return value of `fn`.
 M.with_buf = function(filepath, fn, opts)
     opts = opts or {}
     local existing_bufnr = vim.fn.bufnr(filepath)
-    local was_loaded = existing_bufnr ~= -1 and vim.fn.bufloaded(existing_bufnr) == 1
+    local was_loaded = existing_bufnr ~= -1 and vim.api.nvim_buf_is_loaded(existing_bufnr)
 
     local bufnr = vim.fn.bufadd(filepath)
     if not was_loaded then
         local saved_eventignore = vim.o.eventignore
         vim.o.eventignore = "all"
-        vim.fn.execute("call bufload(" .. bufnr .. ")")
-        vim.api.nvim_set_option_value("filetype", "python", { buf = bufnr })
+        vim.fn.bufload(bufnr)
+        vim.bo[bufnr].filetype = "python"
         vim.o.eventignore = saved_eventignore
     end
 
-    local result = fn(bufnr)
+    local ok, result = pcall(fn, bufnr)
 
-    if not was_loaded then
+    if not was_loaded and vim.api.nvim_buf_is_valid(bufnr) then
         vim.api.nvim_buf_delete(bufnr, { force = opts.force_delete or false })
     end
 
+    if not ok then
+        error(result)
+    end
+
     return result
 end
 

lua/pytrize/warn.lua

@@ -1,9 +1,7 @@
 local M = {}
 
 M.warn = function(msg)
-    msg = vim.fn.escape(msg, '"'):gsub("\\n", "\n")
-    -- vim.cmd(string.format('echohl WarningMsg | echo "Pytrize Warning: %s" | echohl None', msg))
-    vim.notify(vim.split(string.format("Pytrize Warning: %s", msg), "\n"), vim.log.levels.WARN)
+    vim.notify(string.format("Pytrize: %s", msg), vim.log.levels.WARN)
 end
 
 return M

tests/pytrize/utils_spec.lua

@@ -1,31 +1,36 @@
 local utils = require("pytrize.utils")
 
-describe("min", function()
-  it("returns the smaller value", function()
-    assert.are.equal(1, utils.min(1, 2))
-    assert.are.equal(1, utils.min(2, 1))
-  end)
+describe("with_buf", function()
+    it("loads a file into a buffer, calls fn, and cleans up", function()
+        local tmp = vim.fn.tempname() .. ".py"
+        local f = io.open(tmp, "w")
+        f:write("x = 1\n")
+        f:close()
 
-  it("returns the value when equal", function()
-    assert.are.equal(5, utils.min(5, 5))
-  end)
+        local result = utils.with_buf(tmp, function(bufnr)
+            local lines = vim.api.nvim_buf_get_lines(bufnr, 0, -1, false)
+            return lines[1]
+        end, { force_delete = true })
 
-  it("works with negative numbers", function()
-    assert.are.equal(-3, utils.min(-3, -1))
-  end)
-end)
+        assert.are.equal("x = 1", result)
+
+        -- buffer should have been cleaned up
+        assert.are.equal(-1, vim.fn.bufnr(tmp))
+        vim.fn.delete(tmp)
+    end)
 
-describe("max", function()
-  it("returns the larger value", function()
-    assert.are.equal(2, utils.max(1, 2))
-    assert.are.equal(2, utils.max(2, 1))
-  end)
+    it("propagates errors from fn", function()
+        local tmp = vim.fn.tempname() .. ".py"
+        local f = io.open(tmp, "w")
+        f:write("x = 1\n")
+        f:close()
 
-  it("returns the value when equal", function()
-    assert.are.equal(5, utils.max(5, 5))
-  end)
+        assert.has_error(function()
+            utils.with_buf(tmp, function(_)
+                error("test error")
+            end, { force_delete = true })
+        end)
 
-  it("works with negative numbers", function()
-    assert.are.equal(-1, utils.max(-3, -1))
-  end)
+        vim.fn.delete(tmp)
+    end)
 end)