Refactor MediaQuality and ReflectionUtils to improve enum handling and type safety

Author: Dev4Mod

app/src/main/java/com/wmods/wppenhacer/xposed/features/media/MediaQuality.java

@@ -90,10 +90,12 @@ protected void afterHookedMethod(MethodHookParam param) throws Throwable {
                     var resizeVideo = param.getResult();
                     boolean isHighResolution;
                     boolean isEnum = false;
-                    if (param.args[0] instanceof Enum<?>) {
+                    var enumObj = ReflectionUtils.getArg(param.args, Enum.class, 0);
+                    var intParams = ReflectionUtils.findArrayOfType(param.args, Integer.class);
+                    if (enumObj != null) {
                         isEnum = true;
-                        var hightResolution = Enum.valueOf((Class<Enum>) param.args[0].getClass(), "RESOLUTION_1080P");
-                        isHighResolution = hightResolution == param.args[0];
+                        var hightResolution = Enum.valueOf((Class<Enum>) enumObj.getClass(), "RESOLUTION_1080P");
+                        isHighResolution = hightResolution == enumObj;
                     } else {
                         isHighResolution = (int) param.args[1] == 3;
                     }
@@ -106,9 +108,9 @@ protected void afterHookedMethod(MethodHookParam param) throws Throwable {
 
                             if (mediaFields.isEmpty()) {
                                 if (isEnum) {
-                                    width = (int) param.args[3];
-                                    height = (int) param.args[4];
-                                    rotationAngle = (int) param.args[5];
+                                    width = intParams.get(2).second;
+                                    height = intParams.get(3).second;
+                                    rotationAngle = intParams.get(4).second;
                                 } else {
                                     JSONObject mediaFields = (JSONObject) XposedHelpers.callMethod(param.args[0], "A00");
                                     width = mediaFields.getInt("widthPx");

app/src/main/java/com/wmods/wppenhacer/xposed/utils/ReflectionUtils.java

@@ -267,19 +267,19 @@ public static int findIndexOfType(Object[] args, Class<?> type) {
         return -1;
     }
 
-    public static List<Pair<Integer, Object>> findArrayOfType(Object[] args, Class<?> type) {
-        var result = new ArrayList<Pair<Integer, Object>>();
+    public static <T> List<Pair<Integer, T>> findArrayOfType(Object[] args, Class<T> type) {
+        var result = new ArrayList<Pair<Integer, T>>();
         for (int i = 0; i < args.length; i++) {
             var arg = args[i];
             if (arg == null) continue;
-            if (arg instanceof Class) {
-                if (type.isAssignableFrom((Class) arg)) {
-                    result.add(new Pair<>(i, arg));
+            if (arg instanceof Class clazz) {
+                if (type.isAssignableFrom(clazz)) {
+                    result.add(new Pair<>(i, type.cast(clazz)));
                 }
                 continue;
             }
             if (type.isAssignableFrom(arg.getClass()) || type.isInstance(arg)) {
-                result.add(new Pair<>(i, arg));
+                result.add(new Pair<>(i, type.cast(arg)));
             }
         }
         return result;
@@ -334,7 +334,7 @@ public synchronized static boolean isCalledFromMethod(Method method) {
     public static <T> T getArg(Object[] args, Class<T> typeClass, int i) {
         var list = findArrayOfType(args, typeClass);
         if (list.size() <= i) return null;
-        return typeClass.cast(list.get(i).second);
+        return list.get(i).second;
     }
 
     public static void setObjectField(Field field, Object instance, Object value) {