Explorar el Código

Add Beam support

master
lint hace 4 años
padre
commit
20fbe88bc7
Se han modificado 3 ficheros con 347 adiciones y 0 borrados
  1. +14
    -0
      prefs/Resources/Root.plist
  2. +56
    -0
      tweak/Beam.h
  3. +277
    -0
      tweak/Beam.xm

+ 14
- 0
prefs/Resources/Root.plist Ver fichero

@@ -110,6 +110,20 @@
<key>label</key>
<string>Enable Antenna</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>isBeamEnabled</string>
<key>label</key>
<string>Enable Beam</string>
</dict>
<dict>
<key>cell</key>
<string>PSGroupCell</string>

+ 56
- 0
tweak/Beam.h Ver fichero

@@ -0,0 +1,56 @@
/* -- Comment Interfaces -- */
@interface Comment
@property(strong, nonatomic) NSString *author;
@property(strong, nonatomic) NSString *content;
@property(strong, nonatomic) NSString *identifier;
@property(strong, nonatomic) id markdownString;
@end
@interface CommentCell : UIView
@property(strong, nonatomic) id comment;
@property(strong, nonatomic) id commentContentView;
@property(strong, nonatomic) id authorButton;
@property(assign, nonatomic) BOOL isCollapsed;
@property(assign, nonatomic) BOOL commentDidChange;
-(void) reloadContents;
//custom elements
@property(strong, nonatomic) UIButton *undeleteButton;
@end
/* -- Post Interfaces -- */
@interface Post
@property(strong, nonatomic) NSString *author;
@property(strong, nonatomic) NSString *content;
@property(strong, nonatomic) NSString *identifier;
@property(strong, nonatomic) id markdownString;
@property(strong, nonatomic) NSNumber *isSelfText;
@end
@interface PostDetailEmbeddedViewController
@property(strong, nonatomic) UITableView *tableView;
@property(strong, nonatomic) NSArray *content;
//custom elements
@property(strong, nonatomic) id selfTextView;
@property(strong, nonatomic) id metadataView;
@property(strong, nonatomic) id post;
@end
@interface PostToolbarView : UIView
@property(strong, nonatomic) id post;
//custom elements
@property(strong, nonatomic) UIButton *undeleteButton;
@end
@interface PostSelfTextPartCell
-(id) _viewControllerForAncestor;
@end
@interface PostMetadataView
@property(strong, nonatomic) id post;
@end

+ 277
- 0
tweak/Beam.xm Ver fichero

@@ -0,0 +1,277 @@
#import "Beam.h"
static BOOL isBeamEnabled;
static BOOL isTFDeletedOnly;
static CGFloat pushshiftRequestTimeoutValue;
%group Beam
%hook CommentCell
%property(strong, nonatomic) UIButton *undeleteButton;
-(void) layoutSubviews{
%orig;
UIButton *undeleteButton = [self undeleteButton];
if (undeleteButton){
if ([self isCollapsed]){
[undeleteButton setHidden:YES];
} else {
[undeleteButton setHidden:NO];
}
} else {
NSString *commentBody = [[self comment] content];
HBLogDebug(@"body: %@", commentBody);
if ((isTFDeletedOnly && ([commentBody isEqualToString:@"[deleted]"] || [commentBody isEqualToString:@"[removed]"])) || !isTFDeletedOnly){
CGFloat authorTextHeight = [[self authorButton] frame].size.height;
undeleteButton = [UIButton buttonWithType:UIButtonTypeCustom];
[undeleteButton addTarget:self action:@selector(handleUndeleteCommentAction:) forControlEvents:UIControlEventTouchUpInside];
[undeleteButton setImage:[UIImage imageWithContentsOfFile:@"/var/mobile/Library/Application Support/TFDidThatSay/eye160dark.png"] forState:UIControlStateNormal];
undeleteButton.imageView.contentMode = UIViewContentModeScaleAspectFit;
undeleteButton.imageEdgeInsets = UIEdgeInsetsMake(5, 5, 5, 5);
undeleteButton.frame = CGRectMake([[UIScreen mainScreen] bounds].size.width - authorTextHeight - 5, 5, authorTextHeight, authorTextHeight);
[[self commentContentView] addSubview:undeleteButton];
[self setUndeleteButton:undeleteButton];
}
}
}
%new
-(void) handleUndeleteCommentAction:(id) sender{
[sender setEnabled:NO];
id comment = [self 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 identifier]]]];
[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 setContent:body];
[comment setMarkdownString:nil];
[self setCommentDidChange:YES];
[self performSelectorOnMainThread:@selector(reloadContents) withObject:nil waitUntilDone:YES];
[[MSHookIvar<id>(self, "delegate") tableView] performSelectorOnMainThread:@selector(reloadData) withObject:nil waitUntilDone:NO];
[sender setEnabled:YES];
}];
}
%end
%hook PostSelfTextPartCell
-(void) setSelected:(BOOL) arg1 animated:(BOOL) arg2{
%orig;
id postViewController = [self _viewControllerForAncestor];
if ([postViewController isMemberOfClass:objc_getClass("beam.PostDetailEmbeddedViewController")]){
[postViewController setSelfTextView:self];
[postViewController setPost:[self post]];
}
}
%end
%hook PostMetadataView
-(void) layoutSubviews{
%orig;
id postViewController = MSHookIvar<id>(self, "delegate");
if ([postViewController isMemberOfClass:objc_getClass("beam.PostDetailEmbeddedViewController")]){
[postViewController setMetadataView:self];
}
}
%end
%hook PostToolbarView
%property(strong, nonatomic) UIButton *undeleteButton;
-(void) layoutSubviews{
%orig;
if (![self undeleteButton] && [[[self post] isSelfText] boolValue] && [MSHookIvar<id>(self, "delegate") isMemberOfClass:objc_getClass("beam.PostDetailEmbeddedViewController")]){
NSString *postBody = [[self post] content];
if ((isTFDeletedOnly && ([postBody isEqualToString:@"[deleted]"] || [postBody isEqualToString:@"[removed]"])) || !isTFDeletedOnly){
id moreButton = MSHookIvar<id>(self, "moreButton");
CGFloat buttonHeight = [moreButton frame].size.height;
UIButton *undeleteButton = [UIButton buttonWithType:UIButtonTypeCustom];
[undeleteButton addTarget:MSHookIvar<id>(self, "delegate") action:@selector(handleUndeletePostAction:) forControlEvents:UIControlEventTouchUpInside];
[undeleteButton setImage:[UIImage imageWithContentsOfFile:@"/var/mobile/Library/Application Support/TFDidThatSay/eye160dark.png"] forState:UIControlStateNormal];
undeleteButton.imageView.contentMode = UIViewContentModeScaleAspectFit;
undeleteButton.imageEdgeInsets = UIEdgeInsetsMake(5, 5, 5, 5);
undeleteButton.frame = CGRectMake([moreButton frame].origin.x - buttonHeight, 1, buttonHeight, buttonHeight);
[self addSubview:undeleteButton];
[self setUndeleteButton:undeleteButton];
}
}
}
%end
%hook PostDetailEmbeddedViewController
%property(strong, nonatomic) id selfTextView;
%property(strong, nonatomic) id metadataView;
%property(strong, nonatomic) id post;
%new
-(void) handleUndeletePostAction:(id) sender{
[sender setEnabled:NO];
id post = [self post];
if (post){
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 identifier]]]];
[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]];
}
[post setAuthor:author];
[post setContent:body];
[post setMarkdownString:nil];
if ([self selfTextView]){
[[self selfTextView] performSelectorOnMainThread:@selector(reloadContents) withObject:nil waitUntilDone:YES];
}
if ([self metadataView]){
[[self metadataView] performSelectorOnMainThread:@selector(setPost:) withObject:post waitUntilDone:YES];
}
[[self tableView] performSelectorOnMainThread:@selector(reloadData) withObject:post waitUntilDone:NO];
[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:@"isBeamEnabled"] != nil){
isBeamEnabled = [[prefs objectForKey:@"isBeamEnabled"] boolValue];
} else {
isBeamEnabled = 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 {
isBeamEnabled = 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:@"beam"]){
if (isBeamEnabled){
CFNotificationCenterAddObserver(CFNotificationCenterGetDarwinNotifyCenter(), NULL, prefsChanged, CFSTR("com.lint.undelete.prefs.changed"), NULL, CFNotificationSuspensionBehaviorDeliverImmediately);
%init(Beam, CommentCell = objc_getClass("beam.CommentCell"), PostDetailEmbeddedViewController = objc_getClass("beam.PostDetailEmbeddedViewController"), PostToolbarView = objc_getClass("beam.PostToolbarView"), PostSelfTextPartCell = objc_getClass("beam.PostSelfTextPartCell"), PostMetadataView = objc_getClass("beam.PostMetadataView"));
}
}
}

Cargando…
Cancelar
Guardar