新增 -willDisplayHeaderView 和 -didEndDisplayingHeaderView回调

SmileZXLee · Nov 4, 2020 · f0c3e64 · f0c3e64
1 parent fa6cb1c
commit f0c3e64
Show file tree

Hide file tree

Showing 5 changed files with 53 additions and 3 deletions.
diff --git a/ZXTableView.podspec b/ZXTableView.podspec
@@ -1,6 +1,6 @@
 Pod::Spec.new do |s|
 s.name         = 'ZXTableView'
-s.version      = '1.1.0'
+s.version      = '1.1.1'
 s.summary      = '快速构建TableView'
 s.homepage     = 'https://github.com/SmileZXLee/ZXTableView'
 s.license      = 'MIT'

diff --git a/ZXTableView/ZXTableView.h b/ZXTableView/ZXTableView.h
@@ -81,6 +81,10 @@ NS_ASSUME_NONNULL_BEGIN
 @property (nonatomic, copy) NSArray<UITableViewRowAction *>* (^zx_editActionsForRowAtIndexPath)(NSIndexPath *indexPath);
 ///cell将要展示，把id改成对应类名即可无需强制转换
 @property (nonatomic, copy) void (^zx_willDisplayCell)(NSIndexPath *indexPath,id cell);
+///headerView将要展示，把id改成对应类名即可无需强制转换
+@property (nonatomic, copy) void (^zx_willDisplayHeaderView)(NSInteger section,id headerView);
+///headerView已经展示完毕，把id改成对应类名即可无需强制转换
+@property (nonatomic, copy) void (^zx_didEndDisplayingHeaderView)(NSInteger section,id headerView);
 ///scrollView滚动事件
 @property (nonatomic, copy) void (^zx_scrollViewDidScroll)(UIScrollView *scrollView);
 ///scrollView缩放事件

diff --git a/ZXTableView/ZXTableView.m b/ZXTableView/ZXTableView.m
@@ -208,7 +208,12 @@ -(CGFloat)tableView:(UITableView *)tableView heightForRowAtIndexPath:(NSIndexPat
 }
 #pragma mark tableView cell 将要展示
 -(void)tableView:(UITableView *)tableView willDisplayCell:(UITableViewCell *)cell forRowAtIndexPath:(NSIndexPath *)indexPath{
-    !self.zx_willDisplayCell ? : self.zx_willDisplayCell(indexPath,cell);
+    if([self.zxDelegate respondsToSelector:@selector(tableView:willDisplayCell:forRowAtIndexPath:)]){
+        [self.zxDelegate tableView:tableView willDisplayCell:cell forRowAtIndexPath:indexPath];
+    }else{
+        !self.zx_willDisplayCell ? : self.zx_willDisplayCell(indexPath,cell);
+    }
+
 }
 #pragma mark tableView HeaderView & FooterView
 -(UIView *)tableView:(UITableView *)tableView viewForHeaderInSection:(NSInteger)section{
@@ -313,6 +318,22 @@ -(CGFloat)tableView:(UITableView *)tableView heightForFooterInSection:(NSInteger
     }
 }
 
+-(void)tableView:(UITableView *)tableView willDisplayHeaderView:(UIView *)view forSection:(NSInteger)section{
+    if([self.zxDelegate respondsToSelector:@selector(tableView:willDisplayHeaderView:forSection:)]){
+        [self.zxDelegate tableView:tableView willDisplayHeaderView:view forSection:section];
+    }else{
+        !self.zx_willDisplayHeaderView ? : self.zx_willDisplayHeaderView(section,view);
+    }
+}
+
+- (void)tableView:(UITableView *)tableView didEndDisplayingHeaderView:(UIView *)view forSection:(NSInteger)section{
+    if([self.zxDelegate respondsToSelector:@selector(tableView:willDisplayHeaderView:forSection:)]){
+        [self.zxDelegate tableView:tableView didEndDisplayingHeaderView:view forSection:section];
+    }else{
+        !self.zx_didEndDisplayingHeaderView ? : self.zx_didEndDisplayingHeaderView(section,view);
+    }
+}
+
 #pragma mark - scrollView相关代理
 -(void)scrollViewDidScroll:(UIScrollView *)scrollView{
     if([self.zxDelegate respondsToSelector:@selector(scrollViewDidScroll:)]){

diff --git a/ZXTableViewDemo/ZXTableViewDemo/ZXTableView/ZXTableView.h b/ZXTableViewDemo/ZXTableViewDemo/ZXTableView/ZXTableView.h
@@ -81,6 +81,10 @@ NS_ASSUME_NONNULL_BEGIN
 @property (nonatomic, copy) NSArray<UITableViewRowAction *>* (^zx_editActionsForRowAtIndexPath)(NSIndexPath *indexPath);
 ///cell将要展示，把id改成对应类名即可无需强制转换
 @property (nonatomic, copy) void (^zx_willDisplayCell)(NSIndexPath *indexPath,id cell);
+///headerView将要展示，把id改成对应类名即可无需强制转换
+@property (nonatomic, copy) void (^zx_willDisplayHeaderView)(NSInteger section,id headerView);
+///headerView已经展示完毕，把id改成对应类名即可无需强制转换
+@property (nonatomic, copy) void (^zx_didEndDisplayingHeaderView)(NSInteger section,id headerView);
 ///scrollView滚动事件
 @property (nonatomic, copy) void (^zx_scrollViewDidScroll)(UIScrollView *scrollView);
 ///scrollView缩放事件

diff --git a/ZXTableViewDemo/ZXTableViewDemo/ZXTableView/ZXTableView.m b/ZXTableViewDemo/ZXTableViewDemo/ZXTableView/ZXTableView.m
@@ -208,7 +208,12 @@ -(CGFloat)tableView:(UITableView *)tableView heightForRowAtIndexPath:(NSIndexPat
 }
 #pragma mark tableView cell 将要展示
 -(void)tableView:(UITableView *)tableView willDisplayCell:(UITableViewCell *)cell forRowAtIndexPath:(NSIndexPath *)indexPath{
-    !self.zx_willDisplayCell ? : self.zx_willDisplayCell(indexPath,cell);
+    if([self.zxDelegate respondsToSelector:@selector(tableView:willDisplayCell:forRowAtIndexPath:)]){
+        [self.zxDelegate tableView:tableView willDisplayCell:cell forRowAtIndexPath:indexPath];
+    }else{
+        !self.zx_willDisplayCell ? : self.zx_willDisplayCell(indexPath,cell);
+    }
+
 }
 #pragma mark tableView HeaderView & FooterView
 -(UIView *)tableView:(UITableView *)tableView viewForHeaderInSection:(NSInteger)section{
@@ -313,6 +318,22 @@ -(CGFloat)tableView:(UITableView *)tableView heightForFooterInSection:(NSInteger
     }
 }
 
+-(void)tableView:(UITableView *)tableView willDisplayHeaderView:(UIView *)view forSection:(NSInteger)section{
+    if([self.zxDelegate respondsToSelector:@selector(tableView:willDisplayHeaderView:forSection:)]){
+        [self.zxDelegate tableView:tableView willDisplayHeaderView:view forSection:section];
+    }else{
+        !self.zx_willDisplayHeaderView ? : self.zx_willDisplayHeaderView(section,view);
+    }
+}
+
+- (void)tableView:(UITableView *)tableView didEndDisplayingHeaderView:(UIView *)view forSection:(NSInteger)section{
+    if([self.zxDelegate respondsToSelector:@selector(tableView:willDisplayHeaderView:forSection:)]){
+        [self.zxDelegate tableView:tableView didEndDisplayingHeaderView:view forSection:section];
+    }else{
+        !self.zx_didEndDisplayingHeaderView ? : self.zx_didEndDisplayingHeaderView(section,view);
+    }
+}
+
 #pragma mark - scrollView相关代理
 -(void)scrollViewDidScroll:(UIScrollView *)scrollView{
     if([self.zxDelegate respondsToSelector:@selector(scrollViewDidScroll:)]){