cloudcmd/lib/server/socket.js

/* module make possible connectoin thru socket.io on a server */

(function(){
    'use strict';
    
    var main            = global.cloudcmd.main,
        SRVDIR          = main.SRVDIR,
        
        io              = main.require('socket.io'),
        update          = main.srvrequire('update'),
        exec            = main.child_process.exec,
        Util            = main.util,
        
        ClientFuncs     = [],
        OnMessageFuncs  = [],
        INFO_LOG_LEVEL  = 2,
        Win32_b         = main.WIN32;
    
    /**
     * function listen on servers port
     * @pServer {Object} started server object
     */
    exports.listen = function(pServer){
        var lRet;
        if(io){
            io.set('log level', INFO_LOG_LEVEL);
            var lListen = io.listen(pServer),
                /* number of connections */
                lConnNum = 0;
            
            lRet = lListen.sockets.on('connection', function (socket){
                ++lConnNum;
                socket.send('{"stdout":"client connected"}');
                
                Util.log('server connected');
                
                if(!OnMessageFuncs[lConnNum])
                    OnMessageFuncs[lConnNum] = onMessage(lConnNum, socket);
                
                var lConn_func = OnMessageFuncs[lConnNum];
                
                socket.on('message', lConn_func);
                
            });
        }
        
        return lRet;
    };
    
    /**
     * function gets onMessage function 
     * that execute needed command
     * 
     * @param pConnNum, pSocket
     */
    function onMessage(pConnNum, pSocket){
        return function(pCommand) {
                Util.log(pCommand);
                
                if( Util.isContainStr(pCommand, 'cloudcmd') ){
                    pCommand = Util.removeStr(pCommand, 'cloudcmd');
                    
                    if( Util.isContainStr(pCommand, ' ') ){
                        pCommand = Util.removeStr(pCommand, ' ');
                        
                        if( Util.isContainStr(pCommand, 'update') && update )
                            update.get();
                        
                        if( Util.strCmp(pCommand, 'exit') )
                            if(main.WIN32)
                                pCommand = 'taskkill -f /PID ' + process.pid;
                            else
                                pCommand = 'kill -9 ' + process.pid;
                    }
                    else {
                        var lMsg = {
                            stdout : 'cloudcmd exit - for shutdown cloudcmd',
                            stderr : null
                        };
                        
                        lMsg = JSON.stringify(lMsg);
                        pSocket.send(lMsg);
                        
                        Util.log('received from client: ' + pCommand);
                        Util.log('sended to client: ' + lMsg);
                    }
                }
                
                /* if we on windows and command is build in
                 * change code page to unicode becouse
                 * windows use unicode on non English versions
                 */
                if(Win32_b){
                    
                    var lWinCommand = pCommand.toUpperCase();
                    if( Win32Commands.indexOf(lWinCommand) > 0 )
                        pCommand = 'chcp 65001 |' + pCommand;
                }
                if(!ClientFuncs[pConnNum])
                    ClientFuncs[pConnNum] = getExec(pSocket);
                
                var lExec_func = ClientFuncs[pConnNum];
                
                exec(pCommand, lExec_func);
            };
    }
    
    
    /**
     * function send result of command to client
     * @param pSocket
     */
    function getExec(pSocket){
        return function(pError, pStdout, pStderr) {
            if (pError !== null) {
                Util.log('exec error: ' + pError);
            }
            
            var lExec = {
                stdout : pStdout,
                stderr : pStderr || pError
            },            
                lExec_str = JSON.stringify(lExec);
            
            pSocket.send(lExec_str);
        };
    }
    
    
    /* windows commands thet require 
     * unicode charset on locales 
     * different then English
     */
    var Win32Commands   = ['ASSOC', 'AT', 'ATTRIB', 'BREAK', 'CACLS', 'CALL',
                            'CD', 'CHCP', 'CHDIR', 'CHKDSK', 'CHKNTFS', 'CLS',
                            'CMD', 'COLOR', 'COMP', 'COMPACT', 'CONVERT', 'COPY',
                            'DATE', 'DEL', 'DIR', 'DISKCOMP', 'DISKCOPY', 'DOSKEY',
                            'ECHO', 'ENDLOCAL', 'ERASE', 'EXIT', 'FC', 'FIND',
                            'FINDSTR', 'FOR', 'FORMAT', 'FTYPE', 'GOTO', 'GRAFTABL',
                            'HELP', 'IF', 'LABEL', 'MD', 'MKDIR', 'MODE', 'MORE',
                            'MOVE', 'PATH', 'PAUSE', 'POPD', 'PRINT', 'PROMPT',
                            'PUSHD', 'RD', 'RECOVER', 'REM', 'REN', 'RENAME',
                            'REPLACE', 'RMDIR', 'SET', 'SETLOCAL', 'SHIFT', 'SORT',
                            'START', 'SUBST', 'TIME', 'TITLE', 'TREE', 'TYPE',
                            'VER', 'VERIFY', 'VOL', 'XCOPY'];
})();