項目需要對`WKWebView`進行Cookie同步，在網(wǎng)上找了好多方法绝淡，經(jīng)過測試：
第一種（無效）：通過`evaluateJavaScript`注入叹谁，這種方法是注入不了Cookie的混坞，因為js沒有權(quán)限
第二種（無效）：通過請求頭的方法超营，`[request setValue:@"cookie string" forHTTPHeaderField:@"Cookie"]`庆揩，這種方法js不能通過document.cookie獲取到cookie
第二種：通過`addUserScript`方法镊叁，如果你的WKWebView已經(jīng)登錄了唾琼，只需要在`WKWebView`的初始化的時候注入一次就好了，但是如果你的`WKWebView`是要在頁面上動態(tài)同步`Cookie`的話疯趟，還需要其它操作拘哨，下面是我的解決方案。
初始化WKWebView注入Cookie
- (instancetype)initWithFrame:(CGRect)frame configuration:(nonnull WKWebViewConfiguration *)configuration
{
//一定要在[super initWithFrame:frame configuration:configuration]之前設(shè)置迅办，否則無效
WKUserContentController *userContentController = WKUserContentController.new;
WKUserScript *cookieScript = [[WKUserScript alloc] initWithSource:[self updateCookies] injectionTime:WKUserScriptInjectionTimeAtDocumentStart forMainFrameOnly:NO];
[userContentController addUserScript:cookieScript];
configuration.userContentController = userContentController;
self = [super initWithFrame:frame configuration:configuration];
if (self) {
self.scrollView.bounces = NO;
self.scrollView.showsHorizontalScrollIndicator = NO;
[self sizeToFit];
self.backgroundColor = HSQColorGray250;
self.allowsBackForwardNavigationGestures = YES;
[self addSubview:self.progressView];
[self addObserver:self forKeyPath:@"estimatedProgress" options:NSKeyValueObservingOptionNew | NSKeyValueObservingOptionInitial context:nil];
}
return self;
}
這里拼接是Cookie
- (NSString *)updateCookies
{
NSMutableString *cookieValue = [NSMutableString stringWithFormat:@""];
NSHTTPCookieStorage *cookieJar = [NSHTTPCookieStorage sharedHTTPCookieStorage];
//這邊是在登錄接口保存了登錄Cookie
NSString *loginCookie = [[NSUserDefaults standardUserDefaults] objectForKey:kDMUserLoginCookie];
NSMutableString *mHsqCookie = [[NSMutableString alloc] init];
NSMutableString *mDwdCookie = [[NSMutableString alloc] init];
for (NSHTTPCookie *cookie in cookieJar.cookies) {
//這邊是做域名攔截
if ([cookie.domain containsString:@".xxxx.net"]) {
NSString *hsqCookie = [NSString stringWithFormat:@"document.cookie='%@=%@;%@=%@';", cookie.name, cookie.value, @"domain", cookie.domain];
//用來處理垮域問題
NSString *dwdCookie = [NSString stringWithFormat:@"document.cookie='%@=%@;%@=%@';", cookie.name, cookie.value, @"domain", @"yyyy.net"];
[mHsqCookie appendString:hsqCookie];
[mDwdCookie appendString:dwdCookie];
}
}
//拼接登錄Cookie
if (!IsNilOrNull(loginCookie) && ![cookieValue containsString:HsqSidCookie]) {
[mHsqCookie appendString:[NSString stringWithFormat:@"document.cookie='%@domain=.xxxx.net';", loginCookie]];
[mDwdCookie appendString:[NSString stringWithFormat:@"document.cookie='%@domain=yyyy.net';", loginCookie]];
}
[cookieValue appendString:mHsqCookie];
[cookieValue appendString:mDwdCookie];
return cookieValue;
}
接口請求拿到登錄Cookie保存在本地
// 保存登錄cookie
- (void)saveLoginCookie:(NSURLSessionDataTask *)task
{
if ([task.response isKindOfClass:[NSHTTPURLResponse class]]) {
NSHTTPURLResponse *response = (NSHTTPURLResponse *)task.response;
NSDictionary *allFields = response.allHeaderFields;
NSArray<NSHTTPCookie*> *cookies = [NSHTTPCookie cookiesWithResponseHeaderFields:allFields forURL:[NSURL URLWithString:[NSString stringWithFormat:@"api地址"]]];
[cookies enumerateObjectsUsingBlock:^(NSHTTPCookie * _Nonnull obj, NSUInteger idx, BOOL * _Nonnull stop) {
if ([obj.domain containsString:@".xxxx.net"] && [obj.name isEqualToString:HsqSidCookie] && ![obj.value isEqualToString:@"deleted"]) {
NSString *hsqCookie = [NSString stringWithFormat:@"%@=%@;", obj.name, obj.value];
[[NSUserDefaults standardUserDefaults] setObject:hsqCookie forKey:kDMUserLoginCookie];
[[NSUserDefaults standardUserDefaults] synchronize];
*stop = YES;
}
}];
}
}
動態(tài)同步cookie
- (WKNavigation *)reload
{
[self.configuration.userContentController removeAllUserScripts];
WKUserScript * cookieScript = [[WKUserScript alloc] initWithSource:[self updateCookies] injectionTime:WKUserScriptInjectionTimeAtDocumentStart forMainFrameOnly:NO];
[self.configuration.userContentController addUserScript:cookieScript];
return [super reload];
}
