/**
 * 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) {
    NSString *type = transformConfig[@"type"];
    NSString *property = transformConfig[@"property"];

    CGFloat value;
    if ([type isEqualToString: @"animated"]) {
      NSNumber *nodeTag = transformConfig[@"nodeTag"];
      RCTAnimatedNode *node = self.parentNodes[nodeTag];
      if (!node.hasUpdated || ![node isKindOfClass:[RCTValueAnimatedNode class]]) {
        continue;
      }
      RCTValueAnimatedNode *parentNode = (RCTValueAnimatedNode *)node;
      value = parentNode.value;
    } else {
      value = [transformConfig[@"value"] floatValue];
    }

    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