Add badge to Model Collector

barryvdh · Apr 16, 2020 · 95c31aa · 95c31aa
1 parent aaf9274
commit 95c31aa
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/src/DataCollector/ModelsCollector.php b/src/DataCollector/ModelsCollector.php
@@ -14,6 +14,7 @@
 class ModelsCollector extends DataCollector implements DataCollectorInterface, Renderable
 {
     public $models = [];
+    public $count = 0;
 
     /**
      * @param Dispatcher $events
@@ -25,6 +26,7 @@ public function __construct(Dispatcher $events)
                 foreach (array_filter($models) as $model) {
                     $class = get_class($model);
                     $this->models[$class] = ($this->models[$class] ?? 0) + 1;
+                    $this->count++;
                 }
             }
         });
@@ -34,7 +36,7 @@ public function collect()
     {
         ksort($this->models, SORT_NUMERIC);
 
-        return array_reverse($this->models);
+        return ['data' => array_reverse($this->models), 'count' => $this->count];
     }
 
     /**
@@ -54,8 +56,12 @@ public function getWidgets()
             "models" => [
                 "icon" => "cubes",
                 "widget" => "PhpDebugBar.Widgets.HtmlVariableListWidget",
-                "map" => "models",
+                "map" => "models.data",
                 "default" => "{}"
+            ],
+            'models:badge' => [
+                'map' => 'models.count',
+                'default' => 0
             ]
         ];
     }