/*
 * Copyright 2023 The Android Open Source Project
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package com.google.android.material.motion;

import com.google.android.material.R;

import static androidx.annotation.RestrictTo.Scope.LIBRARY_GROUP;
import static com.google.android.material.animation.AnimationUtils.lerp;
import static java.lang.Math.max;
import static java.lang.Math.min;

import android.animation.Animator;
import android.animation.AnimatorListenerAdapter;
import android.animation.AnimatorSet;
import android.animation.ObjectAnimator;
import android.animation.ValueAnimator;
import android.content.res.Resources;
import android.graphics.Rect;
import android.os.Build.VERSION;
import android.os.Build.VERSION_CODES;
import android.util.DisplayMetrics;
import android.view.RoundedCorner;
import android.view.View;
import android.view.WindowInsets;
import androidx.activity.BackEventCompat;
import androidx.annotation.NonNull;
import androidx.annotation.Nullable;
import androidx.annotation.RequiresApi;
import androidx.annotation.RestrictTo;
import androidx.annotation.VisibleForTesting;
import com.google.android.material.animation.AnimationUtils;
import com.google.android.material.internal.ClippableRoundedCornerLayout;
import com.google.android.material.internal.ViewUtils;

/**
 * Utility class for main container views usually filling the entire screen (e.g., search view) that
 * support back progress animations.
 *
 * @hide
 */
@RestrictTo(LIBRARY_GROUP)
public class MaterialMainContainerBackHelper extends MaterialBackAnimationHelper<View> {

  private static final float MIN_SCALE = 0.9f;

  private final float minEdgeGap;
  private final float maxTranslationY;

  private float initialTouchY;
  @Nullable private Rect initialHideToClipBounds;
  @Nullable private Rect initialHideFromClipBounds;
  @Nullable private float[] expandedCornerRadii;

  public MaterialMainContainerBackHelper(@NonNull View view) {
    super(view);

    Resources resources = view.getResources();
    minEdgeGap = resources.getDimension(R.dimen.m3_back_progress_main_container_min_edge_gap);
    maxTranslationY =
        resources.getDimension(R.dimen.m3_back_progress_main_container_max_translation_y);
  }

  @Nullable
  public Rect getInitialHideToClipBounds() {
    return initialHideToClipBounds;
  }

  @Nullable
  public Rect getInitialHideFromClipBounds() {
    return initialHideFromClipBounds;
  }

  public void startBackProgress(@NonNull BackEventCompat backEvent, @Nullable View collapsedView) {
    super.onStartBackProgress(backEvent);

    startBackProgress(backEvent.getTouchY(), collapsedView);
  }

  @VisibleForTesting
  public void startBackProgress(float touchY, @Nullable View collapsedView) {
    initialHideToClipBounds = ViewUtils.calculateRectFromBounds(view);
    if (collapsedView != null) {
      initialHideFromClipBounds = ViewUtils.calculateOffsetRectFromBounds(view, collapsedView);
    }
    initialTouchY = touchY;
  }

  public void updateBackProgress(
      @NonNull BackEventCompat backEvent, @Nullable View collapsedView, float collapsedCornerSize) {
    if (super.onUpdateBackProgress(backEvent) == null) {
      return;
    }

    if (collapsedView != null && collapsedView.getVisibility() != View.INVISIBLE) {
      collapsedView.setVisibility(View.INVISIBLE);
    }

    boolean leftSwipeEdge = backEvent.getSwipeEdge() == BackEventCompat.EDGE_LEFT;
    updateBackProgress(
        backEvent.getProgress(), leftSwipeEdge, backEvent.getTouchY(), collapsedCornerSize);
  }

  @VisibleForTesting
  public void updateBackProgress(
      float progress, boolean leftSwipeEdge, float touchY, float collapsedCornerSize) {
    progress = interpolateProgress(progress);

    float width = view.getWidth();
    float height = view.getHeight();
    if (width <= 0f || height <= 0f) {
      return;
    }

    float scale = lerp(1, MIN_SCALE, progress);

    float availableHorizontalSpace = max(0, (width - MIN_SCALE * width) / 2 - minEdgeGap);
    float translationX = lerp(0, availableHorizontalSpace, progress) * (leftSwipeEdge ? 1 : -1);

    float availableVerticalSpace = max(0, (height - scale * height) / 2 - minEdgeGap);
    float maxTranslationY = min(availableVerticalSpace, this.maxTranslationY);
    float yDelta = touchY - initialTouchY;
    float yProgress = Math.abs(yDelta) / height;
    float translationYDirection = Math.signum(yDelta);
    float translationY = AnimationUtils.lerp(0, maxTranslationY, yProgress) * translationYDirection;

    if (Float.isNaN(scale) || Float.isNaN(translationX) || Float.isNaN(translationY)) {
      return;
    }

    view.setScaleX(scale);
    view.setScaleY(scale);
    view.setTranslationX(translationX);
    view.setTranslationY(translationY);
    if (view instanceof ClippableRoundedCornerLayout) {
      ((ClippableRoundedCornerLayout) view)
          .updateCornerRadii(
              lerpCornerRadii(getExpandedCornerRadii(), collapsedCornerSize, progress));
    }
  }

  public void finishBackProgress(long duration, @Nullable View collapsedView) {
    AnimatorSet resetAnimator = createResetScaleAndTranslationAnimator(collapsedView);
    resetAnimator.setDuration(duration);
    resetAnimator.start();

    resetInitialValues();
  }

  public void cancelBackProgress(@Nullable View collapsedView) {
    if (super.onCancelBackProgress() == null) {
      return;
    }

    AnimatorSet cancelAnimatorSet = createResetScaleAndTranslationAnimator(collapsedView);
    if (view instanceof ClippableRoundedCornerLayout) {
      cancelAnimatorSet.playTogether(createCornerAnimator((ClippableRoundedCornerLayout) view));
    }
    cancelAnimatorSet.setDuration(cancelDuration);
    cancelAnimatorSet.start();

    resetInitialValues();
  }

  private void resetInitialValues() {
    initialTouchY = 0f;
    initialHideToClipBounds = null;
    initialHideFromClipBounds = null;
  }

  @NonNull
  private AnimatorSet createResetScaleAndTranslationAnimator(@Nullable View collapsedView) {
    AnimatorSet animatorSet = new AnimatorSet();
    animatorSet.playTogether(
        ObjectAnimator.ofFloat(view, View.SCALE_X, 1),
        ObjectAnimator.ofFloat(view, View.SCALE_Y, 1),
        ObjectAnimator.ofFloat(view, View.TRANSLATION_X, 0),
        ObjectAnimator.ofFloat(view, View.TRANSLATION_Y, 0));
    animatorSet.addListener(
        new AnimatorListenerAdapter() {
          @Override
          public void onAnimationEnd(Animator animation) {
            if (collapsedView != null) {
              collapsedView.setVisibility(View.VISIBLE);
            }
          }
        });
    return animatorSet;
  }

  @NonNull
  private ValueAnimator createCornerAnimator(
      ClippableRoundedCornerLayout clippableRoundedCornerLayout) {
    ValueAnimator cornerAnimator =
        ValueAnimator.ofObject(
            (fraction, startValue, endValue) ->
                lerpCornerRadii((float[]) startValue, (float[]) endValue, fraction),
            clippableRoundedCornerLayout.getCornerRadii(),
            getExpandedCornerRadii());
    cornerAnimator.addUpdateListener(
        animation ->
            clippableRoundedCornerLayout.updateCornerRadii((float[]) animation.getAnimatedValue()));
    return cornerAnimator;
  }

  private static float[] lerpCornerRadii(float[] startValue, float[] endValue, float fraction) {
    return new float[] {
      lerp(startValue[0], endValue[0], fraction),
      lerp(startValue[1], endValue[1], fraction),
      lerp(startValue[2], endValue[2], fraction),
      lerp(startValue[3], endValue[3], fraction),
      lerp(startValue[4], endValue[4], fraction),
      lerp(startValue[5], endValue[5], fraction),
      lerp(startValue[6], endValue[6], fraction),
      lerp(startValue[7], endValue[7], fraction)
    };
  }

  private static float[] lerpCornerRadii(float[] startValue, float endValue, float fraction) {
    return new float[] {
      lerp(startValue[0], endValue, fraction),
      lerp(startValue[1], endValue, fraction),
      lerp(startValue[2], endValue, fraction),
      lerp(startValue[3], endValue, fraction),
      lerp(startValue[4], endValue, fraction),
      lerp(startValue[5], endValue, fraction),
      lerp(startValue[6], endValue, fraction),
      lerp(startValue[7], endValue, fraction)
    };
  }

  @NonNull
  public float[] getExpandedCornerRadii() {
    if (expandedCornerRadii == null) {
      expandedCornerRadii = calculateExpandedCornerRadii();
    }
    return expandedCornerRadii;
  }

  public void clearExpandedCornerRadii() {
    expandedCornerRadii = null;
  }

  private float[] calculateExpandedCornerRadii() {
    if (VERSION.SDK_INT >= VERSION_CODES.S) {
      final WindowInsets insets = view.getRootWindowInsets();
      if (insets != null) {
        DisplayMetrics displayMetrics = view.getResources().getDisplayMetrics();
        int screenWidth = displayMetrics.widthPixels;
        int screenHeight = displayMetrics.heightPixels;

        int[] location = new int[2];
        view.getLocationOnScreen(location);
        int x = location[0];
        int y = location[1];

        int width = view.getWidth();
        int height = view.getHeight();

        int topLeft =
            x == 0 && y == 0 ? getRoundedCornerRadius(insets, RoundedCorner.POSITION_TOP_LEFT) : 0;
        int topRight =
            x + width >= screenWidth && y == 0
                ? getRoundedCornerRadius(insets, RoundedCorner.POSITION_TOP_RIGHT)
                : 0;
        int bottomRight =
            x + width >= screenWidth && y + height >= screenHeight
                ? getRoundedCornerRadius(insets, RoundedCorner.POSITION_BOTTOM_RIGHT)
                : 0;
        int bottomLeft =
            x == 0 && y + height >= screenHeight
                ? getRoundedCornerRadius(insets, RoundedCorner.POSITION_BOTTOM_LEFT)
                : 0;

        return new float[] {
          topLeft, topLeft, topRight, topRight, bottomRight, bottomRight, bottomLeft, bottomLeft
        };
      }
    }
    return new float[] {0, 0, 0, 0, 0, 0, 0, 0};
  }

  @RequiresApi(VERSION_CODES.S)
  private int getRoundedCornerRadius(WindowInsets insets, int position) {
    final RoundedCorner roundedCorner = insets.getRoundedCorner(position);
    return roundedCorner != null ? roundedCorner.getRadius() : 0;
  }
}