Add role: builder

WilliamYuhangLee · Mar 28, 2019 · ada2c31 · ada2c31
1 parent 8fb0671
commit ada2c31
Show file tree

Hide file tree

Showing 4 changed files with 49 additions and 2 deletions.
diff --git a/main.js b/main.js
@@ -1,5 +1,6 @@
 var roleHarvester = require("role.harvester");
 var roleUpgrader = require("role.upgrader");
+var roleBuilder = require("role.builder");
 
 module.exports.loop = function () {
     for (let name in Game.creeps) {
@@ -8,6 +9,8 @@ module.exports.loop = function () {
             roleHarvester.run(creep);
         } else if (creep.memory.role === "upgrader") {
             roleUpgrader.run(creep);
+        } else if (creep.memory.role === "builder") {
+            roleBuilder.run(creep);
         }
     }
 };
diff --git a/role.builder.js b/role.builder.js
@@ -0,0 +1,34 @@
+/*
+ * Module code goes here. Use 'module.exports' to export things:
+ * module.exports.thing = 'a thing';
+ *
+ * You can import it from another modules like this:
+ * var mod = require('role.harvester');
+ * mod.thing == 'a thing'; // true
+ */
+var roleUpgrader = require("role.upgrader");
+
+module.exports = {
+    run: function (creep) {
+        if (creep.carry.energy === 0) {
+            creep.memory.collecting = true;
+        } else if (creep.carry.energy === creep.carryCapacity) {
+            creep.memory.collecting = false;
+        }
+        if (creep.memory.collecting) {
+            let source = creep.pos.findClosestByRange(FIND_SOURCES);
+            if (creep.harvest(source) === ERR_NOT_IN_RANGE) {
+                creep.moveTo(source);
+            }
+        } else {
+            let construction = creep.pos.findClosestByRange(FIND_CONSTRUCTION_SITES);
+            if (construction) {
+                if (creep.build(construction) === ERR_NOT_IN_RANGE) {
+                    creep.moveTo(construction);
+                }
+            } else {
+                roleUpgrader.run(creep);
+            }
+        }
+    }
+};
diff --git a/role.harvester.js b/role.harvester.js
@@ -9,7 +9,12 @@
 
 module.exports = {
     run: function (creep) {
-        if (creep.carry.energy < creep.carryCapacity) {
+        if (creep.carry.energy === 0) {
+            creep.memory.collecting = true;
+        } else if (creep.carry.energy === creep.carryCapacity) {
+            creep.memory.collecting = false;
+        }
+        if (creep.memory.collecting) {
             let source = creep.pos.findClosestByRange(FIND_SOURCES);
             if (creep.harvest(source) === ERR_NOT_IN_RANGE) {
                 creep.moveTo(source);

diff --git a/role.upgrader.js b/role.upgrader.js
@@ -9,7 +9,12 @@
 
 module.exports = {
     run: function (creep) {
-        if (creep.carry.energy < creep.carryCapacity) {
+        if (creep.carry.energy === 0) {
+            creep.memory.collecting = true;
+        } else if (creep.carry.energy === creep.carryCapacity) {
+            creep.memory.collecting = false;
+        }
+        if (creep.memory.collecting) {
             let source = creep.pos.findClosestByRange(FIND_SOURCES);
             if (creep.harvest(source) === ERR_NOT_IN_RANGE) {
                 creep.moveTo(source);