change header hidden method

nghialv · Feb 9, 2015 · 14ca51f · 14ca51f
1 parent 887512d
commit 14ca51f
Showing 1 changed file with 4 additions and 16 deletions.
diff --git a/Source/MYTableViewManager.swift b/Source/MYTableViewManager.swift
@@ -348,27 +348,15 @@ public extension MYTableViewManager {
         footerViewData[section] = data
     }
 
-    public func enableHeaderViewInSection(section: Int) {
+    public func setHeaderViewInSection(section: Int, hidden: Bool) {
         if let data = headerViewData[section] {
-            data.isEnabled = true
+            data.isEnabled = !hidden
         }
     }
 
-    public func disableHeaderViewInSection(section: Int) {
-        if let data = headerViewData[section] {
-            data.isEnabled = false
-        }
-    }
-
-    public func enableFooterViewInSection(section: Int) {
-        if let data = footerViewData[section] {
-            data.isEnabled = true
-        }
-    }
-
-    public func disableFooterViewInSection(section: Int) {
+    public func setFooterViewInSection(section: Int, hidden: Bool) {
         if let data = footerViewData[section] {
-            data.isEnabled = false
+            data.isEnabled = !hidden
         }
     }
 }