#import "Slide.h"
#import "assets/TFHelper.h"
#import "assets/MMMarkdown/MMMarkdown.h"

static BOOL isSlideEnabled;
static BOOL isTFDeletedOnly;
static CGFloat pushshiftRequestTimeoutValue;

%group Slide

@implementation FontGenerator

+(UIFont *) fontOfSize:(CGFloat) size submission:(BOOL) isSubmission willOffset:(BOOL) willOffset{
	NSUserDefaults *userDefaults = [NSUserDefaults standardUserDefaults];
	
	NSString *fontName;
	CGFloat fontSize = size;
	
	if (willOffset){
		if (isSubmission){
			fontSize += ([userDefaults objectForKey:@"POST_FONT_SIZE"] == nil) ? 0 : [userDefaults integerForKey:@"POST_FONT_SIZE"];
		} else {
			fontSize += ([userDefaults objectForKey:@"COMMENT_FONT_SIZE"] == nil) ? -2 : [userDefaults integerForKey:@"COMMENT_FONT_SIZE"];
		}	
	}
	
	if ([userDefaults stringForKey:(isSubmission ? @"postfont" : @"commentfont")] == nil){
		fontName =  isSubmission ? @"AvenirNext-DemiBold" : @"AvenirNext-Medium";
	} else {
		fontName = [userDefaults stringForKey:(isSubmission ? @"postfont" : @"commentfont")];
	}
	
	UIFont *font = [UIFont fontWithName:fontName size:fontSize]; 
	
	if (!font){
		font = [UIFont systemFontOfSize:fontSize];
	}
	
	return font;
}
 
+(UIFont *) boldFontOfSize:(CGFloat) size submission:(BOOL) isSubmission willOffset:(BOOL) willOffset {
	UIFont *font = [self fontOfSize:size submission:isSubmission willOffset:willOffset];
	
	if ([font.fontName isEqualToString:[UIFont systemFontOfSize:10].fontName]){
		return [UIFont boldSystemFontOfSize:font.pointSize];
	} else {
		UIFontDescriptor *desc = [font.fontDescriptor fontDescriptorWithSymbolicTraits:UIFontDescriptorTraitBold];
		
		if (desc == nil){
			return font;
		} else {
			return [UIFont fontWithDescriptor:desc size: 0];
		}
	}
}

+(UIFont *) italicFontOfSize:(CGFloat) size submission:(BOOL) isSubmission willOffset:(BOOL) willOffset {
	UIFont *font = [self fontOfSize:size submission:isSubmission willOffset:willOffset];
	
	if ([font.fontName isEqualToString:[UIFont systemFontOfSize:10].fontName]){
		return [UIFont italicSystemFontOfSize:font.pointSize];
	} else {
		UIFontDescriptor *desc = [font.fontDescriptor fontDescriptorWithSymbolicTraits:UIFontDescriptorTraitItalic];
		
		if (desc == nil){
			return font;
		} else {
			return [UIFont fontWithDescriptor:desc size: 0];
		}
	}
}

@end


@implementation ColorUtil 

+(UIColor *) accentColorForSub:(NSString *) subreddit{
	NSUserDefaults *userDefaults = [NSUserDefaults standardUserDefaults];
	NSData *colorData = [userDefaults dataForKey:[NSString stringWithFormat:@"accent+%@", subreddit]];
	UIColor *color = [NSKeyedUnarchiver unarchiveObjectWithData:colorData];
	if (color) {
		return color;
	} else {
		UIColor *baseAccentColor = [NSKeyedUnarchiver unarchiveObjectWithData:[userDefaults dataForKey:@"accentcolor"]];
		
		if (baseAccentColor){
			return baseAccentColor;
		} else {
			return [UIColor colorWithRed:0.161 green:0.475 blue:1.0 alpha:1.0];
		}
	}
}

+(UIColor *) fontColorForTheme:(NSString *)theme{
	UIColor *fontColor;
	NSUserDefaults *userDefaults = [NSUserDefaults standardUserDefaults];
	
	//if only switch blocks worked with strings...
	if ([theme isEqualToString:@"light"]) {
		fontColor = [UIColor colorWithRed:0 green:0 blue:0 alpha:0.87];
	} else if ([theme isEqualToString:@"dark"]) {
		fontColor = [UIColor colorWithRed:1.0 green:1.0 blue:1.0 alpha:0.87];
	} else if ([theme isEqualToString:@"black"]) {
		fontColor = [UIColor colorWithRed:1.0 green:1.0 blue:1.0 alpha:0.87];
	} else if ([theme isEqualToString:@"blue"]) {
		fontColor = [UIColor colorWithRed:1.0 green:1.0 blue:1.0 alpha:0.87];
	} else if ([theme isEqualToString:@"sepia"]) {
		fontColor = [UIColor colorWithRed:0.243 green:.239 blue:.212 alpha:0.87]; 
	} else if ([theme isEqualToString:@"red"]) {
		fontColor = [UIColor colorWithRed:1.0 green:0.969 blue:0.929 alpha:0.87]; 
	} else if ([theme isEqualToString:@"deep"]) {
		fontColor = [UIColor colorWithRed:1.0 green:1.0 blue:1.0 alpha:0.87];
	} else if ([theme isEqualToString:@"mint"]) {
		fontColor = [UIColor colorWithRed:0.035 green:0.212 blue:0.059 alpha:0.87];
	} else if ([theme isEqualToString:@"cream"]) {
		fontColor = [UIColor colorWithRed:0.267 green:0.255 blue:0.224 alpha:0.87];
	} else if ([theme isEqualToString:@"acontrast"]) {
		fontColor = [UIColor colorWithRed:1.0 green:1.0 blue:1.0 alpha:0.87];
	} else if ([theme isEqualToString:@"pink"]) {
		fontColor = [UIColor colorWithRed:0.149 green:0.157 blue:0.267 alpha:0.87];
	} else if ([theme isEqualToString:@"solarize"]) {
		fontColor = [UIColor colorWithRed:0.514 green:0.580 blue:0.588 alpha:0.87];
	} else if (!theme) {
		fontColor = [UIColor colorWithRed:0 green:0 blue:0 alpha:0.87];
	} else {
		NSString *customThemeString = [[userDefaults stringForKey:[NSString stringWithFormat:@"Theme+%@", theme]] stringByRemovingPercentEncoding];
		if (customThemeString) {
			NSString *customFontColorHex = [customThemeString componentsSeparatedByString:@"#"][4];
			fontColor =  [UIColor colorWithHex:customFontColorHex]; 
		} else {
			fontColor = [UIColor colorWithRed:0 green:0 blue:0 alpha:0.87];
		}
	}
	
	return fontColor;
}

+(UIColor *) backgroundColorForTheme:(NSString *) theme{
	UIColor *backgroundColor;
	NSUserDefaults *userDefaults = [NSUserDefaults standardUserDefaults];

	if ([theme isEqualToString:@"light"]) {
		backgroundColor = [UIColor colorWithRed:0.352 green:0.352 blue:0.352 alpha:1.0];
	} else if ([theme isEqualToString:@"dark"]) {
		backgroundColor = [UIColor colorWithRed:0.051 green:0.051 blue:0.051 alpha:1.0];
	} else if ([theme isEqualToString:@"black"]) {
		backgroundColor = [UIColor colorWithRed:0 green:0 blue:0 alpha:1.0];
	} else if ([theme isEqualToString:@"blue"]) {
		backgroundColor = [UIColor colorWithRed:0.071 green:0.094 blue:0.106 alpha:1.0];
	} else if ([theme isEqualToString:@"sepia"]) {
		backgroundColor = [UIColor colorWithRed:0.310 green:0.302 blue:0.267 alpha:1.0]; 
	} else if ([theme isEqualToString:@"red"]) {
		backgroundColor = [UIColor colorWithRed:0.075 green:0.055 blue:0.051 alpha:1.0]; 
	} else if ([theme isEqualToString:@"deep"]) {
		backgroundColor = [UIColor colorWithRed:0.035 green:0.035 blue:0.043 alpha:1.0];
	} else if ([theme isEqualToString:@"mint"]) {
		backgroundColor = [UIColor colorWithRed:0.364 green:0.376 blue:0.357 alpha:1.0];
	} else if ([theme isEqualToString:@"cream"]) {
		backgroundColor = [UIColor colorWithRed:0.322 green:0.314 blue:0.286 alpha:1.0];
	} else if ([theme isEqualToString:@"acontrast"]) {
		backgroundColor = [UIColor colorWithRed:0.027 green:0.027 blue:0.24 alpha:1.0];
	} else if ([theme isEqualToString:@"pink"]) {
		backgroundColor = [UIColor colorWithRed:1.0 green:0.376 blue:0.357 alpha:1.0];
	} else if ([theme isEqualToString:@"solarize"]) {
		backgroundColor = [UIColor colorWithRed:0.040 green:0.067 blue:0.082 alpha:1.0];
	} else if (!theme) {
		backgroundColor = [UIColor colorWithRed:0.352 green:0.352 blue:0.352 alpha:1.0];
	} else {
		NSString *customThemeString = [[userDefaults stringForKey:[NSString stringWithFormat:@"Theme+%@", theme]] stringByRemovingPercentEncoding];
		if (customThemeString) {
			NSString *customFontColorHex = [customThemeString componentsSeparatedByString:@"#"][3];
			backgroundColor =  [UIColor colorWithHex:customFontColorHex]; 
		} else {
			backgroundColor = [UIColor colorWithRed:0.352 green:0.352 blue:0.352 alpha:1.0];
		}
	} 
	
	return backgroundColor;
}

@end


static UIButton * createUndeleteButton(){
	
	UIButton *undeleteButton = [UIButton buttonWithType:UIButtonTypeCustom];		
	UIImage *undeleteImage = [UIImage imageWithContentsOfFile:@"/var/mobile/Library/Application Support/TFDidThatSay/eye160white.png"];
	
	UIGraphicsBeginImageContextWithOptions(CGSizeMake(20, 20), NO, 0);
	[undeleteImage drawInRect:CGRectMake(0, 0, 20, 20)];
	undeleteImage = UIGraphicsGetImageFromCurrentImageContext();
	UIGraphicsEndImageContext();
	
	UIGraphicsBeginImageContextWithOptions(CGSizeMake(35, 35), NO, 0);
	CGContextRef context = UIGraphicsGetCurrentContext();
	UIGraphicsPushContext(context);
	[undeleteImage drawAtPoint:CGPointMake(7.5, 7.5)];
	UIGraphicsPopContext();
	undeleteImage = UIGraphicsGetImageFromCurrentImageContext();
	UIGraphicsEndImageContext();

	[undeleteButton setImage:undeleteImage forState:UIControlStateNormal];
	undeleteButton.imageView.contentMode = UIViewContentModeScaleAspectFit;
	
	return undeleteButton;
}


//because it wont compile without this
%hook RComment
%end


%hook UIColor

%new
+(UIColor *) colorWithHex:(NSString *) arg1 {
	if (!arg1){
		NSString *firstChar = [arg1 substringToIndex:1];
		if ([firstChar isEqualToString:@"#"]){
			arg1 = [arg1 substringWithRange:NSMakeRange(1, [arg1 length]-1)];
		}
		
		unsigned rgbValue = 0;
		NSScanner *scanner = [NSScanner scannerWithString:arg1];
		[scanner scanHexInt:&rgbValue];
			
		return [UIColor colorWithRed:((rgbValue & 0xFF0000) >> 16)/255.0 green:((rgbValue & 0xFF00) >> 8)/255.0 blue:(rgbValue & 0xFF)/255.0 alpha:0.87];
	} else {
		return nil;
	}
}

%new 
-(NSString *) hexString {
	const CGFloat *components = CGColorGetComponents(self.CGColor);

    CGFloat r = components[0];
    CGFloat g = components[1];
    CGFloat b = components[2];

    return [NSString stringWithFormat:@"#%02lX%02lX%02lX", lroundf(r * 255), lroundf(g * 255), lroundf(b * 255)];
}

%end


%hook CommentDepthCell

-(void) doShortClick{
	%orig;
	
	[self addUndeleteButtonToMenu];
}

-(void) doLongClick	{
	%orig;
	
	[self addUndeleteButtonToMenu];
}

%new 
-(void) addUndeleteButtonToMenu{
	
	NSString *body = [MSHookIvar<id>(self, "comment") body];
	
	if ((isTFDeletedOnly && ([body isEqualToString:@"[deleted]"] || [body isEqualToString:@"[removed]"])) || !isTFDeletedOnly){
	
		id controller = MSHookIvar<id>(self, "parent");
		
		if (MSHookIvar<id>(controller, "menuCell")){
			
			UIStackView *menu = MSHookIvar<UIStackView *>(self, "menu");
			
			if (![[[[menu arrangedSubviews] lastObject] actionsForTarget:self forControlEvent:UIControlEventTouchUpInside] containsObject:@"handleUndeleteComment:"]){
				UIButton *undeleteButton = createUndeleteButton();
				[undeleteButton addTarget:self action:@selector(handleUndeleteComment:) forControlEvents:UIControlEventTouchUpInside];
			
				[menu addArrangedSubview:undeleteButton];
			}
		}
	}
}

%new
-(void) handleUndeleteComment:(id) sender{
	
	[sender setEnabled:NO];

	id comment = MSHookIvar<id>(self, "comment");
	
	[%c(TFHelper) getUndeleteDataWithID:[[comment id] componentsSeparatedByString:@"_"][1] isComment:YES timeout:pushshiftRequestTimeoutValue extraData:@{@"sender" : sender} completionTarget:self completionSelector:@selector(completeUndeleteCommentAction:)];
}

%new
-(void) completeUndeleteCommentAction:(NSDictionary *) data{
	
	NSUserDefaults *userDefaults = [NSUserDefaults standardUserDefaults];
	id textStackDisplayView = MSHookIvar<id>(self, "commentBody");
	id comment = MSHookIvar<id>(self, "comment");
	
	NSString *author = data[@"author"];
	NSString *body = data[@"body"];
	
	//Attributed string generation rewrote from Slide_for_Reddit.TextDisplayStackView.createAttributedChunk(...) 
	
	UIFont *font = [%c(FontGenerator) fontOfSize:MSHookIvar<CGFloat>(textStackDisplayView, "fontSize") submission:NO willOffset:YES];
		
	NSString *themeName = [userDefaults stringForKey:@"theme"];
	UIColor *fontColor = [%c(ColorUtil) fontColorForTheme:themeName];
	UIColor *accentColor = [%c(ColorUtil) accentColorForSub:[comment subreddit]];
	
	NSString *html = [%c(MMMarkdown) HTMLStringWithMarkdown:body extensions:MMMarkdownExtensionsGitHubFlavored error:nil];
	html = [[html stringByReplacingOccurrencesOfString:@"<sup>" withString:@"<font size=\"1\">"] stringByReplacingOccurrencesOfString:@"</sup>" withString:@"</font>"];
	html = [[html stringByReplacingOccurrencesOfString:@"<del>" withString:@"<font color=\"green\">"] stringByReplacingOccurrencesOfString:@"</del>" withString:@"</font>"];
	html = [[html stringByReplacingOccurrencesOfString:@"<code>" withString:@"<font color=\"blue\">"] stringByReplacingOccurrencesOfString:@"</code>" withString:@"</font>"];
	html = [html stringByTrimmingCharactersInSet:[NSCharacterSet whitespaceAndNewlineCharacterSet]];
	
	DTHTMLAttributedStringBuilder *dthtmlBuilder = [[%c(DTHTMLAttributedStringBuilder) alloc] initWithHTML:[html dataUsingEncoding:NSUTF8StringEncoding] options:@{@"DTUseiOS6Attributes": @YES, @"DTDefaultTextColor": fontColor, @"DTDefaultFontSize": @([font pointSize])} documentAttributes:nil];
	
	NSMutableAttributedString *htmlAttributedString = [[NSMutableAttributedString alloc] initWithAttributedString:[dthtmlBuilder generatedAttributedString]];
	NSRange htmlStringRange = NSMakeRange(0, [htmlAttributedString length]);
	
	[[htmlAttributedString mutableString] replaceOccurrencesOfString:@"\t•\t" withString:@" • " options:0 range: htmlStringRange];
	[[htmlAttributedString mutableString] replaceOccurrencesOfString:@"\t◦\t" withString:@"  ◦ " options:0 range: htmlStringRange];
	[[htmlAttributedString mutableString] replaceOccurrencesOfString:@"\t▪\t" withString:@"   ▪ " options:0	 range: htmlStringRange];
	
	[htmlAttributedString removeAttribute:@"CTForegroundColorFromContext" range:htmlStringRange];
	
	[htmlAttributedString enumerateAttributesInRange:htmlStringRange options:NSAttributedStringEnumerationLongestEffectiveRangeNotRequired usingBlock:^(NSDictionary *attributes, NSRange range, BOOL *stop) {
		for (NSString *key in attributes){
			
			if ([(UIColor *) attributes[key] isKindOfClass:[UIColor class]]){
				UIColor *attrColor = (UIColor *) attributes[key];
				if ([[attrColor hexString] isEqualToString:@"#0000FF"]){
					UIFont *tempFont = [UIFont fontWithName:@"Courier" size:font.pointSize];
					
					[htmlAttributedString setAttributes:@{NSForegroundColorAttributeName: accentColor, NSBackgroundColorAttributeName: [%c(ColorUtil) backgroundColorForTheme:themeName], NSFontAttributeName: (tempFont ? tempFont : font)} range:range];
				} else if ([[attrColor hexString] isEqualToString:@"#008000"]) {
					[htmlAttributedString setAttributes:@{NSForegroundColorAttributeName: fontColor, NSFontAttributeName:font} range:range];
				}
			} else if ([(NSURL *) attributes[key] isKindOfClass:[NSURL class]]){
				NSURL *attrUrl = (NSURL *)attributes[key];

				if (([userDefaults objectForKey:@"ENLARGE_LINKS"] == nil) ? YES : [userDefaults boolForKey:@"ENLARGE_LINKS"]){
					[htmlAttributedString addAttribute:NSFontAttributeName value:[%c(FontGenerator) boldFontOfSize:18 submission:NO willOffset:YES] range:range];
				}
				
				[htmlAttributedString addAttribute:NSForegroundColorAttributeName value:accentColor range:range]; 
				[htmlAttributedString addAttribute:NSUnderlineColorAttributeName value:[UIColor clearColor] range:range];
	
				//skipping showLinkContentType b/c not necessary and spoilers b/c MMMarkdown doesn't support them

				[htmlAttributedString yy_setTextHighlightRange:range color: accentColor backgroundColor:nil userInfo:@{@"url": attrUrl}];
				break; 
			} 
		}
	}];
	
	[htmlAttributedString beginEditing];
	[htmlAttributedString enumerateAttribute:NSFontAttributeName inRange:NSMakeRange(0, [htmlAttributedString length]) options:0 usingBlock:^(id value, NSRange range, BOOL *stop){
		
		UIFont *attrFont = (UIFont *)value;
		
		BOOL isBold = (attrFont.fontDescriptor.symbolicTraits & UIFontDescriptorTraitBold) != 0;
		BOOL isItalic = (attrFont.fontDescriptor.symbolicTraits & UIFontDescriptorTraitItalic) != 0;
		
		UIFont *newFont = font;
		
		if (isBold){
			newFont = [%c(FontGenerator) boldFontOfSize:attrFont.pointSize submission:NO willOffset:NO];
		} else if (isItalic){
			newFont = [%c(FontGenerator) italicFontOfSize:attrFont.pointSize submission:NO willOffset:NO];
		}
	
		[htmlAttributedString removeAttribute:NSFontAttributeName range:range];
		[htmlAttributedString addAttribute:NSFontAttributeName value:newFont range:range];	
	
	}];
	[htmlAttributedString endEditing];
	
	NSMutableAttributedString *newCommentText = [MSHookIvar<NSMutableAttributedString *>(self, "cellContent") initWithAttributedString:htmlAttributedString];
	NSAttributedString *tempAttributedString = [[NSAttributedString alloc] initWithString:@""];
	[newCommentText appendAttributedString:tempAttributedString]; //to keep the compiler happy
	
	[comment setAuthor:author];
	[comment setBody:body];	
	
	id controller = MSHookIvar<id>(self, "parent");
	
	[self showMenu:nil];
	[MSHookIvar<id>(controller, "tableView") reloadData];
	
	[data[@"sender"] setEnabled:YES];
}

%end

%end


static void loadPrefs(){
	NSMutableDictionary *prefs = [[NSMutableDictionary alloc] initWithContentsOfFile:@"/User/Library/Preferences/com.lint.undelete.prefs.plist"];
	
	if (prefs){
		
		if ([prefs objectForKey:@"isSlideEnabled"] != nil){
			isSlideEnabled = [[prefs objectForKey:@"isSlideEnabled"] boolValue];
		} else {
			isSlideEnabled = YES;
		}
		
		if ([prefs objectForKey:@"isTFDeletedOnly"] != nil){
			isTFDeletedOnly = [[prefs objectForKey:@"isTFDeletedOnly"] boolValue];
		} else {
			isTFDeletedOnly = YES;
		}
		
		if ([prefs objectForKey:@"requestTimeoutValue"] != nil){
			pushshiftRequestTimeoutValue = [[prefs objectForKey:@"requestTimeoutValue"] doubleValue];
		} else {
			pushshiftRequestTimeoutValue = 10;
		}
		
	} else {
		isSlideEnabled = YES;
		isTFDeletedOnly = YES;
		pushshiftRequestTimeoutValue = 10;
	}	
}

static void prefsChanged(CFNotificationCenterRef center, void *observer, CFStringRef name, const void *object, CFDictionaryRef userInfo) {
  loadPrefs();
}


%ctor {
	loadPrefs();
	
	NSString* processName = [[NSProcessInfo processInfo] processName];

	if ([processName isEqualToString:@"Slide for Reddit"]){
		if (isSlideEnabled){
			
			CFNotificationCenterAddObserver(CFNotificationCenterGetDarwinNotifyCenter(), NULL, prefsChanged, CFSTR("com.lint.undelete.prefs.changed"), NULL, CFNotificationSuspensionBehaviorDeliverImmediately);
			
			%init(Slide, CommentDepthCell = objc_getClass("Slide_for_Reddit.CommentDepthCell"), RComment = objc_getClass("Slide_for_Reddit.RSubmission"));
		}
	}
}