PSPDFActionSheet.m

//
//  PSPDFActionSheet.m
//  PSPDFKit
//
//  Copyright (c) 2012 Peter Steinberger. All rights reserved.
//

#import "PSPDFActionSheet.h"
#import <objc/runtime.h>

@interface PSPDFActionSheet() <UIActionSheetDelegate> {
    id<UIActionSheetDelegate> _realDelegate;
    NSMutableDictionary *_buttonTitleToBlockDict;
    void (^_destroyBlock)();
}
@end

@implementation PSPDFActionSheet

///////////////////////////////////////////////////////////////////////////////////////////
#pragma mark - NSObject

- (id)initWithFrame:(CGRect)frame
{
    if ((self = [super initWithFrame:frame])) {
        
        self.delegate = self;
        
        // Create the blocks storage for handling all button actions
        _buttonTitleToBlockDict = [[NSMutableDictionary alloc] init];
    }
    return self;
}

- (id)initWithTitle:(NSString *)title {
    if ((self = [super initWithTitle:title delegate:self cancelButtonTitle:nil destructiveButtonTitle:nil otherButtonTitles:nil])) {
        // Does nothing
    }
    return self;
}

- (void)dealloc {
    self.delegate = nil;
}

///////////////////////////////////////////////////////////////////////////////////////////
#pragma mark - Public

- (void)setDelegate:(id<UIActionSheetDelegate>)delegate {
    if(delegate == nil) {
        [super setDelegate:nil];
        _realDelegate = nil;
    }else {
        [super setDelegate:self];
        if (delegate != self) {
            _realDelegate = delegate;
        }
    }
}

- (void)setDestructiveButtonWithTitle:(NSString *)title block:(void (^)())block {
    assert([title length] > 0 && "sheet destructive button title must not be empty");

    [self addButtonWithTitle:title block:block];
    self.destructiveButtonIndex = (self.numberOfButtons - 1);
}

- (void)setCancelButtonWithTitle:(NSString *)title block:(void (^)())block {
    assert([title length] > 0 && "sheet cancel button title must not be empty");

    [self addButtonWithTitle:title block:block];
    self.cancelButtonIndex = (self.numberOfButtons - 1);
}

- (void)addButtonWithTitle:(NSString *)title block:(void (^)())block {
    assert([title length] > 0 && "cannot add button with empty title");
    
    if (block) _buttonTitleToBlockDict[title] = [block copy];
    
    [self addButtonWithTitle:title];
}

- (NSUInteger)buttonCount {
    return self.numberOfButtons;
}

- (void)destroy {
    [_buttonTitleToBlockDict removeAllObjects];
}

- (void)setDestroyBlock:(void (^)())block {
    _destroyBlock = [block copy];
}

///////////////////////////////////////////////////////////////////////////////////////////
#pragma mark - UIActionSheet

- (void)dismissWithClickedButtonIndex:(NSInteger)buttonIndex animated:(BOOL)animated {
    [super dismissWithClickedButtonIndex:buttonIndex animated:animated];
    [self destroy];
}

///////////////////////////////////////////////////////////////////////////////////////////
#pragma mark - UIActionSheetDelegate

// Called when a button is clicked. The view will be automatically dismissed after this call returns
- (void)actionSheet:(UIActionSheet *)actionSheet clickedButtonAtIndex:(NSInteger)buttonIndex {
    // Run the button's block
    NSString* title = [self buttonTitleAtIndex:buttonIndex];
    id block = _buttonTitleToBlockDict[title];
    if (block) ((void (^)())block)();

    if ([_realDelegate respondsToSelector:_cmd]) {
        [_realDelegate actionSheet:actionSheet clickedButtonAtIndex:buttonIndex];
    }
}

// Called when we cancel a view (eg. the user clicks the Home button). This is not called when the user clicks the cancel button.
// If not defined in the delegate, we simulate a click in the cancel button
- (void)actionSheetCancel:(UIActionSheet *)actionSheet {
    if ([_realDelegate respondsToSelector:_cmd]) {
        [_realDelegate actionSheetCancel:actionSheet];
    }
}

// before animation and showing view
- (void)willPresentActionSheet:(UIActionSheet *)actionSheet {
    if ([_realDelegate respondsToSelector:_cmd]) {
        [_realDelegate willPresentActionSheet:actionSheet];
    }
}

// after animation
- (void)didPresentActionSheet:(UIActionSheet *)actionSheet {
    if ([_realDelegate respondsToSelector:_cmd]) {
        [_realDelegate didPresentActionSheet:actionSheet];
    }
}

// before animation and hiding view
- (void)actionSheet:(UIActionSheet *)actionSheet willDismissWithButtonIndex:(NSInteger)buttonIndex {
    if ([_realDelegate respondsToSelector:_cmd]) {
        [_realDelegate actionSheet:actionSheet willDismissWithButtonIndex:buttonIndex];
    }
}

// after animation
- (void)actionSheet:(UIActionSheet *)actionSheet didDismissWithButtonIndex:(NSInteger)buttonIndex {
    if ([_realDelegate respondsToSelector:_cmd]) {
        [_realDelegate actionSheet:actionSheet didDismissWithButtonIndex:buttonIndex];
    }

    [self destroy];
    if (_destroyBlock) {
        _destroyBlock();
    }
    self.delegate = nil;
}

@end