#import "ASButtonNode.h"

#import <AsyncDisplayKit/ASThread.h>

@interface ASButtonNode ()

{

ASDN::RecursiveMutex _propertyLock;

NSAttributedString *_normalAttributedTitle;

NSAttributedString *_highlightedAttributedTitle;

NSAttributedString *_disabledAttributedTitle;

UIImage *_normalImage;

UIImage *_highlightedImage;

UIImage *_disabledImage;

}

@implementation ASButtonNode

@synthesize contentSpacing = _contentSpacing;

@synthesize laysOutHorizontally = _laysOutHorizontally;

- (instancetype)init

{

if (self = [super init]) {

_contentSpacing = 8.0;

_laysOutHorizontally = YES;

_titleNode = [[ASTextNode alloc] init];

_imageNode = [[ASImageNode alloc] init];

[self addSubnode:_titleNode];

[self addSubnode:_imageNode];

[self addTarget:self action:@selector(controlEventUpdated:) forControlEvents:ASControlNodeEventAllEvents];

}

return self;

}

- (void)controlEventUpdated:(ASControlNode *)node

{

[self updateImage];

[self updateTitle];

}

- (void)updateImage

{

ASDN::MutexLocker l(_propertyLock);

UIImage *newImage;

if (self.enabled == NO && _disabledImage) {

newImage = _disabledImage;

} else if (self.highlighted && _highlightedImage) {

newImage = _highlightedImage;

} else {

newImage = _normalImage;

}

if (newImage != self.imageNode.image) {

self.imageNode.image = newImage;

[self setNeedsLayout];

}

}

- (void)updateTitle

{

ASDN::MutexLocker l(_propertyLock);

NSAttributedString *newTitle;

if (self.enabled == NO && _disabledAttributedTitle) {

newTitle = _disabledAttributedTitle;

} else if (self.highlighted && _highlightedAttributedTitle) {

newTitle = _highlightedAttributedTitle;

} else {

newTitle = _normalAttributedTitle;

}

if (newTitle != self.titleNode.attributedString) {

self.titleNode.attributedString = newTitle;

[self setNeedsLayout];

}

}

- (CGFloat)contentSpacing

{

ASDN::MutexLocker l(_propertyLock);

return _contentSpacing;

}

- (void)setContentSpacing:(CGFloat)contentSpacing

{

ASDN::MutexLocker l(_propertyLock);

if (contentSpacing == _contentSpacing)

return;

_contentSpacing = contentSpacing;

[self setNeedsLayout];

}

- (BOOL)laysOutHorizontally

{

ASDN::MutexLocker l(_propertyLock);

return _laysOutHorizontally;

}

- (void)setLaysOutHorizontally:(BOOL)laysOutHorizontally

{

ASDN::MutexLocker l(_propertyLock);

if (laysOutHorizontally == _laysOutHorizontally)

return;

_laysOutHorizontally = laysOutHorizontally;

[self setNeedsLayout];

}

- (NSAttributedString *)attributedTitleForState:(ASButtonState)state

{

ASDN::MutexLocker l(_propertyLock);

switch (state) {

case ASButtonStateNormal:

return _normalAttributedTitle;

case ASButtonStateHighlighted:

return _highlightedAttributedTitle;

case ASButtonStateDisabled:

return _disabledAttributedTitle;

}

}

- (void)setAttributedTitle:(NSAttributedString *)title forState:(ASButtonState)state

{

ASDN::MutexLocker l(_propertyLock);

switch (state) {

case ASButtonStateNormal:

_normalAttributedTitle = [title copy];

break;

case ASButtonStateHighlighted:

_highlightedAttributedTitle = [title copy];

break;

case ASButtonStateDisabled:

_disabledAttributedTitle = [title copy];

break;

}

[self updateTitle];

}

- (UIImage *)imageForState:(ASButtonState)state

{

ASDN::MutexLocker l(_propertyLock);

switch (state) {

case ASButtonStateNormal:

return _normalImage;

case ASButtonStateHighlighted:

return _highlightedImage;

case ASButtonStateDisabled:

return _disabledImage;

}

}

- (void)setImage:(UIImage *)image forState:(ASButtonState)state

{

ASDN::MutexLocker l(_propertyLock);

switch (state) {

case ASButtonStateNormal:

_normalImage = image;

break;

case ASButtonStateHighlighted:

_highlightedImage = image;

break;

case ASButtonStateDisabled:

_disabledImage = image;

break;

}

[self updateImage];

}

- (ASLayoutSpec *)layoutSpecThatFits:(ASSizeRange)constrainedSize

{

ASStackLayoutSpec *stack = [[ASStackLayoutSpec alloc] init];

stack.direction = self.laysOutHorizontally ? ASStackLayoutDirectionHorizontal : ASStackLayoutDirectionVertical;

stack.spacing = self.contentSpacing;

stack.justifyContent = ASStackLayoutJustifyContentCenter;

stack.alignItems = ASStackLayoutAlignItemsCenter;

NSMutableArray *children = [[NSMutableArray alloc] initWithCapacity:2];

if (self.imageNode.image) {

[children addObject:self.imageNode];

}

if (self.titleNode.attributedString.length > 0) {

[children addObject:self.titleNode];

}

stack.children = children;

return stack;

}

- (void)layout

{

[super layout];

self.imageNode.hidden = self.imageNode.image == nil;

self.titleNode.hidden = self.titleNode.attributedString.length > 0 == NO;

}