Browse Source

Method "yii\helpers\base\FileHelper::findFiles()" has been added.

tags/2.0.0-beta
Klimov Paul 12 years ago
parent
commit
dfda6663f6
  1. 97
      framework/yii/helpers/base/FileHelper.php
  2. 29
      tests/unit/framework/helpers/FileHelperTest.php

97
framework/yii/helpers/base/FileHelper.php

@ -191,4 +191,101 @@ class FileHelper
rmdir($directory);
}
}
/**
* 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:
* <ul>
* <li>fileTypes: array, list of file name suffix (without dot). Only files with these suffixes will be returned.</li>
* <li>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
* '.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.
* </li>
* <li>level: integer, recursion depth, default=-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.
* </li>
* </ul>
* @return array files found under the directory. The file list is sorted.
*/
public static function findFiles($dir, array $options = array())
{
$fileTypes = array();
$exclude = array();
$level = -1;
extract($options);
$list = static::findFilesRecursive($dir, '', $fileTypes, $exclude, $level);
sort($list);
return $list;
}
/**
* Returns the files found under the specified directory and subdirectories.
* 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 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)
{
$list = array();
$handle = opendir($dir);
while (($file = readdir($handle)) !== false) {
if ($file === '.' || $file === '..') {
continue;
}
$path = $dir . DIRECTORY_SEPARATOR . $file;
$isFile = is_file($path);
if (static::validatePath($base, $file, $isFile, $fileTypes, $exclude)) {
if ($isFile) {
$list[] = $path;
} elseif ($level) {
$list = array_merge($list, static::findFilesRecursive($path, $base . DIRECTORY_SEPARATOR . $file, $fileTypes, $exclude, $level-1));
}
}
}
closedir($handle);
return $list;
}
/**
* 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 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.
* 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)
{
foreach ($exclude as $e) {
if ($file === $e || strpos($base . DIRECTORY_SEPARATOR . $file, $e) === 0) {
return false;
}
}
if (!$isFile || empty($fileTypes)) {
return true;
}
if (($type = pathinfo($file, PATHINFO_EXTENSION)) !== '') {
return in_array($type, $fileTypes);
} else {
return false;
}
}
}

29
tests/unit/framework/helpers/FileHelperTest.php

@ -95,4 +95,33 @@ class FileHelperTest extends TestCase
$this->assertFalse(file_exists($dirName), 'Unable to remove directory!');
}
public function testFindFiles()
{
$basePath = $this->testFilePath;
$expectedFiles = array();
$dirName = $basePath . DIRECTORY_SEPARATOR . 'test_dir_for_remove';
mkdir($dirName, 0777, true);
$files = array(
'file1.txt' => 'file 1 content',
'file2.txt' => 'file 2 content',
);
foreach ($files as $name => $content) {
$fileName = $dirName . DIRECTORY_SEPARATOR . $name;
file_put_contents($fileName, $content);
$expectedFiles[] = $fileName;
}
$subDirName = $dirName . DIRECTORY_SEPARATOR . 'test_sub_dir';
mkdir($subDirName, 0777, true);
foreach ($files as $name => $content) {
$fileName = $subDirName . DIRECTORY_SEPARATOR . $name;
file_put_contents($fileName, $content);
$expectedFiles[] = $fileName;
}
$foundFiles = FileHelper::findFiles($dirName);
sort($expectedFiles);
$this->assertEquals($expectedFiles, $foundFiles);
}
}
Loading…
Cancel
Save