TFDidThatSay/tweak/Apollo.xm

#import "Apollo.h"
#import "assets/TFHelper.h"

static BOOL isTFDeletedOnly;
static BOOL isApolloEnabled;
static CGFloat pushshiftRequestTimeoutValue;
static BOOL shouldApolloHaveButton;

%group Apollo

NSDictionary* apolloBodyAttributes = nil;
BOOL shouldAddUndeleteCell = NO;
id apolloCommentCell;
id apolloCommentController;

%hook ApolloButtonNode
%end

%hook IconActionTableViewCell
%end


%hook RKComment

-(BOOL) isDeleted{
	return NO;
}

-(BOOL) isModeratorRemoved{
	return NO;
}

%end


%hook RKLink
%property(strong, nonatomic) NSString *undeleteAuthor;

-(id) author{
	
	if ([self undeleteAuthor]){
		return [self undeleteAuthor];
	} else {
		return %orig;
	}
}

%end


%hook MarkdownRenderer

+(id) attributedStringFromHTML:(id)arg1 attributes:(id) arg2 compact:(BOOL) arg3{

	apolloBodyAttributes = [arg2 copy];
	
	return %orig;
}

%end


%hook ActionController

-(id) tableView:(id) arg1 cellForRowAtIndexPath:(NSIndexPath *) arg2{
	
	if (shouldAddUndeleteCell){
		if ([arg2 row] == [self tableView:arg1 numberOfRowsInSection:0] - 1){
			
			id undeleteCell = [arg1 dequeueReusableCellWithIdentifier:@"IconActionCell" forIndexPath:arg2];
			id prevCell = [arg1 dequeueReusableCellWithIdentifier:@"IconActionCell"];
			
			UIImageView *prevCellImageView = MSHookIvar<UIImageView *>(prevCell, "iconImageView");
			CGSize prevImageSize = [[prevCellImageView image] size];
			
			UIImage *undeleteImage; 
			
			if (@available(iOS 13.0, *)){
				
				undeleteImage = [UIImage systemImageNamed:@"eye"];
				
				if (!undeleteImage){
					undeleteImage = [UIImage imageWithContentsOfFile:@"/var/mobile/Library/Application Support/TFDidThatSay/eye160dark.png"];
				}
			} else {
				undeleteImage = [UIImage imageWithContentsOfFile:@"/var/mobile/Library/Application Support/TFDidThatSay/eye160dark.png"];
			}
			
			CGFloat undeleteImageSizeValue = prevImageSize.width > prevImageSize.height ? prevImageSize.width : prevImageSize.height;
			
			if (undeleteImageSizeValue == 0){
				undeleteImageSizeValue = 25;
			}
			
			UIGraphicsBeginImageContextWithOptions(CGSizeMake(undeleteImageSizeValue, undeleteImageSizeValue), NO, 0);
			[undeleteImage drawInRect:CGRectMake(0, 0, undeleteImageSizeValue, undeleteImageSizeValue)];
			undeleteImage = [UIGraphicsGetImageFromCurrentImageContext() imageWithRenderingMode:UIImageRenderingModeAlwaysTemplate];
			UIGraphicsEndImageContext();
			
			UILabel *undeleteLabel = MSHookIvar<UILabel *>(undeleteCell, "actionTitleLabel");
			UIImageView *undeleteImageView = MSHookIvar<UIImageView *>(undeleteCell, "iconImageView");
			
			undeleteLabel.text = @"TF Did That Say?";
			undeleteImageView.image = undeleteImage;
			
			return undeleteCell;
		}
	}
	
	return %orig;	
}

-(void) tableView:(id) arg1 didSelectRowAtIndexPath:(NSIndexPath *)arg2{
	
	if (shouldAddUndeleteCell){
		if ([arg2 row] == [self tableView:arg1 numberOfRowsInSection:0] - 1){

			if (apolloCommentCell){
				[apolloCommentCell undeleteCellWasSelected];
			} else {
				[apolloCommentController undeleteCellWasSelected];
			}
		}
	} 
	
	%orig;
}

-(NSInteger) tableView:(id) arg1 numberOfRowsInSection:(NSInteger) arg2{
	
	if (shouldAddUndeleteCell){
		return %orig + 1;
	} else {
		return %orig;
	}
}

-(id) animationControllerForDismissedController:(id) arg1{
	
	shouldAddUndeleteCell = NO;
	
	return %orig;
}

%end


%hook CommentCellNode
%property(strong,nonatomic) UIButton *undeleteButton;

-(void) moreOptionsTappedWithSender:(id) arg1{
	
	if (!shouldApolloHaveButton){
		
		NSString *commentBody = [MSHookIvar<RKComment *>(self, "comment") body];
		
		if ([%c(TFHelper) shouldShowUndeleteButtonWithInfo:commentBody isDeletedOnly:isTFDeletedOnly]){
			shouldAddUndeleteCell = YES;
			apolloCommentCell = self;
			apolloCommentController = nil;
		}		
	}
	
	%orig;
}

-(void) longPressedWithGestureRecognizer:(id) arg1{
	
	if (!shouldApolloHaveButton){
		
		NSString *commentBody = [MSHookIvar<RKComment *>(self, "comment") body];
		
		if ([%c(TFHelper) shouldShowUndeleteButtonWithInfo:commentBody isDeletedOnly:isTFDeletedOnly]){
			shouldAddUndeleteCell = YES;
			apolloCommentCell = self;
			apolloCommentController = nil;
		}
	}
	
	%orig;
}

-(void) didLoad {
	%orig;
	
	if (shouldApolloHaveButton){
	
		NSString *commentBody = [MSHookIvar<RKComment *>(self, "comment") body];
		
		if ([%c(TFHelper) shouldShowUndeleteButtonWithInfo:commentBody isDeletedOnly:isTFDeletedOnly]){
		
			CGFloat imageSize = 20.0f;

			UIButton *undeleteButton = [UIButton buttonWithType:UIButtonTypeCustom];
			[undeleteButton addTarget:self action:@selector(didTapUndeleteButton:) forControlEvents:UIControlEventTouchUpInside];
			undeleteButton.frame = CGRectMake(0, 0, imageSize, imageSize);
			
			UIImage* undeleteImage = [UIImage imageWithContentsOfFile:@"/var/mobile/Library/Application Support/TFDidThatSay/eye160dark.png"];
			[undeleteButton setImage:undeleteImage forState:UIControlStateNormal];

			[[self view] addSubview:undeleteButton];
			[self setUndeleteButton:undeleteButton];
		}
	}
}

-(void) _layoutSublayouts{
	%orig;
	
	if (shouldApolloHaveButton){
		if ([self undeleteButton]){
		
			CGFloat imageSize = 20.0f;

			id moreNode = MSHookIvar<id>(self, "moreOptionsNode");
			id ageNode = MSHookIvar<id>(self, "ageNode");

			CGRect nodeFrame = [moreNode frame];
			CGFloat centerHeight = (nodeFrame.size.height + nodeFrame.origin.y * 2) / 2.0f;
			CGFloat nodeSpacing = [ageNode frame].origin.x - nodeFrame.origin.x - nodeFrame.size.width;

			[[self undeleteButton] setFrame:CGRectMake(nodeFrame.origin.x - imageSize - nodeSpacing, centerHeight - (imageSize / 2), imageSize, imageSize)];
		}
	}
}

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

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

%new
-(void) undeleteCellWasSelected{

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

%new
-(void) completeUndeleteCommentAction:(NSDictionary *) data{
	
	RKComment *comment = MSHookIvar<RKComment *>(self, "comment");
	id bodyNode = MSHookIvar<id>(self, "bodyNode");
	id authorNode = MSHookIvar<id>(self, "authorNode");
	
	NSString *author = data[@"author"];
	NSString *body = data[@"body"];
	
	[comment setAuthor:author];
	[comment setBody:body];
	
	NSAttributedString *prevAuthorAttributedString = [authorNode attributedTitleForState:UIControlStateNormal];
	NSDictionary *authorStringAttributes = [prevAuthorAttributedString attributesAtIndex:0 longestEffectiveRange:nil inRange:NSMakeRange(0, [prevAuthorAttributedString length])];
	NSAttributedString *newAuthorAttributedString = [[NSAttributedString alloc] initWithString:author attributes:authorStringAttributes];

	[authorNode setAttributedTitle:newAuthorAttributedString forState:UIControlStateNormal];
	[bodyNode setAttributedString:[%c(MarkdownRenderer) attributedStringFromMarkdown:body withAttributes:apolloBodyAttributes]];
	
	if ([data objectForKey:@"sender"]) {
		[data[@"sender"] setEnabled:YES];
	}
}

%end


%hook CommentsViewController
%property(strong, nonatomic) id headerCellNode; 

-(void) moreOptionsBarButtonItemTappedWithSender:(id) arg1{
	
	RKLink *post = MSHookIvar<RKLink *>(self, "link");
	NSString *postBody = [post selfText];
	
	if ([post isSelfPost]){
		if ([%c(TFHelper) shouldShowUndeleteButtonWithInfo:postBody isDeletedOnly:isTFDeletedOnly]){
			shouldAddUndeleteCell = YES;
			apolloCommentCell = nil;
			apolloCommentController = self;
		}
	}
	
	%orig;
}

%new
-(void) undeleteCellWasSelected{
	
	RKLink *post = MSHookIvar<RKLink *>(self, "link");
	
	[%c(TFHelper) getUndeleteDataWithID:[[post fullName] componentsSeparatedByString:@"_"][1] isComment:NO timeout:pushshiftRequestTimeoutValue extraData:nil completionTarget:self completionSelector:@selector(completeUndeletePostAction:)];
}

%new
-(void) completeUndeletePostAction:(NSDictionary *) data{
	
	RKLink *post = MSHookIvar<RKLink *>(self, "link");
	
	id headerCellNode = [self headerCellNode];
	id bodyNode = MSHookIvar<id>(headerCellNode, "bodyNode");
	id postInfoNode = MSHookIvar<id>(headerCellNode, "postInfoNode");
	id authorNode = MSHookIvar<id>(postInfoNode, "authorButtonNode");
	
	NSString *author = data[@"author"];
	NSString *authorTextString = [NSString stringWithFormat:@"by %@", author];
	NSString *body = data[@"body"];
	
	[post setUndeleteAuthor:author];
	[post setSelfText:body];
	
	NSAttributedString *prevAuthorAttributedString = [authorNode attributedTitleForState:UIControlStateNormal];
	NSDictionary *authorStringAttributes = [prevAuthorAttributedString attributesAtIndex:0 longestEffectiveRange:nil inRange:NSMakeRange(0, [prevAuthorAttributedString length])];
	NSAttributedString* newAuthorAttributedString = [[NSAttributedString alloc] initWithString:authorTextString attributes:authorStringAttributes];

	[authorNode setAttributedTitle:newAuthorAttributedString forState:UIControlStateNormal];
	[bodyNode setAttributedString:[%c(MarkdownRenderer) attributedStringFromMarkdown:body withAttributes:apolloBodyAttributes]];
}

%end


%hook CommentsHeaderCellNode

-(void) didLoad{
	%orig;
	
	[[self closestViewController] setHeaderCellNode:self];
}

-(void) _layoutSublayouts{
	%orig;
	
	[[self closestViewController] setHeaderCellNode:self];
}

-(void) longPressedWithGestureRecognizer:(id) arg1{
	
	RKLink *post = MSHookIvar<RKLink *>(self, "link");
	NSString *postBody = [post selfText];
	
	if ([post isSelfPost]){
		if ([%c(TFHelper) shouldShowUndeleteButtonWithInfo:postBody isDeletedOnly:isTFDeletedOnly]){
			shouldAddUndeleteCell = YES;
			apolloCommentCell = nil;
			apolloCommentController = self;
		}
	}
	
	%orig;
}

%end

%end


static void loadPrefs(){
	NSMutableDictionary *prefs = [[NSMutableDictionary alloc] initWithContentsOfFile:@"/User/Library/Preferences/com.lint.undelete.prefs.plist"];
	
	if (prefs){
		
		if ([prefs objectForKey:@"isApolloEnabled"] != nil) {
			isApolloEnabled = [[prefs objectForKey:@"isApolloEnabled"] boolValue];
		} else {
			isApolloEnabled = 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;
		}
		
		if ([prefs objectForKey:@"shouldApolloHaveButton"] != nil){
			shouldApolloHaveButton = [[prefs objectForKey:@"shouldApolloHaveButton"] boolValue];
		} else {
			shouldApolloHaveButton = NO;
		}

	} else {
		isApolloEnabled = YES;
		isTFDeletedOnly = YES;
		pushshiftRequestTimeoutValue = 10;
		shouldApolloHaveButton = NO;
	}	
}

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:@"Apollo"]){
		if (isApolloEnabled){
			
			CFNotificationCenterAddObserver(CFNotificationCenterGetDarwinNotifyCenter(), NULL, prefsChanged, CFSTR("com.lint.undelete.prefs.changed"), NULL, CFNotificationSuspensionBehaviorDeliverImmediately);
			
			%init(Apollo, CommentsHeaderCellNode = objc_getClass("Apollo.CommentsHeaderCellNode"), CommentCellNode = objc_getClass("Apollo.CommentCellNode"), ApolloButtonNode = objc_getClass("Apollo.ApolloButtonNode"), ActionController = objc_getClass("Apollo.ActionController"), IconActionTableViewCell = objc_getClass("Apollo.IconActionTableViewCell"), CommentsViewController = objc_getClass("Apollo.CommentsViewController"));
		}
	}
}