CliRunnable.cpp

Go to the documentation of this file.

/*
* This file is part of Project SkyFire https://www.projectskyfire.org.
* See LICENSE.md file for Copyright information
*/

 
#include "Common.h"
#include "Configuration/Config.h"
#include "ObjectMgr.h"
#include "World.h"
#include "WorldSession.h"
 
#include "AccountMgr.h"
#include "Chat.h"
#include "CliRunnable.h"
#include "Language.h"
#include "Log.h"
#include "MapManager.h"
#include "Player.h"
#include "Util.h"
 
#if PLATFORM != PLATFORM_WINDOWS
#include <readline/readline.h>
#include <readline/history.h>
 
char* command_finder(const char* text, int state)
{
    static size_t idx, len;
    const char* ret;
    std::vector<ChatCommand> const& cmd = ChatHandler::getCommandTable();
 
    if (!state)
    {
        idx = 0;
        len = strlen(text);
    }
 
    while (idx < cmd.size())
    {
        ret = cmd[idx].Name;
        if (!cmd[idx].AllowConsole)
        {
            ++idx;
            continue;
        }
 
        ++idx;
        //printf("Checking %s \n", cmd[idx].Name);
        if (strncmp(ret, text, len) == 0)
            return strdup(ret);
    }
 
    return ((char*)NULL);
}
 
char** cli_completion(const char* text, int start, int /*end*/)
{
    char** matches = NULL;
 
    if (start)
        rl_bind_key('\t', rl_abort);
    else
        matches = rl_completion_matches((char*)text, &command_finder);
    return matches;
}
 
int cli_hook_func()
{
    if (World::IsStopped())
        rl_done = 1;
    return 0;
}
 
#endif
 
void utf8print(void* /*arg*/, const char* str)
{
#if PLATFORM == PLATFORM_WINDOWS
    wchar_t wtemp_buf[6000];
    size_t wtemp_len = 6000 - 1;
    if (!Utf8toWStr(str, strlen(str), wtemp_buf, wtemp_len))
        return;
 
    char temp_buf[6000];
    CharToOemBuffW(&wtemp_buf[0], &temp_buf[0], wtemp_len + 1);
    printf(temp_buf);
#else
    {
        printf("%s", str);
        fflush(stdout);
    }
#endif
}
 
void commandFinished(void*, bool /*success*/)
{
    printf("SF> ");
    fflush(stdout);
}

void CliRunnable::Run()
{
    //SF_LOG_INFO("server.worldserver", "");
#if PLATFORM != PLATFORM_WINDOWS
    rl_attempted_completion_function = cli_completion;
    rl_event_hook = cli_hook_func;
#endif
 
    if (sConfigMgr->GetBoolDefault("BeepAtStart", true))
        printf("\a");                                       // \a = Alert
 
    // print this here the first time
    // later it will be printed after command queue updates
    printf("SF>");

    while (!World::IsStopped())
    {
        fflush(stdout);
 
        char* command_str;             // = fgets(commandbuf, sizeof(commandbuf), stdin);
 
#if PLATFORM == PLATFORM_WINDOWS
        char commandbuf[256];
        command_str = fgets(commandbuf, sizeof(commandbuf), stdin);
#else
        command_str = readline("SF>");
        rl_bind_key('\t', rl_complete);
#endif
 
        if (command_str != NULL)
        {
            for (int x = 0; command_str[x]; ++x)
                if (command_str[x] == '\r' || command_str[x] == '\n')
                {
                    command_str[x] = 0;
                    break;
                }
 
            if (!*command_str)
            {
#if PLATFORM == PLATFORM_WINDOWS
                printf("SF>");
#else
                free(command_str);
#endif
                continue;
            }
 
            std::string command;
            if (!consoleToUtf8(command_str, command))         // convert from console encoding to utf8
            {
#if PLATFORM == PLATFORM_WINDOWS
                printf("SF>");
#else
                free(command_str);
#endif
                continue;
            }
 
            fflush(stdout);
            sWorld->QueueCliCommand(new CliCommandHolder(NULL, command.c_str(), &utf8print, &commandFinished));
#if PLATFORM != PLATFORM_WINDOWS
            add_history(command.c_str());
            free(command_str);
#endif
        }
        else if (feof(stdin))
        {
            World::StopNow(SHUTDOWN_EXIT_CODE);
        }
    }
}