drogon/drogon_ctl/help.cc

/**
 *
 *  help.cc
 *  An Tao
 *
 *  Copyright 2018, An Tao.  All rights reserved.
 *  https://github.com/an-tao/drogon
 *  Use of this source code is governed by a MIT license
 *  that can be found in the License file.
 *
 *  Drogon
 *
 */

#include "help.h"
#include <drogon/DrClassMap.h>
#include <iostream>
#include <memory>
using namespace drogon_ctl;

void help::handleCommand(std::vector<std::string> &parameters)
{
    if (parameters.size() == 0)
    {
        std::cout << "usage: drogon_ctl [-v | --version] [-h | --help] "
                     "<command> [<args>]"
                  << std::endl;
        std::cout << "commands list:" << std::endl;
        for (auto &className : drogon::DrClassMap::getAllClassName())
        {
            auto classPtr = std::shared_ptr<DrObjectBase>(
                drogon::DrClassMap::newObject(className));
            if (classPtr)
            {
                auto cmdHdlPtr =
                    std::dynamic_pointer_cast<CommandHandler>(classPtr);
                if (cmdHdlPtr)
                {
                    if (!cmdHdlPtr->isTopCommand())
                        continue;
                    auto pos = className.rfind("::");
                    if (pos != std::string::npos)
                    {
                        className = className.substr(pos + 2);
                    }
                    while (className.length() < 24)
                        className.append(" ");
                    std::cout << className << cmdHdlPtr->script() << std::endl;
                }
            }
        }
    }
    else
    {
        auto cmd = std::string("drogon_ctl::") + parameters[0];

        auto classPtr =
            std::shared_ptr<DrObjectBase>(drogon::DrClassMap::newObject(cmd));
        if (classPtr)
        {
            auto cmdHdlPtr =
                std::dynamic_pointer_cast<CommandHandler>(classPtr);
            if (cmdHdlPtr)
            {
                if (cmdHdlPtr->isTopCommand())
                    std::cout << cmdHdlPtr->detail() << std::endl;
            }
        }
    }
}