unleash.unleash/migrations/20161212101749-better-strategy-parameter-definitions.js

/* eslint camelcase: "off" */

'use strict';

const async = require('async');

exports.up = function(db, callback) {
    const populateNewData = cb => {
        db.all(
            'select name, parameters_template from strategies',
            (err, results) => {
                const updateSQL = results
                    .map(({ name, parameters_template }) => {
                        const parameters = [];
                        Object.keys(parameters_template || {}).forEach(p => {
                            parameters.push({
                                name: p,
                                type: parameters_template[p],
                                description: '',
                                required: false,
                            });
                        });
                        return { name, parameters };
                    })
                    .map(
                        strategy => `
                UPDATE strategies 
                SET parameters='${JSON.stringify(strategy.parameters)}'
                WHERE name='${strategy.name}';`,
                    )
                    .join('\n');

                db.runSql(updateSQL, cb);
            },
        );
    };

    async.series(
        [
            db.addColumn.bind(db, 'strategies', 'parameters', { type: 'json' }),
            populateNewData.bind(db),
            db.removeColumn.bind(db, 'strategies', 'parameters_template'),
        ],
        callback,
    );
};

exports.down = function(db, callback) {
    const populateOldData = cb => {
        db.all('select name, parameters from strategies', (err, results) => {
            const updateSQL = results
                .map(({ name, parameters }) => {
                    const parameters_template = {};
                    parameters.forEach(p => {
                        parameters_template[p.name] = p.type;
                    });

                    return { name, parameters_template };
                })
                .map(
                    strategy => `
                UPDATE strategies 
                SET parameters_template='${JSON.stringify(
                    strategy.parameters_template,
                )}'
                WHERE name='${strategy.name}';`,
                )
                .join('\n');

            db.runSql(updateSQL, cb);
        });
    };

    async.series(
        [
            db.addColumn.bind(db, 'strategies', 'parameters_template', {
                type: 'json',
            }),
            populateOldData.bind(db),
            db.removeColumn.bind(db, 'strategies', 'parameters'),
        ],
        callback,
    );
};