Added RobotLoader::excludeDirectory() (#9)

Author: vlastavesely

readme.md

@@ -45,7 +45,7 @@ When RobotLoader encounters duplicate class name during indexing, it throws an e
 The `$loader->setAutoRefresh(TRUE or FALSE)` determines whether RobotLoader should reindex the scripts if asked for nonexistent class.
 This feature should be disabled on production server.
 
-If you want RobotLoader to skip some directory, create a file there called `netterobots.txt`:
+If you want RobotLoader to skip some directory, use `$loader->excludeDirectory('temp')` or create a file there called `netterobots.txt`:
 
 ```
 Disallow: /Zend

src/RobotLoader/RobotLoader.php

@@ -13,6 +13,14 @@
 
 /**
  * Nette auto loader is responsible for loading classes and interfaces.
+ *
+ * <code>
+ * $loader = new Nette\Loaders\RobotLoader;
+ * $loader->addDirectory('app');
+ * $loader->excludeDirectory('app/exclude');
+ * $loader->setTempDirectory('temp');
+ * $loader->register();
+ * </code>
  */
 class RobotLoader
 {
@@ -32,6 +40,9 @@ class RobotLoader
 	/** @var array */
 	private $scanPaths = [];
 
+	/** @var array */
+	private $excludeDirs = [];
+
 	/** @var array of class => [file, time] */
 	private $classes = [];
 
@@ -116,6 +127,18 @@ public function addDirectory($path)
 	}
 
 
+	/**
+	 * Excludes path or paths from list.
+	 * @param  string|string[]  absolute path
+	 * @return static
+	 */
+	public function excludeDirectory($path)
+	{
+		$this->excludeDirs = array_merge($this->excludeDirs, (array) $path);
+		return $this;
+	}
+
+
 	/**
 	 * @return array of class => filename
 	 */
@@ -192,7 +215,7 @@ private function createFileIterator($dir)
 
 		$ignoreDirs = is_array($this->ignoreDirs) ? $this->ignoreDirs : preg_split('#[,\s]+#', $this->ignoreDirs);
 		$disallow = [];
-		foreach ($ignoreDirs as $item) {
+		foreach (array_merge($ignoreDirs, $this->excludeDirs) as $item) {
 			if ($item = realpath($item)) {
 				$disallow[str_replace('\\', '/', $item)] = TRUE;
 			}
@@ -415,7 +438,7 @@ private function getCacheFile()
 	 */
 	protected function getCacheKey()
 	{
-		return [$this->ignoreDirs, $this->acceptFiles, $this->scanPaths];
+		return [$this->ignoreDirs, $this->acceptFiles, $this->scanPaths, $this->excludeDirs];
 	}
 
 }

tests/Loaders/RobotLoader.phpt

@@ -19,6 +19,7 @@ $loader->addDirectory(__DIR__ . '/file/interface.php'); // as file
 $loader->addDirectory(__DIR__ . '/file/class.const.php');
 $loader->addDirectory(__DIR__ . '/file/trait.php');
 $loader->addDirectory(__DIR__ . '/files.robots');
+$loader->excludeDirectory(__DIR__ . '/files/exclude');
 $loader->register();
 
 Assert::false(class_exists('ConditionalClass'));   // files/conditional.class.php
@@ -38,3 +39,5 @@ Assert::false(class_exists('Disallowed4'));   // files.robots\subdir\disallowed4
 Assert::false(class_exists('Disallowed5'));   // files.robots\subdir\subdir2\disallowed5\class.php
 Assert::false(class_exists('Disallowed6'));   // files.robots\subdir\subdir2\class.php
 Assert::true(class_exists('Allowed2'));       // files.robots\subdir\subdir2\allowed.php
+
+Assert::false(class_exists('ExcludedClass')); // files/exclude/excluded.php

tests/Loaders/files/exclude/excluded.php

@@ -0,0 +1,4 @@
+<?php
+
+class ExcludedClass
+{}