FS2 gist to pull from a stream until N items that return true for the provided predicate

FS2TakeWhileRepeat.scala

   def takeWhileRepeat[F[_],O](n: Long, f: O => Boolean): Pipe[F,O,O] = {

      def go(s: Stream[F,O], wasTrueCount : Int) : Pull[F,O,Unit] = {

        if(wasTrueCount == n) {
          Pull.done
        }
        else {
          s.pull.uncons1.flatMap {

            case Some((o, tl)) =>

              if(f(o)) {
                Pull.output1(o) >> go(tl, wasTrueCount + 1)
              }
              else {
                Pull.output1(o) >> go(tl, wasTrueCount)
              }

            case None =>
              Pull.done

          }
        }
      }

      in => go(in, 0).stream

    }