From 390fb733b3ab3befaa76d3be7ed47f292f3ab64b Mon Sep 17 00:00:00 2001
From: arily <lbsaaaaa@gmail.com>
Date: Mon, 22 Jul 2019 19:38:32 +0900
Subject: [PATCH] Iterator

---
 src/Client.php                     |  12 +++
 src/Iterators/ResponseIterator.php | 146 +++++++++++++++++++++++++++++++++++++
 2 files changed, 158 insertions(+)
 create mode 100644 src/Iterators/ResponseIterator.php

diff --git a/src/Client.php b/src/Client.php
index ecb9d31..2125fce 100644
--- a/src/Client.php
+++ b/src/Client.php
@@ -176,6 +176,17 @@ class Client implements Interfaces\ClientInterface
         return $parse ? $this->rosario($response) : $response;
     }
 
+
+	/**
+	 * Read using Iterators to improve performance on large dataset
+	 *
+	 * @return Iterators\ResponseIterator
+	 */
+	public function readAsIterator()
+    {
+    	return new Iterators\ResponseIterator($this->read(false));
+    }
+
     /**
      * This method was created by memory save reasons, it convert response
      * from RouterOS to readable array in safe way.
@@ -220,6 +231,7 @@ class Client implements Interfaces\ClientInterface
         return $result;
     }
 
+
     /**
      * Parse response from Router OS
      *
diff --git a/src/Iterators/ResponseIterator.php b/src/Iterators/ResponseIterator.php
new file mode 100644
index 0000000..3fba772
--- /dev/null
+++ b/src/Iterators/ResponseIterator.php
@@ -0,0 +1,146 @@
+<?php
+/**
+ * Created by PhpStorm.
+ * User: kousakananako
+ * Date: 2019/05/30
+ * Time: 2:07
+ */
+
+namespace RouterOS\Iterators;
+
+
+/**
+ * Class ResponseIterator
+ * @package RouterOS\Iterators
+ */
+class ResponseIterator implements \Iterator, \ArrayAccess, \Countable {
+	public $parsed = [];
+	public $raw = [];
+	public $current;
+	public $length;
+	public function __construct($raw) {
+		$this->current = 0;
+		// This RAW should't be an error
+		$positions = array_keys($raw, '!re');
+		$this->length = count($positions);
+		$count     = count($raw);
+		$result    = [];
+
+		if (isset($positions[1])) {
+
+			foreach ($positions as $key => $position) {
+				// Get length of future block
+				$length = isset($positions[$key + 1])
+					? $positions[$key + 1] - $position + 1
+					: $count - $position;
+
+				// Convert array to simple items
+				$item = array_slice($raw,$position,$length);
+
+				// Save as result
+				$result[] = $item;
+			}
+
+		} else {
+			$result = [$raw];
+		}
+
+		$this->raw = $result;
+	}
+	public function next(){
+		++$this->current;
+	}
+	public function current() {
+		if (isset($this->parsed[$this->current])){
+			return $this->parsed[$this->current];
+		} elseif (isset($this->raw[$this->current])){
+			return $this->parseResponse($this->raw[$this->current])[0];
+		} else {
+			return FALSE;
+		}
+	}
+	public function key() {
+		return $this->current;
+	}
+	public function valid() {
+		return isset($this->raw[$this->current]);
+	}
+	public function count() {
+		return count($this->raw);
+	}
+	public function rewind() {
+		$this->current = 0;
+	}
+	public function offsetSet($offset, $value) {
+		if (is_null($offset)) {
+			$this->parsed[] = $value;
+			throw new \RuntimeException('don\'t append to me It will cause a Bug sometime I PROMISE');
+		} else {
+			$this->parsed[$offset] = $value;
+		}
+	}
+	public function offsetExists($offset) {
+		return isset($this->raw[$offset]) && $this->raw[$offset] !== ['!re'];
+	}
+	public function offsetUnset($offset) {
+		unset($this->parsed[$offset]);
+		unset($this->raw[$offset]);
+	}
+	public function offsetGet($offset) {
+		if (isset($this->parsed[$offset])){
+			return $this->parsed[$offset];
+		} elseif(isset($this->raw[$offset]) && $this->raw[$offset] !== NULL) {
+			$f = $this->parseResponse($this->raw[$offset]);
+			if ($f !==[]){
+				$r = $this->parsed[$offset] = $f[0];
+				return $r;
+			}
+		} else {
+			return FALSE;
+		}
+	}
+	public function flush(){
+		$this->raw = [];
+		$this->parsed = [];
+	}
+	private function parseResponse(array $response): array
+	{
+		$result = [];
+		$i      = -1;
+		$lines  = \count($response);
+		foreach ($response as $key => $value) {
+			switch ($value) {
+				case '!re':
+					$i++;
+					break;
+				case '!fatal':
+					$result = $response;
+					break 2;
+				case '!trap':
+				case '!done':
+					// Check for =ret=, .tag and any other following messages
+					for ($j = $key + 1; $j <= $lines; $j++) {
+						// If we have lines after current one
+						if (isset($response[$j])) {
+							$this->pregResponse($response[$j], $matches);
+							if (isset($matches[1][0], $matches[2][0])) {
+								$result['after'][$matches[1][0]] = $matches[2][0];
+							}
+						}
+					}
+					break 2;
+				default:
+					$this->pregResponse($value, $matches);
+					if (isset($matches[1][0], $matches[2][0])) {
+						$result[$i][$matches[1][0]] = $matches[2][0];
+					}
+					break;
+			}
+		}
+		return $result;
+	}
+	private function pregResponse(string $value, &$matches)
+	{
+		preg_match_all('/^[=|\.](.*)=(.*)/', $value, $matches);
+	}
+}