From 9e198c976600b88f8d73e25333ba2effec5b823f Mon Sep 17 00:00:00 2001
From: iska <iska.gp@gmail.com>
Date: Fri, 24 Oct 2014 01:06:21 +0200
Subject: [PATCH] Add init method to Tokenizer header and extend to accept
 string parameter

---
 HTMLKit/HTMLTokenizer.h | 2 ++
 HTMLKit/HTMLTokenizer.m | 4 +++-
 2 files changed, 5 insertions(+), 1 deletion(-)

diff --git a/HTMLKit/HTMLTokenizer.h b/HTMLKit/HTMLTokenizer.h
index 30726a0..faeb866 100644
--- a/HTMLKit/HTMLTokenizer.h
+++ b/HTMLKit/HTMLTokenizer.h
@@ -15,4 +15,6 @@
  */
 @interface HTMLTokenizer : NSObject <NSFastEnumeration>
 
+- (instancetype)initWithString:(NSString *)string;
+
 @end
diff --git a/HTMLKit/HTMLTokenizer.m b/HTMLKit/HTMLTokenizer.m
index b6ce87b..66eb95d 100644
--- a/HTMLKit/HTMLTokenizer.m
+++ b/HTMLKit/HTMLTokenizer.m
@@ -53,7 +53,7 @@
 
 #pragma mark - Lifecycle
 
-- (instancetype)init
+- (instancetype)initWithString:(NSString *)string
 {
 	self = [super init];
 	if (self) {
@@ -62,6 +62,8 @@
 		[self setupStateMachine];
 
 		_tokens = [NSMutableArray new];
+
+		_inputStreamReader = [[HTMLInputStreamReader alloc] initWithString:string];
 	}
 	return self;
 }