hooks.js

const augment = callback => () => {
  const useState = {
    init: true,
    value: void 0,
    update(value) {
      useState.value = value;
      callback(hooks)
    }
  };
  const hooks = {
    useState(value) {
      if (useState.init) {
        useState.init = false;
        useState.value = value;
      }
      return [useState.value, useState.update];
    }
  };
  callback(hooks);
};

const counter = augment(({useState}) => {
  const [i, update] = useState(0);
  console.log(i);
  setTimeout(update, 1000, i + 1);
});

// once invoked, it should log the `i` from 0 to N
// each second the `i` should be incremented + 1
counter();

// 0
// 1
// 2
// ...