tp2 step 3 solution

Author: ctruchi

tp2/src/main/kotlin/fmt/kotlin/fundamentals/Tp2.kt

@@ -1,7 +1,7 @@
 package fmt.kotlin.fundamentals
 
 fun main() {
-    println(getFirstPrimeNumbers())
+    println(getFirstPrimeNumbers(1, 2, 3))
 }
 
 private const val MIN = 0
@@ -13,10 +13,22 @@ private const val prefix = "["
 private const val separator = ", "
 private const val suffix = "]"
 
-fun getFirstPrimeNumbers(nbToFind: Int = 10): String = when {
-    nbToFind <= MIN -> ERROR_MIN
-    nbToFind > MAX -> ERROR_MAX
-    else -> computeFirstPrimeNumbers(nbToFind)
+
+fun getFirstPrimeNumbers(vararg nbToFindBatches: Int): String {
+    var firstLine = true
+    var result = ""
+    for (nbToFind in nbToFindBatches) {
+        if (!firstLine) {
+            result += "\n"
+        }
+        result += when {
+            nbToFind <= MIN -> ERROR_MIN
+            nbToFind > MAX -> ERROR_MAX
+            else -> computeFirstPrimeNumbers(nbToFind)
+        }
+        firstLine = false
+    }
+    return result
 }
 
 private fun computeFirstPrimeNumbers(nbToFind: Int = 10): String {