From 912d539dd024903c532ca6746d46ab343ec3f988 Mon Sep 17 00:00:00 2001
From: Nathaniel Catchpole <catch@35733.no-reply.drupal.org>
Date: Tue, 23 Feb 2016 18:01:17 +0900
Subject: [PATCH] Issue #2671708 by alexpott, damiankloip: Add a
 RegexDirectoryIterator

---
 .../FileSystem/RegexDirectoryIterator.php     |  47 ++++++++
 .../FileSystem/RegexDirectoryIteratorTest.php | 114 ++++++++++++++++++
 2 files changed, 161 insertions(+)
 create mode 100644 core/lib/Drupal/Component/FileSystem/RegexDirectoryIterator.php
 create mode 100644 core/tests/Drupal/Tests/Component/FileSystem/RegexDirectoryIteratorTest.php

diff --git a/core/lib/Drupal/Component/FileSystem/RegexDirectoryIterator.php b/core/lib/Drupal/Component/FileSystem/RegexDirectoryIterator.php
new file mode 100644
index 00000000000..675dfc84682
--- /dev/null
+++ b/core/lib/Drupal/Component/FileSystem/RegexDirectoryIterator.php
@@ -0,0 +1,47 @@
+<?php
+
+/**
+ * @file
+ * Contains \Drupal\Component\FileSystem\RegexDirectoryIterator.
+ */
+
+namespace Drupal\Component\FileSystem;
+
+/**
+ * Iterates over files whose names match a regular expression in a directory.
+ */
+class RegexDirectoryIterator extends \FilterIterator {
+
+  /**
+   * The regular expression to match.
+   *
+   * @var string
+   */
+  protected $regex;
+
+  /**
+   * RegexDirectoryIterator constructor.
+   *
+   * @param string $path
+   *   The path to scan.
+   * @param string $regex
+   *   The regular expression to match, including delimiters. For example,
+       /\.yml$/ would list only files ending in .yml.
+   */
+  public function __construct($path, $regex) {
+    // Use FilesystemIterator to not iterate over the the . and .. directories.
+    $iterator = new \FilesystemIterator($path);
+    parent::__construct($iterator);
+    $this->regex = $regex;
+  }
+
+  /**
+   * Implements \FilterIterator::accept().
+   */
+  public function accept() {
+    /** @var \SplFileInfo $file_info */
+    $file_info = $this->getInnerIterator()->current();
+    return $file_info->isFile() && preg_match($this->regex, $file_info->getFilename());
+  }
+
+}
diff --git a/core/tests/Drupal/Tests/Component/FileSystem/RegexDirectoryIteratorTest.php b/core/tests/Drupal/Tests/Component/FileSystem/RegexDirectoryIteratorTest.php
new file mode 100644
index 00000000000..b4b75c86a05
--- /dev/null
+++ b/core/tests/Drupal/Tests/Component/FileSystem/RegexDirectoryIteratorTest.php
@@ -0,0 +1,114 @@
+<?php
+
+/**
+ * @file
+ * Contains \Drupal\Tests\Component\FileSystem\RegexDirectoryIteratorTest.
+ */
+
+namespace Drupal\Tests\Component\FileSystem;
+
+use Drupal\Component\FileSystem\RegexDirectoryIterator;
+use Drupal\Tests\UnitTestCase;
+use org\bovigo\vfs\vfsStream;
+
+/**
+ * @coversDefaultClass \Drupal\Component\FileSystem\RegexDirectoryIterator
+ * @group FileSystem
+ */
+class RegexDirectoryIteratorTest extends UnitTestCase {
+
+  /**
+   * @covers ::accept
+   * @dataProvider providerTestRegexDirectoryIterator
+   */
+  public function testRegexDirectoryIterator(array $directory, $regex, array $expected) {
+    vfsStream::setup('root', NULL, $directory);
+    $iterator = new RegexDirectoryIterator(vfsStream::url('root'), $regex);
+
+    // Create an array of filenames to assert against.
+    $file_list = array_map(function(\SplFileInfo $file) {
+      return $file->getFilename();
+    }, array_values(iterator_to_array($iterator)));
+
+    $this->assertSame($expected, $file_list);
+  }
+
+  /**
+   * Provider for self::testRegexDirectoryIterator().
+   */
+  public function providerTestRegexDirectoryIterator() {
+    return [
+      [
+        [
+          '1.yml' => '',
+        ],
+        '/\.yml$/',
+        [
+          '1.yml',
+        ],
+      ],
+      [
+        [
+          '1.yml' => '',
+          '2.yml' => '',
+          '3.txt' => '',
+        ],
+        '/\.yml$/',
+        [
+          '1.yml',
+          '2.yml',
+        ],
+      ],
+      [
+        [
+          '1.yml' => '',
+          '2.yml' => '',
+          '3.txt' => '',
+        ],
+        '/\.txt/',
+        [
+          '3.txt',
+        ],
+      ],
+      [
+        [
+          '1.yml' => '',
+          // Ensure we don't recurse in directories even if that match the
+          // regex.
+          '2.yml' => [
+            '3.yml' => '',
+            '4.yml' => '',
+          ],
+          '3.txt' => '',
+        ],
+        '/\.yml$/',
+        [
+          '1.yml',
+        ],
+      ],
+      [
+        [
+          '1.yml' => '',
+          '2.yml' => '',
+          '3.txt' => '',
+        ],
+        '/^\d/',
+        [
+          '1.yml',
+          '2.yml',
+          '3.txt'
+        ],
+      ],
+      [
+        [
+          '1.yml' => '',
+          '2.yml' => '',
+          '3.txt' => '',
+        ],
+        '/^\D/',
+        [],
+      ],
+    ];
+  }
+
+}