From 1393f99b39906f85689f7781da188e6fb414eaf6 Mon Sep 17 00:00:00 2001
From: Arpit Chakladar <54011232+arpitchakladar@users.noreply.github.com>
Date: Tue, 14 Feb 2023 11:53:09 +0530
Subject: Added some style changes.

---
 src/utils/observe-dom.ts | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

(limited to 'src/utils/observe-dom.ts')

diff --git a/src/utils/observe-dom.ts b/src/utils/observe-dom.ts
index 30a07b4..8ea1a98 100644
--- a/src/utils/observe-dom.ts
+++ b/src/utils/observe-dom.ts
@@ -1,8 +1,12 @@
-export const observeDOM = (callback: () => boolean) => {
+export const observeDOM = (callback: () => boolean, element: HTMLElement = document as unknown as HTMLElement) => {
 	const observer = new MutationObserver(() => {
 		if (callback()) {
 			observer.disconnect();
 		}
 	});
-	observer.observe(document, {childList: true, subtree: true});
+
+	observer.observe(element, {
+		childList: true,
+		subtree: true
+	});
 };
-- 
cgit v1.2.3