/**
 * Copyright (c) 2015-present, Facebook, Inc.
 * All rights reserved.
 *
 * This source code is licensed under the BSD-style license found in the
 * LICENSE file in the root directory of this source tree. An additional grant
 * of patent rights can be found in the PATENTS file in the same directory.
 */

#import "RCTTransformAnimatedNode.h"
#import "RCTValueAnimatedNode.h"

@implementation RCTTransformAnimatedNode
{
  NSMutableDictionary<NSString *, NSObject *> *_updatedPropsDictionary;
}

- (instancetype)initWithTag:(NSNumber *)tag
                     config:(NSDictionary<NSString *, id> *)config;
{
  if ((self = [super initWithTag:tag config:config])) {
    _updatedPropsDictionary = [NSMutableDictionary new];
  }
  return self;
}

- (NSDictionary *)updatedPropsDictionary
{
  return _updatedPropsDictionary;
}

- (void)performUpdate
{
  [super performUpdate];

  CATransform3D transform = CATransform3DIdentity;

  NSArray<NSDictionary *> *transformConfigs = self.config[@"transforms"];
  for (NSDictionary *transformConfig in transformConfigs) {
    NSNumber *nodeTag = transformConfig[@"nodeTag"];

    RCTAnimatedNode *node = self.parentNodes[nodeTag];
    if (node.hasUpdated && [node isKindOfClass:[RCTValueAnimatedNode class]]) {
      RCTValueAnimatedNode *parentNode = (RCTValueAnimatedNode *)node;

      NSString *property = transformConfig[@"property"];
      CGFloat value = parentNode.value;

      if ([property isEqualToString:@"scale"]) {
        transform = CATransform3DScale(transform, value, value, 1);

      } else if ([property isEqualToString:@"scaleX"]) {
        transform = CATransform3DScale(transform, value, 1, 1);

      } else if ([property isEqualToString:@"scaleY"]) {
        transform = CATransform3DScale(transform, 1, value, 1);

      } else if ([property isEqualToString:@"translateX"]) {
        transform = CATransform3DTranslate(transform, value, 0, 0);

      } else if ([property isEqualToString:@"translateY"]) {
        transform = CATransform3DTranslate(transform, 0, value, 0);

      } else if ([property isEqualToString:@"rotate"]) {
        transform = CATransform3DRotate(transform, value, 0, 0, 1);

      } else if ([property isEqualToString:@"rotateX"]) {
        transform = CATransform3DRotate(transform, value, 1, 0, 0);

      } else if ([property isEqualToString:@"rotateY"]) {
        transform = CATransform3DRotate(transform, value, 0, 1, 0);

      } else if ([property isEqualToString:@"perspective"]) {
        transform.m34 = 1.0 / -value;
      }
    }
  }

  _updatedPropsDictionary[@"transform"] = [NSValue valueWithCATransform3D:transform];
}

- (void)cleanupAnimationUpdate
{
  [super cleanupAnimationUpdate];
  [_updatedPropsDictionary removeAllObjects];
}

@end
