From 6c8eb71d46489790c18928ef76b6aae9341e92cb Mon Sep 17 00:00:00 2001
From: smartfridge <37928912+smartfrigde@users.noreply.github.com>
Date: Sat, 10 Dec 2022 22:39:30 +0100
Subject: [PATCH] Enable titlebar fix

---
 src/preload/preload.ts | 12 +++++++++++-
 1 file changed, 11 insertions(+), 1 deletion(-)

diff --git a/src/preload/preload.ts b/src/preload/preload.ts
index f5e84f2..e987162 100644
--- a/src/preload/preload.ts
+++ b/src/preload/preload.ts
@@ -4,7 +4,7 @@ import "./capturer";
 import "./patch";
 import * as fs from "fs";
 import * as path from "path";
-import {injectHummusTitlebar, injectTitlebar} from "./titlebar";
+import {fixTitlebar, injectHummusTitlebar, injectTitlebar} from "./titlebar";
 import {sleep, addStyle, addScript} from "../utils";
 import {injectMobileStuff} from "./mobile";
 var version = ipcRenderer.sendSync("displayVersion");
@@ -61,6 +61,16 @@ if (window.location.href.indexOf("splash.html") > -1) {
         addScript(fs.readFileSync(path.join(__dirname, "../", "/content/js/rpc.js"), "utf8"));
         const cssPath = path.join(__dirname, "../", "/content/css/discord.css");
         addStyle(fs.readFileSync(cssPath, "utf8"));
+        if (document.getElementById("window-controls-container") == null) {
+            console.warn("Titlebar didn't inject, retrying...");
+            if (ipcRenderer.sendSync("titlebar")) {
+                if (channel == "hummus") {
+                    injectHummusTitlebar();
+                } else {
+                    fixTitlebar();
+                }
+            }
+        }
         await updateLang();
     });
 }