From 7d377614cb10a067b8c334f8ec2dbeb29169482f Mon Sep 17 00:00:00 2001
From: Unknown <pilotofsparrow@gmail.com>
Date: Fri, 13 Jul 2018 12:39:02 +0300
Subject: [PATCH] Added onViewStateRestored callback to viewcontroller

---
 .../navigation/fragments/ViewControllerFragment.java     | 8 ++++++++
 .../navigation/viewcontrollers/ViewController.java       | 9 +++++++++
 2 files changed, 17 insertions(+)

diff --git a/src/main/java/ru/touchin/roboswag/components/navigation/fragments/ViewControllerFragment.java b/src/main/java/ru/touchin/roboswag/components/navigation/fragments/ViewControllerFragment.java
index e334a04..aaa5496 100644
--- a/src/main/java/ru/touchin/roboswag/components/navigation/fragments/ViewControllerFragment.java
+++ b/src/main/java/ru/touchin/roboswag/components/navigation/fragments/ViewControllerFragment.java
@@ -234,6 +234,14 @@ public class ViewControllerFragment<TActivity extends FragmentActivity, TState e
         }
     }
 
+    @Override
+    public void onViewStateRestored(@Nullable Bundle savedInstanceState) {
+        super.onViewStateRestored(savedInstanceState);
+        if (viewController != null) {
+            viewController.onViewStateRestored(savedInstanceState);
+        }
+    }
+
     @Override
     protected void onStart(@NonNull final View view, @NonNull final TActivity activity) {
         super.onStart(view, activity);
diff --git a/src/main/java/ru/touchin/roboswag/components/navigation/viewcontrollers/ViewController.java b/src/main/java/ru/touchin/roboswag/components/navigation/viewcontrollers/ViewController.java
index 436d9fa..6912025 100644
--- a/src/main/java/ru/touchin/roboswag/components/navigation/viewcontrollers/ViewController.java
+++ b/src/main/java/ru/touchin/roboswag/components/navigation/viewcontrollers/ViewController.java
@@ -320,6 +320,15 @@ public class ViewController<TActivity extends FragmentActivity, TState extends P
         return null;
     }
 
+    /**
+     * Calls when {@link ViewController} saved state has been restored into the view hierarchy.
+     * Happens at {@link ViewControllerFragment#onViewStateRestored}.
+     */
+    @CallSuper
+    public void onViewStateRestored(@Nullable Bundle savedInstanceState) {
+        // do nothing
+    }
+
     /**
      * Calls when {@link ViewController} have started.
      * Happens at {@link ViewControllerFragment#onStart()}.