function* counter() {
  let count = 1;
  while (true) {
    const reset = yield count++;
    if (reset) {
      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);