From db725a90bad7525f23d4756117c8baae07f4b9b7 Mon Sep 17 00:00:00 2001
From: Andrew Thompson <andrew@hijacked.us>
Date: Thu, 14 Jul 2011 10:32:20 -0400
Subject: [PATCH] Add support for parameterized modules to the parse transform

---
 src/lager_transform.erl | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/src/lager_transform.erl b/src/lager_transform.erl
index 8927610..60e18f5 100644
--- a/src/lager_transform.erl
+++ b/src/lager_transform.erl
@@ -33,6 +33,10 @@ parse_transform(AST, _Options) ->
 
 walk_ast(Acc, []) ->
     lists:reverse(Acc);
+walk_ast(Acc, [{attribute, _, module, {Module, _PmodArgs}}=H|T]) ->
+    %% A wild parameterized module appears!
+    put(module, Module),
+    walk_ast([H|Acc], T);
 walk_ast(Acc, [{attribute, _, module, Module}=H|T]) ->
     put(module, Module),
     walk_ast([H|Acc], T);