@assistant-ui/store 0.0.2 → 0.0.4

package/dist/useAssistantClient.js

@@ -1,187 +1,227 @@
+"use client";
+
 // src/useAssistantClient.tsx
-import { useMemo } from "react";
 import { useResource } from "@assistant-ui/tap/react";
 import {
   resource,
   tapMemo,
-  tapResource,
   tapResources,
   tapEffectEvent,
-  tapInlineResource
+  tapInlineResource,
+  tapEffect,
+  tapRef,
+  tapResource
 } from "@assistant-ui/tap";
-import { asStore } from "./asStore.js";
-import { useAssistantContextValue } from "./AssistantContext.js";
-import { splitScopes } from "./utils/splitScopes.js";
+import { StoreResource } from "./utils/StoreResource.js";
+import {
+  useAssistantContextValue,
+  DefaultAssistantClient,
+  createRootAssistantClient
+} from "./utils/react-assistant-context.js";
+import {
+  splitClients
+} from "./utils/splitClients.js";
 import {
-  EventManager,
   normalizeEventSelector
-} from "./EventContext.js";
-import { withStoreContextProvider } from "./StoreContext.js";
-var RootScopeStoreResource = resource(
+} from "./types/events.js";
+import { NotificationManager } from "./utils/NotificationManager.js";
+import { withAssistantTapContextProvider } from "./utils/tap-assistant-context.js";
+import { tapClientResource } from "./tapClientResource.js";
+import { getClientIndex } from "./utils/tap-client-stack-context.js";
+import {
+  PROXIED_ASSISTANT_STATE_SYMBOL,
+  createProxiedAssistantState
+} from "./utils/proxied-assistant-state.js";
+var RootClientResource = resource(
   ({
     element,
-    events,
-    parent
+    emit,
+    clientRef
   }) => {
-    return withStoreContextProvider(
-      { events, parent },
-      () => tapInlineResource(element)
+    const { methods, state } = withAssistantTapContextProvider(
+      { clientRef, emit },
+      () => tapClientResource(element)
     );
+    return tapMemo(() => ({ methods }), [state]);
   }
 );
-var RootScopeResource = resource(
+var RootClientAccessorResource = resource(
   ({
-    scopeName,
     element,
-    events,
-    parent
+    notifications,
+    clientRef
   }) => {
-    const store = tapResource(
-      asStore(RootScopeStoreResource({ element, events, parent }))
+    const store = tapInlineResource(
+      StoreResource(
+        RootClientResource({ element, emit: notifications.emit, clientRef })
+      )
     );
+    tapEffect(() => {
+      return store.subscribe(notifications.notifySubscribers);
+    }, [store, notifications]);
     return tapMemo(() => {
-      const scopeFunction = (() => store.getState().api);
-      scopeFunction.source = "root";
-      scopeFunction.query = {};
-      return [
-        scopeName,
-        {
-          scopeFunction,
-          subscribe: store.subscribe,
-          flushSync: store.flushSync
-        }
-      ];
-    }, [scopeName, store]);
+      const clientFunction = () => store.getState().methods;
+      clientFunction.source = "root";
+      clientFunction.query = {};
+      return clientFunction;
+    }, [store]);
   }
 );
-var RootScopesResource = resource(
-  ({ scopes, parent }) => {
-    const events = tapInlineResource(EventManager());
-    const resultEntries = tapResources(
-      Object.entries(scopes).map(
-        ([scopeName, element]) => RootScopeResource(
-          {
-            scopeName,
-            element,
-            events,
-            parent
-          },
-          { key: scopeName }
-        )
-      )
+var NoOpRootClientsAccessorsResource = resource(() => {
+  return tapMemo(
+    () => ({ clients: {}, subscribe: void 0, on: void 0 }),
+    []
+  );
+});
+var RootClientsAccessorsResource = resource(
+  ({
+    clients: inputClients,
+    clientRef
+  }) => {
+    const notifications = tapInlineResource(NotificationManager());
+    tapEffect(
+      () => clientRef.parent.subscribe(notifications.notifySubscribers),
+      [clientRef, notifications]
+    );
+    const results = tapResources(
+      inputClients,
+      (element) => RootClientAccessorResource({
+        element,
+        notifications,
+        clientRef
+      }),
+      [notifications, clientRef]
     );
-    const on = (selector, callback) => {
-      const { event } = normalizeEventSelector(selector);
-      return events.on(event, callback);
-    };
     return tapMemo(() => {
-      if (resultEntries.length === 0) {
-        return {
-          scopes: {},
-          on
-        };
-      }
       return {
-        scopes: Object.fromEntries(
-          resultEntries.map(([scopeName, { scopeFunction }]) => [
-            scopeName,
-            scopeFunction
-          ])
-        ),
-        subscribe: (callback) => {
-          const unsubscribes = resultEntries.map(([, { subscribe }]) => {
-            return subscribe(() => {
-              console.log("Callback called for");
-              callback();
-            });
+        clients: results,
+        subscribe: notifications.subscribe,
+        on: function(selector, callback) {
+          if (!this) {
+            throw new Error(
+              "const { on } = useAssistantClient() is not supported. Use aui.on() instead."
+            );
+          }
+          const { scope, event } = normalizeEventSelector(selector);
+          if (scope !== "*") {
+            const source = this[scope].source;
+            if (source === null) {
+              throw new Error(
+                `Scope "${scope}" is not available. Use { scope: "*", event: "${event}" } to listen globally.`
+              );
+            }
+          }
+          const localUnsub = notifications.on(event, (payload, clientStack) => {
+            if (scope === "*") {
+              callback(payload);
+              return;
+            }
+            const scopeClient = this[scope]();
+            const index = getClientIndex(scopeClient);
+            if (scopeClient === clientStack[index]) {
+              callback(payload);
+            }
           });
+          if (scope !== "*" && clientRef.parent[scope].source === null)
+            return localUnsub;
+          const parentUnsub = clientRef.parent.on(selector, callback);
           return () => {
-            unsubscribes.forEach((unsubscribe) => unsubscribe());
+            localUnsub();
+            parentUnsub();
           };
-        },
-        flushSync: () => {
-          resultEntries.forEach(([, { flushSync }]) => {
-            flushSync();
-          });
-        },
-        on
+        }
       };
-    }, [...resultEntries, events]);
+    }, [results, notifications, clientRef]);
   }
 );
-var useRootScopes = (rootScopes, parent) => {
-  return useResource(RootScopesResource({ scopes: rootScopes, parent }));
+var getMeta = (props, clientRef, memo) => {
+  if ("source" in props && "query" in props) return props;
+  if (memo.dep === props) return memo.meta;
+  const meta = props.getMeta(clientRef.current);
+  memo.meta = meta;
+  memo.dep = props;
+  return meta;
 };
-var DerivedScopeResource = resource(
+var DerivedClientAccessorResource = resource(
   ({
-    scopeName,
     element,
-    parentClient
+    clientRef
   }) => {
-    const get = tapEffectEvent(element.props.get);
-    const source = element.props.source;
-    const query = element.props.query;
+    const get = tapEffectEvent(() => element.props);
     return tapMemo(() => {
-      const scopeFunction = (() => get(parentClient));
-      scopeFunction.source = source;
-      scopeFunction.query = query;
-      return [scopeName, scopeFunction];
-    }, [scopeName, get, source, JSON.stringify(query), parentClient]);
+      const clientFunction = () => get().get(clientRef.current);
+      const metaMemo = {};
+      Object.defineProperties(clientFunction, {
+        source: {
+          get: () => getMeta(get(), clientRef, metaMemo).source
+        },
+        query: {
+          get: () => getMeta(get(), clientRef, metaMemo).query
+        }
+      });
+      return clientFunction;
+    }, [clientRef]);
   }
 );
-var DerivedScopesResource = resource(
+var DerivedClientsAccessorsResource = resource(
   ({
-    scopes,
-    parentClient
+    clients,
+    clientRef
   }) => {
-    const resultEntries = tapResources(
-      Object.entries(scopes).map(
-        ([scopeName, element]) => DerivedScopeResource(
-          {
-            scopeName,
-            element,
-            parentClient
-          },
-          { key: scopeName }
-        )
-      )
+    return tapResources(
+      clients,
+      (element) => DerivedClientAccessorResource({
+        element,
+        clientRef
+      }),
+      [clientRef]
     );
-    return tapMemo(() => {
-      return Object.fromEntries(resultEntries);
-    }, [...resultEntries]);
   }
 );
-var useDerivedScopes = (derivedScopes, parentClient) => {
-  return useResource(
-    DerivedScopesResource({ scopes: derivedScopes, parentClient })
-  );
-};
-var useExtendedAssistantClientImpl = (scopes) => {
+var AssistantClientResource = resource(
+  ({
+    baseClient,
+    clients
+  }) => {
+    const { rootClients, derivedClients } = splitClients(clients, baseClient);
+    const clientRef = tapRef({
+      parent: baseClient,
+      current: null
+    }).current;
+    const rootFields = tapResource(
+      Object.keys(rootClients).length > 0 ? RootClientsAccessorsResource({ clients: rootClients, clientRef }) : NoOpRootClientsAccessorsResource()
+    );
+    const derivedFields = tapInlineResource(
+      DerivedClientsAccessorsResource({ clients: derivedClients, clientRef })
+    );
+    const client = tapMemo(() => {
+      const proto = baseClient === DefaultAssistantClient ? createRootAssistantClient() : baseClient;
+      const client2 = Object.create(proto);
+      Object.assign(client2, rootFields.clients, derivedFields, {
+        subscribe: rootFields.subscribe ?? baseClient.subscribe,
+        on: rootFields.on ?? baseClient.on,
+        [PROXIED_ASSISTANT_STATE_SYMBOL]: createProxiedAssistantState(client2)
+      });
+      return client2;
+    }, [baseClient, rootFields, derivedFields]);
+    if (clientRef.current === null) {
+      clientRef.current = client;
+    }
+    tapEffect(() => {
+      clientRef.current = client;
+    });
+    return client;
+  }
+);
+function useAssistantClient(clients) {
   const baseClient = useAssistantContextValue();
-  const { rootScopes, derivedScopes } = splitScopes(scopes);
-  const rootFields = useRootScopes(rootScopes, baseClient);
-  const derivedFields = useDerivedScopes(derivedScopes, baseClient);
-  return useMemo(() => {
-    return {
-      ...baseClient,
-      ...rootFields.scopes,
-      ...derivedFields,
-      subscribe: rootFields.subscribe ?? baseClient.subscribe,
-      flushSync: rootFields.flushSync ?? baseClient.flushSync,
-      on: rootFields.on ?? baseClient.on
-    };
-  }, [baseClient, rootFields, derivedFields]);
-};
-function useAssistantClient(scopes) {
-  if (scopes) {
-    return useExtendedAssistantClientImpl(scopes);
-  } else {
-    return useAssistantContextValue();
+  if (clients) {
+    return useResource(AssistantClientResource({ baseClient, clients }));
   }
+  return baseClient;
 }
 export {
-  useAssistantClient,
-  useDerivedScopes,
-  useRootScopes
+  AssistantClientResource,
+  useAssistantClient
 };
 //# sourceMappingURL=useAssistantClient.js.map

package/dist/useAssistantClient.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../src/useAssistantClient.tsx"],"sourcesContent":["import { useMemo } from \"react\";\nimport { useResource } from \"@assistant-ui/tap/react\";\nimport {\n  resource,\n  tapMemo,\n  tapResource,\n  tapResources,\n  tapEffectEvent,\n  tapInlineResource,\n  ResourceElement,\n} from \"@assistant-ui/tap\";\nimport type {\n  AssistantClient,\n  AssistantScopes,\n  ScopesInput,\n  ScopeField,\n  ScopeInput,\n  DerivedScopeProps,\n} from \"./types\";\nimport { asStore } from \"./asStore\";\nimport { useAssistantContextValue } from \"./AssistantContext\";\nimport { splitScopes } from \"./utils/splitScopes\";\nimport {\n  EventManager,\n  normalizeEventSelector,\n  type AssistantEvent,\n  type AssistantEventCallback,\n  type AssistantEventSelector,\n} from \"./EventContext\";\nimport { withStoreContextProvider } from \"./StoreContext\";\n\n/**\n * Resource that renders a store with the store context provider.\n * This ensures the context is re-established on every re-render.\n */\nconst RootScopeStoreResource = resource(\n  <K extends keyof AssistantScopes>({\n    element,\n    events,\n    parent,\n  }: {\n    element: ScopeInput<AssistantScopes[K]>;\n    events: EventManager;\n    parent: AssistantClient;\n  }) => {\n    return withStoreContextProvider({ events, parent }, () =>\n      tapInlineResource(element),\n    );\n  },\n);\n\n/**\n * Resource for a single root scope\n * Returns a tuple of [scopeName, {scopeFunction, subscribe, flushSync}]\n */\nconst RootScopeResource = resource(\n  <K extends keyof AssistantScopes>({\n    scopeName,\n    element,\n    events,\n    parent,\n  }: {\n    scopeName: K;\n    element: ScopeInput<AssistantScopes[K]>;\n    events: EventManager;\n    parent: AssistantClient;\n  }) => {\n    const store = tapResource(\n      asStore(RootScopeStoreResource({ element, events, parent })),\n    );\n\n    return tapMemo(() => {\n      const scopeFunction = (() => store.getState().api) as ScopeField<\n        AssistantScopes[K]\n      >;\n      scopeFunction.source = \"root\";\n      scopeFunction.query = {};\n\n      return [\n        scopeName,\n        {\n          scopeFunction,\n          subscribe: store.subscribe,\n          flushSync: store.flushSync,\n        },\n      ] as const;\n    }, [scopeName, store]);\n  },\n);\n\n/**\n * Resource for all root scopes\n * Mounts each root scope and returns an object mapping scope names to their stores\n */\nconst RootScopesResource = resource(\n  ({ scopes, parent }: { scopes: ScopesInput; parent: AssistantClient }) => {\n    const events = tapInlineResource(EventManager());\n\n    const resultEntries = tapResources(\n      Object.entries(scopes).map(([scopeName, element]) =>\n        RootScopeResource(\n          {\n            scopeName: scopeName as keyof AssistantScopes,\n            element: element as ScopeInput<\n              AssistantScopes[keyof AssistantScopes]\n            >,\n            events,\n            parent,\n          },\n          { key: scopeName },\n        ),\n      ),\n    );\n\n    const on = <TEvent extends AssistantEvent>(\n      selector: AssistantEventSelector<TEvent>,\n      callback: AssistantEventCallback<TEvent>,\n    ) => {\n      const { event } = normalizeEventSelector(selector);\n      return events.on(event, callback);\n    };\n\n    return tapMemo(() => {\n      if (resultEntries.length === 0) {\n        return {\n          scopes: {},\n          on,\n        };\n      }\n\n      return {\n        scopes: Object.fromEntries(\n          resultEntries.map(([scopeName, { scopeFunction }]) => [\n            scopeName,\n            scopeFunction,\n          ]),\n        ) as {\n          [K in keyof typeof scopes]: ScopeField<AssistantScopes[K]>;\n        },\n        subscribe: (callback: () => void) => {\n          const unsubscribes = resultEntries.map(([, { subscribe }]) => {\n            return subscribe(() => {\n              console.log(\"Callback called for\");\n              callback();\n            });\n          });\n          return () => {\n            unsubscribes.forEach((unsubscribe) => unsubscribe());\n          };\n        },\n        flushSync: () => {\n          resultEntries.forEach(([, { flushSync }]) => {\n            flushSync();\n          });\n        },\n        on,\n      };\n    }, [...resultEntries, events]);\n  },\n);\n\n/**\n * Hook to mount and access root scopes\n */\nexport const useRootScopes = (\n  rootScopes: ScopesInput,\n  parent: AssistantClient,\n) => {\n  return useResource(RootScopesResource({ scopes: rootScopes, parent }));\n};\n\n/**\n * Resource for a single derived scope\n * Returns a tuple of [scopeName, scopeFunction] where scopeFunction has source and query\n */\nconst DerivedScopeResource = resource(\n  <K extends keyof AssistantScopes>({\n    scopeName,\n    element,\n    parentClient,\n  }: {\n    scopeName: K;\n    element: ResourceElement<\n      AssistantScopes[K],\n      DerivedScopeProps<AssistantScopes[K]>\n    >;\n    parentClient: AssistantClient;\n  }) => {\n    const get = tapEffectEvent(element.props.get);\n    const source = element.props.source;\n    const query = element.props.query;\n    return tapMemo(() => {\n      const scopeFunction = (() => get(parentClient)) as ScopeField<\n        AssistantScopes[K]\n      >;\n      scopeFunction.source = source;\n      scopeFunction.query = query;\n\n      return [scopeName, scopeFunction] as const;\n    }, [scopeName, get, source, JSON.stringify(query), parentClient]);\n  },\n);\n\n/**\n * Resource for all derived scopes\n * Builds stable scope functions with source and query metadata\n */\nconst DerivedScopesResource = resource(\n  ({\n    scopes,\n    parentClient,\n  }: {\n    scopes: ScopesInput;\n    parentClient: AssistantClient;\n  }) => {\n    const resultEntries = tapResources(\n      Object.entries(scopes).map(([scopeName, element]) =>\n        DerivedScopeResource(\n          {\n            scopeName: scopeName as keyof AssistantScopes,\n            element: element as ScopeInput<\n              AssistantScopes[keyof AssistantScopes]\n            >,\n            parentClient,\n          },\n          { key: scopeName },\n        ),\n      ),\n    );\n\n    return tapMemo(() => {\n      return Object.fromEntries(resultEntries) as {\n        [K in keyof typeof scopes]: ScopeField<AssistantScopes[K]>;\n      };\n    }, [...resultEntries]);\n  },\n);\n\n/**\n * Hook to mount and access derived scopes\n */\nexport const useDerivedScopes = (\n  derivedScopes: ScopesInput,\n  parentClient: AssistantClient,\n) => {\n  return useResource(\n    DerivedScopesResource({ scopes: derivedScopes, parentClient }),\n  );\n};\n\nconst useExtendedAssistantClientImpl = (\n  scopes: ScopesInput,\n): AssistantClient => {\n  const baseClient = useAssistantContextValue();\n  const { rootScopes, derivedScopes } = splitScopes(scopes);\n\n  // Mount the scopes to keep them alive\n  const rootFields = useRootScopes(rootScopes, baseClient);\n  const derivedFields = useDerivedScopes(derivedScopes, baseClient);\n\n  return useMemo(() => {\n    // Merge base client with extended client\n    // If baseClient is the default proxy, spreading it will be a no-op\n    return {\n      ...baseClient,\n      ...rootFields.scopes,\n      ...derivedFields,\n      subscribe: rootFields.subscribe ?? baseClient.subscribe,\n      flushSync: rootFields.flushSync ?? baseClient.flushSync,\n      on: rootFields.on ?? baseClient.on,\n    } as AssistantClient;\n  }, [baseClient, rootFields, derivedFields]);\n};\n\n/**\n * Hook to access or extend the AssistantClient\n *\n * @example Without config - returns the client from context:\n * ```typescript\n * const client = useAssistantClient();\n * const fooState = client.foo.getState();\n * ```\n *\n * @example With config - creates a new client with additional scopes:\n * ```typescript\n * const client = useAssistantClient({\n *   message: DerivedScope({\n *     source: \"thread\",\n *     query: { type: \"index\", index: 0 },\n *     get: () => messageApi,\n *   }),\n * });\n * ```\n */\nexport function useAssistantClient(): AssistantClient;\nexport function useAssistantClient(scopes: ScopesInput): AssistantClient;\nexport function useAssistantClient(scopes?: ScopesInput): AssistantClient {\n  if (scopes) {\n    return useExtendedAssistantClientImpl(scopes);\n  } else {\n    return useAssistantContextValue();\n  }\n}\n"],"mappings":";AAAA,SAAS,eAAe;AACxB,SAAS,mBAAmB;AAC5B;AAAA,EACE;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,OAEK;AASP,SAAS,eAAe;AACxB,SAAS,gCAAgC;AACzC,SAAS,mBAAmB;AAC5B;AAAA,EACE;AAAA,EACA;AAAA,OAIK;AACP,SAAS,gCAAgC;AAMzC,IAAM,yBAAyB;AAAA,EAC7B,CAAkC;AAAA,IAChC;AAAA,IACA;AAAA,IACA;AAAA,EACF,MAIM;AACJ,WAAO;AAAA,MAAyB,EAAE,QAAQ,OAAO;AAAA,MAAG,MAClD,kBAAkB,OAAO;AAAA,IAC3B;AAAA,EACF;AACF;AAMA,IAAM,oBAAoB;AAAA,EACxB,CAAkC;AAAA,IAChC;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,EACF,MAKM;AACJ,UAAM,QAAQ;AAAA,MACZ,QAAQ,uBAAuB,EAAE,SAAS,QAAQ,OAAO,CAAC,CAAC;AAAA,IAC7D;AAEA,WAAO,QAAQ,MAAM;AACnB,YAAM,iBAAiB,MAAM,MAAM,SAAS,EAAE;AAG9C,oBAAc,SAAS;AACvB,oBAAc,QAAQ,CAAC;AAEvB,aAAO;AAAA,QACL;AAAA,QACA;AAAA,UACE;AAAA,UACA,WAAW,MAAM;AAAA,UACjB,WAAW,MAAM;AAAA,QACnB;AAAA,MACF;AAAA,IACF,GAAG,CAAC,WAAW,KAAK,CAAC;AAAA,EACvB;AACF;AAMA,IAAM,qBAAqB;AAAA,EACzB,CAAC,EAAE,QAAQ,OAAO,MAAwD;AACxE,UAAM,SAAS,kBAAkB,aAAa,CAAC;AAE/C,UAAM,gBAAgB;AAAA,MACpB,OAAO,QAAQ,MAAM,EAAE;AAAA,QAAI,CAAC,CAAC,WAAW,OAAO,MAC7C;AAAA,UACE;AAAA,YACE;AAAA,YACA;AAAA,YAGA;AAAA,YACA;AAAA,UACF;AAAA,UACA,EAAE,KAAK,UAAU;AAAA,QACnB;AAAA,MACF;AAAA,IACF;AAEA,UAAM,KAAK,CACT,UACA,aACG;AACH,YAAM,EAAE,MAAM,IAAI,uBAAuB,QAAQ;AACjD,aAAO,OAAO,GAAG,OAAO,QAAQ;AAAA,IAClC;AAEA,WAAO,QAAQ,MAAM;AACnB,UAAI,cAAc,WAAW,GAAG;AAC9B,eAAO;AAAA,UACL,QAAQ,CAAC;AAAA,UACT;AAAA,QACF;AAAA,MACF;AAEA,aAAO;AAAA,QACL,QAAQ,OAAO;AAAA,UACb,cAAc,IAAI,CAAC,CAAC,WAAW,EAAE,cAAc,CAAC,MAAM;AAAA,YACpD;AAAA,YACA;AAAA,UACF,CAAC;AAAA,QACH;AAAA,QAGA,WAAW,CAAC,aAAyB;AACnC,gBAAM,eAAe,cAAc,IAAI,CAAC,CAAC,EAAE,EAAE,UAAU,CAAC,MAAM;AAC5D,mBAAO,UAAU,MAAM;AACrB,sBAAQ,IAAI,qBAAqB;AACjC,uBAAS;AAAA,YACX,CAAC;AAAA,UACH,CAAC;AACD,iBAAO,MAAM;AACX,yBAAa,QAAQ,CAAC,gBAAgB,YAAY,CAAC;AAAA,UACrD;AAAA,QACF;AAAA,QACA,WAAW,MAAM;AACf,wBAAc,QAAQ,CAAC,CAAC,EAAE,EAAE,UAAU,CAAC,MAAM;AAC3C,sBAAU;AAAA,UACZ,CAAC;AAAA,QACH;AAAA,QACA;AAAA,MACF;AAAA,IACF,GAAG,CAAC,GAAG,eAAe,MAAM,CAAC;AAAA,EAC/B;AACF;AAKO,IAAM,gBAAgB,CAC3B,YACA,WACG;AACH,SAAO,YAAY,mBAAmB,EAAE,QAAQ,YAAY,OAAO,CAAC,CAAC;AACvE;AAMA,IAAM,uBAAuB;AAAA,EAC3B,CAAkC;AAAA,IAChC;AAAA,IACA;AAAA,IACA;AAAA,EACF,MAOM;AACJ,UAAM,MAAM,eAAe,QAAQ,MAAM,GAAG;AAC5C,UAAM,SAAS,QAAQ,MAAM;AAC7B,UAAM,QAAQ,QAAQ,MAAM;AAC5B,WAAO,QAAQ,MAAM;AACnB,YAAM,iBAAiB,MAAM,IAAI,YAAY;AAG7C,oBAAc,SAAS;AACvB,oBAAc,QAAQ;AAEtB,aAAO,CAAC,WAAW,aAAa;AAAA,IAClC,GAAG,CAAC,WAAW,KAAK,QAAQ,KAAK,UAAU,KAAK,GAAG,YAAY,CAAC;AAAA,EAClE;AACF;AAMA,IAAM,wBAAwB;AAAA,EAC5B,CAAC;AAAA,IACC;AAAA,IACA;AAAA,EACF,MAGM;AACJ,UAAM,gBAAgB;AAAA,MACpB,OAAO,QAAQ,MAAM,EAAE;AAAA,QAAI,CAAC,CAAC,WAAW,OAAO,MAC7C;AAAA,UACE;AAAA,YACE;AAAA,YACA;AAAA,YAGA;AAAA,UACF;AAAA,UACA,EAAE,KAAK,UAAU;AAAA,QACnB;AAAA,MACF;AAAA,IACF;AAEA,WAAO,QAAQ,MAAM;AACnB,aAAO,OAAO,YAAY,aAAa;AAAA,IAGzC,GAAG,CAAC,GAAG,aAAa,CAAC;AAAA,EACvB;AACF;AAKO,IAAM,mBAAmB,CAC9B,eACA,iBACG;AACH,SAAO;AAAA,IACL,sBAAsB,EAAE,QAAQ,eAAe,aAAa,CAAC;AAAA,EAC/D;AACF;AAEA,IAAM,iCAAiC,CACrC,WACoB;AACpB,QAAM,aAAa,yBAAyB;AAC5C,QAAM,EAAE,YAAY,cAAc,IAAI,YAAY,MAAM;AAGxD,QAAM,aAAa,cAAc,YAAY,UAAU;AACvD,QAAM,gBAAgB,iBAAiB,eAAe,UAAU;AAEhE,SAAO,QAAQ,MAAM;AAGnB,WAAO;AAAA,MACL,GAAG;AAAA,MACH,GAAG,WAAW;AAAA,MACd,GAAG;AAAA,MACH,WAAW,WAAW,aAAa,WAAW;AAAA,MAC9C,WAAW,WAAW,aAAa,WAAW;AAAA,MAC9C,IAAI,WAAW,MAAM,WAAW;AAAA,IAClC;AAAA,EACF,GAAG,CAAC,YAAY,YAAY,aAAa,CAAC;AAC5C;AAwBO,SAAS,mBAAmB,QAAuC;AACxE,MAAI,QAAQ;AACV,WAAO,+BAA+B,MAAM;AAAA,EAC9C,OAAO;AACL,WAAO,yBAAyB;AAAA,EAClC;AACF;","names":[]}
+{"version":3,"sources":["../src/useAssistantClient.tsx"],"sourcesContent":["\"use client\";\n\nimport { useResource } from \"@assistant-ui/tap/react\";\nimport {\n  resource,\n  tapMemo,\n  tapResources,\n  tapEffectEvent,\n  tapInlineResource,\n  tapEffect,\n  tapRef,\n  tapResource,\n} from \"@assistant-ui/tap\";\nimport type {\n  AssistantClient,\n  AssistantClientAccessor,\n  ClientNames,\n  ClientElement,\n  ClientMeta,\n} from \"./types/client\";\nimport { Derived, DerivedElement } from \"./Derived\";\nimport { StoreResource } from \"./utils/StoreResource\";\nimport {\n  useAssistantContextValue,\n  DefaultAssistantClient,\n  createRootAssistantClient,\n} from \"./utils/react-assistant-context\";\nimport {\n  DerivedClients,\n  RootClients,\n  splitClients,\n} from \"./utils/splitClients\";\nimport {\n  normalizeEventSelector,\n  type AssistantEventName,\n  type AssistantEventCallback,\n  type AssistantEventSelector,\n} from \"./types/events\";\nimport { NotificationManager } from \"./utils/NotificationManager\";\nimport { withAssistantTapContextProvider } from \"./utils/tap-assistant-context\";\nimport { tapClientResource } from \"./tapClientResource\";\nimport { getClientIndex } from \"./utils/tap-client-stack-context\";\nimport {\n  PROXIED_ASSISTANT_STATE_SYMBOL,\n  createProxiedAssistantState,\n} from \"./utils/proxied-assistant-state\";\n\nconst RootClientResource = resource(\n  <K extends ClientNames>({\n    element,\n    emit,\n    clientRef,\n  }: {\n    element: ClientElement<K>;\n    emit: NotificationManager[\"emit\"];\n    clientRef: { parent: AssistantClient; current: AssistantClient | null };\n  }) => {\n    const { methods, state } = withAssistantTapContextProvider(\n      { clientRef, emit },\n      () => tapClientResource(element),\n    );\n    return tapMemo(() => ({ methods }), [state]);\n  },\n);\n\nconst RootClientAccessorResource = resource(\n  <K extends ClientNames>({\n    element,\n    notifications,\n    clientRef,\n  }: {\n    element: ClientElement<K>;\n    notifications: NotificationManager;\n    clientRef: { parent: AssistantClient; current: AssistantClient | null };\n  }): AssistantClientAccessor<K> => {\n    const store = tapInlineResource(\n      StoreResource(\n        RootClientResource({ element, emit: notifications.emit, clientRef }),\n      ),\n    );\n\n    tapEffect(() => {\n      return store.subscribe(notifications.notifySubscribers);\n    }, [store, notifications]);\n\n    return tapMemo(() => {\n      const clientFunction = () => store.getState().methods;\n      clientFunction.source = \"root\" as const;\n      clientFunction.query = {};\n      return clientFunction;\n    }, [store]);\n  },\n);\n\nconst NoOpRootClientsAccessorsResource = resource(() => {\n  return tapMemo(\n    () => ({ clients: {}, subscribe: undefined, on: undefined }),\n    [],\n  );\n});\n\nconst RootClientsAccessorsResource = resource(\n  ({\n    clients: inputClients,\n    clientRef,\n  }: {\n    clients: RootClients;\n    clientRef: { parent: AssistantClient; current: AssistantClient | null };\n  }) => {\n    const notifications = tapInlineResource(NotificationManager());\n\n    tapEffect(\n      () => clientRef.parent.subscribe(notifications.notifySubscribers),\n      [clientRef, notifications],\n    );\n\n    const results = tapResources(\n      inputClients,\n      (element) =>\n        RootClientAccessorResource({\n          element: element!,\n          notifications,\n          clientRef,\n        }),\n      [notifications, clientRef],\n    );\n\n    return tapMemo(() => {\n      return {\n        clients: results,\n        subscribe: notifications.subscribe,\n        on: function <TEvent extends AssistantEventName>(\n          this: AssistantClient,\n          selector: AssistantEventSelector<TEvent>,\n          callback: AssistantEventCallback<TEvent>,\n        ) {\n          if (!this) {\n            throw new Error(\n              \"const { on } = useAssistantClient() is not supported. Use aui.on() instead.\",\n            );\n          }\n\n          const { scope, event } = normalizeEventSelector(selector);\n\n          if (scope !== \"*\") {\n            const source = this[scope as ClientNames].source;\n            if (source === null) {\n              throw new Error(\n                `Scope \"${scope}\" is not available. Use { scope: \"*\", event: \"${event}\" } to listen globally.`,\n              );\n            }\n          }\n\n          const localUnsub = notifications.on(event, (payload, clientStack) => {\n            if (scope === \"*\") {\n              callback(payload);\n              return;\n            }\n\n            const scopeClient = this[scope as ClientNames]();\n            const index = getClientIndex(scopeClient);\n            if (scopeClient === clientStack[index]) {\n              callback(payload);\n            }\n          });\n          if (\n            scope !== \"*\" &&\n            clientRef.parent[scope as ClientNames].source === null\n          )\n            return localUnsub;\n\n          const parentUnsub = clientRef.parent.on(selector, callback);\n\n          return () => {\n            localUnsub();\n            parentUnsub();\n          };\n        },\n      };\n    }, [results, notifications, clientRef]);\n  },\n);\n\ntype MetaMemo<K extends ClientNames> = {\n  meta?: ClientMeta<K>;\n  dep?: unknown;\n};\n\nconst getMeta = <K extends ClientNames>(\n  props: Derived.Props<K>,\n  clientRef: { parent: AssistantClient; current: AssistantClient | null },\n  memo: MetaMemo<K>,\n): ClientMeta<K> => {\n  if (\"source\" in props && \"query\" in props) return props;\n  if (memo.dep === props) return memo.meta!;\n  const meta = props.getMeta(clientRef.current!);\n  memo.meta = meta;\n  memo.dep = props;\n  return meta;\n};\n\nconst DerivedClientAccessorResource = resource(\n  <K extends ClientNames>({\n    element,\n    clientRef,\n  }: {\n    element: DerivedElement<K>;\n    clientRef: { parent: AssistantClient; current: AssistantClient | null };\n  }) => {\n    const get = tapEffectEvent(() => element.props);\n\n    return tapMemo(() => {\n      const clientFunction = () => get().get(clientRef.current!);\n      const metaMemo = {};\n      Object.defineProperties(clientFunction, {\n        source: {\n          get: () => getMeta(get(), clientRef, metaMemo).source,\n        },\n        query: {\n          get: () => getMeta(get(), clientRef, metaMemo).query,\n        },\n      });\n      return clientFunction;\n    }, [clientRef]);\n  },\n);\n\nconst DerivedClientsAccessorsResource = resource(\n  ({\n    clients,\n    clientRef,\n  }: {\n    clients: DerivedClients;\n    clientRef: { parent: AssistantClient; current: AssistantClient | null };\n  }) => {\n    return tapResources(\n      clients,\n      (element) =>\n        DerivedClientAccessorResource({\n          element: element!,\n          clientRef,\n        }),\n      [clientRef],\n    );\n  },\n);\n\n/**\n * Resource that creates an extended AssistantClient.\n */\nexport const AssistantClientResource = resource(\n  ({\n    baseClient,\n    clients,\n  }: {\n    baseClient: AssistantClient;\n    clients: useAssistantClient.Props;\n  }): AssistantClient => {\n    const { rootClients, derivedClients } = splitClients(clients, baseClient);\n\n    const clientRef = tapRef({\n      parent: baseClient,\n      current: null as AssistantClient | null,\n    }).current;\n\n    const rootFields = tapResource(\n      Object.keys(rootClients).length > 0\n        ? RootClientsAccessorsResource({ clients: rootClients, clientRef })\n        : NoOpRootClientsAccessorsResource(),\n    );\n\n    const derivedFields = tapInlineResource(\n      DerivedClientsAccessorsResource({ clients: derivedClients, clientRef }),\n    );\n\n    const client = tapMemo(() => {\n      // Swap DefaultAssistantClient -> createRootAssistantClient at root to change error message\n      const proto =\n        baseClient === DefaultAssistantClient\n          ? createRootAssistantClient()\n          : baseClient;\n      const client = Object.create(proto) as AssistantClient;\n      Object.assign(client, rootFields.clients, derivedFields, {\n        subscribe: rootFields.subscribe ?? baseClient.subscribe,\n        on: rootFields.on ?? baseClient.on,\n        [PROXIED_ASSISTANT_STATE_SYMBOL]: createProxiedAssistantState(client),\n      });\n      return client;\n    }, [baseClient, rootFields, derivedFields]);\n\n    if (clientRef.current === null) {\n      clientRef.current = client;\n    }\n\n    tapEffect(() => {\n      clientRef.current = client;\n    });\n\n    return client;\n  },\n);\n\nexport namespace useAssistantClient {\n  export type Props = {\n    [K in ClientNames]?: ClientElement<K> | DerivedElement<K>;\n  };\n}\n\nexport function useAssistantClient(): AssistantClient;\nexport function useAssistantClient(\n  clients: useAssistantClient.Props,\n): AssistantClient;\nexport function useAssistantClient(\n  clients?: useAssistantClient.Props,\n): AssistantClient {\n  const baseClient = useAssistantContextValue();\n  if (clients) {\n    return useResource(AssistantClientResource({ baseClient, clients }));\n  }\n  return baseClient;\n}\n"],"mappings":";;;AAEA,SAAS,mBAAmB;AAC5B;AAAA,EACE;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,OACK;AASP,SAAS,qBAAqB;AAC9B;AAAA,EACE;AAAA,EACA;AAAA,EACA;AAAA,OACK;AACP;AAAA,EAGE;AAAA,OACK;AACP;AAAA,EACE;AAAA,OAIK;AACP,SAAS,2BAA2B;AACpC,SAAS,uCAAuC;AAChD,SAAS,yBAAyB;AAClC,SAAS,sBAAsB;AAC/B;AAAA,EACE;AAAA,EACA;AAAA,OACK;AAEP,IAAM,qBAAqB;AAAA,EACzB,CAAwB;AAAA,IACtB;AAAA,IACA;AAAA,IACA;AAAA,EACF,MAIM;AACJ,UAAM,EAAE,SAAS,MAAM,IAAI;AAAA,MACzB,EAAE,WAAW,KAAK;AAAA,MAClB,MAAM,kBAAkB,OAAO;AAAA,IACjC;AACA,WAAO,QAAQ,OAAO,EAAE,QAAQ,IAAI,CAAC,KAAK,CAAC;AAAA,EAC7C;AACF;AAEA,IAAM,6BAA6B;AAAA,EACjC,CAAwB;AAAA,IACtB;AAAA,IACA;AAAA,IACA;AAAA,EACF,MAIkC;AAChC,UAAM,QAAQ;AAAA,MACZ;AAAA,QACE,mBAAmB,EAAE,SAAS,MAAM,cAAc,MAAM,UAAU,CAAC;AAAA,MACrE;AAAA,IACF;AAEA,cAAU,MAAM;AACd,aAAO,MAAM,UAAU,cAAc,iBAAiB;AAAA,IACxD,GAAG,CAAC,OAAO,aAAa,CAAC;AAEzB,WAAO,QAAQ,MAAM;AACnB,YAAM,iBAAiB,MAAM,MAAM,SAAS,EAAE;AAC9C,qBAAe,SAAS;AACxB,qBAAe,QAAQ,CAAC;AACxB,aAAO;AAAA,IACT,GAAG,CAAC,KAAK,CAAC;AAAA,EACZ;AACF;AAEA,IAAM,mCAAmC,SAAS,MAAM;AACtD,SAAO;AAAA,IACL,OAAO,EAAE,SAAS,CAAC,GAAG,WAAW,QAAW,IAAI,OAAU;AAAA,IAC1D,CAAC;AAAA,EACH;AACF,CAAC;AAED,IAAM,+BAA+B;AAAA,EACnC,CAAC;AAAA,IACC,SAAS;AAAA,IACT;AAAA,EACF,MAGM;AACJ,UAAM,gBAAgB,kBAAkB,oBAAoB,CAAC;AAE7D;AAAA,MACE,MAAM,UAAU,OAAO,UAAU,cAAc,iBAAiB;AAAA,MAChE,CAAC,WAAW,aAAa;AAAA,IAC3B;AAEA,UAAM,UAAU;AAAA,MACd;AAAA,MACA,CAAC,YACC,2BAA2B;AAAA,QACzB;AAAA,QACA;AAAA,QACA;AAAA,MACF,CAAC;AAAA,MACH,CAAC,eAAe,SAAS;AAAA,IAC3B;AAEA,WAAO,QAAQ,MAAM;AACnB,aAAO;AAAA,QACL,SAAS;AAAA,QACT,WAAW,cAAc;AAAA,QACzB,IAAI,SAEF,UACA,UACA;AACA,cAAI,CAAC,MAAM;AACT,kBAAM,IAAI;AAAA,cACR;AAAA,YACF;AAAA,UACF;AAEA,gBAAM,EAAE,OAAO,MAAM,IAAI,uBAAuB,QAAQ;AAExD,cAAI,UAAU,KAAK;AACjB,kBAAM,SAAS,KAAK,KAAoB,EAAE;AAC1C,gBAAI,WAAW,MAAM;AACnB,oBAAM,IAAI;AAAA,gBACR,UAAU,KAAK,iDAAiD,KAAK;AAAA,cACvE;AAAA,YACF;AAAA,UACF;AAEA,gBAAM,aAAa,cAAc,GAAG,OAAO,CAAC,SAAS,gBAAgB;AACnE,gBAAI,UAAU,KAAK;AACjB,uBAAS,OAAO;AAChB;AAAA,YACF;AAEA,kBAAM,cAAc,KAAK,KAAoB,EAAE;AAC/C,kBAAM,QAAQ,eAAe,WAAW;AACxC,gBAAI,gBAAgB,YAAY,KAAK,GAAG;AACtC,uBAAS,OAAO;AAAA,YAClB;AAAA,UACF,CAAC;AACD,cACE,UAAU,OACV,UAAU,OAAO,KAAoB,EAAE,WAAW;AAElD,mBAAO;AAET,gBAAM,cAAc,UAAU,OAAO,GAAG,UAAU,QAAQ;AAE1D,iBAAO,MAAM;AACX,uBAAW;AACX,wBAAY;AAAA,UACd;AAAA,QACF;AAAA,MACF;AAAA,IACF,GAAG,CAAC,SAAS,eAAe,SAAS,CAAC;AAAA,EACxC;AACF;AAOA,IAAM,UAAU,CACd,OACA,WACA,SACkB;AAClB,MAAI,YAAY,SAAS,WAAW,MAAO,QAAO;AAClD,MAAI,KAAK,QAAQ,MAAO,QAAO,KAAK;AACpC,QAAM,OAAO,MAAM,QAAQ,UAAU,OAAQ;AAC7C,OAAK,OAAO;AACZ,OAAK,MAAM;AACX,SAAO;AACT;AAEA,IAAM,gCAAgC;AAAA,EACpC,CAAwB;AAAA,IACtB;AAAA,IACA;AAAA,EACF,MAGM;AACJ,UAAM,MAAM,eAAe,MAAM,QAAQ,KAAK;AAE9C,WAAO,QAAQ,MAAM;AACnB,YAAM,iBAAiB,MAAM,IAAI,EAAE,IAAI,UAAU,OAAQ;AACzD,YAAM,WAAW,CAAC;AAClB,aAAO,iBAAiB,gBAAgB;AAAA,QACtC,QAAQ;AAAA,UACN,KAAK,MAAM,QAAQ,IAAI,GAAG,WAAW,QAAQ,EAAE;AAAA,QACjD;AAAA,QACA,OAAO;AAAA,UACL,KAAK,MAAM,QAAQ,IAAI,GAAG,WAAW,QAAQ,EAAE;AAAA,QACjD;AAAA,MACF,CAAC;AACD,aAAO;AAAA,IACT,GAAG,CAAC,SAAS,CAAC;AAAA,EAChB;AACF;AAEA,IAAM,kCAAkC;AAAA,EACtC,CAAC;AAAA,IACC;AAAA,IACA;AAAA,EACF,MAGM;AACJ,WAAO;AAAA,MACL;AAAA,MACA,CAAC,YACC,8BAA8B;AAAA,QAC5B;AAAA,QACA;AAAA,MACF,CAAC;AAAA,MACH,CAAC,SAAS;AAAA,IACZ;AAAA,EACF;AACF;AAKO,IAAM,0BAA0B;AAAA,EACrC,CAAC;AAAA,IACC;AAAA,IACA;AAAA,EACF,MAGuB;AACrB,UAAM,EAAE,aAAa,eAAe,IAAI,aAAa,SAAS,UAAU;AAExE,UAAM,YAAY,OAAO;AAAA,MACvB,QAAQ;AAAA,MACR,SAAS;AAAA,IACX,CAAC,EAAE;AAEH,UAAM,aAAa;AAAA,MACjB,OAAO,KAAK,WAAW,EAAE,SAAS,IAC9B,6BAA6B,EAAE,SAAS,aAAa,UAAU,CAAC,IAChE,iCAAiC;AAAA,IACvC;AAEA,UAAM,gBAAgB;AAAA,MACpB,gCAAgC,EAAE,SAAS,gBAAgB,UAAU,CAAC;AAAA,IACxE;AAEA,UAAM,SAAS,QAAQ,MAAM;AAE3B,YAAM,QACJ,eAAe,yBACX,0BAA0B,IAC1B;AACN,YAAMA,UAAS,OAAO,OAAO,KAAK;AAClC,aAAO,OAAOA,SAAQ,WAAW,SAAS,eAAe;AAAA,QACvD,WAAW,WAAW,aAAa,WAAW;AAAA,QAC9C,IAAI,WAAW,MAAM,WAAW;AAAA,QAChC,CAAC,8BAA8B,GAAG,4BAA4BA,OAAM;AAAA,MACtE,CAAC;AACD,aAAOA;AAAA,IACT,GAAG,CAAC,YAAY,YAAY,aAAa,CAAC;AAE1C,QAAI,UAAU,YAAY,MAAM;AAC9B,gBAAU,UAAU;AAAA,IACtB;AAEA,cAAU,MAAM;AACd,gBAAU,UAAU;AAAA,IACtB,CAAC;AAED,WAAO;AAAA,EACT;AACF;AAYO,SAAS,mBACd,SACiB;AACjB,QAAM,aAAa,yBAAyB;AAC5C,MAAI,SAAS;AACX,WAAO,YAAY,wBAAwB,EAAE,YAAY,QAAQ,CAAC,CAAC;AAAA,EACrE;AACA,SAAO;AACT;","names":["client"]}

package/dist/useAssistantEvent.d.ts

@@ -1,3 +1,3 @@
-import type { AssistantEvent, AssistantEventCallback, AssistantEventSelector } from "./EventContext";
-export declare const useAssistantEvent: <TEvent extends AssistantEvent>(selector: AssistantEventSelector<TEvent>, callback: AssistantEventCallback<TEvent>) => void;
+import type { AssistantEventName, AssistantEventCallback, AssistantEventSelector } from "./types/events";
+export declare const useAssistantEvent: <TEvent extends AssistantEventName>(selector: AssistantEventSelector<TEvent>, callback: AssistantEventCallback<TEvent>) => void;
 //# sourceMappingURL=useAssistantEvent.d.ts.map

package/dist/useAssistantEvent.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"useAssistantEvent.d.ts","sourceRoot":"","sources":["../src/useAssistantEvent.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EACV,cAAc,EACd,sBAAsB,EACtB,sBAAsB,EACvB,MAAM,gBAAgB,CAAC;AAGxB,eAAO,MAAM,iBAAiB,GAAI,MAAM,SAAS,cAAc,EAC7D,UAAU,sBAAsB,CAAC,MAAM,CAAC,EACxC,UAAU,sBAAsB,CAAC,MAAM,CAAC,SAUzC,CAAC"}
+{"version":3,"file":"useAssistantEvent.d.ts","sourceRoot":"","sources":["../src/useAssistantEvent.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EACV,kBAAkB,EAClB,sBAAsB,EACtB,sBAAsB,EACvB,MAAM,gBAAgB,CAAC;AAGxB,eAAO,MAAM,iBAAiB,GAAI,MAAM,SAAS,kBAAkB,EACjE,UAAU,sBAAsB,CAAC,MAAM,CAAC,EACxC,UAAU,sBAAsB,CAAC,MAAM,CAAC,SAOzC,CAAC"}

package/dist/useAssistantEvent.js

@@ -1,15 +1,12 @@
 // src/useAssistantEvent.ts
 import { useEffect, useEffectEvent } from "react";
 import { useAssistantClient } from "./useAssistantClient.js";
-import { normalizeEventSelector } from "./EventContext.js";
+import { normalizeEventSelector } from "./types/events.js";
 var useAssistantEvent = (selector, callback) => {
-  const client = useAssistantClient();
+  const aui = useAssistantClient();
   const callbackRef = useEffectEvent(callback);
   const { scope, event } = normalizeEventSelector(selector);
-  useEffect(
-    () => client.on({ scope, event }, callbackRef),
-    [client, scope, event]
-  );
+  useEffect(() => aui.on({ scope, event }, callbackRef), [aui, scope, event]);
 };
 export {
   useAssistantEvent

package/dist/useAssistantEvent.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../src/useAssistantEvent.ts"],"sourcesContent":["import { useEffect, useEffectEvent } from \"react\";\nimport { useAssistantClient } from \"./useAssistantClient\";\nimport type {\n  AssistantEvent,\n  AssistantEventCallback,\n  AssistantEventSelector,\n} from \"./EventContext\";\nimport { normalizeEventSelector } from \"./EventContext\";\n\nexport const useAssistantEvent = <TEvent extends AssistantEvent>(\n  selector: AssistantEventSelector<TEvent>,\n  callback: AssistantEventCallback<TEvent>,\n) => {\n  const client = useAssistantClient();\n  const callbackRef = useEffectEvent(callback);\n\n  const { scope, event } = normalizeEventSelector(selector);\n  useEffect(\n    () => client.on({ scope, event }, callbackRef),\n    [client, scope, event],\n  );\n};\n"],"mappings":";AAAA,SAAS,WAAW,sBAAsB;AAC1C,SAAS,0BAA0B;AAMnC,SAAS,8BAA8B;AAEhC,IAAM,oBAAoB,CAC/B,UACA,aACG;AACH,QAAM,SAAS,mBAAmB;AAClC,QAAM,cAAc,eAAe,QAAQ;AAE3C,QAAM,EAAE,OAAO,MAAM,IAAI,uBAAuB,QAAQ;AACxD;AAAA,IACE,MAAM,OAAO,GAAG,EAAE,OAAO,MAAM,GAAG,WAAW;AAAA,IAC7C,CAAC,QAAQ,OAAO,KAAK;AAAA,EACvB;AACF;","names":[]}
+{"version":3,"sources":["../src/useAssistantEvent.ts"],"sourcesContent":["import { useEffect, useEffectEvent } from \"react\";\nimport { useAssistantClient } from \"./useAssistantClient\";\nimport type {\n  AssistantEventName,\n  AssistantEventCallback,\n  AssistantEventSelector,\n} from \"./types/events\";\nimport { normalizeEventSelector } from \"./types/events\";\n\nexport const useAssistantEvent = <TEvent extends AssistantEventName>(\n  selector: AssistantEventSelector<TEvent>,\n  callback: AssistantEventCallback<TEvent>,\n) => {\n  const aui = useAssistantClient();\n  const callbackRef = useEffectEvent(callback);\n\n  const { scope, event } = normalizeEventSelector(selector);\n  useEffect(() => aui.on({ scope, event }, callbackRef), [aui, scope, event]);\n};\n"],"mappings":";AAAA,SAAS,WAAW,sBAAsB;AAC1C,SAAS,0BAA0B;AAMnC,SAAS,8BAA8B;AAEhC,IAAM,oBAAoB,CAC/B,UACA,aACG;AACH,QAAM,MAAM,mBAAmB;AAC/B,QAAM,cAAc,eAAe,QAAQ;AAE3C,QAAM,EAAE,OAAO,MAAM,IAAI,uBAAuB,QAAQ;AACxD,YAAU,MAAM,IAAI,GAAG,EAAE,OAAO,MAAM,GAAG,WAAW,GAAG,CAAC,KAAK,OAAO,KAAK,CAAC;AAC5E;","names":[]}

package/dist/useAssistantState.d.ts

@@ -1,4 +1,4 @@
-import type { AssistantState } from "./types";
+import type { AssistantState } from "./types/client";
 /**
  * Hook to access a slice of the assistant state with automatic subscription
  *
@@ -7,7 +7,7 @@ import type { AssistantState } from "./types";
  *
  * @example
  * ```typescript
- * const client = useAssistantClient({
+ * const aui = useAssistantClient({
  *   foo: RootScope({ ... }),
  * });
  *

package/dist/useAssistantState.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"useAssistantState.d.ts","sourceRoot":"","sources":["../src/useAssistantState.tsx"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAmB,cAAc,EAAE,MAAM,SAAS,CAAC;AAsC/D;;;;;;;;;;;;;;GAcG;AACH,eAAO,MAAM,iBAAiB,GAAI,CAAC,EACjC,UAAU,CAAC,KAAK,EAAE,cAAc,KAAK,CAAC,KACrC,CAuBF,CAAC"}
+{"version":3,"file":"useAssistantState.d.ts","sourceRoot":"","sources":["../src/useAssistantState.tsx"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,gBAAgB,CAAC;AAIrD;;;;;;;;;;;;;;GAcG;AACH,eAAO,MAAM,iBAAiB,GAAI,CAAC,EACjC,UAAU,CAAC,KAAK,EAAE,cAAc,KAAK,CAAC,KACrC,CAmBF,CAAC"}

package/dist/useAssistantState.js

@@ -1,50 +1,21 @@
 // src/useAssistantState.tsx
-import { useMemo, useSyncExternalStore, useDebugValue } from "react";
+import { useSyncExternalStore, useDebugValue } from "react";
 import { useAssistantClient } from "./useAssistantClient.js";
-var ProxiedAssistantState = class _ProxiedAssistantState {
-  #client;
-  constructor(client) {
-    this.#client = client;
-  }
-  #getScope(key) {
-    const scopeField = this.#client[key];
-    if (!scopeField) {
-      throw new Error(`Scope "${String(key)}" not found in client`);
-    }
-    const api = scopeField();
-    const state = api.getState();
-    return state;
-  }
-  // Create a Proxy to dynamically handle property access
-  static create(client) {
-    const instance = new _ProxiedAssistantState(client);
-    return new Proxy(instance, {
-      get(target, prop) {
-        if (typeof prop === "string" && prop in client) {
-          return target.#getScope(prop);
-        }
-        return void 0;
-      }
-    });
-  }
-};
+import { getProxiedAssistantState } from "./utils/proxied-assistant-state.js";
 var useAssistantState = (selector) => {
-  const client = useAssistantClient();
-  const proxiedState = useMemo(
-    () => ProxiedAssistantState.create(client),
-    [client]
-  );
+  const aui = useAssistantClient();
+  const proxiedState = getProxiedAssistantState(aui);
   const slice = useSyncExternalStore(
-    client.subscribe,
+    aui.subscribe,
     () => selector(proxiedState),
     () => selector(proxiedState)
   );
-  useDebugValue(slice);
-  if (slice instanceof ProxiedAssistantState) {
+  if (slice === proxiedState) {
     throw new Error(
       "You tried to return the entire AssistantState. This is not supported due to technical limitations."
     );
   }
+  useDebugValue(slice);
   return slice;
 };
 export {

package/dist/useAssistantState.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../src/useAssistantState.tsx"],"sourcesContent":["import { useMemo, useSyncExternalStore, useDebugValue } from \"react\";\nimport type { AssistantClient, AssistantState } from \"./types\";\nimport { useAssistantClient } from \"./useAssistantClient\";\n\n/**\n * Proxied state that lazily accesses scope states\n */\nclass ProxiedAssistantState {\n  #client: AssistantClient;\n\n  constructor(client: AssistantClient) {\n    this.#client = client;\n  }\n\n  #getScope<K extends keyof AssistantState>(key: K): AssistantState[K] {\n    const scopeField = this.#client[key];\n    if (!scopeField) {\n      throw new Error(`Scope \"${String(key)}\" not found in client`);\n    }\n\n    const api = scopeField();\n    const state = api.getState();\n    return state as AssistantState[K];\n  }\n\n  // Create a Proxy to dynamically handle property access\n  static create(client: AssistantClient): AssistantState {\n    const instance = new ProxiedAssistantState(client);\n    return new Proxy(instance, {\n      get(target, prop) {\n        if (typeof prop === \"string\" && prop in client) {\n          return target.#getScope(prop as keyof AssistantState);\n        }\n        return undefined;\n      },\n    }) as unknown as AssistantState;\n  }\n}\n\n/**\n * Hook to access a slice of the assistant state with automatic subscription\n *\n * @param selector - Function to select a slice of the state\n * @returns The selected state slice\n *\n * @example\n * ```typescript\n * const client = useAssistantClient({\n *   foo: RootScope({ ... }),\n * });\n *\n * const bar = useAssistantState((state) => state.foo.bar);\n * ```\n */\nexport const useAssistantState = <T,>(\n  selector: (state: AssistantState) => T,\n): T => {\n  const client = useAssistantClient();\n\n  const proxiedState = useMemo(\n    () => ProxiedAssistantState.create(client),\n    [client],\n  );\n\n  const slice = useSyncExternalStore(\n    client.subscribe,\n    () => selector(proxiedState),\n    () => selector(proxiedState),\n  );\n\n  useDebugValue(slice);\n\n  if (slice instanceof ProxiedAssistantState) {\n    throw new Error(\n      \"You tried to return the entire AssistantState. This is not supported due to technical limitations.\",\n    );\n  }\n\n  return slice;\n};\n"],"mappings":";AAAA,SAAS,SAAS,sBAAsB,qBAAqB;AAE7D,SAAS,0BAA0B;AAKnC,IAAM,wBAAN,MAAM,uBAAsB;AAAA,EAC1B;AAAA,EAEA,YAAY,QAAyB;AACnC,SAAK,UAAU;AAAA,EACjB;AAAA,EAEA,UAA0C,KAA2B;AACnE,UAAM,aAAa,KAAK,QAAQ,GAAG;AACnC,QAAI,CAAC,YAAY;AACf,YAAM,IAAI,MAAM,UAAU,OAAO,GAAG,CAAC,uBAAuB;AAAA,IAC9D;AAEA,UAAM,MAAM,WAAW;AACvB,UAAM,QAAQ,IAAI,SAAS;AAC3B,WAAO;AAAA,EACT;AAAA;AAAA,EAGA,OAAO,OAAO,QAAyC;AACrD,UAAM,WAAW,IAAI,uBAAsB,MAAM;AACjD,WAAO,IAAI,MAAM,UAAU;AAAA,MACzB,IAAI,QAAQ,MAAM;AAChB,YAAI,OAAO,SAAS,YAAY,QAAQ,QAAQ;AAC9C,iBAAO,OAAO,UAAU,IAA4B;AAAA,QACtD;AACA,eAAO;AAAA,MACT;AAAA,IACF,CAAC;AAAA,EACH;AACF;AAiBO,IAAM,oBAAoB,CAC/B,aACM;AACN,QAAM,SAAS,mBAAmB;AAElC,QAAM,eAAe;AAAA,IACnB,MAAM,sBAAsB,OAAO,MAAM;AAAA,IACzC,CAAC,MAAM;AAAA,EACT;AAEA,QAAM,QAAQ;AAAA,IACZ,OAAO;AAAA,IACP,MAAM,SAAS,YAAY;AAAA,IAC3B,MAAM,SAAS,YAAY;AAAA,EAC7B;AAEA,gBAAc,KAAK;AAEnB,MAAI,iBAAiB,uBAAuB;AAC1C,UAAM,IAAI;AAAA,MACR;AAAA,IACF;AAAA,EACF;AAEA,SAAO;AACT;","names":[]}
+{"version":3,"sources":["../src/useAssistantState.tsx"],"sourcesContent":["import { useSyncExternalStore, useDebugValue } from \"react\";\nimport type { AssistantState } from \"./types/client\";\nimport { useAssistantClient } from \"./useAssistantClient\";\nimport { getProxiedAssistantState } from \"./utils/proxied-assistant-state\";\n\n/**\n * Hook to access a slice of the assistant state with automatic subscription\n *\n * @param selector - Function to select a slice of the state\n * @returns The selected state slice\n *\n * @example\n * ```typescript\n * const aui = useAssistantClient({\n *   foo: RootScope({ ... }),\n * });\n *\n * const bar = useAssistantState((state) => state.foo.bar);\n * ```\n */\nexport const useAssistantState = <T,>(\n  selector: (state: AssistantState) => T,\n): T => {\n  const aui = useAssistantClient();\n  const proxiedState = getProxiedAssistantState(aui);\n\n  const slice = useSyncExternalStore(\n    aui.subscribe,\n    () => selector(proxiedState),\n    () => selector(proxiedState),\n  );\n\n  if (slice === proxiedState) {\n    throw new Error(\n      \"You tried to return the entire AssistantState. This is not supported due to technical limitations.\",\n    );\n  }\n\n  useDebugValue(slice);\n\n  return slice;\n};\n"],"mappings":";AAAA,SAAS,sBAAsB,qBAAqB;AAEpD,SAAS,0BAA0B;AACnC,SAAS,gCAAgC;AAiBlC,IAAM,oBAAoB,CAC/B,aACM;AACN,QAAM,MAAM,mBAAmB;AAC/B,QAAM,eAAe,yBAAyB,GAAG;AAEjD,QAAM,QAAQ;AAAA,IACZ,IAAI;AAAA,IACJ,MAAM,SAAS,YAAY;AAAA,IAC3B,MAAM,SAAS,YAAY;AAAA,EAC7B;AAEA,MAAI,UAAU,cAAc;AAC1B,UAAM,IAAI;AAAA,MACR;AAAA,IACF;AAAA,EACF;AAEA,gBAAc,KAAK;AAEnB,SAAO;AACT;","names":[]}

package/dist/utils/BaseProxyHandler.d.ts

@@ -0,0 +1,23 @@
+/**
+ * Handles common proxy introspection properties.
+ * Returns the appropriate value for toStringTag, toJSON, and props that should return undefined.
+ * Returns `false` if the prop should be handled by the subclass.
+ */
+export declare const handleIntrospectionProp: (prop: string | symbol, name: string) => unknown | false;
+export declare abstract class BaseProxyHandler implements ProxyHandler<object> {
+    abstract get(_: unknown, prop: string | symbol): unknown;
+    abstract ownKeys(): ArrayLike<string | symbol>;
+    abstract has(_: unknown, prop: string | symbol): boolean;
+    getOwnPropertyDescriptor(_: unknown, prop: string | symbol): {
+        value: {} | null;
+        writable: boolean;
+        enumerable: boolean;
+        configurable: boolean;
+    } | undefined;
+    set(): boolean;
+    setPrototypeOf(): boolean;
+    defineProperty(): boolean;
+    deleteProperty(): boolean;
+    preventExtensions(): boolean;
+}
+//# sourceMappingURL=BaseProxyHandler.d.ts.map

package/dist/utils/BaseProxyHandler.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"BaseProxyHandler.d.ts","sourceRoot":"","sources":["../../src/utils/BaseProxyHandler.ts"],"names":[],"mappings":"AAEA;;;;GAIG;AACH,eAAO,MAAM,uBAAuB,GAClC,MAAM,MAAM,GAAG,MAAM,EACrB,MAAM,MAAM,KACX,OAAO,GAAG,KAMZ,CAAC;AAEF,8BAAsB,gBAAiB,YAAW,YAAY,CAAC,MAAM,CAAC;IACpE,QAAQ,CAAC,GAAG,CAAC,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,MAAM,GAAG,MAAM,GAAG,OAAO;IACxD,QAAQ,CAAC,OAAO,IAAI,SAAS,CAAC,MAAM,GAAG,MAAM,CAAC;IAC9C,QAAQ,CAAC,GAAG,CAAC,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,MAAM,GAAG,MAAM,GAAG,OAAO;IAExD,wBAAwB,CAAC,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,MAAM,GAAG,MAAM;;;;;;IAW1D,GAAG;IAGH,cAAc;IAGd,cAAc;IAGd,cAAc;IAGd,iBAAiB,IAAI,OAAO;CAG7B"}

package/dist/utils/BaseProxyHandler.js

@@ -0,0 +1,41 @@
+// src/utils/BaseProxyHandler.ts
+var INTROSPECTION_PROPS = /* @__PURE__ */ new Set(["$$typeof", "nodeType", "then"]);
+var handleIntrospectionProp = (prop, name) => {
+  if (prop === Symbol.toStringTag) return name;
+  if (typeof prop === "symbol") return void 0;
+  if (prop === "toJSON") return () => name;
+  if (INTROSPECTION_PROPS.has(prop)) return void 0;
+  return false;
+};
+var BaseProxyHandler = class {
+  getOwnPropertyDescriptor(_, prop) {
+    const value = this.get(_, prop);
+    if (value === void 0) return void 0;
+    return {
+      value,
+      writable: false,
+      enumerable: true,
+      configurable: false
+    };
+  }
+  set() {
+    return false;
+  }
+  setPrototypeOf() {
+    return false;
+  }
+  defineProperty() {
+    return false;
+  }
+  deleteProperty() {
+    return false;
+  }
+  preventExtensions() {
+    return false;
+  }
+};
+export {
+  BaseProxyHandler,
+  handleIntrospectionProp
+};
+//# sourceMappingURL=BaseProxyHandler.js.map

package/dist/utils/BaseProxyHandler.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../src/utils/BaseProxyHandler.ts"],"sourcesContent":["const INTROSPECTION_PROPS = new Set([\"$$typeof\", \"nodeType\", \"then\"]);\n\n/**\n * Handles common proxy introspection properties.\n * Returns the appropriate value for toStringTag, toJSON, and props that should return undefined.\n * Returns `false` if the prop should be handled by the subclass.\n */\nexport const handleIntrospectionProp = (\n  prop: string | symbol,\n  name: string,\n): unknown | false => {\n  if (prop === Symbol.toStringTag) return name;\n  if (typeof prop === \"symbol\") return undefined;\n  if (prop === \"toJSON\") return () => name;\n  if (INTROSPECTION_PROPS.has(prop)) return undefined;\n  return false;\n};\n\nexport abstract class BaseProxyHandler implements ProxyHandler<object> {\n  abstract get(_: unknown, prop: string | symbol): unknown;\n  abstract ownKeys(): ArrayLike<string | symbol>;\n  abstract has(_: unknown, prop: string | symbol): boolean;\n\n  getOwnPropertyDescriptor(_: unknown, prop: string | symbol) {\n    const value = this.get(_, prop);\n    if (value === undefined) return undefined;\n    return {\n      value,\n      writable: false,\n      enumerable: true,\n      configurable: false,\n    };\n  }\n\n  set() {\n    return false;\n  }\n  setPrototypeOf() {\n    return false;\n  }\n  defineProperty() {\n    return false;\n  }\n  deleteProperty() {\n    return false;\n  }\n  preventExtensions(): boolean {\n    return false;\n  }\n}\n"],"mappings":";AAAA,IAAM,sBAAsB,oBAAI,IAAI,CAAC,YAAY,YAAY,MAAM,CAAC;AAO7D,IAAM,0BAA0B,CACrC,MACA,SACoB;AACpB,MAAI,SAAS,OAAO,YAAa,QAAO;AACxC,MAAI,OAAO,SAAS,SAAU,QAAO;AACrC,MAAI,SAAS,SAAU,QAAO,MAAM;AACpC,MAAI,oBAAoB,IAAI,IAAI,EAAG,QAAO;AAC1C,SAAO;AACT;AAEO,IAAe,mBAAf,MAAgE;AAAA,EAKrE,yBAAyB,GAAY,MAAuB;AAC1D,UAAM,QAAQ,KAAK,IAAI,GAAG,IAAI;AAC9B,QAAI,UAAU,OAAW,QAAO;AAChC,WAAO;AAAA,MACL;AAAA,MACA,UAAU;AAAA,MACV,YAAY;AAAA,MACZ,cAAc;AAAA,IAChB;AAAA,EACF;AAAA,EAEA,MAAM;AACJ,WAAO;AAAA,EACT;AAAA,EACA,iBAAiB;AACf,WAAO;AAAA,EACT;AAAA,EACA,iBAAiB;AACf,WAAO;AAAA,EACT;AAAA,EACA,iBAAiB;AACf,WAAO;AAAA,EACT;AAAA,EACA,oBAA6B;AAC3B,WAAO;AAAA,EACT;AACF;","names":[]}