mirror of
https://github.com/titanscouting/tra-analysis.git
synced 2024-11-14 07:06:17 +00:00
89 lines
2.7 KiB
JavaScript
89 lines
2.7 KiB
JavaScript
|
var util = require('util'),
|
||
|
net = require('net'),
|
||
|
tls = require('tls'),
|
||
|
url = require('url'),
|
||
|
driver = require('websocket-driver'),
|
||
|
API = require('./api'),
|
||
|
Event = require('./api/event');
|
||
|
|
||
|
var DEFAULT_PORTS = {'http:': 80, 'https:': 443, 'ws:':80, 'wss:': 443},
|
||
|
SECURE_PROTOCOLS = ['https:', 'wss:'];
|
||
|
|
||
|
var Client = function(_url, protocols, options) {
|
||
|
options = options || {};
|
||
|
|
||
|
this.url = _url;
|
||
|
this._driver = driver.client(this.url, {maxLength: options.maxLength, protocols: protocols});
|
||
|
|
||
|
['open', 'error'].forEach(function(event) {
|
||
|
this._driver.on(event, function() {
|
||
|
self.headers = self._driver.headers;
|
||
|
self.statusCode = self._driver.statusCode;
|
||
|
});
|
||
|
}, this);
|
||
|
|
||
|
var proxy = options.proxy || {},
|
||
|
endpoint = url.parse(proxy.origin || this.url),
|
||
|
port = endpoint.port || DEFAULT_PORTS[endpoint.protocol],
|
||
|
secure = SECURE_PROTOCOLS.indexOf(endpoint.protocol) >= 0,
|
||
|
onConnect = function() { self._onConnect() },
|
||
|
netOptions = options.net || {},
|
||
|
originTLS = options.tls || {},
|
||
|
socketTLS = proxy.origin ? (proxy.tls || {}) : originTLS,
|
||
|
self = this;
|
||
|
|
||
|
netOptions.host = socketTLS.host = endpoint.hostname;
|
||
|
netOptions.port = socketTLS.port = port;
|
||
|
|
||
|
originTLS.ca = originTLS.ca || options.ca;
|
||
|
socketTLS.servername = socketTLS.servername || endpoint.hostname;
|
||
|
|
||
|
this._stream = secure
|
||
|
? tls.connect(socketTLS, onConnect)
|
||
|
: net.connect(netOptions, onConnect);
|
||
|
|
||
|
if (proxy.origin) this._configureProxy(proxy, originTLS);
|
||
|
|
||
|
API.call(this, options);
|
||
|
};
|
||
|
util.inherits(Client, API);
|
||
|
|
||
|
Client.prototype._onConnect = function() {
|
||
|
var worker = this._proxy || this._driver;
|
||
|
worker.start();
|
||
|
};
|
||
|
|
||
|
Client.prototype._configureProxy = function(proxy, originTLS) {
|
||
|
var uri = url.parse(this.url),
|
||
|
secure = SECURE_PROTOCOLS.indexOf(uri.protocol) >= 0,
|
||
|
self = this,
|
||
|
name;
|
||
|
|
||
|
this._proxy = this._driver.proxy(proxy.origin);
|
||
|
|
||
|
if (proxy.headers) {
|
||
|
for (name in proxy.headers) this._proxy.setHeader(name, proxy.headers[name]);
|
||
|
}
|
||
|
|
||
|
this._proxy.pipe(this._stream, {end: false});
|
||
|
this._stream.pipe(this._proxy);
|
||
|
|
||
|
this._proxy.on('connect', function() {
|
||
|
if (secure) {
|
||
|
var options = {socket: self._stream, servername: uri.hostname};
|
||
|
for (name in originTLS) options[name] = originTLS[name];
|
||
|
self._stream = tls.connect(options);
|
||
|
self._configureStream();
|
||
|
}
|
||
|
self._driver.io.pipe(self._stream);
|
||
|
self._stream.pipe(self._driver.io);
|
||
|
self._driver.start();
|
||
|
});
|
||
|
|
||
|
this._proxy.on('error', function(error) {
|
||
|
self._driver.emit('error', error);
|
||
|
});
|
||
|
};
|
||
|
|
||
|
module.exports = Client;
|