扩展支持更多的任务生命周期钩子

monkeyWie · monkeyWie · commit 1d9c98cde495 · 2018-11-09T13:36:30.000+08:00
diff --git a/main/src/main/java/org/pdown/gui/extension/HookScript.java b/main/src/main/java/org/pdown/gui/extension/HookScript.java
@@ -5,7 +5,12 @@
 public class HookScript {
 
   public static final String EVENT_RESOLVE = "resolve";
+  public static final String EVENT_START = "start";
+  public static final String EVENT_RESUME = "resume";
+  public static final String EVENT_PAUSE = "pause";
   public static final String EVENT_ERROR = "error";
+  public static final String EVENT_DONE = "done";
+  public static final String EVENT_DELETE = "delete";
 
   private Event[] events;
   private String script;
diff --git a/main/src/main/java/org/pdown/gui/extension/util/ExtensionUtil.java b/main/src/main/java/org/pdown/gui/extension/util/ExtensionUtil.java
@@ -13,23 +13,39 @@
 import java.io.InputStreamReader;
 import java.io.OutputStream;
 import java.nio.file.Files;
+import java.nio.file.Path;
 import java.nio.file.Paths;
 import java.nio.file.StandardCopyOption;
+import java.util.concurrent.CountDownLatch;
+import java.util.function.Function;
 import java.util.stream.Collectors;
+import javax.script.Invocable;
+import javax.script.ScriptContext;
 import javax.script.ScriptEngine;
 import javax.script.ScriptException;
+import javax.script.SimpleScriptContext;
 import org.pdown.core.util.FileUtil;
 import org.pdown.gui.DownApplication;
 import org.pdown.gui.content.PDownConfigContent;
 import org.pdown.gui.extension.ExtensionContent;
 import org.pdown.gui.extension.ExtensionInfo;
+import org.pdown.gui.extension.HookScript;
+import org.pdown.gui.extension.HookScript.Event;
 import org.pdown.gui.extension.Meta;
 import org.pdown.gui.extension.jsruntime.JavascriptEngine;
+import org.pdown.gui.http.controller.NativeController;
+import org.pdown.gui.http.util.HttpHandlerUtil;
 import org.pdown.gui.util.AppUtil;
 import org.pdown.gui.util.ConfigUtil;
+import org.pdown.rest.form.TaskForm;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+import org.springframework.util.StringUtils;
 
 public class ExtensionUtil {
 
+  private static final Logger LOGGER = LoggerFactory.getLogger(ExtensionUtil.class);
+
   /**
    * 安装扩展
    */
@@ -52,7 +68,9 @@ public static void update(String server, String path, String files) throws Excep
       //备份扩展配置
       String configPath = extDir + File.separator + Meta.CONFIG_FILE;
       if (FileUtil.exists(configPath)) {
-        Files.copy(Paths.get(configPath), Paths.get(extTmpPath + File.separator + Meta.CONFIG_FILE), StandardCopyOption.REPLACE_EXISTING);
+        Path bakConfigPath = Paths.get(extTmpPath + File.separator + Meta.CONFIG_FILE);
+        FileUtil.createFileSmart(bakConfigPath.toFile().getAbsolutePath());
+        Files.copy(Paths.get(configPath), bakConfigPath, StandardCopyOption.REPLACE_EXISTING);
       }
       String configBakPath = extDir + File.separator + Meta.CONFIG_FILE + ".bak";
       if (FileUtil.exists(configBakPath)) {
@@ -158,4 +176,69 @@ public static ScriptEngine buildExtensionRuntimeEngine(ExtensionInfo extensionIn
     engine.eval(new FileReader(Paths.get(extensionInfo.getMeta().getFullPath(), extensionInfo.getHookScript().getScript()).toFile()));
     return engine;
   }
+
+  /**
+   * 运行一个js方法
+   */
+  public static Object invoke(ExtensionInfo extensionInfo, Event event, Object param, boolean async) throws NoSuchMethodException, ScriptException, FileNotFoundException, InterruptedException {
+    //初始化js引擎
+    ScriptEngine engine = ExtensionUtil.buildExtensionRuntimeEngine(extensionInfo);
+    Invocable invocable = (Invocable) engine;
+    //执行resolve方法
+    Object result = invocable.invokeFunction(StringUtils.isEmpty(event.getMethod()) ? event.getOn() : event.getMethod(), param);
+    //结果为null或者异步调用直接返回
+    if (result == null || async) {
+      return result;
+    }
+    final Object[] ret = {null};
+    //判断是不是返回Promise对象
+    ScriptContext ctx = new SimpleScriptContext();
+    ctx.setAttribute("result", result, ScriptContext.ENGINE_SCOPE);
+    boolean isPromise = (boolean) engine.eval("!!result&&typeof result=='object'&&typeof result.then=='function'", ctx);
+    if (isPromise) {
+      //如果是返回的Promise则等待执行完成
+      CountDownLatch countDownLatch = new CountDownLatch(1);
+      invocable.invokeMethod(result, "then", (Function) o -> {
+        try {
+          ret[0] = o;
+        } catch (Exception e) {
+          LOGGER.error("An exception occurred while resolve()", e);
+        } finally {
+          countDownLatch.countDown();
+        }
+        return null;
+      });
+      invocable.invokeMethod(result, "catch", (Function) o -> {
+        countDownLatch.countDown();
+        return null;
+      });
+      //等待解析完成
+      countDownLatch.await();
+    } else {
+      ret[0] = result;
+    }
+    return ret[0];
+  }
+
+  public static void main(String[] args) {
+    String url = "https://d.pcs.baidu.com/file/ab83d33b3f250a6ff472b8ffa17c3e5f?fid=336129479-250528-831181624029689&dstime=1541581115&rt=sh&sign=FDtAERVY-DCb740ccc5511e5e8fedcff06b081203-aILjqoJW3OEzB4%2Bu7Wnxz63PUho%3D&expires=8h&chkv=1&chkbd=0&chkpc=et&dp-logid=7206180716394015240&dp-callid=0&shareid=3786359813&r=663179228";
+    String[] urlArray = url.split("\\?");
+    StringBuilder params = new StringBuilder(urlArray[1]);
+    String path = urlArray[0].substring(urlArray[0].lastIndexOf("/") + 1);
+    params.append("&path=" + path)
+        .append("&check_blue=1")
+        .append("&clienttype=8")
+        .append("&devuid=BDIMXV2-O_9A2DB23216984690875184DCA864434E-C_0-D_Z4Y7YWL9-M_408D5C4224FB-V_D8F90423")
+        .append("&dtype=1")
+        .append("&eck=1")
+        .append("&ehps=1")
+        .append("&err_ver=1")
+        .append("&es=1")
+        .append("&esl=1")
+        .append("&method=locatedownload")
+        .append("&ver=4")
+        .append("&version=2.1.13.11")
+        .append("&version_app=6.4.0.6");
+    System.out.println(params.toString());
+  }
 }
diff --git a/main/src/main/java/org/pdown/gui/http/controller/NativeController.java b/main/src/main/java/org/pdown/gui/http/controller/NativeController.java
@@ -445,46 +445,14 @@ public FullHttpResponse onResolve(Channel channel, FullHttpRequest request) thro
           Event event = extensionInfo.getHookScript().hasEvent(HookScript.EVENT_RESOLVE, taskRequest.getUrl());
           if (event != null) {
             try {
-              //初始化js引擎
-              ScriptEngine engine = ExtensionUtil.buildExtensionRuntimeEngine(extensionInfo);
-              Invocable invocable = (Invocable) engine;
               //执行resolve方法
-              Object result = invocable.invokeFunction(StringUtils.isEmpty(event.getMethod()) ? HookScript.EVENT_RESOLVE : event.getMethod(), taskRequest);
+              Object result = ExtensionUtil.invoke(extensionInfo, event, taskRequest, false);
               if (result != null) {
-                final TaskForm[] taskForm = {null};
-                //判断是不是返回Promise对象
-                ScriptContext ctx = new SimpleScriptContext();
-                ctx.setAttribute("result", result, ScriptContext.ENGINE_SCOPE);
-                boolean isPromise = (boolean) engine.eval("!!result&&typeof result=='object'&&typeof result.then=='function'", ctx);
                 ObjectMapper objectMapper = new ObjectMapper();
-                if (isPromise) {
-                  //如果是返回的Promise则等待执行完成
-                  CountDownLatch countDownLatch = new CountDownLatch(1);
-                  invocable.invokeMethod(result, "then", (Function) o -> {
-                    try {
-                      String temp = objectMapper.writeValueAsString(o);
-                      taskForm[0] = objectMapper.readValue(temp, TaskForm.class);
-                    } catch (Exception e) {
-                      LOGGER.error("An exception occurred while resolve()", e);
-                    } finally {
-                      countDownLatch.countDown();
-                    }
-                    return null;
-                  });
-                  invocable.invokeMethod(result, "catch", (Function) o -> {
-                    countDownLatch.countDown();
-                    return null;
-                  });
-                  //等待解析完成
-                  countDownLatch.await();
-                } else {
-                  String temp = objectMapper.writeValueAsString(result);
-                  taskForm[0] = objectMapper.readValue(temp, TaskForm.class);
-                }
+                String temp = objectMapper.writeValueAsString(result);
+                TaskForm taskForm = objectMapper.readValue(temp, TaskForm.class);
                 //有一个扩展解析成功的话直接返回
-                if (taskForm[0] != null) {
-                  return HttpHandlerUtil.buildJson(taskForm[0], Include.NON_DEFAULT);
-                }
+                return HttpHandlerUtil.buildJson(taskForm, Include.NON_DEFAULT);
               }
             } catch (Exception e) {
               LOGGER.error("An exception occurred while resolve()", e);
diff --git a/main/src/main/java/org/pdown/gui/rest/HttpDownAppCallback.java b/main/src/main/java/org/pdown/gui/rest/HttpDownAppCallback.java
@@ -1,22 +1,25 @@
 package org.pdown.gui.rest;
 
+import com.fasterxml.jackson.databind.ObjectMapper;
 import java.util.HashMap;
 import java.util.List;
 import java.util.Map;
-import javax.script.Invocable;
-import javax.script.ScriptEngine;
 import org.pdown.core.boot.HttpDownBootstrap;
+import org.pdown.core.entity.HttpResponseInfo;
 import org.pdown.core.util.HttpDownUtil;
 import org.pdown.gui.extension.ExtensionContent;
 import org.pdown.gui.extension.ExtensionInfo;
 import org.pdown.gui.extension.HookScript;
 import org.pdown.gui.extension.HookScript.Event;
 import org.pdown.gui.extension.util.ExtensionUtil;
+import org.pdown.rest.content.HttpDownContent;
 import org.pdown.rest.controller.HttpDownRestCallback;
 import org.pdown.rest.entity.DownInfo;
 import org.pdown.rest.form.HttpRequestForm;
+import org.pdown.rest.form.TaskForm;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
+import org.springframework.beans.BeanUtils;
 import org.springframework.util.StringUtils;
 
 public class HttpDownAppCallback extends HttpDownRestCallback {
@@ -26,56 +29,96 @@ public class HttpDownAppCallback extends HttpDownRestCallback {
   @Override
   public void onStart(HttpDownBootstrap httpDownBootstrap) {
     super.onStart(httpDownBootstrap);
+    commonHook(httpDownBootstrap, HookScript.EVENT_START, false);
   }
 
   @Override
   public void onResume(HttpDownBootstrap httpDownBootstrap) {
     super.onResume(httpDownBootstrap);
+    commonHook(httpDownBootstrap, HookScript.EVENT_RESUME, false);
+  }
+
+  @Override
+  public void onPause(HttpDownBootstrap httpDownBootstrap) {
+    super.onPause(httpDownBootstrap);
+    commonHook(httpDownBootstrap, HookScript.EVENT_PAUSE, false);
   }
 
   @Override
   public void onError(HttpDownBootstrap httpDownBootstrap) {
     super.onError(httpDownBootstrap);
+    commonHook(httpDownBootstrap, HookScript.EVENT_ERROR, true);
+  }
+
+  @Override
+  public void onDone(HttpDownBootstrap httpDownBootstrap) {
+    super.onDone(httpDownBootstrap);
+    commonHook(httpDownBootstrap, HookScript.EVENT_DONE, true);
+  }
+
+  private void commonHook(HttpDownBootstrap httpDownBootstrap, String event, boolean async) {
     DownInfo downInfo = findDownInfo(httpDownBootstrap);
-    if (downInfo != null) {
+    Map<String, Object> taskInfo = buildTaskInfo(downInfo);
+    if (taskInfo != null) {
       //遍历扩展模块是否有对应的处理
       List<ExtensionInfo> extensionInfos = ExtensionContent.get();
-      Map<String, Object> taskForm = new HashMap<>();
-      taskForm.put("id", downInfo.getId());
-      taskForm.put("data", downInfo.getData());
-      taskForm.put("request", HttpRequestForm.parse(downInfo.getBootstrap().getRequest()));
-      taskForm.put("response", downInfo.getBootstrap().getResponse());
       for (ExtensionInfo extensionInfo : extensionInfos) {
         if (extensionInfo.getMeta().isEnabled()) {
           if (extensionInfo.getHookScript() != null
               && !StringUtils.isEmpty(extensionInfo.getHookScript().getScript())) {
-            Event event = extensionInfo.getHookScript().hasEvent(HookScript.EVENT_ERROR, HttpDownUtil.getUrl(httpDownBootstrap.getRequest()));
-            if (event != null) {
+            Event e = extensionInfo.getHookScript().hasEvent(event, HttpDownUtil.getUrl(httpDownBootstrap.getRequest()));
+            if (e != null) {
               try {
-                //初始化js引擎
-                ScriptEngine engine = ExtensionUtil.buildExtensionRuntimeEngine(extensionInfo);
-                Invocable invocable = (Invocable) engine;
-                //执行error方法
-                invocable.invokeFunction(StringUtils.isEmpty(event.getMethod()) ? HookScript.EVENT_ERROR : event.getMethod(), taskForm);
-              } catch (Exception e) {
-                LOGGER.error("An exception occurred while error()", e);
+                //执行钩子函数
+                Object result = ExtensionUtil.invoke(extensionInfo, e, taskInfo, async);
+                if (result != null) {
+                  ObjectMapper objectMapper = new ObjectMapper();
+                  String temp = objectMapper.writeValueAsString(result);
+                  TaskForm taskForm = objectMapper.readValue(temp, TaskForm.class);
+                  if (taskForm.getRequest() != null) {
+                    httpDownBootstrap.setRequest(
+                        HttpDownUtil.buildRequest(taskForm.getRequest().getMethod(),
+                            taskForm.getRequest().getUrl(),
+                            taskForm.getRequest().getHeads(),
+                            taskForm.getRequest().getBody())
+                    );
+                  }
+                  if (taskForm.getResponse() != null) {
+                    httpDownBootstrap.setResponse(taskForm.getResponse());
+                  }
+                  if (taskForm.getData() != null) {
+                    downInfo.setData(taskForm.getData());
+                  }
+                  HttpDownContent.getInstance().save();
+                }
+              } catch (Exception ex) {
+                LOGGER.error("An hook exception occurred while " + event + "()", ex);
               }
             }
-
           }
         }
       }
     }
   }
 
-  @Override
-  public void onProgress(HttpDownBootstrap httpDownBootstrap) {
-    super.onProgress(httpDownBootstrap);
+  private Map<String, Object> buildTaskInfo(DownInfo downInfo) {
+    if (downInfo != null) {
+      Map<String, Object> taskForm = new HashMap<>();
+      taskForm.put("id", downInfo.getId());
+      taskForm.put("data", clone(downInfo.getData(), new HashMap<String, Object>()));
+      taskForm.put("request", HttpRequestForm.parse(downInfo.getBootstrap().getRequest()));
+      taskForm.put("response", clone(downInfo.getBootstrap().getResponse(), new HttpResponseInfo()));
+      return taskForm;
+    }
+    return null;
   }
 
-  @Override
-  public void onDone(HttpDownBootstrap httpDownBootstrap) {
-    super.onDone(httpDownBootstrap);
+  private Object clone(Object source, Object target) {
+    if (source != null && target != null) {
+      BeanUtils.copyProperties(source, target);
+      return target;
+    }
+    return null;
   }
 
 }
