From f59f4243be190e6cff91ce271c11d14c0eab72e7 Mon Sep 17 00:00:00 2001
From: arichorn <78001398+arichorn@users.noreply.github.com>
Date: Thu, 14 Sep 2023 18:56:39 -0500
Subject: [PATCH] Update OLED Dark Theme

Added stuff so that OLED Dark Theme works correctly on the latest version of YouTube.
---
 Source/Themes.xm | 21 +++++++++++++++++++++
 1 file changed, 21 insertions(+)

diff --git a/Source/Themes.xm b/Source/Themes.xm
index 452a34e..669d38c 100644
--- a/Source/Themes.xm
+++ b/Source/Themes.xm
@@ -521,6 +521,27 @@ UIColor* raisedColor = [UIColor blackColor];
 
 // Others
 %hook _ASDisplayView
+- (void)layoutSubviews {
+    %orig;
+    if (isDarkMode()) {
+    UIResponder *responder = [self nextResponder];
+    while (responder != nil) {
+        if ([responder isKindOfClass:NSClassFromString(@"YTActionSheetDialogViewController")]) {
+            self.backgroundColor = [UIColor blackColor];
+        }
+        if ([responder isKindOfClass:NSClassFromString(@"YTPanelLoadingStrategyViewController")]) {
+            self.backgroundColor = [UIColor blackColor];
+        }
+        if ([responder isKindOfClass:NSClassFromString(@"YTTabHeaderElementsViewController")]) {
+            self.backgroundColor = [UIColor blackColor];
+        }
+        if ([responder isKindOfClass:NSClassFromString(@"YTEditSheetControllerElementsContentViewController")]) {
+            self.backgroundColor = [UIColor blackColor];
+        }
+        responder = [responder nextResponder];
+      }
+   }
+}
 - (void)didMoveToWindow {
     %orig;
     if (isDarkMode()) {