silverstripe / silverstripe-routewhitelist

code/SiteTreeWhitelistExtension.php

Spacing +2 added lines, -2 removed lines

@@ -1,14 +1,14 @@
 <?php
 class SiteTreeWhitelistExtension extends DataExtension {
 
-	public function onAfterWrite(){
+	public function onAfterWrite() {
 		//top level page change should trigger re-generation of whitelist
 		if ($this->owner->getParentType() === 'root') {
 			WhitelistGenerator::generateWhitelist();
 		}
 	}
 	
-	public function onAfterDelete(){
+	public function onAfterDelete() {
 		if ($this->owner->getParentType() === 'root') {
 			WhitelistGenerator::generateWhitelist();
 		}

code/WhitelistGenerator.php

Spacing +22 added lines, -22 removed lines

@@ -1,20 +1,20 @@ 
 <?php
 class WhitelistGenerator extends Object implements Flushable {
 
-	public static function generateWhitelist(){
+	public static function generateWhitelist() {
 		$whitelist = self::generateWhitelistRules();
 		self::syncCacheFilesystem($whitelist);
 	}
 
-	public static function generateWhitelistRules(){
+	public static function generateWhitelistRules() {
 		//get all URL rules
 		$rules = Config::inst()->get('Director', 'rules');
 
 		$allTopLevelRules = array();
 		foreach ($rules as $pattern => $controllerOptions) {
 			//match first portion of the URL, either delimited by slash or colon or end-of-line
-			if (preg_match('/^(.*?)(\/|:|$)/', $pattern, $matches)){
-				if (!empty($matches[1])){
+			if (preg_match('/^(.*?)(\/|:|$)/', $pattern, $matches)) {
+				if (!empty($matches[1])) {
 					array_push($allTopLevelRules, $matches[1]);
 				}
 			}
@@ -27,18 +27,18 @@ 
 			$filteredRules = array_merge($filteredRules, $addToWhitelist);
 		}
 		
-		foreach($allTopLevelRules as $rule) {
+		foreach ($allTopLevelRules as $rule) {
 			if (strpos($rule, '$') !== false) {
 				if ($rule === '$Controller') {
 					//special case for Controllers, add all possible controllers
 					$subControllers = ClassInfo::subclassesFor(new Controller());
 
-					foreach ($subControllers as $controller){
-						array_push($filteredRules, $controller);    //add the controller name as a link
+					foreach ($subControllers as $controller) {
+						array_push($filteredRules, $controller); //add the controller name as a link
 					}
 
 				} elseif ($rule === '$URLSegment') {
-					$topLevelPagesArray = array();  //temporary array to store top-level pages
+					$topLevelPagesArray = array(); //temporary array to store top-level pages
 
 					//special case for SiteTree, add all possible top Level Pages
 					$topLevelPages = SiteTree::get()->filter('ParentID', 0);
@@ -55,7 +55,7 @@ 
 
 					$filteredRules = array_merge($filteredRules, $topLevelPagesArray, $oldTopLevelPagesArray);
 				} else {
-					user_error('Unknown wildcard URL match found: '.$rule, E_WARNING);
+					user_error('Unknown wildcard URL match found: ' . $rule, E_WARNING);
 				}
 			} else {
 				//add the rule to a new list of rules
@@ -90,7 +90,7 @@ 
 
 		//create list of files to create
 		$toCreate = array();
-		foreach ($whitelist as $listItem){
+		foreach ($whitelist as $listItem) {
 			if (!in_array($listItem, $whitelistFolderContents)) {
 				if (!empty($listItem)) {    //don't include empty files, such as the file for /
 					array_push($toCreate, $listItem);
@@ -100,8 +100,8 @@ 
 
 		//create list of files to delete
 		$toDelete = array();
-		foreach ($whitelistFolderContents as $file){
-			if (!in_array($file, array('','..','.','.htaccess'))) {    //exclude things that should stay in the folder
+		foreach ($whitelistFolderContents as $file) {
+			if (!in_array($file, array('', '..', '.', '.htaccess'))) {    //exclude things that should stay in the folder
 				if (!in_array($file, $whitelist)) {
 					array_push($toDelete, $file);
 				}
@@ -127,11 +127,11 @@ 
 	 * @param $currentTopLevelPages
 	 * @return array URLs of past top-level pages
 	 */
-	protected static function find_old_top_level_pages($currentTopLevelPages){
+	protected static function find_old_top_level_pages($currentTopLevelPages) {
 		$oldPageURLs = array();
 
 		$queryClass = 'SQLSelect';
-		if (!class_exists($queryClass) && class_exists('SQLQuery')){
+		if (!class_exists($queryClass) && class_exists('SQLQuery')) {
 			$queryClass = 'SQLQuery';
 		}
 		
@@ -141,13 +141,13 @@ 
 			array(
 				'ParentID = 0',
 				'WasPublished = 1',
-				'URLSegment NOT IN (\''.implode("','",array_filter($currentTopLevelPages)).'\')'
+				'URLSegment NOT IN (\'' . implode("','", array_filter($currentTopLevelPages)) . '\')'
 			)
 		);
 
 		$records = $query->execute();
 		if ($records) {
-			foreach($records as $record) {
+			foreach ($records as $record) {
 				array_push($oldPageURLs, $record['URLSegment']);
 			}
 		}
@@ -155,25 +155,25 @@ 
 		return $oldPageURLs;
 	}
 
-    public static function ensureWhitelistFolderExists(){
+    public static function ensureWhitelistFolderExists() {
 		$dir = BASE_PATH . DIRECTORY_SEPARATOR . Config::inst()->get('WhitelistGenerator', 'dir');
 		if (!file_exists($dir)) {
 			mkdir($dir); //create a new whitelist dir
-			chmod($dir,0777);    //make sure it is readable by the web-server user
+			chmod($dir, 0777); //make sure it is readable by the web-server user
 			//create a htaccess file to ensure that the whitelist cache directory is not web-accessible
-			file_put_contents($dir.DIRECTORY_SEPARATOR.'.htaccess', "Deny from all\n");
+			file_put_contents($dir . DIRECTORY_SEPARATOR . '.htaccess', "Deny from all\n");
 		}
 	}
 
-	public static function clearWhitelist(){
+	public static function clearWhitelist() {
 		$dir = BASE_PATH . DIRECTORY_SEPARATOR . Config::inst()->get('WhitelistGenerator', 'dir');
 		if (file_exists($dir)) {
-			array_map('unlink', glob($dir."/*"));
+			array_map('unlink', glob($dir . "/*"));
 		}
 	}
 
 	public static function flush() {
-		self::ensureWhitelistFolderExists();    //only create folder on flush, not on sitetree changes
+		self::ensureWhitelistFolderExists(); //only create folder on flush, not on sitetree changes
 		self::generateWhitelist();
 	}
 }

tests/WhitelistGeneratorTest.php

Spacing +5 added lines, -5 removed lines

@@ -8,7 +8,7 @@ 
         parent::setUp();
     }
 
-	public function testGenerateWhitelist(){
+	public function testGenerateWhitelist() {
 		$whitelist = WhitelistGenerator::generateWhitelistRules();
 
 		$top1 = $this->objFromFixture('SiteTree', 'top1');
@@ -107,13 +107,13 @@ 
 
 	function testSiteTreeVersionsIncludedInWhitelist() {
 		$top1 = $this->objFromFixture('SiteTree', 'top1');
-		$top1->publish('Stage', 'Live');    //publish the page so it has been live and needs redirecting to
+		$top1->publish('Stage', 'Live'); //publish the page so it has been live and needs redirecting to
 
 		$newSegment = 'new-url-segment';
 		$oldSegment = $top1->URLSegment;
 		$top1->URLSegment = $newSegment;
 		$top1->write();
-		$top1->publish('Stage', 'Live');    //publish again with a new URL
+		$top1->publish('Stage', 'Live'); //publish again with a new URL
 
 
 		$whitelist = WhitelistGenerator::generateWhitelistRules();
@@ -147,7 +147,7 @@ 
 		
 		$whitelist = WhitelistGenerator::generateWhitelistRules();
 
-		foreach($addToList as $add) {
+		foreach ($addToList as $add) {
 			$this->assertContains($add, $whitelist);
 		}
 	}
@@ -161,7 +161,7 @@ 
 
 		$whitelist = WhitelistGenerator::generateWhitelistRules();
 
-		foreach($removeFromList as $remove) {
+		foreach ($removeFromList as $remove) {
 			$this->assertNotContains($remove, $whitelist);
 		}
 	}