package com.swmansion.reanimated.layoutReanimation;

import android.view.View;
import android.view.ViewGroup;
import com.facebook.react.uimanager.IllegalViewOperationException;
import com.facebook.react.uimanager.NativeViewHierarchyManager;
import com.facebook.react.uimanager.ViewManager;
import com.swmansion.reanimated.ReactNativeUtils;
import java.lang.reflect.InvocationTargetException;
import java.lang.reflect.Method;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.HashMap;
import java.util.List;

public class Snapshot {
  public static final String WIDTH = "width";
  public static final String HEIGHT = "height";
  public static final String ORIGIN_X = "originX";
  public static final String ORIGIN_Y = "originY";
  public static final String TRANSFORM_MATRIX = "transformMatrix";
  public static final String GLOBAL_ORIGIN_X = "globalOriginX";
  public static final String GLOBAL_ORIGIN_Y = "globalOriginY";
  public static final String BORDER_RADIUS = "borderRadius";
  public static final String BORDER_TOP_LEFT_RADIUS = "borderTopLeftRadius";
  public static final String BORDER_TOP_RIGHT_RADIUS = "borderTopRightRadius";
  public static final String BORDER_BOTTOM_LEFT_RADIUS = "borderBottomLeftRadius";
  public static final String BORDER_BOTTOM_RIGHT_RADIUS = "borderBottomRightRadius";

  public static final String CURRENT_WIDTH = "currentWidth";
  public static final String CURRENT_HEIGHT = "currentHeight";
  public static final String CURRENT_ORIGIN_X = "currentOriginX";
  public static final String CURRENT_ORIGIN_Y = "currentOriginY";
  public static final String CURRENT_TRANSFORM_MATRIX = "currentTransformMatrix";
  public static final String CURRENT_GLOBAL_ORIGIN_X = "currentGlobalOriginX";
  public static final String CURRENT_GLOBAL_ORIGIN_Y = "currentGlobalOriginY";
  public static final String CURRENT_BORDER_RADIUS = "currentBorderRadius";
  public static final String CURRENT_BORDER_TOP_LEFT_RADIUS = "currentBorderTopLeftRadius";
  public static final String CURRENT_BORDER_TOP_RIGHT_RADIUS = "currentBorderTopRightRadius";
  public static final String CURRENT_BORDER_BOTTOM_LEFT_RADIUS = "currentBorderBottomLeftRadius";
  public static final String CURRENT_BORDER_BOTTOM_RIGHT_RADIUS = "currentBorderBottomRightRadius";

  public static final String TARGET_WIDTH = "targetWidth";
  public static final String TARGET_HEIGHT = "targetHeight";
  public static final String TARGET_ORIGIN_X = "targetOriginX";
  public static final String TARGET_ORIGIN_Y = "targetOriginY";
  public static final String TARGET_TRANSFORM_MATRIX = "targetTransformMatrix";
  public static final String TARGET_GLOBAL_ORIGIN_X = "targetGlobalOriginX";
  public static final String TARGET_GLOBAL_ORIGIN_Y = "targetGlobalOriginY";
  public static final String TARGET_BORDER_RADIUS = "targetBorderRadius";
  public static final String TARGET_BORDER_TOP_LEFT_RADIUS = "targetBorderTopLeftRadius";
  public static final String TARGET_BORDER_TOP_RIGHT_RADIUS = "targetBorderTopRightRadius";
  public static final String TARGET_BORDER_BOTTOM_LEFT_RADIUS = "targetBorderBottomLeftRadius";
  public static final String TARGET_BORDER_BOTTOM_RIGHT_RADIUS = "targetBorderBottomRightRadius";

  public View view;
  public ViewGroup parent;
  public ViewManager viewManager;
  public ViewManager parentViewManager;
  public int width;
  public int height;
  public int originX;
  public int originY;
  public int globalOriginX;
  public int globalOriginY;
  public List<Float> transformMatrix =
      new ArrayList<>(Arrays.asList(1f, 0f, 0f, 0f, 1f, 0f, 0f, 0f, 1f));
  public int originXByParent;
  public int originYByParent;
  public ReactNativeUtils.BorderRadii borderRadii;
  private float[] identityMatrix = {1, 0, 0, 0, 1, 0, 0, 0, 1};

  public static ArrayList<String> targetKeysToTransform =
      new ArrayList<>(
          Arrays.asList(
              Snapshot.TARGET_WIDTH,
              Snapshot.TARGET_HEIGHT,
              Snapshot.TARGET_ORIGIN_X,
              Snapshot.TARGET_ORIGIN_Y,
              Snapshot.TARGET_GLOBAL_ORIGIN_X,
              Snapshot.TARGET_GLOBAL_ORIGIN_Y,
              Snapshot.TARGET_BORDER_RADIUS,
              Snapshot.TARGET_BORDER_TOP_LEFT_RADIUS,
              Snapshot.TARGET_BORDER_TOP_RIGHT_RADIUS,
              Snapshot.TARGET_BORDER_BOTTOM_LEFT_RADIUS,
              Snapshot.TARGET_BORDER_BOTTOM_RIGHT_RADIUS));
  public static ArrayList<String> currentKeysToTransform =
      new ArrayList<>(
          Arrays.asList(
              Snapshot.CURRENT_WIDTH,
              Snapshot.CURRENT_HEIGHT,
              Snapshot.CURRENT_ORIGIN_X,
              Snapshot.CURRENT_ORIGIN_Y,
              Snapshot.CURRENT_GLOBAL_ORIGIN_X,
              Snapshot.CURRENT_GLOBAL_ORIGIN_Y,
              Snapshot.CURRENT_BORDER_RADIUS,
              Snapshot.CURRENT_BORDER_TOP_LEFT_RADIUS,
              Snapshot.CURRENT_BORDER_TOP_RIGHT_RADIUS,
              Snapshot.CURRENT_BORDER_BOTTOM_LEFT_RADIUS,
              Snapshot.CURRENT_BORDER_BOTTOM_RIGHT_RADIUS));

  Snapshot(View view, NativeViewHierarchyManager viewHierarchyManager) {
    parent = (ViewGroup) view.getParent();
    try {
      viewManager = viewHierarchyManager.resolveViewManager(view.getId());
      parentViewManager = viewHierarchyManager.resolveViewManager(parent.getId());
    } catch (IllegalViewOperationException | NullPointerException e) {
      // do nothing
    }
    width = view.getWidth();
    height = view.getHeight();
    originX = view.getLeft();
    originY = view.getTop();
    this.view = view;
    int[] location = new int[2];
    view.getLocationOnScreen(location);
    globalOriginX = location[0];
    globalOriginY = location[1];
    borderRadii = new ReactNativeUtils.BorderRadii(0, 0, 0, 0, 0);
  }

  private int[] tryGetRealPosition(View view) {
    int[] location = new int[2];
    View currentView = view;
    while (currentView != null) {
      location[0] += currentView.getX();
      location[1] += currentView.getY();
      if (ScreensHelper.isScreen(currentView)
          && ScreensHelper.isScreensCoordinatorLayout(currentView.getParent())) {
        View screen = currentView;
        Class<?> screenClass = screen.getClass();
        try {
          Method getContainer = screenClass.getMethod("getContainer");
          currentView = (View) getContainer.invoke(screen);
        } catch (NoSuchMethodException
            | InvocationTargetException
            | IllegalAccessException ignored) {
        }
      } else if (currentView.getParent() instanceof View) {
        currentView = (View) currentView.getParent();
      } else {
        break;
      }
    }
    return location;
  }

  public Snapshot(View view) {
    int[] location = new int[2];
    view.getLocationOnScreen(location);
    if (location[0] == 0 && location[1] == 0) {
      /*
       In certain cases, when a view is correctly attached to the screen and has computed
       the correct layout, but is not visible on the screen, `getLocationOnScreen` may return
       incorrect values [0, 0]. This behavior can occur during tab changes in bottom tabs.
      */
      location = tryGetRealPosition(view);
    }
    originX = location[0];
    originY = location[1];
    width = view.getWidth();
    height = view.getHeight();

    View transformedView = findTransformedView(view);
    if (transformedView != null) {
      float[] transformMatrixArray = new float[9];
      transformedView.getMatrix().getValues(transformMatrixArray);
      transformMatrix = new ArrayList<>();
      for (int i = 0; i < 9; i++) {
        transformMatrix.add(transformMatrixArray[i]);
      }
      transformMatrix.set(0, transformedView.getScaleX());
      transformMatrix.set(4, transformedView.getScaleY());
      transformMatrix.set(2, transformedView.getTranslationX());
      transformMatrix.set(5, transformedView.getTranslationY());

      originX -= (width - width * transformedView.getScaleX()) / 2;
      originY -= (height - height * transformedView.getScaleY()) / 2;
    }
    originXByParent = view.getLeft();
    originYByParent = view.getTop();
    borderRadii = ReactNativeUtils.getBorderRadii(view);
  }

  private void addTargetConfig(HashMap<String, Object> data) {
    data.put(Snapshot.TARGET_ORIGIN_Y, originY);
    data.put(Snapshot.TARGET_ORIGIN_X, originX);
    data.put(Snapshot.TARGET_GLOBAL_ORIGIN_Y, globalOriginY);
    data.put(Snapshot.TARGET_GLOBAL_ORIGIN_X, globalOriginX);
    data.put(Snapshot.TARGET_HEIGHT, height);
    data.put(Snapshot.TARGET_WIDTH, width);
    data.put(Snapshot.TARGET_TRANSFORM_MATRIX, transformMatrix);
    data.put(Snapshot.TARGET_BORDER_RADIUS, borderRadii.full);
    data.put(Snapshot.TARGET_BORDER_TOP_LEFT_RADIUS, borderRadii.topLeft);
    data.put(Snapshot.TARGET_BORDER_TOP_RIGHT_RADIUS, borderRadii.topRight);
    data.put(Snapshot.TARGET_BORDER_BOTTOM_LEFT_RADIUS, borderRadii.bottomLeft);
    data.put(Snapshot.TARGET_BORDER_BOTTOM_RIGHT_RADIUS, borderRadii.bottomRight);
  }

  private void addCurrentConfig(HashMap<String, Object> data) {
    data.put(Snapshot.CURRENT_ORIGIN_Y, originY);
    data.put(Snapshot.CURRENT_ORIGIN_X, originX);
    data.put(Snapshot.CURRENT_GLOBAL_ORIGIN_Y, globalOriginY);
    data.put(Snapshot.CURRENT_GLOBAL_ORIGIN_X, globalOriginX);
    data.put(Snapshot.CURRENT_HEIGHT, height);
    data.put(Snapshot.CURRENT_WIDTH, width);
    data.put(Snapshot.CURRENT_TRANSFORM_MATRIX, transformMatrix);
    data.put(Snapshot.CURRENT_BORDER_RADIUS, borderRadii.full);
    data.put(Snapshot.CURRENT_BORDER_TOP_LEFT_RADIUS, borderRadii.topLeft);
    data.put(Snapshot.CURRENT_BORDER_TOP_RIGHT_RADIUS, borderRadii.topRight);
    data.put(Snapshot.CURRENT_BORDER_BOTTOM_LEFT_RADIUS, borderRadii.bottomLeft);
    data.put(Snapshot.CURRENT_BORDER_BOTTOM_RIGHT_RADIUS, borderRadii.bottomRight);
  }

  private void addBasicConfig(HashMap<String, Object> data) {
    data.put(Snapshot.ORIGIN_Y, originY);
    data.put(Snapshot.ORIGIN_X, originX);
    data.put(Snapshot.GLOBAL_ORIGIN_Y, globalOriginY);
    data.put(Snapshot.GLOBAL_ORIGIN_X, globalOriginX);
    data.put(Snapshot.HEIGHT, height);
    data.put(Snapshot.WIDTH, width);
    data.put(Snapshot.TRANSFORM_MATRIX, transformMatrix);
    data.put(Snapshot.BORDER_RADIUS, borderRadii.full);
    data.put(Snapshot.BORDER_TOP_LEFT_RADIUS, borderRadii.topLeft);
    data.put(Snapshot.BORDER_TOP_RIGHT_RADIUS, borderRadii.topRight);
    data.put(Snapshot.BORDER_BOTTOM_LEFT_RADIUS, borderRadii.bottomLeft);
    data.put(Snapshot.BORDER_BOTTOM_RIGHT_RADIUS, borderRadii.bottomRight);
  }

  public HashMap<String, Object> toTargetMap() {
    HashMap<String, Object> data = new HashMap<>();
    addTargetConfig(data);
    return data;
  }

  public HashMap<String, Object> toCurrentMap() {
    HashMap<String, Object> data = new HashMap<>();
    addCurrentConfig(data);
    return data;
  }

  public HashMap<String, Object> toBasicMap() {
    HashMap<String, Object> data = new HashMap<>();
    addBasicConfig(data);
    return data;
  }

  private View findTransformedView(View view) {
    View transformedView = null;
    boolean isTransformed = false;
    do {
      if (transformedView == null) {
        transformedView = view;
      } else {
        if (!(transformedView.getParent() instanceof View)) {
          break;
        }
        transformedView = (View) transformedView.getParent();
      }
      if (transformedView == null) {
        break;
      }
      float[] transformArray = new float[9];
      transformedView.getMatrix().getValues(transformArray);
      isTransformed = !Arrays.equals(transformArray, identityMatrix);
    } while (!isTransformed
        && transformedView != null
        && !transformedView.getClass().getSimpleName().equals("Screen"));
    return (isTransformed && transformedView != null) ? transformedView : null;
  }
}