From 39b4df6ee92dd4e25ee7e0cbc52a797ca835b558 Mon Sep 17 00:00:00 2001
From: Pavel Gurov <pegurov@avito.ru>
Date: Thu, 13 Apr 2017 11:37:08 +0300
Subject: [PATCH] moved indexTitle to TableSection

---
 Sources/TableDirector.swift | 19 +++++++++++++------
 Sources/TableSection.swift  |  1 +
 2 files changed, 14 insertions(+), 6 deletions(-)

diff --git a/Sources/TableDirector.swift b/Sources/TableDirector.swift
index 925aeb0..3d6a175 100644
--- a/Sources/TableDirector.swift
+++ b/Sources/TableDirector.swift
@@ -31,8 +31,7 @@ open class TableDirector: NSObject, UITableViewDataSource, UITableViewDelegate {
     private weak var scrollDelegate: UIScrollViewDelegate?
     private var cellRegisterer: TableCellRegisterer?
     public private(set) var rowHeightCalculator: RowHeightCalculator?
-    public var sectionIndexTitles: [String]?
-    public var onSectionForIndex: ((_ title: String, _ index: Int) -> Int)?
+    private var sectionsIndex = [Int]()
     
     @available(*, deprecated, message: "Produced incorrect behaviour")
     open var shouldUsePrototypeCellHeightCalculation: Bool = false {
@@ -197,13 +196,21 @@ open class TableDirector: NSObject, UITableViewDataSource, UITableViewDelegate {
     // MARK: UITableViewDataSource - Index
     
     public func sectionIndexTitles(for tableView: UITableView) -> [String]? {
-        return sectionIndexTitles
+        
+        var indexTitles = [String]()
+        sectionsIndex = []
+        sections.enumerated().forEach { index, section in
+            
+            if let title = section.indexTitle {
+                indexTitles.append(title)
+                sectionsIndex.append(index)
+            }
+        }
+        return indexTitles.isEmpty ? nil : indexTitles
     }
     
     public func tableView(_ tableView: UITableView, sectionForSectionIndexTitle title: String, at index: Int) -> Int {
-        guard let onSectionForIndex = onSectionForIndex else { return 0 }
-        
-        return onSectionForIndex(title, index)
+        return sectionsIndex[index]
     }
     
     // MARK: UITableViewDelegate - actions
diff --git a/Sources/TableSection.swift b/Sources/TableSection.swift
index bc50522..ddc93e7 100644
--- a/Sources/TableSection.swift
+++ b/Sources/TableSection.swift
@@ -26,6 +26,7 @@ open class TableSection {
     
     open var headerTitle: String?
     open var footerTitle: String?
+    open var indexTitle: String?
     
     open var headerView: UIView?
     open var footerView: UIView?