Mirroar/hivemind

import StructureSource from 'dispatcher/resource-source/structure';

interface LinkSourceTask extends StructureSourceTask {
    type: 'link';
    target: Id<StructureLink>;
}

export default class LinkSource extends StructureSource<LinkSourceTask> {
    constructor(readonly room: Room) {
        super(room);
    }

    getType(): 'link' {
        return 'link';
    }

    getHighestPriority() {
        return 5;
    }

    getTasks(context: ResourceSourceContext) {
        if (!this.room.linkNetwork) return [];
        if (context.resourceType && context.resourceType !== RESOURCE_ENERGY) return [];

        return this.cacheEmptyTaskListFor('', 25, () => {
            const options: LinkSourceTask[] = [];

            for (const link of this.room.linkNetwork.neutralLinks) {
                if (link.store[RESOURCE_ENERGY] === 0) continue;

                const option: LinkSourceTask = {
                    priority: 5,
                    weight: link.store[RESOURCE_ENERGY] / 100,
                    type: this.getType(),
                    target: link.id,
                    resourceType: RESOURCE_ENERGY,
                };

                if (context.creep.pos.getRangeTo(link) > 10) {
                    // Don't go out of your way to empty the link, do it when nearby, e.g. at storage.
                    option.priority--;
                }

                option.priority -= this.room.getCreepsWithOrder(this.getType(), link.id).length * 2;
                option.priority -= this.room.getCreepsWithOrder('getEnergy', link.id).length * 2;
                option.priority -= this.room.getCreepsWithOrder('getResource', link.id).length * 2;

                options.push(option);
            }

            return options;
        });
    }

    isValid(task: LinkSourceTask, context: ResourceSourceContext) {
        if (!super.isValid(task, context)) return false;
        if (!this.room.linkNetwork) return false;
        if (this.room.linkNetwork.energy <= this.room.linkNetwork.maxEnergy) return false;

        return true;
    }
}