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

feat: Allow index reordering in after_each option template

Browse Source
This commit is contained in:
danymat
2022-01-29 13:38:45 +01:00
parent fe44b20a57
commit 30f98ed999
1 changed files with 27 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -133,7 +133,10 @@ return function(parent, data, template, required_type)
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 if opts.after_each then
table.insert(result, conditional_prefix_inserter(prefix, opts.after_each)) table.insert(
result,
conditional_prefix_inserter(prefix, opts.after_each):format(value)
)
end end
end end
elseif type(inserted_type) == "table" and data[opts.required] then elseif type(inserted_type) == "table" and data[opts.required] then
@@ -161,7 +164,29 @@ return function(parent, data, template, required_type)
) )
table.insert(result, inserted) table.insert(result, inserted)
if opts.after_each then if opts.after_each then
table.insert(result, conditional_prefix_inserter(prefix, opts.after_each)) if type(opts.after_each) == "table" then
local _v = {}
local index_types = opts.after_each["index_types"]
for _, i in ipairs(index_types) do
table.insert(_v, _values[i])
end
_values = _v
table.insert(
result,
conditional_prefix_inserter(
prefix,
opts.after_each[1]:format(unpack(_values))
)
)
else
table.insert(
result,
conditional_prefix_inserter(
prefix,
opts.after_each:format(unpack(_values))
)
)
end
end end
end end
end end