@assistant-ui/store 0.1.6 → 0.2.0

package/src/utils/splitClients.ts

@@ -1,130 +0,0 @@
-import { Derived, DerivedElement } from "../Derived";
-import type {
-  AssistantClient,
-  ClientElement,
-  ClientNames,
-} from "../types/client";
-import { getDefaultPeers } from "../attachDefaultPeers";
-import type { useAui } from "../useAui";
-import { tapMemo } from "@assistant-ui/tap";
-
-export type RootClients = Partial<
-  Record<ClientNames, ClientElement<ClientNames>>
->;
-export type DerivedClients = Partial<
-  Record<ClientNames, DerivedElement<ClientNames>>
->;
-
-/**
- * Splits a clients object into root clients and derived clients.
- *
- * @param clients - The clients input object to split
- * @returns An object with { rootClients, derivedClients }
- *
- * @example
- * ```typescript
- * const clients = {
- *   foo: RootClient({ ... }),
- *   bar: Derived({ ... }),
- * };
- *
- * const { rootClients, derivedClients } = splitClients(clients);
- * // rootClients = { foo: ... }
- * // derivedClients = { bar: ... }
- * ```
- */
-function splitClients(clients: useAui.Props, baseClient: AssistantClient) {
-  const rootClients: RootClients = {};
-  const derivedClients: DerivedClients = {};
-
-  for (const [key, clientElement] of Object.entries(clients) as [
-    keyof useAui.Props,
-    NonNullable<useAui.Props[keyof useAui.Props]>,
-  ][]) {
-    if (clientElement.type === Derived) {
-      derivedClients[key] = clientElement as DerivedElement<ClientNames>;
-    } else {
-      rootClients[key] = clientElement as ClientElement<ClientNames>;
-    }
-  }
-
-  // Recursively gather all default peers, flattening nested ones
-  const gatherDefaultPeers = (
-    clientElement: ClientElement<ClientNames>,
-    visited = new Set<ClientElement<ClientNames>>(),
-  ): Array<
-    [ClientNames, ClientElement<ClientNames> | DerivedElement<ClientNames>]
-  > => {
-    // Prevent infinite loops
-    if (visited.has(clientElement)) return [];
-    visited.add(clientElement);
-
-    const defaultPeers = getDefaultPeers(clientElement.type);
-    if (!defaultPeers) return [];
-
-    const result: Array<
-      [ClientNames, ClientElement<ClientNames> | DerivedElement<ClientNames>]
-    > = [];
-
-    for (const [key, peerElement] of Object.entries(defaultPeers) as [
-      ClientNames,
-      ClientElement<ClientNames> | DerivedElement<ClientNames>,
-    ][]) {
-      result.push([key, peerElement]);
-
-      // If this peer is a root client with its own default peers, recursively gather them
-      if (peerElement.type !== Derived<ClientNames>) {
-        const nestedPeers = gatherDefaultPeers(
-          peerElement as ClientElement<ClientNames>,
-          visited,
-        );
-        result.push(...nestedPeers);
-      }
-    }
-
-    return result;
-  };
-
-  // Apply flattened default peers for each root client
-  for (const [_clientKey, clientElement] of Object.entries(rootClients) as [
-    ClientNames,
-    ClientElement<ClientNames>,
-  ][]) {
-    const allPeers = gatherDefaultPeers(clientElement);
-
-    for (const [key, peerElement] of allPeers) {
-      // Skip if already exists (first wins)
-      if (
-        key in rootClients ||
-        key in derivedClients ||
-        baseClient[key].source !== null
-      )
-        continue;
-
-      if (peerElement.type === Derived<ClientNames>) {
-        derivedClients[key] = peerElement as DerivedElement<ClientNames>;
-      } else {
-        rootClients[key] = peerElement as ClientElement<ClientNames>;
-      }
-    }
-  }
-
-  return { rootClients, derivedClients };
-}
-
-const tapShallowMemoObject = <T extends object>(object: T) => {
-  // biome-ignore lint/correctness/useExhaustiveDependencies: shallow memo
-  return tapMemo(() => object, [...Object.entries(object).flat()]);
-};
-
-export const tapSplitClients = (
-  clients: useAui.Props,
-  baseClient: AssistantClient,
-) => {
-  const { rootClients, derivedClients } = splitClients(clients, baseClient);
-
-  return {
-    rootClients: tapShallowMemoObject(rootClients),
-    derivedClients: tapShallowMemoObject(derivedClients),
-  };
-};

package/src/utils/tap-assistant-context.ts

@@ -1,58 +0,0 @@
-import {
-  createResourceContext,
-  tap,
-  withContextProvider,
-  tapEffectEvent,
-} from "@assistant-ui/tap";
-import type {
-  AssistantEventName,
-  AssistantEventPayload,
-} from "../types/events";
-import type { AssistantClient } from "../types/client";
-import { tapClientStack, type ClientStack } from "./tap-client-stack-context";
-
-type EmitFn = <TEvent extends Exclude<AssistantEventName, "*">>(
-  event: TEvent,
-  payload: AssistantEventPayload[TEvent],
-  clientStack: ClientStack,
-) => void;
-
-export type AssistantTapContextValue = {
-  clientRef: { parent: AssistantClient; current: AssistantClient | null };
-  emit: EmitFn;
-};
-
-const AssistantTapContext =
-  createResourceContext<AssistantTapContextValue | null>(null);
-
-export const withAssistantTapContextProvider = <TResult>(
-  value: AssistantTapContextValue,
-  fn: () => TResult,
-) => {
-  return withContextProvider(AssistantTapContext, value, fn);
-};
-
-const tapAssistantTapContext = () => {
-  const ctx = tap(AssistantTapContext);
-  if (!ctx) throw new Error("AssistantTapContext is not available");
-
-  return ctx;
-};
-
-export const tapAssistantClientRef = () => {
-  return tapAssistantTapContext().clientRef;
-};
-
-export const tapAssistantEmit = () => {
-  const { emit } = tapAssistantTapContext();
-  const clientStack = tapClientStack();
-
-  return tapEffectEvent(
-    <TEvent extends Exclude<AssistantEventName, "*">>(
-      event: TEvent,
-      payload: AssistantEventPayload[TEvent],
-    ) => {
-      emit(event, payload, clientStack);
-    },
-  );
-};

package/src/utils/tap-client-stack-context.ts

@@ -1,51 +0,0 @@
-import {
-  createResourceContext,
-  tap,
-  withContextProvider,
-  tapMemo,
-} from "@assistant-ui/tap";
-import type { ClientMethods } from "../types/client";
-
-/**
- * Symbol used to get the client index from a ClientProxy.
- */
-export const SYMBOL_CLIENT_INDEX = Symbol("assistant-ui.store.clientIndex");
-
-/**
- * Get the index of a client (its position in the client stack when created).
- */
-export const getClientIndex = (client: ClientMethods): number => {
-  return (client as unknown as { [SYMBOL_CLIENT_INDEX]: number })[
-    SYMBOL_CLIENT_INDEX
-  ];
-};
-
-/**
- * The client stack - an array of clients representing the current hierarchy.
- */
-export type ClientStack = readonly ClientMethods[];
-
-const ClientStackContext = createResourceContext<ClientStack>([]);
-
-/**
- * Get the current client stack inside a tap resource.
- */
-export const tapClientStack = (): ClientStack => {
-  return tap(ClientStackContext);
-};
-
-/**
- * Execute a callback with a client pushed onto the stack.
- * The stack is duplicated, not mutated.
- */
-export const tapWithClientStack = <T>(
-  client: ClientMethods,
-  callback: () => T,
-): T => {
-  const currentStack = tapClientStack();
-  const newStack = tapMemo(
-    () => [...currentStack, client],
-    [currentStack, client],
-  );
-  return withContextProvider(ClientStackContext, newStack, callback);
-};

package/src/wrapperResource.ts

@@ -1,17 +0,0 @@
-import {
-  type ResourceElement,
-  Resource,
-  resource,
-  withKey,
-} from "@assistant-ui/tap";
-
-export const wrapperResource = <R, P>(
-  fn: (props: ResourceElement<P>) => R,
-): Resource<R, ResourceElement<P>> => {
-  const res = resource(fn);
-  return (props: ResourceElement<P>) => {
-    const el = res(props);
-    if (props.key === undefined) return el;
-    return withKey(props.key, el);
-  };
-};