ASTMangler supports multiple placeholders

This commit is contained in:
Felipe Sere
2016-03-13 14:56:43 +00:00
parent e077286b9d
commit 89dac5bb05
2 changed files with 38 additions and 3 deletions

View File

@@ -1,8 +1,25 @@
defmodule ASTMangler do
def expand([do: remainder], replacements) when is_list(replacements) do
[do: expand(remainder, replacements)]
end
def expand(ast, replacements) when is_list(replacements) do
{node, _acc} = Macro.prewalk(ast, replacements, fn(node, acc) -> pre(node, acc) end)
node
end
def expand(ast, replacement) do
Macro.prewalk(ast, fn(node) -> update(node, replacement) end)
expand(ast, [replacement])
end
def update(:__, replacement), do: replacement
def update(node, _), do: node
def pre(:__, [first | remainder]), do: {first, remainder}
def pre(node, acc), do: {node, acc}
def count(ast) do
{_node, acc} = Macro.prewalk(ast, 0, fn(node, acc) -> count(node, acc) end)
acc
end
def count(:__, acc), do: {node, acc+1}
def count(node, acc), do: {node, acc}
end