From 70f025899a95d3131856ea8778716d52498f6094 Mon Sep 17 00:00:00 2001
From: Laimonas Turauskas <laimiux@gmail.com>
Date: Mon, 11 Dec 2023 14:26:19 -0800
Subject: [PATCH] Fix firstModel timing when re-creating the view. (#318)

---
 .../instacart/formula/android/FormulaFragment.kt   | 14 ++++++++++++--
 1 file changed, 12 insertions(+), 2 deletions(-)

diff --git a/formula-android/src/main/java/com/instacart/formula/android/FormulaFragment.kt b/formula-android/src/main/java/com/instacart/formula/android/FormulaFragment.kt
index 2ad51618..1ea8f23a 100644
--- a/formula-android/src/main/java/com/instacart/formula/android/FormulaFragment.kt
+++ b/formula-android/src/main/java/com/instacart/formula/android/FormulaFragment.kt
@@ -1,5 +1,6 @@
 package com.instacart.formula.android
 
+import android.content.Context
 import android.os.Bundle
 import android.os.SystemClock
 import android.view.LayoutInflater
@@ -30,7 +31,7 @@ class FormulaFragment : Fragment(), BaseFormulaFragment<Any> {
         requireArguments().getParcelable<FragmentKey>(ARG_CONTRACT)!!
     }
 
-    private val initializedAtMillis = SystemClock.uptimeMillis()
+    private var initializedAtMillis: Long? = SystemClock.uptimeMillis()
 
     private var featureView: FeatureView<Any>? = null
     private val stateRelay: BehaviorRelay<Any> = BehaviorRelay.create()
@@ -38,6 +39,13 @@ class FormulaFragment : Fragment(), BaseFormulaFragment<Any> {
 
     private var lifecycleCallback: FragmentLifecycleCallback? = null
 
+    override fun onAttach(context: Context) {
+        super.onAttach(context)
+        if (initializedAtMillis == null) {
+            initializedAtMillis = SystemClock.uptimeMillis()
+        }
+    }
+
     override fun onCreateView(inflater: LayoutInflater, container: ViewGroup?, savedInstanceState: Bundle?): View? {
         val viewFactory = FormulaFragmentDelegate.viewFactory(this) ?: run {
             // No view factory, no view
@@ -53,7 +61,7 @@ class FormulaFragment : Fragment(), BaseFormulaFragment<Any> {
         super.onViewCreated(view, savedInstanceState)
         featureView?.let { value ->
             val state = FeatureView.State(
-                initializedAtMillis = initializedAtMillis,
+                initializedAtMillis = initializedAtMillis ?: SystemClock.uptimeMillis(),
                 fragmentId = getFormulaFragmentId(),
                 environment = FormulaFragmentDelegate.fragmentEnvironment(),
                 observable = stateRelay,
@@ -100,6 +108,8 @@ class FormulaFragment : Fragment(), BaseFormulaFragment<Any> {
     }
 
     override fun onDestroyView() {
+        initializedAtMillis = null
+
         cancelable?.cancel()
         cancelable = null