Blame view

node_modules/are-we-there-yet/tracker-group.js 3.16 KB
aaac7fed   liuqimichale   add
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
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
  'use strict'
  var util = require('util')
  var TrackerBase = require('./tracker-base.js')
  var Tracker = require('./tracker.js')
  var TrackerStream = require('./tracker-stream.js')
  
  var TrackerGroup = module.exports = function (name) {
    TrackerBase.call(this, name)
    this.parentGroup = null
    this.trackers = []
    this.completion = {}
    this.weight = {}
    this.totalWeight = 0
    this.finished = false
    this.bubbleChange = bubbleChange(this)
  }
  util.inherits(TrackerGroup, TrackerBase)
  
  function bubbleChange (trackerGroup) {
    return function (name, completed, tracker) {
      trackerGroup.completion[tracker.id] = completed
      if (trackerGroup.finished) return
      trackerGroup.emit('change', name || trackerGroup.name, trackerGroup.completed(), trackerGroup)
    }
  }
  
  TrackerGroup.prototype.nameInTree = function () {
    var names = []
    var from = this
    while (from) {
      names.unshift(from.name)
      from = from.parentGroup
    }
    return names.join('/')
  }
  
  TrackerGroup.prototype.addUnit = function (unit, weight) {
    if (unit.addUnit) {
      var toTest = this
      while (toTest) {
        if (unit === toTest) {
          throw new Error(
            'Attempted to add tracker group ' +
            unit.name + ' to tree that already includes it ' +
            this.nameInTree(this))
        }
        toTest = toTest.parentGroup
      }
      unit.parentGroup = this
    }
    this.weight[unit.id] = weight || 1
    this.totalWeight += this.weight[unit.id]
    this.trackers.push(unit)
    this.completion[unit.id] = unit.completed()
    unit.on('change', this.bubbleChange)
    if (!this.finished) this.emit('change', unit.name, this.completion[unit.id], unit)
    return unit
  }
  
  TrackerGroup.prototype.completed = function () {
    if (this.trackers.length === 0) return 0
    var valPerWeight = 1 / this.totalWeight
    var completed = 0
    for (var ii = 0; ii < this.trackers.length; ii++) {
      var trackerId = this.trackers[ii].id
      completed += valPerWeight * this.weight[trackerId] * this.completion[trackerId]
    }
    return completed
  }
  
  TrackerGroup.prototype.newGroup = function (name, weight) {
    return this.addUnit(new TrackerGroup(name), weight)
  }
  
  TrackerGroup.prototype.newItem = function (name, todo, weight) {
    return this.addUnit(new Tracker(name, todo), weight)
  }
  
  TrackerGroup.prototype.newStream = function (name, todo, weight) {
    return this.addUnit(new TrackerStream(name, todo), weight)
  }
  
  TrackerGroup.prototype.finish = function () {
    this.finished = true
    if (!this.trackers.length) this.addUnit(new Tracker(), 1, true)
    for (var ii = 0; ii < this.trackers.length; ii++) {
      var tracker = this.trackers[ii]
      tracker.finish()
      tracker.removeListener('change', this.bubbleChange)
    }
    this.emit('change', this.name, 1, this)
  }
  
  var buffer = '                                  '
  TrackerGroup.prototype.debug = function (depth) {
    depth = depth || 0
    var indent = depth ? buffer.substr(0, depth) : ''
    var output = indent + (this.name || 'top') + ': ' + this.completed() + '\n'
    this.trackers.forEach(function (tracker) {
      if (tracker instanceof TrackerGroup) {
        output += tracker.debug(depth + 1)
      } else {
        output += indent + ' ' + tracker.name + ': ' + tracker.completed() + '\n'
      }
    })
    return output
  }