From 71619e18b1d2e4fc454e3bc85d7c2b181a3c6526 Mon Sep 17 00:00:00 2001
From: Kasra Bigdeli <kasra85@gmail.com>
Date: Fri, 4 Aug 2023 21:34:58 -0700
Subject: [PATCH] Fixed feature flags

---
 src/user/FeatureFlags.ts   | 8 ++++++--
 src/user/pro/ProManager.ts | 6 ++----
 2 files changed, 8 insertions(+), 6 deletions(-)

diff --git a/src/user/FeatureFlags.ts b/src/user/FeatureFlags.ts
index 572d113..b207169 100644
--- a/src/user/FeatureFlags.ts
+++ b/src/user/FeatureFlags.ts
@@ -6,8 +6,8 @@ import Logger from '../utils/Logger'
 export default class FeatureFlags {
     static instance: FeatureFlags
 
-    featureFlags: any | undefined
-    static IS_PRO_ENABLED: 'isProEnabled'
+    private featureFlags: any | undefined
+    static IS_PRO_ENABLED = 'isProEnabled'
 
     static get(datastore: DataStore) {
         if (!FeatureFlags.instance) {
@@ -22,6 +22,10 @@ export default class FeatureFlags {
         self.featureFlags = self.datastore.getFeatureFlags()
     }
 
+    getFeatureFlags(): any | undefined {
+        return this.featureFlags
+    }
+
     private refreshFeatureFlags() {
         const self = this
         Promise.resolve() //
diff --git a/src/user/pro/ProManager.ts b/src/user/pro/ProManager.ts
index 8a40e51..6b97801 100644
--- a/src/user/pro/ProManager.ts
+++ b/src/user/pro/ProManager.ts
@@ -127,13 +127,11 @@ export default class ProManager {
                 return self.proDataStore.getApiKey()
             })
             .then(function (apiKey) {
+                const flags = self.featureFlagsProvider.getFeatureFlags()
                 return {
                     isSubscribed: !!apiKey,
                     isFeatureFlagEnabled:
-                        self.featureFlagsProvider.featureFlags &&
-                        self.featureFlagsProvider.featureFlags[
-                            FeatureFlags.IS_PRO_ENABLED
-                        ],
+                        flags && flags[FeatureFlags.IS_PRO_ENABLED],
                 }
             })
     }