Electron使用时拦截HTTP请求的解决方案

背景

最近在做一个Web和Electron共用一份代码的工程，由于使用到了第三方的库（我们是在线地图），该库的认证方式是请求时加key，并且它在后台会校验referer。

于是问题就来了，Electron是运行在本地的，http请求的时候，是不会带有授权Web站点的referer的。导致认证失败，无法调用所需的api接口。

分析

既然HTTP请求发送的时候不带referer，那么我们给他加上去，是不是就可以了呢？

对策

方法1：自定义HTTP协议，在发送请求时，补上referer

1// 注册自定义协议
2protocol.registerStandardSchemes(['app', '自定义协议'], { secure: true })
3// 自定义协议的具体实现
4protocol.registerHttpProtocol('自定义协议', (req, cb) => {
5        cb({
6            url: “特定的url地址”,
7            referrer: "http://被授权的referrer头信息"
8        })
9    })
10// 发送请求时可以使用 自定义协议：//url地址 的形式来发送请求       
11
12发送请求时可以使用 自定义协议：//url地址 的形式来发送请求。当然，我们这个例子里面只是追加了referrer，其实根据实际需求还可以做很多其它事情。相关api文档：https://electronjs.org/docs/api/protocol#protocolregisterhttpprotocolscheme-handler-completion

方法2：修改窗口会话中发送请求前的行为

本来以为方法1以及很完美的解决了问题，但是，在自定义协议认证成功后，在api的使用过程中，api内部主动发送了http请求，这时的api请求没有使用自定义协议，再次认证失败了。于是我想到有个办法，很天真的认为可以解决问题。失败的做法如下：

1protocol.interceptHttpProtocol("http", (req, cb) => {
2        // 此处可以追加url是否匹配的逻辑判断
3        cb({
4            url:req.url,
5            referrer: "http://授权的referrer头信息"
6        })
7    })
8
9相关api文档：https://electronjs.org/docs/api/protocol#protocolintercepthttpprotocolscheme-handler-completion

我直接拦截了原生的http协议，追加referrer信息，发生的结果是，http->截获http->修改后发送http->截获http->修改后发送http->......发生了无限循环！！！

于是只能另辟蹊径，寻找其余解决方案。发现了webquest的行为可以捕获。于是在onBeforeSendHeaders的时机，追加Referer信息。正确的做法如下：

1// 需要拦截的URL地址 
2   const xxx_filter = {
3        urls: ["https://*.xxx.com/*", "http://*.yyy.com/*"]
4    }    
5   session.defaultSession.webRequest.onBeforeSendHeaders(xxx_filter, (details, callback) => {
6        details.requestHeaders['Referer'] = 'http://授权的referer头信息'
7        callback({ requestHeaders: details.requestHeaders });
8    })
9
10相关api文档：https://electronjs.org/docs/api/web-request

总结

通过对webrequest的捕获，解决了本次的问题，并且把自定义协议那个案也可以替换掉。webrequest还有好几个阶段的行为可以自定义，可以适用于不同场景的需求。