git ssb

1
var fs = require('fs')
2
var path = require('path')
3
var pull = require('pull-stream')
4
var File = require('pull-file')
5
var Split = require('pull-split')
6
var PNG = require('pngjs').PNG
7
var common = require('./common')
8
9
var start = new Date('2015-08-05').getTime()
10
var startDay = Math.floor(start / 86400000)
11
var startWeek = Math.floor(start / 86400000 / 7)
12
var startMonth = Math.floor(start / 86400000 / 30.43666666666666666666)
13
var startT
14
15
var now = Date.now()
16
var nowDay = Math.floor(now / 86400000)
17
var nowWeek = Math.floor(now / 86400000 / 7)
18
var nowMonth = Math.floor(now / 86400000 / 30.43666666666666666666)
19
20
var series = process.argv[2]
21
if (!series) throw new TypeError('Missing series argument')
22
23
if (series === 'day') {
24
  width = nowDay - startDay
25
  startT = startDay
26
} else if (series === 'week') {
27
  width = nowWeek - startWeek
28
  startT = startWeek
29
} else if (series === 'month') {
30
  width = nowMonth - startMonth
31
  startT = startMonth
32
} else {
33
  throw new TypeError('Invalid series')
34
}
35
36
var info = JSON.parse(fs.readFileSync(path.join(__dirname, series + '.json')))
37
var ids = JSON.parse(fs.readFileSync(path.join(__dirname, 'ids.json')))
38
39
var width
40
var height = info.max
41
42
var png = new PNG({
43
  width: width,
44
  height: height,
45
  colorType: 6
46
})
47
48
var userColors = {}
49
var userNumbers = {}
50
51
for (var i = 0; i < ids.length; i++) {
52
  var id = ids[i]
53
  userNumbers[id] = i
54
  var x = i / ids.length
55
  var colorI = Math.floor(x*20)
56
  var inbetweenAmount = x*20 - colorI
57
  var c = common.interpolate(common.colors20[colorI], common.colors20[colorI+1], inbetweenAmount)
58
  c[3] = 255
59
  userColors[id] = c
60
}
61
62
function setPixel(data, x, y, color) {
63
  var i = (y * width + x) << 2
64
  data[i] = color[0]
65
  data[i+1] = color[1]
66
  data[i+2] = color[2]
67
  data[i+3] = color[3]
68
}
69
70
function compareIds(a, b) {
71
  return userNumbers[a] - userNumbers[b]
72
}
73
74
pull(
75
  File(path.join(__dirname, series + '.jsonp')),
76
  Split(),
77
  pull.filter(Boolean),
78
  pull.map(JSON.parse),
79
  pull.drain(function (interval) {
80
    var x = interval.time - startT
81
    var ids = interval.ids.sort(compareIds)
82
    for (var i = 0; i < ids.length; i++) {
83
      var id = ids[i]
84
      var y = height - i
85
      var color = userColors[id] || [0, 0, 0, 255]
86
      setPixel(png.data, x, y, color)
87
    }
88
  }, function (err) {
89
    if (err) throw err
90
    png.pack().pipe(fs.createWriteStream(series + '.png'))
91
  })
92
)
93