Packages

beamoji

0.1.0
0.1.0

🪄

HexDocs

Current section

Files

Jump to
Readme
1 Version
3 Dependencies
0 Dependants
Files
Activity
beamoji src beamoji_utils.erl
Raw

src/beamoji_utils.erl 

-module(beamoji_utils).



-export(['🪄'/2]).



-type translator() :: fun((erl_parse:abstract_form()) -> erl_parse:abstract_form()).



-spec '🪄'(translator(), erl_parse:abstract_form()) -> erl_parse:abstract_form().

'🪄'(Translator, {atom, Anno, Atom}) ->

    {atom, Anno, Translator(Atom)};

'🪄'(Translator, {function, Anno, Name, Arity, Clauses}) when is_atom(Name) ->

    {function, Anno, Translator(Name), Arity, Clauses};

'🪄'(Translator, {attribute, Anno, record, {Name, Fields}}) when is_atom(Name) ->

    {attribute, Anno, record, {Translator(Name), Fields}};

'🪄'(Translator, {attribute, Anno, AttrName, Es})

    when AttrName == export; AttrName == export_type ->

    {attribute, Anno, AttrName, '🪄🪄'(Translator, Es)};

'🪄'(Translator, {attribute, Anno, import, {Mod, Es}}) ->

    {attribute, Anno, import, {Mod, '🪄🪄'(Translator, Es)}};

'🪄'(Translator, {attribute, Anno, AttrName, {Name, Type, Variables}})

    when AttrName == type; AttrName == opaque ->

    {attribute, Anno, AttrName, {Translator(Name), Type, Variables}};

'🪄'(Translator, {attribute, Anno, AttrName, {{Name, Arity}, Clauses}})

    when AttrName == spec; AttrName == callback ->

    {attribute, Anno, AttrName, {{Translator(Name), Arity}, Clauses}};

'🪄'(Translator, {attribute, Anno, spec, {{Module, Name, Arity}, Clauses}}) ->

    {attribute, Anno, spec, {{Module, Translator(Name), Arity}, Clauses}};

'🪄'(_Translator, Ast) ->

    Ast.



'🪄🪄'(Translator, Es) ->

    [{Translator(Name), Arity} || {Name, Arity} <- Es].