coderfanjie
diff --git a/‎Behavioral/Iterator/Book.php
Lines changed: 10 additions & 4 deletions b/‎Behavioral/Iterator/Book.php
Lines changed: 10 additions & 4 deletions
diff --git a/‎Behavioral/Iterator/BookList.php
Lines changed: 34 additions & 10 deletions b/‎Behavioral/Iterator/BookList.php
Lines changed: 34 additions & 10 deletions
diff --git a/‎Behavioral/Iterator/BookListIterator.php
Lines changed: 0 additions & 86 deletions b/‎Behavioral/Iterator/BookListIterator.php
Lines changed: 0 additions & 86 deletions
diff --git a/‎Behavioral/Iterator/BookListReverseIterator.php
Lines changed: 0 additions & 87 deletions b/‎Behavioral/Iterator/BookListReverseIterator.php
Lines changed: 0 additions & 87 deletions
diff --git a/‎Behavioral/Iterator/README.rst
Lines changed: 2 additions & 15 deletions b/‎Behavioral/Iterator/README.rst
Lines changed: 2 additions & 15 deletions
@@ -4,27 +4,33 @@
 
 class Book
 {
+    /**
+     * @var string
+     */
     private $author;
 
+    /**
+     * @var string
+     */
     private $title;
 
-    public function __construct($title, $author)
+    public function __construct(string $title, string $author)
     {
         $this->author = $author;
         $this->title = $title;
     }
 
-    public function getAuthor()
+    public function getAuthor(): string
     {
         return $this->author;
     }
 
-    public function getTitle()
+    public function getTitle(): string
     {
         return $this->title;
     }
 
-    public function getAuthorAndTitle()
+    public function getAuthorAndTitle(): string
     {
         return $this->getTitle().' by '.$this->getAuthor();
     }
 
@@ -2,18 +2,17 @@
 
 namespace DesignPatterns\Behavioral\Iterator;
 
-class BookList implements \Countable
+class BookList implements \Countable, \Iterator
 {
+    /**
+     * @var Book[]
+     */
     private $books;
 
-    public function getBook($bookNumberToGet)
-    {
-        if (isset($this->books[$bookNumberToGet])) {
-            return $this->books[$bookNumberToGet];
-        }
-
-        return;
-    }
+    /**
+     * @var int
+     */
+    private $currentIndex = 0;
 
     public function addBook(Book $book)
     {
@@ -30,8 +29,33 @@ public function removeBook(Book $bookToRemove)
         }
     }
 
-    public function count()
+    public function count(): int
     {
         return count($this->books);
     }
+
+    public function current(): Book
+    {
+        return $this->books[array_keys($this->books)[$this->currentIndex]];
+    }
+
+    public function key(): int
+    {
+        return array_keys($this->books)[$this->currentIndex];
+    }
+
+    public function next()
+    {
+        $this->currentIndex++;
+    }
+
+    public function rewind()
+    {
+        $this->currentIndex = 0;
+    }
+
+    public function valid(): bool
+    {
+        return isset(array_keys($this->books)[$this->currentIndex]);
+    }
 }
@@ -4,8 +4,7 @@
 Purpose
 -------
 
-To make an object iterable and to make it appear like a collection of
-objects.
+To make an object iterable and to make it appear like a collection of objects.
 
 Examples
 --------
@@ -45,18 +44,6 @@ BookList.php
    :language: php
    :linenos:
 
-BookListIterator.php
-
-.. literalinclude:: BookListIterator.php
-   :language: php
-   :linenos:
-
-BookListReverseIterator.php
-
-.. literalinclude:: BookListReverseIterator.php
-   :language: php
-   :linenos:
-
 Test
 ----
 
@@ -67,4 +54,4 @@ Tests/IteratorTest.php
    :linenos:
 
 .. _`GitHub`: https://github.com/domnikl/DesignPatternsPHP/tree/master/Behavioral/Iterator
-.. __: http://en.wikipedia.org/wiki/Iterator_pattern
+.. __: http://en.wikipedia.org/wiki/Iterator_pattern
Original file line number	Diff line number	Diff line change
`@@ -4,27 +4,33 @@`
`4`	`4`
`5`	`5`	`class Book`
`6`	`6`	`{`
	`7`	`+ /**`
	`8`	`+ * @var string`
	`9`	`+ */`
`7`	`10`	`private $author;`
`8`	`11`
	`12`	`+ /**`
	`13`	`+ * @var string`
	`14`	`+ */`
`9`	`15`	`private $title;`
`10`	`16`
`11`		`- public function __construct($title, $author)`
	`17`	`+ public function __construct(string $title, string $author)`
`12`	`18`	`{`
`13`	`19`	`$this->author = $author;`
`14`	`20`	`$this->title = $title;`
`15`	`21`	`}`
`16`	`22`
`17`		`- public function getAuthor()`
	`23`	`+ public function getAuthor(): string`
`18`	`24`	`{`
`19`	`25`	`return $this->author;`
`20`	`26`	`}`
`21`	`27`
`22`		`- public function getTitle()`
	`28`	`+ public function getTitle(): string`
`23`	`29`	`{`
`24`	`30`	`return $this->title;`
`25`	`31`	`}`
`26`	`32`
`27`		`- public function getAuthorAndTitle()`
	`33`	`+ public function getAuthorAndTitle(): string`
`28`	`34`	`{`
`29`	`35`	`return $this->getTitle().' by '.$this->getAuthor();`
`30`	`36`	`}`