Iteration. Not working, but code structure is there

lua/treewalker/init.lua

@@ -1,4 +1,4 @@
-local util     = require('treewalker.util')
+local util = require('treewalker.util')
 local movement = require('treewalker.movement')
 local swapping = require('treewalker.swapping')
 

lua/treewalker/nodes.lua

@@ -181,4 +181,10 @@ function M.get_at_row(row)
   return vim.treesitter.get_node({ pos = { row - 1, col } })
 end
 
+-----Getting the size of the node is no trivial thing. Different nodes behave differently.
+-----@param row integer
+-----@return TSNode|nil
+--function M.get_real_range(row)
+--end
+
 return M

lua/treewalker/swapping.lua

@@ -38,12 +38,21 @@ end
 
 ---@return nil
 function M.swap_down()
+  local current = nodes.get_current()
   local target, row, line = targets.down()
 
-  if target and row and line then
+  if not target or not row or not line then
     --util.log("no down candidate")
-    ops.jump(row, target)
+    return
   end
+
+  local current_range = nodes.range(current)
+  local target_range = nodes.range(target)
+
+  ops.swap(
+    { current_range[1], current_range[3] },
+    { target_range[1], target_range[3] }
+  )
 end
 
 return M

plugin/init.lua

@@ -21,17 +21,20 @@ local subcommands = {
     tw().move_in()
   end,
 
-  ---@param subcommand "right" | "left" | "up" | "down"
-  Swap = function(subcommand)
-    if subcommand == "right" then
-      tw().swap_right()
-    elseif subcommand == "left" then
-      tw().swap_left()
-    elseif subcommand == "up" then
-      tw().swap_up()
-    elseif subcommand == "down" then
-      tw().swap_down()
-    end
+  SwapUp = function()
+    tw().swap_up()
+  end,
+
+  SwapDown = function()
+    tw().swap_down()
+  end,
+
+  SwapRight = function()
+    tw().swap_in()
+  end,
+
+  SwapLeft = function()
+    tw().swap_out()
   end
 }