From 01b7213fbd0a47e7df94a46b29aa31814ebc787b Mon Sep 17 00:00:00 2001 From: danymat Date: Sat, 29 Jan 2022 11:40:32 +0100 Subject: [PATCH] feat: Add fallback node for custom placement --- lua/neogen/generators/default.lua | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/lua/neogen/generators/default.lua b/lua/neogen/generators/default.lua index 995125a..ee9d18b 100644 --- a/lua/neogen/generators/default.lua +++ b/lua/neogen/generators/default.lua @@ -75,7 +75,12 @@ return function(parent, data, template, required_type) if append.position == "after" then local child_node = nodes:matching_child_nodes(parent, append.child_name)[1] - if child_node ~= nil then + if not child_node and append.fallback then + local fallback = nodes:matching_child_nodes(parent, append.fallback)[1] + if fallback then + row_to_place, col_to_place, _, _ = fallback:range() + end + else row_to_place, col_to_place, _, _ = child_node:range() end end