//
//  MIT License (MIT)
//
//  Copyright (c) 2018 Oleg Geier
//
//  Permission is hereby granted, free of charge, to any person obtaining a copy of
//  this software and associated documentation files (the "Software"), to deal in
//  the Software without restriction, including without limitation the rights to
//  use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies
//  of the Software, and to permit persons to whom the Software is furnished to do
//  so, subject to the following conditions:
//
//  The above copyright notice and this permission notice shall be included in all
//  copies or substantial portions of the Software.
//
//  THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
//  IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
//  FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
//  AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
//  LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
//  OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
//  SOFTWARE.

#import <libxml/xmlerror.h>

#import "RSXMLParser.h"
#import "RSXMLData.h"
#import "RSXMLError.h"

@interface RSXMLParser()
@property (nonatomic) RSSAXParser *parser;
@property (nonatomic) NSData *xmlData;
@property (nonatomic, copy) NSError *xmlInputError;
@end


@implementation RSXMLParser

+ (BOOL)isFeedParser { return NO; } // override
+ (BOOL)isOPMLParser { return NO; } // override
+ (BOOL)isHTMLParser { return NO; } // override
- (id)xmlParserWillReturnDocument { return nil; } // override

/**
 Designated initializer. Runs a check whether it matches the detected parser in @c RSXMLData.
 Keeps an internal pointer to the @c RSXMLData and initializes a new @c RSSAXParser.
 */
+ (instancetype)parserWithXMLData:(nonnull RSXMLData *)xmlData {
	if ([xmlData.parserClass isSubclassOfClass:[super class]]) {
		return [[xmlData.parserClass alloc] initWithXMLData:xmlData];
	}
	return [[super alloc] initWithXMLData:xmlData];
}

/**
 Internal initializer. Use the class initializer to automatically initialize to proper subclass.
 Keeps an internal pointer to the @c RSXMLData and initializes a new @c RSSAXParser.
 */
- (instancetype)initWithXMLData:(nonnull RSXMLData *)xmlData {
	self = [super init];
	if (self) {
		_documentURI = [xmlData.urlString copy];
		_xmlInputError = [xmlData.parserError copy];
		[self checkIfParserMatches:xmlData.parserClass];
		_xmlData = xmlData.data;
		if (!_xmlData) {
			_xmlInputError = RSXMLMakeError(RSXMLErrorNoData);
		}
		_parser = [[RSSAXParser alloc] initWithDelegate:self];
	}
	return self;
}

/**
 Parse the XML data on whatever thread this method is called.
 
 @param error Sets @c error if parser gets unrecognized data or libxml runs into a parsing error.
 @return The parsed object. The object type depends on the underlying data. @c RSParsedFeed, @c RSOPMLItem or @c RSHTMLMetadata.
 */
- (id _Nullable)parseSync:(NSError **)error {
	if (_xmlInputError) {
		if (error) *error = _xmlInputError;
		return nil;
	}
	
	if ([self respondsToSelector:@selector(xmlParserWillStartParsing)] && ![self xmlParserWillStartParsing])
		return nil;

	@autoreleasepool {
		xmlResetLastError();
		[_parser parseBytes:_xmlData.bytes numberOfBytes:_xmlData.length];
		if (error) {
			xmlErrorPtr err = xmlGetLastError();
			if (err && err->level == XML_ERR_FATAL) {
				int errCode = err->code;
				char * msg = err->message;
				NSString *errMsg = [[NSString stringWithFormat:@"%s", msg] stringByTrimmingCharactersInSet:[NSCharacterSet whitespaceAndNewlineCharacterSet]];
				*error = [NSError errorWithDomain:kLIBXMLParserErrorDomain code:errCode userInfo:@{NSLocalizedDescriptionKey: errMsg}];
			}
			xmlResetLastError();
		}
	}
	return [self xmlParserWillReturnDocument];
}

/**
 Dispatch new background thread, parse the data synchroniously on the background thread and exec callback on the main thread.
 */
- (void)parseAsync:(void(^)(id parsedDocument, NSError *error))block {
	dispatch_async(dispatch_get_global_queue(QOS_CLASS_UTILITY, 0), ^{ // QOS_CLASS_DEFAULT
		@autoreleasepool {
			NSError *error;
			id obj = [self parseSync:&error];
			dispatch_async(dispatch_get_main_queue(), ^{
				block(obj, error);
			});
		}
	});
}

/// @return @c YES if @c .xmlInputError is @c nil.
- (BOOL)canParse {
	return (self.xmlInputError == nil);
}


#pragma mark - Check Parser Type Matches


/**
 @return Returns either @c ExpectingFeed, @c ExpectingOPML, @c ExpectingHTML.
 @return @c RSXMLErrorNoData for an unexpected class (e.g., if @c RSXMLParser is used directly).
 */
- (RSXMLError)getExpectedErrorForClass:(Class<RSXMLParserDelegate>)cls {
	if ([cls isFeedParser])
		return RSXMLErrorExpectingFeed;
	if ([cls isOPMLParser])
		return RSXMLErrorExpectingOPML;
	if ([cls isHTMLParser])
		return RSXMLErrorExpectingHTML;
	return RSXMLErrorNoData; // will result in 'Unknown format'
}

/**
 Check whether parsing class matches the expected parsing class. If not set @c .xmlInputError along the way.
 
 @return @c YES if @c parserClass matches, @c NO otherwise. If @c NO is returned, @c parserError is set also.
 */
- (BOOL)checkIfParserMatches:(Class<RSXMLParserDelegate>)xmlParserClass {
	if (!xmlParserClass)
		return NO;
	if (xmlParserClass != [self class]) { // && !_xmlInputError
		RSXMLError current = [self getExpectedErrorForClass:xmlParserClass];
		RSXMLError expected = [self getExpectedErrorForClass:[self class]];
		if (current != expected) {
			_xmlInputError =  RSXMLMakeErrorWrongParser(expected, current);
			return NO;
		}
	}
	return YES; // only if no error was set (not now, nor before)
}

@end