stf-DeviceFarmer/res/app/components/stf/device/device-service.js

var oboe = require('oboe')
var _ = require('lodash')
var Promise = require('bluebird')

module.exports = function DeviceServiceFactory($rootScope, $http, socket) {
  var deviceService = {}

  function Tracker($scope, options) {
    var devices = []
      , devicesBySerial = Object.create(null)
      , scopedSocket = socket.scoped($scope)

    function notify() {
      // Not great. Consider something else
      if (!$scope.$$phase) {
        $scope.$digest()
      }
    }

    function get(data) {
      return devices[devicesBySerial[data.serial]]
    }

    function insert(data) {
      devicesBySerial[data.serial] = devices.push(data) - 1
      notify()
    }

    function modify(oldData, newData) {
      _.assign(oldData, newData)
      notify()
    }

    function remove(data) {
      var index = devicesBySerial[data.serial]
      if (index >= 0) {
        devices.splice(index, 1)
        delete devicesBySerial[data.serial]
        notify()
      }
    }

    scopedSocket.on('device.add', function (data) {
      var device = get(data)
      if (device) {
        modify(device, data)
      }
      else if (options.auto) {
        insert(data)
      }
    })

    scopedSocket.on('device.remove', function (data) {
      var device = get(data)
      if (device) {
        modify(device, data)
      }
    })

    scopedSocket.on('device.change', function (data) {
      var device = get(data)
      if (device) {
        modify(device, data)
      }
    })

    this.add = function(device) {
      remove(device)
      insert(device)
    }

    this.devices = devices
  }

  deviceService.trackAll = function ($scope) {
    var tracker = new Tracker($scope, {
      auto: true
    })

    oboe('/api/v1/devices')
      .node('devices[*]', function (device) {
        tracker.add(device)
      })

    return tracker
  }

  deviceService.get = function (serial, $scope) {
    var tracker = new Tracker($scope, {
      auto: false
    })

    return $http.get('/api/v1/devices/' + serial)
      .then(function (response) {
        tracker.add(response.data.device)
        return response.data.device
      })
  }

  return deviceService
}