diff --git a/HTMLKit/HTMLToken.h b/HTMLKit/HTMLToken.h index 11bfba5..c943614 100644 --- a/HTMLKit/HTMLToken.h +++ b/HTMLKit/HTMLToken.h @@ -9,6 +9,7 @@ #import @class HTMLDOCTYPEToken; +@class HTMLTagToken; @class HTMLStartTagToken; @class HTMLEndTagToken; @class HTMLCommentToken; @@ -43,6 +44,7 @@ typedef NS_ENUM(NSUInteger, HTMLTokenType) - (BOOL)isParseError; - (HTMLDOCTYPEToken *)asDoctypeToken; +- (HTMLTagToken *)asTagToken; - (HTMLStartTagToken *)asStartTagToken; - (HTMLEndTagToken *)asEndTagToken; - (HTMLCommentToken *)asCommentToken; diff --git a/HTMLKit/HTMLToken.m b/HTMLKit/HTMLToken.m index 678c9bd..ec4f4d0 100644 --- a/HTMLKit/HTMLToken.m +++ b/HTMLKit/HTMLToken.m @@ -58,6 +58,11 @@ return (HTMLDOCTYPEToken *)self; } +- (HTMLTagToken *)asTagToken +{ + return (HTMLTagToken *)self; +} + - (HTMLStartTagToken *)asStartTagToken { return (HTMLStartTagToken *)self;