Initial import.

RSXML/RSHTMLLinkParser.m

@@ -0,0 +1,151 @@
+//
+//  RSHTMLLinkParser.m
+//  RSXML
+//
+//  Created by Brent Simmons on 8/7/16.
+//  Copyright © 2016 Ranchero Software, LLC. All rights reserved.
+//
+
+#import <libxml/xmlstring.h>
+#import "RSHTMLLinkParser.h"
+#import "RSSAXHTMLParser.h"
+#import "RSSAXParser.h"
+#import "RSXMLData.h"
+#import "RSXMLInternal.h"
+
+
+@interface RSHTMLLinkParser() <RSSAXHTMLParserDelegate>
+
+@property (nonatomic, readonly) NSMutableArray *links;
+@property (nonatomic, readonly) RSXMLData *xmlData;
+@property (nonatomic, readonly) NSMutableArray *dictionaries;
+@property (nonatomic, readonly) NSURL *baseURL;
+
+@end
+
+
+@interface RSHTMLLink()
+
+@property (nonatomic, readwrite) NSString *urlString; //absolute
+@property (nonatomic, readwrite) NSString *text;
+@property (nonatomic, readwrite) NSString *title; //title attribute inside anchor tag
+
+@end
+
+
+@implementation RSHTMLLinkParser
+
+
+#pragma mark - Class Methods
+
++ (NSArray *)htmlLinksWithData:(RSXMLData *)xmlData {
+
+	RSHTMLLinkParser *parser = [[self alloc] initWithXMLData:xmlData];
+	return parser.links;
+}
+
+
+#pragma mark - Init
+
+- (instancetype)initWithXMLData:(RSXMLData *)xmlData {
+
+	NSParameterAssert(xmlData.data);
+	NSParameterAssert(xmlData.urlString);
+
+	self = [super init];
+	if (!self) {
+		return nil;
+	}
+
+	_links = [NSMutableArray new];
+	_xmlData = xmlData;
+	_dictionaries = [NSMutableArray new];
+	_baseURL = [NSURL URLWithString:xmlData.urlString];
+
+	[self parse];
+
+	return self;
+}
+
+
+#pragma mark - Parse
+
+- (void)parse {
+
+	RSSAXHTMLParser *parser = [[RSSAXHTMLParser alloc] initWithDelegate:self];
+	[parser parseData:self.xmlData.data];
+	[parser finishParsing];
+}
+
+
+- (RSHTMLLink *)currentLink {
+
+	return self.links.lastObject;
+}
+
+
+static NSString *kHrefKey = @"href";
+
+- (NSString *)urlStringFromDictionary:(NSDictionary *)d {
+
+	NSString *href = [d rsxml_objectForCaseInsensitiveKey:kHrefKey];
+	if (!href) {
+		return nil;
+	}
+
+	NSURL *absoluteURL = [NSURL URLWithString:href relativeToURL:self.baseURL];
+	return absoluteURL.absoluteString;
+}
+
+
+static NSString *kTitleKey = @"title";
+
+- (NSString *)titleFromDictionary:(NSDictionary *)d {
+
+	return [d rsxml_objectForCaseInsensitiveKey:kTitleKey];
+}
+
+
+- (void)handleLinkAttributes:(NSDictionary *)d {
+
+	RSHTMLLink *link = self.currentLink;
+	link.urlString = [self urlStringFromDictionary:d];
+	link.title = [self titleFromDictionary:d];
+}
+
+
+static const char *kAnchor = "a";
+static const NSInteger kAnchorLength = 2;
+
+- (void)saxParser:(RSSAXHTMLParser *)SAXParser XMLStartElement:(const xmlChar *)localName attributes:(const xmlChar **)attributes {
+
+	if (!RSSAXEqualTags(localName, kAnchor, kAnchorLength)) {
+		return;
+	}
+
+	RSHTMLLink *link = [RSHTMLLink new];
+	[self.links addObject:link];
+
+	NSDictionary *d = [SAXParser attributesDictionary:attributes];
+	if (!RSXMLIsEmpty(d)) {
+		[self handleLinkAttributes:d];
+	}
+
+	[SAXParser beginStoringCharacters];
+}
+
+
+- (void)saxParser:(RSSAXParser *)SAXParser XMLEndElement:(const xmlChar *)localName {
+
+	if (!RSSAXEqualTags(localName, kAnchor, kAnchorLength)) {
+		return;
+	}
+
+	self.currentLink.text = SAXParser.currentStringWithTrimmedWhitespace;
+}
+
+@end
+
+@implementation RSHTMLLink
+
+@end