封装一个匹配特定字符可点击的Label

这个功能用的最多的就是在微博里面，里面有用户名@南山忆，网址http://www.jianshu.com/users/774b1d5616a7/latest_articles，#话题# 等需要特别现实，还需要有点击事件，这种情况下普通的label显示已经满足不了需求了。需要我们来自己写一个可实现相应功能的label。
　　好了废话不多说，让我们开始，大致流程如下：

屏幕快照 2016-09-09 上午11.35.48.png

1、主要用到的属性

textStorage，layoutManager，textContainer的关系
1）NSTextStorage是NSMutableAttributedString的子类，由于可以灵活地往文字添加或修改属性，所以非常适用于保存并修改文字属性。
2）NSLayoutManager用来管理NSTextStorage其中的文字内容的排版布局。
3）NSTextContainer定义了一个矩形区域用于存放已经进行了排版并设置好属性的文字
三者关系如图所示

label.png

//是NSMutableAttributedString的子类，由于可以灵活地往文字添加或修改属性，所以非常适用于保存并修改文字属性。
@property (nonatomic, strong) NSTextStorage *textStorage;
//管理NSTextStorage其中的文字内容的排版布局。
@property (nonatomic, strong) NSLayoutManager *layoutManager;
//定义了一个矩形区域用于存放已经进行了排版并设置好属性的文字
@property (nonatomic, strong) NSTextContainer *textContainer;

需要重写系统方法，这里讲主要的

- (instancetype)initWithFrame:(CGRect)frame
{
    self = [super initWithFrame:frame];
    if (self) {
       //准备文本
        [self prepareTextSystem];
    }
    return self;
}

//设置容器大小为label的尺寸
- (void)layoutSubviews{
    self.textContainer.size = self.frame.size;
}

//重写drawTextInRect方法绘制背景，文字
- (void)drawTextInRect:(CGRect)rect{
    //如果有选中的range，改变文字背景颜色。
    if (_selectedRangeValue) {
        UIColor *selectColor = _isSelected ? [UIColor colorWithWhite:0.6 alpha:0.2] : [UIColor clearColor];
        //设置文字背景颜色
        [self.textStorage addAttribute:NSBackgroundColorAttributeName value:selectColor range:self.selectedRangeValue.rangeValue];
        //绘制背景颜色
        [self.layoutManager drawBackgroundForGlyphRange:self.selectedRangeValue.rangeValue atPoint:CGPointMake(0, 0)];
    }
    NSRange range = NSMakeRange(0, self.textStorage.length);
    [self.layoutManager drawGlyphsForGlyphRange:range atPoint:CGPointZero];
}

2、准备文本显示

这里面会用到正则表达式，正则表达式是一门挺深的学问。有不太了解的童鞋可以参考这篇文章

- (void)prepareTextSystem {

    //将布局添加到storage
    [self.textStorage addLayoutManager:self.layoutManager];
    //将容器添加到布局中
    [self.layoutManager addTextContainer:self.textContainer];
    self.userInteractionEnabled = YES;
    //设置左右边距
    self.textContainer.lineFragmentPadding = 10;
}
- (void)prepareText{
    NSAttributedString *attrString;
//拿到文本，或者富文本，拿不到置为空
    if (self.attributedText) {
        attrString = self.attributedText;
    }else if (self.text){
        attrString = [[NSAttributedString alloc]initWithString:self.text];
    }else {
        attrString = [[NSAttributedString alloc]initWithString:@""];
    }
    self.selectedRangeValue = nil;
    //设置折行模式
    NSMutableAttributedString *attrMString = [self addLineBreak:attrString];
    [self.textStorage setAttributedString:attrMString];
    
    //正则匹配，匹配我们需要的@XXX和URL
     self.linkRanges = [self getRanges:@"http(s)?://([\\w-]+\\.)+[\\w-]+(/[\\w- ./?%&=]*)?"];
    UIColor *attrColor = _attrStrColor == nil ? [UIColor blueColor] : _attrStrColor;
    for (NSValue *rangeValue in self.linkRanges) {
        [self.textStorage addAttribute:NSForegroundColorAttributeName value:attrColor range:rangeValue.rangeValue];
    }
    
    self.userRanges = [self getRanges:@"@\\w{1,}?:"];
    for (NSValue *rangeValue in self.userRanges) {
        [self.textStorage addAttribute:NSForegroundColorAttributeName value:attrColor range:rangeValue.rangeValue];
    }
    //匹配完成调用setNeedsDisplay，来触发drawTextInRect绘制文字
    [self setNeedsDisplay];
}
//添加折行模式
- (NSMutableAttributedString*)addLineBreak:(NSAttributedString *) attrString{
    NSMutableAttributedString *attrMString = [[NSMutableAttributedString alloc]initWithAttributedString:attrString];
    if (attrMString.length == 0) {
        return attrMString;
    }
    NSRange range = NSMakeRange(0, 0);
    NSMutableDictionary *dict = (NSMutableDictionary*)[attrMString attributesAtIndex:0 effectiveRange:&range];
    //里面有NSShadow,NSParagraphStyle,NSFont,NSColor四个值
    NSLog(@"%@",dict);
    
    NSMutableParagraphStyle *paragraphStyle = [dict[NSParagraphStyleAttributeName] mutableCopy] ;
    //设置paragraphStyle，如果不为空，则设置折行模式,为空则自己生成并add
    if (paragraphStyle) {
        paragraphStyle.lineBreakMode = NSLineBreakByCharWrapping;
    }else {
        paragraphStyle = [[NSMutableParagraphStyle alloc]init];
        paragraphStyle.lineBreakMode = NSLineBreakByCharWrapping;
    }
    [attrMString addAttribute:NSParagraphStyleAttributeName value:paragraphStyle range:range];
    return attrMString;
}

3、正则的匹配

- (NSMutableArray<NSValue *> *)getRanges:(NSString*)pattern{
    //生成匹配规则
    NSRegularExpression *regular = [[NSRegularExpression alloc]initWithPattern:pattern options:NSRegularExpressionCaseInsensitive error:nil];
//根据匹配规则匹配相应的文字
    NSArray *results = [regular matchesInString:self.textStorage.string options:NSMatchingReportCompletion range:NSMakeRange(0, self.textStorage.string.length)];
//由于OC的数组只能放对象，NSRange是不能放进去的，所以在此转换为NSValue来放入数组
    NSMutableArray *ranges = [NSMutableArray array];
    for (NSTextCheckingResult *result in results) {
        NSValue *value = [NSValue valueWithRange:result.range];
        [ranges addObject:value];
    }
    return ranges;
}

4、点击交互处理

- (void)touchesBegan:(NSSet<UITouch *> *)touches withEvent:(UIEvent *)event{
  //记录是否有选中
    _isSelected = YES;
//拿到touch的位置
    CGPoint selectPoint = [[touches anyObject] locationInView:self];
//获取touch的位置在那个range范围内，拿到selectedRange
    self.selectedRangeValue = [self getSelectRange:selectPoint];
    if (!_selectedRangeValue) {
        [super touchesBegan:touches withEvent:event];
    }
}
- (void)touchesEnded:(NSSet<UITouch *> *)touches withEvent:(UIEvent *)event{
    if (!_selectedRangeValue) {
        [super touchesEnded:touches withEvent:event];
        return;
    }
//是否有选中置为NO
    _isSelected = NO;
//更新label显示
    [self setNeedsDisplay];
//回调Block触发相应的事件
    NSString *contentText = [self.textStorage.string substringWithRange:_selectedRangeValue.rangeValue];
    switch (_tapHandleType) {
        case TapHandleUser:
            if (_userTaphandle) {
            _userTaphandle(self,contentText,_selectedRangeValue.rangeValue);
            }
            break;
            case TapHandlelink:
            if (_linkTapHandle) {
             _linkTapHandle(self,contentText,_selectedRangeValue.rangeValue);
            }
            break;
        default:
            break;
    }
    
}
//获取点击位置在哪个range范围内
-(NSValue*)getSelectRange:(CGPoint)selectPoint{
    if (self.textStorage.length == 0) {
        return nil;
    }
    //根据电机的位置得到点击的是label内的第几个字符索引
    NSInteger index = [self.layoutManager glyphIndexForPoint:selectPoint inTextContainer:self.textContainer];
//遍历所有range,确定index在哪个range范围内
    for (NSValue *rangeValue in self.linkRanges) {
        NSRange range = rangeValue.rangeValue;
        if (index >= range.location && index <range.location + range.length) {
//一旦遍历到就记录点击的范围类型，调用setNeedsDisplay触发文字label重新绘制
            _tapHandleType = TapHandlelink;
            [self setNeedsDisplay];
//返回range范围
            return rangeValue;
        }
    }
    for (NSValue *rangeValue in self.userRanges) {
        NSRange range = rangeValue.rangeValue;
        if (index >= range.location && index <range.location + range.length) {
            _tapHandleType = TapHandleUser;
            [self setNeedsDisplay];
            return rangeValue;
        }
    }
    _tapHandleType = TapHandleNone;
    return nil;
}

5、总结

1、这里面主要用到了三个重要的属性：NSTextStorage、NSLayoutManager、NSTextContainer要高明白其中的关系。上面有图介绍。
　　2、正则表达式的使用，现在基本上大部分语言都支持正则，里面的学问很多，需要多加学习。
　　3、点击事件的处理。
这是GitHub地址可以在此查看源码。本文为原创，如需转载请注明出处。