UITextField输入字数限制实现

开发过程中，我们会收到产品各种需求，比如使用手机号码登录或注册时限制输入字数为11位，那应该怎么实现呢？UITextField系统代理方法也没有提供一个相关的方法，我使用如下方法实现。

一、实现

给UITextField创建Category类别UITextField+LimitLength，利用runtime给UITextField实力对象绑定一个限制输入的最大长度Value，添加一个“检查和过滤超限的字符串”方法，在适当时机调用过滤超限字符串，达到限制输入的目的。

1. UITextField+LimitLength.h文件实现如下：

//
//  UITextField+LimitLength.h
//
//  Created by yyz on 2017/6/17.
//  Copyright © 2017年 yyz. All rights reserved.
//

#import <UIKit/UIKit.h>

@interface UITextField (LimitLength)

/**
 绑定限制输入的最大长度(建议在代理方法textFieldDidBeginEditing:中调用)

 @param inputLimitMaxLength 限制输入的最大长度
 */
- (void)setInputLimitMaxLength:(NSUInteger)inputLimitMaxLength;

/**
 移除限制输入的最大长度(建议在代理方法textFieldDidEndEditing:中调用)
 */
- (void)removeInputLimitMaxLength;

/**
 检查和过滤超限的字符串

 @return 超出限制返回YES，否则返回NO
 */
- (BOOL)checkAndFilterBeyondLimitsCharacters;

@end

2. UITextField+LimitLength.m文件实现如下：

//
//  UITextField+LimitLength.m
//
//  Created by yyz on 2017/6/17.
//  Copyright © 2017年 yyz. All rights reserved.
//

#import <objc/runtime.h>
#import "UITextField+LimitLength.h"

@implementation UITextField (LimitLength)

- (void)setInputLimitMaxLength:(NSUInteger)inputLimitMaxLength
{
    objc_setAssociatedObject(self,
                             @selector(checkAndFilterBeyondLimitsCharacters),
                             @(inputLimitMaxLength),
                             OBJC_ASSOCIATION_RETAIN);
}

- (NSUInteger)getCurrentInputLimitMaxLength
{
    return [objc_getAssociatedObject(self, @selector(checkAndFilterBeyondLimitsCharacters)) unsignedIntegerValue];
}

- (void)removeInputLimitMaxLength
{
    objc_setAssociatedObject(self,
                             @selector(checkAndFilterBeyondLimitsCharacters),
                             nil,
                             OBJC_ASSOCIATION_ASSIGN);
}

- (BOOL)checkAndFilterBeyondLimitsCharacters
{
    BOOL beyondLimits = NO;
    NSUInteger inputMaxLength =  [self getCurrentInputLimitMaxLength];
    if (self && inputMaxLength) {
        NSString *toBeString = self.text;
        
        // 获取高亮部分
        UITextRange *selectedRange = [self markedTextRange];
        UITextPosition *position = [self positionFromPosition:selectedRange.start offset:0];
        
        // 没有高亮选择的字，则对已输入的文字进行字数统计和限制
        if (!position) {
            if (toBeString.length > inputMaxLength) {
                beyondLimits = YES;
                NSRange rangeIndex = [toBeString rangeOfComposedCharacterSequenceAtIndex:inputMaxLength];
                if (rangeIndex.length == 1) {
                    self.text = [toBeString substringToIndex:inputMaxLength];
                } else {
                    self.text = [toBeString substringToIndex:rangeIndex.location];
                }
            }
        }
    }
    
    return beyondLimits;
}

@end

二、使用示例

给UITextField实力对象添加一个UIControlEventEditingChangedTarget事件，在action方法中调用过滤超限的方法。

UITextField *inputMobileTF = [[UITextField alloc] init];
[inputMobileTF addTarget:self
                  action:@selector(inputMobileTextFieldValueDidChanged:)
        forControlEvents:UIControlEventEditingChanged];
...// 其他代码省略

- (void)inputMobileTextFieldValueDidChanged:(UITextField *)textField
{
    [textField checkAndFilterBeyondLimitsCharacters]; // 过滤超限字符串
    NSLog(@"%@", textField.text);// 得到的就是过滤后的结果
}

在UITextFieldDelegate代理方法中调用设置限制输入字数相关方法即可，如下所示：

#pragma mark - UITextFieldDelegate
- (void)textFieldDidBeginEditing:(UITextField *)textField
{
    if (textField.tag == 100) {
        [textField setInputLimitMaxLength:11]; // 手机号限制输入11位
    } else if (textField.tag == 101) {
        [textField setInputLimitMaxLength:6];  // 验证码限制输入6位
    }
}

- (void)textFieldDidEndEditing:(UITextField *)textField
{
    [textField removeInputLimitMaxLength]; // 移除输入限制
}

三、源码地址

工程GitHub地址

UITextField+LimitLength扩展类GitHub地址