[VXP]: refactor code.

Author: tiann

VirtualApp/lib/src/main/java/com/lody/virtual/client/hook/proxies/view/AutoFillManagerStub.java

@@ -4,10 +4,8 @@
 import android.content.ComponentName;
 import android.util.Log;
 
-import com.lody.virtual.client.core.VirtualCore;
 import com.lody.virtual.client.hook.base.BinderInvocationProxy;
-import com.lody.virtual.client.hook.base.MethodProxy;
-import com.lody.virtual.client.hook.utils.MethodParameterUtils;
+import com.lody.virtual.client.hook.base.ReplaceLastPkgMethodProxy;
 import com.lody.virtual.helper.utils.ArrayUtils;
 
 import java.lang.reflect.Field;
@@ -48,63 +46,35 @@ public void inject() throws Throwable {
             Log.e(TAG, "AutoFillManagerStub inject error.", tr);
             return;
         }
-        addMethodProxy(new MethodProxy() {
-            @Override
-            public String getMethodName() {
-                return "startSession";
-            }
-            @Override
-            public boolean beforeCall(Object who, Method method, Object... args) {
-                if (args == null) {
-                    return super.beforeCall(who, method, args);
-                }
-                int length = args.length;
-                if (length == 0) {
-                    return super.beforeCall(who, method, args);
-                }
-
-                int lastIndex = length - 1;
-                Object lastParam = args[lastIndex];
-                if (lastParam instanceof String) {
-                    MethodParameterUtils.replaceLastAppPkg(args);
-                } else if (lastParam instanceof ComponentName){
-                    ComponentName lastComponent = (ComponentName) lastParam;
-                    ComponentName newComponent = new ComponentName(VirtualCore.get().getHostPkg(),
-                            lastComponent.getClassName());
-                    args[lastIndex] = newComponent;
-                }
-                return super.beforeCall(who, method, args);
-            }
-        });
 
-        addMethodProxy(new MethodProxy() {
-            @Override
-            public String getMethodName() {
-                return "updateOrRestartSession";
-            }
-            @Override
-            public boolean beforeCall(Object who, Method method, Object... args) {
-                MethodParameterUtils.replaceLastAppPkg(args);
-                int index = ArrayUtils.indexOfLast(args, ComponentName.class);
-                if (index > 0) {
-                    ComponentName origComponent = (ComponentName) args[index];
-                    ComponentName newComponent = new ComponentName(getHostPkg(), origComponent.getClassName());
-                    args[index] = newComponent;
-                }
-                return super.beforeCall(who, method, args);
-            }
-        });
-        addMethodProxy(new MethodProxy() {
-            @Override
-            public String getMethodName() {
-                return "isServiceEnabled";
-            }
-            @Override
-            public boolean beforeCall(Object who, Method method, Object... args) {
-                MethodParameterUtils.replaceLastAppPkg(args);
-                return super.beforeCall(who, method, args);
+        addMethodProxy(new ReplacePkgAndComponentProxy("startSession"));
+        addMethodProxy(new ReplacePkgAndComponentProxy("updateOrRestartSession"));
+        addMethodProxy(new ReplaceLastPkgMethodProxy("isServiceEnabled"));
+    }
+
+    static class ReplacePkgAndComponentProxy extends ReplaceLastPkgMethodProxy {
+
+        ReplacePkgAndComponentProxy(String name) {
+            super(name);
+        }
+
+        @Override
+        public boolean beforeCall(Object who, Method method, Object... args) {
+            replaceLastAppComponent(args, getHostPkg());
+            return super.beforeCall(who, method, args);
+        }
+
+        static ComponentName replaceLastAppComponent(Object[] args, String hostPkg) {
+            int index = ArrayUtils.indexOfLast(args, ComponentName.class);
+            if (index != -1) {
+                ComponentName orig = (ComponentName) args[index];
+                ComponentName newComponent = new ComponentName(hostPkg, orig.getClassName());
+                args[index] = newComponent;
+                return newComponent;
             }
-        });
+            return null;
+        }
     }
 
+
 }