#import "UIBarButtonItem+Proxy.h"

#import "NSBag.h"
#import <ObjCRuntimeUtils/RuntimeUtils.h>

static const void *setEnabledListenerBagKey = &setEnabledListenerBagKey;
static const void *setTitleListenerBagKey = &setTitleListenerBagKey;
static const void *customDisplayNodeKey = &customDisplayNodeKey;
static const void *backButtonAppearanceKey = &backButtonAppearanceKey;
static const void *customActionKey = &customActionKey;

@implementation UIBarButtonItem (Proxy)

+ (void)load
{
    static dispatch_once_t onceToken;
    dispatch_once(&onceToken, ^
    {
        [RuntimeUtils swizzleInstanceMethodOfClass:[UIBarButtonItem class] currentSelector:@selector(setEnabled:) newSelector:@selector(_c1e56039_setEnabled:)];
        [RuntimeUtils swizzleInstanceMethodOfClass:[UIBarButtonItem class] currentSelector:@selector(setTitle:) newSelector:@selector(_c1e56039_setTitle:)];
    });
}

- (instancetype)initWithCustomDisplayNode:(ASDisplayNode *)customDisplayNode {
    self = [self init];
    if (self != nil) {
        [self setAssociatedObject:customDisplayNode forKey:customDisplayNodeKey];
    }
    return self;
}

- (instancetype)initWithBackButtonAppearanceWithTitle:(NSString *)title target:(id)target action:(SEL)action {
    self = [self initWithTitle:title style:UIBarButtonItemStylePlain target:target action:action];
    if (self != nil) {
        [self setAssociatedObject:@true forKey:backButtonAppearanceKey];
    }
    return self;
}

- (ASDisplayNode *)customDisplayNode {
    return [self associatedObjectForKey:customDisplayNodeKey];
}

- (bool)backButtonAppearance {
    return [[self associatedObjectForKey:backButtonAppearanceKey] boolValue];
}

- (void)setCustomAction:(void (^)())customAction {
    [self setAssociatedObject:[customAction copy] forKey:customActionKey];
}

- (void)_c1e56039_setEnabled:(BOOL)enabled
{
    [self _c1e56039_setEnabled:enabled];
    
    [(NSBag *)[self associatedObjectForKey:setEnabledListenerBagKey] enumerateItems:^(UIBarButtonItemSetEnabledListener listener)
    {
        listener(enabled);
    }];
}

- (void)_c1e56039_setTitle:(NSString *)title
{
    [self _c1e56039_setTitle:title];
    
    [(NSBag *)[self associatedObjectForKey:setTitleListenerBagKey] enumerateItems:^(UIBarButtonItemSetTitleListener listener)
    {
        listener(title);
    }];
}

- (void)performActionOnTarget
{
    void (^customAction)() = [self associatedObjectForKey:customActionKey];
    if (customAction) {
        customAction();
        return;
    }
    
    if (self.target == nil) {
        return;
    }
    
#pragma clang diagnostic push
#pragma clang diagnostic ignored "-Warc-performSelector-leaks"
    [self.target performSelector:self.action];
#pragma clang diagnostic pop
}

- (NSInteger)addSetTitleListener:(UIBarButtonItemSetTitleListener)listener
{
    NSBag *bag = [self associatedObjectForKey:setTitleListenerBagKey];
    if (bag == nil)
    {
        bag = [[NSBag alloc] init];
        [self setAssociatedObject:bag forKey:setTitleListenerBagKey];
    }
    return [bag addItem:[listener copy]];
}

- (void)removeSetTitleListener:(NSInteger)key
{
    [(NSBag *)[self associatedObjectForKey:setTitleListenerBagKey] removeItem:key];
}

- (NSInteger)addSetEnabledListener:(UIBarButtonItemSetEnabledListener)listener
{
    NSBag *bag = [self associatedObjectForKey:setEnabledListenerBagKey];
    if (bag == nil)
    {
        bag = [[NSBag alloc] init];
        [self setAssociatedObject:bag forKey:setEnabledListenerBagKey];
    }
    return [bag addItem:[listener copy]];
}

- (void)removeSetEnabledListener:(NSInteger)key
{
    [(NSBag *)[self associatedObjectForKey:setEnabledListenerBagKey] removeItem:key];
}

@end