PFNavigationDropdownMenu.m

//
//  PFNavigationDropdownMenu.m
//  PFNavigationDropdownMenu
//
//  Created by Cee on 02/08/2015.
//  Copyright (c) 2015 Cee. All rights reserved.
//

#import "PFNavigationDropdownMenu.h"
#import "PFTableView.h"

@interface PFNavigationDropdownMenu()
@property (nonatomic, strong) UIView *tableContainerView;
@property (nonatomic, strong) PFConfiguration *configuration;
@property (nonatomic, assign) CGRect mainScreenBounds;
@property (nonatomic, strong) UIButton *menuButton;
@property (nonatomic, strong) UILabel *menuTitle;
@property (nonatomic, strong) UIImageView *menuArrow;
@property (nonatomic, strong) UIView *backgroundView;
@property (nonatomic, strong) PFTableView *tableView;
@property (nonatomic, strong) NSArray *items;
@property (nonatomic, assign) BOOL isShown;
@property (nonatomic, assign) CGFloat navigationBarHeight;
@end

@implementation PFNavigationDropdownMenu

- (instancetype)initWithFrame:(CGRect)frame
                        title:(NSString *)title
                        items:(NSArray *)items
                containerView:(UIView *)containerView
{
    self = [super initWithFrame:frame];
    if (self) {
        // Init properties
        self.configuration = [[PFConfiguration alloc] init];
        self.tableContainerView = containerView;
        self.navigationBarHeight = 44;
        self.mainScreenBounds = [UIScreen mainScreen].bounds;
        self.isShown = NO;
        self.items = items;
        
        // Init button as navigation title
        self.menuButton = [[UIButton alloc] initWithFrame:frame];
        [self.menuButton addTarget:self action:@selector(menuButtonTapped:) forControlEvents:UIControlEventTouchUpInside];
        [self addSubview:self.menuButton];
        
        self.menuTitle = [[UILabel alloc] initWithFrame:frame];
        self.menuTitle.text = title;
        self.menuTitle.textColor = [UINavigationBar appearance].titleTextAttributes[NSForegroundColorAttributeName];
        self.menuTitle.textAlignment = NSTextAlignmentCenter;
        self.menuTitle.font = self.configuration.cellTextLabelFont;
        [self.menuButton addSubview:self.menuTitle];
        
        self.menuArrow = [[UIImageView alloc] initWithImage:self.configuration.arrowImage];
        [self.menuButton addSubview:self.menuArrow];
        
        // Init table view
        self.tableView = [[PFTableView alloc] initWithFrame:CGRectMake(self.mainScreenBounds.origin.x,
                                                                       self.mainScreenBounds.origin.y,
                                                                       self.mainScreenBounds.size.width,
                                                                       self.mainScreenBounds.size.height + 300 - 64)
                                                      items:items
                                              configuration:self.configuration];
        __weak typeof(self) weakSelf = self;
        self.tableView.selectRowAtIndexPathHandler = ^(NSUInteger indexPath){
            __strong typeof(weakSelf) strongSelf = weakSelf;
            if (strongSelf.didSelectItemAtIndexHandler)
            {
                strongSelf.didSelectItemAtIndexHandler(indexPath);
            }
            [strongSelf setMenuTitleText:items[indexPath]];
            [strongSelf hideMenu];
            strongSelf.isShown = NO;
            [strongSelf layoutSubviews];
        };
        
    }
    return self;
}

- (void)layoutSubviews
{
    [super layoutSubviews];
    [self.menuTitle sizeToFit];
    self.menuTitle.center = CGPointMake(self.frame.size.width / 2.f, self.frame.size.height / 2.f);
    [self.menuArrow sizeToFit];
    self.menuArrow.center = CGPointMake(CGRectGetMaxX(self.menuTitle.frame) + self.configuration.arrowPadding, self.frame.size.height / 2.f);
}

- (void)showMenu
{
    // Table view header
    UIView *headerView = [[UIView alloc] initWithFrame:CGRectMake(0, 0, self.frame.size.width, 300)];
    headerView.backgroundColor = self.configuration.cellBackgroundColor;
    self.tableView.tableHeaderView = headerView;
    
    // Reload data to dismiss highlight color of selected cell
    [self.tableView reloadData];
    
    // Init background view (under table view)
    self.backgroundView = [[UIView alloc] initWithFrame:self.mainScreenBounds];
    self.backgroundView.backgroundColor = self.configuration.maskBackgroundColor;
    
    // Add background view & table view to container view
    [self.tableContainerView addSubview:self.backgroundView];
    [self.tableContainerView addSubview:self.tableView];
    
    // Rotate arrow
    [self rotateArrow];
    
    // Change background alpha
    self.backgroundView.alpha = 0;
    
    // Animation
    self.tableView.frame = CGRectMake(self.tableView.frame.origin.x,
                                      -(CGFloat)(self.items.count) * self.configuration.cellHeight - 300,
                                      self.tableView.frame.size.width,
                                      self.tableView.frame.size.height);
    
    [UIView animateWithDuration:self.configuration.animationDuration * 1.5f
                          delay:0
         usingSpringWithDamping:.7
          initialSpringVelocity:.5
                        options:0
                     animations:^{
                         self.tableView.frame = CGRectMake(self.tableView.frame.origin.x,
                                                           -300,
                                                           self.tableView.frame.size.width,
                                                           self.tableView.frame.size.height);
                         self.backgroundView.alpha = self.configuration.maskBackgroundOpacity;
                         
                     }
                     completion:nil];
}

- (void)hideMenu
{
    // Rotate arrow
    [self rotateArrow];
    
    // Change background alpha
    self.backgroundView.alpha = self.configuration.maskBackgroundOpacity;
    
    [UIView animateWithDuration:self.configuration.animationDuration * 1.5f
                          delay:0
         usingSpringWithDamping:.7
          initialSpringVelocity:.5
                        options:0
                     animations:^{
                         self.tableView.frame = CGRectMake(self.tableView.frame.origin.x,
                                                           -200,
                                                           self.tableView.frame.size.width,
                                                           self.tableView.frame.size.height);
                         self.backgroundView.alpha = self.configuration.maskBackgroundOpacity;
                         
                     }
                     completion:nil];
    
    [UIView animateWithDuration:self.configuration.animationDuration
                          delay:0
                        options:UIViewAnimationOptionTransitionNone
                     animations:^{
                         self.tableView.frame = CGRectMake(self.tableView.frame.origin.x,
                                                           -(CGFloat)(self.items.count) * self.configuration.cellHeight - 300,
                                                           self.tableView.frame.size.width,
                                                           self.tableView.frame.size.height);
                         self.backgroundView.alpha = 0;
                     } completion:^(BOOL finished) {
                         [self.tableView removeFromSuperview];
                         [self.backgroundView removeFromSuperview];
                     }];
    
}

- (void)rotateArrow
{
    __weak typeof(self) weakSelf = self;
    [UIView animateWithDuration:self.configuration.animationDuration
                     animations:^{
                         __strong typeof(weakSelf) strongSelf = weakSelf;
                         strongSelf.menuArrow.transform = CGAffineTransformRotate(strongSelf.menuArrow.transform, 180 * (CGFloat)(M_PI / 180));
                     }];
}

- (void)setMenuTitleText:(NSString *)title
{
    self.menuTitle.text = title;
}

- (void)menuButtonTapped:(UIButton *)sender
{
    self.isShown = !self.isShown;
    if (self.isShown) {
        [self showMenu];
    } else {
        [self hideMenu];
    }
}

#pragma mark - Open/Close State
- (void)openMenu
{
    self.isShown = YES;
    [self showMenu];
}

- (void)closeMenu
{
    self.isShown = NO;
    [self hideMenu];
}

- (BOOL)isOpen
{
    return [self isShown];
}

#pragma mark - Item Selection
- (void)selectItemAtIndex:(NSUInteger)index
{
    if (index >= self.items.count) {
        index = self.items.count - 1;
    }
    [self.tableView selectIndex:index];
}

#pragma mark - Setters
- (void)setCellHeight:(CGFloat)cellHeight
{
    self.configuration.cellHeight = cellHeight;
}

- (void)setCellBackgroundColor:(UIColor *)cellBackgroundColor
{
    self.configuration.cellBackgroundColor = cellBackgroundColor;
}

- (void)setCellTextLabelColor:(UIColor *)cellTextLabelColor
{
    self.configuration.cellTextLabelColor = cellTextLabelColor;
}

- (void)setCellTextLabelFont:(UIFont *)cellTextLabelFont
{
    self.configuration.cellTextLabelFont = cellTextLabelFont;
    self.menuTitle.font = self.configuration.cellTextLabelFont;
}

- (void)setCellSelectionColor:(UIColor *)cellSelectionColor
{
    self.configuration.cellSelectionColor = cellSelectionColor;
}

- (void)setCheckMarkImage:(UIImage *)checkMarkImage
{
    self.configuration.checkMarkImage = checkMarkImage;
}

- (void)setAnimationDuration:(NSTimeInterval)animationDuration
{
    self.configuration.animationDuration = animationDuration;
}

- (void)setArrowImage:(UIImage *)arrowImage
{
    self.configuration.arrowImage = arrowImage;
    self.menuArrow.image = self.configuration.arrowImage;
}

- (void)setArrowPadding:(CGFloat)arrowPadding
{
    self.configuration.arrowPadding = arrowPadding;
}

- (void)setMaskBackgroundColor:(UIColor *)maskBackgroundColor
{
    self.configuration.maskBackgroundColor = maskBackgroundColor;
}

- (void)setMaskBackgroundOpacity:(CGFloat)maskBackgroundOpacity
{
    self.configuration.maskBackgroundOpacity = maskBackgroundOpacity;
}
@end