encapsulated-collections.4 : convert withJourneyAt to extension

Duncan McGregor · dmcg · commit efecaecfc596 · 2021-06-23T22:43:19.000+01:00
diff --git a/src/main/java/travelator/itinerary/Route.kt b/src/main/java/travelator/itinerary/Route.kt
@@ -5,7 +5,7 @@ import java.time.Duration
 import java.util.*
 
 class Route(
-    private val journeys: List<Journey>
+    val journeys: List<Journey>
 ) {
     fun size(): Int = journeys.size
 
@@ -20,11 +20,12 @@ class Route(
             get(size() - 1).arrivalTime
         )
 
-    fun withJourneyAt(index: Int, replacedBy: Journey): Route {
-        val newJourneys = ArrayList(journeys)
-        newJourneys[index] = replacedBy
-        return Route(newJourneys)
-    }
+}
+
+fun Route.withJourneyAt(index: Int, replacedBy: Journey): Route {
+    val newJourneys = ArrayList(journeys)
+    newJourneys[index] = replacedBy
+    return Route(newJourneys)
 }
 
 val Route.departsFrom: Location
