#import <UIKit/UIKit.h>

@interface UIViewController (NavigationBar)
@property (nonatomic, assign) IBInspectable BOOL hidesNavigationBarWhenPush;
@end

#import "UIViewController+NavigationBar.h"
#import <objc/runtime.h>

@implementation UIViewController (NavigationBar)

const void *kHidesNavigationBarWhenPush = "kHidesNavigationBarWhenPush";

+ (void)load {
    static dispatch_once_t onceToken;
    dispatch_once(&onceToken, ^{
        Class class = [self class];
        
        SEL originalSelector = @selector(viewWillAppear:);
        SEL swizzledSelector = @selector(_viewWillAppear:);
        
        Method originalMethod = class_getInstanceMethod(class, originalSelector);
        Method swizzledMethod = class_getInstanceMethod(class , swizzledSelector);
        
        BOOL didAddMethod = class_addMethod(class, originalSelector, method_getImplementation(swizzledMethod), method_getTypeEncoding(swizzledMethod));
        
        if (didAddMethod) {
            class_replaceMethod(class, swizzledSelector, method_getImplementation(originalMethod), method_getTypeEncoding(originalMethod));
        } else {
            method_exchangeImplementations(originalMethod, swizzledMethod);
        }
        
    });
}

- (void)_viewWillAppear:(BOOL)animated {
    [self _viewWillAppear:animated];
//    self.hidesBottomBarWhenPushed = false;
    [self.navigationController setNavigationBarHidden:self.hidesNavigationBarWhenPush animated:animated];
}

- (BOOL)hidesNavigationBarWhenPush {
    return objc_getAssociatedObject(self, kHidesNavigationBarWhenPush);
}

- (void)setHidesNavigationBarWhenPush:(BOOL)hidesNavigationBarWhenPush {
    objc_setAssociatedObject(self, kHidesNavigationBarWhenPush, [NSNumber numberWithBool:hidesNavigationBarWhenPush], OBJC_ASSOCIATION_ASSIGN);
}

@end