Adding class in the display::badge function

main/inc/lib/display.lib.php

@@ -1376,12 +1376,31 @@ class Display {
         return $div;
     }
     
-    function badge($count) {        
-        $count = intval($count);
+    function badge($count, $type ="warning") {        
+        $class = '';
+        
+        switch ($type) {
+            case 'success':
+                $class = 'badge-success';
+                break;
+            case 'warning':
+                $class = 'badge-warning';
+                break;
+            case 'important':
+                $class = 'badge-important';
+                break;
+            case 'info':
+                $class = 'badge-info';
+                break;
+            case 'inverse':
+                $class = 'badge-inverse';
+                break;            
+        }
+        
         if (!empty($count)) {
-            return ' <span class="badge badge-warning">'.$count.'</span>';
+            return ' <span class="badge '.$class.'">'.$count.'</span>';
         }
-        return '';
+        return null;
     }
     
     function label($content, $type = null) {