Merge pull request #120 from devmjun/feature/add_cornerRadius_value_o…

…n_underlineFocusView Add cornerRadius value on UnderlineFocusView
kazuhiro4949 · Apr 22, 2020 · 21fc974 · 21fc974
2 parents cb1401f + 5b0f013
commit 21fc974
Showing 1 changed file with 18 additions and 8 deletions.
diff --git a/PagingKit/Menu Cells/Underline/UnderlineFocusView.swift b/PagingKit/Menu Cells/Underline/UnderlineFocusView.swift
@@ -35,7 +35,7 @@ public class UnderlineFocusView: UIView {
             underlineView.backgroundColor = underlineColor
         }
     }
-
+    
     /// The color of underline
     public var underlineHeight = CGFloat(4) {
         didSet {
@@ -45,15 +45,25 @@ public class UnderlineFocusView: UIView {
 
     public var underlineWidth: CGFloat? = nil {
         didSet {
-             if let underlineWidth = underlineWidth {
-                 widthConstraint.isActive = true
-                 widthConstraint.constant = underlineWidth
-             } else {
-                 widthConstraint.isActive = false
-             }
+            if let underlineWidth = underlineWidth {
+                widthConstraint.isActive = true
+                widthConstraint.constant = underlineWidth
+            } else {
+                widthConstraint.isActive = false
+            }
         }
     }
 
+    public var masksToBounds: Bool {
+        get { return underlineView.layer.masksToBounds }
+        set { underlineView.layer.masksToBounds = newValue }
+    }
+
+    public var cornerRadius: CGFloat {
+        get { return underlineView.layer.cornerRadius }
+        set { underlineView.layer.cornerRadius = newValue }
+    }
+
     private let widthConstraint: NSLayoutConstraint
     private let heightConstraint: NSLayoutConstraint
     private let underlineView = UIView()
@@ -97,7 +107,7 @@ public class UnderlineFocusView: UIView {
         super.init(frame: frame)
         setup()
     }
-
+    
     private func setup() {
         addSubview(underlineView)
         underlineView.translatesAutoresizingMaskIntoConstraints = false