feat(config): stay_on_exec (#3)

config.lua

@@ -17,6 +17,7 @@ local defaults = {
 
 	mode = "hybrid", -- "modal" | "hold" | "hybrid" | "forever"
 	smart_modifiers = true, -- like smartcase but for all root key modifiers
+	stay_on_exec = true, -- stay on tree level or reset tree after executing
 	stop_on_unknown_key = false,
 	timeout = 0, -- ms
 	labels = "1234567890abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ!\"#$%&'()*+,-./:;<=>?@\\^_`{|}~",

modal.lua

@@ -546,7 +546,9 @@ function trunner:step_into(node)
 
 	node:remove_temp_successors()
 
-	if node:is_leaf() then
+	local is_exec = node:is_leaf()
+
+	if is_exec then
 		next_tree = self:execute(node)
 	else
 		next_tree = node
@@ -571,6 +573,13 @@ function trunner:step_into(node)
 		return
 	end
 
+	if is_exec then
+		local opts = node:opts()
+		if not opts.stay_on_exec then
+			self:set_tree(node:get_root_tree())
+		end
+	end
+
 	self.continue_external = false
 end
 

tree.lua

@@ -301,6 +301,15 @@ function M:successors()
 	return tree_objects
 end
 
+function M:get_root_tree()
+	local t = self
+	while t:pred() do
+		t = t:pred()
+	end
+
+	return t
+end
+
 function M:is_leaf()
 	local succs = self._succs
 	if not succs or vim.tbl_isempty(succs) then