cogsbox-state 0.5.476 → 0.5.478

package/src/store.ts

@@ -136,6 +136,7 @@ export type ShadowMetadata = {
   stateSource?: 'default' | 'server' | 'localStorage';
   lastServerSync?: number;
   isDirty?: boolean;
+  isRaw?: boolean;
   baseServerState?: any;
   arrayKeys?: string[];
   fields?: Record<string, any>;
@@ -816,11 +817,24 @@ export const getGlobalStore = create<CogsGlobalState>((set, get) => ({
           newMeta.validation = existingMeta.validation;
         }
 
-        // 3. Preserve component registrations, which only exist on the live target state.
+        // 3. Preserve component registrations
         if (existingMeta.components) {
           newMeta.components = existingMeta.components;
         }
 
+        // 4. Preserve clientActivityState (DOM ref registrations)
+        if (
+          existingMeta.clientActivityState &&
+          !sourceMeta.clientActivityState
+        ) {
+          newMeta.clientActivityState = existingMeta.clientActivityState;
+        }
+
+        // 5. Preserve pluginMetaData
+        if (existingMeta.pluginMetaData && !sourceMeta.pluginMetaData) {
+          newMeta.pluginMetaData = existingMeta.pluginMetaData;
+        }
+
         target._meta = newMeta;
       }
       // --- END: CORRECTED METADATA MERGE ---
@@ -906,6 +920,33 @@ export const getGlobalStore = create<CogsGlobalState>((set, get) => ({
       shadowStateStore.get(key) || shadowStateStore.get(`[${key}`);
     let preservedMetadata: Partial<ShadowMetadata> = {};
 
+    // NEW: Collect runtime metadata (clientActivityState, pluginMetaData) from ALL nodes
+    const preservedRuntimeMeta = new Map<
+      string,
+      {
+        clientActivityState?: ClientActivityState;
+        pluginMetaData?: Map<string, Record<string, any>>;
+      }
+    >();
+
+    if (existingRoot) {
+      const collectRuntimeMeta = (node: any, pathSegments: string[]) => {
+        if (!node || typeof node !== 'object') return;
+        if (node._meta?.clientActivityState || node._meta?.pluginMetaData) {
+          preservedRuntimeMeta.set(pathSegments.join('\x00'), {
+            clientActivityState: node._meta.clientActivityState,
+            pluginMetaData: node._meta.pluginMetaData,
+          });
+        }
+        for (const k in node) {
+          if (k !== '_meta') {
+            collectRuntimeMeta(node[k], [...pathSegments, k]);
+          }
+        }
+      };
+      collectRuntimeMeta(existingRoot, []);
+    }
+
     if (existingRoot?._meta) {
       const {
         components,
@@ -924,7 +965,6 @@ export const getGlobalStore = create<CogsGlobalState>((set, get) => ({
     shadowStateStore.delete(key);
     shadowStateStore.delete(`[${key}`);
 
-    // Get all available schemas for this state
     const options = get().getInitialOptions(key);
     const syncSchemas = get().getInitialOptions('__syncSchemas');
 
@@ -938,12 +978,31 @@ export const getGlobalStore = create<CogsGlobalState>((set, get) => ({
       },
     };
 
-    // Build with context so type info is stored
     const newRoot = buildShadowNode(key, initialState, context);
 
     if (!newRoot._meta) newRoot._meta = {};
     Object.assign(newRoot._meta, preservedMetadata);
 
+    // NEW: Restore runtime metadata to matching nodes in rebuilt tree
+    const restoreRuntimeMeta = (node: any, pathSegments: string[]) => {
+      if (!node || typeof node !== 'object') return;
+      const pathKey = pathSegments.join('\x00');
+      const saved = preservedRuntimeMeta.get(pathKey);
+      if (saved) {
+        if (!node._meta) node._meta = {};
+        if (saved.clientActivityState)
+          node._meta.clientActivityState = saved.clientActivityState;
+        if (saved.pluginMetaData)
+          node._meta.pluginMetaData = saved.pluginMetaData;
+      }
+      for (const k in node) {
+        if (k !== '_meta') {
+          restoreRuntimeMeta(node[k], [...pathSegments, k]);
+        }
+      }
+    };
+    restoreRuntimeMeta(newRoot, []);
+
     const storageKey = Array.isArray(initialState) ? `[${key}` : key;
     shadowStateStore.set(storageKey, newRoot);
   },
@@ -1354,6 +1413,7 @@ export const getGlobalStore = create<CogsGlobalState>((set, get) => ({
     const rootMeta = get().getShadowMetadata(stateKey, []) || {};
     const components = new Map(rootMeta.components);
     components.set(fullComponentId, registration);
+
     get().setShadowMetadata(stateKey, [], { components });
   },
 
@@ -1585,3 +1645,71 @@ export const getGlobalStore = create<CogsGlobalState>((set, get) => ({
     }),
   getSyncInfo: (key) => get().syncInfoStore.get(key) || null,
 }));
+
+export function getAllFieldElements(stateKey: string): HTMLElement[] {
+  const elements: HTMLElement[] = [];
+
+  const rootNode =
+    shadowStateStore.get(stateKey) || shadowStateStore.get(`[${stateKey}`);
+
+  if (!rootNode) return elements;
+
+  const collectElements = (node: any) => {
+    if (!node || typeof node !== 'object') return;
+
+    // Check this node for elements
+    if (node._meta?.clientActivityState?.elements) {
+      node._meta.clientActivityState.elements.forEach((entry: any) => {
+        if (entry.domRef?.current) {
+          elements.push(entry.domRef.current);
+        }
+      });
+    }
+
+    // Recurse into children
+    for (const key in node) {
+      if (key !== '_meta') {
+        collectElements(node[key]);
+      }
+    }
+  };
+
+  collectElements(rootNode);
+
+  return elements;
+}
+
+export function setAllFieldsDisabled(
+  stateKey: string,
+  disabled: boolean
+): void {
+  const rootNode =
+    shadowStateStore.get(stateKey) || shadowStateStore.get(`[${stateKey}`);
+  if (!rootNode) return;
+
+  const disableElements = (node: any) => {
+    if (!node || typeof node !== 'object') return;
+
+    if (node._meta?.clientActivityState?.elements) {
+      node._meta.clientActivityState.elements.forEach((entry: any) => {
+        const el = entry.domRef?.current;
+        if (!el) return;
+
+        if ('disabled' in el) {
+          (el as HTMLInputElement).disabled = disabled;
+        } else {
+          el.style.pointerEvents = disabled ? 'none' : '';
+          el.setAttribute('aria-disabled', String(disabled));
+        }
+      });
+    }
+
+    for (const key in node) {
+      if (key !== '_meta') {
+        disableElements(node[key]);
+      }
+    }
+  };
+
+  disableElements(rootNode);
+}

package/src/utility.ts

@@ -1,5 +1,3 @@
-import type { TransformedStateType, CogsInitialState } from './CogsState';
-
 export const isObject = (item: any): item is Record<string, any> => {
   return (
     item && typeof item === 'object' && !Array.isArray(item) && item !== null
@@ -346,35 +344,5 @@ export type DebouncedFunction<F extends (...args: any[]) => any> = F & {
 };
 
 export function transformStateFunc<State extends unknown>(initialState: State) {
-  const isInitialStateType = (state: any): state is CogsInitialState<State> => {
-    return Object.values(state).some((value) =>
-      value?.hasOwnProperty('initialState')
-    );
-  };
-  let initalOptions: GenericObject = {};
-  const transformInitialState = (
-    state: CogsInitialState<State>
-  ): GenericObject | GenericObject[] => {
-    const transformedState: GenericObject | GenericObject[] = {};
-    Object.entries(state).forEach(([key, value]) => {
-      if (value?.initialState) {
-        initalOptions = { ...(initalOptions ?? {}), [key]: value };
-
-        transformedState[key] = value.initialState;
-      } else {
-        transformedState[key] = value;
-      }
-    });
-
-    return transformedState;
-  };
-
-  const transformedInitialState = isInitialStateType(initialState)
-    ? (transformInitialState(initialState) as State)
-    : (initialState as State);
-
-  return [transformedInitialState, initalOptions] as [
-    TransformedStateType<State>,
-    GenericObject,
-  ];
+  return [initialState as State, {}] as [State, GenericObject];
 }