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

#import "TextCellNode.h"
#import <AsyncDisplayKit/AsyncDisplayKit.h>
#import <AsyncDisplayKit/ASDisplayNode+Beta.h>

#ifndef USE_ASTEXTNODE_2
#define USE_ASTEXTNODE_2 1
#endif

@interface TextCellNode()
{
#if USE_ASTEXTNODE_2
  ASTextNode2 *_label1;
  ASTextNode2 *_label2;
#else
  ASTextNode *_label1;
  ASTextNode *_label2;
#endif
}
@end

@implementation TextCellNode

- (instancetype)initWithText1:(NSString *)text1 text2:(NSString *)text2
{
  self = [super init];
  if (self) {
    self.automaticallyManagesSubnodes = YES;
    self.clipsToBounds = YES;
#if USE_ASTEXTNODE_2
    _label1 = [[ASTextNode2 alloc] init];
    _label2 = [[ASTextNode2 alloc] init];
#else 
    _label1 = [[ASTextNode alloc] init];
    _label2 = [[ASTextNode alloc] init];
#endif

    _label1.attributedText = [[NSAttributedString alloc] initWithString:text1];
    _label2.attributedText = [[NSAttributedString alloc] initWithString:text2];

    _label1.maximumNumberOfLines = 1;
    _label1.truncationMode = NSLineBreakByTruncatingTail;
    _label2.maximumNumberOfLines = 1;
    _label2.truncationMode = NSLineBreakByTruncatingTail;

    [self simpleSetupYogaLayout];
  }
  return self;
}

/** 
  This is to text a row with two labels, the first should be truncated with "...".
  Layout is like: [l1Container[_label1], label2].
  This shows a bug of ASTextNode2.
 */
- (void)simpleSetupYogaLayout
{
  [self.style yogaNodeCreateIfNeeded];
  [_label1.style yogaNodeCreateIfNeeded];
  [_label2.style yogaNodeCreateIfNeeded];

  _label1.style.flexGrow = 0;
  _label1.style.flexShrink = 1;
  _label1.backgroundColor = [UIColor lightGrayColor];

  _label2.style.flexGrow = 0;
  _label2.style.flexShrink = 0;
  _label2.backgroundColor = [UIColor greenColor];

  ASDisplayNode *l1Container = [ASDisplayNode yogaVerticalStack];

  // TODO(fix ASTextNode2): next two line will show the bug of TextNode2
  // which works for ASTextNode though
  // see discussion here: https://github.com/TextureGroup/Texture/pull/553
  l1Container.style.alignItems = ASStackLayoutAlignItemsCenter;
  _label1.style.alignSelf = ASStackLayoutAlignSelfStart;

  l1Container.style.flexGrow = 0;
  l1Container.style.flexShrink = 1;

  l1Container.yogaChildren = @[_label1];

  self.style.justifyContent = ASStackLayoutJustifyContentSpaceBetween;
  self.style.alignItems = ASStackLayoutAlignItemsStart;
  self.style.flexDirection = ASStackLayoutDirectionHorizontal;
  self.yogaChildren = @[l1Container, _label2];
}

@end