`useState` implementation with `useReducer`

useState.js

import React from "react";

const useStateReducer = (prevState, dispatchArg) =>
  typeof dispatchArg === "function" ? dispatchArg(prevState) : dispatchArg;

const useStateInitializer = (initialArg) =>
  typeof initialArg === "function" ? initialArg() : initialArg;

function useState(initialValue) {
  return React.useReducer(useStateReducer, initialValue, useStateInitializer);
}

function Counter() {
  const [count, setCount] = useState(0);
  const increment = () => setCount(count + 1);
  // const increment = () => setCount((c) => c + 1);
  // const increment = () => setCount(() => 1);

  return (
    <>
      <div>{count}</div>
      <button onClick={increment}>+1</button>
    </>
  );
}

export default function App() {
  return (
    <div className="App">
      <Counter />
    </div>
  );
}