Writer gets messages collector, not whole phpdebugbar

snapshotpl · Apr 22, 2015 · e6b21f2 · e6b21f2
1 parent 7e0fd61
commit e6b21f2
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 8 deletions.
diff --git a/src/ZfSnapPhpDebugBar/Log/Writer/PhpDebugBar.php b/src/ZfSnapPhpDebugBar/Log/Writer/PhpDebugBar.php
@@ -2,7 +2,7 @@
 
 namespace ZfSnapPhpDebugBar\Log\Writer;
 
-use DebugBar\DebugBar;
+use DebugBar\DataCollector\MessagesCollector;
 use Zend\Log\Writer\AbstractWriter;
 
 /**
@@ -13,17 +13,17 @@
 class PhpDebugBar extends AbstractWriter
 {
     /**
-     * @var DebugBar
+     * @var MessagesCollector
      */
-    protected $debugbar;
+    protected $messagesCollector;
 
     /**
      *
-     * @param DebugBar $debugbar
+     * @param MessagesCollector $debugbar
      */
-    public function __construct(DebugBar $debugbar)
+    public function __construct(MessagesCollector $debugbar)
     {
-        $this->debugbar = $debugbar;
+        $this->messagesCollector = $debugbar;
     }
 
     /**
@@ -32,7 +32,7 @@ public function __construct(DebugBar $debugbar)
     protected function doWrite(array $event)
     {
         $priority = $this->priorityMap($event['priorityName']);
-        $this->debugbar['messages']->addMessage($event['message'], $priority);
+        $this->messagesCollector->addMessage($event['message'], $priority);
     }
 
     /**

diff --git a/src/ZfSnapPhpDebugBar/Log/Writer/PhpDebugBarFactory.php b/src/ZfSnapPhpDebugBar/Log/Writer/PhpDebugBarFactory.php
@@ -22,7 +22,9 @@ public function createService(ServiceLocatorInterface $serviceLocator)
         /* @var $debugbar \DebugBar\DebugBar */
         $debugbar = $serviceLocator->get('debugbar');
 
-        return new PhpDebugBar($debugbar);
+        $messagesCollector = $debugbar['messages'];
+
+        return new PhpDebugBar($messagesCollector);
     }
 
 }