Merged conflicts

Author: ivanbruel

Source/SeparatorCollectionViewFlowLayout.swift

@@ -1,7 +1,9 @@
 //
 //  SeparatorCollectionViewFlowLayout.swift
+//  ChicByChoice
 //
 //  Created by Ivan Bruel on 10/12/15.
+//  Copyright © 2015 Chic by Choice. All rights reserved.
 //
 
 import Foundation
@@ -21,6 +23,14 @@ public class SeparatorCollectionViewFlowLayout: UICollectionViewFlowLayout {
     }
   }
 
+  private static let topSeparatorKind = "SeparatorCollectionViewFlowLayout.Top"
+  private static let bottomSeparatorKind = "SeparatorCollectionViewFlowLayout.Bottom"
+  private static let leftSeparatorKind = "SeparatorCollectionViewFlowLayout.Left"
+  private static let rightSeparatorKind = "SeparatorCollectionViewFlowLayout.Right"
+
+  private static let separatorKinds = [leftSeparatorKind, rightSeparatorKind, topSeparatorKind,
+    bottomSeparatorKind]
+
   init(separatorWidth: CGFloat = 1, separatorColor: UIColor = UIColor.blackColor()) {
     self.separatorWidth = separatorWidth
     self.separatorColor = separatorColor
@@ -39,13 +49,13 @@ public class SeparatorCollectionViewFlowLayout: UICollectionViewFlowLayout {
     super.prepareLayout()
 
     registerClass(SeparatorView.self,
-      forDecorationViewOfKind: SeparatorKind.topSeparator)
+      forDecorationViewOfKind: SeparatorCollectionViewFlowLayout.topSeparatorKind)
     registerClass(SeparatorView.self,
-      forDecorationViewOfKind: SeparatorKind.bottomSeparator)
+      forDecorationViewOfKind: SeparatorCollectionViewFlowLayout.bottomSeparatorKind)
     registerClass(SeparatorView.self,
-      forDecorationViewOfKind: SeparatorKind.rightSeparator)
+      forDecorationViewOfKind: SeparatorCollectionViewFlowLayout.leftSeparatorKind)
     registerClass(SeparatorView.self,
-      forDecorationViewOfKind: SeparatorKind.leftSeparator)
+      forDecorationViewOfKind: SeparatorCollectionViewFlowLayout.rightSeparatorKind)
   }
 
   override public func layoutAttributesForDecorationViewOfKind(elementKind: String,
@@ -61,12 +71,27 @@ public class SeparatorCollectionViewFlowLayout: UICollectionViewFlowLayout {
 
       let baseFrame = cellAttributes.frame
 
-      guard let transformedFrame = SeparatorKind.init(rawValue: elementKind)?.transform(frame: baseFrame, with: separatorWidth) else {
-        return nil
+      switch elementKind {
+      case SeparatorCollectionViewFlowLayout.rightSeparatorKind:
+        layoutAttributes.frame = CGRect(x: baseFrame.maxX,
+          y: baseFrame.minY - separatorWidth, width: separatorWidth,
+          height: baseFrame.height + separatorWidth * 2)
+      case SeparatorCollectionViewFlowLayout.leftSeparatorKind:
+        layoutAttributes.frame = CGRect(x: baseFrame.minX - separatorWidth,
+          y: baseFrame.minY - separatorWidth, width: separatorWidth,
+          height: baseFrame.height + separatorWidth * 2)
+      case SeparatorCollectionViewFlowLayout.topSeparatorKind:
+        layoutAttributes.frame = CGRect(x: baseFrame.minX,
+          y: baseFrame.minY - separatorWidth, width: baseFrame.width,
+          height: separatorWidth)
+      case SeparatorCollectionViewFlowLayout.bottomSeparatorKind:
+        layoutAttributes.frame = CGRect(x: baseFrame.minX,
+          y: baseFrame.maxY, width: baseFrame.width,
+          height: separatorWidth)
+      default:
+        break
       }
 
-      layoutAttributes.frame = transformedFrame
-
       layoutAttributes.zIndex = -1
       layoutAttributes.color = separatorColor
 
@@ -83,9 +108,10 @@ public class SeparatorCollectionViewFlowLayout: UICollectionViewFlowLayout {
       baseLayoutAttributes.filter { $0.representedElementCategory == .Cell }.forEach {
         (layoutAttribute) -> () in
 
-        layoutAttributes += SeparatorKind.elements.flatMap { (kind) -> UICollectionViewLayoutAttributes? in
-          layoutAttributesForDecorationViewOfKind(kind.rawValue, atIndexPath: layoutAttribute.indexPath)
-        }
+        layoutAttributes += SeparatorCollectionViewFlowLayout.separatorKinds.flatMap {
+          (kind) -> UICollectionViewLayoutAttributes? in
+            layoutAttributesForDecorationViewOfKind(kind, atIndexPath: layoutAttribute.indexPath)
+          }
       }
 
       return layoutAttributes
@@ -107,44 +133,7 @@ private class SeparatorView: UICollectionReusableView {
     guard let coloredLayoutAttributes = layoutAttributes as? ColoredViewLayoutAttributes else {
       return
     }
-
     backgroundColor = coloredLayoutAttributes.color
   }
 
 }
-
-private enum SeparatorKind: String {
-  case topSeparator = "SeparatorCollectionViewFlowLayout.Top"
-  case bottomSeparator = "SeparatorCollectionViewFlowLayout.Bottom"
-  case leftSeparator = "SeparatorCollectionViewFlowLayout.Left"
-  case rightSeparator = "SeparatorCollectionViewFlowLayout.Right"
-
-  static var elements: [SeparatorKind] {
-    get {
-      return [.leftSeparator, .rightSeparator, .topSeparator, .bottomSeparator]
-    }
-  }
-
-  func transform(frame baseFrame: CGRect, with separatorWidth: CGFloat) -> CGRect {
-    switch self {
-
-    case .topSeparator:
-      return CGRect(x: baseFrame.minX, y: baseFrame.minY - separatorWidth, width: baseFrame.width, height: separatorWidth)
-
-    case .bottomSeparator:
-      return CGRect(x: baseFrame.minX, y: baseFrame.maxY, width: baseFrame.width, height: separatorWidth)
-
-    case .leftSeparator:
-      return CGRect(x: baseFrame.minX - separatorWidth, y: baseFrame.minY - separatorWidth, width: separatorWidth, height: baseFrame.height + separatorWidth * 2)
-
-    case .rightSeparator:
-      return CGRect(x: baseFrame.maxX, y: baseFrame.minY - separatorWidth, width: separatorWidth, height: baseFrame.height + separatorWidth * 2)
-    }
-  }
-}
-
-extension UICollectionViewFlowLayout {
-  private func registerClass(viewClass: AnyClass?, forDecorationViewOfKind elementKind: SeparatorKind) {
-    registerClass(viewClass, forDecorationViewOfKind: elementKind.rawValue)
-  }
-}