balena-supervisor/src/utils.coffee
2015-10-13 13:27:58 -03:00

72 lines
2.4 KiB
CoffeeScript

Promise = require 'bluebird'
_ = require 'lodash'
fs = Promise.promisifyAll require 'fs'
config = require './config'
mixpanel = require 'mixpanel'
request = Promise.promisifyAll require 'request'
mixpanelClient = mixpanel.init(config.mixpanelToken)
exports.mixpanelProperties = mixpanelProperties =
username: require('/boot/config.json').username
exports.mixpanelTrack = (event, properties = {}) ->
# Allow passing in an error directly and having it assigned to the error property.
if properties instanceof Error
properties = error: properties
# If the properties has an error argument that is an Error object then it treats it nicely,
# rather than letting it become `{}`
if properties.error instanceof Error
properties.error =
message: properties.error.message
stack: properties.error.stack
console.log('Event:', event, JSON.stringify(properties))
# Mutation is bad, and it should feel bad
properties = _.assign(_.cloneDeep(properties), mixpanelProperties)
mixpanelClient.track(event, properties)
# Returns an array of the host's ip address(es) by parsing the host /proc/net/fib_trie
exports.findIpAddrs = ->
fs.readFileAsync('/mnt/fib_trie', 'utf8')
.then (fibtrie) ->
prevLine = ''
fibtrie.split('\n')
.map (line) ->
line = line.trim()
# We only care about LOCAL routes (not UNICAST or BROADCAST)
if line.match(/LOCAL$/)
# Then we make sure the previous line was an ending branch (and hence contains an IP - 127.0.0.0 has
# BROADCAST and LOCAL entries)
if prevLine.match(/^\|--/)
# Then we remove the ending branch bit
maybeAddr = prevLine.replace(/^\|--/, '').trim()
# And ignore loopback/docker interfaces.
# TODO: Docker interface can technically be on another address range if 172.17
if !maybeAddr.match(/^(127.0.0.1|172.17.)/)
ipAddr = maybeAddr
prevLine = line
return ipAddr
.filter(Boolean)
# Helps in blinking the LED from the given end point.
exports.blink = (ms = 200) ->
fs.writeFileAsync(config.ledFile, 1)
.delay(ms)
.then -> fs.writeFileAsync(config.ledFile, 0)
# Helps in checking connectivity by pseudo-pinging our endpoint.
exports.checkConnectivity = ->
# We avoid using ICMP as this traffic is sometimes restricted/dropped. Good
# ol' port 80 HTTP should always be available :-)
request
.getAsync(config.heartbeatEndpoint)
.spread (response) ->
return response.statusCode in [ 200, 304 ]
.catch ->
return false