feat: grab state from dom for requests

leafsphp · Apr 14, 2023 · 1bfdbbc · 1bfdbbc
1 parent 83f6146
commit 1bfdbbc
Showing 1 changed file with 17 additions and 4 deletions.
diff --git a/client/src/server/connection.ts b/client/src/server/connection.ts
@@ -9,16 +9,24 @@ export default class Connection {
         uiData: Record<string, any>,
         dom: typeof Dom
     ) {
-        const component: Element = uiData.element.closest('[ui-state]');
-        const componentData = component.getAttribute('ui-state') ?? '{}';
+        const pageState: Record<string, any> = {};
+        const component: HTMLElement = uiData.element.closest('[ui-state]');
+        const componentData = JSON.parse(component?.getAttribute('ui-state') ?? '{}');
+        const components = document.querySelectorAll('[ui-state]');
+
+        components.forEach((i) => {
+            const attr = JSON.parse(i.getAttribute('ui-state') ?? '{}');
+            pageState[attr.key] = attr;
+        });
 
         const payload = {
             type,
             payload: {
                 params: [],
                 method: uiData.method,
                 methodArgs: uiData.methodArgs,
-                data: componentData,
+                component: componentData?.key,
+                data: pageState,
             }
         };
 
@@ -47,7 +55,12 @@ export default class Connection {
                 response.text().then(response => {
                     const data = JSON.parse(response);
                     window._leafUIConfig.data = data.state;
-                    dom.diff(data.html, document.body!);
+                    dom.diff(
+                        data.html,
+                        component.nodeName === 'HTML' || !component
+                            ? document.body!
+                            : component
+                    );
                 });
             } else {
                 error(await response.text().then(res => res));