Implement tableView(_:accessoryButtonTappedForRowWith:)

Author: bcylin

Source/Protocol/RowCompatible.swift

@@ -27,7 +27,12 @@
 import Foundation
 
 /// This protocol defines the compatible interface of a `NavigationRow` regardless of its associated cell type.
-public protocol NavigationRowCompatible: Row, RowStyle {}
+public protocol NavigationRowCompatible: Row, RowStyle {
+  #if os(iOS)
+  /// A closure that will be invoked when the accessory button is selected.
+  var accessoryButtonAction: ((Row) -> Void)? { get }
+  #endif
+}
 
 
 /// This protocol defines the compatible interface of a `TapActionRow` regardless of its associated cell type.

Source/QuickTableViewController.swift

@@ -165,6 +165,17 @@ open class QuickTableViewController: UIViewController, UITableViewDataSource, UI
     }
   }
 
+  #if os(iOS)
+  public func tableView(_ tableView: UITableView, accessoryButtonTappedForRowWith indexPath: IndexPath) {
+    switch tableContents[indexPath.section].rows[indexPath.row] {
+    case let row as NavigationRowCompatible:
+      row.accessoryButtonAction?(row)
+    default:
+      break
+    }
+  }
+  #endif
+
 }
 
 

Tests/ViewController/QuickTableViewDelegateSpec.swift

@@ -264,6 +264,34 @@ internal final class QuickTableViewDelegateSpec: QuickSpec {
           }
         }
       }
+
+      #if os(iOS)
+
+      // MARK: - tableView(_:accessoryButtonTappedForRowWith:)
+      describe("tableView(_:accessoryButtonTappedForRowWith:)") {
+        let controller = QuickTableViewController()
+        _ = controller.view
+        var selectedIndex = -1
+
+        controller.tableContents = [
+          Section(title: "NavigationRow", rows: [
+            NavigationRow(text: "", detailText: .none, accessoryButtonAction: { _ in selectedIndex = 0 }),
+            CustomNavigationRow(text: "", detailText: .none, accessoryButtonAction: { _ in selectedIndex = 1 }),
+            NavigationRow<CustomCell>(text: "", detailText: .none, accessoryButtonAction: { _ in selectedIndex = 2 }),
+            CustomNavigationRow<CustomCell>(text: "", detailText: .none, accessoryButtonAction: { _ in selectedIndex = 3 })
+          ])
+        ]
+
+        for index in 0...3 {
+          it("should invoke action when accessory button at \(index) is selected") {
+            controller.tableView(controller.tableView, accessoryButtonTappedForRowWith: IndexPath(row: index, section: 0))
+            expect(selectedIndex).toEventually(equal(index))
+          }
+        }
+      }
+
+      #endif
+
     }
   }