From c172c0e8d2ca7d13126aa68c708442e3cb883a99 Mon Sep 17 00:00:00 2001
From: Andrew Thompson <andrew@hijacked.us>
Date: Fri, 24 Jun 2011 16:51:34 -0400
Subject: [PATCH] Fix parse_transform to recurse the AST more

---
 src/lager_transform.erl | 15 +++++----------
 1 file changed, 5 insertions(+), 10 deletions(-)

diff --git a/src/lager_transform.erl b/src/lager_transform.erl
index db8d20b..ff74485 100644
--- a/src/lager_transform.erl
+++ b/src/lager_transform.erl
@@ -103,17 +103,12 @@ transform_statement({call, Line, {remote, Line1, {atom, Line2, boston_lager},
         end,
         transform_statement({call, Line, {remote, Line1, {atom, Line2, lager},
               {atom, Line3, Severity}}, NewArgs});
-transform_statement({'case', Line, Expr, Clauses}) ->
-    {'case', Line, Expr, walk_clauses([], Clauses)};
-transform_statement({'if', Line, Clauses}) ->
-    {'if', Line, walk_clauses([], Clauses)};
-transform_statement({block, Line, Body}) ->
-    {block, Line, walk_body([], Body)};
-transform_statement({lc, Line, Expression, Generator}) ->
-    {lc, Line, transform_statement(Expression), Generator};
-transform_statement({match, Line, Var, Expression}) ->
-    {match, Line, Var, transform_statement(Expression)};
+transform_statement(Stmt) when is_tuple(Stmt) ->
+    list_to_tuple(transform_statement(tuple_to_list(Stmt)));
+transform_statement(Stmt) when is_list(Stmt) ->
+    [transform_statement(S) || S <- Stmt];
 transform_statement(Stmt) ->
+    %io:format("Statement ~p~n", [Stmt]),
     Stmt.