diff --git a/HTMLKit/CSSTokenizer.h b/HTMLKit/CSSTokenizer.h index 2913093..03a78f0 100644 --- a/HTMLKit/CSSTokenizer.h +++ b/HTMLKit/CSSTokenizer.h @@ -39,6 +39,7 @@ typedef NS_ENUM(int, CSSToken) - (instancetype)initWithString:(NSString *)string; - (CSSToken)nextToken; +- (CSSToken)nextNonSpaceToken; - (NSString *)currentTokenText; @end diff --git a/HTMLKit/CSSTokenizer.m b/HTMLKit/CSSTokenizer.m index edb87b6..ada62c6 100644 --- a/HTMLKit/CSSTokenizer.m +++ b/HTMLKit/CSSTokenizer.m @@ -42,6 +42,15 @@ return _currentToken; } +- (CSSToken)nextNonSpaceToken +{ + CSSToken token; + do { + token = self.nextToken; + } while (token == CSSTokenSpace); + return token; +} + - (NSString *)currentTokenText { if (_currentToken == 0) {