fix: fix main files

jetbrains-academy · Nov 6, 2023 · b77a984 · b77a984
1 parent 2aceb19
commit b77a984
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 10 deletions.
diff --git a/...actice/FactoryMethodPattern/src/main/kotlin/jetbrains/refactoring/course/patterns/Main.kt b/...actice/FactoryMethodPattern/src/main/kotlin/jetbrains/refactoring/course/patterns/Main.kt
@@ -1,7 +1,7 @@
 package jetbrains.refactoring.course.patterns
 
-fun main() {
-    val transportName = "car"
+fun main(args: Array<String>) {
+    val transportName = args[0]
     val transport = if (transportName == "car") {
         Car()
     } else if (transportName == "bicycle") {
@@ -10,4 +10,5 @@ fun main() {
         throw IllegalArgumentException("Unknown transport")
     }
     val driveMessage = transport.drive()
+    println(driveMessage)
 }
diff --git a/...yMethodPatternPractice/Task/src/main/kotlin/jetbrains/refactoring/course/patterns/Main.kt b/...yMethodPatternPractice/Task/src/main/kotlin/jetbrains/refactoring/course/patterns/Main.kt
@@ -2,13 +2,6 @@ package jetbrains.refactoring.course.patterns
 
 fun main(args: Array<String>) {
     val transportName = args[0]
-    val transport = if (transportName == "car") {
-        Car()
-    } else if (transportName == "bicycle") {
-        Bicycle()
-    } else {
-        throw IllegalArgumentException("Unknown transport")
-    }
-    val driveMessage = transport.drive()
+    val driveMessage = TransportationServiceFactory().getTransportation(transportName).drive()
     println(driveMessage)
 }