mirror of
https://github.com/zhigang1992/RestKit.git
synced 2026-04-02 09:31:32 +08:00
98 lines
3.0 KiB
Objective-C
98 lines
3.0 KiB
Objective-C
//
|
|
// RKSearchEngine.m
|
|
// Two Toasters
|
|
//
|
|
// Created by Blake Watters on 8/26/09.
|
|
// Copyright 2009 Two Toasters. All rights reserved.
|
|
//
|
|
|
|
#import "RKSearchEngine.h"
|
|
|
|
@implementation RKSearchEngine
|
|
|
|
@synthesize mode = _mode, tokenizeQuery = _tokenizeQuery, stripWhitespace = _stripWhitespace, caseSensitive = _caseSensitive;
|
|
|
|
+ (id)searchEngine {
|
|
return [[[RKSearchEngine alloc] init] autorelease];
|
|
}
|
|
|
|
- (id)init {
|
|
if (self = [super init]) {
|
|
_mode = RKSearchModeOr;
|
|
_tokenizeQuery = YES;
|
|
_stripWhitespace = YES;
|
|
_caseSensitive = NO;
|
|
}
|
|
|
|
return self;
|
|
}
|
|
|
|
#pragma mark Private
|
|
#pragma mark -
|
|
|
|
- (NSString*)stripWhitespaceIfNecessary:(NSString*)string {
|
|
if (_stripWhitespace) {
|
|
return [string stringByTrimmingCharactersInSet:[NSCharacterSet whitespaceCharacterSet]];
|
|
} else {
|
|
return string;
|
|
}
|
|
}
|
|
|
|
- (NSArray*)tokenizeOrCollect:(NSString*)string {
|
|
if (_tokenizeQuery) {
|
|
return [string componentsSeparatedByString:@" "];
|
|
} else {
|
|
return [NSArray arrayWithObject:string];
|
|
}
|
|
}
|
|
|
|
- (NSArray*)searchWithTerms:(NSArray*)searchTerms onProperties:(NSArray*)properties inCollection:(NSArray*)collection compoundSelector:(SEL)selector {
|
|
NSPredicate* searchPredicate = nil;
|
|
|
|
// do any of these properties contain all of these terms
|
|
NSMutableArray* propertyPredicates = [NSMutableArray array];
|
|
for (NSString* property in properties) {
|
|
NSMutableArray* termPredicates = [NSMutableArray array];
|
|
for (NSString* searchTerm in searchTerms) {
|
|
NSPredicate* predicate;
|
|
if (_caseSensitive) {
|
|
predicate = [NSPredicate predicateWithFormat:@"(%K contains %@)", property, searchTerm];
|
|
} else {
|
|
predicate = [NSPredicate predicateWithFormat:@"(%K contains[cd] %@)", property, searchTerm];
|
|
}
|
|
[termPredicates addObject:predicate];
|
|
}
|
|
|
|
// build an and predicate for all of the search terms
|
|
NSPredicate* termsPredicate = [NSCompoundPredicate performSelector:selector withObject:termPredicates];
|
|
[propertyPredicates addObject:termsPredicate];
|
|
}
|
|
|
|
searchPredicate = [NSCompoundPredicate orPredicateWithSubpredicates:propertyPredicates];
|
|
return [collection filteredArrayUsingPredicate:searchPredicate];
|
|
}
|
|
|
|
#pragma mark Public
|
|
#pragma mark -
|
|
|
|
- (NSArray*)searchFor:(NSString*)searchText inCollection:(NSArray*)collection {
|
|
NSArray* properties = [NSArray arrayWithObject:@"searchableText"];
|
|
return [self searchFor:searchText onProperties:properties inCollection:collection];
|
|
}
|
|
|
|
- (NSArray*)searchFor:(NSString*)searchText onProperties:(NSArray*)properties inCollection:(NSArray*)collection {
|
|
NSString* searchQuery = [[searchText copy] autorelease];
|
|
searchQuery = [self stripWhitespaceIfNecessary:searchQuery];
|
|
NSArray* searchTerms = [self tokenizeOrCollect:searchQuery];
|
|
|
|
if (_mode == RKSearchModeOr) {
|
|
return [self searchWithTerms:searchTerms onProperties:properties inCollection:collection compoundSelector:@selector(orPredicateWithSubpredicates:)];
|
|
} else if (_mode == RKSearchModeAnd) {
|
|
return [self searchWithTerms:searchTerms onProperties:properties inCollection:collection compoundSelector:@selector(andPredicateWithSubpredicates:)];
|
|
} else {
|
|
return nil;
|
|
}
|
|
}
|
|
|
|
@end
|