@luminati-io/luminati-proxy
Version:
A configurable local proxy for luminati.io
1,566 lines (1,518 loc) • 47.3 kB
JavaScript
// LICENSE_CODE ZON ISC
'use strict'; /*jslint node:true, browser:true, es6: true*/
(function(){
var define, process, zerr, assert;
var is_node = typeof module=='object' && module.exports && module.children;
var is_rn = typeof global=='object' && !!global.nativeRequire ||
typeof navigator=='object' && navigator.product=='ReactNative';
if (!is_node)
{
if (is_rn)
{
define = require('./require_node.js').define(module, '../',
require('/util/events.js'), require('/util/array.js'),
require('/util/util.js'));
}
else
define = self.define;
process = {
nextTick: function(fn){ setTimeout(fn, 0); },
env: {},
};
// XXX romank: use zerr.js
// XXX bahaa: require bext/pub/zerr.js for extensions
if (!is_rn && self.hola && self.hola.zerr)
zerr = self.hola.zerr;
else
{
zerr = function(){ console.log.apply(console, arguments); };
zerr.perr = zerr;
zerr.debug = function(){};
zerr.is = function(){ return false; };
zerr.L = {DEBUG: 0};
}
if (!zerr.is)
zerr.is = function(){ return false; };
}
else
{
require('./config.js');
process = global.process||require('_process');
zerr = require('./zerr.js');
assert = require('assert');
define = require('./require_node.js').define(module, '../');
}
// XXX odin: normally this would only be run for !is_node, but 'who' unittests
// loads a stubbed assert
if (typeof assert!='function')
assert = function(){}; // XXX romank: add proper assert
// XXX yuval: /util/events.js -> events when node 6 (support prependListener)
// is here
define(['/util/events.js', '/util/array.js', '/util/util.js'],
function(events, array, zutil){
var E = Etask;
var etask = Etask;
var env = process.env, assign = Object.assign;
E.use_bt = +env.ETASK_BT;
E.root = new Set();
E.assert_extra = +env.ETASK_ASSERT_EXTRA; // to debug internal etask bugs
E.nextTick = process.nextTick;
// XXX arik/romank: hack, rm set_zerr, get zerzerrusing require
E.set_zerr = function(_zerr){ zerr = _zerr; };
E.events = new events();
var cb_pre, cb_post, cb_ctx, longcb_ms, perf_enable;
E.perf_stat = {};
// XXX romang: hack to import in react native
if (is_rn)
E.etask = E;
function _cb_pre(et){ return {start: Date.now()}; }
function _cb_post(et, ctx){
ctx = ctx||cb_ctx;
var ms = Date.now()-ctx.start;
if (longcb_ms && ms>longcb_ms)
{
zerr('long cb '+ms+'ms: '+et.get_name()+', '
+et.run_state.f.toString().slice(0, 128));
}
if (perf_enable)
{
var name = et.get_name();
var perf = E.perf_stat[name] ||
(E.perf_stat[name] = {ms: 0, n: 0, max: 0});
if (perf.max<ms)
perf.max = ms;
perf.ms += ms;
perf.n++;
}
}
function cb_set(){
if (longcb_ms || perf_enable)
{
cb_pre = _cb_pre;
cb_post = _cb_post;
cb_ctx = {start: Date.now()};
}
else
cb_pre = cb_post = cb_ctx = undefined;
}
E.longcb = function(ms){
longcb_ms = ms;
cb_set();
};
E.perf = function(enable){
if (arguments.length)
{
perf_enable = enable;
cb_set();
}
return perf_enable;
};
E.longcb(+env.LONGCB);
E.perf(+env.ETASK_PERF);
function stack_get(){
// new Error(): 200K per second
// http://jsperf.com/error-generation
// Function.caller (same as arguments.callee.caller): 2M per second
// http://jsperf.com/does-function-caller-affect-preformance
// http://jsperf.com/the-arguments-object-s-effect-on-speed/2
var prev = Error.stackTraceLimit, err;
Error.stackTraceLimit = 4;
err = new Error();
Error.stackTraceLimit = prev;
return err;
}
function Etask(opt, states){
if (!(this instanceof Etask))
return new Etask(opt, states);
if (Array.isArray(opt) || typeof opt=='function')
{
states = opt;
opt = undefined;
}
opt = typeof opt=='string' && {name: opt} || opt || {};
if (typeof states=='function')
{
if (states.constructor.name=='GeneratorFunction')
return E._generator(null, states, opt);
states = [states];
}
// performance: set all fields to undefined
this.cur_state = this.states = this._finally = this.error =
this.at_return = this.next_state = this.use_retval = this.running =
this.at_continue = this.cancel = this.wait_timer = this.retval =
this.run_state = this._stack = this.down = this.up = this.child =
this.name = this._name = this.parent = this.cancelable =
this.tm_create = this._alarm = this.tm_completed = this.parent_type =
this.info = this.then_waiting = this.free = this.parent_guess =
this.child_guess = this.wait_retval = undefined;
// init fields
this.name = opt.name;
this._name = this.name===undefined ? 'noname' : this.name;
this.cancelable = opt.cancel;
this.then_waiting = new Set();
this.child = new Set();
this.child_guess = new Set();
this.cur_state = -1;
this.states = [];
this._stack = Etask.use_bt ? stack_get() : undefined;
this.tm_create = Date.now();
this.info = {};
var idx = this.states.idx = {};
for (var i=0; i<states.length; i++)
{
var pstate = states[i], t;
if (typeof pstate!='function')
assert(0, 'invalid state type');
t = this._get_func_type(pstate);
var state = {f: pstate, label: t.label, try_catch: t.try_catch,
catch: t.catch, finally: t.finally, cancel: t.cancel,
sig: undefined};
if (i==0 && opt.state0_args)
{
state.f = state.f.bind.apply(state.f,
[this].concat(opt.state0_args));
}
if (state.label)
idx[state.label] = i;
assert((state.catch||state.try_catch?1:0)
+(state.finally?1:0)+(state.cancel?1:0)<=1,
'invalid multiple state types');
state.sig = state.finally||state.cancel;
if (state.finally)
{
assert(this._finally===undefined, 'more than 1 finally$');
this._finally = i;
}
if (state.cancel)
{
assert(this.cancel===undefined, 'more than 1 cancel$');
this.cancel = i;
}
this.states[i] = state;
}
var _this = this;
E.root.add(this);
var in_run = E.in_run_top();
if (opt.spawn_parent)
this.spawn_parent(opt.spawn_parent);
else if (opt.up)
opt.up._set_down(this);
else if (in_run)
this._spawn_parent_guess(in_run);
if (opt.init)
opt.init.call(this);
if (opt.async)
{
var wait_retval = this._set_wait_retval();
E.nextTick(function(){
if (_this.running!==undefined)
return;
_this._got_retval(wait_retval);
});
}
else
this._next_run();
return this;
}
zutil.inherits(Etask, events.EventEmitter);
E.prototype._root_remove = function(){
assert(!this.parent, 'cannot remove from root when has parent');
if (!E.root.delete(this))
assert(0, 'etask not in root\n'+E.ps({MARK: this}));
};
E.prototype._parent_remove = function(){
if (this.up)
{
var up = this.up;
this.up = this.up.down = undefined;
if (up.tm_completed)
up._check_free();
return;
}
if (this.parent_guess)
this._parent_guess_remove();
if (!this.parent)
return this._root_remove();
if (!this.parent.child.delete(this))
{
assert(0, 'etask child not in parent\n'
+E.ps({MARK: [['child', this], ['parent', this.parent]]}));
}
if (this.parent.tm_completed)
this.parent._check_free();
this.parent = undefined;
};
E.prototype._check_free = function(){
if (this.down || this.child.size)
return;
this._parent_remove();
this.free = true;
};
E.prototype._call_err = function(e){
E.ef(e);
// XXX derry: add assert(0, 'etask err in signal: '+e);
};
E.prototype.emit_safe = function(){
try { this.emit.apply(this, arguments); }
catch(e){ this._call_err(e); }
};
E.prototype._call_safe = function(state_fn){
try { return state_fn.call(this); }
catch(e){ this._call_err(e); }
};
E.prototype._complete = function(){
if (zerr.is(zerr.L.DEBUG))
zerr.debug(this._name+': close');
this.tm_completed = Date.now();
this.parent_type = this.up ? 'call' : 'spawn';
if (this.error)
this.emit_safe('uncaught', this.error);
if (this._finally!==undefined)
{
var ret = this._call_safe(this.states[this._finally].f);
if (E.is_err(ret))
this._set_retval(ret);
}
this.emit_safe('finally');
this.emit_safe('ensure');
if (this.error && !this.up && !this.parent && !this.parent_guess)
E.events.emit('uncaught', this);
if (this.parent)
this.parent.emit('child', this);
if (this.up && (this.down || this.child.size))
{
var up = this.up;
this.up = this.up.down = undefined;
this.parent = up;
up.child.add(this);
}
this._check_free();
this._del_wait_timer();
this.del_alarm();
this._ecancel_child();
this.emit_safe('finally1');
for (let v of this.then_waiting.values())
{
this.then_waiting.delete(v);
v();
}
};
E.prototype._next = function(rv){
if (this.tm_completed)
return true;
rv = rv||{ret: undefined, err: undefined};
var states = this.states;
var state = this.at_return ? states.length :
this.next_state!==undefined ? this.next_state :
this.cur_state+1;
this.retval = rv.ret;
this.error = rv.err;
if (rv.err!==undefined)
{
if (zerr.on_exception)
zerr.on_exception(rv.err);
if (this.run_state.try_catch)
{
this.use_retval = true;
for (; state<states.length && states[state].sig; state++);
}
else
for (; state<states.length && !states[state].catch; state++);
}
else
{
for (; state<states.length && (states[state].sig || states[state]
.catch); state++)
{
}
}
this.cur_state = state;
this.run_state = states[state];
this.next_state = undefined;
if (this.cur_state<states.length)
return false;
this._complete();
return true;
};
E.prototype._next_run = function(rv){
if (this._next(rv))
return;
this._run();
};
E.prototype._handle_rv = function(rv){
var wait_retval, _this = this, ret = rv.ret;
if (ret===this.retval); // fast-path: retval already set
else if (!ret);
else if (ret instanceof Etask)
{
if (!ret.tm_completed)
{
this._set_down(ret);
wait_retval = this._set_wait_retval();
ret.then_waiting.add(function(){
_this._got_retval(wait_retval, E.err_res(ret.error,
ret.retval));
});
return true;
}
rv.err = ret.error;
rv.ret = ret.retval;
}
else if (ret instanceof Etask_err)
{
rv.err = ret.error;
rv.ret = undefined;
}
else if (typeof ret.then=='function') // promise
{
wait_retval = this._set_wait_retval();
ret.then(function(_ret){ _this._got_retval(wait_retval, _ret); },
function(err){ _this._got_retval(wait_retval, E.err(err)); });
return true;
}
// generator
else if (typeof ret.next=='function' && typeof ret.throw=='function')
{
rv.ret = E._generator(ret, this.states[this.cur_state]);
return this._handle_rv(rv);
}
return false;
};
E.prototype._set_retval = function(ret){
if (ret===this.retval && !this.error); // fast-path retval already set
else if (!ret)
{
this.retval = ret;
this.error = undefined;
}
else if (ret instanceof Etask)
{
if (ret.tm_completed)
{
this.retval = ret.retval;
this.error = ret.error;
}
}
else if (ret instanceof Etask_err)
{
this.retval = undefined;
this.error = ret.error;
}
else if (typeof ret.then=='function'); // promise
// generator
else if (typeof ret.next=='function' && typeof ret.throw=='function');
else
{
this.retval = ret;
this.error = undefined;
}
return ret;
};
E.prototype._set_wait_retval = function(){
return this.wait_retval = new Etask_wait(this, 'wait_int'); };
E.in_run = [];
E.in_run_top = function(){ return E.in_run[E.in_run.length-1]; };
E.prototype._run = function(){
var rv = {ret: undefined, err: undefined};
while (1)
{
var _cb_ctx;
var arg = this.error && !this.use_retval ? this.error : this.retval;
this.use_retval = false;
this.running = true;
rv.ret = rv.err = undefined;
E.in_run.push(this);
if (zerr.is(zerr.L.DEBUG))
zerr.debug(this._name+':S'+this.cur_state+': running');
if (cb_pre)
_cb_ctx = cb_pre(this);
try { rv.ret = this.run_state.f.call(this, arg); }
catch(e){
rv.err = e;
if (rv.err instanceof Error)
rv.err.etask = this;
}
if (cb_post)
cb_post(this, _cb_ctx);
this.running = false;
E.in_run.pop();
for (let vv of this.child_guess.values())
{
this.child_guess.delete(vv);
vv.parent_guess = undefined;
}
if (rv.ret instanceof Etask_wait)
{
var wait_completed = false, wait = rv.ret;
if (!this.at_continue && !wait.ready)
{
this.wait_retval = wait;
if (wait.op=='wait_child')
wait_completed = this._set_wait_child(wait);
if (wait.timeout)
this._set_wait_timer(wait.timeout);
if (!wait_completed)
return;
this.wait_retval = undefined;
}
rv.ret = this.at_continue ? this.at_continue.ret :
wait.ready && !wait.completed ? wait.ready.ret : undefined;
wait.completed = true;
}
this.at_continue = undefined;
if (this._handle_rv(rv))
return;
if (this._next(rv))
return;
}
};
E.prototype._set_down = function(down){
if (this.down)
assert(0, 'caller already has a down\n'+this.ps());
if (down.parent_guess)
down._parent_guess_remove();
assert(!down.parent, 'returned etask already has a spawn parent');
assert(!down.up, 'returned etask already has a caller parent');
down._parent_remove();
this.down = down;
down.up = this;
};
var func_type_cache = {};
E.prototype._get_func_type = function(func, on_fail){
var name = func.name;
var type = func_type_cache[name];
if (type)
return type;
type = func_type_cache[name] = {name: undefined, label: undefined,
try_catch: undefined, catch: undefined, finally: undefined,
cancel: undefined};
if (!name)
return type;
type.name = name;
var n = name.split('$');
if (n.length==1)
{
type.label = n[0];
return type;
}
if (n.length>2)
return type;
if (n[1].length)
type.label = n[1];
var f = n[0].split('_');
for (var j=0; j<f.length; j++)
{
if (f[j]=='try')
{
type.try_catch = true;
if (f[j+1]=='catch')
j++;
}
else if (f[j]=='catch')
type['catch'] = true;
else if (f[j]=='finally' || f[j]=='ensure')
type.finally = true;
else if (f[j]=='cancel')
type.cancel = true;
else
{
return void (on_fail||assert.bind(null, false))(
'unknown func name '+name);
}
}
return type;
};
E.prototype.spawn = function(child, replace){
if (!(child instanceof Etask) && child && typeof child.then=='function')
{
var promise = child;
child = etask([function(){ return promise; }]);
}
if (!(child instanceof Etask)) // promise already completed?
{
this.emit('child', child);
return child;
}
if (!replace && child.parent)
assert(0, 'child already has a parent\n'+child.parent.ps());
child.spawn_parent(this);
return child;
};
E.prototype._spawn_parent_guess = function(parent){
this.parent_guess = parent;
parent.child_guess.add(this);
};
E.prototype._parent_guess_remove = function(){
if (!this.parent_guess.child_guess.delete(this))
assert(0, 'etask not in parent_guess\n'+E.ps({MARK: this}));
this.parent_guess = undefined;
};
E.prototype.spawn_parent = function(parent){
if (this.up)
assert(0, 'child already has an up\n'+this.up.ps());
if (this.tm_completed && !this.parent)
return;
this._parent_remove();
if (parent && parent.free)
parent = undefined;
if (!parent)
return void E.root.add(this);
parent.child.add(this);
this.parent = parent;
};
E.prototype.set_state = function(name){
var state = this.states.idx[name];
assert(state!==undefined, 'named func "'+name+'" not found');
return this.next_state = state;
};
E.prototype.finally = function(cb){
this.prependListener('finally', cb); };
E.prototype.goto_fn = function(name){
return this.goto.bind(this, name); };
E.prototype.goto = function(name, promise){
this.set_state(name);
var state = this.states[this.next_state];
assert(!state.sig, 'goto to sig');
return this.continue(promise);
};
E.prototype.loop = function(promise){
this.next_state = this.cur_state;
return promise;
};
E.prototype._set_wait_timer = function(timeout){
var _this = this;
this.wait_timer = setTimeout(function(){
_this.wait_timer = undefined;
_this._next_run({ret: undefined, err: 'timeout'});
}, timeout);
};
E.prototype._del_wait_timer = function(){
if (this.wait_timer)
this.wait_timer = clearTimeout(this.wait_timer);
this.wait_retval = undefined;
};
E.prototype._get_child_running = function(){
for (let v of this.child.values())
{
if (!v.tm_completed)
return v;
}
};
E.prototype._set_wait_child = function(wait_retval){
var _this = this, child = wait_retval.child;
var cond = wait_retval.cond, wait_on;
assert(!cond || child=='any', 'condition supported only for "any" '+
'option, you can add support if needed');
if (child=='any')
{
if (!this._get_child_running())
return true;
wait_on = function(){
_this.once('child', function(_child){
if (!cond || cond.call(_child, _child.retval))
return _this._got_retval(wait_retval, {child: _child});
if (!_this._get_child_running())
return _this._got_retval(wait_retval);
wait_on();
});
};
wait_on();
}
else if (child=='all')
{
let child_et = this._get_child_running();
if (!child_et)
return true;
wait_on = function(){
_this.once('child', function(){
let ch_child_et = _this._get_child_running();
if (!ch_child_et)
return _this._got_retval(wait_retval);
wait_on(ch_child_et);
});
};
wait_on(child_et);
}
else
{
assert(child, 'no child provided');
assert(this===child.parent, 'child does not belong to parent');
if (child.tm_completed)
return true;
child.once('finally', function(){
return _this._got_retval(wait_retval, {child: child}); });
}
this.emit_safe('wait_on_child');
};
E.prototype._got_retval = function(wait_retval, res){
if (this.wait_retval!==wait_retval || wait_retval.completed)
return;
wait_retval.completed = true;
// inline _next_run to reduce stack depth
if (!this._next(E._res2rv(res)))
this._run();
};
E.prototype.continue_fn = function(){
return this.continue.bind(this); };
E.continue_depth = 0;
E.prototype.continue = function(promise, sync){
this.wait_retval = undefined;
this._set_retval(promise);
if (this.tm_completed)
return promise;
if (this.down)
this.down._ecancel();
this._del_wait_timer();
var rv = {ret: promise, err: undefined};
if (this.running)
{
this.at_continue = rv;
return promise;
}
if (this._handle_rv(rv))
return rv.ret;
var _this = this;
if (E.is_final(promise) &&
(!E.continue_depth && !E.in_run.length || sync))
{
E.continue_depth++;
this._next_run(rv);
E.continue_depth--;
}
else // avoid high stack depth
E.nextTick(function(){ _this._next_run(rv); });
return promise;
};
E.prototype._ecancel = function(){
if (this.tm_completed)
return this;
this.emit_safe('cancel');
if (this.cancel!==undefined)
return this._call_safe(this.states[this.cancel].f);
if (this.cancelable)
return this.return();
};
E.prototype._ecancel_child = function(){
if (!this.child.size)
return;
// copy array, since ecancel has side affects and can modify array
var child = Array.from(this.child.values());
for (var i=0; i<child.length; i++)
child[i]._ecancel();
};
E.prototype.return_fn = function(){
return this.return.bind(this); };
E.prototype.return = function(promise){
if (this.tm_completed)
return this._set_retval(promise);
this.at_return = true;
this.next_state = undefined;
return this.continue(promise, true);
};
E.prototype.del_alarm = function(){
var a = this._alarm;
if (!a)
return;
clearTimeout(a.id);
if (a.cb)
this.removeListener('sig_alarm', a.cb);
this._alarm = undefined;
};
E.prototype.alarm_left = function(){
var a = this._alarm;
if (!a)
return 0;
return a.start-Date.now();
};
E.prototype._operation_opt = function(opt){
if (opt.goto)
return {ret: this.goto(opt.goto, opt.ret)};
if (opt.throw)
return {ret: this.throw(opt.throw)};
if (opt.return!==undefined)
return {ret: this.return(opt.return)};
if (opt.continue!==undefined)
return {ret: this.continue(opt.continue)};
};
E.prototype.alarm = function(ms, cb){
var _this = this, opt, a;
if (cb && typeof cb!='function')
{
opt = cb;
cb = function(){
var v;
if (!(v = _this._operation_opt(opt)))
assert(0, 'invalid alarm cb opt');
return v.ret;
};
}
this.del_alarm();
a = this._alarm = {ms: ms, cb: cb, start: Date.now()};
a.id = setTimeout(function(){
_this._alarm = undefined;
_this.emit('sig_alarm');
}, a.ms);
if (cb)
this.once('sig_alarm', cb);
};
function Etask_wait(et, op, timeout){
this.timeout = timeout;
this.et = et;
this.op = op;
this.child = this.at_child = this.cond = undefined;
this.ready = this.completed = undefined;
}
Etask_wait.prototype.continue = function(res){
if (this.completed)
return;
if (!this.et.wait_retval)
return void(this.ready = {ret: res});
if (this!==this.et.wait_retval)
return;
this.et.continue(res);
};
Etask_wait.prototype.continue_fn = function(){
return this.continue.bind(this); };
Etask_wait.prototype.throw = function(err){
return this.continue(E.err(err)); };
Etask_wait.prototype.throw_fn = function(){
return this.throw.bind(this); };
E.prototype.wait = function(timeout){
return new Etask_wait(this, 'wait', timeout); };
E.prototype.wait_child = function(child, timeout, cond){
if (typeof timeout=='function')
{
cond = timeout;
timeout = 0;
}
var wait = new Etask_wait(this, 'wait_child', timeout);
wait.child = child;
wait.at_child = null;
wait.cond = cond;
return wait;
};
E.prototype.throw_fn = function(err){
return err ? this.throw.bind(this, err) : this.throw.bind(this); };
E.prototype.throw = function(err){
return this.continue(E.err(err)); };
E.prototype.get_name = function(flags){
/* anon: Context.<anonymous> (/home/yoni/zon1/pkg/util/test.js:1740:7)
* with name: Etask.etask1_1 (/home/yoni/zon1/pkg/util/test.js:1741:11) */
var stack = this._stack instanceof Error ? this._stack.stack.split('\n') :
undefined;
var caller;
flags = flags||{};
if (stack)
{
caller = /^ {4}at (.*)$/.exec(stack[4]);
caller = caller ? caller[1] : undefined;
}
var names = [];
if (this.name)
names.push(this.name);
if (caller && !(this.name && flags.SHORT_NAME))
names.push(caller);
if (!names.length)
names.push('noname');
return names.join(' ');
};
E.prototype.state_str = function(){
return this.cur_state+(this.next_state ? '->'+this.next_state : ''); };
E.prototype.get_depth = function(){
var i=0, et = this;
for (; et; et = et.up, i++);
return i;
};
function trim_space(s){
if (s[s.length-1]!=' ')
return s;
return s.slice(0, -1);
}
function ms_to_str(ms){ // from date.js
var s = ''+ms;
return s.length<=3 ? s+'ms' : s.slice(0, -3)+'.'+s.slice(-3)+'s';
}
E.prototype.get_time_passed = function(){
return ms_to_str(Date.now()-this.tm_create); };
E.prototype.get_time_completed = function(){
return ms_to_str(Date.now()-this.tm_completed); };
E.prototype.get_info = function(){
var info = this.info, s = '', _i;
if (!info)
return '';
for (var i in info)
{
_i = info[i];
if (!_i)
continue;
if (s!=='')
s += ' ';
if (typeof _i=='function')
s += _i();
else
s += _i;
}
return trim_space(s);
};
// light-weight efficient etask/promise error value
function Etask_err(err){ this.error = err || new Error(); }
E.Etask_err = Etask_err;
E.err = function(err){ return new Etask_err(err); };
E.is_err = function(v){
return v instanceof Etask && v.error!==undefined ||
v instanceof Etask_err;
};
E.err_res = function(err, res){ return err ? E.err(err) : res; };
E._res2rv = function(res){
return E.is_err(res) ? {ret: undefined, err: res.error}
: {ret: res, err: undefined};
};
E.is_final = function(v){
return !v || typeof v.then!='function' || v instanceof Etask_err ||
v instanceof Etask && !!v.tm_completed;
};
// promise compliant .then() implementation for Etask and Etask_err.
// for unit-test comfort, also .otherwise(), .catch(), .ensure(), resolve() and
// reject() are implemented.
E.prototype.then = function(on_res, on_err){
var _this = this;
function on_done(){
if (!_this.error)
return !on_res ? _this.retval : on_res(_this.retval);
return !on_err ? E.err(_this.error) : on_err(_this.error);
}
if (this.tm_completed)
return etask('then_completed', [function(){ return on_done(); }]);
var then_wait = etask('then_wait', [function(){ return this.wait(); }]);
this.then_waiting.add(function(){
try { then_wait.continue(on_done()); }
catch(e){ then_wait.throw(e); }
});
return then_wait;
};
E.prototype.otherwise = E.prototype.catch = function(on_err){
return this.then(null, on_err); };
E.prototype.ensure = function(on_ensure){
return this.then(function(res){ on_ensure(); return res; },
function(err){ on_ensure(); throw err; });
};
Etask_err.prototype.then = function(on_res, on_err){
var _this = this;
return etask('then_err', [function(){
return !on_err ? E.err(_this.error) : on_err(_this.error);
}]);
};
Etask_err.prototype.otherwise = Etask_err.prototype.catch = function(on_err){
return this.then(null, on_err); };
Etask_err.prototype.ensure = function(on_ensure){
this.then(null, function(){ on_ensure(); });
return this;
};
E.resolve = function(res){ return etask([function(){ return res; }]); };
E.reject = function(err){ return etask([function(){ throw err; }]); };
E.prototype.wait_ext = function(promise){
if (!promise || typeof promise.then!='function')
return promise;
var wait = this.wait();
promise.then(wait.continue_fn(), wait.throw_fn());
return wait;
};
E.prototype.longname = function(flags){
flags = flags||{TIME: 1};
var s = '', _s;
if (this.running)
s += 'RUNNING ';
s += this.get_name(flags)+(!this.tm_completed ? '.'+this.state_str() : '')
+' ';
if (this.tm_completed)
s += 'COMPLETED'+(flags.TIME ? ' '+this.get_time_completed() : '')+' ';
if (flags.TIME)
s += this.get_time_passed()+' ';
if (_s = this.get_info())
s += _s+' ';
return trim_space(s);
};
E.prototype.stack = function(flags){
var et = this, s = '';
flags = assign({STACK: 1, RECURSIVE: 1, GUESS: 1}, flags);
while (et)
{
var _s = et.longname(flags)+'\n';
if (et.up)
et = et.up;
else if (et.parent)
{
_s = (et.parent_type=='call' ? 'CALL' : 'SPAWN')+' '+_s;
et = et.parent;
}
else if (et.parent_guess && flags.GUESS)
{
_s = 'SPAWN? '+_s;
et = et.parent_guess;
}
else
et = undefined;
if (flags.TOPDOWN)
s = _s+s;
else
s += _s;
}
return s;
};
E.prototype._ps = function(pre_first, pre_next, flags){
var i, s = '', task_trail, et = this, child_guess;
if (++flags.limit_n>=flags.LIMIT)
return flags.limit_n==flags.LIMIT ? '\nLIMIT '+flags.LIMIT+'\n': '';
/* get top-most et */
for (; et.up; et = et.up);
/* print the sp frames */
for (var first = 1; et; et = et.down, first = 0)
{
s += first ? pre_first : pre_next;
first = 0;
if (flags.MARK && (i = flags.MARK.sp.indexOf(et))>=0)
s += (flags.MARK.name[i]||'***')+' ';
s += et.longname(flags)+'\n';
if (flags.RECURSIVE)
{
var stack_trail = et.down ? '.' : ' ';
var child = et.child;
if (flags.GUESS)
child = new Set([...child, ...et.child_guess]);
i = 0;
for (let child_i of child.values())
{
task_trail = i<child.size-1 ? '|' : stack_trail;
child_guess = child_i.parent_guess ? '\\? ' :
child_i.parent_type=='call' ? '\\> ' : '\\_ ';
s += child_i._ps(pre_next+task_trail+child_guess,
pre_next+task_trail+' ', flags);
i++;
}
}
}
return s;
};
function ps_flags(flags){
var m, _m;
if (m = flags.MARK)
{
if (!Array.isArray(m))
_m = {sp: [m], name: []};
else if (!Array.isArray(flags.MARK[0]))
_m = {sp: m, name: []};
else
{
_m = {sp: [], name: []};
for (var i=0; i<m.length; i++)
{
_m.name.push(m[i][0]);
_m.sp.push(m[i][1]);
}
}
flags.MARK = _m;
}
}
E.prototype.ps = function(flags){
flags = assign({STACK: 1, RECURSIVE: 1, LIMIT: 10000000, TIME: 1,
GUESS: 1}, flags, {limit_n: 0});
ps_flags(flags);
return this._ps('', '', flags);
};
E._longname_root = function(){
return (zerr.prefix ? zerr.prefix+'pid '+process.pid+' ' : '')+'root'; };
E.ps = function(flags){
var s = '', task_trail;
flags = assign({STACK: 1, RECURSIVE: 1, LIMIT: 10000000, TIME: 1,
GUESS: 1}, flags, {limit_n: 0});
ps_flags(flags);
s += E._longname_root()+'\n';
var child = Array.from(E.root.values());
if (flags.GUESS)
{
child = [];
E.root.forEach(root_i=>{
if (!root_i.parent_guess)
child.push(root_i);
});
}
child.forEach((child_i, i)=>{
task_trail = i<child.length-1 ? '|' : ' ';
s += child_i._ps(task_trail+'\\_ ', task_trail+' ', flags);
});
return s;
};
function assert_tree_unique(a){
var i;
for (i=0; i<a.length-1; i++)
assert(!a.includes(a[i], i+1));
}
E.prototype._assert_tree = function(opt){
var et;
opt = opt||{};
assert_tree_unique(this.child);
assert(this.parent);
if (this.down)
{
et = this.down;
assert(et.up===this);
assert(!et.parent);
assert(!et.parent_guess);
this.down._assert_tree(opt);
}
for (let _et of this.child.values())
{
assert(_et.parent===this);
assert(!_et.parent_guess);
assert(!_et.up);
_et._assert_tree(opt);
}
if (this.child_guess.size)
assert(E.in_run.includes(this));
for (let _et of this.child_guess.values())
{
assert(_et.parent_guess===this);
assert(!_et.parent);
assert(!_et.up);
}
};
E._assert_tree = function(opt){
opt = opt||{};
assert_tree_unique(E.root);
for (let et of this.child.values())
{
assert(!et.parent);
assert(!et.up);
et._assert_tree(opt);
}
};
E.prototype._assert_parent = function(){
if (this.up)
return assert(!this.parent && !this.parent_guess);
assert(this.parent && this.parent_guess,
'parent_guess together with parent');
if (this.parent)
{
var child = this.parent ? this.parent.child : E.root;
assert(child.has(this),
'cannot find in parent '+(this.parent ? '' : 'root'));
}
else if (this.parent_guess)
{
assert(this.parent_guess.child_guess.has(this),
'cannot find in parent_guess');
assert(E.in_run.includes(this.parent_guess));
}
};
E.prototype.return_child = function(){
// copy array, since return() has side affects and can modify array
var child = Array.from(this.child.values());
for (var i=0; i<child.length; i++)
child[i].return();
};
E.sleep = function(ms){
var timer;
ms = ms||0;
return etask({name: 'sleep', cancel: true}, [function(){
this.info.ms = ms+'ms';
timer = setTimeout(this.continue_fn(), ms);
return this.wait();
}, function finally$(){
clearTimeout(timer);
}]);
};
var ebreak_obj = {ebreak: 1};
E.prototype.break = function(ret){
return this.throw({ebreak: ebreak_obj, ret: ret}); };
E.for = function(cond, inc, opt, states){
if (Array.isArray(opt) || typeof opt=='function')
{
states = opt;
opt = {};
}
if (typeof states=='function')
states = [states];
opt = opt||{};
return etask({name: 'for', cancel: true, init: opt.init_parent},
[function loop(){
return !cond || cond.call(this);
}, function try_catch$(res){
if (!res)
return this.return();
return etask({name: 'for_iter', cancel: true, init: opt.init},
states||[]);
}, function(){
if (this.error)
{
if (this.error.ebreak===ebreak_obj)
return this.return(this.error.ret);
return this.throw(this.error);
}
return inc && inc.call(this);
}, function(){
return this.goto('loop');
}]);
};
E.for_each = function(obj, states){
var keys = Object.keys(obj);
var iter = {obj: obj, keys: keys, i: 0, key: undefined, val: undefined};
function init_iter(){ this.iter = iter; }
return E.for(function(){
this.iter = this.iter||iter;
iter.key = keys[iter.i];
iter.val = obj[keys[iter.i]];
return iter.i<keys.length;
},
function(){ return iter.i++; },
{init: init_iter, init_parent: init_iter},
states);
};
E.while = function(cond, states){ return E.for(cond, null, states); };
// all([opt, ]a_or_o)
E.all = function(a_or_o, ao2){
var i, j, opt = {};
if (ao2)
{
opt = a_or_o;
a_or_o = ao2;
}
if (Array.isArray(a_or_o))
{
var a = Array.from(a_or_o);
i = 0;
return etask({name: 'all_a', cancel: true}, [function(){
for (j=0; j<a.length; j++)
this.spawn(a[j]);
}, function try_catch$loop(){
if (i>=a.length)
return this.return(a);
this.info.at = 'at '+i+'/'+a.length;
var _a = a[i];
if (_a instanceof Etask)
_a.spawn_parent();
return _a;
}, function(res){
if (this.error)
{
if (!opt.allow_fail)
return this.throw(this.error);
res = E.err(this.error);
}
a[i] = res;
i++;
return this.goto('loop');
}]);
}
else if (a_or_o instanceof Object)
{
var keys = Object.keys(a_or_o), o = {};
i = 0;
return etask({name: 'all_o', cancel: true}, [function(){
for (j=0; j<keys.length; j++)
this.spawn(a_or_o[keys[j]]);
}, function try_catch$loop(){
if (i>=keys.length)
return this.return(o);
var _i = keys[i], _a = a_or_o[_i];
this.info.at = 'at '+_i+' '+i+'/'+keys.length;
if (_a instanceof Etask)
_a.spawn_parent();
return _a;
}, function(res){
if (this.error)
{
if (!opt.allow_fail)
return this.throw(this.error);
res = E.err(this.error);
}
o[keys[i]] = res;
i++;
return this.goto('loop');
}]);
}
assert(0, 'invalid type');
};
E.all_limit = function(limit, arr_iter, cb){
var at = 0;
var iter = !Array.isArray(arr_iter) ? arr_iter : function(){
if (at<arr_iter.length)
return cb.call(this, arr_iter[at++]);
};
return etask({name: 'all_limit', cancel: true}, [function(){
var next;
if (!(next = iter.call(this)))
return this.goto('done');
this.spawn(next);
this.loop();
if (this.child.size>=limit)
return this.wait_child('any');
}, function done(){
return this.wait_child('all');
}]);
};
// _apply(opt, func[, _this], args)
// _apply(opt, object, method, args)
E._apply = function(opt, func, _this, args){
var func_name;
if (typeof _this=='string') // class with '.method' string call
{
assert(_this[0]=='.', 'invalid method '+_this);
var method = _this.slice(1), _class = func;
func = _class[method];
_this = _class;
assert(_this instanceof Object, 'invalid method .'+method);
func_name = method;
}
else if (Array.isArray(_this) && !args)
{
args = _this;
_this = null;
}
opt.name = opt.name||func_name||func.name;
return etask(opt, [function(){
var et = this, ret_sync, returned = 0;
args = Array.from(args);
args.push(function cb(err, res){
if (typeof opt.ret_sync=='string' && !returned)
{
// hack to wait for result
var a = arguments;
returned++;
return void E.nextTick(function(){ cb.apply(null, a); });
}
var nfn = opt.nfn===undefined || opt.nfn ? 1 : 0;
if (opt.ret_o)
{
var o = {}, i;
if (Array.isArray(opt.ret_o))
{
for (i=0; i<opt.ret_o.length; i++)
o[opt.ret_o[i]] = arguments[i+nfn];
}
else if (typeof opt.ret_o=='string')
o[opt.ret_o] = array.slice(arguments, nfn);
else
assert(0, 'invalid opt.ret_o');
if (typeof opt.ret_sync=='string')
o[opt.ret_sync] = ret_sync;
res = o;
}
else if (opt.ret_a)
res = array.slice(arguments, nfn);
else if (!nfn)
res = err;
et.continue(nfn ? E.err_res(err, res) : res);
});
ret_sync = func.apply(_this, args);
if (Array.isArray(opt.ret_sync))
opt.ret_sync[0][opt.ret_sync[1]] = ret_sync;
returned++;
return this.wait();
}]);
};
// nfn_apply([opt, ]object, method, args)
// nfn_apply([opt, ]func, this, args)
E.nfn_apply = function(opt, func, _this, args){
var _opt = {nfn: 1, cancel: 1};
if (typeof opt=='function' || typeof func=='string')
{
args = _this;
_this = func;
func = opt;
opt = _opt;
}
else
opt = assign(_opt, opt);
return E._apply(opt, func, _this, args);
};
// cb_apply([opt, ]object, method, args)
// cb_apply([opt, ]func, this, args)
E.cb_apply = function(opt, func, _this, args){
var _opt = {nfn: 0};
if (typeof opt=='function' || typeof func=='string')
{
args = _this;
_this = func;
func = opt;
opt = _opt;
}
else
opt = assign(_opt, opt);
return E._apply(opt, func, _this, args);
};
E.prototype.continue_nfn = function(){
return function(err, res){ this.continue(E.err_res(err, res)); }
.bind(this);
};
E.augment = function(_prototype, method, e_method){
var i, opt = {};
if (method instanceof Object && !Array.isArray(method))
{
assign(opt, method);
method = arguments[2];
e_method = arguments[3];
}
if (Array.isArray(method))
{
if (e_method)
opt.prefix = e_method;
for (i=0; i<method.length; i++)
E.augment(_prototype, opt, method[i]);
return;
}
opt.prefix = opt.prefix||'e_';
if (!e_method)
e_method = opt.prefix+method;
var fn = _prototype[method];
_prototype[e_method] = function(){
return etask._apply({name: e_method, nfn: 1}, fn, this, arguments); };
};
E.wait = function(timeout){
return etask({name: 'wait', cancel: true},
[function(){ return this.wait(timeout); }]);
};
E.to_nfn = function(promise, cb, opt){
return etask({name: 'to_nfn', async: true}, [function try_catch$(){
return promise;
}, function(res){
var ret = [this.error];
if (opt && opt.ret_a)
ret = ret.concat(res);
else
ret.push(res);
cb.apply(null, ret);
}]);
};
function etask_fn(opt, states, push_this){
if (Array.isArray(opt) || typeof opt=='function')
{
states = opt;
opt = undefined;
}
var is_generator = typeof states=='function' &&
states.constructor.name=='GeneratorFunction';
return function(){
var _opt = assign({}, opt);
_opt.state0_args = Array.from(arguments);
if (push_this)
_opt.state0_args.unshift(this);
if (is_generator)
return E._generator(null, states, _opt);
return new Etask(_opt, states);
};
}
E.fn = function(opt, states){ return etask_fn(opt, states, false); };
E._fn = function(opt, states){ return etask_fn(opt, states, true); };
E._generator = function(gen, ctor, opt){
opt = opt||{};
opt.name = opt.name || ctor && ctor.name || 'generator';
if (opt.cancel===undefined)
opt.cancel = true;
var done;
return new Etask(opt, [function(){
this.generator = gen = gen||ctor.apply(this, opt.state0_args||[]);
this.generator_ctor = ctor;
return {ret: undefined, err: undefined};
}, function try_catch$loop(rv){
var res;
try { res = rv.err ? gen.throw(rv.err) : gen.next(rv.ret); }
catch(e){ return this.return(E.err(e)); }
if (res.done)
{
done = true;
return this.return(res.value);
}
return res.value;
}, function(ret){
return this.goto('loop', this.error ?
{ret: undefined, err: this.error} : {ret: ret, err: undefined});
}, function finally$(){
// https://kangax.github.io/compat-table/es6/#test-generators_%GeneratorPrototype%.return
// .return() supported only in node>=6.x.x
if (!done && gen && gen.return)
try { gen.return(); } catch(e){}
}]);
};
E.ef = function(err){ // error filter
if (zerr.on_exception)
zerr.on_exception(err);
return err;
};
// similar to setInterval
// opt==10000 (or opt.ms==10000) - call states every 10 seconds
// opt.mode=='smart' - default mode, like setInterval. If states take
// longer than 'ms' to execute, next execution is delayed.
// opt.mode=='fixed' - always sleep 10 seconds between states
// opt.mode=='spawn' - spawn every 10 seconds
E.interval = function(opt, states){
if (typeof opt=='number')
opt = {ms: opt};
if (opt.mode=='fixed')
{
return E.for(null, function(){ return etask.sleep(opt.ms); },
states);
}
if (opt.mode=='smart' || !opt.mode)
{
var now;
return E.for(function(){ now = Date.now(); return true; },
function(){
var delay = zutil.clamp(0, now+opt.ms-Date.now(), Infinity);
return etask.sleep(delay);
}, states);
}
if (opt.mode=='spawn')
{
var stopped = false;
return etask([function loop(){
etask([function try_catch$(){
return etask(states);
}, function(res){
if (!this.error)
return;
if (this.error.ebreak!==ebreak_obj)
return this.throw(this.error);
stopped = true;
}]);
}, function(){
if (stopped)
return this.return();
return etask.sleep(opt.ms);
}, function(){
if (stopped) // stopped during sleep by prev long iteration
return this.return();
return this.goto('loop');
}]);
}
throw new Error('unexpected mode '+opt.mode);
};
E._class = function(cls){
var proto = cls.prototype, keys = Object.getOwnPropertyNames(proto);
for (var i=0; i<keys.length; i++)
{
var key = keys[i];
var descr = Object.getOwnPropertyDescriptor(proto, key);
if (descr.get||descr.set)
continue;
var p = proto[key];
if (p && p.constructor && p.constructor.name=='GeneratorFunction')
proto[key] = E._fn(p);
}
return cls;
};
E.shutdown = function(){
var prev;
while (E.root.size)
{
var e = E.root.values().next().value;
if (e==prev)
{
assert(e.tm_completed);
zerr.zexit('etask root not removed after return - '+
'fix non-cancelable child etask');
}
prev = e;
e.return();
}
};
return Etask; }); }());