#include <stdafx.h>
#include <windows.h>
#include "ScriptAPI.h"

static void ConcatArgs(duk_context* ctx, stdstr& out);

void ScriptAPI::Define_console(duk_context* ctx)
{
    const DukPropListEntry props[] = {
        { "print",  DukCFunction(js_console_print) },
        { "log",    DukCFunction(js_console_log) },
        { "error",  DukCFunction(js_console_error) },
        { "clear",  DukCFunction(js_console_clear) },
        { "listen", DukCFunction(js_console_listen) },
        { nullptr }
    };

    DefineGlobalInterface(ctx, "console", props);
}

duk_ret_t ScriptAPI::js_console_print(duk_context* ctx)
{
    stdstr out;
    ConcatArgs(ctx, out);
    GetInstance(ctx)->System()->ConsolePrint("%s", out.c_str());
    return 0;
}

duk_ret_t ScriptAPI::js_console_log(duk_context* ctx)
{
    stdstr out;
    ConcatArgs(ctx, out);
    GetInstance(ctx)->System()->ConsoleLog("%s", out.c_str());
    return 0;
}

duk_ret_t ScriptAPI::js_console_error(duk_context* ctx)
{
    CheckArgs(ctx, { Arg_OptAny });

    if (duk_is_error(ctx, 0))
    {
        duk_get_prop_string(ctx, 0, "stack");
        const char* message = duk_get_string(ctx, -1);
        GetInstance(ctx)->System()->ConsoleLog("%s", message);
        return 0;
    }

    GetInstance(ctx)->System()->ConsoleLog("%s", duk_safe_to_string(ctx, 0));
    return 0;
}

duk_ret_t ScriptAPI::js_console_clear(duk_context* ctx)
{
    CheckArgs(ctx, {});
    GetInstance(ctx)->System()->ConsoleClear();
    return 0;
}

duk_ret_t ScriptAPI::js_console_listen(duk_context* ctx)
{
    CScriptInstance* inst = GetInstance(ctx);

    duk_push_global_object(ctx);
    duk_bool_t haveListener = duk_has_prop_string(ctx, -1, HS_gInputListener);

    if (duk_is_function(ctx, 0))
    {
        duk_pull(ctx, 0);
        duk_put_prop_string(ctx, -2, HS_gInputListener);
        if (!haveListener)
        {
            inst->IncRefCount();
        }
        return 0;
    }
    else if (duk_is_null(ctx, 0))
    {
        if (haveListener)
        {
            duk_del_prop_string(ctx, -1, HS_gInputListener);
            inst->DecRefCount();
        }
        return 0;
    }

    return ThrowInvalidArgsError(ctx);
}

void ConcatArgs(duk_context* ctx, stdstr& out)
{
    out = "";
    duk_idx_t nargs = duk_get_top(ctx);
    
    // note: global JSON.stringify must be intact

    duk_get_global_string(ctx, "JSON");
    duk_get_prop_string(ctx, -1, "stringify");
    duk_remove(ctx, -2);

    for (duk_idx_t n = 0; n < nargs; n++)
    {
        if (n != 0)
        {
            out += " ";
        }

        if (duk_is_object(ctx, n))
        {
            duk_dup(ctx, n);
            out += duk_safe_to_string(ctx, -1);
            out += " ";
            duk_pop(ctx);
            duk_dup(ctx, -1);
            duk_dup(ctx, n);
            duk_push_null(ctx);
            duk_push_int(ctx, 2);
            duk_pcall(ctx, 3);
            out += duk_safe_to_string(ctx, -1);
            duk_pop(ctx);
        }
        else
        {
            out += duk_safe_to_string(ctx, n);
        }
    }
}