Optimize single element find

Author: mvorisek

src/Element/Element.php

@@ -139,24 +139,17 @@ public function waitFor($timeout, $callback)
     }
 
     /**
-     * {@inheritdoc}
-     */
-    public function find($selector, $locator)
-    {
-        $items = $this->findAll($selector, $locator);
-
-        return count($items) ? current($items) : null;
-    }
-
-    /**
-     * {@inheritdoc}
+     * @param string       $selector selector engine name
+     * @param string|array $locator  selector locator
+     *
+     * @return NodeElement[]
      */
-    public function findAll($selector, $locator)
+    private function findAllWithLimit($selector, $locator, ?int $limit): array
     {
         if ('named' === $selector) {
-            $items = $this->findAll('named_exact', $locator);
+            $items = $this->findAllWithLimit('named_exact', $locator);
             if (empty($items)) {
-                $items = $this->findAll('named_partial', $locator);
+                $items = $this->findAllWithLimit('named_partial', $locator);
             }
 
             return $items;
@@ -165,9 +158,35 @@ public function findAll($selector, $locator)
         $xpath = $this->selectorsHandler->selectorToXpath($selector, $locator);
         $xpath = $this->xpathManipulator->prepend($xpath, $this->getXpath());
 
+        if ($limit !== null) {
+            if ($limit === 1) {
+                $xpath = '(' . $xpath . ')[1]';
+            } else {
+                $xpath = '(' . $xpath . ')[position() <= ' . $limit . ']';
+            }
+        }
+
         return $this->getDriver()->find($xpath);
     }
 
+    /**
+     * {@inheritdoc}
+     */
+    public function find($selector, $locator)
+    {
+        $items = $this->findAllWithLimit($selector, $locator, 1);
+
+        return count($items) > 0 ? current($items) : null;
+    }
+
+    /**
+     * {@inheritdoc}
+     */
+    public function findAll($selector, $locator)
+    {
+        return $this->findAllWithLimit($selector, $locator, null);
+    }
+
     /**
      * {@inheritdoc}
      */