增加IPassContext抽象

Author: luiox

.idea/uiDesigner.xml

@@ -11,14 +11,14 @@ public interface AbstractPass {
      *
      * @param context 上下文
      */
-    default void doInitialization(@NotNull PassContext context) {
+    default void doInitialization(@NotNull IPassContext context) {
     }
 
     /**
      * 在所有转换之后统一的清理操作
      *
      * @param context 上下文
      */
-    default void doFinalization(@NotNull PassContext context) {
+    default void doFinalization(@NotNull IPassContext context) {
     }
 }

morpher-api/src/main/java/com/github/luiox/morpher/transformer/AbstractPass.java

@@ -0,0 +1,110 @@
+package com.github.luiox.morpher.transformer;
+
+import com.github.luiox.morpher.jar.IJarCaches;
+import com.github.luiox.morpher.jar.JarCachesEntry;
+import com.github.luiox.morpher.jar.JarCachesEntryType;
+import org.jetbrains.annotations.NotNull;
+import org.jetbrains.annotations.Nullable;
+import org.objectweb.asm.tree.ClassNode;
+
+import java.util.ArrayList;
+import java.util.HashMap;
+import java.util.List;
+import java.util.Map;
+import java.util.concurrent.atomic.AtomicInteger;
+import java.util.function.Predicate;
+
+public class BasicPassContext implements IPassContext{
+    // pass声明周期操作
+    private Map<AbstractPass, LifecycleHook> lifecycleHookMap = new HashMap<>();
+
+    /**
+     * 生命周期钩子注册
+     */
+    public LifecycleHook regLifecycleHook(AbstractPass owner) {
+        if (!lifecycleHookMap.containsKey(owner)) {
+            lifecycleHookMap.put(owner, new LifecycleHook());
+        }
+        return lifecycleHookMap.get(owner);
+    }
+
+    public void runBeforeAllClassesHooks() {
+        for (var entry : lifecycleHookMap.entrySet()) {
+            var hook = entry.getValue().beforeAllClassesHook;
+            if (hook != null) {
+                hook.run();
+            }
+        }
+    }
+
+    public void runAfterAllClassesHook() {
+        for (var entry : lifecycleHookMap.entrySet()) {
+            var hook = entry.getValue().afterAllClassesHook;
+            if (hook != null) {
+                hook.run();
+            }
+        }
+    }
+
+    public void setNewClassNode(ClassNode newNode) {
+        currentClass = newNode;
+    }
+
+    public static class LifecycleHook {
+        Runnable beforeAllClassesHook;
+        Runnable afterAllClassesHook;
+
+        public LifecycleHook beforeAllClasses(Runnable hook) {
+            beforeAllClassesHook = hook;
+            return this;
+        }
+
+        public LifecycleHook afterAllClasses(Runnable hook) {
+            afterAllClassesHook = hook;
+            return this;
+        }
+    }
+
+    // 计数器
+    private Map<Class<?>, Map<String, AtomicInteger>> passCounters = new HashMap<>();
+
+    public AtomicInteger getCounter(Class<?> clazz, String name) {
+        passCounters.computeIfAbsent(clazz, k -> new HashMap<>()).computeIfAbsent(name, k -> new AtomicInteger());
+        return passCounters.get(clazz).get(name);
+    }
+
+    // 基础的上下文信息
+    private ClassNode currentClass;
+    // 如果为null就不过滤，其次就是传入一个谓词，用于判断是否需要过滤，如果类名让这个谓词返回true那么就被保留
+    private Predicate<String> currentClassFilter;
+
+    private AbstractPass prevPass;
+
+    public void setCurrentClassFilter(Predicate<String> currentClassFilter) {
+        this.currentClassFilter = currentClassFilter;
+    }
+
+    public Predicate<String> getCurrentClassFilter() {
+        return currentClassFilter;
+    }
+
+    @Override
+    public @NotNull ClassNode currentClass() {
+        return currentClass;
+    }
+
+    @Override
+    public void setCurrentClass(@NotNull ClassNode classNode) {
+        currentClass = classNode;
+    }
+
+    @Override
+    public @Nullable AbstractPass prevPass() {
+        return prevPass;
+    }
+
+    @Override
+    public void setPrevPass(@NotNull AbstractPass pass) {
+        prevPass = pass;
+    }
+}

morpher-api/src/main/java/com/github/luiox/morpher/transformer/BasicPassContext.java

@@ -7,5 +7,5 @@
  * Class级别的Pass
  */
 public abstract class ClassPass implements AbstractPass {
-    public abstract void run(@NotNull ClassNode classNode, @NotNull PassContext context);
+    public abstract void run(@NotNull ClassNode classNode, @NotNull IPassContext context);
 }

morpher-api/src/main/java/com/github/luiox/morpher/transformer/ClassPass.java

@@ -7,5 +7,5 @@
  */
 public abstract class GlobalPass implements AbstractPass {
     // 全局级别的pass
-    public abstract void run(@NotNull PassContext context);
+    public abstract void run(@NotNull IPassContext context);
 }

morpher-api/src/main/java/com/github/luiox/morpher/transformer/GlobalPass.java

@@ -0,0 +1,35 @@
+package com.github.luiox.morpher.transformer;
+
+import org.jetbrains.annotations.NotNull;
+import org.jetbrains.annotations.Nullable;
+import org.objectweb.asm.tree.ClassNode;
+
+public interface IPassContext {
+    /**
+     * 当前正在处理的ClassNode，只对MethodPass和ClassPass有效
+     * @return 当前的ClassNode
+     */
+    @NotNull ClassNode currentClass();
+
+    /**
+     * 设置当前正在处理的ClassNode
+     * @param classNode class
+     */
+    void setCurrentClass(@NotNull ClassNode classNode);
+
+    /**
+     * 上一个pass
+     * @return 如果此pass是第一个pass，那么上一个pass是null
+     */
+    default @Nullable AbstractPass prevPass(){
+        return null;
+    }
+
+    /**
+     * 设置之前运行的pass
+     * @param pass pass
+     */
+    void setPrevPass(@NotNull AbstractPass pass);
+
+
+}

morpher-api/src/main/java/com/github/luiox/morpher/transformer/IPassContext.java

@@ -1,5 +1,5 @@
 package com.github.luiox.morpher.transformer;
 
 public interface IPassRunner {
-    void transform(PassContext context);
+    void transform(IPassContext context);
 }

morpher-api/src/main/java/com/github/luiox/morpher/transformer/IPassRunner.java

@@ -7,5 +7,5 @@
  * 方法级别的Pass
  */
 public abstract class MethodPass implements AbstractPass {
-    public abstract void run(@NotNull MethodNode methodNode, @NotNull PassContext context);
+    public abstract void run(@NotNull MethodNode methodNode, @NotNull IPassContext context);
 }