OpenRaider/src/commands/Command.cpp

/*!
 * \file src/commands/Command.cpp
 * \brief Commands
 *
 * \author xythobuz
 */

#include <fstream>
#include <iomanip>

#include "global.h"
#include "Log.h"
#include "utils/strings.h"
#include "commands/Command.h"
#include "commands/CommandAnimate.h"
#include "commands/CommandBind.h"
#include "commands/CommandEngine.h"
#include "commands/CommandGame.h"
#include "commands/CommandMove.h"
#include "commands/CommandRender.h"
#include "commands/CommandSet.h"

std::vector<std::shared_ptr<Command>> Command::commands;

Command::~Command() {
}

void Command::printHelp() {
    getLog() << "No help available!" << Log::endl;
}

void Command::fillCommandList() {
    commands.clear();
    commands.push_back(std::shared_ptr<Command>(new CommandLoad()));
    commands.push_back(std::shared_ptr<Command>(new CommandBind()));
    commands.push_back(std::shared_ptr<Command>(new CommandSet()));
    commands.push_back(std::shared_ptr<Command>(new CommandGet()));
    commands.push_back(std::shared_ptr<Command>(new CommandScreenshot()));
    commands.push_back(std::shared_ptr<Command>(new CommandAnimate()));
    commands.push_back(std::shared_ptr<Command>(new CommandMove()));
    commands.push_back(std::shared_ptr<Command>(new CommandMode()));
    commands.push_back(std::shared_ptr<Command>(new CommandRenderflag()));
    commands.push_back(std::shared_ptr<Command>(new CommandPos()));
    commands.push_back(std::shared_ptr<Command>(new CommandViewmodel()));
    commands.push_back(std::shared_ptr<Command>(new CommandPigtail()));
    commands.push_back(std::shared_ptr<Command>(new CommandPonypos()));
    commands.push_back(std::shared_ptr<Command>(new CommandQuit()));
}

int Command::command(std::string c) {
    assert(commands.size() > 0);

    // Remove comment, if any
    size_t comment = c.find_first_of('#');
    if (comment != std::string::npos)
        c.erase(comment);

    // Execute command
    std::stringstream command(c);
    std::string cmd;
    command >> cmd;
    command >> std::boolalpha >> std::ws;

    if (cmd.length() == 0)
        return 0;

    // Print help
    if (cmd == "help") {
        std::string arg;
        command >> arg;
        if (arg.length() == 0) {
            // List all available commands
            getLog() << "Available commands:" << Log::endl;
            getLog() << std::right << std::setw(11);
            getLog() << "help" << " - print command help" << Log::endl;
            for (auto& x : commands) {
                if (x) {
                    getLog() << std::right << std::setw(11);
                    getLog() << x->name() << " - " << x->brief() << Log::endl;
                }
            }
            getLog() << "Use help COMMAND to get additional info" << Log::endl;
            getLog() << "Pass BOOLs as true or false" << Log::endl;
            return 0;
        } else {
            // Show help for a specific command
            for (auto& x : commands) {
                if (x) {
                    if (x->name() == arg) {
                        x->printHelp();
                        return 0;
                    }
                }
            }
            getLog() << "Unknown command: \"" << arg << "\"" << Log::endl;
            return -1;
        }
    }

    // Execute command
    for (auto& x : commands) {
        if (x) {
            if (x->name() == cmd) {
                return x->execute(command);
            }
        }
    }

    getLog() << "Unknown command: \"" << cmd << "\"" << Log::endl;
    return -1;
}

int Command::executeFile(std::string file) {
    std::string configFile = expandHomeDirectory(file);
    getLog() << "Loading config from \"" << configFile << "\"..." << Log::endl;

    std::ifstream f(configFile);
    if (!f) {
        getLog() << "Could not open file!" << Log::endl;
        return -1;
    }

    for (std::string line; std::getline(f, line);) {
        if (line.length() == 0)
            continue;

        int error = Command::command(line);
        if (error != 0)
            getLog() << "Error Code: " << error << Log::endl;
    }

    f.close();
    return 0;
}

std::string Command::autoComplete(std::string begin) {
    std::vector<std::string> candidates;

    std::string help("help");
    if (begin.size() <= help.size()) {
        if (begin.compare(0, begin.size(), help, 0, begin.size()) == 0) {
            candidates.push_back(help);
        }
    }

    for (auto& x : commands) {
        if (x) {
            std::string name = x->name();
            if (begin.size() <= name.size()) {
                if (begin.compare(0, begin.size(), name, 0, begin.size()) == 0) {
                    candidates.push_back(name);
                }
            }
        }
    }

    if (candidates.size() == 0) {
        return "";
    } else if (candidates.size() == 1) {
        return candidates.at(0);
    } else {
        std::string common = candidates.at(0);
        for (int i = 0; i < candidates.size(); i++) {
            getLog() << candidates.at(i);
            if (i < (candidates.size() - 1))
                getLog() << "/";

            for (int c = 0; (c < common.size()) && (c < candidates.at(i).size()); c++) {
                if (common.at(c) != candidates.at(i).at(c)) {
                    common.erase(c);
                    break;
                }
            }
        }
        getLog() << Log::endl;
        return common;
    }
}