Added magic and each methods to the collection.

Plus some tests

Author: paquettg

src/PHPHtmlParser/Dom/Collection.php

@@ -1,10 +1,10 @@
 <?php
 namespace PHPHtmlParser\Dom;
 
-use IteratorAggregate;
-use ArrayIterator;
-use ArrayAccess;
 use Countable;
+use ArrayAccess;
+use ArrayIterator;
+use IteratorAggregate;
 
 class Collection implements IteratorAggregate, ArrayAccess, Countable {
 
@@ -15,6 +15,54 @@ class Collection implements IteratorAggregate, ArrayAccess, Countable {
 	 */
 	protected $collection = [];
 
+	/**
+	 * Attempts to call the method on the first node in
+	 * the collection.
+	 *
+	 * @param string $method
+	 * @param array $arguments
+	 * @return mixed;
+	 */
+	public function __call($method, $arguments)
+	{
+		$node = reset($this->collection);
+		if ($node instanceof Node)
+		{
+			return call_user_func_array([$node, $method], $arguments);
+		}
+	}
+
+	/**
+	 * Attempts to apply the magic get to the first node
+	 * in the collection.
+	 *
+	 * @param mixed $key
+	 * @return mixed
+	 */
+	public function __get($key)
+	{
+		$node = reset($this->collection);
+		if ($node instanceof Node)
+		{
+			return $node->$key;
+		}
+	}
+
+	/** 
+	 * Applies the magic string method to the first node in
+	 * the collection.
+	 *
+	 * @return string
+	 */
+	public function __toString()
+	{
+		$node = reset($this->collection);
+		if ($node instanceof Node)
+		{
+			return (string) $node;
+		}
+	}
+
 	/** 
 	 * Returns the count of the collection.
 	 *
@@ -84,4 +132,16 @@ public function offsetGet($offset)
     {
         return isset($this->collection[$offset]) ? $this->collection[$offset] : null;
     }
+
+	/**
+	 * Similar to jQuery "each" method. Calls the callback with each
+	 * Node in this collection.
+	 */
+    public function each($callback)
+    {
+    	foreach ($this->collection as $key => $value)
+    	{
+    		$callback($value, $key);
+    	}
+    }
 }

tests/CollectionTest.php

@@ -0,0 +1,45 @@
+<?php
+
+use PHPHtmlParser\Selector;
+use PHPHtmlParser\Dom\HtmlNode;
+use PHPHtmlParser\Dom\Tag;
+
+class CollectionTest extends PHPUnit_Framework_TestCase {
+	
+	public function testEach()
+	{
+		$root   = new HtmlNode(new Tag('root'));
+		$parent = new HtmlNode(new Tag('div'));
+		$child1 = new HtmlNode(new Tag('a'));
+		$child2 = new HtmlNode(new Tag('p'));
+		$child3 = new HtmlNode(new Tag('a'));
+		$root->addChild($parent);
+		$parent->addChild($child1);
+		$parent->addChild($child2);
+		$child2->addChild($child3);
+
+		$selector   = new Selector('a');
+		$collection = $selector->find($root);
+		$count      = 0;
+		$collection->each(function ($node) use (&$count) {
+			++$count;
+		});
+		$this->assertEquals(2, $count);
+	}
+
+	public function testCallMagic()
+	{
+		$root   = new HtmlNode(new Tag('root'));
+		$parent = new HtmlNode(new Tag('div'));
+		$child1 = new HtmlNode(new Tag('a'));
+		$child2 = new HtmlNode(new Tag('p'));
+		$child3 = new HtmlNode(new Tag('a'));
+		$root->addChild($parent);
+		$parent->addChild($child1);
+		$parent->addChild($child2);
+		$child2->addChild($child3);
+
+		$selector = new Selector('div * a');
+		$this->assertEquals($child3->id(), $selector->find($root)->id());
+	}
+}