drive/apps/drive-web/src/components/with-atom.tsx

import { type PrimitiveAtom, useAtom } from "jotai"

export function WithAtom<Value>({
	atom,
	children,
}: {
	atom: PrimitiveAtom<Value>
	children: (
		value: Value,
		setValue: (value: Value | ((current: Value) => Value)) => void,
	) => React.ReactNode
}) {
	const [value, setValue] = useAtom(atom)
	return children(value, setValue)
}