From 8776a8ba38123c822530e5f659c626c8db616217 Mon Sep 17 00:00:00 2001
From: jeff
Date: Wed, 24 Sep 2014 01:00:54 -0400
Subject: HOW DO I WRITE SO MANY BUGS?!?

---
 .../enigma/bytecode/MethodParameterWriter.java     | 22 ++++++++++++++++++++--
 1 file changed, 20 insertions(+), 2 deletions(-)

(limited to 'src/cuchaz/enigma/bytecode')

diff --git a/src/cuchaz/enigma/bytecode/MethodParameterWriter.java b/src/cuchaz/enigma/bytecode/MethodParameterWriter.java
index a8d3983..adea7ea 100644
--- a/src/cuchaz/enigma/bytecode/MethodParameterWriter.java
+++ b/src/cuchaz/enigma/bytecode/MethodParameterWriter.java
@@ -15,9 +15,13 @@ import java.util.List;
 
 import javassist.CtBehavior;
 import javassist.CtClass;
+import javassist.CtConstructor;
+import javassist.CtMethod;
 import javassist.bytecode.Descriptor;
 import cuchaz.enigma.mapping.ArgumentEntry;
+import cuchaz.enigma.mapping.BehaviorEntry;
 import cuchaz.enigma.mapping.ClassEntry;
+import cuchaz.enigma.mapping.ConstructorEntry;
 import cuchaz.enigma.mapping.MethodEntry;
 import cuchaz.enigma.mapping.Translator;
 
@@ -42,12 +46,26 @@ public class MethodParameterWriter
 				continue;
 			}
 			
+			// get the behavior entry
+			BehaviorEntry behaviorEntry;
+			if( behavior instanceof CtMethod )
+			{
+				behaviorEntry = new MethodEntry( classEntry, behavior.getMethodInfo().getName(), behavior.getSignature() );
+			}
+			else if( behavior instanceof CtConstructor )
+			{
+				behaviorEntry = new ConstructorEntry( classEntry, behavior.getSignature() );
+			}
+			else
+			{
+				throw new Error( "Unsupported behavior type: " + behavior.getClass().getName() );
+			}
+			
 			// get the list of parameter names
-			MethodEntry methodEntry = new MethodEntry( classEntry, behavior.getMethodInfo().getName(), behavior.getSignature() );
 			List<String> names = new ArrayList<String>( numParams );
 			for( int i=0; i<numParams; i++ )
 			{
-				names.add( m_translator.translate( new ArgumentEntry( methodEntry, i, "" ) ) );
+				names.add( m_translator.translate( new ArgumentEntry( behaviorEntry, i, "" ) ) );
 			}
 			
 			// save the mappings to the class
-- 
cgit v1.2.3