Custom pagelet ID's

2 files changed, 4 insertions, 4 deletions

diff --git a/include/classes/BigPipe/DemoPagelet.php b/include/classes/BigPipe/DemoPagelet.php
index a6c61e1..83689ba 100755
--- a/include/classes/BigPipe/DemoPagelet.php
+++ b/include/classes/BigPipe/DemoPagelet.php
@@ -3,8 +3,8 @@ namespace BigPipe;
 
 class DemoPagelet extends Pagelet {
 
-	public function __construct($priority = Pagelet::PRIORITY_NORMAL) {
-		parent::__construct($priority);
+	public function __construct($customID = NULL, $priority = Pagelet::PRIORITY_NORMAL) {
+		parent::__construct($customID, $priority);
 
 		$message = '%s: PhaseDoneJS for phase %s';
 
diff --git a/include/classes/BigPipe/Pagelet.php b/include/classes/BigPipe/Pagelet.php
index d1c335f..3b5530c 100755
--- a/include/classes/BigPipe/Pagelet.php
+++ b/include/classes/BigPipe/Pagelet.php
@@ -28,9 +28,9 @@ class Pagelet {
 	const PHASE_LOADJS  = 3; # After all the JS resources have been loaded
 	const PHASE_EXECJS  = 4; # After the static JS code has been executed
 
-	public function __construct($priority = self::PRIORITY_NORMAL) {
+	public function __construct($customID = NULL, $priority = self::PRIORITY_NORMAL) {
 		$this->phaseDoneJS = array_pad([], 5, []);
-		$this->ID = 'P'.++self::$count;
+		$this->ID = is_string($customID) ? $customID : 'P'.++self::$count;
 
 		BigPipe::addPagelet($this, $priority);
 	}