/
setHeader.js
46 lines (40 loc) · 1.05 KB
/
setHeader.js
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
'use strict'
/*
* Adds a header to responses matching an URL. Default matches all urls
*/
module.exports = {
proxy: setHeader
}
var fs = require('fs'),
url = require('url')
/**
* // in ~/.magicproxyrc:
* setHeader: [
* {
* "url": "http://www.example.com/my/endpoint"
* "header": "Content-Type",
* "value": "text/urmom"
* },
* ]
*/
function setHeader(req, res, plugin) {
(plugin.config.setHeader || [])
.filter(function (headerAdd) {
return req.url === headerAdd.url;
})
.forEach(function (headerAdd) {
var res_end = res.end
var res_write = res.write
var res_writeHead = res.writeHead
var written = ''
res.writeHead = new Function
res.write = function (s) {written += s}
res.end = function (s) {
res.setHeader(headerAdd.header, headerAdd.value)
res_writeHead.call(res, res.statusCode)
res_write.call(res, written)
res_write.call(res, s || '')
res_end.call(res)
}
})
}