package com.swmansion.reanimated.transitions;

import android.animation.Animator;
import android.animation.AnimatorSet;
import android.animation.ObjectAnimator;
import android.view.View;
import android.view.ViewGroup;
import androidx.transition.Transition;
import androidx.transition.TransitionListenerAdapter;
import androidx.transition.TransitionValues;
import androidx.transition.Visibility;

public class Scale extends Visibility {

  static final String PROPNAME_SCALE_X = "scale:scaleX";
  static final String PROPNAME_SCALE_Y = "scale:scaleY";

  @Override
  public void captureStartValues(TransitionValues transitionValues) {
    super.captureStartValues(transitionValues);
    transitionValues.values.put(PROPNAME_SCALE_X, transitionValues.view.getScaleX());
    transitionValues.values.put(PROPNAME_SCALE_Y, transitionValues.view.getScaleY());
  }

  public Scale setDisappearedScale(float disappearedScale) {
    if (disappearedScale < 0f) {
      throw new IllegalArgumentException("disappearedScale cannot be negative!");
    }
    return this;
  }

  private Animator createAnimation(
      final View view, float startScale, float endScale, TransitionValues values) {
    final float initialScaleX = view.getScaleX();
    final float initialScaleY = view.getScaleY();
    float startScaleX = initialScaleX * startScale;
    float endScaleX = initialScaleX * endScale;
    float startScaleY = initialScaleY * startScale;
    float endScaleY = initialScaleY * endScale;

    if (values != null) {
      Float savedScaleX = (Float) values.values.get(PROPNAME_SCALE_X);
      Float savedScaleY = (Float) values.values.get(PROPNAME_SCALE_Y);
      // if saved value is not equal initial value it means that previous
      // transition was interrupted and in the onTransitionEnd
      // we've applied endScale. we should apply proper value to
      // continue animation from the interrupted state
      if (savedScaleX != null && savedScaleX != initialScaleX) {
        startScaleX = savedScaleX;
      }
      if (savedScaleY != null && savedScaleY != initialScaleY) {
        startScaleY = savedScaleY;
      }
    }

    view.setScaleX(startScaleX);
    view.setScaleY(startScaleY);

    AnimatorSet animator = new AnimatorSet();
    animator.playTogether(
        ObjectAnimator.ofFloat(view, View.SCALE_X, startScaleX, endScaleX),
        ObjectAnimator.ofFloat(view, View.SCALE_Y, startScaleY, endScaleY));
    addListener(
        new TransitionListenerAdapter() {
          @Override
          public void onTransitionEnd(Transition transition) {
            view.setScaleX(initialScaleX);
            view.setScaleY(initialScaleY);
            transition.removeListener(this);
          }
        });
    return animator;
  }

  @Override
  public Animator onAppear(
      ViewGroup sceneRoot, View view, TransitionValues startValues, TransitionValues endValues) {
    return createAnimation(view, 0f, 1f, startValues);
  }

  @Override
  public Animator onDisappear(
      ViewGroup sceneRoot, View view, TransitionValues startValues, TransitionValues endValues) {
    return createAnimation(view, 1f, 0f, startValues);
  }
}