import { useAtom } from "jotai";

export function useChat() {
  const [messages] = useAtom(messagesAtom);
  const [isPending] = useAtom(isPendingAtom);
  const [input, setInput] = useAtom(inputAtom);
  // ... more atoms

  return {
    messages,
    isPending,
    input,
    setInput,
    // ... more atoms
  };
}

const MessageList = () => {
  // we only need `messages` and `isPending` in this component, but all atoms are mounted and will be re-evaluated every time one of them changes
  const { messages, isPending } = useChat();
  return (
    <div>
      {isPending && <p>Loading...</p>}
      {messages.map((message) => (
        <p key={message.id}>{message.text}</p>
      ))}
    </div>
  );
};

import { useAtom } from "jotai-lazy";

export function useChat() {
  const messagesObject = useAtom(messagesAtom);
  const isPendingObject = useAtom(isPendingAtom);
  const inputObject = useAtom(inputAtom);
  // ... more atoms

  return {
    get messages() {
      return messagesObject[0];
    },
    get isPending() {
      return isPendingObject[0];
    },
    get input() {
      return inputObject[0];
    },
    setInput: inputObject[1],
    // ... more atoms
  };
}

const trackUserAtom = atom(async () => {
  await fetch("https://api.example.com/tracks");
});

const App = () => {
  useAtom(trackUserAtom); // this won't trigger the fetch
  const [] = useAtom(trackUserAtom); // this will not trigger the fetch

  useAtom(trackUserAtom)[0]; // this will trigger the fetch
  const [_] = useAtom(trackUserAtom); // this will trigger the fetch
  const [,] = useAtom(trackUserAtom); // this will trigger the fetch

  return <div>App</div>;
};