Version 4.5.0

applications/nexus/sources/CommissionRule/Iterator.php

@@ -12,7 +12,7 @@
 namespace IPS\nexus\CommissionRule;
 
 /* To prevent PHP errors (extending class does not exist) revealing path */
-if ( !defined( '\IPS\SUITE_UNIQUE_KEY' ) )
+if ( !\defined( '\IPS\SUITE_UNIQUE_KEY' ) )
 {
 	header( ( isset( $_SERVER['SERVER_PROTOCOL'] ) ? $_SERVER['SERVER_PROTOCOL'] : 'HTTP/1.0' ) . ' 403 Forbidden' );
 	exit;
@@ -22,7 +22,7 @@ if ( !defined( '\IPS\SUITE_UNIQUE_KEY' ) )
  * Commission Rule Filter Iterator
  * @note	When we require PHP 5.4+ this can just be replaced with a CallbackFilterIterator
  */
-class _Iterator extends \FilterIterator
+class _Iterator extends \FilterIterator implements \Countable
 {
 	/**
 	 * @brief	Member
@@ -39,6 +39,11 @@ class _Iterator extends \FilterIterator
 	 */
 	protected $valueOfPurchases = NULL;
 	
+	/**
+	 * @brief	Number of referral rules
+	 */
+	protected $numberOfRules= NULL;
+	
 	/**
 	 * Constructor
 	 *
@@ -146,4 +151,18 @@ class _Iterator extends \FilterIterator
 		}
 		return $this->valueOfPurchases;
 	}
+	
+	/**
+	 * Countable
+	 *
+	 * @return	int
+	 */
+	public function count(): int
+	{
+		if ( $this->numberOfRules === NULL )
+		{
+			$this->numberOfRules = (int) \IPS\Db::i()->select( 'COUNT(*)', 'nexus_referral_rules' )->first();
+		}
+		return $this->numberOfRules;
+	}
 }