素数の時にJOJOと出力するKotlin

gistfile1.kt

package jojo

fun main(args: Array<String>) {
  jojoIterator() forEach {println(it)}
}

fun jojoIterator(): java.util.Iterator<Any> {
  fun next(number: Int): Any {
    if (number == 1) {
      number
    } else if (number == 2) {
      "JOJO"
    } else {
      if ((2..Math.sqrt(number.toDouble()).toInt()).any {number % it == 0}) {
        number
      } else {
        "JOJO"
      }
    }
  }
  var number = 1
  return iterate {
    if (number <= 100) {
      next(number++)
    } else {
      null
    }
  }
}