draft

Author: mtumilowicz

src/main/kotlin/dsl/FsmSpec.kt

@@ -5,7 +5,7 @@ class FsmSpec(
     val initialState: String
 ) {
 
-    fun add(transitionRecipe: TransitionSpec.() -> TransitionSpec): FsmSpec {
+    fun add(transitionRecipe: TransitionSpec.() -> Unit): FsmSpec {
         transitions.add(Transition.create(transitionRecipe))
         return this
     }

src/main/kotlin/dsl/Transition.kt

@@ -3,9 +3,8 @@ package dsl
 data class Transition(val event: Event, val stateFlow: StateFlow) {
 
     companion object {
-        fun create(transitionSpec: TransitionSpec.() -> TransitionSpec): Transition {
-            return TransitionSpec()
-                .transitionSpec()
+        fun create(transitionSpec: TransitionSpec.() -> Unit): Transition {
+            return TransitionSpec().apply(transitionSpec)
                 .build()
         }
     }

src/main/kotlin/dsl/TransitionSpec.kt

@@ -6,19 +6,16 @@ class TransitionSpec(
     private var stateTo: String = ""
 ) {
 
-    fun on(event: String): TransitionSpec {
+    fun on(event: String) {
         this.event = event
-        return this
     }
 
-    fun from(state: String): TransitionSpec {
+    fun from(state: String) {
         stateFrom = state
-        return this
     }
 
-    fun into(state: String): TransitionSpec {
+    fun into(state: String) {
         stateTo = state
-        return this
     }
 
     fun build(): Transition {

src/test/kotlin/dsl/FsmTest.kt

@@ -9,8 +9,16 @@ class FsmTest {
     fun a() {
 //        when:
         val fsm = Fsm.create(initialState = "locked") {
-            add { on("coin").from("locked").into("unlocked") }
-            add { on("pass").from("unlocked").into("locked") }
+            add {
+                on("coin")
+                from("locked")
+                into("unlocked")
+            }
+            add {
+                on("pass")
+                from("unlocked")
+                into("locked")
+            }
         }
 
 //        then: