Mighel88
/
TFDidThatSay
複製自 lint/TFDidThatSay
關注 1
收藏 0
複製 0
程式碼 問題 0 合併請求 0 版本發佈 0 Wiki 活動
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
31 提交
1 分支
目錄樹: bb0d0d9e95
分支列表 標籤
${ item.name }
建立分支 ${ searchTerm }
從 'bb0d0d9e95'
${ noResults }
TFDidThatSay/tweak/Slide.xm

577 line
21KB
原始文件 Blame 歷史記錄 

  1. 

  2. #import "Slide.h"

  3. #import "assets/TFHelper.h"

  4. #import "assets/MMMarkdown/MMMarkdown.h"

  5. 

  6. static BOOL isSlideEnabled;

  7. static BOOL isTFDeletedOnly;

  8. static CGFloat pushshiftRequestTimeoutValue;

  9. 

  10. %group Slide

  11. 

  12. BOOL shouldSlideOverridePostInfo = NO;

  13. NSString *slidePostOverrideAuthor;

  14. NSString *slidePostOverrideBodyHtml;

  15. 

  16. @implementation FontGenerator

  17. 

  18. +(UIFont *) fontOfSize:(CGFloat) size submission:(BOOL) isSubmission willOffset:(BOOL) willOffset{

  19. 	NSUserDefaults *userDefaults = [NSUserDefaults standardUserDefaults];

  20. 	

  21. 	NSString *fontName;

  22. 	CGFloat fontSize = size;

  23. 	

  24. 	if (willOffset){

  25. 		if (isSubmission){

  26. 			fontSize += ([userDefaults objectForKey:@"POST_FONT_SIZE"] == nil) ? 0 : [userDefaults integerForKey:@"POST_FONT_SIZE"];

  27. 		} else {

  28. 			fontSize += ([userDefaults objectForKey:@"COMMENT_FONT_SIZE"] == nil) ? -2 : [userDefaults integerForKey:@"COMMENT_FONT_SIZE"];

  29. 		}	

  30. 	}

  31. 	

  32. 	if ([userDefaults stringForKey:(isSubmission ? @"postfont" : @"commentfont")] == nil){

  33. 		fontName =  isSubmission ? @"AvenirNext-DemiBold" : @"AvenirNext-Medium";

  34. 	} else {

  35. 		fontName = [userDefaults stringForKey:(isSubmission ? @"postfont" : @"commentfont")];

  36. 	}

  37. 	

  38. 	UIFont *font = [UIFont fontWithName:fontName size:fontSize]; 

  39. 	

  40. 	if (!font){

  41. 		font = [UIFont systemFontOfSize:fontSize];

  42. 	}

  43. 	

  44. 	return font;

  45. }

  46.  

  47. +(UIFont *) boldFontOfSize:(CGFloat) size submission:(BOOL) isSubmission willOffset:(BOOL) willOffset {

  48. 	UIFont *font = [self fontOfSize:size submission:isSubmission willOffset:willOffset];

  49. 	

  50. 	if ([font.fontName isEqualToString:[UIFont systemFontOfSize:10].fontName]){

  51. 		return [UIFont boldSystemFontOfSize:font.pointSize];

  52. 	} else {

  53. 		UIFontDescriptor *desc = [font.fontDescriptor fontDescriptorWithSymbolicTraits:UIFontDescriptorTraitBold];

  54. 		

  55. 		if (desc == nil){

  56. 			return font;

  57. 		} else {

  58. 			return [UIFont fontWithDescriptor:desc size: 0];

  59. 		}

  60. 	}

  61. }

  62. 

  63. +(UIFont *) italicFontOfSize:(CGFloat) size submission:(BOOL) isSubmission willOffset:(BOOL) willOffset {

  64. 	UIFont *font = [self fontOfSize:size submission:isSubmission willOffset:willOffset];

  65. 	

  66. 	if ([font.fontName isEqualToString:[UIFont systemFontOfSize:10].fontName]){

  67. 		return [UIFont italicSystemFontOfSize:font.pointSize];

  68. 	} else {

  69. 		UIFontDescriptor *desc = [font.fontDescriptor fontDescriptorWithSymbolicTraits:UIFontDescriptorTraitItalic];

  70. 		

  71. 		if (desc == nil){

  72. 			return font;

  73. 		} else {

  74. 			return [UIFont fontWithDescriptor:desc size: 0];

  75. 		}

  76. 	}

  77. }

  78. 

  79. @end

  80. 

  81. 

  82. @implementation ColorUtil 

  83. 

  84. +(UIColor *) accentColorForSub:(NSString *) subreddit{

  85. 	NSUserDefaults *userDefaults = [NSUserDefaults standardUserDefaults];

  86. 	NSData *colorData = [userDefaults dataForKey:[NSString stringWithFormat:@"accent+%@", subreddit]];

  87. 	UIColor *color = [NSKeyedUnarchiver unarchiveObjectWithData:colorData];

  88. 	if (color) {

  89. 		return color;

  90. 	} else {

  91. 		UIColor *baseAccentColor = [NSKeyedUnarchiver unarchiveObjectWithData:[userDefaults dataForKey:@"accentcolor"]];

  92. 		

  93. 		if (baseAccentColor){

  94. 			return baseAccentColor;

  95. 		} else {

  96. 			return [UIColor colorWithRed:0.161 green:0.475 blue:1.0 alpha:1.0];

  97. 		}

  98. 	}

  99. }

  100. 

  101. +(UIColor *) fontColorForTheme:(NSString *)theme{

  102. 	UIColor *fontColor;

  103. 	NSUserDefaults *userDefaults = [NSUserDefaults standardUserDefaults];

  104. 	

  105. 	//if only switch blocks worked with strings...

  106. 	if ([theme isEqualToString:@"light"]) {

  107. 		fontColor = [UIColor colorWithRed:0 green:0 blue:0 alpha:0.87];

  108. 	} else if ([theme isEqualToString:@"dark"]) {

  109. 		fontColor = [UIColor colorWithRed:1.0 green:1.0 blue:1.0 alpha:0.87];

  110. 	} else if ([theme isEqualToString:@"black"]) {

  111. 		fontColor = [UIColor colorWithRed:1.0 green:1.0 blue:1.0 alpha:0.87];

  112. 	} else if ([theme isEqualToString:@"blue"]) {

  113. 		fontColor = [UIColor colorWithRed:1.0 green:1.0 blue:1.0 alpha:0.87];

  114. 	} else if ([theme isEqualToString:@"sepia"]) {

  115. 		fontColor = [UIColor colorWithRed:0.243 green:.239 blue:.212 alpha:0.87]; 

  116. 	} else if ([theme isEqualToString:@"red"]) {

  117. 		fontColor = [UIColor colorWithRed:1.0 green:0.969 blue:0.929 alpha:0.87]; 

  118. 	} else if ([theme isEqualToString:@"deep"]) {

  119. 		fontColor = [UIColor colorWithRed:1.0 green:1.0 blue:1.0 alpha:0.87];

  120. 	} else if ([theme isEqualToString:@"mint"]) {

  121. 		fontColor = [UIColor colorWithRed:0.035 green:0.212 blue:0.059 alpha:0.87];

  122. 	} else if ([theme isEqualToString:@"cream"]) {

  123. 		fontColor = [UIColor colorWithRed:0.267 green:0.255 blue:0.224 alpha:0.87];

  124. 	} else if ([theme isEqualToString:@"acontrast"]) {

  125. 		fontColor = [UIColor colorWithRed:1.0 green:1.0 blue:1.0 alpha:0.87];

  126. 	} else if ([theme isEqualToString:@"pink"]) {

  127. 		fontColor = [UIColor colorWithRed:0.149 green:0.157 blue:0.267 alpha:0.87];

  128. 	} else if ([theme isEqualToString:@"solarize"]) {

  129. 		fontColor = [UIColor colorWithRed:0.514 green:0.580 blue:0.588 alpha:0.87];

  130. 	} else if (!theme) {

  131. 		fontColor = [UIColor colorWithRed:0 green:0 blue:0 alpha:0.87];

  132. 	} else {

  133. 		NSString *customThemeString = [[userDefaults stringForKey:[NSString stringWithFormat:@"Theme+%@", theme]] stringByRemovingPercentEncoding];

  134. 		if (customThemeString) {

  135. 			NSString *customFontColorHex = [customThemeString componentsSeparatedByString:@"#"][4];

  136. 			fontColor =  [UIColor colorWithHex:customFontColorHex]; 

  137. 		} else {

  138. 			fontColor = [UIColor colorWithRed:0 green:0 blue:0 alpha:0.87];

  139. 		}

  140. 	}

  141. 	

  142. 	return fontColor;

  143. }

  144. 

  145. +(UIColor *) backgroundColorForTheme:(NSString *) theme{

  146. 	UIColor *backgroundColor;

  147. 	NSUserDefaults *userDefaults = [NSUserDefaults standardUserDefaults];

  148. 

  149. 	if ([theme isEqualToString:@"light"]) {

  150. 		backgroundColor = [UIColor colorWithRed:0.352 green:0.352 blue:0.352 alpha:1.0];

  151. 	} else if ([theme isEqualToString:@"dark"]) {

  152. 		backgroundColor = [UIColor colorWithRed:0.051 green:0.051 blue:0.051 alpha:1.0];

  153. 	} else if ([theme isEqualToString:@"black"]) {

  154. 		backgroundColor = [UIColor colorWithRed:0 green:0 blue:0 alpha:1.0];

  155. 	} else if ([theme isEqualToString:@"blue"]) {

  156. 		backgroundColor = [UIColor colorWithRed:0.071 green:0.094 blue:0.106 alpha:1.0];

  157. 	} else if ([theme isEqualToString:@"sepia"]) {

  158. 		backgroundColor = [UIColor colorWithRed:0.310 green:0.302 blue:0.267 alpha:1.0]; 

  159. 	} else if ([theme isEqualToString:@"red"]) {

  160. 		backgroundColor = [UIColor colorWithRed:0.075 green:0.055 blue:0.051 alpha:1.0]; 

  161. 	} else if ([theme isEqualToString:@"deep"]) {

  162. 		backgroundColor = [UIColor colorWithRed:0.035 green:0.035 blue:0.043 alpha:1.0];

  163. 	} else if ([theme isEqualToString:@"mint"]) {

  164. 		backgroundColor = [UIColor colorWithRed:0.364 green:0.376 blue:0.357 alpha:1.0];

  165. 	} else if ([theme isEqualToString:@"cream"]) {

  166. 		backgroundColor = [UIColor colorWithRed:0.322 green:0.314 blue:0.286 alpha:1.0];

  167. 	} else if ([theme isEqualToString:@"acontrast"]) {

  168. 		backgroundColor = [UIColor colorWithRed:0.027 green:0.027 blue:0.24 alpha:1.0];

  169. 	} else if ([theme isEqualToString:@"pink"]) {

  170. 		backgroundColor = [UIColor colorWithRed:1.0 green:0.376 blue:0.357 alpha:1.0];

  171. 	} else if ([theme isEqualToString:@"solarize"]) {

  172. 		backgroundColor = [UIColor colorWithRed:0.040 green:0.067 blue:0.082 alpha:1.0];

  173. 	} else if (!theme) {

  174. 		backgroundColor = [UIColor colorWithRed:0.352 green:0.352 blue:0.352 alpha:1.0];

  175. 	} else {

  176. 		NSString *customThemeString = [[userDefaults stringForKey:[NSString stringWithFormat:@"Theme+%@", theme]] stringByRemovingPercentEncoding];

  177. 		if (customThemeString) {

  178. 			NSString *customFontColorHex = [customThemeString componentsSeparatedByString:@"#"][3];

  179. 			backgroundColor =  [UIColor colorWithHex:customFontColorHex]; 

  180. 		} else {

  181. 			backgroundColor = [UIColor colorWithRed:0.352 green:0.352 blue:0.352 alpha:1.0];

  182. 		}

  183. 	} 

  184. 	

  185. 	return backgroundColor;

  186. }

  187. 

  188. @end

  189. 

  190. 

  191. static UIButton *createUndeleteButton(){

  192. 	

  193. 	UIButton *undeleteButton = [UIButton buttonWithType:UIButtonTypeCustom];		

  194. 	UIImage *undeleteImage = [UIImage imageWithContentsOfFile:@"/var/mobile/Library/Application Support/TFDidThatSay/eye160white.png"];

  195. 	

  196. 	UIGraphicsBeginImageContextWithOptions(CGSizeMake(20, 20), NO, 0);

  197. 	[undeleteImage drawInRect:CGRectMake(0, 0, 20, 20)];

  198. 	undeleteImage = UIGraphicsGetImageFromCurrentImageContext();

  199. 	UIGraphicsEndImageContext();

  200. 	

  201. 	UIGraphicsBeginImageContextWithOptions(CGSizeMake(35, 35), NO, 0);

  202. 	CGContextRef context = UIGraphicsGetCurrentContext();

  203. 	UIGraphicsPushContext(context);

  204. 	[undeleteImage drawAtPoint:CGPointMake(7.5, 7.5)];

  205. 	UIGraphicsPopContext();

  206. 	undeleteImage = UIGraphicsGetImageFromCurrentImageContext();

  207. 	UIGraphicsEndImageContext();

  208. 

  209. 	[undeleteButton setImage:undeleteImage forState:UIControlStateNormal];

  210. 	undeleteButton.imageView.contentMode = UIViewContentModeScaleAspectFit;

  211. 	

  212. 	return undeleteButton;

  213. }

  214. 

  215. 

  216. %hook UIColor

  217. 

  218. %new

  219. +(UIColor *) colorWithHex:(NSString *) arg1 {

  220. 	if (!arg1){

  221. 		NSString *firstChar = [arg1 substringToIndex:1];

  222. 		if ([firstChar isEqualToString:@"#"]){

  223. 			arg1 = [arg1 substringWithRange:NSMakeRange(1, [arg1 length]-1)];

  224. 		}

  225. 		

  226. 		unsigned rgbValue = 0;

  227. 		NSScanner *scanner = [NSScanner scannerWithString:arg1];

  228. 		[scanner scanHexInt:&rgbValue];

  229. 			

  230. 		return [UIColor colorWithRed:((rgbValue & 0xFF0000) >> 16)/255.0 green:((rgbValue & 0xFF00) >> 8)/255.0 blue:(rgbValue & 0xFF)/255.0 alpha:0.87];

  231. 	} else {

  232. 		return nil;

  233. 	}

  234. }

  235. 

  236. %new 

  237. -(NSString *) hexString {

  238. 	const CGFloat *components = CGColorGetComponents(self.CGColor);

  239. 

  240.     CGFloat r = components[0];

  241.     CGFloat g = components[1];

  242.     CGFloat b = components[2];

  243. 

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

  245. }

  246. 

  247. %end

  248. 

  249. 

  250. %hook CommentDepthCell

  251. 

  252. -(void) doShortClick{

  253. 	%orig;

  254. 	

  255. 	[self addUndeleteButtonToMenu];

  256. }

  257. 

  258. -(void) doLongClick	{

  259. 	%orig;

  260. 	

  261. 	[self addUndeleteButtonToMenu];

  262. }

  263. 

  264. %new 

  265. -(void) addUndeleteButtonToMenu{

  266. 	

  267. 	NSString *commentBody = [MSHookIvar<id>(self, "comment") body];

  268. 

  269. 	if ([%c(TFHelper) shouldShowUndeleteButtonWithInfo:commentBody isDeletedOnly:isTFDeletedOnly]){

  270. 	

  271. 		id controller = MSHookIvar<id>(self, "parent");

  272. 		

  273. 		if (MSHookIvar<id>(controller, "menuCell")){

  274. 			

  275. 			UIStackView *menu = MSHookIvar<UIStackView *>(self, "menu");

  276. 			

  277. 			if (![[[[menu arrangedSubviews] lastObject] actionsForTarget:self forControlEvent:UIControlEventTouchUpInside] containsObject:@"handleUndeleteComment:"]){

  278. 				UIButton *undeleteButton = createUndeleteButton();

  279. 				[undeleteButton addTarget:self action:@selector(handleUndeleteComment:) forControlEvents:UIControlEventTouchUpInside];

  280. 			

  281. 				[menu addArrangedSubview:undeleteButton];

  282. 			}

  283. 		}

  284. 	}

  285. }

  286. 

  287. %new

  288. -(void) handleUndeleteComment:(id) sender{

  289. 	

  290. 	[sender setEnabled:NO];

  291. 

  292. 	id comment = MSHookIvar<id>(self, "comment");

  293. 	

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

  295. }

  296. 

  297. %new

  298. -(void) completeUndeleteCommentAction:(NSDictionary *) data{

  299. 	

  300. 	NSUserDefaults *userDefaults = [NSUserDefaults standardUserDefaults];

  301. 	id textStackDisplayView = MSHookIvar<id>(self, "commentBody");

  302. 	id comment = MSHookIvar<id>(self, "comment");

  303. 	

  304. 	NSString *author = data[@"author"];

  305. 	NSString *body = data[@"body"];

  306. 	

  307. 	//Attributed string generation rewrote from Slide_for_Reddit.TextDisplayStackView.createAttributedChunk(...) 

  308. 	

  309. 	UIFont *font = [%c(FontGenerator) fontOfSize:MSHookIvar<CGFloat>(textStackDisplayView, "fontSize") submission:NO willOffset:YES];

  310. 	

  311. 	NSString *themeName = [userDefaults stringForKey:@"theme"];

  312. 	UIColor *fontColor = [%c(ColorUtil) fontColorForTheme:themeName];

  313. 	UIColor *accentColor = [%c(ColorUtil) accentColorForSub:[comment subreddit]];

  314. 	

  315. 	NSString *html = [%c(MMMarkdown) HTMLStringWithMarkdown:body extensions:MMMarkdownExtensionsGitHubFlavored error:nil];

  316. 	html = [[html stringByReplacingOccurrencesOfString:@"<sup>" withString:@"<font size=\"1\">"] stringByReplacingOccurrencesOfString:@"</sup>" withString:@"</font>"];

  317. 	html = [[html stringByReplacingOccurrencesOfString:@"<del>" withString:@"<font color=\"green\">"] stringByReplacingOccurrencesOfString:@"</del>" withString:@"</font>"];

  318. 	html = [[html stringByReplacingOccurrencesOfString:@"<code>" withString:@"<font color=\"blue\">"] stringByReplacingOccurrencesOfString:@"</code>" withString:@"</font>"];

  319. 	html = [html stringByTrimmingCharactersInSet:[NSCharacterSet whitespaceAndNewlineCharacterSet]];

  320. 	

  321. 	DTHTMLAttributedStringBuilder *dthtmlBuilder = [[%c(DTHTMLAttributedStringBuilder) alloc] initWithHTML:[html dataUsingEncoding:NSUTF8StringEncoding] options:@{@"DTUseiOS6Attributes": @YES, @"DTDefaultTextColor": fontColor, @"DTDefaultFontSize": @([font pointSize])} documentAttributes:nil];

  322. 	

  323. 	NSMutableAttributedString *htmlAttributedString = [[NSMutableAttributedString alloc] initWithAttributedString:[dthtmlBuilder generatedAttributedString]];

  324. 	NSRange htmlStringRange = NSMakeRange(0, [htmlAttributedString length]);

  325. 	

  326. 	[[htmlAttributedString mutableString] replaceOccurrencesOfString:@"\t•\t" withString:@" • " options:0 range: htmlStringRange];

  327. 	[[htmlAttributedString mutableString] replaceOccurrencesOfString:@"\t◦\t" withString:@"  ◦ " options:0 range: htmlStringRange];

  328. 	[[htmlAttributedString mutableString] replaceOccurrencesOfString:@"\t▪\t" withString:@"   ▪ " options:0	 range: htmlStringRange];

  329. 	

  330. 	[htmlAttributedString removeAttribute:@"CTForegroundColorFromContext" range:htmlStringRange];

  331. 	

  332. 	[htmlAttributedString enumerateAttributesInRange:htmlStringRange options:NSAttributedStringEnumerationLongestEffectiveRangeNotRequired usingBlock:^(NSDictionary *attributes, NSRange range, BOOL *stop) {

  333. 		for (NSString *key in attributes){

  334. 			

  335. 			if ([(UIColor *) attributes[key] isKindOfClass:[UIColor class]]){

  336. 				UIColor *attrColor = (UIColor *) attributes[key];

  337. 				if ([[attrColor hexString] isEqualToString:@"#0000FF"]){

  338. 					UIFont *tempFont = [UIFont fontWithName:@"Courier" size:font.pointSize];

  339. 					

  340. 					[htmlAttributedString setAttributes:@{NSForegroundColorAttributeName: accentColor, NSBackgroundColorAttributeName: [%c(ColorUtil) backgroundColorForTheme:themeName], NSFontAttributeName: (tempFont ? tempFont : font)} range:range];

  341. 				} else if ([[attrColor hexString] isEqualToString:@"#008000"]) {

  342. 					[htmlAttributedString setAttributes:@{NSForegroundColorAttributeName: fontColor, NSFontAttributeName:font} range:range];

  343. 				}

  344. 			} else if ([(NSURL *) attributes[key] isKindOfClass:[NSURL class]]){

  345. 				NSURL *attrUrl = (NSURL *)attributes[key];

  346. 

  347. 				if (([userDefaults objectForKey:@"ENLARGE_LINKS"] == nil) ? YES : [userDefaults boolForKey:@"ENLARGE_LINKS"]){

  348. 					[htmlAttributedString addAttribute:NSFontAttributeName value:[%c(FontGenerator) boldFontOfSize:18 submission:NO willOffset:YES] range:range];

  349. 				}

  350. 				

  351. 				[htmlAttributedString addAttribute:NSForegroundColorAttributeName value:accentColor range:range]; 

  352. 				[htmlAttributedString addAttribute:NSUnderlineColorAttributeName value:[UIColor clearColor] range:range];

  353. 	

  354. 				//skipping showLinkContentType b/c not necessary and spoilers b/c MMMarkdown doesn't support them

  355. 

  356. 				[htmlAttributedString yy_setTextHighlightRange:range color: accentColor backgroundColor:nil userInfo:@{@"url": attrUrl}];

  357. 				break; 

  358. 			} 

  359. 		}

  360. 	}];

  361. 	

  362. 	[htmlAttributedString beginEditing];

  363. 	[htmlAttributedString enumerateAttribute:NSFontAttributeName inRange:NSMakeRange(0, [htmlAttributedString length]) options:0 usingBlock:^(id value, NSRange range, BOOL *stop){

  364. 		

  365. 		UIFont *attrFont = (UIFont *)value;

  366. 		

  367. 		BOOL isBold = (attrFont.fontDescriptor.symbolicTraits & UIFontDescriptorTraitBold) != 0;

  368. 		BOOL isItalic = (attrFont.fontDescriptor.symbolicTraits & UIFontDescriptorTraitItalic) != 0;

  369. 		

  370. 		UIFont *newFont = font;

  371. 		

  372. 		if (isBold){

  373. 			newFont = [%c(FontGenerator) boldFontOfSize:attrFont.pointSize submission:NO willOffset:NO];

  374. 		} else if (isItalic){

  375. 			newFont = [%c(FontGenerator) italicFontOfSize:attrFont.pointSize submission:NO willOffset:NO];

  376. 		}

  377. 	

  378. 		[htmlAttributedString removeAttribute:NSFontAttributeName range:range];

  379. 		[htmlAttributedString addAttribute:NSFontAttributeName value:newFont range:range];	

  380. 	

  381. 	}];

  382. 	[htmlAttributedString endEditing];

  383. 	

  384. 	NSMutableAttributedString *newCommentText = [MSHookIvar<NSMutableAttributedString *>(self, "cellContent") initWithAttributedString:htmlAttributedString];

  385. 	NSAttributedString *tempAttributedString = [[NSAttributedString alloc] initWithString:@""];

  386. 	[newCommentText appendAttributedString:tempAttributedString]; //to keep the compiler happy

  387. 	

  388. 	[comment setAuthor:author];

  389. 	[comment setBody:body];	

  390. 	

  391. 	id controller = MSHookIvar<id>(self, "parent");

  392. 	

  393. 	[self showMenu:nil];

  394. 	[MSHookIvar<id>(controller, "tableView") reloadData];

  395. 	

  396. 	[data[@"sender"] setEnabled:YES];

  397. }

  398. 

  399. %end

  400. 

  401. 

  402. %hook RSubmission

  403. 

  404. -(id) author{

  405. 	if (shouldSlideOverridePostInfo){

  406. 		return slidePostOverrideAuthor;

  407. 	} else {

  408. 		return %orig;

  409. 	}

  410. }

  411. 

  412. -(id) htmlBody{

  413. 	if (shouldSlideOverridePostInfo){

  414. 		return slidePostOverrideBodyHtml;

  415. 	}

  416. 	else {

  417. 		return %orig;

  418. 	}

  419. }

  420. 

  421. %end

  422. 

  423. 

  424. %hook CommentViewController

  425. %property(strong, nonatomic) UIButton *undeleteButton;

  426. 

  427. -(void) viewDidLoad {

  428. 	%orig;

  429. 	

  430. 	shouldSlideOverridePostInfo = NO;

  431. 	slidePostOverrideAuthor = nil;

  432. 	slidePostOverrideBodyHtml = nil;

  433. 			

  434. 	UIButton *undeleteButton = createUndeleteButton();

  435. 	[undeleteButton setImage:[[undeleteButton currentImage] imageWithRenderingMode:UIImageRenderingModeAlwaysTemplate] forState:UIControlStateNormal];

  436. 	[undeleteButton addTarget:self action:@selector(handleUndeletePost:) forControlEvents:UIControlEventTouchUpInside];

  437. 	[self setUndeleteButton:undeleteButton];

  438. }

  439. 

  440. -(void) viewDidLayoutSubviews{

  441. 	%orig;

  442. 	

  443. 	if ([self undeleteButton]){

  444. 		if ([self shouldAddUndeleteButtonToToolbar]){

  445. 			[self addUndeleteButtonToToolbar];

  446. 		}

  447. 	}

  448. }

  449. 

  450. -(void) loadAll:(id) arg1{

  451. 	%orig;

  452. 	

  453. 	if ([self undeleteButton]){

  454. 		if ([self shouldAddUndeleteButtonToToolbar]){

  455. 			[self addUndeleteButtonToToolbar];

  456. 		}

  457. 	}

  458. }

  459. 

  460. %new

  461. -(BOOL) shouldAddUndeleteButtonToToolbar{

  462. 	id post = MSHookIvar<id>(self, "submission");

  463. 	

  464. 	if ([post isSelf]){

  465. 		

  466. 		NSString *postBody = [post body];

  467. 		

  468. 		if ([%c(TFHelper) shouldShowUndeleteButtonWithInfo:postBody isDeletedOnly:isTFDeletedOnly]){

  469. 			return YES;

  470. 		}

  471. 	}

  472. 	

  473. 	return NO;

  474. }

  475. 

  476. %new

  477. -(void) addUndeleteButtonToToolbar{

  478. 	UIToolbar *toolbar = [[self navigationController] toolbar];

  479. 	NSMutableArray *toolbarItems = [[toolbar items] mutableCopy];

  480. 	UIView *firstView = [toolbarItems[0] customView];

  481. 	

  482. 	if (firstView){	

  483. 		UIColor *tintColor = [toolbar tintColor];

  484. 		

  485. 		UIButton *undeleteButton = [self undeleteButton];

  486. 		[undeleteButton setTintColor:tintColor];

  487. 		

  488. 		if ([firstView isMemberOfClass:[UIView class]]){

  489. 			if (![undeleteButton isDescendantOfView:firstView]){

  490. 				[firstView addSubview:undeleteButton];

  491. 				[undeleteButton setFrame:firstView.bounds];

  492. 			}

  493. 		} else if ([firstView isMemberOfClass:[UIButton class]] && undeleteButton != firstView){

  494. 			UIBarButtonItem *undeleteItem = [[UIBarButtonItem alloc] initWithCustomView:undeleteButton];

  495. 			

  496. 			[toolbarItems insertObject:toolbarItems[1] atIndex:0];

  497. 			[toolbarItems insertObject:undeleteItem atIndex:0];

  498. 			[toolbar setItems:toolbarItems animated:NO];

  499. 		}

  500. 	}

  501. }

  502. 

  503. %new

  504. -(void) handleUndeletePost:(id) sender{

  505. 	shouldSlideOverridePostInfo = YES;

  506. 	

  507. 	[sender setEnabled:NO];

  508. 	

  509. 	[%c(TFHelper) getUndeleteDataWithID:[MSHookIvar<id>(self, "submission") name] isComment:NO timeout:pushshiftRequestTimeoutValue extraData:@{@"sender" : sender} completionTarget:self completionSelector:@selector(completeUndeletePostAction:)];	

  510. }

  511. 

  512. %new

  513. -(void) completeUndeletePostAction:(NSDictionary *) data{

  514. 	

  515. 	slidePostOverrideAuthor = data[@"author"];

  516. 	slidePostOverrideBodyHtml = [%c(MMMarkdown) HTMLStringWithMarkdown:data[@"body"] extensions:MMMarkdownExtensionsGitHubFlavored error:nil];

  517. 	

  518. 	[self refresh:nil];

  519. 	

  520. 	[data[@"sender"] setEnabled:YES];

  521. }

  522. 

  523. %end

  524. 

  525. %end

  526. 

  527. 

  528. static void loadPrefs(){

  529. 	NSMutableDictionary *prefs = [[NSMutableDictionary alloc] initWithContentsOfFile:@"/User/Library/Preferences/com.lint.undelete.prefs.plist"];

  530. 	

  531. 	if (prefs){

  532. 		

  533. 		if ([prefs objectForKey:@"isSlideEnabled"] != nil){

  534. 			isSlideEnabled = [[prefs objectForKey:@"isSlideEnabled"] boolValue];

  535. 		} else {

  536. 			isSlideEnabled = YES;

  537. 		}

  538. 		

  539. 		if ([prefs objectForKey:@"isTFDeletedOnly"] != nil){

  540. 			isTFDeletedOnly = [[prefs objectForKey:@"isTFDeletedOnly"] boolValue];

  541. 		} else {

  542. 			isTFDeletedOnly = YES;

  543. 		}

  544. 		

  545. 		if ([prefs objectForKey:@"requestTimeoutValue"] != nil){

  546. 			pushshiftRequestTimeoutValue = [[prefs objectForKey:@"requestTimeoutValue"] doubleValue];

  547. 		} else {

  548. 			pushshiftRequestTimeoutValue = 10;

  549. 		}

  550. 		

  551. 	} else {

  552. 		isSlideEnabled = YES;

  553. 		isTFDeletedOnly = YES;

  554. 		pushshiftRequestTimeoutValue = 10;

  555. 	}	

  556. }

  557. 

  558. static void prefsChanged(CFNotificationCenterRef center, void *observer, CFStringRef name, const void *object, CFDictionaryRef userInfo) {

  559.   loadPrefs();

  560. }

  561. 

  562. 

  563. %ctor {

  564. 	loadPrefs();

  565. 	

  566. 	NSString* processName = [[NSProcessInfo processInfo] processName];

  567. 

  568. 	if ([processName isEqualToString:@"Slide for Reddit"]){

  569. 		if (isSlideEnabled){

  570. 			

  571. 			CFNotificationCenterAddObserver(CFNotificationCenterGetDarwinNotifyCenter(), NULL, prefsChanged, CFSTR("com.lint.undelete.prefs.changed"), NULL, CFNotificationSuspensionBehaviorDeliverImmediately);

  572. 			

  573. 			%init(Slide, CommentDepthCell = objc_getClass("Slide_for_Reddit.CommentDepthCell"), RSubmission = objc_getClass("Slide_for_Reddit.RSubmission"), CommentViewController = objc_getClass("Slide_for_Reddit.CommentViewController"));

  574. 		}

  575. 	}

  576. }