Added WebDriverAware functionality

README.md

@@ -137,6 +137,27 @@ public class SampleTest {
 }
 ```
 
+Need access to the WebDriver in your HtmlElement?
+-------------------------------------------------
+Use WebDriverAwareDecorator in your page objects
+
+```java
+PageFactory.initElements(new WebDriverAwareDecorator(new HtmlElementLocatorFactory(driver), driver), this);
+```
+
+Implement WebDriverAware on your HtmlElement
+
+```
+public class MyComponent extends HtmlElement implements WebDriverAware {
+	WebDriver driver = null;
+
+	@Override
+	public void setWebDriver(WebDriver driver) {
+		this.driver = driver;
+	}
+}
+```
+
 Questions?
 ----------
 In case you can't find an answer in documentation and examples provided above, you can ask it on StackOverflow with [![htmlelements](https://img.shields.io/badge/stackoverflow-htmlelements-orange.svg?style=flat)](http://stackoverflow.com/questions/tagged/htmlelements) tag.

htmlelements-java/src/main/java/ru/yandex/qatools/htmlelements/loader/decorator/WebDriverAware.java

@@ -0,0 +1,7 @@
+package ru.yandex.qatools.htmlelements.loader.decorator;
+
+import org.openqa.selenium.WebDriver;
+
+public interface WebDriverAware {
+    public void setWebDriver(WebDriver driver);
+}

htmlelements-java/src/main/java/ru/yandex/qatools/htmlelements/loader/decorator/WebDriverAwareDecorator.java

@@ -0,0 +1,44 @@
+package ru.yandex.qatools.htmlelements.loader.decorator;
+
+import java.lang.reflect.Field;
+import java.util.List;
+
+import org.openqa.selenium.WebDriver;
+
+import ru.yandex.qatools.htmlelements.element.HtmlElement;
+import ru.yandex.qatools.htmlelements.loader.decorator.HtmlElementDecorator;
+import ru.yandex.qatools.htmlelements.pagefactory.CustomElementLocatorFactory;
+
+public class WebDriverAwareDecorator extends HtmlElementDecorator {
+    private WebDriver driver;
+
+    public WebDriverAwareDecorator(CustomElementLocatorFactory factory, WebDriver driver) {
+        super(factory);
+
+        this.driver = driver;
+    }
+
+    @Override
+    protected <T extends HtmlElement> T decorateHtmlElement(ClassLoader loader, Field field) {
+        T element = super.decorateHtmlElement(loader, field);
+
+        if (element instanceof WebDriverAware) {
+            ((WebDriverAware)element).setWebDriver(driver);
+        }        
+
+        return element;
+    }
+
+    @Override
+    protected <T extends HtmlElement> List<T> decorateHtmlElementList(ClassLoader loader, Field field) {
+        List<T> elements = super.decorateHtmlElementList(loader, field);
+
+        if (!elements.isEmpty() && elements.get(0) instanceof WebDriverAware) {
+            for (T element : elements) {
+                ((WebDriverAware)element).setWebDriver(driver);
+            }
+        }
+
+        return elements;
+    }
+}