Register pre/post scripts for each step of the upgrade

pydio · Nov 29, 2014 · 65aa360 · 65aa360
1 parent 4aca084
commit 65aa360
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/core/src/plugins/action.updater/class.AjaXplorerUpgrader.php b/core/src/plugins/action.updater/class.AjaXplorerUpgrader.php
@@ -36,6 +36,7 @@ class AjaXplorerUpgrader
     private $debugMode = FALSE;
     private $cleanFile = "UPGRADE/CLEAN-FILES";
     private $additionalScript = "UPGRADE/PHP-SCRIPT";
+    private $stepTriggerPrefix = "UPGRADE/PHP-";
     private $releaseNote = "UPGRADE/NOTE";
     private $htmlInstructions = "UPGRADE/NOTE-HTML";
     private $dbUpgrade = "UPGRADE/DB-UPGRADE";
@@ -120,12 +121,15 @@ public function hasNextStep()
     public function execute()
     {
         $stepKeys = array_keys($this->steps);
+        $stepName = $stepKeys[$this->step];
         try {
+            $this->executeStepTrigger($stepName, "pre");
             if (method_exists($this, $stepKeys[$this->step])) {
                 $this->result = call_user_func(array($this, $stepKeys[$this->step]));
             } else {
                 $this->result = "Skipping step, method not found";
             }
+            $this->executeStepTrigger($stepName, "post");
         } catch (Exception $e) {
             $this->error = $e->getMessage();
         }
@@ -351,6 +355,16 @@ public function specificTask()
 
     }
 
+    protected function executeStepTrigger($stepName, $trigger = "pre")
+    {
+        $scriptName = $this->workingFolder."/".$this->stepTriggerPrefix."-".$trigger."-".$stepName.".php";
+        if(!is_file($scriptName)) return "";
+        ob_start();
+        include($scriptName);
+        $output = ob_get_flush();
+        return "Executed specific task for ".$trigger."-".$stepName.": ".$output;
+    }
+
     public function updateVersion()
     {
         // Finally copy VERSION file