com.therouter.plugin.agp8.TheRouterTask Maven / Gradle / Ivy
The newest version!
package com.therouter.plugin.agp8;
import com.google.gson.Gson;
import com.google.gson.GsonBuilder;
import com.google.gson.reflect.TypeToken;
import com.therouter.plugin.AddCodeVisitor;
import com.therouter.plugin.BuildConfig;
import com.therouter.plugin.LogUI;
import com.therouter.plugin.RouteItem;
import com.therouter.plugin.TheRouterExtension;
import com.therouter.plugin.TheRouterInjects;
import com.therouter.plugin.TheRouterPlugin;
import com.therouter.plugin.utils.TheRouterPluginUtils;
import org.codehaus.groovy.runtime.ResourceGroovyMethods;
import org.gradle.api.DefaultTask;
import org.gradle.api.file.Directory;
import org.gradle.api.file.FileTree;
import org.gradle.api.file.RegularFile;
import org.gradle.api.file.RegularFileProperty;
import org.gradle.api.provider.ListProperty;
import org.gradle.api.tasks.InputFiles;
import org.gradle.api.tasks.OutputFile;
import org.gradle.api.tasks.TaskAction;
import org.objectweb.asm.ClassReader;
import org.objectweb.asm.ClassWriter;
import org.objectweb.asm.tree.ClassNode;
import org.objectweb.asm.tree.FieldNode;
import java.io.*;
import java.net.URI;
import java.net.URISyntaxException;
import java.nio.charset.StandardCharsets;
import java.util.*;
import java.util.jar.JarEntry;
import java.util.jar.JarFile;
import java.util.jar.JarOutputStream;
public abstract class TheRouterTask extends DefaultTask {
private TheRouterExtension theRouterExtension;
private File asmTargetFile;
private File allClassFile;
private File flowTaskFile;
private File routeFile;
private boolean isFirst;
@InputFiles
public abstract ListProperty getAllJars();
@InputFiles
public abstract ListProperty getAllDirectories();
@OutputFile
public abstract RegularFileProperty getOutputFile();
public void setTheRouterExtension(TheRouterExtension theRouterExtension) {
this.theRouterExtension = theRouterExtension;
}
public void setAsmTargetFile(File file) {
this.asmTargetFile = file;
}
public void setAllClassFile(File file) {
this.allClassFile = file;
}
public void setFlowTaskFile(File flowTaskFile) {
this.flowTaskFile = flowTaskFile;
}
public void setRouteFile(File routeFile) {
this.routeFile = routeFile;
}
public void setFirst(boolean first) {
isFirst = first;
}
@TaskAction
public void taskAction() throws ClassNotFoundException, IOException {
System.out.println("TheRouter编译插件：" + LogUI.C_BLACK_GREEN.getValue() + "cn.therouter:" + BuildConfig.NAME + ":" + BuildConfig.VERSION + LogUI.E_NORMAL.getValue());
System.out.println("JDK Version::" + System.getProperty("java.version"));
System.out.println("Gradle Version::" + getProject().getGradle().getGradleVersion());
System.out.println("本次是增量编译::" + !isFirst);
System.out.println("checkRouteMap::" + theRouterExtension.checkRouteMap);
System.out.println("checkFlowDepend::" + theRouterExtension.checkFlowDepend);
if (isFirst) {
System.out.println("首次编译速度会慢是正常的，实现原理请查看：\nhttps://kymjs.com/code/2024/10/31/01/\n");
}
System.out.println("----------------------TheRouter build start------------------------------");
theRouterTransform();
System.out.println("----------------------TheRouter build finish-----------------------------");
}
private void theRouterTransform() throws ClassNotFoundException, IOException {
String tempText = "";
if (TheRouterPluginUtils.needCheckRouteItemClass(theRouterExtension.checkRouteMap)) {
tempText = TheRouterPluginUtils.getTextFromFile(allClassFile);
}
final String allClassText = tempText;
final String asmTargetText = TheRouterPluginUtils.getTextFromFile(asmTargetFile);
final String routeText = TheRouterPluginUtils.getTextFromFile(routeFile);
final String flowTaskText = TheRouterPluginUtils.getTextFromFile(flowTaskFile);
File theRouterJar = null;
JarEntry theRouterServiceProvideInjecter = null;
JarOutputStream jarOutput = new JarOutputStream(new BufferedOutputStream(new FileOutputStream(getOutputFile().get().getAsFile())));
Set addedEntries = new HashSet<>();
for (RegularFile file : getAllJars().get()) {
File jar = file.getAsFile();
JarFile jarFile = new JarFile(jar);
for (Enumeration e = jarFile.entries(); e.hasMoreElements(); ) {
JarEntry jarEntry = e.nextElement();
String name = jarEntry.getName();
if (name.contains("META-INF/") || !addedEntries.add(name)) {
// 如果已添加该条目，则跳过
continue;
}
if (!allClassText.contains(name) && TheRouterPluginUtils.needCheckRouteItemClass(theRouterExtension.checkRouteMap)) {
TheRouterPluginUtils.addTextToFile(allClassFile, name, theRouterExtension.debug);
}
if (isFirst && name.contains("TheRouterServiceProvideInjecter")) {
theRouterJar = jar;
theRouterServiceProvideInjecter = jarEntry;
} else {
if (!name.contains("$")) {
if (name.contains(TheRouterInjects.PREFIX_ROUTER_MAP)) {
if (!asmTargetText.contains(name)) {
TheRouterPluginUtils.addTextToFile(asmTargetFile, name, theRouterExtension.debug);
}
ClassReader reader = new ClassReader(jarFile.getInputStream(jarEntry));
ClassNode cn = new ClassNode();
reader.accept(cn, 0);
List fieldList = cn.fields;
for (FieldNode fieldNode : fieldList) {
if (TheRouterInjects.FIELD_ROUTER_MAP.equals(fieldNode.name)) {
String v = fieldNode.value.toString();
if (!routeText.contains(v)) {
TheRouterPluginUtils.addTextToFileIgnoreCheck(routeFile, v, theRouterExtension.debug);
}
}
}
} else if (name.contains(TheRouterInjects.PREFIX_SERVICE_PROVIDER)) {
if (!asmTargetText.contains(name)) {
TheRouterPluginUtils.addTextToFile(asmTargetFile, name, theRouterExtension.debug);
}
if (!theRouterExtension.checkFlowDepend.isEmpty()) {
ClassReader reader = new ClassReader(jarFile.getInputStream(jarEntry));
ClassNode cn = new ClassNode();
reader.accept(cn, 0);
List fieldList = cn.fields;
for (FieldNode fieldNode : fieldList) {
if (TheRouterInjects.FIELD_FLOW_TASK_JSON.equals(fieldNode.name)) {
String v = fieldNode.value.toString();
if (!flowTaskText.contains(v)) {
TheRouterPluginUtils.addTextToFileIgnoreCheck(flowTaskFile, v, theRouterExtension.debug);
}
}
}
}
} else if (name.contains(TheRouterInjects.SUFFIX_AUTOWIRED)) {
if (!asmTargetText.contains(name)) {
TheRouterPluginUtils.addTextToFile(asmTargetFile, name, theRouterExtension.debug);
}
}
}
try (InputStream inputStream = jarFile.getInputStream(jarEntry)) {
jarOutput.putNextEntry(new JarEntry(name));
byte[] buffer = new byte[1024];
int bytesRead;
while ((bytesRead = inputStream.read(buffer)) != -1) {
jarOutput.write(buffer, 0, bytesRead);
}
jarOutput.closeEntry();
} catch (IOException ex) {
ex.printStackTrace();
}
}
}
jarFile.close();
}
for (Directory directory : getAllDirectories().get()) {
for (File file : directory.getAsFileTree()) {
String name = directory.getAsFile().toURI().relativize(file.toURI()).getPath().replace(File.separatorChar, '/');
if (name.contains("META-INF/") || !addedEntries.add(name)) {
// 如果已添加该条目，则跳过
continue;
}
if (!allClassText.contains(name) && TheRouterPluginUtils.needCheckRouteItemClass(theRouterExtension.checkRouteMap)) {
TheRouterPluginUtils.addTextToFile(allClassFile, name, theRouterExtension.debug);
}
if (!name.contains("$")) {
if (name.contains(TheRouterInjects.PREFIX_ROUTER_MAP)) {
if (!asmTargetText.contains(name)) {
TheRouterPluginUtils.addTextToFile(asmTargetFile, name, theRouterExtension.debug);
}
try {
ClassReader reader = new ClassReader(new FileInputStream(file));
ClassNode cn = new ClassNode();
reader.accept(cn, 0);
List fieldList = cn.fields;
for (FieldNode fieldNode : fieldList) {
if (TheRouterInjects.FIELD_ROUTER_MAP.equals(fieldNode.name)) {
String v = fieldNode.value.toString();
if (!routeText.contains(v)) {
TheRouterPluginUtils.addTextToFileIgnoreCheck(routeFile, v, theRouterExtension.debug);
}
}
}
} catch (Exception e) {
e.printStackTrace();
}
} else if (name.contains(TheRouterInjects.PREFIX_SERVICE_PROVIDER)) {
if (!asmTargetText.contains(name)) {
TheRouterPluginUtils.addTextToFile(asmTargetFile, name, theRouterExtension.debug);
}
try {
ClassReader reader = new ClassReader(new FileInputStream(file));
ClassNode cn = new ClassNode();
reader.accept(cn, 0);
List fieldList = cn.fields;
for (FieldNode fieldNode : fieldList) {
if (TheRouterInjects.FIELD_FLOW_TASK_JSON.equals(fieldNode.name)) {
String v = fieldNode.value.toString();
if (!flowTaskText.contains(v)) {
TheRouterPluginUtils.addTextToFileIgnoreCheck(flowTaskFile, v, theRouterExtension.debug);
}
}
}
} catch (Exception e) {
e.printStackTrace();
}
} else if (name.contains(TheRouterInjects.SUFFIX_AUTOWIRED)) {
if (!asmTargetText.contains(name)) {
TheRouterPluginUtils.addTextToFile(asmTargetFile, name, theRouterExtension.debug);
}
}
}
try (FileInputStream inputStream = new FileInputStream(file)) {
jarOutput.putNextEntry(new JarEntry(name));
byte[] buffer = new byte[1024];
int bytesRead;
while ((bytesRead = inputStream.read(buffer)) != -1) {
jarOutput.write(buffer, 0, bytesRead);
}
jarOutput.closeEntry();
} catch (Exception e) {
e.printStackTrace();
}
}
}
if (isFirst && theRouterJar != null && theRouterServiceProvideInjecter != null) {
Map serviceProvideMap = new HashMap<>();
Set autowiredSet = new HashSet<>();
Set routeSet = new HashSet<>();
for (String name : TheRouterPluginUtils.getSetFromFile(asmTargetFile)) {
name = name.substring(0, name.length() - TheRouterInjects.DOT_CLASS.length());
if (name.contains(TheRouterInjects.PREFIX_ROUTER_MAP)) {
routeSet.add(name.trim());
} else if (name.contains(TheRouterInjects.PREFIX_SERVICE_PROVIDER)) {
serviceProvideMap.put(name.trim().substring(2), BuildConfig.VERSION);
} else if (name.contains(TheRouterInjects.SUFFIX_AUTOWIRED)) {
autowiredSet.add(name.trim());
}
}
JarFile jarFile = new JarFile(theRouterJar);
jarOutput.putNextEntry(new JarEntry(theRouterServiceProvideInjecter.getName()));
ClassReader cr = new ClassReader(jarFile.getInputStream(theRouterServiceProvideInjecter));
ClassWriter cw = new ClassWriter(ClassWriter.COMPUTE_FRAMES);
AddCodeVisitor cv = new AddCodeVisitor(cw, serviceProvideMap, autowiredSet, routeSet, false);
cr.accept(cv, ClassReader.SKIP_DEBUG);
byte[] bytes = cw.toByteArray();
jarOutput.write(bytes);
jarOutput.closeEntry();
jarFile.close();
}
jarOutput.close();
Set pageSet = new HashSet<>();
Gson gson = new GsonBuilder().setPrettyPrinting().create();
TheRouterInjects.routeMapStringSet.addAll(TheRouterPluginUtils.getSetFromFile(routeFile));
for (String routeMapString : TheRouterInjects.routeMapStringSet) {
pageSet.addAll(gson.fromJson(routeMapString, new TypeToken>() {
}.getType()));
}
// 让第三方Activity也支持路由，第三方页面的路由表可以在assets中添加
// 获取项目目录下 assets/therouter/routeMap.json 文件
File assetRouteMap = new File(getProject().getProjectDir(), "src/main/assets/therouter/routeMap.json");
// 如果文件存在
if (assetRouteMap.exists()) {
// 如果 checkRouteMap 配置为 DELETE
if (TheRouterPlugin.DELETE.equalsIgnoreCase(theRouterExtension.checkRouteMap)) {
System.out.println("---------TheRouter delete route map------------------------------------------");
// 删除并重新创建 routeMap.json
assetRouteMap.delete();
try {
assetRouteMap.createNewFile();
} catch (IOException e) {
e.printStackTrace();
}
} else {
// 读取 JSON 文件内容
String assetString = null;
try {
assetString = ResourceGroovyMethods.getText(assetRouteMap, StandardCharsets.UTF_8.displayName());
} catch (IOException e) {
e.printStackTrace();
}
System.out.println("---------TheRouter get route map from: /assets/therouter/routeMap.json-------");
try {
// 将 JSON 字符串反序列化为 RouteItem 列表
List assetsList = gson.fromJson(assetString, new TypeToken>() {
}.getType());
if (assetsList == null) {
assetsList = new ArrayList<>();
}
// 如果 assetsList 中的 RouteItem 不在 pageSet 中，则添加到 pageSet
for (RouteItem item : assetsList) {
if (!pageSet.contains(item)) {
pageSet.add(item);
}
}
} catch (Exception e) {
e.printStackTrace();
}
}
} else {
// 如果文件不存在，创建父目录和文件
System.out.println("---------TheRouter route map does not exist: /assets/therouter/routeMap.json-------");
try {
assetRouteMap.getParentFile().mkdirs();
assetRouteMap.createNewFile();
} catch (IOException e) {
e.printStackTrace();
}
}
Map> result = new HashMap<>();
// 遍历 pageSet 中的每个 routeItem
for (RouteItem routeItem : pageSet) {
String url = routeItem.path;
// 检查 URL 是否包含查询参数（?）
if (url.contains("?")) {
try {
// 使用 URI 解析路径
URI uri = new URI(routeItem.path);
// 假设 URI 中有属性方法来获取查询参数（这里的 getProperties 需要替换为具体获取参数的方法）
Map map = extractQueryParams(uri);
// 将查询参数存入 routeItem 的 params 中
routeItem.params.putAll(map);
// 将 URL 去掉查询参数部分
url = url.substring(0, url.indexOf('?'));
} catch (URISyntaxException e) {
e.printStackTrace();
}
}
// 获取当前 URL 对应的 RouteItem 列表
List routeList = result.get(url);
if (routeList == null) {
routeList = new ArrayList<>();
result.put(url, routeList);
}
// 将当前 routeItem 添加到列表中
routeList.add(routeItem);
}
// 检查路由表合法性
for (List routeItems : result.values()) {
String className = null;
// 遍历每个 RouteItem
for (RouteItem routeItem : routeItems) {
if (className == null) {
className = routeItem.className;
} else if (!className.equals(routeItem.className)) {
throw new RuntimeException("Multiple Activity to single Url: " + className + " and " + routeItem.className);
}
// 检查路由表是否为空
if (TheRouterPluginUtils.needCheckRouteItemClass(theRouterExtension.checkRouteMap)) {
boolean classNotFound = true;
// 遍历 mergeClass 以检查 routeItem.className
TheRouterInjects.allClass.addAll(TheRouterPluginUtils.getSetFromFile(allClassFile));
for (String item : TheRouterInjects.allClass) {
// routeItem.className 格式为 com.therouter.demo.shell.TestActivity
// item 格式为 com/therouter/demo/shell/TestActivity
if (item.contains(routeItem.className.replace(".", "/"))) {
classNotFound = false;
break;
}
}
if (classNotFound) {
if (TheRouterPlugin.ERROR.equalsIgnoreCase(theRouterExtension.checkRouteMap)) {
throw new ClassNotFoundException(routeItem.className + " in /assets/therouter/routeMap.json");
} else if (TheRouterPlugin.WARNING.equalsIgnoreCase(theRouterExtension.checkRouteMap)) {
System.out.println(LogUI.C_WARN.getValue() + "[" + routeItem.className + " in /assets/therouter/routeMap.json]" + LogUI.E_NORMAL.getValue());
}
}
}
}
}
List pageList = new ArrayList<>(pageSet);
Collections.sort(pageList);
String json = gson.toJson(pageList);
ResourceGroovyMethods.write(assetRouteMap, json, StandardCharsets.UTF_8.displayName());
System.out.println("---------TheRouter create new route map--------------");
Map> flowTaskDependMap = new HashMap<>();
Set stringSet = TheRouterPluginUtils.getSetFromFile(flowTaskFile);
for (String str : stringSet) {
Map map = TheRouterInjects.gson.fromJson(str, HashMap.class);
TheRouterInjects.flowTaskMap.putAll(map);
}
for (String key : TheRouterInjects.flowTaskMap.keySet()) {
Set value = flowTaskDependMap.get(key);
if (value == null) {
value = new HashSet<>();
}
String dependsOn = TheRouterInjects.flowTaskMap.get(key);
if (dependsOn != null && !dependsOn.isBlank()) {
String[] dependencies = dependsOn.split(",");
for (String depend : dependencies) {
if (!depend.isBlank()) {
value.add(depend.trim());
}
}
}
flowTaskDependMap.put(key, value);
}
if (!theRouterExtension.checkFlowDepend.isEmpty()) {
for (String taskName : flowTaskDependMap.keySet()) {
Set dependencies = flowTaskDependMap.get(taskName);
for (String dependency : dependencies) {
if (!flowTaskDependMap.containsKey(dependency)) {
if (TheRouterPlugin.ERROR.equalsIgnoreCase(theRouterExtension.checkFlowDepend)) {
throw new RuntimeException("\n\n==========================================\n" +
"TheRouter:: FlowTask::\n" +
"Can not found Task: [" + dependency + "] from " + taskName + " dependsOn\n" +
"==========================================\n\n");
} else if (TheRouterPlugin.WARNING.equalsIgnoreCase(theRouterExtension.checkFlowDepend)) {
System.out.println();
System.out.println(LogUI.C_WARN.getValue() + "==========================================" + LogUI.E_NORMAL.getValue());
System.out.println(LogUI.C_WARN.getValue() + "TheRouter:: FlowTask:: " + LogUI.E_NORMAL.getValue());
System.out.println(LogUI.C_WARN.getValue() + "Can not found Task: [" + dependency + "] from " + taskName + " dependsOn" + LogUI.E_NORMAL.getValue());
System.out.println(LogUI.C_WARN.getValue() + "==========================================" + LogUI.E_NORMAL.getValue());
System.out.println();
}
}
}
}
}
// 遍历 flowTaskDependMap 的 keySet 并调用 fillTodoList
for (String key : flowTaskDependMap.keySet()) {
TheRouterPluginUtils.fillTodoList(flowTaskDependMap, key);
}
if (theRouterExtension.showFlowDepend) {
// 再次遍历 flowTaskDependMap 的 keySet 并调用 fillNode
for (String key : flowTaskDependMap.keySet()) {
TheRouterPluginUtils.fillNode(TheRouterPluginUtils.createNode(flowTaskDependMap, key), null);
}
System.out.println();
System.out.println(LogUI.C_WARN.getValue() + "TheRouter:: FlowTask::dependency " + LogUI.E_NORMAL.getValue());
System.out.println(LogUI.C_WARN.getValue() + "==========================================" + LogUI.E_NORMAL.getValue());
// 对 dependStack 排序并打印
List dependStack = new ArrayList<>(TheRouterPluginUtils.dependStack);
Collections.sort(dependStack);
for (String it : dependStack) {
System.out.println(LogUI.C_WARN.getValue() + "[Root --> " + it + "]" + LogUI.E_NORMAL.getValue());
}
System.out.println(LogUI.C_WARN.getValue() + "==========================================" + LogUI.E_NORMAL.getValue());
System.out.println();
}
System.out.println("---------TheRouter check flow task map--------------");
}
private void checkBuildCache() {
if (isFirst) {
if (theRouterExtension.lang.equals("en")) {
throw new RuntimeException("\nTheRouter has module additions or removals; please rebuild it again. \nYou can visit the link for more details：\nhttps://kymjs.com/code/2024/10/31/01/\n\n\n");
} else {
throw new RuntimeException("\nTheRouter 有模块增减，请再构建一次。\n可访问链接查看详细原因：\nhttps://kymjs.com/code/2024/10/31/01/\n\n\n");
}
}
}
// 一个辅助方法，用来从 URI 中提取查询参数
private static Map extractQueryParams(URI uri) {
Map queryPairs = new LinkedHashMap<>();
String query = uri.getQuery();
if (query != null) {
String[] pairs = query.split("&");
for (String pair : pairs) {
int idx = pair.indexOf("=");
queryPairs.put(pair.substring(0, idx), pair.substring(idx + 1));
}
}
return queryPairs;
}
}