Add convenience method to cast a Token to a Tag Token

2015-03-01 03:27:50 +01:00
parent 6217bc4299
commit f79530c7f6
2 changed files with 7 additions and 0 deletions
@@ -9,6 +9,7 @@
 #import <Foundation/Foundation.h>

@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;
@@ -58,6 +58,11 @@
 	return (HTMLDOCTYPEToken *)self;
 }

+- (HTMLTagToken *)asTagToken
+{
+	return (HTMLTagToken *)self;
+}
+
 - (HTMLStartTagToken *)asStartTagToken
 {
 	return (HTMLStartTagToken *)self;