fix: dashboard icon inject and refactor some init code

src/Renderer/App.js

@@ -3,14 +3,14 @@ const Settings = require('./Settings');
 const Instance = require('./Instance');
 const AppLock = require('./AppLock');
 const { whatsappReady } = require('./whatsapp-inject');
+const Deferred = require('./Deferred');
 
 class App {
 	constructor() {
 		console.log('Initializing WALC');
 		this.initialized = false;
 		this.awaitApp();
 		ipcRenderer.on('renderTray', () => this.renderTray());
-		ipcRenderer.on('ready', () => this.init());
 		ipcRenderer.on('setFullWidth', (e, status) => this.setFullWidth(status));
 
 		window.WALC = {
@@ -19,12 +19,12 @@ class App {
 		};
 	}
 
-	awaitApp() {
+	async awaitApp() {
+		const pageReady = new Deferred();
 		const observer = new MutationObserver(() => {
-			const sidebar = document.querySelector('#app [data-testid=chatlist-header]');
-			if(sidebar) {
-				setTimeout(() => this.init(), 1000);
-				observer.disconnect();
+			const isReady = document.querySelector('#app [data-icon=\'chat\']');
+			if(isReady) {
+				setTimeout(() => pageReady.resolve(), 2000);
 			}
 		});
 
@@ -34,6 +34,15 @@ class App {
 				subtree: true,
 			});
 		}, 2000);
+
+		await Promise.race([
+			pageReady.promise,
+			new Promise((resolve) => {
+				ipcRenderer.once('ready', () => resolve());
+			}),
+		]);
+		this.init();
+		observer.disconnect();
 	}
 
 	async init() {

src/Renderer/AppLock.js

@@ -59,6 +59,9 @@ class AppLock {
 	}
 
 	async _initWhatsapp() {
+		if (!window.Store) {
+			return;
+		}
 		const user = window.Store.User.getMeUser() || {};
 		const myContact = await window.Store.Contact.find(user._serialized);
 		this.overlay.querySelector('.app-lock-user h1').textContent = myContact.displayName;

src/Renderer/Deferred.js

@@ -0,0 +1,8 @@
+module.exports = class Deferred {
+  constructor() {
+    this.promise = new Promise((resolve, reject)=> {
+      this.reject = reject
+      this.resolve = resolve
+    })
+  }
+}

src/Renderer/Instance.js

@@ -34,7 +34,8 @@ class Instance {
 	}
 
 	installDashboardIcon(icon) {
-		const container = document.querySelector('[data-testid=chatlist-header] div:first-child');
+		const profilePhoto = document.querySelector('[aria-label="profile photo"]');
+		const container = profilePhoto.parentElement;
 		this.image = document.createElement('div');
 		container.style.display = 'flex';
 		container.style.alignItems = 'center';