react-native-navigation/lib/android/app/src/main/java/com/reactnativenavigation/views/element/ElementTransitionManager.kt

package com.reactnativenavigation.views.element

import android.animation.AnimatorSet
import android.view.View
import androidx.core.view.doOnLayout
import com.facebook.react.uimanager.util.ReactFindViewUtil.OnViewFoundListener
import com.facebook.react.uimanager.util.ReactFindViewUtil.findView
import com.reactnativenavigation.parse.AnimationOptions
import com.reactnativenavigation.parse.NestedAnimationsOptions
import com.reactnativenavigation.utils.Functions.Func1
import com.reactnativenavigation.viewcontrollers.ViewController

open class ElementTransitionManager {
    private val animatorCreator: TransitionAnimatorCreator = TransitionAnimatorCreator()

    fun createTransitions(animation: NestedAnimationsOptions, fromScreen: ViewController<*>, toScreen: ViewController<*>, onAnimatorsCreated: Func1<TransitionSet?>) {
        val sharedElements = animation.sharedElements
        val elementTransitions = animation.elementTransitions
        if (!sharedElements.hasValue() && !elementTransitions.hasValue) {
            onAnimatorsCreated.run(TransitionSet())
            return
        }
        val transitionSet = TransitionSet()
        for (transitionOptions in sharedElements.get()) {
            val transition = SharedElementTransition(toScreen, transitionOptions!!)
            findView(fromScreen.view, transition.fromId)?.let { transition.from = it }
            findView(toScreen.view, object : OnViewFoundListener {
                override fun getNativeId(): String {
                    return transition.toId
                }

                override fun onViewFound(view: View) {
                    view.doOnLayout {
                        transition.to = view
                        if (transition.isValid()) transitionSet.add(transition)
                        if (transitionSet.size() == sharedElements.get().size + elementTransitions.transitions.size) {
                            onAnimatorsCreated.run(transitionSet)
                        }
                    }
                }
            })
        }
        for (transitionOptions in elementTransitions.transitions) {
            val transition = ElementTransition(transitionOptions)
            findView(fromScreen.view, transition.id)?.let {
                transition.view = it
                transition.viewController = fromScreen
                transitionSet.add(transition)
            }
            if (transition.isValid()) continue
            findView(toScreen.view, object : OnViewFoundListener {
                override fun getNativeId(): String {
                    return transition.id
                }

                override fun onViewFound(view: View) {
                    view.doOnLayout {
                        transition.view = view
                        transition.viewController = toScreen
                        transitionSet.add(transition)
                        if (transitionSet.size() == sharedElements.get().size + elementTransitions.transitions.size) {
                            onAnimatorsCreated.run(transitionSet)
                        }
                    }
                }
            })
        }
    }

    fun createAnimators(fadeAnimation: AnimationOptions?, transitionSet: TransitionSet?): AnimatorSet {
        return animatorCreator.create(fadeAnimation!!, transitionSet!!)
    }

}