diff --git a/Yep/ViewControllers/Login/LoginVerifyMobileViewController.swift b/Yep/ViewControllers/Login/LoginVerifyMobileViewController.swift index cb02283c5..78ae005d7 100644 --- a/Yep/ViewControllers/Login/LoginVerifyMobileViewController.swift +++ b/Yep/ViewControllers/Login/LoginVerifyMobileViewController.swift @@ -231,5 +231,14 @@ extension LoginVerifyMobileViewController: UITextFieldDelegate { return true } */ + + func textField(textField: UITextField, shouldChangeCharactersInRange range: NSRange, replacementString string: String) -> Bool { + let currentCharacterCount = textField.text?.characters.count ?? 0 + if range.length + range.location > currentCharacterCount { + return false + } + let newLength = currentCharacterCount + string.characters.count - range.length + return newLength <= YepConfig.verifyCodeLength() + } } diff --git a/Yep/ViewControllers/Register/RegisterVerifyMobileViewController.swift b/Yep/ViewControllers/Register/RegisterVerifyMobileViewController.swift index 5997e90a4..ad5314e70 100644 --- a/Yep/ViewControllers/Register/RegisterVerifyMobileViewController.swift +++ b/Yep/ViewControllers/Register/RegisterVerifyMobileViewController.swift @@ -225,5 +225,14 @@ extension RegisterVerifyMobileViewController: UITextFieldDelegate { return true } */ + + func textField(textField: UITextField, shouldChangeCharactersInRange range: NSRange, replacementString string: String) -> Bool { + let currentCharacterCount = textField.text?.characters.count ?? 0 + if range.length + range.location > currentCharacterCount { + return false + } + let newLength = currentCharacterCount + string.characters.count - range.length + return newLength <= YepConfig.verifyCodeLength() + } }