From 9ff6b49ca2c559d6938715fd0b48bb8c3e2c1d19 Mon Sep 17 00:00:00 2001
From: Michael Williamson <mike@zwobble.org>
Date: Thu, 14 Jul 2016 15:16:47 +0100
Subject: Simply method generation

---
 .../zwobble/couscous/util/asm/Implementations.java | 32 ++++++++++++++++++++++
 1 file changed, 32 insertions(+)
 create mode 100644 src/main/java/org/zwobble/couscous/util/asm/Implementations.java

(limited to 'src/main/java/org/zwobble/couscous/util/asm/Implementations.java')

diff --git a/src/main/java/org/zwobble/couscous/util/asm/Implementations.java b/src/main/java/org/zwobble/couscous/util/asm/Implementations.java
new file mode 100644
index 0000000..ee6911c
--- /dev/null
+++ b/src/main/java/org/zwobble/couscous/util/asm/Implementations.java
@@ -0,0 +1,32 @@
+package org.zwobble.couscous.util.asm;
+
+import net.bytebuddy.description.method.MethodDescription;
+import net.bytebuddy.dynamic.scaffold.InstrumentedType;
+import net.bytebuddy.implementation.Implementation;
+import net.bytebuddy.implementation.bytecode.ByteCodeAppender;
+import net.bytebuddy.implementation.bytecode.StackManipulation;
+import net.bytebuddy.jar.asm.MethodVisitor;
+
+import java.util.function.Function;
+
+public class Implementations {
+    public static Implementation stackManipulation(Function<Implementation.Target, StackManipulation> build) {
+        return new Implementation() {
+            @Override
+            public InstrumentedType prepare(InstrumentedType instrumentedType) {
+                return instrumentedType;
+            }
+
+            @Override
+            public ByteCodeAppender appender(Target implementationTarget) {
+                return new ByteCodeAppender() {
+                    @Override
+                    public Size apply(MethodVisitor methodVisitor, Context implementationContext, MethodDescription instrumentedMethod) {
+                        return new ByteCodeAppender.Simple(build.apply(implementationTarget))
+                            .apply(methodVisitor, implementationContext, instrumentedMethod);
+                    }
+                };
+            }
+        };
+    }
+}
-- 
cgit v1.2.3