Add new models collector (barryvdh#947)

Author: reinink

config/debugbar.php

@@ -9,7 +9,7 @@
      |
      | Debugbar is enabled by default, when debug is set to true in app.php.
      | You can override the value by setting enable to true or false instead of null.
-     | 
+     |
      | You can provide an array of URI's that must be ignored (eg. 'api/*')
      |
      */
@@ -79,7 +79,7 @@
      |
      */
     'error_handler' => false,
-    
+
     /*
      |--------------------------------------------------------------------------
      | Clockwork integration
@@ -122,6 +122,7 @@
         'files'           => false, // Show the included files
         'config'          => false, // Display config settings
         'cache'           => false, // Display cache events
+        'models'          => false, // Display models
     ],
 
     /*

src/DataCollector/ModelsCollector.php

@@ -0,0 +1,54 @@
+<?php
+
+namespace Barryvdh\Debugbar\DataCollector;
+
+use Barryvdh\Debugbar\DataFormatter\SimpleFormatter;
+use DebugBar\DataCollector\MessagesCollector;
+use Illuminate\Events\Dispatcher;
+use Illuminate\Support\Str;
+
+/**
+ * Collector for Models.
+ */
+class ModelsCollector extends MessagesCollector
+{
+    public $models = [];
+
+    /**
+     * @param Dispatcher $events
+     */
+    public function __construct(Dispatcher $events)
+    {
+        parent::__construct('models');
+        $this->setDataFormatter(new SimpleFormatter());
+
+        $events->listen('eloquent.*', function ($event, $models) {
+            if (Str::contains($event, 'eloquent.retrieved')) {
+                foreach ($models as $model) {
+                    $class = get_class($model);
+                    $this->models[$class] = ($this->models[$class] ?? 0) + 1;
+                }
+            }
+        });
+    }
+
+    public function collect()
+    {
+        foreach ($this->models as $type => $count) {
+            $this->addMessage($count, $type);
+        }
+
+        return [
+            'count' => array_sum($this->models),
+            'messages' => $this->getMessages(),
+        ];
+    }
+
+    public function getWidgets()
+    {
+        $widgets = parent::getWidgets();
+        $widgets['models']['icon'] = 'cubes';
+
+        return $widgets;
+    }
+}

src/LaravelDebugbar.php

@@ -7,6 +7,7 @@
 use Barryvdh\Debugbar\DataCollector\GateCollector;
 use Barryvdh\Debugbar\DataCollector\LaravelCollector;
 use Barryvdh\Debugbar\DataCollector\LogsCollector;
+use Barryvdh\Debugbar\DataCollector\ModelsCollector;
 use Barryvdh\Debugbar\DataCollector\MultiAuthCollector;
 use Barryvdh\Debugbar\DataCollector\QueryCollector;
 use Barryvdh\Debugbar\DataCollector\SessionCollector;
@@ -404,6 +405,15 @@ function ($query, $bindings = null, $time = null, $connectionName = null) use ($
             }
         }
 
+        if ($this->shouldCollect('models', false)) {
+            try {
+                $modelsCollector = $this->app->make('Barryvdh\Debugbar\DataCollector\ModelsCollector');
+                $this->addCollector($modelsCollector);
+            } catch (\Exception $e){
+                // No Models collector
+            }
+        }
+
         if ($this->shouldCollect('mail', true) && class_exists('Illuminate\Mail\MailServiceProvider')) {
             try {
                 $mailer = $this->app['mailer']->getSwiftMailer();