重定向修改jar

replugin-plugin-gradle/src/main/groovy/com/qihoo360/replugin/gradle/plugin/inner/InjectorVersion.groovy

@@ -0,0 +1,32 @@
+
+package com.qihoo360.replugin.gradle.plugin.inner
+
+import com.qihoo360.replugin.gradle.plugin.AppConstant
+
+/**
+ * @author 247321453
+ */
+public class InjectorVersion {
+
+    def String jarPath
+
+    def String jarMd5
+
+    def String pluginVersion
+
+    InjectorVersion(){
+
+    }
+
+    InjectorVersion(String jarPath, String jarMd5, String pluginVersion) {
+        this.jarPath = jarPath
+        this.jarMd5 = jarMd5
+        this.pluginVersion = pluginVersion
+    }
+
+    InjectorVersion(File jar){
+        this.jarPath = jar.getAbsolutePath()
+        this.jarMd5 = Util.md5File(jar)
+        this.pluginVersion = AppConstant.VER
+    }
+}

replugin-plugin-gradle/src/main/groovy/com/qihoo360/replugin/gradle/plugin/inner/Util.groovy

@@ -22,11 +22,18 @@ import com.android.build.api.transform.JarInput
 import com.android.build.api.transform.TransformInput
 import com.android.build.gradle.internal.scope.GlobalScope
 import com.android.sdklib.IAndroidTarget
+import com.google.common.reflect.TypeToken
+import com.google.gson.Gson
+import com.google.gson.GsonBuilder
+import com.google.gson.stream.JsonReader
+import com.qihoo360.replugin.gradle.plugin.AppConstant
+import org.apache.commons.codec.digest.DigestUtils
 import org.apache.commons.io.FileUtils
 import com.google.common.base.Charsets
 import com.google.common.hash.Hashing
 import org.gradle.api.Project
 
+import java.lang.reflect.Field
 import java.nio.file.Files
 import java.nio.file.Paths
 import java.util.zip.ZipFile
@@ -65,7 +72,8 @@ public class Util {
         def projectDir = project.getRootDir().absolutePath
 
         println ">>> Unzip Jar ..."
-
+        Map<String, InjectorVersion> injectorMap = readJarInjectorHistory(project)
+        boolean needSave = false
         inputs.each { TransformInput input ->
 
             input.directoryInputs.each { DirectoryInput dirInput ->
@@ -95,26 +103,146 @@ public class Util {
                     }
 
                 } else {
+                    //重定向jar
+                    if (jarPath.contains(File.separator + FD_INTERMEDIATES + File.separator + "replugin-jar")) {
+                        //
+                    } else {
+                        String md5 = md5File(jar);
+                        File reJar = new File(project.getBuildDir().path +
+                                File.separator + FD_INTERMEDIATES + File.separator + "replugin-jar"
+                                + File.separator + md5 + ".jar");
+                        String reJarPath = reJar.getAbsolutePath()
+
+                        boolean needInject = false
+                        if (reJar.exists()) {
+                            //检查修改插件版本
+                            InjectorVersion injectorVersion = injectorMap.get(jar.getAbsolutePath());
+                            if (injectorVersion != null) {
+                                if (!AppConstant.VER.equals(injectorVersion.pluginVersion)) {
+                                    //版本变化了
+                                    needInject = true
+                                } else {
+                                    if (!md5.equals(injectorVersion.jarMd5)) {
+                                        //原始jar内容变化
+                                        needInject = true
+                                    }
+                                }
+                            } else {
+                                //无记录
+                                needInject = true
+                            }
+                        } else {
+                            FileUtils.copyFile(jar, reJar)
+                            needInject = true;
+                        }
+                        //设置重定向jar
+                        setJarInput(jarInput, reJar)
+                        if (needInject) {
+                            includeJars << reJarPath
+                            map.put(reJarPath, reJarPath)
+
+                            /* 将 jar 包解压，并将解压后的目录加入 classpath */
+                            // println ">>> 解压Jar${jarPath}"
+                            String jarZipDir = reJar.getParent() + File.separatorChar + reJar.getName().replace('.jar', '')
+                            if (unzip(jarPath, jarZipDir)) {
+                                classPath << jarZipDir
+                                //保存修改的插件版本号
+                                needSave = true
+                                injectorMap.put(jar.getAbsolutePath(), new InjectorVersion(jar))
+
+                                visitor.setBaseDir(jarZipDir)
+                                Files.walkFileTree(Paths.get(jarZipDir), visitor)
+                            }
+                            // 删除 jar
+                            FileUtils.forceDelete(reJar)
+                        } else {
+                            map.remove(reJarPath)
+                            includeJars.remove(reJarPath)
+                        }
+                    }
+                }
+            }
+        }
+        if (needSave) {
+            saveJarInjectorHistory(project, injectorMap)
+        }
+        return classPath
+    }
 
-                    includeJars << jarPath
-                    map.put(jarPath, jarPath)
+    /**
+     * 计算jar的md5
+     */
+    def static md5File(File jar) {
+        FileInputStream fileInputStream = new FileInputStream(jar);
+        String md5 = DigestUtils.md5Hex(fileInputStream);
+        fileInputStream.close()
+        return md5
+    }
 
-                    /* 将 jar 包解压，并将解压后的目录加入 classpath */
-                    // println ">>> 解压Jar${jarPath}"
-                    String jarZipDir = jar.getParent() + File.separatorChar + jar.getName().replace('.jar', '')
-                    if (unzip(jarPath, jarZipDir)) {
-                        classPath << jarZipDir
+    /**
+     * 读取修改jar的记录
+     */
+    def static readJarInjectorHistory(Project project) {
+        File file = new File(project.getBuildDir(), FD_INTERMEDIATES
+                + File.separator + "replugin-jar" + File.separator + "version.json");
+        if (!file.exists()) {
+            return new HashMap<String, InjectorVersion>();
+        }
+        Gson gson = new GsonBuilder()
+                .create();
+        FileReader fileReader = new FileReader(file)
+        JsonReader jsonReader = new JsonReader(fileReader);
+        Map<String, InjectorVersion> injectorMap = gson.fromJson(jsonReader, new TypeToken<Map<String, InjectorVersion>>() {
+        }.getType());
+        jsonReader.close()
+        if (injectorMap == null) {
+            injectorMap = new HashMap<String, InjectorVersion>();
+        }
+        return injectorMap;
+    }
 
-                        visitor.setBaseDir(jarZipDir)
-                        Files.walkFileTree(Paths.get(jarZipDir), visitor)
-                    }
+    /**
+     * 保存修改jar的记录
+     */
+    def static saveJarInjectorHistory(Project project, Map<String, InjectorVersion> injectorMap) {
+        Gson gson = new GsonBuilder()
+                .setPrettyPrinting()
+                .create();
+        File file = new File(project.getBuildDir(), FD_INTERMEDIATES
+                + File.separator + "replugin-jar" + File.separator + "version.json");
+        if (file.exists()) {
+            file.delete()
+        } else {
+            File dir = file.getParentFile();
+            if (!dir.exists()) {
+                dir.mkdirs()
+            }
+        }
+        file.createNewFile()
+        FileWriter fileWriter = new FileWriter(file);
+        String json = gson.toJson(injectorMap);
+        fileWriter.write(json)
+        fileWriter.close()
+    }
 
-                    // 删除 jar
-                    FileUtils.forceDelete(jar)
-                }
+    /**
+     * 反射，修改引用的jar路径
+     */
+    def static setJarInput(JarInput jarInput, File rejar) {
+        Field fileField = null;
+        Class<?> clazz = jarInput.getClass();
+        while (fileField == null && clazz != Object.class) {
+            try {
+                fileField = clazz.getDeclaredField("file");
+            } catch (Exception e) {
+                //ignore
+                clazz = clazz.getSuperclass();
             }
         }
-        return classPath
+        if (fileField != null) {
+            fileField.setAccessible(true);
+            fileField.set(jarInput, rejar);
+        }
     }
 
     /**