WKWebView 的 cookie 问题

￡神魔★判官ぃ 2023-03-13 13:17 2阅读 0赞

# WKWebView 的 cookie 问题 #

场景一：原生登录之后立马跳转到 H5 页面

原因分析：NSHTTPCookieStorage 中的 cookie 不会立马（主动？）同步到 WKWebView，会导致进入 H5 页面后 cookie 没带上。

解决方案：初始化 WKWebView 的时候将 NSHTTPCookieStorage 中的 cookie 注入其中。

示例代码：

- (void)setupWebView {
        WKUserContentController *userContentController = [WKUserContentController new];
        NSString *cookieJS = [self updateCookieJSString];
        WKUserScript *cookieScript = [[WKUserScript alloc] initWithSource:cookieJS injectionTime:WKUserScriptInjectionTimeAtDocumentStart forMainFrameOnly:NO];
        [userContentController addUserScript:cookieScript];
        WKWebViewConfiguration *configuration = [WKWebViewConfiguration new];
        configuration.processPool = [WKProcessPool new];
        configuration.userContentController = userContentController;
        self.webView = [[WKWebView alloc] initWithFrame:CGRectZero configuration:configuration];
    }
    
    - (NSString *)updateCookieJSString {
        NSMutableString *script = [NSMutableString string];
        [script appendString:@"var cookieNames = document.cookie.split('; ').map(function(cookie) { return cookie.split('=')[0] } );\n"];
        NSArray *cookies = [[NSHTTPCookieStorage sharedHTTPCookieStorage] cookies];
        for (NSHTTPCookie *cookie in cookies) {
            if ([cookie.value rangeOfString:@"'"].location != NSNotFound) {
                continue;
            }
            [script appendFormat:@"if (cookieNames.indexOf('%@') == -1) { document.cookie='%@'; };\n", cookie.name, [self cookieString:cookie]];
        }
        return script;
    }
    
    - (NSString *)cookieString:(NSHTTPCookie *)cookie {
        NSString *string = [NSString stringWithFormat:@"%@=%@;domain=%@;path=%@",
                            cookie.name,
                            cookie.value,
                            cookie.domain,
                            cookie.path ?: @"/"];
        if (cookie.secure) {
            string = [string stringByAppendingString:@";secure=true"];
        }
        return string;
    }

### 场景二：在 H5 页面内调起原生登录，登录成功后 webView 带不上 cookie ###

原因分析：同场景一。

解决方案：在登录成功的回调方法里面，通过原生调 JS 的方式运行一段 JS 代码，把 NSHTTPCookieStorage 中的 cookie 通过 “document.cookie = ” 的方式注入到 webView 中。但是有一个弊端：运行完这段 JS 代码后 cookie 可能不会立即生效，大概会延迟一秒左右。

示例代码：

- (void)loginSuccess {
        [webView evaluateJavaScript:[self updateCookieJSString] completionHandler:^(id _Nullable result, NSError * _Nullable error) {
            dispatch_after(dispatch_time(DISPATCH_TIME_NOW, 1 * NSEC_PER_SEC), dispatch_get_main_queue(), ^{
                // 需要在登录成功后立马执行的操作。。。
            }];
        }];
    }