From c6e38540c58ae9f458d753339c2fa87d561769a0 Mon Sep 17 00:00:00 2001
From: diegommm <diegoaugustomolina@gmail.com>
Date: Fri, 30 Nov 2018 02:40:59 -0300
Subject: [PATCH] ADD OPTION TO RETURN ONLY FIRST ELEMENT

- Add magic method __invoke for convenience.

- Add support for returning only first element of the array when passed
"first" in $as_array or only its name, attributes, text or children if
passed any of those words instead.
---
 selector.inc | 37 +++++++++++++++++++++++++++++++++++--
 1 file changed, 35 insertions(+), 2 deletions(-)

diff --git a/selector.inc b/selector.inc
index affef19..309e282 100644
--- a/selector.inc
+++ b/selector.inc
@@ -26,9 +26,42 @@ class SelectorDOM {
     }
   }
   
+  public function __invoke($selector, $as_array = true) {
+    return $this->select( $selector, $as_array );
+  }
+  
   public function select($selector, $as_array = true) {
-    $elements = $this->xpath->evaluate(selector_to_xpath($selector));
-    return $as_array ? elements_to_array($elements) : $elements;
+    $ret = $this->xpath->evaluate(selector_to_xpath($selector));
+    if ( $as_array !== false ){
+        $ret = elements_to_array($ret);
+        if ( $as_array !== true ){
+            if ( count( $ret ) > 0 ){
+                switch ( $as_array ){
+                    case "first":
+                        $ret = $ret[0];
+                        break;
+                    case "name":
+                        $ret = $ret[0]["name"];
+                        break;
+                    case "attributes":
+                        $ret = $ret[0]["attributes"];
+                        break;
+                    case "text":
+                        $ret = $ret[0]["text"];
+                        break;
+                    case "children":
+                        $ret = $ret[0]["children"];
+                        break;
+                    default:
+                        $ret = null;
+                        break;
+                }
+            } else {
+                return null;
+            }
+        }
+    }
+    return $ret;
   }
 }