main.m
#import "NSString+isImage.h"
#import "NSString+isURL.h"
#import <Foundation/Foundation.h>
void image(NSString * image)
{
if ([image isImage]) {
NSLog(@"是图片");
}
else {
NSLog(@"不是图片");
}
}
void url(NSString * url)
{
if ([url isURL]) {
NSLog(@"是网址");
}
else {
NSLog(@"不是网址");
}
}
int main(int argc, const char * argv[])
{
@autoreleasepool {
NSString * url1 = @"http://www.baidu.com";
NSString * url2 = @"https://www.taobao.com";
url(url1);
url(url2);
NSString * image1 = @"bizhi.jpg";
NSString * image2 = @"bizhi.png";
NSString * image3 = @"bizhi.jpeg";
NSString * image4 = @"bizhi.gif";
image(image1);
image(image2);
image(image3);
image(image4);
}
return 0;
}
NSString+isURL.h
#import <Foundation/Foundation.h>
@interface NSString (isURL)
-(BOOL)isURL;
@end
NSString+isURL.m
#import "NSString+isURL.h"
@implementation NSString (isURL)
- (BOOL)isURL
{
return ([self hasPrefix:@"http://"] || [self hasPrefix:@"https://"]);
}
@end
NSString+isImage.h
#import <Foundation/Foundation.h>
@interface NSString (isImage)
- (BOOL)isImage;
@end
NSString+isImage.m
#import "NSString+isImage.h"
@implementation NSString (isImage)
- (BOOL)isImage
{
return [self hasSuffix:@".jpg"] || [self hasSuffix:@".png"] || [self hasSuffix:@".jpeg"] || [self hasSuffix:@".gif"];
}
@end