leigh/neogen
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: Allow after_each option in template

Browse Source
This commit is contained in:
danymat
2022-01-23 12:50:13 +01:00
parent 2973ff1489
commit 510282622b
1 changed files with 6 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
lua/neogen/generators/default.lua
View File

@@ -126,6 +126,9 @@ neogen.default_generator = function(parent, data, template, required_type)
for _, value in ipairs(data[inserted_type]) do for _, value in ipairs(data[inserted_type]) do
local inserted = conditional_prefix_inserter(prefix, formatted_string:format(value)) local inserted = conditional_prefix_inserter(prefix, formatted_string:format(value))
table.insert(result, inserted) table.insert(result, inserted)
if opts.after_each then
table.insert(result, conditional_prefix_inserter(prefix, opts.after_each))
end
end end
elseif type(inserted_type) == "table" and data[opts.required] then elseif type(inserted_type) == "table" and data[opts.required] then
-- First item in the template item can be a table. -- First item in the template item can be a table.
@@ -151,6 +154,9 @@ neogen.default_generator = function(parent, data, template, required_type)
formatted_string:format(unpack(_values)) formatted_string:format(unpack(_values))
) )
table.insert(result, inserted) table.insert(result, inserted)
if opts.after_each then
table.insert(result, conditional_prefix_inserter(prefix, opts.after_each))
end
end end
end end
end end