From ed9b5cdfc648e86fd463bfa8d86b94c41671e14c Mon Sep 17 00:00:00 2001
From: jeff
Date: Sun, 8 Feb 2015 21:29:25 -0500
Subject: switch all classes to new signature/type system

---
 src/cuchaz/enigma/mapping/JavassistUtil.java | 83 ++++++++++++++++++++++++++++
 1 file changed, 83 insertions(+)
 create mode 100644 src/cuchaz/enigma/mapping/JavassistUtil.java

(limited to 'src/cuchaz/enigma/mapping/JavassistUtil.java')

diff --git a/src/cuchaz/enigma/mapping/JavassistUtil.java b/src/cuchaz/enigma/mapping/JavassistUtil.java
new file mode 100644
index 0000000..0c446c4
--- /dev/null
+++ b/src/cuchaz/enigma/mapping/JavassistUtil.java
@@ -0,0 +1,83 @@
+package cuchaz.enigma.mapping;
+
+import javassist.CtBehavior;
+import javassist.CtClass;
+import javassist.CtConstructor;
+import javassist.CtField;
+import javassist.CtMethod;
+import javassist.bytecode.Descriptor;
+import javassist.expr.ConstructorCall;
+import javassist.expr.FieldAccess;
+import javassist.expr.MethodCall;
+import javassist.expr.NewExpr;
+
+public class JavassistUtil {
+	
+	public static ClassEntry getClassEntry(CtClass c) {
+		return new ClassEntry(Descriptor.toJvmName(c.getName()));
+	}
+	
+	public static ClassEntry getSuperclassEntry(CtClass c) {
+		return new ClassEntry(Descriptor.toJvmName(c.getClassFile().getSuperclass()));
+	}
+	
+	public static MethodEntry getMethodEntry(CtMethod method) {
+		return new MethodEntry(
+			getClassEntry(method.getDeclaringClass()),
+			method.getName(),
+			new Signature(method.getMethodInfo().getDescriptor())
+		);
+	}
+	
+	public static MethodEntry getMethodEntry(MethodCall call) {
+		return new MethodEntry(
+			new ClassEntry(Descriptor.toJvmName(call.getClassName())),
+			call.getMethodName(),
+			new Signature(call.getSignature())
+		);
+	}
+	
+	public static ConstructorEntry getConstructorEntry(CtConstructor constructor) {
+		return new ConstructorEntry(
+			getClassEntry(constructor.getDeclaringClass()),
+			new Signature(constructor.getMethodInfo().getDescriptor())
+		);
+	}
+	
+	public static ConstructorEntry getConstructorEntry(ConstructorCall call) {
+		return new ConstructorEntry(
+			new ClassEntry(Descriptor.toJvmName(call.getClassName())),
+			new Signature(call.getSignature())
+		);
+	}
+	
+	public static ConstructorEntry getConstructorEntry(NewExpr call) {
+		return new ConstructorEntry(
+			new ClassEntry(Descriptor.toJvmName(call.getClassName())),
+			new Signature(call.getSignature())
+		);
+	}
+	
+	public static BehaviorEntry getBehaviorEntry(CtBehavior behavior) {
+		if (behavior instanceof CtMethod) {
+			return getMethodEntry((CtMethod)behavior);
+		} else if (behavior instanceof CtConstructor) {
+			return getConstructorEntry((CtConstructor)behavior);
+		}
+		throw new Error("behavior is neither Method nor Constructor!");
+	}
+	
+	public static FieldEntry getFieldEntry(CtField field) {
+		return new FieldEntry(
+			getClassEntry(field.getDeclaringClass()),
+			field.getName()
+		);
+	}
+	
+	public static FieldEntry getFieldEntry(FieldAccess call) {
+		return new FieldEntry(
+			new ClassEntry(Descriptor.toJvmName(call.getClassName())),
+			call.getFieldName()
+		);
+	}
+}
-- 
cgit v1.2.3