slepher / astranaut Goto Github PK

-use_macro({macro/0, [unique_var]}).
macro() ->
  quote(
    fun(A) ->
      B = A + 1
      B  = A + 3
      {C, +B} = {A  + 5, A +3}
      {B, C}
   end). 

hello(A) ->
  B = A + 5,
  macro().
=>
hello(A) ->
  B = A + 5,
  fun(A_1@Q_1) ->
      B_1@Q_1 = A_1@Q_1 + 1
      B_2@Q_1 = A_1@Q_1 + 3
      {C_1@Q_1, B_2@Q_1} = {A_1@Q_1  + 5, A_1@Q_1 + 3}
      {B_2@Q_1, C_1@Q_1}
   end.

hello(begin A = A +1, A end, begin A = A+ 2, A end).
{begin A = A +1, A end, begin A = A+ 2, A end}

-module(macro_example).
macro_with_vars_1(Ast) ->
    quote(
      begin
          A = 10,
          B = unquote(Ast),
          A + B
      end
     ).
macro_with_vars_2(Ast) ->
    quote(
      begin
          A = 10,
          B = unquote(Ast),
          A + B
      end
     ).

test_macro_with_vars(N) ->
    A1 = macro_with_vars_1(N),
    A2 = macro_with_vars_2(A1),
    A3 = macro_with_vars_2(N),
    A4 = macro_with_vars_1(A1),
    A1 + A2.

test_macro_with_vars(N) ->
A1 =
begin
  A@macro_example@_1 = 10,
  B@macro_example@_1 = N,
  A@macro_example@_1 + B@macro_example@_1
end,
A2 = 
begin
  A@macro_example@_3 = 10,
  B@macro_example@_3 = A1,
  A@macro_example@_3 + B@macro_example@_3
end,
A3 = 
begin
  A@macro_example@_4 = 10,
  B@macro_example@_4 = N,
  A@macro_example@_4 + B@macro_example@_4
end
A4 =
begin
  A@macro_example@_2 = 10,
  B@macro_example@_2 = A1,
  A@macro_example@_2 + B@macro_example@_2
end
A1 + A2 + A3 + A4.

-module(a).
-behaviour(gen_server).
-use_macro({macro_a, macro/2, [{attrs, [module, line, behaviour]}]}).

hello() ->
  macro_a:macro(world).

-module(macro_a).
-export([macro/2]).

macro(Ast, #{module => Module, line => Line, behaviour => Behaviours} = Attributes) ->
   io:format("module is ~p, behaviour is ~p", [Module, Behaviours]),
   Ast.

astranaut:replace_from_nth(Nodes, N, Forms, Options).