Fixed parse_transform pt_lazy_case

Author: Рощупкин Никита

src/pt_lazy_case.erl

@@ -164,7 +164,7 @@ parse_clause(_, Acc, OtherAST, _, _) ->
 parse_tuple(HeadAstTuple, ListASTClause, [], V1, _) ->
     NewListASTClause = parse_clause(ListASTClause, [], []),
     OffRepeateASTClause = delete_repeate(NewListASTClause, []),
-    case check_load_code(OffRepeateASTClause) of
+    case check_load_code(OffRepeateASTClause, HeadAstTuple) of
         false ->
             {'case', V1, HeadAstTuple, OffRepeateASTClause};
         {true, Data} ->
@@ -179,7 +179,7 @@ parse_tuple(HeadAstTuple, ListASTClause, TailAstTuple, V1, File) ->
             OffRepeateASTClause = delete_repeate(create_new_case(NewListASTClause, CountElems, TailAstTuple, [], V1), []),
             NewCase = {'case', V1, HeadAstTuple, OffRepeateASTClause},
             ChangeCase = parser_case(NewCase, File),
-            case check_load_code(ChangeCase) of
+            case check_load_code(ChangeCase, HeadAstTuple) of
                 false ->
                     ChangeCase;
                 {true, Data} ->
@@ -297,14 +297,17 @@ check_guard(Guard, [Guard | _], _) ->
 check_guard(Guard, [HeadGuard | TailGuard], Acc) ->
     check_guard(Guard, TailGuard, [HeadGuard | Acc]);
 check_guard(Guard, _, Acc) ->
-
     {true, [Guard | Acc]}.
 
-check_load_code({'case', _, _, [{clause, _, [{var, _, _}], _, D3} | _]}) ->
+check_load_code({'case', _, _, [{clause, _, [{var, _, '_'}], _, D3} | _]}, _HeadAstTuple) ->
     {true, D3};
-check_load_code([{clause, _, [{var, _, _}], _, D3} | _]) ->
+check_load_code([{clause, _, [{var, _, '_'}], _, D3} | _], _HeadAstTuple) ->
     {true, D3};
-check_load_code(_) ->
+check_load_code({'case', _, _, [{clause, _, [{var, Line, Var}], _, D3} | _]}, HeadAstTuple) ->
+    {true, [{match, Line, {var, Line, Var}, HeadAstTuple} | D3]};
+check_load_code([{clause, _, [{var, Line, Var}], _, D3} | _], HeadAstTuple) ->
+    {true, [{match, Line, {var, Line, Var}, HeadAstTuple} | D3]};
+check_load_code(_, _) ->
     false.
 
 delete_repeate([{clause, _, Data1, Guard1, _} = HeadASTClause | TailASTClause], Acc) ->

test/pt_lazy_case_tests.erl

@@ -19,7 +19,8 @@ pt_lazy_case_test_() ->
         ?_test(ok = lazy_case_test3()),
         ?_test(ok = lazy_case_test4()),
         ?_test(ok = lazy_case_test5()),
-        ?_test(ok = lazy_case_test6())
+        ?_test(ok = lazy_case_test6()),
+        ?_test(ok = lazy_case_test7())
     ].
 
 lazy_case_test1() ->
@@ -109,3 +110,12 @@ lazy_case_test6() ->
            {_V, _} -> ?assert(false)
         end,
     X.
+
+lazy_case_test7() ->
+    '@lazy_case',
+    X =
+        case {[], ok} of
+           {[], _V} -> _V;
+           {_V, _} -> ?assert(false)
+        end,
+    X.