From d0b0d767250269baf68b6850db6d88b30dea97a7 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Christian=20Ko=CC=88nig?= <hi@chriskoenig.de>
Date: Sat, 10 Feb 2018 09:37:54 +0100
Subject: [PATCH] Added accessibility traits

---
 ...ButtonBarPagerTabStripViewController.swift |  2 ++
 ...ButtonBarPagerTabStripViewController.swift |  2 ++
 Sources/ButtonBarViewCell.swift               | 21 +++++++++++++++++++
 Sources/IndicatorInfo.swift                   | 16 ++++++++++++++
 4 files changed, 41 insertions(+)

diff --git a/Sources/BaseButtonBarPagerTabStripViewController.swift b/Sources/BaseButtonBarPagerTabStripViewController.swift
index f4a3ce5..818d65f 100644
--- a/Sources/BaseButtonBarPagerTabStripViewController.swift
+++ b/Sources/BaseButtonBarPagerTabStripViewController.swift
@@ -131,6 +131,7 @@ open class BaseButtonBarPagerTabStripViewController<ButtonBarCellType: UICollect
         // selectedBar is resized and its contentOffset/scroll is set correctly (the selected
         // tab/cell may end up either skewed or off screen after a rotation otherwise)
         buttonBarView.moveTo(index: currentIndex, animated: false, swipeDirection: .none, pagerScroll: .scrollOnlyIfOutOfScreen)
+        buttonBarView.selectItem(at: IndexPath(item: currentIndex, section: 0), animated: false, scrollPosition: [])
     }
 
     // MARK: - View Rotation
@@ -340,6 +341,7 @@ open class ExampleBaseButtonBarPagerTabStripViewController: BaseButtonBarPagerTa
 
     open override func configure(cell: ButtonBarViewCell, for indicatorInfo: IndicatorInfo) {
         cell.label.text = indicatorInfo.title
+        cell.accessibilityLabel = indicatorInfo.accessibilityLabel
         if let image = indicatorInfo.image {
             cell.imageView.image = image
         }
diff --git a/Sources/ButtonBarPagerTabStripViewController.swift b/Sources/ButtonBarPagerTabStripViewController.swift
index 908142c..3a7103d 100644
--- a/Sources/ButtonBarPagerTabStripViewController.swift
+++ b/Sources/ButtonBarPagerTabStripViewController.swift
@@ -194,6 +194,7 @@ open class ButtonBarPagerTabStripViewController: PagerTabStripViewController, Pa
         // selectedBar is resized and its contentOffset/scroll is set correctly (the selected
         // tab/cell may end up either skewed or off screen after a rotation otherwise)
         buttonBarView.moveTo(index: currentIndex, animated: false, swipeDirection: .none, pagerScroll: .scrollOnlyIfOutOfScreen)
+        buttonBarView.selectItem(at: IndexPath(item: currentIndex, section: 0), animated: false, scrollPosition: [])
     }
 
     // MARK: - Public Methods
@@ -323,6 +324,7 @@ open class ButtonBarPagerTabStripViewController: PagerTabStripViewController, Pa
         let indicatorInfo = childController.indicatorInfo(for: self)
 
         cell.label.text = indicatorInfo.title
+        cell.accessibilityLabel = indicatorInfo.accessibilityLabel
         cell.label.font = settings.style.buttonBarItemFont
         cell.label.textColor = settings.style.buttonBarItemTitleColor ?? cell.label.textColor
         cell.contentView.backgroundColor = settings.style.buttonBarItemBackgroundColor ?? cell.contentView.backgroundColor
diff --git a/Sources/ButtonBarViewCell.swift b/Sources/ButtonBarViewCell.swift
index f116577..933434b 100644
--- a/Sources/ButtonBarViewCell.swift
+++ b/Sources/ButtonBarViewCell.swift
@@ -29,4 +29,25 @@ open class ButtonBarViewCell: UICollectionViewCell {
     @IBOutlet open var imageView: UIImageView!
     @IBOutlet open var label: UILabel!
 
+    public required init?(coder aDecoder: NSCoder) {
+        super.init(coder: aDecoder)
+        
+        isAccessibilityElement = true
+        accessibilityTraits |= UIAccessibilityTraitButton
+        accessibilityTraits |= UIAccessibilityTraitHeader
+    }
+    
+    open override var isSelected: Bool {
+        get {
+            return super.isSelected
+        }
+        set {
+            super.isSelected = newValue
+            if (newValue) {
+                accessibilityTraits |= UIAccessibilityTraitSelected
+            } else {
+                accessibilityTraits &= ~UIAccessibilityTraitSelected
+            }
+        }
+    }
 }
diff --git a/Sources/IndicatorInfo.swift b/Sources/IndicatorInfo.swift
index a20f012..146eb9a 100644
--- a/Sources/IndicatorInfo.swift
+++ b/Sources/IndicatorInfo.swift
@@ -29,21 +29,34 @@ public struct IndicatorInfo {
     public var title: String?
     public var image: UIImage?
     public var highlightedImage: UIImage?
+    public var accessibilityLabel: String?
     public var userInfo: Any?
     
     public init(title: String?) {
         self.title = title
+        self.accessibilityLabel = title
     }
     
     public init(image: UIImage?, highlightedImage: UIImage? = nil, userInfo: Any? = nil) {
         self.image = image
         self.highlightedImage = highlightedImage
+        self.userInfo = userInfo
     }
     
     public init(title: String?, image: UIImage?, highlightedImage: UIImage? = nil, userInfo: Any? = nil) {
         self.title = title
+        self.accessibilityLabel = title
         self.image = image
         self.highlightedImage = highlightedImage
+        self.userInfo = userInfo
+    }
+    
+    public init(title: String?, accessibilityLabel:String?, image: UIImage?, highlightedImage: UIImage? = nil, userInfo: Any? = nil) {
+        self.title = title
+        self.accessibilityLabel = accessibilityLabel
+        self.image = image
+        self.highlightedImage = highlightedImage
+        self.userInfo = userInfo
     }
 
 }
@@ -52,13 +65,16 @@ extension IndicatorInfo : ExpressibleByStringLiteral {
 
     public init(stringLiteral value: String) {
         title = value
+        accessibilityLabel = value
     }
 
     public init(extendedGraphemeClusterLiteral value: String) {
         title = value
+        accessibilityLabel = value
     }
 
     public init(unicodeScalarLiteral value: String) {
         title = value
+        accessibilityLabel = value
     }
 }