diff options
| -rw-r--r-- | phpBB/phpbb/template/context.php | 69 | ||||
| -rw-r--r-- | tests/template/template_test.php | 131 | 
2 files changed, 169 insertions, 31 deletions
diff --git a/phpBB/phpbb/template/context.php b/phpBB/phpbb/template/context.php index c7df29d62c..5d04a09865 100644 --- a/phpBB/phpbb/template/context.php +++ b/phpBB/phpbb/template/context.php @@ -376,45 +376,49 @@ class context  	public function alter_block_array($blockname, array $vararray, $key = false, $mode = 'insert')  	{  		$this->num_rows_is_set = false; -		if (strpos($blockname, '.') !== false) -		{ -			// Nested block. -			$blocks = explode('.', $blockname); -			$blockcount = sizeof($blocks) - 1; -			$block = &$this->tpldata; -			for ($i = 0; $i < $blockcount; $i++) +		// For nested block, $blockcount > 0, for top-level block, $blockcount == 0 +		$blocks = explode('.', $blockname); +		$blockcount = sizeof($blocks) - 1; + +		$block = &$this->tpldata; +		for ($i = 0; $i < $blockcount; $i++) +		{ +			if (($pos = strpos($blocks[$i], '[')) !== false)  			{ -				if (($pos = strpos($blocks[$i], '[')) !== false) +				$name = substr($blocks[$i], 0, $pos); + +				if (strpos($blocks[$i], '[]') === $pos)  				{ -					$name = substr($blocks[$i], 0, $pos); - -					if (strpos($blocks[$i], '[]') === $pos) -					{ -						$index = sizeof($block[$name]) - 1; -					} -					else -					{ -						$index = min((int) substr($blocks[$i], $pos + 1, -1), sizeof($block[$name]) - 1); -					} +					$index = sizeof($block[$name]) - 1;  				}  				else  				{ -					$name = $blocks[$i]; -					$index = sizeof($block[$name]) - 1; +					$index = min((int) substr($blocks[$i], $pos + 1, -1), sizeof($block[$name]) - 1);  				} -				$block = &$block[$name]; -				$block = &$block[$index];  			} - -			$block = &$block[$blocks[$i]]; // Traverse the last block +			else +			{ +				$name = $blocks[$i]; +				$index = sizeof($block[$name]) - 1; +			} +			$block = &$block[$name]; +			$block = &$block[$index];  		} -		else +		$name = $blocks[$i]; + +		// If last block does not exist and we are inserting, and not searching for key, we create it empty; otherwise, nothing to do +		if (!isset($block[$name]))  		{ -			// Top-level block. -			$block = &$this->tpldata[$blockname]; +			if ($mode != 'insert' || is_array($key)) +			{ +				return false; +			} +			$block[$name] = array();  		} +		$block = &$block[$name]; // Now we can traverse the last block +  		// Change key to zero (change first position) if false and to last position if true  		if ($key === false || $key === true)  		{ @@ -454,14 +458,15 @@ class context  				unset($block[($key - 1)]['S_LAST_ROW']);  				$vararray['S_LAST_ROW'] = true;  			} -			else if ($key === 0) +			if ($key <= 0)  			{ +				$key = 0;  				unset($block[0]['S_FIRST_ROW']);  				$vararray['S_FIRST_ROW'] = true;  			}  			// Assign S_BLOCK_NAME -			$vararray['S_BLOCK_NAME'] = $blockname; +			$vararray['S_BLOCK_NAME'] = $name;  			// Re-position template blocks  			for ($i = sizeof($block); $i > $key; $i--) @@ -481,6 +486,12 @@ class context  		// Which block to change?  		if ($mode == 'change')  		{ +			// If key is out of bounds, do not change anything +			if ($key > sizeof($block) || $key < 0) +			{ +				return false; +			} +  			if ($key == sizeof($block))  			{  				$key--; diff --git a/tests/template/template_test.php b/tests/template/template_test.php index 997da80246..69546cc227 100644 --- a/tests/template/template_test.php +++ b/tests/template/template_test.php @@ -528,22 +528,139 @@ EOT  			),  			array(  				'outer', +				array('VARIABLE' => 'changed'), +				0, +				'change', +				<<<EOT +outer - 0 - changed +middle - 0 +middle - 1 +outer - 1 +middle - 0 +middle - 1 +outer - 2 +middle - 0 +middle - 1 +EOT +, +				'Test changing at 0 on top level block', +			), +			array( +				'outer', +				array('VARIABLE' => 'changed'), +				array('S_ROW_NUM' => 2), +				'change', +				<<<EOT +outer - 0 +middle - 0 +middle - 1 +outer - 1 +middle - 0 +middle - 1 +outer - 2 - changed +middle - 0 +middle - 1 +EOT +, +				'Test changing at KEY on top level block', +			), +			array( +				'outer.middle', +				array('VARIABLE' => 'before'), +				false, +				'insert', +				<<<EOT +outer - 0 +middle - 0 +middle - 1 +outer - 1 +middle - 0 +middle - 1 +outer - 2 +middle - 0 - before +middle - 1 +middle - 2 +EOT +, +				'Test inserting before on middle level block', +			), +			array( +				'outer.middle', +				array('VARIABLE' => 'after'), +				true, +				'insert', +				<<<EOT +outer - 0 +middle - 0 +middle - 1 +outer - 1 +middle - 0 +middle - 1 +outer - 2 +middle - 0 +middle - 1 +middle - 2 - after +EOT +, +				'Test inserting after on middle level block', +			), +			array( +				'outer[1].middle',  				array('VARIABLE' => 'pos #1'), +				1, +				'insert', +				<<<EOT +outer - 0 +middle - 0 +middle - 1 +outer - 1 +middle - 0 +middle - 1 - pos #1 +middle - 2 +outer - 2 +middle - 0 +middle - 1 +EOT +, +				'Test inserting at 1 on middle level block', +			), +			array( +				'outer[].middle', +				array('VARIABLE' => 'changed'),  				0,  				'change',  				<<<EOT -outer - 0 - pos #1 +outer - 0  middle - 0  middle - 1  outer - 1  middle - 0  middle - 1  outer - 2 +middle - 0 - changed +middle - 1 +EOT +, +				'Test changing at beginning of last top level block', +			), +			array( +				'outer.middle', +				array('VARIABLE' => 'changed'), +				array('S_ROW_NUM' => 1), +				'change', +				<<<EOT +outer - 0 +middle - 0 +middle - 1 +outer - 1  middle - 0  middle - 1 +outer - 2 +middle - 0 +middle - 1 - changed  EOT  , -				'Test inserting at 1 on top level block', +				'Test changing at beginning of last top level block',  			),  		);  	} @@ -601,6 +718,16 @@ EOT  		$expect = 'outer - 0[outer|4]outer - 1[outer|4]middle - 0[middle|1]outer - 2 - test[outer|4]middle - 0[middle|2]middle - 1[middle|2]outer - 3[outer|4]middle - 0[middle|3]middle - 1[middle|3]middle - 2[middle|3]';  		$this->assertEquals($expect, str_replace(array("\n", "\r", "\t"), '', $this->display('test')), 'Ensuring S_NUM_ROWS is correct after modification'); + +		$this->template->alter_block_array('outer.middle', array()); + +		$expect = 'outer - 0[outer|4]outer - 1[outer|4]middle - 0[middle|1]outer - 2 - test[outer|4]middle - 0[middle|2]middle - 1[middle|2]outer - 3[outer|4]middle - 0[middle|4]middle - 1[middle|4]middle - 2[middle|4]middle - 3[middle|4]'; +		$this->assertEquals($expect, str_replace(array("\n", "\r", "\t"), '', $this->display('test')), 'Ensuring S_NUM_ROWS is correct after insertion at middle level'); + +		$this->template->alter_block_array('outer.middle', array('VARIABLE' => 'test'), 2, 'change'); + +		$expect = 'outer - 0[outer|4]outer - 1[outer|4]middle - 0[middle|1]outer - 2 - test[outer|4]middle - 0[middle|2]middle - 1[middle|2]outer - 3[outer|4]middle - 0[middle|4]middle - 1[middle|4]middle - 2 - test[middle|4]middle - 3[middle|4]'; +		$this->assertEquals($expect, str_replace(array("\n", "\r", "\t"), '', $this->display('test')), 'Ensuring S_NUM_ROWS is correct after modification at middle level');  	}  	public function test_find_key_index()  | 
