//Main Module ---
var harvestScript = require('roleHarvester');
var upgradeScript = require('upgrader');
var buildScript = require('builder');
var defenseScript = require('soldier');
var fixScript = require('repair');
var wallScript = require('wallmaster');

module.exports.loop = function () {
    //Clear dead memory
    for (let name in Memory.creeps) {
        if (Game.creeps[name] == undefined){
            console.log('Dead creeps found, cleaning up...')
            delete Memory.creeps[name]
        }
    }
    //Run Role Modules, Call a different script depending on role in memory
    for(let name in Game.creeps) {
        var creep = Game.creeps[name];
        if (creep.memory.role == 'harvester') {
            harvestScript.run(creep);
        }
        else if (creep.memory.role == 'upgrader') {
            upgradeScript.run(creep);
        }
        else if (creep.memory.role == 'builder'){
            buildScript.run(creep);
        }
        else if (creep.memory.role == 'fixer'){
            fixScript.run(creep);
        }
        else if (creep.memory.role == 'wall'){
            wallScript.run(creep);
        }
    }
    //Check for every combination of Roles and IDs, maintain populations
    var harID0 = _.sum(Game.creeps, (c) =>c.memory.role == 'harvester' && c.memory.ID == 0)
    var harID1 = _.sum(Game.creeps, (c) =>c.memory.role == 'harvester' && c.memory.ID == 1)
    var upID0 = _.sum(Game.creeps, (c) =>c.memory.role == 'upgrader' && c.memory.ID == 0)
    var buildID1 = _.sum(Game.creeps, (c) =>c.memory.role == 'builder' && c.memory.ID == 1)
    var buildID0 = _.sum(Game.creeps, (c) =>c.memory.role == 'builder' && c.memory.ID == 0)
    var fixID0 = _.sum(Game.creeps, (c) =>c.memory.role == 'fixer' && c.memory.ID == 0)
    var wallID1 = _.sum(Game.creeps, (c) =>c.memory.role == 'wall' && c.memory.ID == 1)
    if (harID0 < 3){
        Game.spawns.Spawn1.createCreep([WORK,WORK,WORK,CARRY,CARRY,MOVE,MOVE,MOVE], 'harvester' + Game.time, {role: 'harvester', ID: 0})
    }
    else if (harID1 < 2){
        Game.spawns.Spawn1.createCreep([WORK,WORK,WORK,CARRY,CARRY,MOVE,MOVE,MOVE], 'harvester' + Game.time, {role: 'harvester', ID: 1})
    }
    else if (upID0 < 5){
        Game.spawns.Spawn1.createCreep([WORK,WORK,WORK,CARRY,CARRY,MOVE,MOVE,MOVE], 'upgrader' + Game.time, {role: 'upgrader', ID: 0, upgrading: false})
    }
    else if (buildID1 < 1){
        Game.spawns.Spawn1.createCreep([WORK,WORK,WORK,CARRY,CARRY,MOVE,MOVE,MOVE], 'builder' + Game.time, {role: 'builder', ID: 1, building: false})
    }
    else if (buildID0 < 1){
        Game.spawns.Spawn1.createCreep([WORK,WORK,WORK,CARRY,CARRY,MOVE,MOVE,MOVE], 'builder' + Game.time, {role: 'builder', ID: 0, building: false})
    }
    else if (fixID0 < 1){
        Game.spawns.Spawn1.createCreep([WORK,WORK,WORK,CARRY,CARRY,MOVE,MOVE,MOVE], 'fixer' + Game.time, {role: 'fixer', ID: 0, fixing: false})
    }
    else if (wallID1 < 1){
        Game.spawns.Spawn1.createCreep([WORK,WORK,WORK,CARRY,CARRY,MOVE,MOVE,MOVE], 'wallMaster' + Game.time, {role: 'wall', ID: 1, fixing: false})
    }
}
//Builder module---
module.exports = {
    run: function(creep) {
        if (creep.carry.energy == creep.carryCapacity){
            creep.memory.building = true
            creep.say('Building')
        }
        else if (creep.carry.energy == 0){
            creep.memory.building = false
            creep.say('Collecting')
        }
        if (creep.memory.building == true){
            var buildSite = creep.pos.findClosestByPath(FIND_CONSTRUCTION_SITES);
            if (creep.build(buildSite) == ERR_NOT_IN_RANGE){
                creep.moveTo(buildSite)
            }
        }
        else if (creep.memory.ID == 0 && creep.memory.building == false){
            var source = creep.pos.findClosestByPath(FIND_SOURCES);
            if (creep.harvest(source) == ERR_NOT_IN_RANGE){
                creep.moveTo(source);
            }
        }
        else if (creep.memory.ID == 1 && creep.memory.building == false){
            var farSource = creep.pos.findClosestByPath(FIND_SOURCES)
            if (creep.harvest(farSource) == ERR_NOT_IN_RANGE){
                creep.moveTo(44,29);
            }
        }
    }
};
//Repair module---
module.exports = {
    run: function(creep) {
        if (creep.carry.energy == creep.carryCapacity){
            creep.memory.fixing = true
            creep.say('Repairing')
        }
        else if (creep.carry.energy == 0){
            creep.memory.fixing = false
            creep.say('Collecting')
        }
        if (creep.memory.fixing == true){
            var repairSite = creep.pos.findClosestByPath(FIND_STRUCTURES, {
                filter: (s) => s.hits < s.hitsMax && s.structureType != STRUCTURE_WALL
            });
            if (creep.repair(repairSite) == ERR_NOT_IN_RANGE){
                creep.moveTo(repairSite)
            }
        }
        else if (creep.memory.ID == 0 && creep.memory.fixing == false){
            var source = creep.pos.findClosestByPath(FIND_SOURCES);
            if (creep.harvest(source) == ERR_NOT_IN_RANGE){
                creep.moveTo(source);
            }
        }
        else if (creep.memory.ID == 1 && creep.memory.fixing == false){
            var farSource = creep.pos.findClosestByPath(FIND_SOURCES)
            if (creep.harvest(farSource) == ERR_NOT_IN_RANGE){
                creep.moveTo(44,29);
            }
        }
    }
};
//Harvester module ---
module.exports = {
    run: function(creep) {
        if (creep.carryCapacity == creep.carry.energy){
            var container = creep.pos.findClosestByPath(FIND_MY_STRUCTURES, {
                filter: (s) => s.energy < s.energyCapacity
            });
            if (container != undefined){
                if (creep.transfer(container, RESOURCE_ENERGY) == ERR_NOT_IN_RANGE){
                    creep.moveTo(container);
                }
            }
        }
        else if (creep.memory.ID == 0){
            var source = creep.pos.findClosestByPath(FIND_SOURCES);
            if (creep.harvest(source) == ERR_NOT_IN_RANGE){
                creep.moveTo(source);
            }
        }
        else if (creep.memory.ID == 1){
            var farSource = creep.pos.findClosestByPath(FIND_SOURCES)
            if (creep.harvest(farSource) == ERR_NOT_IN_RANGE){
                creep.moveTo(44,29);
            }
        }
    }
};
//Upgrader module ---
module.exports = {
    run: function(creep) {
        if (creep.carry.energy == creep.carryCapacity){
            creep.memory.upgrading = true
            creep.say('Upgrading')
        }
        else if (creep.carry.energy == 0){
            creep.memory.upgrading = false
            creep.say('Collecting')
        }
        if (creep.memory.upgrading == true){
            if (creep.upgradeController(creep.room.controller) == ERR_NOT_IN_RANGE){
                creep.moveTo(creep.room.controller);
            }
        }
        else if (creep.memory.ID == 0 && creep.memory.upgrading == false){
            var source = creep.pos.findClosestByPath(FIND_SOURCES);
            if (creep.harvest(source) == ERR_NOT_IN_RANGE){
                creep.moveTo(source);
            }
        }
        else if (creep.memory.ID == 1 && creep.memory.upgrading == false){
            var farSource = creep.pos.findClosestByPath(FIND_SOURCES)
            if (creep.harvest(farSource) == ERR_NOT_IN_RANGE){
                creep.moveTo(44,29);
            }
        }
    }
};
// Wall fortifier module ---
module.exports = {
    run: function(creep) {
        if (creep.carry.energy == creep.carryCapacity){
            creep.memory.fixing = true
            creep.say('Repairing Walls')
        }
        else if (creep.carry.energy == 0){
            creep.memory.fixing = false
            creep.say('Collecting')
        }
        if (creep.memory.fixing == true){
            var repairSite = creep.pos.findClosestByPath(FIND_STRUCTURES, {
                filter: (s) => s.structureType == STRUCTURE_WALL && s.hits < 300000
            });
            if (creep.repair(repairSite) == ERR_NOT_IN_RANGE){
                creep.moveTo(repairSite)
            }
        }
        else if (creep.memory.ID == 0 && creep.memory.fixing == false){
            var source = creep.pos.findClosestByPath(FIND_SOURCES);
            if (creep.harvest(source) == ERR_NOT_IN_RANGE){
                creep.moveTo(source);
            }
        }
        else if (creep.memory.ID == 1 && creep.memory.fixing == false){
            var farSource = creep.pos.findClosestByPath(FIND_SOURCES)
            if (creep.harvest(farSource) == ERR_NOT_IN_RANGE){
                creep.moveTo(44,29);
            }
        }
    }
};