From 479ca11ede5794e0efc86b8f7cdec4c699f22a0f Mon Sep 17 00:00:00 2001
From: KostyaDanovsky <kostya.danovsky@gmail.com>
Date: Tue, 29 Aug 2017 15:55:09 +0300
Subject: [PATCH] fix(dashboard): improve dropdowns for electricity and traffic
 charts

---
 .../dashboard/electricity/electricity.component.html  |  3 ++-
 src/app/pages/dashboard/traffic/traffic.component.ts  | 11 ++++++++++-
 2 files changed, 12 insertions(+), 2 deletions(-)

diff --git a/src/app/pages/dashboard/electricity/electricity.component.html b/src/app/pages/dashboard/electricity/electricity.component.html
index 6096b706..91ccb120 100644
--- a/src/app/pages/dashboard/electricity/electricity.component.html
+++ b/src/app/pages/dashboard/electricity/electricity.component.html
@@ -42,7 +42,8 @@
       </div>
 
       <div ngbDropdown [ngClass]="{'ghost-dropdown': currentTheme == 'default'}">
-        <button type="button" class="btn btn-primary" ngbDropdownToggle>
+        <button type="button" ngbDropdownToggle class="btn"
+                [ngClass]="{ 'btn-success': currentTheme == 'default', 'btn-primary': currentTheme != 'default'}">
           {{ type }}
         </button>
         <ul class="dropdown-menu">
diff --git a/src/app/pages/dashboard/traffic/traffic.component.ts b/src/app/pages/dashboard/traffic/traffic.component.ts
index 2a953cc6..c2a226d9 100644
--- a/src/app/pages/dashboard/traffic/traffic.component.ts
+++ b/src/app/pages/dashboard/traffic/traffic.component.ts
@@ -1,4 +1,5 @@
 import { Component } from '@angular/core';
+import { NbThemeService } from '@nebular/theme';
 
 @Component({
   selector: 'ngx-traffic',
@@ -8,7 +9,8 @@ import { Component } from '@angular/core';
       <nb-card-header>
         <span>Traffic Consumption</span>
         <div class="ghost-dropdown" ngbDropdown>
-          <button type="button" class="btn btn-sm btn-primary" ngbDropdownToggle>
+          <button type="button" class="btn btn-sm" ngbDropdownToggle
+                  [ngClass]="{ 'btn-success': currentTheme == 'default', 'btn-primary': currentTheme != 'default'}">
             {{ type }}
           </button>
           <ul class="dropdown-menu">
@@ -25,4 +27,11 @@ import { Component } from '@angular/core';
 export class TrafficComponent {
   type: string = 'month';
   types = ['week', 'month', 'year'];
+  currentTheme: string;
+
+  constructor(private themeService: NbThemeService) {
+    this.themeService.getJsTheme().subscribe(theme => {
+      this.currentTheme = theme.name;
+    });
+  }
 }