自己封装了的AlertController

一直觉得使用系统这个东西写起来特别麻烦，每次都要写一大推东西，还是重复的，今天抽了点时间自己重新封装了一下，解决了自己的强迫症。。。，不多说，直接上代码了。

1.自己定义了一个名为XBZ的UIAlertViewControllerde 分类，.h文件里面

 #import <UIKit/UIKit.h>

 typedef void(^ActionOne)(void);

 typedef void(^ActionTwo)(void);

 @interface UIAlertController (XBZ)

 /**
  只显示标题，默认2s后dismiss

  @param controller 当前弹出的控制器
  @param title 标题
  */
 + (void)alertWithController:(nonnull UIViewController *)controller
                       title:(nonnull NSString *)title;

 /**
  只显示标题，自定义dismiss时间

  @param controller 当前弹出的控制器
  @param title 标题
  @param timerInerval dismiss时间
  */
 + (void)alertWithController:(nonnull UIViewController *)controller
                       title:(nonnull NSString *)title
                timeInterval:(NSTimeInterval)timerInerval;

 /**
  显示标题+消息，默认显示2s后dismiss

  @param controller 当前弹出的控制器
  @param title 标题
  @param message 消息内容
  */
 + (void)alertWithController:(nonnull UIViewController *)controller
                       title:(nonnull NSString *)title
                     message:(nonnull NSString *)message;

 /**
  显示标题+消息，可自定义dismiss时间

  @param controller 当前弹出的控制器
  @param title 标题
  @param message 消息内容
  @param timerInerval dismiss时间
  */
 + (void)alertWithController:(nonnull UIViewController *)controller
                       title:(nonnull NSString *)title
                     message:(nonnull NSString *)message
                timeInterval:(NSTimeInterval)timerInerval;

 /**
  显示默认弹出和按钮风格（常用一）

  @param controller 当前弹出的控制器
  @param title 标题
  @param message 消息内容
  @param titles 按钮标题数组，可只传一个，actionTwo置nil
  @param actionOne 第一个按钮事件
  @param actionTwo 第二个按钮事件
  */
 + (void)alertWithController:(nonnull UIViewController *)controller
                       title:(NSString *)title
                     message:(NSString *)message
                actionTitles:(nonnull NSArray<NSString *> *)titles
                   actionOne:(ActionOne)actionOne
                   actionTwo:(ActionTwo)actionTwo;

 /**
  显示默认按钮风格，可自定义弹出风格

  @param controller 当前弹出的控制器
  @param title 标题
  @param message 消息内容
  @param titles 按钮标题数组，可只传一个，actionTwo置nil
  @param actionOne 第一个按钮事件
  @param actionTwo 第二个按钮事件
  @param isAlertStyle 弹出方式，YES为Alert，NO为Sheet，采用sheet时会自动加上cacel按钮
  */
 + (void)alertWithController:(nonnull UIViewController *)controller
                       title:(NSString *)title
                     message:(NSString *)message
                actionTitles:(nonnull NSArray<NSString *> *)titles
                   actionOne:(ActionOne)actionOne
                   actionTwo:(ActionTwo)actionTwo
                isAlertStyle:(BOOL)isAlertStyle;

 /**
  显示默认弹出风格，可自定义按钮风格（常用二）

  @param controller 当前弹出的控制器
  @param title 标题
  @param message 消息内容
  @param titles 按钮标题数组，可只传一个，actionTwo置nil
  @param actionStyles 自定义按钮风格
  @param actionOne 第一个按钮事件
  @param actionTwo 第二个按钮事件
  */
 + (void)alertWithController:(nonnull UIViewController *)controller
                       title:(NSString *)title
                     message:(NSString *)message
                actionTitles:(nonnull NSArray<NSString *> *)titles
                actionStyles:(NSArray<NSNumber *> *)actionStyles
                   actionOne:(ActionOne)actionOne
                   actionTwo:(ActionOne)actionTwo;

 /**
  自定义弹出风格和按钮风格

  @param controller 当前弹出的控制器
  @param title 标题
  @param message 消息内容
  @param titles 按钮标题数组，可只传一个，actionTwo置nil
  @param actionStyles action风格，数组形式
  @param actionOne 第一个按钮事件
  @param actionTwo 第二个按钮事件
  @param isAlertStyle 弹出方式，YES为Alert，NO为Sheet，采用sheet时会自动加上cacel按钮
  */
 + (void)alertWithController:(nonnull UIViewController *)controller
                       title:(NSString *)title
                     message:(NSString *)message
                actionTitles:(nonnull NSArray<NSString *> *)titles
                actionStyles:(NSArray<NSNumber *> *)actionStyles
                   actionOne:(ActionOne)actionOne
                   actionTwo:(ActionOne)actionTwo
                isAlertStyle:(BOOL)isAlertStyle;

 @end

2.然后是.m文件

 #import "UIAlertController+XBZ.h"

 NSTimeInterval kDefaultTimerInterval = .f;

 @implementation UIAlertController (XBZ)

 + (void)alertWithController:(nonnull UIViewController *)controller
                       title:(NSString *)title {

     [self alertWithController:controller title:title timeInterval:kDefaultTimerInterval];

 }

 + (void)alertWithController:(nonnull UIViewController *)controller
                       title:(NSString *)title
                timeInterval:(NSTimeInterval)timerInerval {

     [self alertWithController:controller title:title message:@"" timeInterval:timerInerval];

 }

 + (void)alertWithController:(nonnull UIViewController *)controller
                       title:(NSString *)title
                     message:(NSString *)message {

     [self alertWithController:controller title:title message:message timeInterval:kDefaultTimerInterval];

 }

 + (void)alertWithController:(nonnull UIViewController *)controller
                       title:(NSString *)title
                     message:(NSString *)message
                timeInterval:(NSTimeInterval)timerInerval {

     UIAlertController *alertController = [UIAlertController alertControllerWithTitle:title message:message preferredStyle:UIAlertControllerStyleAlert];

     [NSTimer scheduledTimerWithTimeInterval:timerInerval target:self selector:@selector(dismissAlertController:) userInfo:alertController repeats:NO];

     [controller presentViewController:alertController animated:YES completion:nil];
 }

 + (void)alertWithController:(nonnull UIViewController *)controller
                       title:(NSString *)title
                     message:(NSString *)message
                actionTitles:(nonnull NSArray<NSString *> *)titles
                   actionOne:(ActionOne)actionOne
                   actionTwo:(ActionTwo)actionTwo {

     [self alertWithController:controller title:title message:message actionTitles:titles actionOne:actionOne actionTwo:actionTwo isAlertStyle:YES];
 }

 + (void)alertWithController:(nonnull UIViewController *)controller
                       title:(NSString *)title
                     message:(NSString *)message
                actionTitles:(nonnull NSArray<NSString *> *)titles
                   actionOne:(ActionOne)actionOne
                   actionTwo:(ActionTwo)actionTwo
                isAlertStyle:(BOOL)isAlertStyle; {

     UIAlertController *alertController = [UIAlertController alertControllerWithTitle:title message:message preferredStyle:isAlertStyle ? UIAlertControllerStyleAlert : UIAlertControllerStyleActionSheet];

     switch (titles.count) {
             break;
         case :
         {
             [alertController addAction:[UIAlertAction actionWithTitle:titles.firstObject style:UIAlertActionStyleDefault handler:^(UIAlertAction * _Nonnull action) {
                 if (actionOne) {
                     actionOne();
                 }
             }]];
         }
             break;
         case :
         {
             [alertController addAction:[UIAlertAction actionWithTitle:titles.firstObject style:UIAlertActionStyleDefault handler:^(UIAlertAction * _Nonnull action) {
                 if (actionOne) {
                     actionOne();
                 }
             }]];
             [alertController addAction:[UIAlertAction actionWithTitle:titles.lastObject style:UIAlertActionStyleDefault handler:^(UIAlertAction * _Nonnull action) {
                 if (actionTwo) {
                     actionTwo();
                 }
             }]];
         }
             break;
         default:
             break;
     }

     if (!isAlertStyle) {
         [alertController addAction:[UIAlertAction actionWithTitle:@"取消" style:UIAlertActionStyleCancel handler:NULL]];
     }

     [controller presentViewController:alertController animated:YES completion:nil];
 }

 + (void)alertWithController:(nonnull UIViewController *)controller
                       title:(NSString *)title
                     message:(NSString *)message
                actionTitles:(nonnull NSArray<NSString *> *)titles
                actionStyles:(NSArray<NSNumber *> *)actionStyles
                   actionOne:(ActionOne)actionOne
                   actionTwo:(ActionOne)actionTwo {

     [self alertWithController:controller title:title message:message actionTitles:titles actionStyles:actionStyles actionOne:actionOne actionTwo:actionTwo isAlertStyle:YES];

 }

 + (void)alertWithController:(nonnull UIViewController *)controller
                       title:(NSString *)title
                     message:(NSString *)message
                actionTitles:(nonnull NSArray<NSString *> *)titles
                actionStyles:(NSArray<NSNumber *> *)actionStyles
                   actionOne:(ActionOne)actionOne
                   actionTwo:(ActionOne)actionTwo
                isAlertStyle:(BOOL)isAlertStyle {

     UIAlertController *alertController = [UIAlertController alertControllerWithTitle:title message:message preferredStyle:isAlertStyle ? UIAlertControllerStyleAlert : UIAlertControllerStyleActionSheet];

     switch (titles.count) {
             break;
         case :
         {
             UIAlertActionStyle style = actionStyles.firstObject ? [actionStyles.firstObject integerValue] : UIAlertActionStyleDefault;
             [alertController addAction:[UIAlertAction actionWithTitle:titles.firstObject style:style handler:^(UIAlertAction * _Nonnull action) {
                 if (actionOne) {
                     actionOne();
                 }
             }]];
         }
             break;
         case :
         {
             UIAlertActionStyle styleOne = actionStyles.firstObject ? [actionStyles.firstObject integerValue] : UIAlertActionStyleDefault;
             UIAlertActionStyle styleTwo = actionStyles.lastObject ? [actionStyles.lastObject integerValue] : UIAlertActionStyleDefault;
             [alertController addAction:[UIAlertAction actionWithTitle:titles.firstObject style:styleOne handler:^(UIAlertAction * _Nonnull action) {
                 if (actionOne) {
                     actionOne();
                 }
             }]];
             [alertController addAction:[UIAlertAction actionWithTitle:titles.lastObject style:styleTwo handler:^(UIAlertAction * _Nonnull action) {
                 if (actionTwo) {
                     actionTwo();
                 }
             }]];
         }
             break;
         default:
             break;
     }

     if (!isAlertStyle) {
         [alertController addAction:[UIAlertAction actionWithTitle:@"取消" style:UIAlertActionStyleCancel handler:NULL]];
     }

     [controller presentViewController:alertController animated:YES completion:nil];

 }

 + (void)dismissAlertController:(NSTimer *)timer {

     UIAlertController *alertController = timer.userInfo;

     [alertController dismissViewControllerAnimated:YES completion:nil];

     [timer invalidate];
     timer = nil;
 }

 @end

3.调用方法~

 - (IBAction)example1:(id)sender {

     [UIAlertController alertWithController:self title:@"这是我的标题，我会自动2s消失"];

 }

 - (IBAction)example2:(id)sender {

     [UIAlertController alertWithController:self title:@"这是我的标题，我会根据时间来让我消失" timeInterval:.f];

 }
 - (IBAction)example3:(id)sender {

     [UIAlertController alertWithController:self title:@"我是带message的，我会自动2s消失" message:@"我是message"];

 }
 - (IBAction)example4:(id)sender {

     [UIAlertController alertWithController:self title:@"我是带message的，我根据时间来让我消失" message:@"我是message" timeInterval:.f];

 }
 - (IBAction)example5:(id)sender {

     [UIAlertController alertWithController:self title:@"我是带按钮的" message:@"我还是个消息" actionTitles:@[@"确定", @"取消"] actionOne:^{

         NSLog(@"点了确定了");

     } actionTwo:^{

         NSLog(@"点了取消了");

     } isAlertStyle:YES];

　　}
 - (IBAction)example6:(id)sender {

     [UIAlertController alertWithController:self title:@"我是能自定义action style的" message:@"我就是个消息" actionTitles:@[@"确定", @"取消"] actionStyles:@[@(UIAlertActionStyleDefault), @(UIAlertActionStyleDestructive)] actionOne:^{

         NSLog(@"点了确定了");

     } actionTwo:^{

         NSLog(@"点了取消了");

     } isAlertStyle:YES];

 }

 - (IBAction)example7:(id)sender {

     [UIAlertController alertWithController:self title:@"我是带按钮的" message:@"我还是个消息" actionTitles:@[@"确定", @"取消"] actionOne:^{

         NSLog(@"点了确定了");

     } actionTwo:^{

         NSLog(@"点了取消了");

     } isAlertStyle:NO];

 }

目前就写了最多两个按钮的情况，代码也算简化了不少，看着没那么乱了，再多的按钮就单独写吧，反正用得不多。-_-

最后附上demo地址吧：https://github.com/BigKingQY/XBZAlertViewController_Demo.git

写完才发现，名字里面多了个view...就不改了...-_-！！