1 line
3.5 KiB
JavaScript
1 line
3.5 KiB
JavaScript
'use strict';Object.defineProperty(exports,'__esModule',{value:true});exports.MuxOutbound=exports.MuxInbound=undefined;var _extends=Object.assign||function(target){for(var i=1;i<arguments.length;i++){var source=arguments[i];for(var key in source){if(Object.prototype.hasOwnProperty.call(source,key)){target[key]=source[key]}}}return target};var _defs=require('./defs');var _defs2=require('../presets/defs');var _utils=require('../utils');function _asyncToGenerator(fn){return function(){var gen=fn.apply(this,arguments);return new Promise(function(resolve,reject){function step(key,arg){try{var info=gen[key](arg);var value=info.value}catch(error){reject(error);return}if(info.done){resolve(value)}else{return Promise.resolve(value).then(function(value){step('next',value)},function(err){step('throw',err)})}}return step('next')})}}class MuxInbound extends _defs.Inbound{constructor(props){super(props);this.onDrain=this.onDrain.bind(this);if(this._config.is_server){const inbound=this.ctx.muxRelay.getInbound();inbound.on('drain',this.onDrain)}else{}}get name(){return'mux:inbound'}get bufferSize(){if(this._config.is_client){const totalBufferSize=0;return totalBufferSize}else{const inbound=this.ctx.muxRelay.getInbound();if(inbound){return inbound.bufferSize}else{return 0}}}onBroadcast(action){const socket=this.ctx.socket;switch(action.type){case _defs2.CONNECT_TO_REMOTE:socket&&socket.pause();break;case _defs2.CONNECTED_TO_REMOTE:socket&&socket.resume();break;case _defs2.PRESET_FAILED:this.onPresetFailed(action);break;default:break;}}onPresetFailed(action){var _this=this;return _asyncToGenerator(function*(){var _action$payload=action.payload;const name=_action$payload.name,message=_action$payload.message;_utils.logger.error(`[${_this.name}] [${_this.remote}] preset "${name}" fail to process: ${message}`)})()}onDrain(){this.emit('drain')}write(buffer){if(this._config.is_server){var _ctx=this.ctx;const muxRelay=_ctx.muxRelay,cid=_ctx.cid;muxRelay.encode(buffer,{cid})}}end(){this.close()}close(){const doClose=()=>{if(this._config.is_server){var _ctx2=this.ctx;const muxRelay=_ctx2.muxRelay,cid=_ctx2.cid;const inbound=muxRelay.getInbound();if(inbound){inbound.removeListener('drain',this.onDrain)}muxRelay.destroySubRelay(cid)}if(!this._destroyed){this._destroyed=true;this.emit('close')}};if(this.bufferSize>0){this.once('drain',doClose)}else{doClose()}}}exports.MuxInbound=MuxInbound;class MuxOutbound extends _defs.Outbound{constructor(props){super(props);this._isFirstFrame=true;this.onDrain=this.onDrain.bind(this);if(this._config.is_client){const outbound=this.ctx.muxRelay.getOutbound();outbound.on('drain',this.onDrain)}else{}}get bufferSize(){if(this._config.is_client){const outbound=this.ctx.muxRelay.getOutbound();if(outbound){return outbound.bufferSize}else{return 0}}else{const totalBufferSize=0;return totalBufferSize}}onDrain(){this.emit('drain')}write(buffer){if(this._config.is_client){var _ctx3=this.ctx;const muxRelay=_ctx3.muxRelay,proxyRequest=_ctx3.proxyRequest,cid=_ctx3.cid;if(this._isFirstFrame){this._isFirstFrame=false;muxRelay.encode(buffer,_extends({cid},proxyRequest))}else{muxRelay.encode(buffer,{cid})}}}end(){this.close()}close(){const doClose=()=>{if(this._config.is_client){var _ctx4=this.ctx;const muxRelay=_ctx4.muxRelay,cid=_ctx4.cid;const outbound=muxRelay.getOutbound();if(outbound){outbound.removeListener('drain',this.onDrain)}muxRelay.destroySubRelay(cid)}if(!this._destroyed){this._destroyed=true;this.emit('close')}};if(this.bufferSize>0){this.once('drain',doClose)}else{doClose()}}}exports.MuxOutbound=MuxOutbound; |