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 +- (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; }