Support custom number of digits feature

Author: vinothkeepworks

Example/KWVerificationCodeView/AppDelegate.swift

@@ -32,4 +32,3 @@ class AppDelegate: UIResponder, UIApplicationDelegate {
     func applicationWillTerminate(_ application: UIApplication) {
     }
 }
-

Example/KWVerificationCodeView/VerificationCodeViewController.swift

@@ -14,15 +14,15 @@ class VerificationCodeViewController: UIViewController {
   // MARK: - IBOutlets
   @IBOutlet weak var verificationCodeView: KWVerificationCodeView!
   @IBOutlet weak var submitButton: UIButton!
-  
+
   // MARK: - Lifecycle
   override func viewDidLoad() {
     super.viewDidLoad()
-    
+
     submitButton.isEnabled = false
     verificationCodeView.delegate = self
   }
-    
+
   // MARK: - IBAction
   @IBAction func submitButtonTapped(_ sender: UIButton) {
     if verificationCodeView.hasValidCode() {

Example/Tests/Tests.swift

@@ -3,27 +3,27 @@ import XCTest
 import KWVerificationCodeView
 
 class Tests: XCTestCase {
-    
+
     override func setUp() {
         super.setUp()
         // Put setup code here. This method is called before the invocation of each test method in the class.
     }
-    
+
     override func tearDown() {
         // Put teardown code here. This method is called after the invocation of each test method in the class.
         super.tearDown()
     }
-    
+
     func testExample() {
         // This is an example of a functional test case.
         XCTAssert(true, "Pass")
     }
-    
+
     func testPerformanceExample() {
         // This is an example of a performance test case.
-        self.measure() {
+        self.measure {
             // Put the code you want to measure the time of here.
         }
     }
-    
+
 }

KWVerificationCodeView/Classes/Extensions/StringExtension.swift

@@ -12,7 +12,7 @@ extension String {
   func trim() -> String {
     return trimmingCharacters(in: CharacterSet.whitespaces)
   }
-  
+
   func range(from nsRange: NSRange) -> Range<String.Index>? {
     guard
       let from16 = utf16.index(utf16.startIndex, offsetBy: nsRange.location, limitedBy: utf16.endIndex),

KWVerificationCodeView/Classes/KWTextFieldView.swift

@@ -15,31 +15,31 @@ protocol KWTextFieldDelegate: class {
 }
 
 @IBDesignable class KWTextFieldView: UIView {
-  
+
   // MARK: - Constants
   static let maxCharactersLength = 1
-  
+
   // MARK: - IBInspectables
   @IBInspectable var underlineColor: UIColor = UIColor.darkGray {
     didSet {
       underlineView.backgroundColor = underlineColor
     }
   }
-  
+
   @IBInspectable var underlineSelectedColor: UIColor = UIColor.black
-  
+
   @IBInspectable var textColor: UIColor = UIColor.darkText {
     didSet {
       numberTextField.textColor = textColor
     }
   }
-  
+
   @IBInspectable var textSize: CGFloat = 24.0 {
     didSet {
       numberTextField.font = UIFont.systemFont(ofSize: textSize)
     }
   }
-  
+
   @IBInspectable var textFont: String = "" {
     didSet {
       if let font = UIFont(name: textFont, size: textSize) {
@@ -49,77 +49,82 @@ protocol KWTextFieldDelegate: class {
       }
     }
   }
-  
+
   @IBInspectable var textFieldBackgroundColor: UIColor = UIColor.clear {
     didSet {
       numberTextField.backgroundColor = textFieldBackgroundColor
     }
   }
-  
+
   @IBInspectable var textFieldTintColor: UIColor = UIColor.blue {
     didSet {
       numberTextField.tintColor = textFieldTintColor
     }
   }
-  
+
   @IBInspectable var darkKeyboard: Bool = false {
     didSet {
       keyboardAppearance = darkKeyboard ? .dark : .light
       numberTextField.keyboardAppearance = keyboardAppearance
     }
   }
-  
+
   // MARK: - IBOutlets
   @IBOutlet weak var numberTextField: UITextField!
   @IBOutlet weak private var underlineView: UIView!
-  
+
   // MARK: - Variables
   private var keyboardAppearance = UIKeyboardAppearance.default
   weak var delegate: KWTextFieldDelegate?
-  
+
   // MARK: - Lifecycle
   override init(frame: CGRect) {
     super.init(frame: frame)
-    
-    loadViewFromNib()
+
+    setup()
   }
-  
+
   required public init?(coder aDecoder: NSCoder) {
     super.init(coder: aDecoder)
-    
-    loadViewFromNib()
-    numberTextField.delegate = self
-    NotificationCenter.default.addObserver(self, selector: #selector(textFieldDidChange(_:)), name: NSNotification.Name.UITextFieldTextDidChange, object: numberTextField)
+
+    setup()
   }
-  
+
   deinit {
     NotificationCenter.default.removeObserver(self)
   }
-  
+
+  // MARK: - Private Methods
+  private func setup() {
+    loadViewFromNib()
+    numberTextField.delegate = self
+    NotificationCenter.default.addObserver(self, selector: #selector(textFieldDidChange(_:)), name: NSNotification.Name.UITextFieldTextDidChange, object: numberTextField)
+  }
+
   // MARK: - Public Methods
   public func activate() {
     numberTextField.becomeFirstResponder()
-    if numberTextField.text?.characters.count == 0 {
+    if numberTextField.text?.count == 0 {
       numberTextField.text = " "
     }
   }
-  
+
   public func deactivate() {
     numberTextField.resignFirstResponder()
   }
-  
+
   public func reset() {
     numberTextField.text = " "
     updateUnderline()
   }
-  
+
   // MARK: - FilePrivate Methods
   dynamic fileprivate func textFieldDidChange(_ notification: Foundation.Notification) {
-    if numberTextField.text?.characters.count == 0 {
+    if numberTextField.text?.count == 0 {
       numberTextField.text = " "
     }
   }
-  
+
   fileprivate func updateUnderline() {
     underlineView.backgroundColor = numberTextField.text?.trim() != "" ? underlineSelectedColor : underlineColor
   }
@@ -130,18 +135,18 @@ extension KWTextFieldView: UITextFieldDelegate {
   public func textField(_ textField: UITextField, shouldChangeCharactersIn range: NSRange, replacementString string: String) -> Bool {
     let currentString = numberTextField.text!
     let newString = currentString.replacingCharacters(in: textField.text!.range(from: range)!, with: string)
-    
-    if newString.characters.count > type(of: self).maxCharactersLength {
+
+    if newString.count > type(of: self).maxCharactersLength {
       delegate?.moveToNext(self)
       textField.text = string
-    } else if newString.characters.count == 0 {
+    } else if newString.count == 0 {
       delegate?.moveToPrevious(self, oldCode: textField.text!)
       numberTextField.text = " "
     }
-    
+
     delegate?.didChangeCharacters()
     updateUnderline()
-    
-    return newString.characters.count <= type(of: self).maxCharactersLength
+
+    return newString.count <= type(of: self).maxCharactersLength
   }
 }