support multicursor.nvim

Author: jake-stewart

lua/oil/view.lua

@@ -257,6 +257,32 @@ local function get_first_mutable_column_col(adapter, ranges)
   return min_col
 end
 
+--- @param bufnr integer
+--- @param adapter oil.Adapter
+--- @param mode false|"name"|"editable"
+--- @param cur integer[]
+--- @return integer[] | nil
+local function calc_constrained_cursor_pos(bufnr, adapter, mode, cur)
+  local parser = require("oil.mutator.parser")
+  local line = vim.api.nvim_buf_get_lines(bufnr, cur[1] - 1, cur[1], true)[1]
+  local column_defs = columns.get_supported_columns(adapter)
+  local result = parser.parse_line(adapter, line, column_defs)
+  if result and result.ranges then
+    local min_col
+    if mode == "editable" then
+      min_col = get_first_mutable_column_col(adapter, result.ranges)
+    elseif mode == "name" then
+      min_col = result.ranges.name[1]
+    else
+      error(string.format('Unexpected value "%s" for option constrain_cursor', mode))
+    end
+    if cur[2] < min_col then
+      return { cur[1], min_col }
+    end
+  end
+end
+
+
 ---Force cursor to be after hidden/immutable columns
 ---@param bufnr integer
 ---@param mode false|"name"|"editable"
@@ -267,30 +293,33 @@ local function constrain_cursor(bufnr, mode)
   if bufnr ~= vim.api.nvim_get_current_buf() then
     return
   end
-  local parser = require("oil.mutator.parser")
 
   local adapter = util.get_adapter(bufnr, true)
   if not adapter then
     return
   end
 
-  local cur = vim.api.nvim_win_get_cursor(0)
-  local line = vim.api.nvim_buf_get_lines(bufnr, cur[1] - 1, cur[1], true)[1]
-  local column_defs = columns.get_supported_columns(adapter)
-  local result = parser.parse_line(adapter, line, column_defs)
-  if result and result.ranges then
-    local min_col
-    if mode == "editable" then
-      min_col = get_first_mutable_column_col(adapter, result.ranges)
-    elseif mode == "name" then
-      min_col = result.ranges.name[1]
-    else
-      error(string.format('Unexpected value "%s" for option constrain_cursor', mode))
-    end
-    if cur[2] < min_col then
-      vim.api.nvim_win_set_cursor(0, { cur[1], min_col })
+  local mc = package.loaded["multicursor-nvim"]
+  if mc then
+    mc.onSafeState(function()
+      mc.action(function(ctx)
+        ctx:forEachCursor(function(cursor)
+          local new_cur = calc_constrained_cursor_pos(
+            bufnr, adapter, mode, { cursor:line(), cursor:col() - 1 })
+          if new_cur then
+            cursor:setPos({new_cur[1], new_cur[2] + 1})
+          end
+        end)
+      end)
+    end, { once = true })
+  else
+    local cur = vim.api.nvim_win_get_cursor(0)
+    local new_cur = calc_constrained_cursor_pos(bufnr, adapter, mode, cur)
+    if new_cur then
+      vim.api.nvim_win_set_cursor(0, new_cur)
     end
   end
+
 end
 
 ---Redraw original path virtual text for trash buffer