import { useCallback } from "react";
import { useOpenAIGlobal } from "./useOpenAIGlobal";
import type { SetStateAction } from "react";
// Implementation
export function useWidgetState<T extends Record<string, unknown>>(): readonly [
T | null | undefined,
(state: SetStateAction<T | null | undefined>) => void,
] {
// 1. Get the live state and the global setter function
const currentState = useOpenAIGlobal<{ State: T }, "widgetState">(
"widgetState"
);
const globalSetter = useOpenAIGlobal<{ State: T }, "setWidgetState">(
"setWidgetState"
);
// 2. Determine the state to return, respecting precedence
// (Global state > default state)
// 3. Create a stable, React-like setter function
const setState = useCallback(
(action: SetStateAction<T | null | undefined>) => {
if (!globalSetter) {
console.warn("setWidgetState is not available on window.openai");
return;
}
// Calculate the new state, supporting updater functions
const newState =
typeof action === "function"
? action(currentState) // Pass the current state to the updater
: action;
if (newState != null) {
globalSetter(newState).catch((err) =>
console.warn("setWidgetState could not be set on window.openai", err)
);
}
},
[globalSetter, currentState]
); // `state` is a dependency to correctly resolve updater functions
return [currentState, setState] as const;
}
