tmp.remoteHarvest.js

const roads = require('plan.roads');
const reserver = require('role2.reserver');
const harvester = require('role2.dropHarvester')
const relocater = require('role.relocater');
const builder = require('role2.builder');
const hauler = require('role2.hauler');

const SPAWN = Game.spawns.Spawn1;
const destRoom = ((SPAWN || {}).room || {}).name;

module.exports = {
reserve: function(roomName, spawn=SPAWN) {
    return reserver.spawn(spawn, roomName);
},
harvest: function(roomName, spawn=SPAWN) {
    return harvester.spawnRemote(spawn, roomName);
},
build: function(roomName, spawn=SPAWN) {
    var mem = {};
    relocater.setMem(mem, roomName, builder.ROLE);
    const body = builder.mkBody(spawn.room.energyAvailable);
    const name = `${builder.ROLE}-${roomName}-${Game.time}`;
    return spawn.spawnCreep(body, name, {memory: mem});
},
haul: function(roomName, spawn=SPAWN) {
    return hauler.spawnRemote(spawn, roomName, destRoom);
},
roads: function(roomName, spawn=SPAWN) {
    // XXX calculate this stuff.
    /*
    const spawnPos = spawn.pos;
    const store = spawn.room.storage.pos;
    
    const resCtrl = new RoomPosition(7, 14, destRoom);
    roads.plan(spawnPos, resCtrl);
    
    const srcOne = new RoomPosition(28, 39, destRoom);
    roads.plan(spawnPos, srcOne);
    roads.plan(srcOne, store);
    
    const srcTwo = new RoomPosition(42, 40, destRoom);
    roads.plan(spawnPos, srcTwo);
    */
},
run: function(roomName) {
    // it's a bit tricky to count what's still spawning or headed to the room
    // so assume this is only run every 500 ticks or so...
    // if(Game.time % 500 != 0) return false;
    
    var spawns = _.filter(Game.spawns, s => {return !s.spawning && s.room.energyAvailable > 1000});
    if(spawns.length == 0) return;
    
    const room = Game.rooms[roomName];
    if(room) {
        // XXX Look for all creeps *assigned* to the room, not just the ones in it.
        const kinds = _.groupBy(room.find(FIND_MY_CREEPS).filter(c => {
            if(c.memory.role == reserver.ROLE) return c.ticksToLive > 100;
            return c.ticksToLive > 600;
        }), c => c.memory.role);
        const numRole = role => (kinds[role] || []).length;
        console.log(`remoteHarvest ${roomName} state:`, JSON.stringify(_.mapValues(kinds, (v, k) => numRole(k))));
        const ctrl = room.controller;
        if(numRole(reserver.ROLE) < 1 && !ctrl.reservation || ctrl.reservation.ticksToEnd < 300) {
            let spawn = spawns.shift();
            let ret = module.exports.reserve(roomName, spawn);
            console.log(`${spawn} spawning more reservers: ${ret}`);
            if(ret != OK) spawns.push(spawn);
            if(spawns.length == 0) return;
        }
        // would be nice to send (especially) harvesters in before the old ones expire.
        if(numRole(harvester.ROLE) < 2) {
            let spawn = spawns.shift();
            let ret = module.exports.harvest(roomName, spawn);
            console.log(`${spawn} spawning more harvesters: ${ret}`);
            if(ret != OK) spawns.push(spawn);
            if(spawns.length == 0) return;
        }
        // TODO(baptr): Check these from Memory.creeps instead of room.creeps
        if(numRole(hauler.ROLE) < 2) {
            let spawn = spawns.shift();
            let ret = module.exports.haul(roomName, spawn);
            console.log(`${spawn} spawning more haulers: ${ret}`);
            if(ret != OK) spawns.push(spawn);
            if(spawns.length == 0) return;
        }
        /*
        if(numRole(builder.ROLE) < 1) {
            let spawn = spawns.shift();
            let ret = module.exports.build(roomName, spawn);
            console.log(`${spawn} spawning more builders: ${ret}`);
            if(ret != OK) spawns.push(spawn);
            if(spawns.length == 0) return;
        }
        */
    } else {
        module.exports.reserve(roomName);
    }
}
};