HSNavigationController

HSNavigationController.h

#import <UIKit/UIKit.h>

@interface HSNavigationController : UINavigationController

@end

HSNavigationController.m

#import "HSNavigationController.h"

@interface HSNavigationController() <UINavigationControllerDelegate>

@end

@implementation HSNavigationController

#pragma mark - NavigationController Delegate

- (void)navigationController:(UINavigationController *)navigationController willShowViewController:(UIViewController *)viewController animated:(BOOL)animated
{
    /* Hide navigation bar if root controller */
    if ([navigationController isEqual:self] && [viewController isEqual:[self.viewControllers firstObject]]) {
        [self setNavigationBarHidden:YES animated:animated];
    } else {
        [self setNavigationBarHidden:NO animated:animated];
    }
}

#pragma mark - Initialization

- (void)setup
{
    self.delegate = self;
}

- (instancetype)init
{
    self = [super init];
    
    if (self) {
        [self setup];
    }
    
    return self;
}

- (instancetype)initWithNavigationBarClass:(Class)navigationBarClass toolbarClass:(Class)toolbarClass
{
    self = [super initWithNavigationBarClass:navigationBarClass toolbarClass:toolbarClass];
    
    if (self) {
        [self setup];
    }
    
    return self;
}

- (instancetype)initWithRootViewController:(UIViewController *)rootViewController
{
    self = [super initWithRootViewController:rootViewController];
    
    if (self) {
        [self setup];
    }
    
    return self;
}

@end