mirror of
https://github.com/Sunnyyoung/WeChatTweak-macOS.git
synced 2025-05-23 14:56:08 +08:00
57 lines
2.2 KiB
Objective-C
57 lines
2.2 KiB
Objective-C
//
|
|
// AutoAuth.m
|
|
// WeChatTweak
|
|
//
|
|
// Created by Sunny Young on 2021/5/8.
|
|
// Copyright © 2021 Sunnyyoung. All rights reserved.
|
|
//
|
|
|
|
#import "AutoAuth.h"
|
|
#import "WeChatTweakHeaders.h"
|
|
#import "WTConfigManager.h"
|
|
|
|
@implementation NSObject(AutoAuth)
|
|
|
|
static void __attribute__((constructor)) tweak(void) {
|
|
[objc_getClass("MMLoginOneClickViewController") jr_swizzleMethod:NSSelectorFromString(@"onLoginButtonClicked:") withMethod:@selector(tweak_onLoginButtonClicked:) error:nil];
|
|
[objc_getClass("MMMainViewController") jr_swizzleMethod:NSSelectorFromString(@"viewDidLoad") withMethod:@selector(tweak_viewDidLoad) error:nil];
|
|
[objc_getClass("LogoutCGI") jr_swizzleMethod:NSSelectorFromString(@"FFVCRecvDataAddDataToMsgChatMgrRecvZZ:") withMethod:@selector(tweak_FFVCRecvDataAddDataToMsgChatMgrRecvZZ:) error:nil];
|
|
[objc_getClass("AccountService") jr_swizzleMethod:NSSelectorFromString(@"FFAddSvrMsgImgVCZZ") withMethod:@selector(tweak_FFAddSvrMsgImgVCZZ) error:nil];
|
|
}
|
|
|
|
- (void)tweak_onLoginButtonClicked:(id)sender {
|
|
AccountService *accountService = [[objc_getClass("MMServiceCenter") defaultCenter] getService:objc_getClass("AccountService")];
|
|
BOOL enabledAutoAuth = [NSUserDefaults.standardUserDefaults boolForKey:WeChatTweakPreferenceAutoAuthKey];
|
|
BOOL canAutoAuth = accountService.canAutoAuth;
|
|
if (enabledAutoAuth && canAutoAuth) {
|
|
[accountService AutoAuth];
|
|
} else {
|
|
[self tweak_onLoginButtonClicked:sender];
|
|
}
|
|
}
|
|
|
|
- (void)tweak_viewDidLoad {
|
|
[self tweak_viewDidLoad];
|
|
if ([NSUserDefaults.standardUserDefaults boolForKey:WeChatTweakPreferenceAutoAuthKey]) {
|
|
MMSessionMgr *mgr = [[objc_getClass("MMServiceCenter") defaultCenter] getService:objc_getClass("MMSessionMgr")];
|
|
[mgr loadSessionData];
|
|
[mgr loadBrandSessionData];
|
|
}
|
|
}
|
|
|
|
- (void)tweak_FFVCRecvDataAddDataToMsgChatMgrRecvZZ:(id)arg {
|
|
if (![NSUserDefaults.standardUserDefaults boolForKey:WeChatTweakPreferenceAutoAuthKey]) {
|
|
[self tweak_FFVCRecvDataAddDataToMsgChatMgrRecvZZ:arg];
|
|
}
|
|
}
|
|
|
|
- (void)tweak_FFAddSvrMsgImgVCZZ {
|
|
if ([NSUserDefaults.standardUserDefaults boolForKey:WeChatTweakPreferenceAutoAuthKey]) {
|
|
return;
|
|
} else {
|
|
[self tweak_FFAddSvrMsgImgVCZZ];
|
|
}
|
|
}
|
|
|
|
@end
|