From 445600534d9bb04580d20ed24f41e6d78c9f2d6b Mon Sep 17 00:00:00 2001
From: Sergey Kopytov <kopytov@me.com>
Date: Mon, 17 Dec 2018 15:16:30 +0300
Subject: [PATCH] add functions to sections

---
 .../TableDirector+Extensions.swift            | 73 +++++++++++++++++++
 1 file changed, 73 insertions(+)

diff --git a/Sources/Extensions/TableKit/TableDirector/TableDirector+Extensions.swift b/Sources/Extensions/TableKit/TableDirector/TableDirector+Extensions.swift
index 804feb5c..609a3573 100644
--- a/Sources/Extensions/TableKit/TableDirector/TableDirector+Extensions.swift
+++ b/Sources/Extensions/TableKit/TableDirector/TableDirector+Extensions.swift
@@ -172,4 +172,77 @@ public extension TableDirector {
         }
     }
 
+    /**
+     Method inserts section with animation
+
+     - parameter section: Section to insert
+     - parameter index: Position to insert
+     - parameter animation: The type of insert animation
+     - parameter manualBeginEndUpdates: Don't call beginUpdates() & endUpdates() inside.
+
+     - returns: self
+     */
+    @discardableResult
+    func insert(section: TableSection,
+                at index: Int,
+                with animation: UITableView.RowAnimation,
+                manualBeginEndUpdates: Bool = false) -> Self {
+
+        insert(section: section, atIndex: index)
+        if manualBeginEndUpdates {
+            tableView?.insertSections([index], with: animation)
+        } else {
+            tableView?.beginUpdates()
+            tableView?.insertSections([index], with: animation)
+            tableView?.endUpdates()
+        }
+
+        return self
+    }
+
+    /**
+     Method removes section with animation
+
+     - parameter index: Position to remove
+     - parameter animation: The type of remove animation
+     - parameter manualBeginEndUpdates: Don't call beginUpdates() & endUpdates() inside.
+
+     - returns: self
+     */
+    @discardableResult
+    func remove(at index: Int,
+                with animation: UITableView.RowAnimation,
+                manualBeginEndUpdates: Bool = false) -> Self {
+
+        delete(sectionAt: index)
+        if manualBeginEndUpdates {
+            tableView?.deleteSections([index], with: animation)
+        } else {
+            tableView?.beginUpdates()
+            tableView?.deleteSections([index], with: animation)
+            tableView?.endUpdates()
+        }
+
+        return self
+    }
+
+    /**
+     Method replace section with animation
+
+     - parameter section: Section to replace
+     - parameter index: Position to replace
+     - parameter animation: The type of replace animation
+     - parameter manualBeginEndUpdates: Don't call beginUpdates() & endUpdates() inside.
+
+     - returns: self
+     */
+    func replace(with section: TableSection,
+                 at index: Int,
+                 with animation: UITableView.RowAnimation,
+                 manualBeginEndUpdates: Bool = false) -> Self {
+
+        remove(at: index, with: animation, manualBeginEndUpdates: manualBeginEndUpdates)
+        return insert(section: section, at: index, with: animation, manualBeginEndUpdates: manualBeginEndUpdates)
+    }
+
 }