Function useStableKeys

packages/agentflow/src/atoms/useStableKeys.ts:18–45

(length: number, prefix: string)

 * @param prefix  Key prefix string (e.g. `'item'` → `'item-0'`, `'item-1'`, …)
 */
export function useStableKeys(length: number, prefix: string): { keys: string[]; removeKey: (index: number) => void } {
    const idCounterRef = useRef(0)

    const [itemKeys, setItemKeys] = useState<string[]>(() => {
        const initial: string[] = []
        while (initial.length < length) {
            initial.push(`${prefix}-${idCounterRef.current++}`)
        }
        return initial
    })

    let effectiveKeys = itemKeys

    if (effectiveKeys.length < length) {
        const nextKeys = [...effectiveKeys]
        while (nextKeys.length < length) {
            nextKeys.push(`${prefix}-${idCounterRef.current++}`)
        }
        setItemKeys(nextKeys)
        effectiveKeys = nextKeys
    }

    const removeKey = useCallback((index: number) => {
        setItemKeys((prev) => prev.filter((_, i) => i !== index))
    }, [])

    return { keys: effectiveKeys, removeKey }
}