akurath/core/cb.shells/main.js

IDE Frontend for codepsaces.io

// Requires
var Q = require('q');
var _ = require('lodash');
var path = require('path');
var os = require('os');
var shux = require('shux');

var utils = require('../utils');


function setup(options, imports, register) {
    var events = imports.events;
    var workspace = imports.workspace;

    // Construct
    var manager = shux();
    var oldCreateShell = manager.createShell.bind(manager);

    // Monkey patch createShell
    manager.createShell = function(id, opts) {  
        opts = opts || {};
        return (
            (opts.command || opts.args)?
                Q(opts.command) :
                manager.getDefaultShell()
        )
        .then(function(shellCmd) {
            // Build opts
            opts = _.defaults(opts, {
                cwd: "/"
            });
            opts.command = shellCmd;
            opts.cwd = path.join(workspace.root, opts.cwd || "/");
            return opts;
        })
        .then(function(opts) {
            return oldCreateShell(id, opts);
        });
    };

    // Simplify shell creation
    manager.createShellCommand = function(shellId, args, opts) {
        var exitCMD = "bash";

        // Spawn shell
        return manager.createShell(_.defaults({}, opts || {}, {
            id: shellId,
            command: 'bash',
            arguments: [
                '-c',
                (_.isString(args) ?
                    [args] :
                    (args || [])
                ).concat([';', exitCMD]).join(' ')
            ],
            cwd: workspace.root
        }));
    };

    // Get the systems default shell
    // this is memoized to only be called once
    manager.getDefaultShell = _.memoize(function() {
        var cmd = {
            'linux': 'getent passwd $USER | cut -d: -f7',
            'darwin': 'finger $USER | grep Shell | cut -d ":" -f3 | cut -d " " -f2'
        }[os.platform()];

        // If on unknown platform
        if(!cmd || !process.env.USER) {
            return Q('bash');
        }

        return utils.exec(cmd).get('stdout')
        .then(function(shellCmd) {
            return shellCmd.trim();
        }, function() {
            // Fallback to bash on failure
            return 'bash';
        });
    });

    // Pre cache call to getDefaultShell
    manager.getDefaultShell();

    // Utility function for connecting emitter to eventbus
    // (converts arguments to data convention)
    var emit = function shellEmit(eventId, shellId) {
        return events.emit(eventId, {
            shellId: shellId
        });
    };

    // Connect shell manger to event bus
    manager.on('exit', _.partial(emit, 'shell.exit'));
    manager.on('spawn', _.partial(emit, 'shell.spawn'));

    manager.on('attach', _.partial(emit, 'shell.attach'));
    manager.on('detach', _.partial(emit, 'shell.detach'));

    // Register
    register(null, {
        "shells": manager
    });
}

// Exports
module.exports = setup;