@assistant-ui/react
Version:
Typescript/React library for AI Chat
51 lines • 1.71 kB
JavaScript
"use client";
import { jsx } from "react/jsx-runtime";
import {
createContext,
forwardRef,
useContext,
useState
} from "react";
import { create } from "zustand";
import { useContentPartRuntime } from "../../context/react/ContentPartContext";
import { createContextStoreHook } from "../../context/react/utils/createContextStoreHook";
const SmoothContext = createContext(null);
const makeSmoothContext = (initialState) => {
const useSmoothStatus2 = create(() => initialState);
return { useSmoothStatus: useSmoothStatus2 };
};
const SmoothContextProvider = ({ children }) => {
const outer = useSmoothContext({ optional: true });
const contentPartRuntime = useContentPartRuntime();
const [context] = useState(
() => makeSmoothContext(contentPartRuntime.getState().status)
);
if (outer) return children;
return /* @__PURE__ */ jsx(SmoothContext.Provider, { value: context, children });
};
const withSmoothContextProvider = (Component) => {
const Wrapped = forwardRef((props, ref) => {
return /* @__PURE__ */ jsx(SmoothContextProvider, { children: /* @__PURE__ */ jsx(Component, { ...props, ref }) });
});
Wrapped.displayName = Component.displayName;
return Wrapped;
};
function useSmoothContext(options) {
const context = useContext(SmoothContext);
if (!options?.optional && !context)
throw new Error(
"This component must be used within a SmoothContextProvider."
);
return context;
}
const { useSmoothStatus, useSmoothStatusStore } = createContextStoreHook(
useSmoothContext,
"useSmoothStatus"
);
export {
SmoothContextProvider,
useSmoothStatus,
useSmoothStatusStore,
withSmoothContextProvider
};
//# sourceMappingURL=SmoothContext.js.map