Add BaconReader support

prefs/Resources/Root.plist

@@ -82,6 +82,20 @@
 			<key>label</key>
 			<string>Enable Slide</string>
 		</dict>
+		<dict>
+			<key>PostNotification</key>
+			<string>com.lint.undelete.prefs.changed</string>
+			<key>cell</key>
+			<string>PSSwitchCell</string>
+			<key>default</key>
+			<true/>
+			<key>defaults</key>
+			<string>com.lint.undelete.prefs</string>
+			<key>key</key>
+			<string>isBaconReaderEnabled</string>
+			<key>label</key>
+			<string>Enable BaconReader</string>
+		</dict>
 		<dict>
 			<key>cell</key>
 			<string>PSGroupCell</string>
@@ -132,6 +146,20 @@
 			<key>label</key>
 			<string>Slide | Deleted only</string>
 		</dict>
+		<dict>
+			<key>PostNotification</key>
+			<string>com.lint.undelete.prefs.changed</string>
+			<key>cell</key>
+			<string>PSSwitchCell</string>
+			<key>default</key>
+			<true/>
+			<key>defaults</key>
+			<string>com.lint.undelete.prefs</string>
+			<key>key</key>
+			<string>isBaconReaderDeletedOnly</string>
+			<key>label</key>
+			<string>BaconReader | Deleted only</string>
+		</dict>
 		<dict>
 			<key>cell</key>
 			<string>PSGroupCell</string>

tweak/BaconReader.h

@@ -0,0 +1,59 @@
+
+/* -- Votable Interfaces -- */
+
+@interface BRVotable 
+@property(strong, nonatomic) NSString *serverID;
+@property(strong, nonatomic) NSAttributedString *attributedDescriptionString;
+@property(assign, nonatomic) CGSize cellSize;
+@end
+
+@interface BRComment : BRVotable
+@property(strong, nonatomic) NSString *author;
+@property(strong, nonatomic) NSString *body;
+@property(strong, nonatomic) NSString *body_html;
+@end
+
+@interface BRStory : BRVotable
+@property(strong, nonatomic) NSString *author;
+@property(strong, nonatomic) NSString *selftext;
+@property(strong, nonatomic) NSString *selftext_html;
+@property(assign, nonatomic) BOOL is_selfValue;
+@end
+
+/* -- Comment Interfaces -- */
+
+@interface CommentCell
+@property(strong, nonatomic) id cellView;
+@end
+
+@interface CommentCellView
+@property(strong, nonatomic) id comment;
+@property(strong, nonatomic) id parentDelegate;
+@end
+
+@interface BRComposeCommentViewController : NSObject
+@property(assign, nonatomic) BOOL editComment;
+@end
+
+/* -- Post Interfaces -- */
+
+@interface  StoryDetailView
+@property(strong, nonatomic) UITableView *tableView;
+-(void) refreshTouched;
+@end
+
+@interface StoryDetailViewController
+@property(strong, nonatomic) id story;
+@property(strong, nonatomic) id detailPage;
+
+//custom elements
+-(void) handleUndeleteCommentAction;
+-(void) handleUndeletePostAction;
+@end
+
+/* -- Other Interfaces -- */
+
+@interface BRUtils
++(id) attributedDescriptionForComment:(id) arg1;
++(id) createAttributedStringFromHTML:(id) arg1 options:(id) arg2;
+@end

tweak/BaconReader.xm

@@ -0,0 +1,263 @@
+
+#import "BaconReader.h"
+#import "assets/MMMarkdown.h"
+
+static BOOL isBaconReaderEnabled;
+static BOOL isBaconReaderDeletedOnly;
+static CGFloat pushshiftRequestTimeoutValue;
+
+%group BaconReader
+
+BOOL shouldHaveBRUndeleteAction = NO;
+NSString *tfPostSelftext;
+NSString *tfPostAuthor;
+id tfCommentCellView;
+id tfStoryController;
+
+
+%hook CommentCell
+%end
+
+%hook CommentCellView
+%end
+
+%hook StoryDetailView
+%end
+
+
+%hook BRComment
+
+-(BOOL) contains_htmlValue{
+	return YES;
+} 
+
+-(BOOL) primitiveContains_htmlValue{
+	return YES;
+}
+
+-(BOOL) is_deletedValue{
+	return NO;
+}
+
+-(BOOL) primitiveIs_deletedValue{
+	return NO;
+}
+
+%end
+
+
+%hook BRStory
+
++(id) storyWithDictionary:(id) arg1 inContext:(id) arg2 {
+	id orig = %orig;
+	
+	if (tfPostSelftext){
+		[orig setSelftext_html:tfPostSelftext];
+		[orig setAuthor:tfPostAuthor];
+		tfPostSelftext = nil;
+		tfPostAuthor = nil;
+	}
+	
+	return orig;
+}
+
+%end
+
+
+%hook UIViewController
+
+-(void) presentViewController:(id) arg1 animated:(BOOL) arg2 completion:(id) arg3 {
+	
+	if ([arg1 isKindOfClass:[UIAlertController class]] && shouldHaveBRUndeleteAction){
+		
+		UIAlertAction *undeleteAction;
+		
+		if (tfCommentCellView){
+			undeleteAction = [UIAlertAction actionWithTitle:@"TF Did That Say?" style:nil handler:^(UIAlertAction* action){[tfStoryController handleUndeleteCommentAction];}];
+		} else {
+			undeleteAction = [UIAlertAction actionWithTitle:@"TF Did That Say?" style:nil handler:^(UIAlertAction* action){[tfStoryController handleUndeletePostAction];}];
+		}
+		 
+		[arg1 addAction:undeleteAction];
+	}
+	
+	%orig;
+}
+
+%end
+
+
+%hook StoryDetailViewController
+
+-(void) showMoreCommentActions:(id) arg1 showAll:(BOOL) arg2 {
+	
+	NSString *commentBody = [[arg1 comment] body];
+	
+	if ((isBaconReaderDeletedOnly && ([commentBody isEqualToString:@"[deleted]"] || [commentBody isEqualToString:@"[removed]"])) || !isBaconReaderDeletedOnly) {
+		shouldHaveBRUndeleteAction = YES;
+		tfCommentCellView = arg1;
+		tfStoryController = self;
+		tfPostSelftext = nil;
+	}
+	
+	%orig;
+	
+	shouldHaveBRUndeleteAction = NO;
+}
+
+-(void) menuTouchedWithSender:(id) arg1 {
+	
+	if ([[self story] is_selfValue]){
+		NSString *postBody = [[self story] selftext];
+		
+		if ((isBaconReaderDeletedOnly && ([postBody isEqualToString:@"[deleted]"] || [postBody isEqualToString:@"[removed]"])) || !isBaconReaderDeletedOnly) {
+			shouldHaveBRUndeleteAction = YES;
+			tfCommentCellView = nil;
+			tfStoryController = self;
+			tfPostSelftext = nil;
+		}
+	}
+	
+	%orig;
+	
+	shouldHaveBRUndeleteAction = NO;
+}
+
+%new
+-(void) handleUndeleteCommentAction{
+	
+	id comment = [tfCommentCellView comment];
+	
+	NSMutableURLRequest *request = [[NSMutableURLRequest alloc] init];
+	NSOperationQueue *queue = [[NSOperationQueue alloc] init];
+
+	[request setURL:[NSURL URLWithString:[NSString stringWithFormat:@"https://api.pushshift.io/reddit/search/comment/?ids=%@&fields=author,body", [comment serverID]]]];
+	[request setHTTPMethod:@"GET"];
+	[request setTimeoutInterval:pushshiftRequestTimeoutValue];
+
+	[NSURLConnection sendAsynchronousRequest:request queue:queue completionHandler:^(NSURLResponse *response, NSData *data, NSError *error) {
+	
+		NSString *author = @"[author]";
+		NSString *body = @"[body]";
+
+		if (data != nil && error == nil){
+			id jsonData = [NSJSONSerialization JSONObjectWithData:data options:0 error:&error];
+			if ([[jsonData objectForKey:@"data"] count] != 0){
+				author = [[jsonData objectForKey:@"data"][0] objectForKey:@"author"];
+				body = [[jsonData objectForKey:@"data"][0] objectForKey:@"body"];
+				if ([body isEqualToString:@"[deleted]"] || [body isEqualToString:@"[removed]"]){
+					body = @"[pushshift was unable to archive this]";
+				}
+			} else {
+				body = @"[pushshift has not archived this yet]";
+			}
+		} else if (error != nil || data == nil){
+			body = [NSString stringWithFormat:@"[an error occured while attempting to contact pushshift api (%@)]", [error localizedDescription]];
+		}
+		
+		[comment setAuthor:author];
+		[comment setBody:body];
+		[comment setBody_html:[%c(MMMarkdown) HTMLStringWithMarkdown:body extensions:MMMarkdownExtensionsGitHubFlavored error:nil]];
+		[comment setAttributedDescriptionString:nil];
+		
+		NSAttributedString *commentAttrString = [%c(BRUtils) attributedDescriptionForComment:comment];
+		[comment setAttributedDescriptionString:commentAttrString];
+		
+		[[[self detailPage] tableView] performSelectorOnMainThread:@selector(reloadData) withObject:nil waitUntilDone:NO];
+	}];
+}
+
+%new
+-(void) handleUndeletePostAction{
+	
+	id post = [self story];
+	
+	NSMutableURLRequest *request = [[NSMutableURLRequest alloc] init];
+	NSOperationQueue *queue = [[NSOperationQueue alloc] init];
+
+	[request setURL:[NSURL URLWithString:[NSString stringWithFormat:@"https://api.pushshift.io/reddit/search/submission/?ids=%@&fields=author,selftext", [post serverID]]]];
+	[request setHTTPMethod:@"GET"];
+	[request setTimeoutInterval:pushshiftRequestTimeoutValue];
+
+	[NSURLConnection sendAsynchronousRequest:request queue:queue completionHandler:^(NSURLResponse *response, NSData *data, NSError *error) {
+		
+		NSString *author = @"[author]";
+		NSString *body = @"[body]";
+
+		if (data != nil && error == nil){
+			id jsonData = [NSJSONSerialization JSONObjectWithData:data options:0 error:&error];
+			if ([[jsonData objectForKey:@"data"] count] != 0){
+				author = [[jsonData objectForKey:@"data"][0] objectForKey:@"author"];
+				body = [[jsonData objectForKey:@"data"][0] objectForKey:@"selftext"];
+				if ([body isEqualToString:@"[deleted]"] || [body isEqualToString:@"[removed]"]){
+					body = @"[pushshift was unable to archive this]";
+				}
+			} else {
+				body = @"[pushshift has not archived this yet]";
+			}
+		} else if (error != nil || data == nil){
+			body = [NSString stringWithFormat:@"[an error occured while attempting to contact pushshift api (%@)]", [error localizedDescription]];
+		}
+		
+		tfPostAuthor = author;
+		tfPostSelftext = [%c(MMMarkdown) HTMLStringWithMarkdown:body extensions:MMMarkdownExtensionsGitHubFlavored error:nil];
+	
+		[[self detailPage] performSelectorOnMainThread:@selector(refreshTouched) withObject:nil waitUntilDone:NO];
+		
+	}];
+}
+
+%end
+
+%end
+
+
+static void loadPrefs(){
+	NSMutableDictionary *prefs = [[NSMutableDictionary alloc] initWithContentsOfFile:@"/User/Library/Preferences/com.lint.undelete.prefs.plist"];
+	
+	if (prefs){
+		
+		if ([prefs objectForKey:@"isBaconReaderEnabled"] != nil){
+			isBaconReaderEnabled = [[prefs objectForKey:@"isBaconReaderEnabled"] boolValue];
+		} else {
+			isBaconReaderEnabled = YES;
+		}
+		
+		if ([prefs objectForKey:@"requestTimeoutValue"] != nil){
+			pushshiftRequestTimeoutValue = [[prefs objectForKey:@"requestTimeoutValue"] doubleValue];
+		} else {
+			pushshiftRequestTimeoutValue = 10;
+		}
+		
+		if ([prefs objectForKey:@"isBaconReaderDeletedOnly"] != nil) {
+			isBaconReaderDeletedOnly = [[prefs objectForKey:@"isBaconReaderDeletedOnly"] boolValue];
+		} else {
+			isBaconReaderDeletedOnly = YES;
+		}
+		
+	} else {
+		isBaconReaderEnabled = YES;
+		isBaconReaderDeletedOnly = 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:@"BaconReader"]){		
+		if (isBaconReaderEnabled){
+			
+			CFNotificationCenterAddObserver(CFNotificationCenterGetDarwinNotifyCenter(), NULL, prefsChanged, CFSTR("com.lint.undelete.prefs.changed"), NULL, CFNotificationSuspensionBehaviorDeliverImmediately);
+			
+			%init(BaconReader, StoryDetailView = objc_getClass("BaconReader.StoryDetailView"), StoryDetailViewController = objc_getClass("BaconReader.StoryDetailViewController"), CommentCellView = objc_getClass("BaconReader.CommentCellView"), CommentCell = objc_getClass("BaconReader.CommentCell"));
+		}
+	}
+}