Simply method generation

1 files changed, 32 insertions, 0 deletions

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);
+                    }
+                };
+            }
+        };
+    }
+}