偷樑換柱 - iOS實現UITextField+Limit

用例分析

在使用UITextField的過程當中，難免會有限制字符個數，字符輸入規則的需求。通常狀況下，會有以下兩種方法：

• 直接設置代理，實現代理方法，- (BOOL)textField:(UITextField *)textField shouldChangeCharactersInRange:(NSRange)range replacementString:(NSString *)string
• 封裝代理過程，利用block來實現回調
• 固然方法不止這兩種，這裏只是舉經常使用的例子，再也不贅述

BNTextField-Limit的方法

依然是利用block回調，不過實現方式有點不一樣。

[testField limitCondition:^BOOL(NSString *inputStr){
        return ![testField.text isEqualToString:@"111"];
    } action:^{
        NSLog(@"limit action");
}];
Or

[testField limitNums:3 action:^{
	NSLog(@"num limit action");
}];
複製代碼

BNTextField-Limit的實現策略

對於UITextField用來作字符限制最好的方法就是使用- (BOOL)textField:(UITextField *)textField shouldChangeCharactersInRange:(NSRange)range replacementString:(NSString *)string這個代理方法，咱們經過判斷string來肯定UITextField是否響應輸入。

接下來就是如何封裝好代理回調的這個過程了 這裏我借鑑了 facebook/**KVOController**的思想，建立一箇中間管理者，來接管代理方法。 不過須要考慮幾個問題：

• 代理釋放的問題
• 多個條件約束
• 如何不影響其餘代理方法

實現過程:

1. 首先，咱們假定，AController內實現了UITextField的delegate，咱們先把delegate的身份接管過來，實現偷樑換柱。

// self 即UITextField 這裏這是一個分類方法
  self.delegate =  UITextFieldDelegateManager.sharedInstance
複製代碼

2. UITextFieldDelegateManager爲中間管理類， keyCode：

@interface UITextFieldDelegateManager : NSObject<UITextFieldDelegate> {
    NSMapTable<id,_LimitInfo *> *_infos;
}

+ (instancetype)sharedInstance;

- (void)addLimitNums:(NSInteger)num key:(id)key target:(id<UITextFieldDelegate>)target action:(void(^)(void))action;

@end
複製代碼

@interface _LimitInfo : NSObject

@property(nonatomic,assign)NSInteger num;
@property(nonatomic,weak)id<UITextFieldDelegate> pinocchio;

@end

複製代碼

這時，咱們的UITextField的delegate成爲了UITextFieldDelegateManager，這樣咱們就「截獲」了AController的delgate身份。 而這裏有一個問題，那就是AController的UITextFieldDelegate內全部方法會失效，這個問題，咱們稍後再說。

3. 實現- (void)addLimitNums:(NSInteger)num key:(id)key target:(id<UITextFieldDelegate>)target action:(void(^)(void))action;

_LimitInfo *info = [_infos objectForKey:key];
    
    if (!info) {
        info = [_LimitInfo new];
        info.pinocchio = target;
    }
    
    info.condition = condition;
    [info setConditionAction:action];
    [_infos setObject:info forKey:key];
複製代碼

這裏Key是UITextField當前實例對象，target是AController，咱們把這二者映射進一個NSMapTable中，NSMapTable的弱引用會使咱們不用擔憂循環引用。其做用和字典同樣。

同時，也解決了多個條件約束的問題。

而_LimitInfo只是對AController的一個包裝，到這時，咱們的AController已經被架空了，成爲了一個受咱們擺佈的傀儡😏，pinocchio保存了AController的實例。

4. 接下來就簡單了，將UITextFieldDelegate在UITextFieldDelegateManager中所有實現出來 主要是咱們的shouldChangeCharactersInRange代理方法

- (BOOL)textField:(UITextField *)textField shouldChangeCharactersInRange:(NSRange)range replacementString:(NSString *)string{
    
    BOOL checkInLimit = NO;
    
    _LimitInfo *info = [self safeReadForKey:textField];
    if (info.condition && !info.condition(string) && string.length > 0) {
        info.conditionAction();
        checkInLimit = YES;
    }
    
    if (info.num != 0) {
        if (info && textField.text.length == info.num && string.length > 0) {
            info.action();
            checkInLimit = YES;
        }
    }
    
    if (checkInLimit) {
        return NO;
    }
    
    if (!info.pinocchio) {
        return YES;
    }
    
    return [info.pinocchio textField:textField shouldChangeCharactersInRange:range replacementString:string];
}
複製代碼

其餘方法相似，具體能夠參見源碼

不過這裏還要注意咱們剛剛提到的問題，經過咱們的pinocchio return [info.pinocchio textField:textField shouldChangeCharactersInRange:range replacementString:string];來控制本來邏輯，否則delegate就失效了

至於代理釋放的問題，我是經過runtime hook UITextField的removeFromSuperview方法，在這個方法調用的時候，將pinocchio從新設置回UITextField的delegate，同時移除緩存。

_LimitInfo* info = [_infos objectForKey:key];
  ((UITextField*)key).delegate = info.pinocchio;
  [_infos removeObjectForKey:key];
複製代碼

5. 至此，一個基於 facebook/KVOController思想的小工具就出爐了，雖然簡單，可是須要這種思想仍是比較巧妙的。

最後貼出源碼地址，歡迎指正# BNTextField-Limit