add appendHTML, prependHTML, addRegexReplace

AvocadoAmber/classes/htmlprocess/htmlprocess.class.php

@@ -4,6 +4,9 @@ class html_process
   protected $latecss = [];
   protected $css = [];
   protected $js = [];
+  protected $beforeBuffer = "";
+  protected $afterBuffer = "";
+  protected $replacer = [];
 
   private function updateLoginTable()
   {
@@ -188,9 +191,24 @@ class html_process
     }*/
   }
 
+  public function appendHtml($html)
+  {
+    $this->afterBuffer .= $html;
+  }
+
+  public function prependHtml($html)
+  {
+    $this->beforeBuffer .= $html;
+  }
+
+  public function addRegexReplace($pattern, $replace)
+  {
+    $this->replacer[] = [$pattern, $replace];
+  }
+
   public function run()
   {
-    global $config, $g5, $member;
+    global $config;
 
     $this->updateLoginTable();
     $this->cleanOldLoginRecords($config['cf_login_minutes']);
@@ -203,6 +221,13 @@ class html_process
     $buffer = $this->injectStyles($buffer, $stylesheet, $latestylesheet);
     $buffer = $this->injectJavascript($buffer, $javascript);
 
+    foreach($this->replacer as $v) {
+      $buffer = preg_replace($v[0], $v[1], $buffer);
+    }
+
+    $buffer = preg_replace('#(</head>[^<]*<body[^>]*>)#', "$1{$this->beforeBuffer}", $buffer);
+    $buffer = preg_replace('#</(?:\s+)?body>#', "</body>" . $this->afterBuffer, $buffer);
+
     if (class_exists("EventHandler")) {
       EventHandler::triggerEvent("amber.renderhtml_before_print", $buffer);
     }