偷樑換柱 - iOS實現UITextField+Limit

BackNotGod發表於2018-10-25

用例分析

在使用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
複製程式碼
  1. 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內所有方法會失效,這個問題,我們稍後再說。

  1. 實現- (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的例項。

  1. 接下來就簡單了,將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];
複製程式碼
  1. 至此,一個基於 facebook/KVOController思想的小工具就出爐了,雖然簡單,但是需要這種思想還是比較巧妙的。

最後貼出原始碼地址,歡迎指正# BNTextField-Limit

相關文章