123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990 |
- 'use strict';
- 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;
|