From 8b0d17d5cc9a08886c41fc0115b96b851fc809c9 Mon Sep 17 00:00:00 2001
From: crain <alf_bk@bk.ru>
Date: Wed, 7 Oct 2020 15:37:09 +0500
Subject: [PATCH] Add transform method into ContentEvent

---
 .../main/java/ru/touchin/lifecycle/event/ContentEvent.kt | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/lifecycle/src/main/java/ru/touchin/lifecycle/event/ContentEvent.kt b/lifecycle/src/main/java/ru/touchin/lifecycle/event/ContentEvent.kt
index 8570d10..e25e9fd 100644
--- a/lifecycle/src/main/java/ru/touchin/lifecycle/event/ContentEvent.kt
+++ b/lifecycle/src/main/java/ru/touchin/lifecycle/event/ContentEvent.kt
@@ -10,4 +10,13 @@ sealed class ContentEvent<out T>(open val data: T?) {
 
     data class Complete<out T>(override val data: T? = null) : ContentEvent<T>(data)
 
+    fun <P> transform(transformation: (T?) -> P): ContentEvent<P> {
+        return when(this) {
+            is Loading -> Loading(transformation(data))
+            is Success -> Success(transformation(data))
+            is Complete -> Complete(transformation(data))
+            is Error -> Error(throwable, transformation(data))
+        }
+    }
+
 }