formatting

Author: tedivm

Diff for: js/screepsstats.js

@@ -1,10 +1,10 @@
 var ScreepsStats = function () {
-  if(!Memory.___screeps_stats) {
+  if (!Memory.___screeps_stats) {
     Memory.___screeps_stats = {}
   }
   this.username = _.get(
-    _.find(Game.structures,(s) => true),'owner.username',
-    _.get(_.find(Game.creeps,(s) => true),'owner.username')
+    _.find(Game.structures, (s) => true), 'owner.username',
+    _.get(_.find(Game.creeps, (s) => true), 'owner.username')
   ) || false
   this.clean()
 }
@@ -58,19 +58,21 @@ ScreepsStats.prototype.runBuiltinStats = function () {
     }
   })
 
-  _.forEach(Game.rooms,(room) => {
+  _.forEach(Game.rooms, (room) => {
 
-    if(!stats[room.name]) {
+    if (!stats[room.name]) {
       stats.rooms[room.name] = {}
     }
 
-    if (_.isEmpty(room.controller)) { return }
+    if (_.isEmpty(room.controller)) {
+      return
+    }
     var controller = room.controller
 
     // Is hostile room? Continue
-    if(!controller.my) {
-      if(!!controller.owner) { // Owner is set but is not this user.
-        if(controller.owner.username != this.username) {
+    if (!controller.my) {
+      if (!!controller.owner) { // Owner is set but is not this user.
+        if (controller.owner.username != this.username) {
           return
         }
       } else {
@@ -83,20 +85,20 @@ ScreepsStats.prototype.runBuiltinStats = function () {
       level: controller.level,
       progress: controller.progress,
       upgradeBlocked: controller.upgradeBlocked,
-      reservation: _.get(controller,'reservation.ticksToEnd'),
+      reservation: _.get(controller, 'reservation.ticksToEnd'),
       ticksToDowngrade: controller.ticksToDowngrade
     })
 
-    if(controller.level > 0) {
+    if (controller.level > 0) {
 
       // Room
-      _.merge(stats.rooms[room.name],{
+      _.merge(stats.rooms[room.name], {
         energyAvailable: room.energyAvailable,
         energyCapacityAvailable: room.energyCapacityAvailable,
       })
 
       // Storage
-      if(room.storage) {
+      if (room.storage) {
         _.defaults(stats, {
           storage: {
             subgroups: true
@@ -107,14 +109,14 @@ ScreepsStats.prototype.runBuiltinStats = function () {
           store: _.sum(room.storage.store),
           resources: {}
         }
-        for(var resourceType in room.storage.store) {
+        for (var resourceType in room.storage.store) {
           stats.storage[room.storage.id].resources[resourceType] = room.storage.store[resourceType]
           stats.storage[room.storage.id][resourceType] = room.storage.store[resourceType]
         }
       }
 
       // Terminals
-      if(room.terminal) {
+      if (room.terminal) {
         _.defaults(stats, {
           terminal: {
             subgroups: true
@@ -125,7 +127,7 @@ ScreepsStats.prototype.runBuiltinStats = function () {
           store: _.sum(room.terminal.store),
           resources: {}
         }
-        for(var resourceType in room.terminal.store) {
+        for (var resourceType in room.terminal.store) {
           stats.terminal[room.terminal.id].resources[resourceType] = room.terminal.store[resourceType]
           stats.terminal[room.terminal.id][resourceType] = room.terminal.store[resourceType]
         }
@@ -144,11 +146,11 @@ ScreepsStats.prototype.runBuiltinStats = function () {
       subgroups: true
     }
   })
-  _.forEach(Game.spawns, function(spawn) {
+  _.forEach(Game.spawns, function (spawn) {
     stats.spawns[spawn.name] = {
       room: spawn.room.name,
       busy: !!spawn.spawning,
-      remainingTime: _.get(spawn,'spawning.remainingTime',0)
+      remainingTime: _.get(spawn, 'spawning.remainingTime', 0)
     }
   })
 
@@ -174,16 +176,16 @@ ScreepsStats.prototype.roomExpensive = function (stats, room) {
   stats.rooms[room.name].sources = {}
   var sources = room.find(FIND_SOURCES)
 
-  _.forEach(sources,(source) => {
+  _.forEach(sources, (source) => {
     stats.sources[source.id] = {
       room: room.name,
       energy: source.energy,
       energyCapacity: source.energyCapacity,
       ticksToRegeneration: source.ticksToRegeneration
     }
-    if(source.energy < source.energyCapacity && source.ticksToRegeneration) {
+    if (source.energy < source.energyCapacity && source.ticksToRegeneration) {
       var energyHarvested = source.energyCapacity - source.energy
-      if(source.ticksToRegeneration < ENERGY_REGEN_TIME) {
+      if (source.ticksToRegeneration < ENERGY_REGEN_TIME) {
         var ticksHarvested = ENERGY_REGEN_TIME - source.ticksToRegeneration
         stats.sources[source.id].averageHarvest = energyHarvested / ticksHarvested
       }
@@ -198,7 +200,7 @@ ScreepsStats.prototype.roomExpensive = function (stats, room) {
   // Mineral Mining
   var minerals = room.find(FIND_MINERALS)
   stats.rooms[room.name].minerals = {}
-  _.forEach(minerals,(mineral) => {
+  _.forEach(minerals, (mineral) => {
     stats.minerals[mineral.id] = {
       room: room.name,
       mineralType: mineral.mineralType,
@@ -212,8 +214,8 @@ ScreepsStats.prototype.roomExpensive = function (stats, room) {
   // Hostiles in Room
   var hostiles = room.find(FIND_HOSTILE_CREEPS)
   stats.rooms[room.name].hostiles = {}
-  _.forEach(hostiles,(hostile) => {
-    if(!stats.rooms[room.name].hostiles[hostile.owner.username]) {
+  _.forEach(hostiles, (hostile) => {
+    if (!stats.rooms[room.name].hostiles[hostile.owner.username]) {
       stats.rooms[room.name].hostiles[hostile.owner.username] = 1
     } else {
       stats.rooms[room.name].hostiles[hostile.owner.username]++
@@ -226,14 +228,14 @@ ScreepsStats.prototype.roomExpensive = function (stats, room) {
 
 ScreepsStats.prototype.removeTick = function (tick) {
 
-  if(Array.isArray(tick)) {
-    for(var index in tick) {
+  if (Array.isArray(tick)) {
+    for (var index in tick) {
       this.removeTick(tick[index])
     }
     return 'ScreepStats: Processed ' + tick.length + ' ticks'
   }
 
-  if(!!Memory.___screeps_stats[tick]) {
+  if (!!Memory.___screeps_stats[tick]) {
     delete Memory.___screeps_stats[tick]
     return 'ScreepStats: Removed tick ' + tick
   } else {
@@ -242,19 +244,24 @@ ScreepsStats.prototype.removeTick = function (tick) {
 }
 
 ScreepsStats.prototype.getStats = function (json) {
-  if(json) {
+  if (json) {
     return JSON.stringify(Memory.___screeps_stats)
   } else {
     return Memory.__screeps_stats
   }
 }
 
 ScreepsStats.prototype.getStatsForTick = function (tick) {
-  if(!Memory.__screeps_stats[tick]) {
+  if (!Memory.__screeps_stats[tick]) {
     return false
   } else {
     return Memory.__screeps_stats[tick]
   }
 }
 
+if (ENABLE_PROFILER && !!Profiler) {
+  ScreepsStats.prototype = Profiler.registerObject(ScreepsStats, 'ScreepsStats')
+}
+
+
 module.exports = ScreepsStats