From 20b17729b2f1e43eea9fd82e8398326c53b9d78b Mon Sep 17 00:00:00 2001
From: Ubiratan Soares <ubiratan.f.soares@gmail.com>
Date: Wed, 24 Jul 2019 01:18:50 +0200
Subject: [PATCH] Improving design

---
 .../norris/architecture/StateTransition.kt      | 17 ++++++++++++++---
 1 file changed, 14 insertions(+), 3 deletions(-)

diff --git a/unidirectional-dataflow/src/main/java/io/dotanuki/norris/architecture/StateTransition.kt b/unidirectional-dataflow/src/main/java/io/dotanuki/norris/architecture/StateTransition.kt
index a2cc9dff..05d1b2b5 100644
--- a/unidirectional-dataflow/src/main/java/io/dotanuki/norris/architecture/StateTransition.kt
+++ b/unidirectional-dataflow/src/main/java/io/dotanuki/norris/architecture/StateTransition.kt
@@ -4,11 +4,22 @@ sealed class StateTransition<out T> {
 
     interface Parameters
 
-    class Unparametrized<T>(val task: suspend () -> T) : StateTransition<T>()
-    class Parametrized<T>(val task: suspend (Parameters) -> T, val parameters: Parameters) : StateTransition<T>()
+    class Unparametrized<T> internal constructor(
+        val task: suspend () -> T
+    ) : StateTransition<T>()
+
+    class Parametrized<T> internal constructor(
+        val task: suspend (Parameters) -> T,
+        val parameters: Parameters
+    ) : StateTransition<T>()
 
     companion object {
+
         operator fun <T> invoke(task: suspend () -> T) = Unparametrized(task)
-        operator fun <T> invoke(task: suspend (Parameters) -> T, data: Parameters) = Parametrized(task, data)
+
+        operator fun <T> invoke(
+            task: suspend (Parameters) -> T,
+            data: Parameters
+        ) = Parametrized(task, data)
     }
 }
\ No newline at end of file