iOs迁至WKWebView跨过的一些坑

wkwebviewconfiguration *configuration = [[wkwebviewconfiguration alloc] init];
nsstring *jscript = @"var meta = document.createelement('meta'); meta.setattribute('name', 'viewport'); meta.setattribute('content', 'width=device-width'); document.getelementsbytagname('head')[0].appendchild(meta);";
wkuserscript *wkuscript = [[wkuserscript alloc] initwithsource:jscript injectiontime:wkuserscriptinjectiontimeatdocumentend formainframeonly:yes];
wkusercontentcontroller *wkucontroller = [[wkusercontentcontroller alloc] init];
[wkucontroller adduserscript:wkuscript];
configuration.usercontentcontroller = wkucontroller;

//jscontext就为其提供着运行环境 h5上下文
- (void)webviewdidfinishload:(uiwebview *)webview{
 jscontext *jscontext = [self.webview valueforkeypath:@"documentview.webview.mainframe.javascriptcontext"];
 self.jscontext = jscontext;
}

// 执行脚本增加js全局变量
[self.jscontext evaluatescript:@"var arr = [3, '3', 'abc'];"];

// ⚠️添加js方法，需要注意的是添加的方法会覆盖原有的js方法，因为我们是在网页加载成功后获取上下文来操作的。
// 无参数的
self.jscontext[@"alertmessage"] = ^() {
 nslog(@"js端调用alertmessage时就会跑到这里来！");
};
// 带参数的，值必须进行转换
 self.jscontext[@"showdict"] = ^(jsvalue *value) {
 nsarray *args = [jscontext currentarguments];
 jsvalue *dictvalue = args[0];
 nsdictionary *dict = dictvalue.todictionary;
 nslog(@"%@",dict);
 };

// 获取js中的arr数据
jsvalue *arrvalue = self.jscontext[@"arr"];

// 异常捕获
self.jscontext.exceptionhandler = ^(jscontext *context, jsvalue *exception) {
 weakself.jscontext.exception = exception;
 nslog(@"exception == %@",exception);
};

// 给js中的对象重新赋值
omjsobject *omobject = [[omjsobject alloc] init];
self.jscontext[@"omobject"] = omobject;
nslog(@"omobject == %d",[omobject getsum:20 num2:40]);

// 我们都知道object必须要遵守jsexport协议时，js可以直接调用object中的方法，并且需要把函数名取个别名。在js端可以调用gets，oc可以继续使用这个getsum这个方法
@protocol omprotocol <jsexport>
// 协议 - 协议方法 
jsexportas(gets, -(int)getsum:(int)num1 num2:(int)num2);
@end

// 直接调用js
nsstring *jsstr = @"var arr = [3, '3', 'abc']; ";
[self.webview evaluatejavascript:jsstr completionhandler:^(id _nullable result, nserror * _nullable error) {
 nslog(@"%@----%@",result, error);
}];

// 下面是注册名称后，js使用messagehandlers调用了指定名称就会进入到代理中

// oc我们添加了js名称后
- (void)viewdidload{
 //...
 [wkucontroller addscriptmessagehandler:self name:@"showtime"];
 configuration.usercontentcontroller = wkucontroller;
}

// js中messagehandlers调用我们在oc中的名称一致时就会进入后面的到oc的代理
window.webkit.messagehandlers.showtime.postmessage('');

// 代理，判断逻辑
- (void)usercontentcontroller:(wkusercontentcontroller *)usercontentcontroller didreceivescriptmessage:(wkscriptmessage *)message{
 if ([message.name isequaltostring:@"showtime"]) {
  nslog(@"来了！");
 }
 nslog(@"message == %@ --- %@",message.name,message.body); 
}

// 最后在dealloc必须移除
[self.usercontentcontroller removescriptmessagehandlerforname:@"showtime"];

//如果是弹窗的必须自己实现代理方法
- (void)webview:(wkwebview *)webview runjavascriptalertpanelwithmessage:(nsstring *)message initiatedbyframe:(wkframeinfo *)frame completionhandler:(void (^)(void))completionhandler
{
 uialertcontroller *alert = [uialertcontroller alertcontrollerwithtitle:@"提醒" message:message preferredstyle:uialertcontrollerstylealert];
 [alert addaction:[uialertaction actionwithtitle:@"知道了" style:uialertactionstylecancel handler:^(uialertaction * _nonnull action) {
  completionhandler();
 }]];
 
 [self presentviewcontroller:alert animated:yes completion:nil];
}

// 使用
self.wjb = [webviewjavascriptbridge bridgeforwebview:self.webview];
// 如果你要在vc中实现 uiwebview的代理方法 就实现下面的代码(否则省略)
[self.wjb setwebviewdelegate:self];

// 注册js方法名称
[self.wjb registerhandler:@"jscallsoc" handler:^(id data, wvjbresponsecallback responsecallback) {
 nslog(@"currentthread == %@",[nsthread currentthread]);
 nslog(@"data == %@ -- %@",data,responsecallback);
}];

// 调用js
dispatch_async(dispatch_get_global_queue(0, 0), ^{
  [self.wjb callhandler:@"occalljsfunction" data:@"oc调用js" responsecallback:^(id responsedata) {
   nslog(@"currentthread == %@",[nsthread currentthread]);
   nslog(@"调用完js后的回调：%@",responsedata);
  }];
});

function setupwebviewjavascriptbridge(callback) {
	if (window.webviewjavascriptbridge) { return callback(webviewjavascriptbridge); }
	if (window.wvjbcallbacks) { return window.wvjbcallbacks.push(callback); }
	window.wvjbcallbacks = [callback];
	var wvjbiframe = document.createelement('iframe');
	wvjbiframe.style.display = 'none';
	wvjbiframe.src = 'https://__bridge_loaded__';
	document.documentelement.appendchild(wvjbiframe);
	settimeout(function() { document.documentelement.removechild(wvjbiframe) }, 0)
}

setupwebviewjavascriptbridge(function(bridge) {
	
	/* initialize your app here */

	bridge.registerhandler('js echo', function(data, responsecallback) {
		console.log("js echo called with:", data)
		responsecallback(data)
	})
	bridge.callhandler('objc echo', {'key':'value'}, function responsecallback(responsedata) {
		console.log("js received response:", responsedata)
	})
})

// 子类
@interface omurlprotocol : nsurlprotocol<nsurlsessiondatadelegate>
@property (nonatomic, strong) nsurlsession *session;
@end

// 注册
[nsurlprotocol registerclass:[omurlprotocol class]];

// 1. 首先会在这里来进行拦截，返回yes则表示需要经过我们自定义处理，no则走系统处理
+ (bool)caninitwithrequest:(nsurlrequest *)request;

// 2.拦截处理将会进入下一个环节, 返回一个标准化的request，可以在这里进行重定向
+ (nsurlrequest *)canonicalrequestforrequest:(nsurlrequest *)request;

// 3.是否成功拦截都会走这个方法, 可以在这里进行一些自定义处理
- (void)startloading;

// 4. 任何网络请求都会走上面的拦截处理，即使我们重定向后还会再走一次或多次流程，需要标记来处理
// 根据request获取标记值来决定是否需要拦截，在caninitwithrequest内处理
+ (nullable id)propertyforkey:(nsstring *)key inrequest:(nsurlrequest *)request;
// 标记
+ (void)setproperty:(id)value forkey:(nsstring *)key inrequest:(nsmutableurlrequest *)request;
// 移除标记
+ (void)removepropertyforkey:(nsstring *)key inrequest:(nsmutableurlrequest *)request;

po session.configuration.protocolclasses
<__nsarrayi 0x600001442d00>(
_nsurlhttpprotocol,
_nsurldataprotocol,
_nsurlftpprotocol,
_nsurlfileprotocol,
nsabouturlprotocol
)

// 自定义返回我们的协议类
- (nsarray *)protocolclasses {
 nsarray *originalprotocols = [omurlprotocol readoriginalprotocols];
 nsmutablearray *newprotocols = [nsmutablearray arraywitharray:originalprotocols];
 [newprotocols addobject:[omurlprotocol class]];
 return newprotocols;
}

// 我们再次打印时发现已经加上我们自定义的协议类了
po session.configuration.protocolclasses
<__nsarraym 0x60000041a4f0>(
_nsurlhttpprotocol,
_nsurldataprotocol,
_nsurlftpprotocol,
_nsurlfileprotocol,
nsabouturlprotocol,
omurlprotocol
)

// 存储系统原有的协议类
+ (void)saveoriginalprotocols: (nsarray<class> *)protocols{
 nsmutablearray *protocolnamearray = [nsmutablearray array];
 for (class protocol in protocols){
  [protocolnamearray addobject:nsstringfromclass(protocol)];
 }
 nslog(@"协议数组为： %@", protocolnamearray);
 [[nsuserdefaults standarduserdefaults] setobject:protocolnamearray forkey:originalprotocolskey];
 [[nsuserdefaults standarduserdefaults] synchronize];
}

// 获取系统原有的协议类
+ (nsarray<class> *)readoriginalprotocols{
 nsarray *classnames = [[nsuserdefaults standarduserdefaults] valueforkey:originalprotocolskey];
 nsmutablearray *origianlprotocols = [nsmutablearray array];
 for (nsstring *name in classnames){
  class class = nsclassfromstring(name);
  [origianlprotocols addobject: class];
 }
 return origianlprotocols;
}

+ (void)hooknsurlsessionconfiguration{
 nsarray *originalprotocols = [nsurlsession sharedsession].configuration.protocolclasses;
 [self saveoriginalprotocols:originalprotocols];
 class cls = nsclassfromstring(@"__nscfurlsessionconfiguration") ?: nsclassfromstring(@"nsurlsessionconfiguration");
 method originalmethod = class_getinstancemethod(cls, @selector(protocolclasses));
 method stubmethod = class_getinstancemethod([self class], @selector(protocolclasses));
 if (!originalmethod || !stubmethod) {
  [nsexception raise:nsinternalinconsistencyexception format:@"没有这个方法 无法交换"];
 }
 method_exchangeimplementations(originalmethod, stubmethod);
}

// 使用脚本来添加cookie

// 获取去cookie数据
- (nsstring *)cookiestring
{
 nsmutablestring *script = [nsmutablestring string];
 [script appendstring:@"var cookienames = document.cookie.split('; ').map(function(cookie) { return cookie.split('=')[0] } );\n"];
 for (nshttpcookie *cookie in [[nshttpcookiestorage sharedhttpcookiestorage] cookies]) {

  if ([cookie.value rangeofstring:@"'"].location != nsnotfound) {
   continue;
  }
  [script appendformat:@"if (cookienames.indexof('%@') == -1) { document.cookie='%@'; };\n", cookie.name, cookie.kc_formatcookiestring];
 }
 return script;
}

// 添加cookie
wkuserscript * cookiescript = [[wkuserscript alloc] initwithsource:[self cookiestring] injectiontime:wkuserscriptinjectiontimeatdocumentstart formainframeonly:no];
[[[wkusercontentcontroller alloc] init] adduserscript: cookiescript];

// 添加一个分类来修复cookie丢失的问题
@interface nsurlrequest (cookie)

- (nsurlrequest *)fixcookie;

@end

@implementation nsurlrequest (cookie)

- (nsurlrequest *)fixcookie{
 nsmutableurlrequest *fixedrequest;
 if ([self iskindofclass:[nsmutableurlrequest class]]) {
  fixedrequest = (nsmutableurlrequest *)self;
 } else {
  fixedrequest = self.mutablecopy;
 }
 //防止cookie丢失
 nsdictionary *dict = [nshttpcookie requestheaderfieldswithcookies:[nshttpcookiestorage sharedhttpcookiestorage].cookies];
 if (dict.count) {
  nsmutabledictionary *mdict = self.allhttpheaderfields.mutablecopy;
  [mdict setvaluesforkeyswithdictionary:dict];
  fixedrequest.allhttpheaderfields = mdict;
 }
 return fixedrequest;
}

@end
 
 
 // 使用场景
- (void)webview:(wkwebview *)webview decidepolicyfornavigationaction:(wknavigationaction *)navigationaction decisionhandler:(void (^)(wknavigationactionpolicy))decisionhandler {
 [navigationaction.request fixcookie];
 decisionhandler(wknavigationactionpolicyallow);
}