102 lines
3.0 KiB
JavaScript
102 lines
3.0 KiB
JavaScript
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}`);
|
|
}
|
|
}
|