kingraph/lib/join.js

Plots family trees using JavaScript and Graphviz

/**
 * Joins everything in `array` recursively.
 *
 *     join(['a', ['b', 'c']], { sep: ',' })
 *     => 'a,b,c'
 *
 * To make an indentation, make a `{indent: [...]}` object.
 *
 *     join(['a {', {indent: ['b', 'c']}, '}'], { sep: '\n' })
 *     => 'a {\n  b\n  c\n}'
 */

function join (array, options) {
    if (!options) options = {};

    var prefix = options.prefix || '';
    var indent = options.indent || '\t';

    return flatten(array)
    .map(frag => {
        if (!frag.indent) return prefix + frag;
        return join(frag.indent, Object.assign({}, options, { prefix: prefix + indent }));
    })
    .join(options && options.sep || '\n');
}

function flatten (array) {
    if (array === null || array === undefined || array === false) return [];
    if (!Array.isArray(array)) return [array];

    return array.reduce((acc, frag) => {
        return acc.concat(flatten(frag));
    }, []);
}

module.exports = join;