1 files changed, 14 insertions, 4 deletions
diff --git a/app/src/main/kotlin/com/pitchedapps/frost/utils/FragmentUtils.kt b/app/src/main/kotlin/com/pitchedapps/frost/utils/FragmentUtils.kt
index cd638068..e7a38227 100644
--- a/app/src/main/kotlin/com/pitchedapps/frost/utils/FragmentUtils.kt
+++ b/app/src/main/kotlin/com/pitchedapps/frost/utils/FragmentUtils.kt
@@ -2,14 +2,24 @@ package com.pitchedapps.frost.utils
 
 import android.os.Bundle
 import android.support.v4.app.Fragment
+import com.pitchedapps.frost.fragments.BaseFragment
 
 /**
  * Created by Allan Wang on 2017-05-29.
  */
 
-fun Fragment.withBundle(creator: (Bundle) -> Unit): Fragment {
-    val bundle = Bundle()
-    creator.invoke(bundle)
-    this.arguments = bundle
+private fun Fragment.bundle(): Bundle {
+    if (this.arguments == null)
+        this.arguments = Bundle()
+    return this.arguments
+}
+
+fun <T : Fragment> T.putString(key: String, value: String): T {
+    this.bundle().putString(key, value)
+    return this
+}
+
+fun <T : Fragment> T.putInt(key: String, value: Int): T {
+    this.bundle().putInt(key, value)
     return this
 }
 \ No newline at end of file