main.js

require('prototype.room');
require('prototype.structure');
require('prototype.creep');
require('constants');

const BaseController=require('controller.base');
const FarmController=require('controller.farm');
const TargetController=require('controller.target');

const HarvestTime = require('role.harvester');
const UpgradeTime = require('role.upgrader');
const BuildTime = require('role.builder');
const MineTime = require('role.miner');
const DeliveryTime = require('role.deliveryBoy');
const DistributeTime = require('role.distributor');
const ExtractTime = require('role.extractor');
const DefendTime = require('role.defender');
const TerminalTime = require('role.terminalWorker');

console.log('Script Reloaded');

//TODO: REPLACE SORT WITH FINDMIN/FINDMAX
module.exports.loop = function() {
    //run all creep AI
    for(let name in Memory.creeps){
        //Delete memory for creeps that no longer exist
        const creep=Game.creeps[name];
        if(!creep) {
            delete Memory.creeps[name];
            continue;
        }

        //Activate creep scripts as corresponding to their role
        creep.room.memory.clean=0;
        switch(creep.memory.role){
            case 'miner':
                MineTime.run(creep);
                break;
            case 'deliveryBoy':
                DeliveryTime.run(creep);
                break;
            case 'distributor':
                DistributeTime.run(creep);
                break;
            case 'harvester':
                HarvestTime.run(creep);
                break;
            case 'upgrader':
                UpgradeTime.run(creep);
                break;
            case 'builder':
                BuildTime.run(creep);
                break;
            case 'extractor':
                ExtractTime.run(creep);
                break;
            case 'defender':
                DefendTime.run(creep);
                break;
            case 'terminalWorker':
                TerminalTime.run(creep);
                break;
            default: //Reset Creeps role to be the beginning of their name
                let name=creep.name;
                creep.memory.role=name.substr(0,name.length-4);
                break;
            //TODO: other creeps that attack
        }
        creep.room.memory.clean=1;
    }

    const groupedCreeps=_.groupBy(Game.creeps,(creep)=>creep.memory.home);

    let isWorking=false;
    //Flag controllers
    for(let roomName in Memory.rooms){
        const room=Game.rooms[roomName];
        if(room) {
            isWorking=true;

            //If there is a script error, this will stop data from getting corrupted
            if(room.memory.clean===0){
                console.log('Room '+room.name+' had an error. Resetting memory');
                room.resetTempMemory();
            }
            room.memory.clean=0;

            //Run the controller for the correct room level (0 if it is a base w/ no spawn)
            if (room.memory.level >= 0) {
                BaseController.run(room,groupedCreeps[room.name]);
            }
            else if (room.memory.level === -1) {
                FarmController.run(room,groupedCreeps[room.name]);
            }
            else if (room.memory.level === -2) {
                TargetController.run(room,groupedCreeps[room.name]);
            }
            else{
                if(room.controller) {
                    if (room.controller.my) {
                        if(room.getSpawns().length===0)
                            room.memory.level=0;
                        else
                            room.memory.level = room.controller.level;
                    }
                    else {
                        room.memory.level = -2;
                    }
                }
                //TODO: define behavior for rooms with no controller (remove unnecessary rooms?)
            }

            room.resetTempMemory();
            room.memory.clean=1; //Flag that shows the room controller was cleared correctly
        }
    }
    if(!isWorking){//Should only be necessary at the very beginning of the script running
        const spawn=Game.spawns.Spawn1;
        if(spawn)
            spawn.room.memory.level=spawn.room.controller.level;
    }
};