var harvester = require('harvester');

var base = Game.spawns.Spawn1;
var source = base.room.find(Game.SOURCES)[0];
var construction_sites = base.room.find(Game.CONSTRUCTION_SITES);
var enemies = base.room.find(Game.HOSTILE_CREEPS);
var outpost = Game.flags.Flag1;

for(var creep_name in Game.creeps) {
    var creep = Game.creeps[creep_name];
    var role = creep.memory.role;

    if(role == "harvester") {
        harvester(creep, source, base);
    } else if (role == "builder") {
        if(creep.energy === 0) {
            creep.moveTo(base);
            base.transferEnergy(creep);
        } else if(construction_sites.length) {
            creep.moveTo(construction_sites[0]);
            creep.build(construction_sites[0]);
        }
    } else if (role == "guard") {
        if (enemies.length) {
            creep.moveTo(enemies[0]);
            creep.attack(enemies[0]);
        } else {
            creep.moveTo(outpost);
        }
    } else if (role == "healer") {

    }
}