open-source/WeChatTweak-macOS
1
0
Fork 0
mirror of https://github.com/Sunnyyoung/WeChatTweak-macOS.git synced 2026-07-29 06:24:31 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fixed potential crash on exit

Browse Source
This commit is contained in:
Sunnyyoung
2017-12-10 17:20:10 +08:00
parent 701b8a3ad0
commit b41fb3910d
4 changed files with 20 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files
WeChatTweak/WeChatTweak.m
+6 -6
View File
@@ -46,7 +46,7 @@ static void __attribute__((constructor)) tweak(void) {
// Method Swizzling
class_addMethod(objc_getClass("AppDelegate"), @selector(applicationDockMenu:), method_getImplementation(class_getInstanceMethod(objc_getClass("AppDelegate"), @selector(tweak_applicationDockMenu:))), "@:@");
[objc_getClass("AppDelegate") jr_swizzleMethod:NSSelectorFromString(@"applicationDidFinishLaunching:") withMethod:@selector(tweak_applicationDidFinishLaunching:) error:nil];
[objc_getClass("AppDelegate") jr_swizzleMethod:NSSelectorFromString(@"applicationShouldTerminate:") withMethod:@selector(tweak_applicationShouldTerminate:) error:nil];
[objc_getClass("LogoutCGI") jr_swizzleMethod:NSSelectorFromString(@"sendLogoutCGIWithCompletion:") withMethod:@selector(tweak_sendLogoutCGIWithCompletion:) error:nil];
[objc_getClass("MessageService") jr_swizzleMethod:NSSelectorFromString(@"onRevokeMsg:") withMethod:@selector(tweak_onRevokeMsg:) error:nil];
[objc_getClass("CUtility") jr_swizzleClassMethod:NSSelectorFromString(@"HasWechatInstance") withClassMethod:@selector(tweak_HasWechatInstance) error:nil];
[objc_getClass("MASPreferencesWindowController") jr_swizzleMethod:NSSelectorFromString(@"initWithViewControllers:") withMethod:@selector(tweak_initWithViewControllers:) error:nil];
@@ -153,13 +153,13 @@ static void __attribute__((constructor)) tweak(void) {
[[AlfredManager sharedInstance] startListener];
}
- (NSApplicationTerminateReply)tweak_applicationShouldTerminate:(NSApplication *)sender {
- (void)tweak_sendLogoutCGIWithCompletion:(id)completion {
BOOL enabledAutoAuth = [[NSUserDefaults standardUserDefaults] boolForKey:WeChatTweakPreferenceAutoAuthKey];
if (enabledAutoAuth) {
return NSTerminateNow;
} else {
return [self tweak_applicationShouldTerminate:sender];
WeChat *wechat = [objc_getClass("WeChat") sharedInstance];
if (enabledAutoAuth && wechat.isAppTerminating) {
return;
}
[self tweak_sendLogoutCGIWithCompletion:completion];
}
#pragma mark - Preferences Window
WeChatTweak/WeChatTweakHeaders.h
+9 -1
View File
@@ -11,6 +11,8 @@
@interface WeChat : NSObject
@property(nonatomic) BOOL isAppTerminating;
+ (instancetype)sharedInstance;
- (void)lock:(id)block;
- (void)showMainWindow;
@@ -92,6 +94,12 @@
@end
@interface LogoutCGI: NSObject
- (void)sendLogoutCGIWithCompletion:(id)arg1;
@end
@protocol MASPreferencesViewController <NSObject>
@property(readonly, nonatomic) NSString *toolbarItemLabel;
@@ -104,7 +112,7 @@
@end
@interface MASPreferencesWindowController : NSWindowController
@interface MASPreferencesWindowController: NSWindowController
- (id)initWithViewControllers:(NSArray *)arg1;