From 7f3bbd6e514585750477f6b5f27e3d996c24902e Mon Sep 17 00:00:00 2001
From: Matt Gibson <mgibson@bitwarden.com>
Date: Wed, 24 Mar 2021 11:15:51 -0500
Subject: [PATCH] Use UserService to determine if user has premium access
 (#311)

---
 src/angular/components/premium.component.ts | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/src/angular/components/premium.component.ts b/src/angular/components/premium.component.ts
index 9bb2ac092a..b37d4d1f2d 100644
--- a/src/angular/components/premium.component.ts
+++ b/src/angular/components/premium.component.ts
@@ -4,6 +4,7 @@ import { ApiService } from '../../abstractions/api.service';
 import { I18nService } from '../../abstractions/i18n.service';
 import { PlatformUtilsService } from '../../abstractions/platformUtils.service';
 import { TokenService } from '../../abstractions/token.service';
+import { UserService } from '../../abstractions/user.service';
 
 export class PremiumComponent implements OnInit {
     isPremium: boolean = false;
@@ -11,10 +12,10 @@ export class PremiumComponent implements OnInit {
     refreshPromise: Promise<any>;
 
     constructor(protected i18nService: I18nService, protected platformUtilsService: PlatformUtilsService,
-        protected tokenService: TokenService, protected apiService: ApiService) { }
+        protected apiService: ApiService, protected userService: UserService) { }
 
     async ngOnInit() {
-        this.isPremium = this.tokenService.getPremium();
+        this.isPremium = await this.userService.canAccessPremium();
     }
 
     async refresh() {
@@ -22,7 +23,7 @@ export class PremiumComponent implements OnInit {
             this.refreshPromise = this.apiService.refreshIdentityToken();
             await this.refreshPromise;
             this.platformUtilsService.showToast('success', null, this.i18nService.t('refreshComplete'));
-            this.isPremium = this.tokenService.getPremium();
+            this.isPremium = await this.userService.canAccessPremium();
         } catch { }
     }