1 | var httpProxy = module.exports,
|
2 | extend = require('util')._extend,
|
3 | parse_url = require('url').parse,
|
4 | EE3 = require('eventemitter3'),
|
5 | http = require('http'),
|
6 | https = require('https'),
|
7 | web = require('./passes/web-incoming'),
|
8 | ws = require('./passes/ws-incoming');
|
9 |
|
10 | httpProxy.Server = ProxyServer;
|
11 |
|
12 |
|
13 |
|
14 |
|
15 |
|
16 |
|
17 |
|
18 |
|
19 |
|
20 |
|
21 |
|
22 |
|
23 |
|
24 |
|
25 |
|
26 |
|
27 |
|
28 | function createRightProxy(type) {
|
29 |
|
30 | return function(options) {
|
31 | return function(req, res /*, [head], [opts] */) {
|
32 | var passes = (type === 'ws') ? this.wsPasses : this.webPasses,
|
33 | args = [].slice.call(arguments),
|
34 | cntr = args.length - 1,
|
35 | head, cbl,
|
36 | proxyOptions = options;
|
37 |
|
38 |
|
39 | if(typeof args[cntr] === 'function') {
|
40 | cbl = args[cntr];
|
41 |
|
42 | cntr--;
|
43 | }
|
44 |
|
45 | if(
|
46 | !(args[cntr] instanceof Buffer) &&
|
47 | args[cntr] !== res
|
48 | ) {
|
49 |
|
50 | proxyOptions = extend({}, options);
|
51 |
|
52 | extend(proxyOptions, args[cntr]);
|
53 |
|
54 | cntr--;
|
55 | }
|
56 |
|
57 | if(args[cntr] instanceof Buffer) {
|
58 | head = args[cntr];
|
59 | }
|
60 |
|
61 |
|
62 | if (!proxyOptions.target) {
|
63 | return this.emit('error', new Error('Must provide a proper URL as target'));
|
64 | }
|
65 |
|
66 | if (typeof proxyOptions.target === 'string')
|
67 | proxyOptions.target = parse_url(proxyOptions.target);
|
68 |
|
69 |
|
70 | for(var i=0; i < passes.length; i++) {
|
71 | |
72 |
|
73 |
|
74 |
|
75 |
|
76 |
|
77 |
|
78 |
|
79 | if(passes[i](req, res, proxyOptions, head, this, cbl)) {
|
80 | break;
|
81 | }
|
82 | }
|
83 | };
|
84 | };
|
85 | }
|
86 | httpProxy.createRightProxy = createRightProxy;
|
87 |
|
88 | function ProxyServer(options) {
|
89 | EE3.call(this);
|
90 |
|
91 | options = options || {};
|
92 | options.prependPath = options.prependPath === false ? false : true;
|
93 |
|
94 | this.web = this.proxyRequest = createRightProxy('web')(options);
|
95 | this.ws = this.proxyWebsocketRequest = createRightProxy('ws')(options);
|
96 | this.options = options;
|
97 |
|
98 | this.webPasses = Object.keys(web).map(function(pass) {
|
99 | return web[pass];
|
100 | });
|
101 |
|
102 | this.wsPasses = Object.keys(ws).map(function(pass) {
|
103 | return ws[pass];
|
104 | });
|
105 |
|
106 | this.on('error', this.onError, this);
|
107 |
|
108 | }
|
109 |
|
110 | require('util').inherits(ProxyServer, EE3);
|
111 |
|
112 | ProxyServer.prototype.onError = function (err) {
|
113 |
|
114 |
|
115 |
|
116 |
|
117 | if(this.listeners('error').length === 1) {
|
118 | throw err;
|
119 | }
|
120 | };
|
121 |
|
122 | ProxyServer.prototype.listen = function(port, hostname) {
|
123 | var self = this,
|
124 | closure = function(req, res) { self.web(req, res); };
|
125 |
|
126 | this._server = this.options.ssl ?
|
127 | https.createServer(this.options.ssl, closure) :
|
128 | http.createServer(closure);
|
129 |
|
130 | if(this.options.ws) {
|
131 | this._server.on('upgrade', function(req, socket, head) { self.ws(req, socket, head); });
|
132 | }
|
133 |
|
134 | this._server.listen(port, hostname);
|
135 |
|
136 | return this;
|
137 | };
|
138 |
|
139 | ProxyServer.prototype.close = function(callback) {
|
140 | var self = this;
|
141 | if (this._server) {
|
142 | this._server.close(done);
|
143 | }
|
144 |
|
145 |
|
146 | function done() {
|
147 | self._server = null;
|
148 | if (callback) {
|
149 | callback.apply(null, arguments);
|
150 | }
|
151 | };
|
152 | };
|
153 |
|
154 | ProxyServer.prototype.before = function(type, passName, callback) {
|
155 | if (type !== 'ws' && type !== 'web') {
|
156 | throw new Error('type must be `web` or `ws`');
|
157 | }
|
158 | var passes = (type === 'ws') ? this.wsPasses : this.webPasses,
|
159 | i = false;
|
160 |
|
161 | passes.forEach(function(v, idx) {
|
162 | if(v.name === passName) i = idx;
|
163 | })
|
164 |
|
165 | if(i === false) throw new Error('No such pass');
|
166 |
|
167 | passes.splice(i, 0, callback);
|
168 | };
|
169 | ProxyServer.prototype.after = function(type, passName, callback) {
|
170 | if (type !== 'ws' && type !== 'web') {
|
171 | throw new Error('type must be `web` or `ws`');
|
172 | }
|
173 | var passes = (type === 'ws') ? this.wsPasses : this.webPasses,
|
174 | i = false;
|
175 |
|
176 | passes.forEach(function(v, idx) {
|
177 | if(v.name === passName) i = idx;
|
178 | })
|
179 |
|
180 | if(i === false) throw new Error('No such pass');
|
181 |
|
182 | passes.splice(i++, 0, callback);
|
183 | };
|