NetNewsWire/Frameworks/RSParser/HTML/RSHTMLMetadata.m

//
//  RSHTMLMetadata.m
//  RSParser
//
//  Created by Brent Simmons on 3/6/16.
//  Copyright © 2016 Ranchero Software, LLC. All rights reserved.
//

#import <RSParser/RSHTMLMetadata.h>
#import <RSParser/RSParserInternal.h>
#import <RSParser/RSHTMLTag.h>

static NSString *urlStringFromDictionary(NSDictionary *d);
static NSString *absoluteURLStringWithRelativeURLString(NSString *relativeURLString, NSString *baseURLString);
static NSString *absoluteURLStringWithDictionary(NSDictionary *d, NSString *baseURLString);
static NSArray *objectsOfClassWithTags(Class class, NSArray *tags, NSString *baseURLString);
static NSString *relValue(NSDictionary *d);
static BOOL typeIsFeedType(NSString *type);

static NSString *kShortcutIconRelValue = @"shortcut icon";
static NSString *kIconRelValue = @"icon";
static NSString *kHrefKey = @"href";
static NSString *kSrcKey = @"src";
static NSString *kAppleTouchIconValue = @"apple-touch-icon";
static NSString *kAppleTouchIconPrecomposedValue = @"apple-touch-icon-precomposed";
static NSString *kSizesKey = @"sizes";
static NSString *kTitleKey = @"title";
static NSString *kRelKey = @"rel";
static NSString *kAlternateKey = @"alternate";
static NSString *kRSSSuffix = @"/rss+xml";
static NSString *kAtomSuffix = @"/atom+xml";
static NSString *kJSONSuffix = @"/json";
static NSString *kTypeKey = @"type";

@interface RSHTMLMetadataAppleTouchIcon ()

- (instancetype)initWithTag:(RSHTMLTag *)tag baseURLString:(NSString *)baseURLString;

@end


@interface RSHTMLMetadataFeedLink ()

- (instancetype)initWithTag:(RSHTMLTag *)tag baseURLString:(NSString *)baseURLString;

@end


@implementation RSHTMLMetadata

#pragma mark - Init

- (instancetype)initWithURLString:(NSString *)urlString tags:(NSArray <RSHTMLTag *> *)tags {

	self = [super init];
	if (!self) {
		return nil;
	}

	_baseURLString = urlString;
	_tags = tags;
	_faviconLink = [self resolvedLinkFromFirstLinkTagWithMatchingRel:kShortcutIconRelValue];
	if (_faviconLink == nil) {
		_faviconLink = [self resolvedLinkFromFirstLinkTagWithMatchingRel:kIconRelValue];
	}
	
	NSArray *appleTouchIconTags = [self appleTouchIconTags];
	_appleTouchIcons = objectsOfClassWithTags([RSHTMLMetadataAppleTouchIcon class], appleTouchIconTags, urlString);

	NSArray *feedLinkTags = [self feedLinkTags];
	_feedLinks = objectsOfClassWithTags([RSHTMLMetadataFeedLink class], feedLinkTags, urlString);

	_openGraphProperties = [[RSHTMLOpenGraphProperties alloc] initWithURLString:urlString tags:tags];

	return self;
}


#pragma mark - Private

- (RSHTMLTag *)firstLinkTagWithMatchingRel:(NSString *)valueToMatch {

	// Case-insensitive.

	for (RSHTMLTag *tag in self.tags) {

		if (tag.type != RSHTMLTagTypeLink) {
			continue;
		}
		NSString *oneRelValue = relValue(tag.attributes);
		if (oneRelValue && [oneRelValue compare:valueToMatch options:NSCaseInsensitiveSearch] == NSOrderedSame) {
			return tag;
		}
	}

	return nil;
}


- (NSArray *)appleTouchIconTags {

	NSMutableArray *tags = [NSMutableArray new];

	for (RSHTMLTag *tag in self.tags) {

		if (tag.type != RSHTMLTagTypeLink) {
			continue;
		}
		NSString *oneRelValue = relValue(tag.attributes).lowercaseString;
		if ([oneRelValue isEqualToString:kAppleTouchIconValue] || [oneRelValue isEqualToString:kAppleTouchIconPrecomposedValue]) {
			[tags addObject:tag];
		}
	}

	return tags;
}


- (NSArray *)feedLinkTags {

	NSMutableArray *tags = [NSMutableArray new];

	for (RSHTMLTag *tag in self.tags) {

		if (tag.type != RSHTMLTagTypeLink) {
			continue;
		}

		NSDictionary *oneDictionary = tag.attributes;
		NSString *oneRelValue = relValue(oneDictionary).lowercaseString;
		if (![oneRelValue isEqualToString:kAlternateKey]) {
			continue;
		}

		NSString *oneType = [oneDictionary rsparser_objectForCaseInsensitiveKey:kTypeKey];
		if (!typeIsFeedType(oneType)) {
			continue;
		}

		if (RSParserStringIsEmpty(urlStringFromDictionary(oneDictionary))) {
			continue;
		}

		[tags addObject:tag];
	}

	return tags;
}


- (NSString *)resolvedLinkFromFirstLinkTagWithMatchingRel:(NSString *)relValue {

	RSHTMLTag *tag = [self firstLinkTagWithMatchingRel:relValue];
	return absoluteURLStringWithDictionary(tag.attributes, self.baseURLString);
}

@end


static NSString *relValue(NSDictionary *d) {

	return [d rsparser_objectForCaseInsensitiveKey:kRelKey];
}


static NSString *urlStringFromDictionary(NSDictionary *d) {

	NSString *urlString = [d rsparser_objectForCaseInsensitiveKey:kHrefKey];
	if (urlString) {
		return urlString;
	}

	return [d rsparser_objectForCaseInsensitiveKey:kSrcKey];
}


static NSString *absoluteURLStringWithRelativeURLString(NSString *relativeURLString, NSString *baseURLString) {

	NSURL *url = [NSURL URLWithString:baseURLString];
	if (!url) {
		return nil;
	}

	NSURL *absoluteURL = [NSURL URLWithString:relativeURLString relativeToURL:url];
	return absoluteURL.absoluteString;
}


static NSString *absoluteURLStringWithDictionary(NSDictionary *d, NSString *baseURLString) {

	NSString *urlString = urlStringFromDictionary(d);
	if (RSParserStringIsEmpty(urlString)) {
		return nil;
	}
	return absoluteURLStringWithRelativeURLString(urlString, baseURLString);
}


static NSArray *objectsOfClassWithTags(Class class, NSArray *tags, NSString *baseURLString) {

	NSMutableArray *objects = [NSMutableArray new];

	for (RSHTMLTag *tag in tags) {

		id oneObject = [[class alloc] initWithTag:tag baseURLString:baseURLString];
		if (oneObject) {
			[objects addObject:oneObject];
		}
	}

	return objects;
}


static BOOL typeIsFeedType(NSString *type) {

	type = type.lowercaseString;
	return [type hasSuffix:kRSSSuffix] || [type hasSuffix:kAtomSuffix] || [type hasSuffix:kJSONSuffix];
}


@implementation RSHTMLMetadataAppleTouchIcon

- (instancetype)initWithTag:(RSHTMLTag *)tag baseURLString:(NSString *)baseURLString {

	self = [super init];
	if (!self) {
		return nil;
	}

	NSDictionary *d = tag.attributes;
	_urlString = absoluteURLStringWithDictionary(d, baseURLString);
	_sizes = [d rsparser_objectForCaseInsensitiveKey:kSizesKey];
	_rel = [d rsparser_objectForCaseInsensitiveKey:kRelKey];

	return self;
}

@end


@implementation RSHTMLMetadataFeedLink

- (instancetype)initWithTag:(RSHTMLTag *)tag baseURLString:(NSString *)baseURLString {

	self = [super init];
	if (!self) {
		return nil;
	}

	NSDictionary *d = tag.attributes;
	_urlString = absoluteURLStringWithDictionary(d, baseURLString);
	_title = [d rsparser_objectForCaseInsensitiveKey:kTitleKey];
	_type = [d rsparser_objectForCaseInsensitiveKey:kTypeKey];

	return self;
}

@end

@interface RSHTMLOpenGraphImage ()

@property (nonatomic, readwrite) NSString *url;
@property (nonatomic, readwrite) NSString *secureURL;
@property (nonatomic, readwrite) NSString *mimeType;
@property (nonatomic, readwrite) CGFloat width;
@property (nonatomic, readwrite) CGFloat height;
@property (nonatomic, readwrite) NSString *altText;

@end

@implementation RSHTMLOpenGraphImage


@end

@interface RSHTMLOpenGraphProperties ()

@property (nonatomic) NSMutableArray *ogImages;
@end

@implementation RSHTMLOpenGraphProperties

- (instancetype)initWithURLString:(NSString *)urlString tags:(NSArray <RSHTMLTag *> *)tags {

	self = [super init];
	if (!self) {
		return nil;
	}

	_ogImages = [NSMutableArray new];

	[self parseTags:tags];
	return self;
}


- (RSHTMLOpenGraphImage *)currentImage {

	return self.ogImages.lastObject;
}


- (RSHTMLOpenGraphImage *)pushImage {

	RSHTMLOpenGraphImage *image = [RSHTMLOpenGraphImage new];
	[self.ogImages addObject:image];
	return image;
}

- (RSHTMLOpenGraphImage *)ensureImage {

	RSHTMLOpenGraphImage *image = [self currentImage];
	if (image != nil) {
		return image;
	}
	return [self pushImage];
}


- (NSArray *)images {

	return self.ogImages;
}

static NSString *ogPrefix = @"og:";
static NSString *ogImage = @"og:image";
static NSString *ogImageURL = @"og:image:url";
static NSString *ogImageSecureURL = @"og:image:secure_url";
static NSString *ogImageType = @"og:image:type";
static NSString *ogImageWidth = @"og:image:width";
static NSString *ogImageHeight = @"og:image:height";
static NSString *ogImageAlt = @"og:image:alt";
static NSString *ogPropertyKey = @"property";
static NSString *ogContentKey = @"content";

- (void)parseTags:(NSArray *)tags {

	for (RSHTMLTag *tag in tags) {

		if (tag.type != RSHTMLTagTypeMeta) {
			continue;
		}

		NSString *propertyName = tag.attributes[ogPropertyKey];
		if (!propertyName || ![propertyName hasPrefix:ogPrefix]) {
			continue;
		}
		NSString *content = tag.attributes[ogContentKey];
		if (!content) {
			continue;
		}

		if ([propertyName isEqualToString:ogImage]) {
			RSHTMLOpenGraphImage *image = [self currentImage];
			if (!image || image.url) { // Most likely case, since og:image will probably appear before other image attributes.
				image = [self pushImage];
			}
			image.url = content;
		}

		else if ([propertyName isEqualToString:ogImageURL]) {
			[self ensureImage].url = content;
		}
		else if ([propertyName isEqualToString:ogImageSecureURL]) {
			[self ensureImage].secureURL = content;
		}
		else if ([propertyName isEqualToString:ogImageType]) {
			[self ensureImage].mimeType = content;
		}
		else if ([propertyName isEqualToString:ogImageAlt]) {
			[self ensureImage].altText = content;
		}
		else if ([propertyName isEqualToString:ogImageWidth]) {
			[self ensureImage].width = [content floatValue];
		}
		else if ([propertyName isEqualToString:ogImageHeight]) {
			[self ensureImage].height = [content floatValue];
		}
	}
}

@end