iOS本地代码url生成短链接

开发中如果要把一个url地址转换成短链接,基本上的操作是搜索第三方在线转换api,通过调用api,生成短链接返回,然后拿来使用。
比如微博的http://t.cn/xxxxxx格式
比如https://www.jianshu.com/u/bc4a806f89c5
通过网上的某个api转换成短链接就是http://b0z.cn/0009Fb。
但是这种做法还是有不舒服的地方,毕竟还得先请求第三方api接口,达不到实时转换,所有就想到了本地代码实现相似格式的短链接。
搜了下别人的代码实现,基本上都是MD5加密转换,然后通用一些字符串映射转换来实现。
下面结合他人跟自己项目需求的情况,实现了一个短链接转换的实现。
通过新建NSString分类来实现:

@interface NSString (ABShortLink)

/// url转换成短链接
- (NSString *)shortLink;

/// 动态列表内容正文里面的url转换成短链接显示
- (NSString *)matchFeedContentUrl;

@end
#import "NSString+ABShortLink.h"
#import 
#import "ABShortLinkCache.h"

@implementation NSString (ABShortLink)

- (NSString *)shortLink
{
    if (self.length <= 0) {
        return @"";
    }
    
    NSArray *chars = @[@"a", @"b", @"c", @"d", @"e", @"f", @"g", @"h", @"i", @"j", @"k", @"l", @"m", @"n", @"o", @"p", @"q", @"r", @"s", @"t", @"u", @"v", @"w", @"x", @"y", @"z", @"0", @"1", @"2", @"3", @"4", @"5", @"6", @"7", @"8", @"9", @"A", @"B", @"C", @"D", @"E", @"F", @"G", @"H", @"I", @"J", @"K", @"L", @"M", @"N", @"O", @"P", @"Q", @"R", @"S", @"T", @"U", @"V", @"W", @"X", @"Y", @"Z"];
//key也是自定义
    NSString *key = @"xxxxxx";
    
    const char *original_str = [self UTF8String];
    unsigned char result[CC_MD5_DIGEST_LENGTH];
    CC_MD5(original_str, (CC_LONG)strlen(original_str), result);
    NSMutableString *hash = [NSMutableString string];
    for (int i = 0; i < 16; i ++) {
        [hash appendFormat:@"%02X", result[i]];
    }
    NSString *urlMd5 = hash.lowercaseString;
    
    NSString *hex = [NSString stringWithFormat:@"%@", [key stringByAppendingFormat:@"%@",urlMd5]];
    
    NSMutableArray *resultUrl = [NSMutableArray arrayWithCapacity:4];
    for (int i = 0; i < 4; i ++) {
        // 把加密字符按照 8 位一组 16 进制与 0x3FFFFFFF 进行位与运算
        NSString *sTempSubString = [hex substringWithRange:NSMakeRange(i * 8, 8)];
        // 这里需要使用 long 型来转换,因为 Inteper只能处理 31 位 , 首位为符号位 , 如果不用 long ,则会越界
        long longOfTemp;
        sscanf([sTempSubString cStringUsingEncoding:NSASCIIStringEncoding], "%lx", &longOfTemp);
        long IHexLong = 0x3FFFFFF & longOfTemp;
        NSString *outChars = @"";
        for (int j = 0; j < 6; j ++) {
            // 把得到的值与 0x0000003D 进行位与运算,取得字符数组 chars 索引
            long index = 0x0000003D & IHexLong;
            //把取得的字符相加
            outChars = [outChars stringByAppendingFormat:@"%@",[chars objectAtIndex:(int)index]];
            //每次循环按位右移5位
            IHexLong = IHexLong >> 5;
        }
        //把字符串存入对应索引的输出数组
        [resultUrl insertObject:outChars atIndex:i];
    }
    NSString *output = [resultUrl objectAtIndex:1];//这里可以返回任意一个元素作为短链接(0,1,2,3)
//此处"http://w.cn/%@"为d自定义的格式,模仿微博
    NSString *output_url = [NSString stringWithFormat:@"http://w.cn/%@",output];
//    DLog(@"原链接url == %@", self);
//    DLog(@"短链接url == %@", output_url);
    
    [[ABShortLinkCache singleton] cacheShortLink:self key:output_url];
    
    return output_url;
}

- (NSString *)matchFeedContentUrl
{
    if (self.length <= 0) {
        return @"";
    }
    
//    NSString *allString = @"[狗头]测试一段网址正则匹配https://github.com/luffySaw我还是一段中文 是连上的。也是www.baidu.com的地址拉,但是我不[奸笑]相信这个傻屌竟然是http://www.sina.cn,而且竟然还是www.apple.cn 哈哈哈哈哈哈";
    NSError *error;
    NSString *regulaStr = @"((http[s]{0,1}|ftp)://[a-zA-Z0-9\\.\\-]+\\.([a-zA-Z]{2,6})(:\\d+)?(/[a-zA-Z0-9\\.\\-~!@#$%^&*+?:_/=<>]*)?)|(www.[a-zA-Z0-9\\.\\-]+\\.([a-zA-Z]{2,6})(:\\d+)?(/[a-zA-Z0-9\\.\\-~!@#$%^&*+?:_/=<>]*)?)|(((http[s]{0,1}|ftp)://|)((?:(?:25[0-5]|2[0-4]\\d|((1\\d{2})|([1-9]?\\d)))\\.){3}(?:25[0-5]|2[0-4]\\d|((1\\d{2})|([1-9]?\\d))))(:\\d+)?(/[a-zA-Z0-9\\.\\-~!@#$%^&*+?:_/=<>]*)?)";
    NSRegularExpression *urlRegex = [NSRegularExpression regularExpressionWithPattern:regulaStr
                                                                               options:NSRegularExpressionCaseInsensitive
                                                                                 error:&error];
    NSArray *matchArray = [urlRegex matchesInString:self options:0 range:NSMakeRange(0, self.length)];
    
    NSAttributedString *attString = [[NSAttributedString alloc]initWithString:self];
    NSMutableAttributedString *resultAttString = [[NSMutableAttributedString alloc] init];
    NSUInteger location = 0;
    for (NSTextCheckingResult *match in matchArray) {
        NSRange range = match.range;
        NSAttributedString *subAttStr = [attString attributedSubstringFromRange:NSMakeRange(location, range.location - location)];
        //先把非url部分加上去
        [resultAttString appendAttributedString:subAttStr];
        //下次循环从url的下一个位置开始
        location = NSMaxRange(range);
        
        NSAttributedString *urlAttStr = [attString attributedSubstringFromRange:range];
        if (urlAttStr.length) {
            //找到url部分,就处理成短链接拼接上去
            NSAttributedString *temp = [[NSAttributedString alloc] initWithString:urlAttStr.string.shortLink];
            [resultAttString appendAttributedString:temp];
        }
        else {
            //找不到url就直接拼接上去
            [resultAttString appendAttributedString:urlAttStr];
        }
    }
    if (location < attString.length) {
        //到这里说明最后面还设有非url部分
        NSRange range = NSMakeRange(location, attString.length - location);
        NSAttributedString *subAttStr = [attString attributedSubstringFromRange:range];
        [resultAttString appendAttributedString:subAttStr];
    }
//    DLog(@"最后 == %@",resultAttString);
    return resultAttString.string;
}

@end

你可能感兴趣的:(iOS本地代码url生成短链接)