From 7967ddead790d89478554cd6a96429fd06958590 Mon Sep 17 00:00:00 2001
From: Aman Harwara <amanharwara@protonmail.com>
Date: Sat, 23 Mar 2024 00:01:54 +0530
Subject: [PATCH] fix: Fixed issue with system color scheme not being applied
 correctly on start

---
 .../src/Domain/Component/ComponentManagerInterface.ts         | 2 +-
 .../snjs/lib/Services/ComponentManager/ComponentManager.ts    | 4 ++--
 packages/ui-services/src/Theme/ThemeManager.ts                | 2 +-
 3 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/packages/services/src/Domain/Component/ComponentManagerInterface.ts b/packages/services/src/Domain/Component/ComponentManagerInterface.ts
index e27c765fe..e2161a195 100644
--- a/packages/services/src/Domain/Component/ComponentManagerInterface.ts
+++ b/packages/services/src/Domain/Component/ComponentManagerInterface.ts
@@ -36,7 +36,7 @@ export interface ComponentManagerInterface {
   getDefaultEditorIdentifier(currentTag?: SNTag): string
 
   isThemeActive(theme: UIFeature<ThemeFeatureDescription>): boolean
-  toggleTheme(theme: UIFeature<ThemeFeatureDescription>): Promise<void>
+  toggleTheme(theme: UIFeature<ThemeFeatureDescription>, skipEntitlementCheck?: boolean): Promise<void>
   getActiveThemes(): UIFeature<ThemeFeatureDescription>[]
   getActiveThemesIdentifiers(): { features: NativeFeatureIdentifier[]; uuids: Uuid[] }
 
diff --git a/packages/snjs/lib/Services/ComponentManager/ComponentManager.ts b/packages/snjs/lib/Services/ComponentManager/ComponentManager.ts
index 87f532d04..4773b2a77 100644
--- a/packages/snjs/lib/Services/ComponentManager/ComponentManager.ts
+++ b/packages/snjs/lib/Services/ComponentManager/ComponentManager.ts
@@ -390,7 +390,7 @@ export class ComponentManager
     return this.viewers.find((viewer) => viewer.sessionKey === key)
   }
 
-  public async toggleTheme(uiFeature: UIFeature<ThemeFeatureDescription>): Promise<void> {
+  public async toggleTheme(uiFeature: UIFeature<ThemeFeatureDescription>, skipEntitlementCheck = false): Promise<void> {
     this.logger.info('Toggling theme', uiFeature.uniqueIdentifier)
 
     if (this.isThemeActive(uiFeature)) {
@@ -399,7 +399,7 @@ export class ComponentManager
     }
 
     const featureStatus = this.features.getFeatureStatus(uiFeature.uniqueIdentifier)
-    if (featureStatus !== FeatureStatus.Entitled) {
+    if (!skipEntitlementCheck && featureStatus !== FeatureStatus.Entitled) {
       return
     }
 
diff --git a/packages/ui-services/src/Theme/ThemeManager.ts b/packages/ui-services/src/Theme/ThemeManager.ts
index a36bd0bcc..72892f6be 100644
--- a/packages/ui-services/src/Theme/ThemeManager.ts
+++ b/packages/ui-services/src/Theme/ThemeManager.ts
@@ -247,7 +247,7 @@ export class ThemeManager extends AbstractUIService {
     } else {
       const theme = themes.find((theme) => theme.featureIdentifier === themeIdentifier)
       if (theme && !this.components.isThemeActive(theme)) {
-        this.components.toggleTheme(theme).catch(console.error)
+        this.components.toggleTheme(theme, true).catch(console.error)
       }
     }
   }