docs/src/callback.js


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69

function noop(){}

function chain(delegateFn, explicitDone){
  var onDoneFn = noop;
  var onErrorFn = function(e){
    console.error(e.stack || e);
    process.exit(-1);
  };
  var waitForCount = 1;
  delegateFn = delegateFn || noop;
  var stackError = new Error('capture stack');

  function decrementWaitFor() {
    waitForCount--;
    if (waitForCount == 0)
      onDoneFn();
  }

  function self(){
    try {
      return delegateFn.apply(self, arguments);
    } catch (error) {
      self.error(error);
    } finally {
      if (!explicitDone)
        decrementWaitFor();
    }
  };
  self.onDone = function(callback){
    onDoneFn = callback;
    return self;
  };
  self.onError = function(callback){
    onErrorFn = callback;
    return self;
  };
  self.waitFor = function(callback){
    if (waitForCount == 0)
      throw new Error("Can not wait on already called callback.");
    waitForCount++;
    return chain(callback).onDone(decrementWaitFor).onError(self.error);
  };

  self.waitMany = function(callback){
    if (waitForCount == 0)
      throw new Error("Can not wait on already called callback.");
    waitForCount++;
    return chain(callback, true).onDone(decrementWaitFor).onError(self.error);
  };

  self.done = function(callback){
    decrementWaitFor();
  };

  self.error = function(error) {
    var stack = stackError.stack.split(/\n\r?/).splice(2);
    var nakedStack = [];
    stack.forEach(function(frame){
      if (!frame.match(/callback\.js:\d+:\d+\)$/))
        nakedStack.push(frame);
    });
    error.stack = error.stack + '\nCalled from:\n' + nakedStack.join('\n');
    onErrorFn(error);
  };

  return self;
}

exports.chain = chain;