20 | * This class is copied from Intimate . 21 | *
22 | * https://github.com/JustKiddingBaby/Intimate 23 | */ 24 | public class JarUtils { 25 | 26 | /** 27 | * 返回推荐的 jar 解压目录 28 | * 29 | * @param path jar 文件路径 30 | * @return 解压的 jar 路径 31 | */ 32 | public static String getExtractJarPath(String path) { 33 | return getExtractJarPath(new File(path)); 34 | } 35 | 36 | /** 37 | * 返回推荐的 jar 解压目录 38 | * 39 | * @param jarFile jar 文件 40 | * @return 解压的 jar 路径 41 | */ 42 | public static String getExtractJarPath(File jarFile) { 43 | return jarFile.getParent() + "/" + jarFile.getName().replace(".jar", ""); 44 | } 45 | 46 | /** 47 | * 压缩 jar目录 48 | * @param unJarFile 49 | * @return 50 | */ 51 | public static String getExtractJarOriginPath(File unJarFile) { 52 | return unJarFile.getParent() + "/" + unJarFile.getName() + ".jar"; 53 | } 54 | 55 | public static void jar(File desJar, File jarDir) throws Exception { 56 | BufferedOutputStream out = new BufferedOutputStream(new FileOutputStream(desJar)); 57 | File[] src = jarDir.listFiles(); 58 | jar(out, src); 59 | } 60 | 61 | public static void jar(OutputStream out, File[] src) throws Exception { 62 | jar(out, src, null, null); 63 | } 64 | 65 | public static void jar(OutputStream out, File src) throws Exception { 66 | jar(out, new File[]{src}, null, null); 67 | } 68 | 69 | public static void jar(OutputStream out, File[] src, String prefix, Manifest man) throws Exception { 70 | JarOutputStream jout = null; 71 | if (man == null) { 72 | jout = new JarOutputStream(out); 73 | } else { 74 | jout = new JarOutputStream(out, man); 75 | } 76 | 77 | if (prefix != null && prefix.trim().length() > 0 && !prefix.equals("/")) { 78 | if (prefix.charAt(0) == '/') { 79 | prefix = prefix.substring(1); 80 | } 81 | if (prefix.charAt(prefix.length() - 1) != '/') { 82 | prefix = prefix + "/"; 83 | } 84 | } else { 85 | prefix = ""; 86 | } 87 | for (File f : src) { 88 | jar(f, prefix, jout); 89 | } 90 | jout.close(); 91 | } 92 | 93 | private static void jar(File src, String prefix, JarOutputStream jout) throws Exception { 94 | if (src.isDirectory()) { 95 | prefix = prefix + src.getName() + "/"; 96 | ZipEntry entry = new ZipEntry(prefix); 97 | entry.setTime(src.lastModified()); 98 | entry.setMethod(ZipEntry.STORED); 99 | entry.setSize(0l); 100 | entry.setCrc(0l); 101 | jout.putNextEntry(entry); 102 | jout.closeEntry(); 103 | File[] files = src.listFiles(); 104 | if (files != null) { 105 | for (File file : files) { 106 | jar(file, prefix, jout); 107 | } 108 | } 109 | } else { 110 | byte[] buffer = new byte[8092]; 111 | ZipEntry entry = new ZipEntry(prefix + src.getName()); 112 | entry.setTime(src.lastModified()); 113 | jout.putNextEntry(entry); 114 | FileInputStream inputStream = new FileInputStream(src); 115 | int len; 116 | while ((len = inputStream.read(buffer, 0, buffer.length)) != -1) { 117 | jout.write(buffer, 0, len); 118 | } 119 | inputStream.close(); 120 | jout.closeEntry(); 121 | } 122 | } 123 | 124 | public static List unJar(File jarFile, File unJarDir) throws Exception { 125 | List list = new ArrayList<>(); 126 | 127 | BufferedInputStream inputStream = new BufferedInputStream(new FileInputStream(jarFile)); 128 | unJar(inputStream, unJarDir, list); 129 | 130 | return list; 131 | } 132 | 133 | public static List unJar(InputStream inputStream, File unJarDir, List list) throws Exception { 134 | if (!unJarDir.exists()) { 135 | unJarDir.mkdirs(); 136 | } 137 | 138 | JarInputStream jin = new JarInputStream(inputStream); 139 | byte[] buffer = new byte[8092]; 140 | 141 | ZipEntry entry = jin.getNextEntry(); 142 | while (entry != null) { 143 | String fileName = entry.getName(); 144 | if (File.separatorChar != '/') { 145 | fileName = fileName.replace('/', File.separatorChar); 146 | } 147 | if (fileName.charAt(fileName.length() - 1) == '/') { 148 | fileName = fileName.substring(0, fileName.length() - 1); 149 | } 150 | if (fileName.charAt(0) == '/') { 151 | fileName = fileName.substring(1); 152 | } 153 | String entryName = entry.getName(); 154 | if (entryName.endsWith(".class")) { 155 | String className = entryName.replace('\\', '.').replace('/', '.'); 156 | list.add(className); 157 | } 158 | 159 | File file = new File(unJarDir, fileName); 160 | if (!file.getName().endsWith(".class")) { 161 | boolean re = file.mkdirs(); 162 | } else { 163 | File parent = file.getParentFile(); 164 | if (!parent.exists()) { 165 | parent.mkdirs(); 166 | } 167 | BufferedOutputStream out = new BufferedOutputStream(new FileOutputStream(file)); 168 | int len = 0; 169 | while ((len = jin.read(buffer, 0, buffer.length)) != -1) { 170 | out.write(buffer, 0, len); 171 | } 172 | out.flush(); 173 | out.close(); 174 | file.setLastModified(entry.getTime()); 175 | } 176 | jin.closeEntry(); 177 | entry = jin.getNextEntry(); 178 | } 179 | 180 | Manifest mf = jin.getManifest(); 181 | if (mf != null) { 182 | File file = new File(unJarDir, "META-INF/MANIFEST.MF"); 183 | File parent = file.getParentFile(); 184 | if (!parent.exists()) { 185 | parent.mkdirs(); 186 | } 187 | OutputStream out = new FileOutputStream(file); 188 | mf.write(out); 189 | out.flush(); 190 | out.close(); 191 | } 192 | 193 | jin.close(); 194 | 195 | return list; 196 | } 197 | 198 | } 199 | -------------------------------------------------------------------------------- /mock-method-plugin/.idea/workspace.xml: -------------------------------------------------------------------------------- 1 | 2 | 3 | 4 | 5 | 6 | 7 | 8 | 9 | 10 | 11 | 12 | 13 | 14 | 15 | 16 | 17 | 18 | 19 | 20 | 21 | 22 | 23 | 24 | 25 | 26 | 27 | 28 | 29 | 30 | 31 | 32 | 33 | 34 | 35 | 36 | 37 | 38 | 39 | 40 | 41 | 42 | 43 | 44 | 45 | Android 46 | 47 | 48 | 49 | 50 | 51 | 52 | 53 | 54 | 55 | 56 | 57 | 58 | 59 | 60 | 61 | 62 | 63 | 64 | 65 | 66 | 67 | 68 | 69 | 70 | 71 | 72 | 73 | 74 | 75 | 76 | 77 | 78 | 79 | 80 | 81 | 82 | 83 | 84 | 85 | 86 | 87 | 88 | 89 | 90 | 91 | 92 | 93 | 94 | 95 | 96 | 97 | 98 | 99 | 100 | 101 | 102 | 103 | 104 | 105 | 106 | 107 | 108 | 109 | 110 | 111 | 112 | 113 | 114 | 115 | 116 | 117 | 118 | 119 | 120 | 121 | 122 | 123 | 124 | 125 | 126 | 127 | 128 | 129 | 130 | 131 | 132 | 133 | 134 | 135 | 136 | 137 | 138 | 139 | 140 | 141 | 1544951263083 142 | 143 | 144 | 1544951263083 145 | 146 | 147 | 148 | 149 | 150 | 151 | 152 | 153 | 154 | 155 | 156 | 157 | 158 | 159 | 160 | 161 | 162 | 163 | 164 | 165 | 166 | 167 | 168 | 169 | 170 | 171 | 172 | 173 | 174 | 175 | 176 | 177 | 178 | 179 | 180 | 181 | 182 | 183 | 184 | 185 | 186 | 187 | 188 | 1.8 189 | 190 | 191 | 192 | 193 | 194 | 195 | 196 | 197 | 198 | 199 | 200 | -------------------------------------------------------------------------------- /mock-method-plugin/src/main/groovy/com/mock/internel/Inject.groovy: -------------------------------------------------------------------------------- 1 | package com.mock.internel 2 | 3 | import com.mock.annotation.MockMethod 4 | import javassist.ClassPool 5 | import javassist.CtClass 6 | import javassist.CtMethod 7 | import org.gradle.api.Project 8 | 9 | 10 | class Inject { 11 | private static ClassPool pool = ClassPool.getDefault() 12 | private static List generatorMockMap = new ArrayList<>(); 13 | 14 | static void addPoolPath(String path) { 15 | pool.appendClassPath(path) 16 | } 17 | 18 | static void generatorMockMap(String path, Project project) { 19 | File jarFile = new File(path) 20 | String jarZipDir = JarUtils.getExtractJarPath(jarFile) 21 | File unJar = new File(jarZipDir) 22 | List strings = JarUtils.unJar(jarFile, unJar) 23 | addPoolPath(unJar.path) 24 | //todo inject代码 25 | if (strings.contains('com.mock.generator.MockMethodGenerator.class')) { 26 | CtClass ctClass = pool.getCtClass('com.mock.generator.MockMethodGenerator') 27 | try { 28 | CtMethod ctMethod = ctClass.getDeclaredMethod('initMackMock') 29 | 30 | StringBuilder builder = new StringBuilder() 31 | generatorMockMap.each { 32 | builder.append(getMockMapDes(it.className, it.methodName, it.values, it.defaultValue) + "\n") 33 | } 34 | println '方法' + builder.toString() 35 | ctMethod.insertBefore(builder.toString()) 36 | ctClass.writeFile(unJar.path) 37 | ctClass.detach() 38 | } catch (Exception e) { 39 | e.printStackTrace() 40 | } finally { 41 | generatorMockMap.clear() 42 | } 43 | } 44 | println '解压路径:' + new File(jarZipDir).path 45 | println '压缩路径:' + new File(path).path 46 | jarFile.delete() 47 | JarUtils.jar(jarFile, unJar) 48 | } 49 | 50 | private 51 | static String getMockMapDes(String className, String methodName, String values, String defaultValue) { 52 | return String.format("mockMethodModels.add(getMockMap(\"%s\",\"%s\",\"%s\",\"%s\"));", className, methodName, values, defaultValue) 53 | } 54 | 55 | static void injectMockMap(String className, String methodName, String values, String defaultValue) { 56 | MockMethodModel mockMethodModel = new MockMethodModel() 57 | mockMethodModel.values = values 58 | mockMethodModel.defaultValue = defaultValue 59 | mockMethodModel.methodName = methodName 60 | mockMethodModel.className = className 61 | generatorMockMap.add(mockMethodModel) 62 | } 63 | 64 | 65 | static void injectJarPath(String path, Project project) { 66 | //注入依赖与generator中的类，所以注入jar path 67 | project.extensions[MockExtension.plugin].subprojects.each { 68 | if (path.contains(it)) { 69 | File jarFile = new File(path) 70 | String jarZipDir = JarUtils.getExtractJarPath(jarFile) 71 | File unJar = new File(jarZipDir) 72 | List strings = JarUtils.unJar(jarFile, unJar) 73 | injectDir(unJar.path, project) 74 | println '解压路径:' + new File(jarZipDir).path 75 | println '压缩路径:' + new File(path).path 76 | jarFile.delete() 77 | JarUtils.jar(jarFile, unJar) 78 | } 79 | } 80 | 81 | } 82 | 83 | 84 | static void injectDir(String path, Project project) { 85 | println("--------path----:" + path) 86 | addPoolPath(path) 87 | processorDir(path, project) 88 | } 89 | 90 | private static void processorDir(String path, Project project) { 91 | File dir = new File(path) 92 | if (dir.isDirectory()) { 93 | dir.eachFileRecurse { File file -> 94 | String classPath = getClassPath(file, path) 95 | String className = classPath.replace('\\', '.').replace('/', '.') 96 | if (className.endsWith(".class") 97 | && !className.contains('R$') 98 | && !className.contains('R.class') 99 | && !className.contains("BuildConfig.class")) { 100 | 101 | project.extensions[MockExtension.plugin].packages.each { String pkg -> 102 | if (className.contains(pkg)) { 103 | className = className.substring(0, className.length() - 6) 104 | println 'className:' + className 105 | try { 106 | injectClass(path, className) 107 | } catch (Exception e) { 108 | e.printStackTrace() 109 | } 110 | } 111 | } 112 | } 113 | } 114 | } 115 | } 116 | 117 | private static String getClassPath(File file, String path) { 118 | String filePath = file.absolutePath 119 | return filePath.substring(path.length() + 1, filePath.length()) 120 | 121 | } 122 | 123 | 124 | private static void injectClass(String path, String className) { 125 | CtClass ctClass = pool.get(className) 126 | CtMethod[] methods = ctClass.getDeclaredMethods() 127 | if (methods == null || methods.length == 0) { 128 | return 129 | } 130 | methods.each { CtMethod method -> 131 | MockMethod mockMethod = method.getAnnotation(MockMethod.class) 132 | if (mockMethod == null) { 133 | return 134 | } 135 | println "method return : " + method.returnType.name 136 | if (!isAllow(method.returnType.name)) { 137 | return 138 | } 139 | injectMockMap(className, method.name, mockMethod.values(), mockMethod.defaultValue()) 140 | println("methodDes:" + getMethodDes(className, method.name, method.returnType.name)) 141 | 142 | method.insertBefore(getMethodDes(className, method.name, method.returnType.name)) 143 | 144 | } 145 | ctClass.writeFile(path) 146 | ctClass.detach() 147 | } 148 | static List dataType = [ 149 | String.class.name, 150 | int.class.name, 151 | long.class.name, 152 | double.class.name, 153 | float.class.name, 154 | boolean.class.name, 155 | short.class.name, 156 | Integer.class.name, 157 | Long.class.name, 158 | Double.class.name, 159 | Float.class.name, 160 | Boolean.class.name, 161 | Short.class.name,].collect() 162 | 163 | static boolean isAllow(String returnType) { 164 | if (returnType == null) { 165 | return false 166 | } 167 | 168 | 169 | 170 | if (dataType.contains(returnType)) { 171 | return true 172 | } 173 | return false 174 | } 175 | 176 | private static String getMethodDes(String className, String methodName, String returnType) { 177 | return String.format("if (com.mock.generator.MockManager.isMock(\"%s\", \"%s\")) {\n" + 178 | " String value = com.mock.generator.MockManager.getValue(\"%s\", \"%s\");\n" + 179 | " return " + getReturnValueByStr(returnType) + ";\n" + 180 | " }", className, methodName, className, methodName) 181 | } 182 | 183 | private static String getReturnValueByStr(String returnType) { 184 | if (returnType.contains("String")) { 185 | return 'value' 186 | } else if (returnType.contains("int")) { 187 | return 'Integer.parseInt(value)' 188 | } else if (returnType.contains("long")) { 189 | return 'Long.parseLong(value)' 190 | } else if (returnType.contains("double")) { 191 | return 'Double.parseDouble(value)' 192 | } else if (returnType.contains("float")) { 193 | return 'Float.parseFloat(value)' 194 | } else if (returnType.contains("boolean")) { 195 | return 'Boolean.parseBoolean(value)' 196 | } else if (returnType.contains("short")) { 197 | return 'Short.parseShort(value)' 198 | } else if (returnType.contains("Integer")) { 199 | return 'Integer.valueOf(value)' 200 | } else if (returnType.contains("Long")) { 201 | return 'Long.valueOf(value)' 202 | } else if (returnType.contains("Double")) { 203 | return 'Double.valueOf(value)' 204 | } else if (returnType.contains("Float")) { 205 | return 'Float.valueOf(value)' 206 | } else if (returnType.contains("Boolean")) { 207 | return 'Boolean.valueOf(value)' 208 | } else if (returnType.contains("Short")) { 209 | return 'Short.valueOf(value)' 210 | } 211 | } 212 | 213 | 214 | } 215 | --------------------------------------------------------------------------------