//
//  ASDisplayNodeSnapshotTests.mm
//  Texture
//
//  Copyright (c) Pinterest, Inc.  All rights reserved.
//  Licensed under Apache 2.0: http://www.apache.org/licenses/LICENSE-2.0
//

#import "ASSnapshotTestCase.h"
#import <AsyncDisplayKit/AsyncDisplayKit.h>
#import <AsyncDisplayKit/ASDisplayNodeInternal.h>

@interface ASDisplayNodeSnapshotTests : ASSnapshotTestCase

@end

@implementation ASDisplayNodeSnapshotTests

- (void)setUp {
  [super setUp];
  ASConfiguration *config = [ASConfiguration new];
  config.experimentalFeatures = ASExperimentalTraitCollectionDidChangeWithPreviousCollection;
  [ASConfigurationManager test_resetWithConfiguration:config];
}

- (void)testBasicHierarchySnapshotTesting
{
  ASDisplayNode *node = [[ASDisplayNode alloc] init];
  node.backgroundColor = [UIColor blueColor];
  
  ASTextNode *subnode = [[ASTextNode alloc] init];
  subnode.backgroundColor = [UIColor whiteColor];
  
  subnode.attributedText = [[NSAttributedString alloc] initWithString:@"Hello"];
  node.automaticallyManagesSubnodes = YES;
  node.layoutSpecBlock = ^(ASDisplayNode * _Nonnull node, ASSizeRange constrainedSize) {
    return [ASInsetLayoutSpec insetLayoutSpecWithInsets:UIEdgeInsetsMake(5, 5, 5, 5) child:subnode];
  };

  ASDisplayNodeSizeToFitSizeRange(node, ASSizeRangeMake(CGSizeZero, CGSizeMake(INFINITY, INFINITY)));
  ASSnapshotVerifyNode(node, nil);
}

NS_INLINE UIImage *BlueImageMake(CGRect bounds)
{
  UIGraphicsBeginImageContextWithOptions(bounds.size, YES, 0);
  [[UIColor blueColor] setFill];
  UIRectFill(bounds);
  UIImage *image = UIGraphicsGetImageFromCurrentImageContext();
  UIGraphicsEndImageContext();
  return image;
}

- (void)testPrecompositedCornerRounding
{
  for (CACornerMask c = 1; c <= kASCACornerAllCorners; c |= (c << 1)) {
    auto node = [[ASImageNode alloc] init];
    auto bounds = CGRectMake(0, 0, 100, 100);
    node.image = BlueImageMake(bounds);
    node.frame = bounds;
    node.cornerRoundingType = ASCornerRoundingTypePrecomposited;
    node.backgroundColor = UIColor.greenColor;
    node.maskedCorners = c;
    node.cornerRadius = 15;
    ASSnapshotVerifyNode(node, ([NSString stringWithFormat:@"%d", (int)c]));
  }
}

- (void)testClippingCornerRounding
{
#if AS_AT_LEAST_IOS13
  if (@available(iOS 13.0, *)) {
    ASConfiguration *config = [ASConfiguration new];
    config.experimentalFeatures = ASExperimentalTraitCollectionDidChangeWithPreviousCollection;
    [ASConfigurationManager test_resetWithConfiguration:config];
  }
#endif

  for (CACornerMask c = 1; c <= kASCACornerAllCorners; c |= (c << 1)) {
    auto node = [[ASImageNode alloc] init];
    auto bounds = CGRectMake(0, 0, 100, 100);
    node.image = BlueImageMake(bounds);
    node.frame = bounds;
    node.cornerRoundingType = ASCornerRoundingTypeClipping;
#if AS_AT_LEAST_IOS13
    if (@available(iOS 13.0, *)) {
      node.backgroundColor = UIColor.systemBackgroundColor;
    } else {
      node.backgroundColor = UIColor.greenColor;
    }
#else
    node.backgroundColor = UIColor.greenColor;
#endif
    node.maskedCorners = c;
    node.cornerRadius = 15;
    // A layout pass is required, because that's where we lay out the clip layers.
    [node.layer layoutIfNeeded];

#if AS_AT_LEAST_IOS13
    if (@available(iOS 13.0, *)) {
      [[UITraitCollection traitCollectionWithUserInterfaceStyle:UIUserInterfaceStyleLight] performAsCurrentTraitCollection:^{
        ASSnapshotVerifyNode(node, ([NSString stringWithFormat:@"%d-light", (int)c]));
      }];

      [[UITraitCollection traitCollectionWithUserInterfaceStyle:UIUserInterfaceStyleDark] performAsCurrentTraitCollection:^{
        ASSnapshotVerifyNode(node, ([NSString stringWithFormat:@"%d-dark", (int)c]));
      }];
    } else {
      ASSnapshotVerifyNode(node, ([NSString stringWithFormat:@"%d", (int)c]));
    }
#else
    ASSnapshotVerifyNode(node, ([NSString stringWithFormat:@"%d", (int)c]));
#endif
  }
}

#if AS_AT_LEAST_IOS13

- (void)testUserInterfaceStyleSnapshotTesting
{
  if (@available(iOS 13.0, *)) {
    ASDisplayNode *node = [[ASDisplayNode alloc] init];
    [node setLayerBacked:YES];

    node.backgroundColor = [UIColor systemBackgroundColor];

    node.style.preferredSize = CGSizeMake(100, 100);
    ASDisplayNodeSizeToFitSizeRange(node, ASSizeRangeMake(CGSizeZero, CGSizeMake(INFINITY, INFINITY)));

    [[UITraitCollection traitCollectionWithUserInterfaceStyle:UIUserInterfaceStyleLight] performAsCurrentTraitCollection:^{
      ASSnapshotVerifyNode(node, @"user_interface_style_light");
    }];

    [[UITraitCollection traitCollectionWithUserInterfaceStyle:UIUserInterfaceStyleDark] performAsCurrentTraitCollection:^{
      ASSnapshotVerifyNode(node, @"user_interface_style_dark");
    }];
  }
}

- (void)testBackgroundDynamicColor {
  if (@available(iOS 13.0, *)) {
    ASDisplayNode *node = [[ASImageNode alloc] init];
    node.backgroundColor = [UIColor systemGray6Color];
    auto bounds = CGRectMake(0, 0, 100, 100);
    node.frame = bounds;
    
    UITraitCollection *tcLight = [UITraitCollection traitCollectionWithUserInterfaceStyle:UIUserInterfaceStyleLight];
    [tcLight performAsCurrentTraitCollection: ^{
      ASSnapshotVerifyNode(node, @"light");
    }];
    
    UITraitCollection *tcDark = [UITraitCollection traitCollectionWithUserInterfaceStyle:UIUserInterfaceStyleDark];
    [tcDark performAsCurrentTraitCollection: ^{
      ASSnapshotVerifyNode(node, @"dark");
    }];
  }
}

- (void)testBackgroundDynamicColorLayerBacked {
  if (@available(iOS 13.0, *)) {
    ASDisplayNode *node = [[ASImageNode alloc] init];
    node.backgroundColor = [UIColor systemGray6Color];
    node.layerBacked = YES;
    auto bounds = CGRectMake(0, 0, 100, 100);
    node.frame = bounds;
    
    UITraitCollection *tcLight = [UITraitCollection traitCollectionWithUserInterfaceStyle:UIUserInterfaceStyleLight];
    [tcLight performAsCurrentTraitCollection: ^{
      ASSnapshotVerifyNode(node, @"light");
    }];
    
    UITraitCollection *tcDark = [UITraitCollection traitCollectionWithUserInterfaceStyle:UIUserInterfaceStyleDark];
    [tcDark performAsCurrentTraitCollection: ^{
      ASSnapshotVerifyNode(node, @"dark");
    }];
  }
}

#endif // #if AS_AT_LEAST_IOS13

@end