@assistant-ui/react/src/utils/tap-store/store.ts

TypeScript/React library for AI Chat

import {
  tapMemo,
  tapEffect,
  ResourceElement,
  resource,
  createResource,
} from "@assistant-ui/tap";
import { Unsubscribe } from "../../types";

export interface Store<TState> {
  /**
   * Get the current state of the store.
   */
  getState(): TState;

  /**
   * Subscribe to the store.
   */
  subscribe(listener: () => void): Unsubscribe;
}

export const asStore = resource(
  <TState, TProps>(element: ResourceElement<TState, TProps>): Store<TState> => {
    const resource = tapMemo(
      () => createResource(element, { mount: false }),
      [element.type],
    );

    tapEffect(() => {
      resource.render(element);
    });

    return resource;
  },
);