diff --git a/framework/yii/helpers/base/FileHelper.php b/framework/yii/helpers/base/FileHelper.php index c5b8250..2f6d559 100644 --- a/framework/yii/helpers/base/FileHelper.php +++ b/framework/yii/helpers/base/FileHelper.php @@ -196,6 +196,10 @@ class FileHelper * Returns the files found under the specified directory and subdirectories. * @param string $dir the directory under which the files will be looked for. * @param array $options options for file searching. Valid options are: + * - filter: callback, a PHP callback that is called for each sub-directory or file. + * If the callback returns false, the the sub-directory or file will not be placed to result set. + * The signature of the callback should be: `function ($base, $name, $isFile)`, where `$base` is the name of directory, + * which contains file or sub-directory, `$name` file or sub-directory name, `$isFile` indicates if object is a file or a directory. * - fileTypes: array, list of file name suffix (without dot). Only files with these suffixes will be returned. * - exclude: array, list of directory and file exclusions. Each exclusion can be either a name or a path. * If a file or directory name or path matches the exclusion, it will not be copied. For example, an exclusion of @@ -209,11 +213,9 @@ class FileHelper */ public static function findFiles($dir, array $options = array()) { - $fileTypes = array(); - $exclude = array(); - $level = -1; - extract($options); - $list = static::findFilesRecursive($dir, '', $fileTypes, $exclude, $level); + $level = array_key_exists('level', $options) ? $options['level'] : -1; + $filterOptions = $options; + $list = static::findFilesRecursive($dir, '', $filterOptions, $level); sort($list); return $list; } @@ -223,18 +225,17 @@ class FileHelper * This method is mainly used by [[findFiles]]. * @param string $dir the source directory. * @param string $base the path relative to the original source directory. - * @param array $fileTypes list of file name suffix (without dot). Only files with these suffixes will be returned. - * @param array $exclude list of directory and file exclusions. Each exclusion can be either a name or a path. - * If a file or directory name or path matches the exclusion, it will not be copied. For example, an exclusion of - * '.svn' will exclude all files and directories whose name is '.svn'. And an exclusion of '/a/b' will exclude - * file or directory '$src/a/b'. Note, that '/' should be used as separator regardless of the value of the DIRECTORY_SEPARATOR constant. + * @param array $filterOptions list of filter options. + * - filter: a PHP callback, which results indicates if file will be returned. + * - fileTypes: list of file name suffix (without dot). Only files with these suffixes will be returned. + * - exclude: list of directory and file exclusions. Each exclusion can be either a name or a path. * @param integer $level recursion depth. It defaults to -1. * Level -1 means searching for all directories and files under the directory; * Level 0 means searching for only the files DIRECTLY under the directory; * level N means searching for those directories that are within N levels. * @return array files found under the directory. */ - protected static function findFilesRecursive($dir, $base, $fileTypes, $exclude, $level) + protected static function findFilesRecursive($dir, $base, array $filterOptions, $level) { $list = array(); $handle = opendir($dir); @@ -244,11 +245,11 @@ class FileHelper } $path = $dir . DIRECTORY_SEPARATOR . $file; $isFile = is_file($path); - if (static::validatePath($base, $file, $isFile, $fileTypes, $exclude)) { + if (static::validatePath($base, $file, $isFile, $filterOptions)) { if ($isFile) { $list[] = $path; } elseif ($level) { - $list = array_merge($list, static::findFilesRecursive($path, $base . DIRECTORY_SEPARATOR . $file, $fileTypes, $exclude, $level-1)); + $list = array_merge($list, static::findFilesRecursive($path, $base . DIRECTORY_SEPARATOR . $file, $filterOptions, $level-1)); } } } @@ -259,30 +260,40 @@ class FileHelper /** * Validates a file or directory, checking if it match given conditions. * @param string $base the path relative to the original source directory - * @param string $file the file or directory name + * @param string $name the file or directory name * @param boolean $isFile whether this is a file - * @param array $fileTypes list of valid file name suffixes (without dot). - * @param array $exclude list of directory and file exclusions. Each exclusion can be either a name or a path. + * @param array $filterOptions list of filter options. + * - filter: a PHP callback, which results indicates if file will be returned. + * - fileTypes: list of file name suffix (without dot). Only files with these suffixes will be returned. + * - exclude: list of directory and file exclusions. Each exclusion can be either a name or a path. * If a file or directory name or path matches the exclusion, false will be returned. For example, an exclusion of * '.svn' will return false for all files and directories whose name is '.svn'. And an exclusion of '/a/b' will return false for * file or directory '$src/a/b'. Note, that '/' should be used as separator regardless of the value of the DIRECTORY_SEPARATOR constant. * @return boolean whether the file or directory is valid */ - protected static function validatePath($base, $file, $isFile, $fileTypes, $exclude) + protected static function validatePath($base, $name, $isFile, array $filterOptions) { - foreach ($exclude as $e) { - if ($file === $e || strpos($base . DIRECTORY_SEPARATOR . $file, $e) === 0) { - return false; - } + if (isset($filterOptions['filter']) && !call_user_func($filterOptions['filter'], $base, $name, $isFile)) { + return false; } - if (!$isFile || empty($fileTypes)) { - return true; + if (!empty($filterOptions['exclude'])) { + foreach ($filterOptions['exclude'] as $e) { + if ($name === $e || strpos($base . DIRECTORY_SEPARATOR . $name, $e) === 0) { + return false; + } + } } - if (($type = pathinfo($file, PATHINFO_EXTENSION)) !== '') { - return in_array($type, $fileTypes); - } else { - return false; + if (!empty($filterOptions['fileTypes'])) { + if (!$isFile) { + return true; + } + if (($type = pathinfo($name, PATHINFO_EXTENSION)) !== '') { + return in_array($type, $filterOptions['fileTypes']); + } else { + return false; + } } + return true; } /** diff --git a/tests/unit/framework/helpers/FileHelperTest.php b/tests/unit/framework/helpers/FileHelperTest.php index c74f20c..781812d 100644 --- a/tests/unit/framework/helpers/FileHelperTest.php +++ b/tests/unit/framework/helpers/FileHelperTest.php @@ -221,6 +221,31 @@ class FileHelperTest extends TestCase /** * @depends testFindFiles */ + public function testFindFileFilter() + { + $dirName = 'test_dir'; + $passedFileName = 'passed.txt'; + $this->createFileStructure(array( + $dirName => array( + $passedFileName => 'passed file content', + 'declined.txt' => 'declined file content', + ), + )); + $basePath = $this->testFilePath; + $dirName = $basePath . DIRECTORY_SEPARATOR . $dirName; + + $options = array( + 'filter' => function($base, $name, $isFile) use ($passedFileName) { + return ($passedFileName == $name); + } + ); + $foundFiles = FileHelper::findFiles($dirName, $options); + $this->assertEquals(array($dirName . DIRECTORY_SEPARATOR . $passedFileName), $foundFiles); + } + + /** + * @depends testFindFiles + */ public function testFindFilesExclude() { $dirName = 'test_dir';