Added observeTree method

package.json

@@ -1,6 +1,6 @@
 {
   "name": "@asmcss/assembler",
-  "version": "0.7.0",
+  "version": "0.7.1",
   "main": "dist/assembler.cjs.js",
   "module": "dist/assembler.es.js",
   "browser": "dist/assembler.js",

src/index.ts

@@ -15,7 +15,7 @@
  */
 
 import {generateStyles} from "./generators";
-import {observeDocument, observeShadow} from "./observers";
+import {observeDocument, observeShadow, observeTree} from "./observers";
 import {getUserSettings} from "./helpers";
 import {generateRootVariables} from "./variables";
 import StyleHandler from "./StyleHandler";
@@ -86,6 +86,12 @@ export function init(options?: {[key: string]: string}): boolean {
 
     observeDocument(document, styleHandler);
 
+    if (window) {
+        window.addEventListener('DOMContentLoaded', function () {
+            observeTree(document.body, styleHandler);
+        });
+    }
+
     return true;
 }
 

src/observers.ts

@@ -21,6 +21,16 @@ let _elementObserver:MutationObserver = null;
 let _shadowRootObserver:MutationObserver = null;
 const observedElements = new WeakMap<HTMLElement, any[]>();
 
+export function observeTree(element: HTMLElement, handler: StyleHandler): void {
+    for (let e = element; e != null; e = e.nextElementSibling as HTMLElement) {
+        if (!observedElements.has(e)) {
+            observe(e, handler);
+        } else {
+            observeTree(e.firstElementChild as HTMLElement, handler);
+        }
+    }
+}
+
 export function observeDocument(document: Document, handler: StyleHandler): void {
     if (_documentObserver === null) {
         _documentObserver = new MutationObserver(function (mutations: MutationRecord[]): void {

types/observers.d.ts

@@ -1,3 +1,4 @@
 import StyleHandler from "./StyleHandler";
+export declare function observeTree(element: HTMLElement, handler: StyleHandler): void;
 export declare function observeDocument(document: Document, handler: StyleHandler): void;
 export declare function observeShadow(shadow: ShadowRoot, handler: StyleHandler): void;