Add action collectors to scoped component browser instances

Author: nie7321

src/Dusk/Browser.php

@@ -2,6 +2,7 @@
 
 namespace BeyondCode\DuskDashboard\Dusk;
 
+use Closure;
 use BeyondCode\DuskDashboard\BrowserActionCollector;
 
 class Browser extends \Laravel\Dusk\Browser
@@ -61,6 +62,30 @@ public function refresh()
         return $browser;
     }
 
+    /** {@inheritdoc} */
+    public function with($selector, Closure $callback)
+    {
+        $action_collector_callback = function ($browser) use ($callback) {
+            print 'Entering the wrapped callback...';
+            $browser->setActionCollector($this->getActionCollector());
+
+            print 'Wrapped callback invoked & updated browser!';
+            return $callback($browser);
+        };
+
+        return parent::with($selector, $action_collector_callback);
+    }
+
+    /** {@inheritDoc} */
+    public function onComponent($component, $parentResolver)
+    {
+        if ($this->getActionCollector() === null) {
+            $this->setActionCollector(new BrowserActionCollector('dog'));
+        }
+
+        parent::onComponent($component, $parentResolver);
+    }
+
     public function getCurrentPageSource()
     {
         $this->ensurejQueryIsAvailable();