From 90db5c2047898f37a92799dd6ba498c8570d59c9 Mon Sep 17 00:00:00 2001
From: Alannah Kearney <hi@alannahkearney.com>
Date: Sat, 1 Jun 2019 22:34:32 +1000
Subject: [PATCH] Added InputTypeFilterIterator class

---
 src/Input/InputTypeFilterIterator.php | 39 +++++++++++++++++++++++++++
 1 file changed, 39 insertions(+)
 create mode 100644 src/Input/InputTypeFilterIterator.php

diff --git a/src/Input/InputTypeFilterIterator.php b/src/Input/InputTypeFilterIterator.php
new file mode 100644
index 0000000..e46e231
--- /dev/null
+++ b/src/Input/InputTypeFilterIterator.php
@@ -0,0 +1,39 @@
+<?php
+
+declare(strict_types=1);
+
+namespace pointybeard\Helpers\Cli\Input;
+
+class InputTypeFilterIterator extends \FilterIterator {
+
+    private $types;
+    private $mode;
+
+    public const FILTER_INCLUDE = 0x0001;
+    public const FILTER_EXCLUDE = 0x0002;
+
+    public function __construct(\Iterator $iterator, array $types=[], int $mode=self::FILTER_INCLUDE)
+    {
+        parent::__construct($iterator);
+
+        $this->types = array_map('strtolower', $types);
+        $this->mode = $mode;
+
+    }
+    public function accept()
+    {
+        $input = $this->getInnerIterator()->current();
+
+        switch($this->mode) {
+            case self::FILTER_EXCLUDE:
+                return !in_array($input->getType(), $this->types);
+                break;
+
+            case self::FILTER_INCLUDE:
+            default:
+                return in_array($input->getType(), $this->types);
+                break;
+
+        }
+    }
+}