package com.navigation.reactnative;

import android.animation.Animator;
import android.animation.AnimatorSet;
import android.animation.ObjectAnimator;
import android.content.Context;
import android.os.Bundle;
import android.view.LayoutInflater;
import android.view.View;
import android.view.ViewGroup;
import android.widget.LinearLayout;

import androidx.annotation.NonNull;
import androidx.annotation.Nullable;
import androidx.fragment.app.Fragment;
import androidx.transition.Transition;

import com.facebook.react.bridge.ReadableArray;
import com.facebook.react.bridge.ReadableMap;
import com.google.android.material.transition.Hold;
import com.google.android.material.transition.MaterialElevationScale;
import com.google.android.material.transition.MaterialFade;
import com.google.android.material.transition.MaterialFadeThrough;
import com.google.android.material.transition.MaterialSharedAxis;

import java.util.HashMap;
import java.util.HashSet;
import java.util.Map;
import java.util.concurrent.TimeUnit;

public class SceneFragment extends Fragment {
    private SceneView scene;
    protected AnimationPropParser.Animator enterAnimator;
    protected AnimationPropParser.Animator returnAnimator;
    protected AnimationPropParser.Animator exitAnimator;
    protected AnimationPropParser.Animator reenterAnimator;

    public SceneFragment() {
        super();
    }

    SceneFragment(SceneView scene, HashSet<String> sharedElements, boolean containerTransform) {
        super();
        this.scene = scene;
        if (sharedElements != null )
            scene.sharedElementMotion = new SharedElementMotion(this, this, sharedElements, containerTransform);
    }

    @Nullable
    @Override
    public View onCreateView(@NonNull LayoutInflater inflater, @Nullable ViewGroup container, @Nullable Bundle savedInstanceState) {
        if (scene != null) {
            SceneFragmentView fragmentView = new SceneFragmentView(getContext());
            if (scene.getParent() != null) {
                ((ViewGroup) scene.getParent()).removeView(scene);
            }
            fragmentView.addView(scene);
            if (scene.sharedElementMotion != null)
                postponeEnterTransition(300, TimeUnit.MILLISECONDS);
            return fragmentView;
        }
        return new View(getContext());
    }

    @Nullable
    @Override
    public Animator onCreateAnimator(int transit, boolean enter, int nextAnim) {
        if (nextAnim == 0 && enterAnimator != null && enter) return transform(enterAnimator, true);
        if (nextAnim == -1 && reenterAnimator != null && enter) return transform(reenterAnimator, true);
        if (nextAnim == 0 && exitAnimator != null && !enter) return transform(exitAnimator, false);
        if (nextAnim == -1 && returnAnimator != null && !enter) return transform(returnAnimator, false);
        return super.onCreateAnimator(transit, enter, nextAnim);
    }

    private Animator transform(AnimationPropParser.Animator anim, boolean from) {
        AnimatorSet animatorSet = new AnimatorSet();
        if (anim.duration != null)
            animatorSet.setDuration(anim.duration);
        for (int i = 0; i < anim.items.length; i++) {
            AnimationPropParser.AnimatorItem item = anim.items[i];
            if ("translate".equals(item.type)) {
                ObjectAnimator xAnimator = new ObjectAnimator();
                xAnimator.setPropertyName("translationX");
                float xVal = item.x.second ? scene.getWidth() * item.x.first / 100 : item.x.first;
                xAnimator.setFloatValues(from ? xVal : 0, from ? 0 : xVal);
                xAnimator.setDuration(item.duration != null ? item.duration : 300);
                ObjectAnimator yAnimator = new ObjectAnimator();
                yAnimator.setPropertyName("translationY");
                float yVal = item.y.second ? scene.getHeight() * item.y.first / 100 : item.y.first;
                yAnimator.setFloatValues(from ? yVal : 0, from ? 0 : yVal);
                yAnimator.setDuration(item.duration != null ? item.duration : 300);
                animatorSet.playTogether(xAnimator, yAnimator);
            }
            if ("scale".equals(item.type)) {
                ObjectAnimator xAnimator = new ObjectAnimator();
                xAnimator.setPropertyName("scaleX");
                float xVal = item.x.second ? item.x.first / 100 : item.x.first;
                xAnimator.setFloatValues(from ? xVal : 1, from ? 1 : xVal);
                xAnimator.setDuration(item.duration != null ? item.duration : 300);
                ObjectAnimator yAnimator = new ObjectAnimator();
                yAnimator.setPropertyName("scaleY");
                float yVal = item.y.second ? item.y.first / 100 : item.y.first;
                yAnimator.setFloatValues(from ? yVal : 1, from ? 1 : yVal);
                yAnimator.setDuration(item.duration != null ? item.duration : 300);
                animatorSet.playTogether(xAnimator, yAnimator);
            }
            if ("alpha".equals(item.type)) {
                ObjectAnimator alphaAnimator = new ObjectAnimator();
                alphaAnimator.setPropertyName("alpha");
                alphaAnimator.setFloatValues(from ? item.x.first : 1, from ? 1 : item.x.first);
                alphaAnimator.setDuration(item.duration != null ? item.duration : 300);
                animatorSet.playTogether(alphaAnimator);
            }
            if ("rotate".equals(item.type)) {
                ObjectAnimator rotateAnimator = new ObjectAnimator();
                rotateAnimator.setPropertyName("rotation");
                rotateAnimator.setFloatValues(from ? item.x.first : 0, from ? 0 : item.x.first);
                rotateAnimator.setDuration(item.duration != null ? item.duration : 300);
                animatorSet.playTogether(rotateAnimator);
            }
        }
        return animatorSet;
    }

    @Override
    public void setEnterTransition(@Nullable Object transition) {
        transition = getTransition((ReadableMap) transition);
        super.setEnterTransition(transition);
    }

    @Override
    public void setExitTransition(@Nullable Object transition) {
        transition = getTransition((ReadableMap) transition);
        super.setExitTransition(transition);
    }

    @Override
    public void setReenterTransition(@Nullable Object transition) {
        transition = getTransition((ReadableMap) transition);
        super.setReenterTransition(transition);
    }

    @Override
    public void setReturnTransition(@Nullable Object transition) {
        transition = getTransition((ReadableMap) transition);
        super.setReturnTransition(transition);
    }

    private Transition getTransition(ReadableMap trans) {
        Transition transition = null;
        if (trans == null) return null;
        ReadableArray items = trans.hasKey("items") ? trans.getArray("items") : null;
        if (items == null || items.size() == 0) return null;
        trans = items.getMap(0);
        String transType = trans != null ? trans.getString("type") : null;
        if (transType == null) return null;
        switch (transType) {
            case "sharedAxis" -> {
                Map<String, Integer> axisMap = new HashMap<>();
                axisMap.put("x", MaterialSharedAxis.X);
                axisMap.put("y", MaterialSharedAxis.Y);
                Integer axis = axisMap.get(trans.getString("axis"));
                transition = new MaterialSharedAxis(axis != null ? axis : MaterialSharedAxis.Z, true);
            }
            case "elevationScale" -> transition = new MaterialElevationScale(true);
            case "fade" -> transition = new MaterialFade();
            case "fadeThrough" -> transition = new MaterialFadeThrough();
            case "hold" -> transition = new Hold();
        }
        return transition;
    }

    @Override
    public void onDestroy() {
        super.onDestroy();
        if (scene != null)
            scene.popped();
    }

    public SceneView getScene() {
        return scene;
    }

    public static class SceneFragmentView extends LinearLayout {
        private final OnLayoutChangeListener onLayoutChangeListener = (v, left, top, right, bottom, oldLeft, oldTop, oldRight, oldBottom) -> {
            ViewGroup stack = (ViewGroup) getParent();
            measure(
                MeasureSpec.makeMeasureSpec(stack.getWidth(), MeasureSpec.EXACTLY),
                MeasureSpec.makeMeasureSpec(stack.getHeight(), MeasureSpec.EXACTLY));
            layout(0, 0, stack.getWidth(), stack.getHeight());
        };

        public SceneFragmentView(Context context) {
            super(context);
        }

        @Override
        protected void onAttachedToWindow() {
            super.onAttachedToWindow();
            ((ViewGroup) getParent()).addOnLayoutChangeListener(onLayoutChangeListener);
        }

        @Override
        protected void onDetachedFromWindow() {
            super.onDetachedFromWindow();
            ((ViewGroup) getParent()).removeOnLayoutChangeListener(onLayoutChangeListener);
        }
    }
}