screep/main.js

const init=require('init')
const harvester = require('harvester');
const picker = require('picker');
const util = require('utils');
const transfer = require('transfer');
module.exports.loop = function () {
    init.init();
    Creep.prototype.action = action;
    if ((_(Game.creeps).filter({ memory: { role: 'harvester' } }).size() + _(Game.creeps).filter({ memory: { role: 'picker' } }).size()) < 2) {
        createHarvestCreep();
    } else {
        if (_(Game.creeps).filter({ memory: { role: 'transfer' } }).size() < 2) {
            createTransferCreep();

        }
    }
    chanage_role();
    assign_task();

}

function action() {
    let role = this.memory.role;
    // this.say(`${role}`);
    switch (role) {
        case "harvester":
            harvester.run(this)
            break;
        case "picker":
            picker.run(this)
            break;
        case "transfer":
            transfer.run(this)
            break;
    }
}

function assign_task() {
    for (const creep_id in Game.creeps) {
        let creep = Game.creeps[creep_id];
        creep.action();
    }

}

function chanage_role() {
    if (_(Game.creeps).filter({ memory: { role: 'transfer' } }).size() > 0) {
        let harvester_array = _(Game.creeps).filter({ memory: { role: 'harvester' } }).value();
        // console.log(harvester_array[0].memory.role);
        for (let harvester_creep of harvester_array) {
            harvester_creep.memory.role = 'picker';
        }
    } else {
        let picker_array = _(Game.creeps).filter({ memory: { role: 'picker' } }).value();
        // console.log(harvester_array[0].memory.role);
        for (let picker_creep of picker_array) {
            picker_creep.memory.role = 'harvester';
        }
    }
}

function get_creep_with_name(inlcude_string) {
    let creeps = [];
    for (const creep_id in Game.creeps) {
        let creep = Game.creeps[creep_id];
        if (creep.name.include(inlcude_string)) {
            creeps.push(creep);
        }
    }
    return creeps;
}

function count(hash_container) {
    return Object.keys(hash_container).length
}

function createHarvestCreep() {
    //create new creep
    let new_creep_name = "AUTO_CREATE_Harvester_" + util.uuid();
    let Re_code;
    Re_code = Game.spawns['Spawn1'].createCreep([MOVE, CARRY, WORK], new_creep_name)
    if (Re_code == new_creep_name) {
        Game.creeps[new_creep_name].memory.role = 'harvester';
        console.log('Create new creep name ' + new_creep_name);
    } else {
        console.log(`failed to create harvester with code ${Re_code}`);
    }
}

function createTransferCreep() {
    //create new creep
    let new_creep_name = "AUTO_CREATE_Transfer_" + util.uuid();
    let Re_code;
    Re_code = Game.spawns['Spawn1'].createCreep([MOVE, CARRY, CARRY], new_creep_name);
    if (new_creep_name == Re_code) {
        Game.creeps[new_creep_name].memory.role = 'transfer';
        console.log('Create new creep name ' + new_creep_name);
    } else {
        console.log(`failed to create transfer with code ${Re_code}`);
    }
}