@schlessera/brain-ui-react 0.34.1 → 0.36.0

package/src/hooks/websocket-handlers/mask.ts

@@ -1,13 +1,12 @@
-import { useMaskStore } from "../../stores/mask-store.js";
 import type { ServerMessageHandlerMap } from "./types.js";
 
 type MaskFrame = "mask_request";
 
 export const maskFrameHandlers = {
-  mask_request: (msg) => {
+  mask_request: (msg, context) => {
     // Opens the editor; the answer travels back from the component, because
     // only the user can say which part of the picture they meant.
-    useMaskStore.getState().open({
+    context.stores.mask.getState().open({
       requestId: msg.requestId,
       imagePath: msg.imagePath,
       ...(msg.instruction ? { instruction: msg.instruction } : {}),

package/src/hooks/websocket-handlers/provider.ts

@@ -1,5 +1,3 @@
-import { useChatStore } from "../../stores/chat-store.js";
-import { useProviderStore } from "../../stores/provider-store.js";
 import type { ServerMessageHandlerMap } from "./types.js";
 
 type ProviderFrame = "session_info";
@@ -13,17 +11,17 @@ export const providerFrameHandlers = {
     // (fails only for since-hidden profiles, which then use the default).
     const ownerBackendId =
       msg.backendId ??
-      useProviderStore
+      context.stores.provider
         .getState()
         .available.find((p) => p.id === msg.providerId)?.backendId;
     if (ownerBackendId) {
-      useChatStore.getState().setSessionBackend(msg.sessionId, ownerBackendId);
+      context.stores.chat.getState().setSessionBackend(msg.sessionId, ownerBackendId);
     }
     // bindDraftSession in the dispatcher handled draft adoption; an info frame
     // may still re-pin the provider picker when it concerns the session in view.
-    const current = useChatStore.getState();
+    const current = context.stores.chat.getState();
     if (context.frameSessionId && context.frameSessionId === current.activeSessionId) {
-      useProviderStore.getState().setPinned(msg.providerId ?? null);
+      context.stores.provider.getState().setPinned(msg.providerId ?? null);
       // Reattachment on a fresh connection: enter streaming mode so deltas
       // append to a live assistant message instead of vanishing.
       const chat = current.buffers[context.frameSessionId];

package/src/hooks/websocket-handlers/types.ts

@@ -1,12 +1,16 @@
 import type { ServerMessage } from "@schlessera/brain-ui-sdk/protocol";
 import type { ChatKey, SessionChat } from "../../stores/chat-store.js";
-import { useChatStore } from "../../stores/chat-store.js";
+import type { ChatState } from "../../stores/chat-state.js";
+import type { BrainStores } from "../../stores/create-stores.js";
 
 export type ServerMessageType = ServerMessage["type"];
 
 export interface DispatchContext {
-  state: ReturnType<typeof useChatStore.getState>;
+  state: ChatState;
+  stores: BrainStores;
   frameSessionId: string | undefined;
+  /** The host-minted turn the frame belongs to, when scoped (rev 2+). */
+  frameTurnId: string | undefined;
   key: ChatKey;
   buffer: () => SessionChat | null | undefined;
   enqueueDelta: (key: ChatKey, kind: "text" | "thinking", text: string) => void;

package/src/index.ts

@@ -9,7 +9,7 @@
  */
 
 // Branding / copy configuration (module singleton, set once at boot).
-export { configureBrainUi, uiConfig, type BrainUiConfig } from "./config.js";
+export { configureBrainUi, createBrainUiConfig, uiConfig, type BrainUiConfig } from "./config.js";
 
 // Top-level surfaces the shell composes.
 export { ConnectionGate } from "./components/connectivity/connection-gate.js";
@@ -79,5 +79,9 @@ export {
 export { ShareIntake } from "./components/chat/share-card.js";
 
 // API surface (typed REST client + backend URL helpers).
-export { api } from "./lib/api-client.js";
-export { apiBase, getWsUrl, getBackendUrl } from "./lib/backend.js";
+export { api, createBrainApi, type BrainApi } from "./lib/api-client.js";
+export { apiBase, apiBaseFor, getWsUrl, getWsUrlFor, getBackendUrl, getBackendUrlFor } from "./lib/backend.js";
+
+// Independent UI roots; connection handlers close over the root they belong to.
+export { createBrainUiRoot, type BrainUiRoot, type BrainUiRootOptions } from "./root.js";
+export { BrainUiProvider, useBrainUiRoot, useBrainApi, useBrainConfig } from "./root-context.js";