function* counter() {
  let count = 1;
  while (true) {
    const reset = yield count;
    count = reset ? 1 : count + 1;
  }
}

const gen = counter();
console.log(gen.next().value);
console.log(gen.next().value);
console.log(gen.next(true).value);
console.log(gen.next().value);