UITableView的编辑

#import "RootViewController.h"
#import "RootView.h"
#define kColor arc4random() % 256 / 255.0
@interface RootViewController ()<UITableViewDataSource,UITableViewDelegate>
@property (nonatomic, strong) RootView *rootView;
// 声明大数组, 用来存放所有学生的信息
@property (nonatomic, strong) NSMutableArray *allDataArray;
// 声明编辑样式的属性
@property (nonatomic, assign) UITableViewCellEditingStyle style;

@end

@implementation RootViewController

- (void)loadView
{
    self.rootView = [[RootView alloc] initWithFrame:[UIScreen mainScreen].bounds];
    self.view = self.rootView;
          
}

- (void)viewDidLoad {
    [super viewDidLoad];
    // Do any additional setup after loading the view.
    // 设置代理
    self.rootView.tableView.dataSource = self;
    self.rootView.tableView.delegate = self;
    // 设置数据
    [self handleData];
    // 处理导航栏
    self.title = @"管理";
    self.navigationController.navigationBar.barTintColor = [UIColor orangeColor];
    // 添加右按钮
    self.navigationItem.rightBarButtonItem = [[UIBarButtonItem alloc] initWithBarButtonSystemItem:UIBarButtonSystemItemEdit target:self action:@selector(rightBarButtonItemAction:)];
    // 添加左按钮
    self.navigationItem.leftBarButtonItem = [[UIBarButtonItem alloc] initWithBarButtonSystemItem:UIBarButtonSystemItemAdd target:self action:@selector(leftBarButtonItemAction:)];
    
}

// 设置数据
- (void)handleData
{
    // 1. 初始化大数组
    self.allDataArray = [NSMutableArray array];
    // 2. 定义三个数组存放每一组学生的姓名
    NSMutableArray *array1 = @[@"阿福", @"曲国伟", @"小楠", @"老司机", @"波波", @"唱吧小樽"].mutableCopy;
    NSMutableArray *array2 = @[@"一打", @"mbBoy", @"lida", @"BoomSky", @"正能量"].mutableCopy;
    NSMutableArray *array3 = @[@"孙超", @"傲然", @"MBBoy", @"🐟神", @"雷🐟"].mutableCopy;
    // 将所有学生存放在大数组中
    [self.allDataArray addObject:array1];
    [self.allDataArray addObject:array2];
    [self.allDataArray addObject:array3];
}

//设置分区个数
- (NSInteger)numberOfSectionsInTableView:(UITableView *)tableView
{
    return self.allDataArray.count;
}

// 设置每个分区的行数
- (NSInteger)tableView:(UITableView *)tableView numberOfRowsInSection:(NSInteger)section
{
    return [self.allDataArray[section] count];
}

// 返回cell对象
- (UITableViewCell *)tableView:(UITableView *)tableView cellForRowAtIndexPath:(NSIndexPath *)indexPath
{
    
//    　　提到UITableView，就必须的说一说NSIndexPath。UITableView声明了一个NSIndexPath的类别，主要用 来标识当前cell的在tableView中的位置，该类别有section和row两个属性，前者标识当前cell处于第几个section中，后者代 表在该section中的第几行。
    static NSString *identifier = @"cell";
    // 1. 从重用池查找cell
    UITableViewCell *cell = [tableView dequeueReusableCellWithIdentifier:identifier];
    // 2. 判断,如果没有找到创建cell对象
    if (!cell) {
        cell = [[UITableViewCell alloc] initWithStyle:UITableViewCellStyleSubtitle reuseIdentifier:identifier];
    }
    // 3. 设置cell数据
    NSArray *array = self.allDataArray[indexPath.section];
    cell.textLabel.text = array[indexPath.row];
    //cell.backgroundColor = [UIColor colorWithRed:arc4random()%256 / 255.0 green:arc4random()%256 / 255.0 blue:arc4random()%256 / 255.0 alpha:1];
    
    cell.selectionStyle = UITableViewCellSelectionStyleNone;// 取消cell点击效果
    //cell.selectedBackgroundView
    
    return cell;
    
}

#pragma mark UITableView 编辑-----

// 实现左按钮点击方法
- (void)leftBarButtonItemAction:(UIBarButtonItem *)sender
{
    //左按钮进行插入
    self.style = UITableViewCellEditingStyleInsert;
    [self.rootView.tableView setEditing:!self.rootView.tableView.editing animated:YES];
}



// 实现右按钮方法
- (void)rightBarButtonItemAction:(UIBarButtonItem *)sender
{
    // 第一步: 设置让tableView处于编辑状态
    //self.rootView.tableView.editing = YES;
//    if (self.rootView.tableView.editing) {
//        self.rootView.tableView.editing = NO;
//    } else {
//        self.rootView.tableView.editing = YES;
//    }
    // 右按钮进行删除
    self.style = UITableViewCellEditingStyleDelete;
    // 优化写法
    [self.rootView.tableView setEditing:!self.rootView.tableView.editing animated:YES];

    
}
// 第二步: 指定哪些cell可以被编辑 (默认所有cell都可以编辑)
- (BOOL)tableView:(UITableView *)tableView canEditRowAtIndexPath:(NSIndexPath *)indexPath
{
    if (indexPath.section == 0) {
        return YES;
    }
    return NO;
}

// 第三步: 设置编辑样式 (默认编辑样式为删除样式)
- (UITableViewCellEditingStyle)tableView:(UITableView *)tableView editingStyleForRowAtIndexPath:(NSIndexPath *)indexPath
{
    return self.style;

}

// 第四步: 完成编辑, 提交编辑状态
- (void)tableView:(UITableView *)tableView commitEditingStyle:(UITableViewCellEditingStyle)editingStyle forRowAtIndexPath:(NSIndexPath *)indexPath
{
    
    // 判断编辑样式
    if (editingStyle == UITableViewCellEditingStyleDelete) {
        // 1.删除数据
        [self.allDataArray[indexPath.section] removeObjectAtIndex:indexPath.row];
        // 2.更新UI
        // 更新一行
        [tableView deleteRowsAtIndexPaths:@[indexPath] withRowAnimation:UITableViewRowAnimationRight];
        // 全部更新
        //[tableView reloadData];
    } else if (editingStyle == UITableViewCellEditingStyleInsert) {
        // 1. 添加数据到数组中
        [self.allDataArray[indexPath.section] insertObject:@"你是不是啥" atIndex:indexPath.row + 1];
        
        // 2. 更行UI
     //   [tableView reloadData];
        //更新一行
        /**
         *   创建新一行的NSIndexPath对象
         */
        
        NSIndexPath *newIndexPath = [NSIndexPath indexPathForRow:indexPath.row + 1 inSection:indexPath.section];

        [tableView insertRowsAtIndexPaths:@[newIndexPath] withRowAnimation:UITableViewRowAnimationBottom];
    }
}

// 返回每一行的高度
- (CGFloat)tableView:(UITableView *)tableView heightForRowAtIndexPath:(NSIndexPath *)indexPath
{
    return 80;
}

// 点击cell取消动画
- (void)tableView:(UITableView *)tableView didSelectRowAtIndexPath:(NSIndexPath *)indexPath
{
    [tableView deselectRowAtIndexPath:indexPath animated:YES];
}