jotai/esm/zustand.js

import { atom } from 'jotai';

function atomWithStore(store) {
  const baseAtom = atom(store.getState());
  baseAtom.onMount = (setValue) => {
    const callback = () => {
      setValue(store.getState());
    };
    const unsub = store.subscribe(callback);
    callback();
    return unsub;
  };
  const derivedAtom = atom(
    (get) => get(baseAtom),
    (get, _set, update) => {
      const newState = typeof update === "function" ? update(get(baseAtom)) : update;
      store.setState(newState, true);
    }
  );
  return derivedAtom;
}

export { atomWithStore };