devel/libmdw/trunk/libmdw/src/CommandManager.cpp

/***************************************************************************
 *   Copyright (C) 2010 by Jean-Baptiste Wons   *
 *   wonsjb@gmail.com   *
 *                                                                         *
 *   This program is free software; you can redistribute it and/or modify  *
 *   it under the terms of the GNU General Public License as published by  *
 *   the Free Software Foundation; either version 2 of the License, or     *
 *   (at your option) any later version.                                   *
 *                                                                         *
 *   This program is distributed in the hope that it will be useful,       *
 *   but WITHOUT ANY WARRANTY; without even the implied warranty of        *
 *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the         *
 *   GNU General Public License for more details.                          *
 *                                                                         *
 *   You should have received a copy of the GNU General Public License     *
 *   along with this program; if not, write to the                         *
 *   Free Software Foundation, Inc.,                                       *
 *   59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.             *
 ***************************************************************************/
#include <cassert>

#include "CommandManager.h"
#include "CommandHandler.h"
#include "Stream.h"

namespace mdw
{

CommandManager *CommandManager::_instance = 0;

CommandManager::CommandManager()
{
}

CommandManager::~CommandManager()
{
}

void CommandManager::initialise()
{
    _instance = new CommandManager();
    _instance->init();
}

CommandManager &CommandManager::getInstance()
{
    assert(_instance != 0);
    return *_instance;
}

CommandHandler *CommandManager::getHandler (const std::vector<std::string> &tokens, int &position)
{
    if (tokens.empty())
    {
        return 0;
    }
    CommandHandler *handler = getHandler(tokens[0]);
    if (handler == 0)
    {
        return 0;
    }
    for (int i = 1; i < tokens.size(); ++i)
    {
        CommandHandler *nextHandler = handler->getHandler(tokens[i]);
        if (nextHandler == 0)
        {
            position = i;
            return handler;
        }
        else
        {
            handler = nextHandler;
        }
    }
}

CommandHandler *CommandManager::getHandler (const std::string &firstLevel)
{
    std::map<std::string, CommandHandler*>::iterator aIter = _commandTree.find(firstLevel);
    if (aIter == _commandTree.end())
    {
        return 0;
    }
    return aIter->second;
}

void CommandManager::execute (const std::vector<std::string> &tokens, Stream &output)
{
    if (tokens.empty())
    {
        return;
    }
    int position;
    CommandHandler *aHandler = getHandler(tokens, position);
    if (aHandler == 0)
    {
        output << "Command not found : " << tokens[0] << endl;
        output << "Possible commands are: " << endl;
        for (std::map<std::string, CommandHandler*>::iterator aIter = _commandTree.begin();
             aIter != _commandTree.end();
             ++aIter)
        {
            output << aIter->first << endl;
        }
    }
    aHandler->execute(tokens, position, output);
}

void CommandManager::init()
{
}

CommandHandler &CommandManager::createNode (const std::string &subHandler, CommandHandler *handler)
{
    std::map<std::string, CommandHandler*>::iterator aIter = _commandTree.find(subHandler);
    if (aIter == _commandTree.end())
    {
        if (handler == 0)
        {
            handler = new CommandHandler();
        }

        _commandTree.insert(make_pair(subHandler, handler));
    }
    else
    {
        if (handler != 0)
        {
            for (std::map<std::string, CommandHandler*>::iterator aIter = aIter->second->_subCommands.begin();
                 aIter != aIter->second->_subCommands.end();
                 ++aIter)
            {
                handler->_subCommands.insert(*aIter);
            }
            _commandTree.insert(make_pair(subHandler, handler));
        }
    }
    return *handler;
}

}

---