UITableView删除添加和移动

#import "RootTableViewController.h"

@interface RootTableViewController ()

@property (nonatomic, strong) NSMutableArray *allDataArray;
@property (nonatomic, assign) UITableViewCellEditingStyle style;

@end

@implementation RootTableViewController

- (void)viewDidLoad {
    [super viewDidLoad];

    self.view.backgroundColor = [UIColor lightGrayColor];

    // 设置导航栏
    self.navigationController.navigationBar.barTintColor = [UIColor orangeColor];
    self.title = @"尹浩";
    self.navigationController.navigationBar.tintColor = [UIColor whiteColor];
    [self.navigationController.navigationBar setTitleTextAttributes:@{NSForegroundColorAttributeName : [UIColor whiteColor], NSFontAttributeName : [UIFont systemFontOfSize:20]}];

    // 处理数据
    [self handleData];

    // 注册tableView
    [self.tableView registerClass:[UITableViewCell class] forCellReuseIdentifier:@"cell"];

    // 添加右按钮
    UIBarButtonItem *button1 = [[UIBarButtonItem alloc] initWithBarButtonSystemItem:UIBarButtonSystemItemEdit target:self action:@selector(rightBarButtonItemClick:)];

    UIBarButtonItem *button2 = [[UIBarButtonItem alloc] initWithBarButtonSystemItem:UIBarButtonSystemItemBookmarks target:self action:@selector(moveClick:)];

    self.navigationItem.rightBarButtonItems = @[button1, button2];

    // 添加左按钮
    self.navigationItem.leftBarButtonItem = [[UIBarButtonItem alloc] initWithBarButtonSystemItem:UIBarButtonSystemItemAdd target:self action:@selector(leftBarButtonItemClick:)];
}

// 处理数据
- (void)handleData {

    // 1.初始化大数组
    self.allDataArray = [NSMutableArray array];

    // 2.定义三个数存放每一组学生的姓名
    NSMutableArray *array1 = @[@"泰隆", @"刀妹", @"卡牌大师", @"提莫", @"艾希", @"蛮王"].mutableCopy;

    NSMutableArray *array2 = @[@"盖聂", @"卫庄", @"天明", @"少羽", @"高月"].mutableCopy;

    NSMutableArray *array3 = @[@"尹浩", @"尹笑", @"尹双浩", @"尹冬冬", @"尹句号"].mutableCopy;

    // 3.将所有学生存放到大数组中
    [self.allDataArray addObject:array1];
    [self.allDataArray addObject:array2];
    [self.allDataArray addObject:array3];

}

#pragma mark - Table view data source

- (NSInteger)numberOfSectionsInTableView:(UITableView *)tableView {

    return self.allDataArray.count;
}

- (NSInteger)tableView:(UITableView *)tableView numberOfRowsInSection:(NSInteger)section {

    return [self.allDataArray[section] count];
}

- (UITableViewCell *)tableView:(UITableView *)tableView cellForRowAtIndexPath:(NSIndexPath *)indexPath {

    UITableViewCell *cell = [tableView dequeueReusableCellWithIdentifier:@"cell" forIndexPath:indexPath];

    // 设置数据
    NSArray *array = [self.allDataArray objectAtIndex:indexPath.section];
    cell.textLabel.text = array[indexPath.row];

    return cell;
}

// 设置行高
- (CGFloat)tableView:(UITableView *)tableView heightForRowAtIndexPath:(NSIndexPath *)indexPath {

    return 60;
}

// 取消选中状态
- (void)tableView:(UITableView *)tableView didSelectRowAtIndexPath:(NSIndexPath *)indexPath {

    [tableView deselectRowAtIndexPath:indexPath animated:YES];
}

#pragma mark - 编辑（删除和添加）

// 右按钮点击事件
- (void)rightBarButtonItemClick:(UIBarButtonItem *)sender {

    self.style = UITableViewCellEditingStyleDelete;

    // 让cell处于编辑状态
    [self.tableView setEditing:!self.tableView.editing animated:YES];

}

// 左按钮点击事件
- (void)leftBarButtonItemClick:(UIBarButtonItem *)sender {

    self.style = UITableViewCellEditingStyleInsert;

    [self.tableView setEditing:!self.tableView.editing animated:YES];
}

// 指定哪些cell可以编辑
- (BOOL)tableView:(UITableView *)tableView canEditRowAtIndexPath:(NSIndexPath *)indexPath {

    if (indexPath.section == 0 || indexPath.section == 1) {
        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 == UITableViewCellEditingStyleInsert) {

        [self.allDataArray[indexPath.section] insertObject:@"星魂" atIndex:indexPath.row + 1];

        NSIndexPath *newIndexPath = [NSIndexPath indexPathForRow:indexPath.row + 1 inSection:indexPath.section];
        [tableView insertRowsAtIndexPaths:@[newIndexPath] withRowAnimation:UITableViewRowAnimationTop];

    } else if (editingStyle == UITableViewCellEditingStyleDelete) {

        [self.allDataArray[indexPath.section] removeObjectAtIndex:indexPath.row];

        [tableView deleteRowsAtIndexPaths:@[indexPath] withRowAnimation:UITableViewRowAnimationRight];
    }
}

#pragma mark - 移动

// 让cell处于编辑状态
- (void)moveClick:(UIBarButtonItem *)sender {

    [self.tableView setEditing:!self.tableView.editing animated:YES];
}

// 设置哪些cell可以移动
- (BOOL)tableView:(UITableView *)tableView canMoveRowAtIndexPath:(NSIndexPath *)indexPath {

    return YES;
}

// 开始移动
- (void)tableView:(UITableView *)tableView moveRowAtIndexPath:(NSIndexPath *)sourceIndexPath toIndexPath:(NSIndexPath *)destinationIndexPath {

    // 获取需要修改的数据
    NSString *sourceName = [self.allDataArray[sourceIndexPath.section] objectAtIndex:sourceIndexPath.row];

    // 先将数据从当前位置移除
    [self.allDataArray[sourceIndexPath.section] removeObjectAtIndex:sourceIndexPath.row];

    // 再将数据插入到对应的位置
    [self.allDataArray[destinationIndexPath.section] insertObject:sourceName atIndex:destinationIndexPath.row];

}

// 防止随意移动
- (NSIndexPath *)tableView:(UITableView *)tableView targetIndexPathForMoveFromRowAtIndexPath:(NSIndexPath *)sourceIndexPath toProposedIndexPath:(NSIndexPath *)proposedDestinationIndexPath {

    if (sourceIndexPath.section == proposedDestinationIndexPath.section) {
        return proposedDestinationIndexPath;
    } else {
        return sourceIndexPath;
    }
}

@end