-
Notifications
You must be signed in to change notification settings - Fork 1
/
EHAlertView.m
110 lines (95 loc) · 2.47 KB
/
EHAlertView.m
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
//
// EHAlertView.m
//
// Created by Antoine Marcadet on 28/09/11.
// Copyright 2011 Antoine Marcadet. All rights reserved.
//
// Permission is given to use this source code file without charge in any
// project, commercial or otherwise, entirely at your risk, with the condition
// that any redistribution (in part or whole) of source code must retain
// this copyright and permission notice. Attribution in compiled projects is
// appreciated but not required.
//
#import "EHAlertView.h"
@implementation EHAlertView
- (void)dealloc
{
#if NS_BLOCKS_AVAILABLE
if (willDismissButtonBlock != nil)
{
[willDismissButtonBlock release];
}
if (clickedButtonBlock != nil)
{
[clickedButtonBlock release];
}
if (didDismissButtonBlock != nil)
{
[didDismissButtonBlock release];
}
#endif
[super dealloc];
}
- (id)initWithTitle:(NSString *)title message:(NSString *)message cancelButtonTitle:(NSString *)cancelButtonTitle otherButtonTitles:(NSString *)otherButtonTitles, ...
{
if ((self = [super initWithTitle:title message:message delegate:self cancelButtonTitle:cancelButtonTitle otherButtonTitles:nil]))
{
va_list args;
va_start(args, otherButtonTitles);
for (NSString *otherButtonTitle = otherButtonTitles; otherButtonTitle != nil; otherButtonTitle = va_arg(args, NSString *))
{
[self addButtonWithTitle:otherButtonTitle];
}
va_end(args);
}
return self;
}
#if NS_BLOCKS_AVAILABLE
- (void)setWillDismissButtonBlock:(EHAlertButtonBlock)block
{
if (willDismissButtonBlock != nil)
{
[willDismissButtonBlock release];
}
willDismissButtonBlock = [block copy];
}
- (void)setClickedButtonBlock:(EHAlertButtonBlock)block
{
if (clickedButtonBlock != nil)
{
[clickedButtonBlock release];
}
clickedButtonBlock = [block copy];
}
- (void)setDidDismissButtonBlock:(EHAlertButtonBlock)block
{
if (didDismissButtonBlock != nil)
{
[didDismissButtonBlock release];
}
didDismissButtonBlock = [block copy];
}
#pragma mark - UIAlertViewDelegate
- (void)alertView:(UIAlertView *)alertView willDismissWithButtonIndex:(NSInteger)buttonIndex
{
if (willDismissButtonBlock != nil)
{
willDismissButtonBlock(buttonIndex);
}
}
- (void)alertView:(UIAlertView *)alertView clickedButtonAtIndex:(NSInteger)buttonIndex
{
if (clickedButtonBlock != nil)
{
clickedButtonBlock(buttonIndex);
}
}
- (void)alertView:(UIAlertView *)alertView didDismissWithButtonIndex:(NSInteger)buttonIndex
{
if (didDismissButtonBlock != nil)
{
didDismissButtonBlock(buttonIndex);
}
}
#endif
@end