@screeps/backend/lib/utils.js

const _ = require('lodash');
const common = require('@screeps/common');
const config = common.configManager.config;
const C = config.common.constants;
const db = common.storage.db;
const env = common.storage.env;
const q = require('q');
const png = require('pngjs').PNG;
const fs = require('fs');
const path = require('path');

exports.roomNameFromXY = function(x,y) {
    if(x < 0) {
        x = 'W'+(-x-1);
    }
    else {
        x = 'E'+(x);
    }
    if(y < 0) {
        y = 'N'+(-y-1);
    }
    else {
        y = 'S'+(y);
    }
    return ""+x+y;
}

exports.roomNameToXY = function(name) {
    var [match,hor,x,ver,y] = name.match(/^(\w)(\d+)(\w)(\d+)$/);
    if(hor == 'W') {
        x = -x-1;
    }
    else {
        x = +x;
        //x--;
    }
    if(ver == 'N') {
        y = -y-1;
    }
    else {
        y = +y;
        //y--;
    }
    return [x,y];
}

exports.translateModulesFromDb = function(modules) {
    modules = modules || {};

    for(var key in modules) {
        var newKey = key.replace(/\$DOT\$/g, '.');
        newKey = newKey.replace(/\$SLASH\$/g, '/');
        newKey = newKey.replace(/\$BACKSLASH\$/g, '\\');
        if(newKey != key) {
            modules[newKey] = modules[key];
            delete modules[key];
        }
    }
    return modules;
};

exports.translateModulesToDb = function(modules) {
    modules = modules || {};

    for(var key in modules) {
        var newKey = key.replace(/\./g,'$DOT$');
        newKey = newKey.replace(/\//g,'$SLASH$');
        newKey = newKey.replace(/\\/g,'$BACKSLASH$');

        if(newKey[0] == '$') {
            delete modules[key];
            continue;
        }

        if(newKey != key) {
            modules[newKey] = modules[key];
            delete modules[key];
        }
    }

    if(!modules.main) {
        modules.main = '';
    }

    return modules;
};

exports.getUserWorldStatus = function(user) {
    return db['rooms.objects'].count({user: ""+user._id})
    .then((objectsCnt) => {
        if(!objectsCnt) {
            return {status: 'empty'};
        }
        return db['rooms.objects'].find({$and: [
            {user: ""+user._id},
            {type: {$in: ['spawn','controller']}}
        ]}).then((objects) => {
            var spawns = false;
            if(objects) {
                objects.forEach((i) => {
                    if (i.type == 'spawn') {
                        if(!_.any(objects, {type: 'controller', room: i.room, user: i.user})) {
                            return;
                        }
                        spawns = true;
                    }
                })
            }
            return {status: spawns ? 'normal' : 'lost'};
        })
    })
};


exports.respawnUser = function(userId) {

    return db['rooms.objects'].find({$and: [{user: ""+userId}, {energy: {$gt: 0}}]})
    .then((data) => {
        if(!data.length) {
            return q.when();
        }
        data = _.map(data, (i) => new Object({type: 'energy', room: i.room, x: i.x, y: i.y, energy: i.energy}));
        return db['rooms.objects'].insert(data);
    })
    .then(() => db['rooms.objects'].removeWhere({$and: [
        {user: ""+userId},
        {type: {$ne: 'controller'}}]}))

    .then(() => db['rooms.flags'].removeWhere({user: ""+userId}))

    .then(() => db['rooms.objects'].find({$and: [{user: ""+userId}, {type: 'controller'}]}))

    .then((controllers) => q.all(_.map(controllers, (i) => db.rooms.update({_id: i.room}, {$set: {status: 'normal'}}))))

    .then(() => db['rooms.objects'].update({$and: [{user: "" + userId}, {type: 'controller'}]}, {
        $set: {
            level: 0,
            hits: 0,
            hitsMax: 0,
            progress: 0,
            progressTotal: 0,
            user: null,
            downgradeTime: null,
            safeMode: null,
            safeModeAvailable: 0,
            safeModeCooldown: null
        }
    }));
};

exports.withHelp = function(array) {
    var fn = array[1];
    fn._help = array[0];
    return fn;
};

exports.generateCliHelp = function(prefix, container) {
    return `Available methods:\r\n`+Object.keys(container).filter(i => _.isFunction(container[i])).map(i => ' - ' + prefix + (container[i]._help || i)).join('\r\n');
};

exports.writePng = function(colors, width, height, filename) {

    var image = new png({width, height});

    for(var y=0; y<height; y++) {
        for(var x=0; x<width; x++) {
            var idx = (width*y + x) << 2;

            image.data[idx] = colors[y][x][0];
            image.data[idx+1] = colors[y][x][1];
            image.data[idx+2] = colors[y][x][2];
            image.data[idx+3] = colors[y][x][3] === undefined ? 255 : colors[y][x][3];
        }
    }

    var defer = q.defer();
    image.pack().pipe(fs.createWriteStream(filename)).on('finish', () => defer.resolve());
    return defer.promise;
};

exports.createTerrainColorsMap = function(terrain, zoomIn) {
    var colors = {},
        width = 50, height = 50;

    for(var y=0; y<height; y++) {
        if(zoomIn) {
            colors[y * 3] = {};
            colors[y * 3 + 1] = {};
            colors[y * 3 + 2] = {};
        }
        else {
            colors[y] = {};
        }
        for(var x=0; x<width; x++) {

            var color;
            if(common.checkTerrain(terrain,x,y,C.TERRAIN_MASK_WALL)) {
                color = [0,0,0];
            }
            else if(common.checkTerrain(terrain,x,y,C.TERRAIN_MASK_SWAMP)) {
                color = [35,37,19];
            }
            else if(x == 0 || y == 0 || x == 49 || y == 49) {
                color = [50,50,50];
            }
            else {
                color = [43,43,43];
            }
            if(zoomIn) {
                for (var dx = 0; dx < 3; dx++) {
                    for (var dy = 0; dy < 3; dy++) {
                        colors[y * 3 + dy][x * 3 + dx] = color;
                    }
                }
            }
            else {
                colors[y][x] = color;
            }
        }
    }

    return colors;
};

exports.writeTerrainToPng = function(terrain, filename, zoomIn) {

    var colors = exports.createTerrainColorsMap(terrain, zoomIn);
    return exports.writePng(colors, 50*(zoomIn?3:1), 50*(zoomIn?3:1), filename);
};

exports.loadBot = function(name) {
    var dir = config.common.bots[name];
    if(!dir) {
        throw new Error(`Bot AI with the name "${name}" doesn't exist`);
    }
    var stat = fs.statSync(dir);
    if (!stat.isDirectory()) {
        throw new Error(`"${dir}" is not a directory`);
    }
    fs.statSync(path.resolve(dir, 'main.js'));
    var files = fs.readdirSync(dir), modules = {};
    files.forEach(file => {
        var m = file.match(/^(.*)\.js$/);
        if(!m) {
            return;
        }
        modules[m[1]] = fs.readFileSync(path.resolve(dir, file), {encoding: 'utf8'});
    });
    return exports.translateModulesToDb(modules);
};

exports.reloadBotUsers = function(name) {

    return db.users.find({bot: name})
        .then(users => {
            if(!users.length) {
                return 'No bot players found';
            }
            var modules = exports.loadBot(name);
            var timestamp = Date.now();

            return db['users.code'].insert(users.map(i => ({
                user: i._id,
                branch: 't'+timestamp,
                timestamp,
                activeWorld: true,
                activeSim: true,
                modules
            })))
            .then(() => db['users.code'].removeWhere({$and: [
                {user: {$in: users.map(i => i._id)}},
                {branch: {$ne: 't'+timestamp}}
            ]}))
                .then(() => 'Reloaded scripts for users: '+users.map(i => i.username).join(', '));
        })
};