使用nodejs将第三方页面嵌入到系统中

最近在公司遇到一个特殊的需求，需要将第三方的页面嵌入到系统中，并更改内部的样式。

---

如何将外来的页面嵌入到系统中？

百度查找了各路大神的方案，得出结论：js不允许访问不同源iframe内的元素。所以解决方案只能写一个后台来搞定这件事情。当时想写java后台，后面想想部署等相对复杂，最终使用了强大的nodejs。

首先尝试前端js能不能获取到不同源的iframe内的dom元素。

tip:试了才知道，映象才深刻。话不多说直接上代码。

html

<div class="col-xl-3">
    <button>删除</button>
</div>
<div class="col-xl-9">
    <!--不同源的百度页面-->
    <iframe  width="100%" height="800px" src="http://www.baidu.com"></iframe>
    <!--同源的百度页面-->
    <!--<iframe width="100%" height="800px" src="/baidu"></iframe>-->
</div>

js

//点击按钮就将百度页面头和大logo去掉
$('button').click(function () {
    var $iframe=$('iframe');
    $iframe.contents().find('head').remove();
    $iframe.contents().find('#lg').remove();
});

结果是点击了没有任何反应。

使用nodejs中转页面

使用express做接口，使用superagent做页面的请求。
routes代码如下：

var express = require('express');
var router = express.Router();
var path = require('path');
//使用agent自动处理cookie
const agent = require('superagent').agent();

router
    .get('/', function(req, res, next) {
      res.sendFile(path.join(__dirname, '../html', 'iframe.html'));
    })
    .get('/baidu',(req,res) => {
      agent.get('http://www.baidu.com',(areq,ares) => {
        res.end(ares.text);
      })
    });

module.exports = router;

当请求/baidu时agent会先将页面请求下来然后再返回给页面，这样就简单地处理了一个不同源问题。

我们修改html的内容，如下：

<div class="col-xl-9">
    <!--不同源的百度页面-->
    <!--<iframe  width="100%" height="800px" src="http://www.baidu.com"></iframe>-->
    <!--同源的百度页面-->
    <iframe width="100%" height="800px" src="/baidu"></iframe>
</div>

再次打开，点击删除按钮页面的head和lg成功被删除。

image

至此可以使用nodejs来解决这个问题。最后贴上实际代码。

js

'use strict';
// 要代理的url
const proxyUri = 'http://xxx.xxx.xxx';

var express = require('express');
var cookie = require('cookie-parser');
var events = require('events');
//本来想在nodejs中处理dom的
var cheerio = require('cheerio');
const superAgent = require('superagent');
const httpProxy = require('http-proxy');

var router = express.Router();

router.use(cookie());
const agent = superAgent.agent(express);

var emitter = new events.EventEmitter();
emitter.on("setCookeie", cookieReset);

let proxy = httpProxy.createProxyServer({
    target: proxyUri,
    changeOrigin: true,
    cookieDomainRewrite: {
        '*': proxyUri
    },
    secure: false,
});

proxy.on('error', function (err, request, response) {
    response.writeHead(500, {
        'Content-Type': 'text/plain',
    });
    console.log(err);
    response.end('server error');
});

//设置跨域访问
router.all('*', function (req, res, next) {
    res.header("Access-Control-Allow-Headers", "Origin, X-Requested-With, Content-Type, Accept");
    res.header("Access-Control-Allow-Methods", "PUT,POST,GET,DELETE,OPTIONS");
    res.header("X-Powered-By", ' 3.2.1');
    res.header('Access-Control-Allow-Origin', "*");
    res.header('Access-Control-Allow-Credentials', true);
    next();
});
router.use(express.static('../static/'));


login();

/**
 * 登录获取cookie
 */
function login(isFlag) {
    agent
        .post('http://xxx.xxx.xxx/login')
        .type('form')
        .send({username: 'admin'})
        .send({password: 'admin'})
        .send({remembercb: 'remember-me'})
        .redirects(0)
        .end(function (sreq, sres) {
            let cookie = sres.headers['set-cookie'];
            emitter.emit("setCookeie", cookie, isFlag);
        });
}

/**
 * 将设置的cookie与express共享
 * @param cookie
 */
function cookieReset(cookie, isFlag) {
    router.all('*', function (req, res, next) {
        res.header('set-cookie', cookie[0]);
        next();
    });
    if (isFlag) return;
    router
        .get('/', function (req, res) {
            res.sendFile('../static/index.html');
        })
        .get('/mit/*', function (req, res) {
            let proxy_url = proxyUri + (req.url).replace('mit', 'cmp');
            console.log('proxy:', proxy_url);
            //这里注意agent的缓存，它会缓存cookie，这样导致某cookie失效后就无法使用了。
            //这里推荐你这样做 superAgent.agent();
            agent
                .get(proxy_url)
                .end(function (areq, ares) {
                    if (ares.text) {
                    res.end(ares.text);
                    }else {
                        res.end('<h2>请求失败</h2>');
                    }
                });
        })
        .all('/mitc/*', function (req, res) {
            if (req.url === '/mitc/login') {
                console.log('cookie超时，请求登录页了');
                login(true);
                res.redirect('/');
            } else {
                proxy.web(req, res);
                return;
            }
        });
};

module.exports = router;

基本思想是：开始使用anget登录到该网站，并将cookie保存并设置给express。之后
使用http-proxy代理来自iframe内的所有请求（iframe内的css、js等）。这里页面cookie会过期，
这里并不希望它调到登录页面，所有拦截它的登录请求，再次登录。

ok成功解决了问题，一个好周末就来了。
出现了一个cookie缓存的bug，如果你和我一样的代码，那么必须注意agent的缓存cookie（原因未知），使用必须看一下。