diff --git a/HTMLKit/HTMLInputStreamReader.h b/HTMLKit/HTMLInputStreamReader.h index 7823a03..f102ba4 100644 --- a/HTMLKit/HTMLInputStreamReader.h +++ b/HTMLKit/HTMLInputStreamReader.h @@ -31,6 +31,7 @@ - (BOOL)consumeHexInt:(unsigned int *)result; - (BOOL)consumeString:(NSString *)string caseSensitive:(BOOL)caseSensitive; - (NSString *)consumeCharactersUpToCharactersInString:(NSString *)characters; +- (NSString *)consumeCharactersUpToString:(NSString *)string; - (void)markCurrentLocation; - (void)rewindToMarkedLocation; diff --git a/HTMLKit/HTMLInputStreamReader.m b/HTMLKit/HTMLInputStreamReader.m index bbe6bd3..c91f029 100644 --- a/HTMLKit/HTMLInputStreamReader.m +++ b/HTMLKit/HTMLInputStreamReader.m @@ -162,6 +162,14 @@ return consumed.length > 0 ? consumed : nil; } +- (NSString *)consumeCharactersUpToString:(NSString *)string +{ + NSString *consumed = [NSString string]; + [_scanner scanUpToString:string intoString:&consumed]; + _location = _scanner.scanLocation; + return consumed; +} + - (void)unconsumeCurrentInputCharacter { _location -= _consume;