Add PHP_CodeCoverage_Filter_Iterator.

Author: sebastianbergmann

PHP/CodeCoverage/Filter/Iterator.php

@@ -0,0 +1,152 @@
+<?php
+/**
+ * PHP_CodeCoverage
+ *
+ * Copyright (c) 2009, Sebastian Bergmann <[email protected]>.
+ * All rights reserved.
+ *
+ * Redistribution and use in source and binary forms, with or without
+ * modification, are permitted provided that the following conditions
+ * are met:
+ *
+ *   * Redistributions of source code must retain the above copyright
+ *     notice, this list of conditions and the following disclaimer.
+ *
+ *   * Redistributions in binary form must reproduce the above copyright
+ *     notice, this list of conditions and the following disclaimer in
+ *     the documentation and/or other materials provided with the
+ *     distribution.
+ *
+ *   * Neither the name of Sebastian Bergmann nor the names of his
+ *     contributors may be used to endorse or promote products derived
+ *     from this software without specific prior written permission.
+ *
+ * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
+ * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
+ * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
+ * FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE
+ * COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT,
+ * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING,
+ * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
+ * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
+ * CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
+ * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN
+ * ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
+ * POSSIBILITY OF SUCH DAMAGE.
+ *
+ * @category   PHP
+ * @package    CodeCoverage
+ * @author     Sebastian Bergmann <[email protected]>
+ * @copyright  2009 Sebastian Bergmann <[email protected]>
+ * @license    http://www.opensource.org/licenses/bsd-license.php  BSD License
+ * @link       http://github.com/sebastianbergmann/php-code-coverage
+ * @since      File available since Release 1.0.0
+ */
+
+require_once 'PHPUnit/Util/Filter.php';
+
+PHPUnit_Util_Filter::addFileToFilter(__FILE__, 'PHPUNIT');
+
+/**
+ * 
+ *
+ * @category   PHP
+ * @package    CodeCoverage
+ * @author     Sebastian Bergmann <[email protected]>
+ * @copyright  2009 Sebastian Bergmann <[email protected]>
+ * @license    http://www.opensource.org/licenses/bsd-license.php  BSD License
+ * @version    Release: @package_version@
+ * @link       http://github.com/sebastianbergmann/php-code-coverage
+ * @since      Class available since Release 1.0.0
+ */
+class PHP_CodeCoverage_Filter_Iterator extends FilterIterator
+{
+    /**
+     * @var array
+     */
+    protected $suffixes = array();
+
+    /**
+     * @var string
+     */
+    protected $prefixes = array();
+
+    /**
+     * @param  Iterator $iterator
+     * @param  array    $suffixes
+     * @param  array    $prefixes
+     */
+    public function __construct(Iterator $iterator, $suffixes = array(), $prefixes = array())
+    {
+        if (is_string($suffixes)) {
+            if (!empty($suffixes)) {
+                $suffixes = array($suffixes);
+            } else {
+                $suffixes = array();
+            }
+        }
+
+        if (!is_array($suffixes)) {
+            throw new InvalidArgumentException('Argument #2 of PHP_CodeCoverage_Filter_Iterator:__construct() is no array or string.');
+        }
+
+        $this->suffixes = $suffixes;
+
+        if (is_string($prefixes)) {
+            if (!empty($prefixes)) {
+                $prefixes = array($prefixes);
+            } else {
+                $prefixes = array();
+            }
+        }
+
+        if (!is_array($prefixes)) {
+            throw new InvalidArgumentException('Argument #3 of PHP_CodeCoverage_Filter_Iterator:__construct() is no array or string.');
+        }
+
+        $this->prefixes = $prefixes;
+
+        parent::__construct($iterator);
+    }
+
+    /**
+     * @return boolean
+     */
+    public function accept()
+    {
+        $string = (string)$this->getInnerIterator()->current();
+
+        if (!empty($this->prefixes)) {
+            $matched = FALSE;
+
+            foreach ($this->prefixes as $prefix) {
+                if (strpos($string, $prefix) === 0) {
+                    $matched = TRUE;
+                    break;
+                }
+            }
+
+            if (!$matched) {
+                return FALSE;
+            }
+        }
+
+        if (!empty($this->suffixes)) {
+            $matched = FALSE;
+
+            foreach ($this->suffixes as $suffix) {
+                if (substr($string, -1 * strlen($suffix)) == $suffix) {
+                    $matched = TRUE;
+                    break;
+                }
+            }
+
+            if (!$matched) {
+                return FALSE;
+            }
+        }
+
+        return TRUE;
+    }
+}
+?>