toolbox/UUAlert.m at master · ryandevore/toolbox · GitHub

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
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
//
//  UUAlert.m
//  Useful Utilities - UIAlertView extensions
//
//	License:
//  You are free to use this code for whatever purposes you desire. The only requirement is that you smile everytime you use it.
//
//  Contact: @cheesemaker or jon@threejacks.com

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

@interface UIAlertController(UUToolbox)
	@property (nonatomic, strong) UIWindow* displayWindow;
@end

@implementation UIAlertController(UUToolbox)
@dynamic displayWindow;

- (void)setDisplayWindow:(UIWindow *)displayWindow
{
    objc_setAssociatedObject(self, @selector(displayWindow), displayWindow, OBJC_ASSOCIATION_RETAIN_NONATOMIC);
}

- (UIWindow*) displayWindow
{
    return objc_getAssociatedObject(self, @selector(displayWindow));
}

@end

@implementation UUAlertViewController

+ (UIAlertController*) initWithTitle:(NSString *)alertTitle
                       message:(NSString *)alertMessage
             completionHandler:(void (^)(NSInteger buttonIndex))completionHandler
                  buttonTitles:(NSString *)defaultButtonTitle, ...
{
	NSMutableArray* argumentArray = [[NSMutableArray alloc] init];
	va_list argList;
	va_start(argList, defaultButtonTitle);
	NSString* otherButtonTitle = va_arg(argList, NSString*);
	if (otherButtonTitle)
	{
		NSString* buttonMessage = va_arg(argList, NSString*);
		while (buttonMessage)
		{
			[argumentArray addObject:[NSString stringWithString:buttonMessage]];
			buttonMessage = va_arg(argList, NSString*);
		}
	}
	va_end(argList);

	UIAlertController* alertController = [UIAlertController alertControllerWithTitle:alertTitle message:alertMessage preferredStyle:UIAlertControllerStyleAlert];
	UIAlertAction* alertAction = [UIAlertAction actionWithTitle:defaultButtonTitle style:UIAlertActionStyleDestructive handler:^(UIAlertAction * _Nonnull action)
	{
		if (completionHandler)
		{
			completionHandler(0);
		}
	}];
	[alertController addAction:alertAction];

	for (NSInteger i = 0; i < argumentArray.count; i++)
	{
		NSString* buttonTitle = [argumentArray objectAtIndex:i];

		UIAlertAction* alertAction = [UIAlertAction actionWithTitle:buttonTitle style:UIAlertActionStyleDefault handler:^(UIAlertAction * _Nonnull action)
		{
			if (completionHandler)
			{
				completionHandler(i + 1);
			}
		}];

		[alertController addAction:alertAction];
	}

	return alertController;
}

//Convenience functions
+ (UIAlertController*) uuOKCancelAlert:(NSString *)title message:(NSString *)message completionHandler:(void (^)(NSInteger buttonIndex))completionHandler
{
	UIAlertController* controller = [UUAlertViewController initWithTitle:title message:message completionHandler:completionHandler buttonTitles:@"Cancel", @"Ok", nil];
	return controller;
}

+ (UIAlertController*) uuOneButtonAlert:(NSString *)title message:(NSString *)message button:(NSString*)button completionHandler:(void (^)(NSInteger buttonIndex))completionHandler
{
	UIAlertController* controller = [UUAlertViewController initWithTitle:title message:message completionHandler:completionHandler buttonTitles:button, nil];
	return controller;
}

+ (UIAlertController*) uuTwoButtonAlert:(NSString *)title message:(NSString *)message buttonOne:(NSString*)buttonOne buttonTwo:(NSString*)buttonTwo completionHandler:(void (^)(NSInteger buttonIndex))completionHandler
{
	UIAlertController* controller = [UUAlertViewController initWithTitle:title message:message completionHandler:completionHandler buttonTitles:buttonOne, buttonTwo, nil];
	return controller;
}

+ (void) uuShowAlertWithTitle:(NSString*)alertTitle
                      message:(NSString*)alertMessage
                  buttonTitle:(NSString*)buttonTitle
            completionHandler:(void (^)(NSInteger buttonIndex))completionHandler
{
	UIAlertController* alertController = [UUAlertViewController initWithTitle:alertTitle message:alertMessage completionHandler:completionHandler buttonTitles:buttonTitle, nil];
	[UUAlertViewController displayAlertController:alertController];
}

+ (void) uuShowOKCancelAlert:(NSString *)title message:(NSString *)message completionHandler:(void (^)(NSInteger buttonIndex))completionHandler
{
	UIAlertController* alertController = [UUAlertViewController uuOKCancelAlert:title message:message completionHandler:completionHandler];
	[UUAlertViewController displayAlertController:alertController];
}

+ (void) uuShowOneButtonAlert:(NSString *)title message:(NSString *)message button:(NSString*)button completionHandler:(void (^)(NSInteger buttonIndex))completionHandler
{
	UIAlertController* alertController = [UUAlertViewController uuOneButtonAlert:title message:message button:button completionHandler:completionHandler];
	[UUAlertViewController displayAlertController:alertController];
}

+ (void) uuShowTwoButtonAlert:(NSString *)title message:(NSString *)message buttonOne:(NSString*)buttonOne buttonTwo:(NSString*)buttonTwo completionHandler:(void (^)(NSInteger buttonIndex))completionHandler
{
	UIAlertController* alertController = [UUAlertViewController uuTwoButtonAlert:title message:message buttonOne:buttonOne buttonTwo:buttonTwo completionHandler:completionHandler];
	[UUAlertViewController displayAlertController:alertController];
}

+ (void) displayAlertController:(UIAlertController*)alertController
{
    UIWindow *topWindow = [UIApplication sharedApplication].windows.lastObject;

	alertController.displayWindow = [[UIWindow alloc] initWithFrame:[UIScreen mainScreen].bounds];
	alertController.displayWindow.rootViewController = [UIViewController new];
	alertController.displayWindow.tintColor = [UIApplication sharedApplication].keyWindow.tintColor;


	alertController.displayWindow.windowLevel = topWindow.windowLevel + 1;
	[alertController.displayWindow makeKeyAndVisible];
	[alertController.displayWindow.rootViewController presentViewController:alertController animated:NO completion:nil];
}


@end