symfony · dator-zz · Feb 23, 2011 · Feb 23, 2011
diff --git a/src/Symfony/Bundle/SwiftmailerBundle/DataCollector/SpoolDataCollector.php b/src/Symfony/Bundle/SwiftmailerBundle/DataCollector/SpoolDataCollector.php
@@ -0,0 +1,62 @@
+<?php
+
+/*
+ * This file is part of the Symfony package.
+ *
+ * (c) Fabien Potencier <fabien.potencier@symfony-project.com>
+ *
+ * For the full copyright and license information, please view the LICENSE
+ * file that was distributed with this source code.
+ */
+
+namespace Symfony\Bundle\SwiftmailerBundle\DataCollector;
+
+use Symfony\Component\HttpKernel\DataCollector\DataCollector;
+use Symfony\Component\HttpFoundation\Request;
+use Symfony\Component\HttpFoundation\Response;
+
+/**
+ * SpoolDataCollector.
+ *
+ * @author Clément JOBEILI <clement.jobeili@gmail.com>
+ */
+class SpoolDataCollector extends DataCollector
+{
+    protected $path;
+
+    public function __construct($path)
+    {
+        $this->path = $path;
+    }
+
+    /**
+     * {@inheritdoc}
+     */
+    public function collect(Request $request, Response $response, \Exception $exception = null)
+    {
+        $messageCount = 0; 
+
+        foreach (new \DirectoryIterator($this->path) as $file) {
+            $file = $file->getRealPath();
+
+            if (strpos($file, '.message')) {
+                $messageCount++;
+            }
+        }
+
+        $this->data['messageCount'] = $messageCount;
+    }
+
+    public function getMessageCount()
+    {
+        return $this->data['messageCount'];
+    }
+
+    /**
+     * {@inheritdoc}
+     */
+    public function getName()
+    {
+        return 'spool';
+    }
+}
diff --git a/src/Symfony/Bundle/SwiftmailerBundle/Resources/config/spool.xml b/src/Symfony/Bundle/SwiftmailerBundle/Resources/config/spool.xml
@@ -22,5 +22,11 @@
     </service>
 
     <service id="swiftmailer.spool" alias="swiftmailer.spool.file" public="false" />
+
+    <service id="swiftmailer.data_collector" class="%swiftmailer.data_collector.class%" public="false">
+        <tag name="data_collector" template="SwiftmailerBundle:Collector:spool" id="spool" />
+        <argument>%swiftmailer.spool.file.path%</argument>
+    </service>
+
   </services>
 </container>
diff --git a/src/Symfony/Bundle/SwiftmailerBundle/Resources/public/images/mail.png b/src/Symfony/Bundle/SwiftmailerBundle/Resources/public/images/mail.png
diff --git a/src/Symfony/Bundle/SwiftmailerBundle/Resources/views/Collector/spool.html.twig b/src/Symfony/Bundle/SwiftmailerBundle/Resources/views/Collector/spool.html.twig
@@ -0,0 +1,7 @@
+{% extends 'WebProfilerBundle:Profiler:layout.html.twig' %}
+
+{% block toolbar %}
+<img style="margin: 0 5px 0 10px; vertical-align: middle; height: 24px" width="24" height="24" alt="Spool" src="data:image/png;base64,iVBORw0KGgoAAAANSUhEUgAAABgAAAAYCAYAAADgdz34AAADNElEQVRIidWVzW4cRRDHf9092zvj3c14vR/MSpgTByR84IAckQfgYr8CDxDEExCJF4A8DQfulh/BjkWkROZgs4ode8N+2DszXcVhdhcbIdtSCBJ/qdU9PV1VXf/6aKOqfEjYD6r9vzBgtra2vnz69Nvv8iJH/85X9WUeqqxe9+bVq9e/PX/+0w+rzZ2d3W+KotR/C3t7e8eqynJEQJHnOcfHp7RaLVqtFiLycAqMQVW5vLwkjmNUmdz8b1WFJEnIsozZbMZwOCSEQBCp5juGiHJ1dcXp6Snee3q9LtbaWzRHqhBCAAwvX/66EnTu/virKiKCcw5rq/O1mq9n2eAza00+Ho8vIgARIS9yRIQnXz0BYxARjLk7viKCtXZF0wKfHh0dHdTrnu+fPfsxUtXq1iEAMHr3jmazuYrFSvAfjC13lmeMMVhrzXg8diEEfK0WrTwIIWCModfrMRwOmUwmdDqdWwrugjEGUeXi7Vucc3S7Hax1GikVjyKCAtZasizj4uKCk5MTNjY28N7fmVnWWq7nc0aXl6RpSpqmq4tFqFYZIwKqlGVARFhfbxPHMWdnZyRJQrPZWnhy0xuDMYbRaESe5/R6Pbz3FEUBJIBil1lU5MUicAGRQFkWeO8ZDAaUZcn5+TlFkRNEKMsqjYui4Pz8DFElyzKcc5RluWBEEVGsMcbG9TqDQYYxZpGmFWVlWaIK7XabEEqGwyFXsxkigdlsxunvJwBstNurOC5lRQRRIZpOJ6MXLw6P37w568RxPQGiim8DBq5nU4KU9Pt9Qgj8MR4zn89xzrH58Saqymg0oh7HeO8XDFa0qyh2f3//l+3t7c93d3c+Ojg4/DmOE0SUoiyYTqZEkeNR6xHWOWre0+126ff7dLpdoloN7z1pmiIhMJ1MVoW69MYuetRMVa+MMUWQwDyfE8pAmj5irdHAWkvkHJFzOGtxizlyrlo7R5qmNJtNrufX5HmOyF/NboUgYoqioLG2RpIkVdaognP31oGqsraQG48n5PM5IuG2gWajkWy01+9Vdh8ajTUAfL0em5tVuv348defbG5+UZTlez/Uvlbj4PBw3zykDbwP/v+P/p/fhU37KZPH8QAAAABJRU5ErkJggg==" />
+{{ collector.messagecount }}
+
+{% endblock %}