From f79530c7f6ec9c6c248c8f4dd7ad742b7cda95b3 Mon Sep 17 00:00:00 2001 From: iska Date: Sun, 1 Mar 2015 03:27:50 +0100 Subject: [PATCH] Add convenience method to cast a Token to a Tag Token --- HTMLKit/HTMLToken.h | 2 ++ HTMLKit/HTMLToken.m | 5 +++++ 2 files changed, 7 insertions(+) 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;