AtomicGameEngine/Source/AtomicJS/Javascript/JSComponentFile.cpp

//
// Copyright (c) 2008-2015 the Urho3D project.
//
// Permission is hereby granted, free of charge, to any person obtaining a copy
// of this software and associated documentation files (the "Software"), to deal
// in the Software without restriction, including without limitation the rights
// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
// copies of the Software, and to permit persons to whom the Software is
// furnished to do so, subject to the following conditions:
//
// The above copyright notice and this permission notice shall be included in
// all copies or substantial portions of the Software.
//
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
// THE SOFTWARE.
//

#include <Atomic/Core/Context.h>
#include <Atomic/IO/Deserializer.h>
#include <Atomic/IO/Log.h>
#include <Atomic/Core/Profiler.h>
#include <Atomic/Resource/ResourceCache.h>
#include <Atomic/IO/Serializer.h>

#include "JSComponentFile.h"
#include "JSComponent.h"
#include "JSVM.h"

namespace Atomic
{

JSComponentFile::JSComponentFile(Context* context) :
    Resource(context),
    scriptClass_(false)
{
}

JSComponentFile::~JSComponentFile()
{

}

void JSComponentFile::RegisterObject(Context* context)
{
    context->RegisterFactory<JSComponentFile>();
}

void JSComponentFile::GetDefaultFieldValue(const String& name, Variant& v)
{
    v = Variant::EMPTY;

    VariantMap::ConstIterator itr = defaultFieldValues_.Find(name);

    if (itr == defaultFieldValues_.End())
    {
        HashMap<String, VariantType>::ConstIterator citr = fields_.Find(name);
        if (citr == fields_.End())
            return;

        switch (citr->second_)
        {
        case VAR_BOOL:
            v = false;
            break;
        case VAR_STRING:
            v = "";
            break;
        case VAR_FLOAT:
            v = 0.0f;
            break;
        case VAR_VECTOR3:
            v = Vector3::ZERO;
            break;
        default:
            break;
        }

        return;

    }

    v = itr->second_;

}

bool JSComponentFile::PushModule()
{
    if (context_->GetEditorContext())
        return false;

    JSVM* vm = JSVM::GetJSVM(NULL);

    duk_context* ctx = vm->GetJSContext();

    const String& path = GetName();
    String pathName, fileName, ext;
    SplitPath(path, pathName, fileName, ext);

	if (path.Contains('/') || path.Contains('\\'))
		pathName += "/" + fileName;
	else
		pathName = fileName;

    duk_get_global_string(ctx, "require");
    duk_push_string(ctx, pathName.CString());

    if (duk_pcall(ctx, 1) != 0)
    {
        vm->SendJSErrorEvent();
        return false;
    }

    return true;

}

JSComponent* JSComponentFile::CreateJSComponent()
{
    JSComponent* component = NULL;

    if (!scriptClass_)
    {
        component =  new JSComponent(context_);
    }
    else
    {

        JSVM* vm = JSVM::GetJSVM(NULL);
        duk_context* ctx = vm->GetJSContext();

        PushModule();

        duk_new(ctx, 0);

        if (duk_is_object(ctx, -1))
        {
            component = js_to_class_instance<JSComponent>(ctx, -1, 0);
            component->scriptClassInstance_ = true;
            // store reference below so pop doesn't gc the component
            component->UpdateReferences();
        }

        duk_pop(ctx);

    }

    if (!component)
    {
        LOGERRORF("Failed to create script class from component file %s", GetName().CString());
        component =  new JSComponent(context_);
    }

    component->SetComponentFile(this);

    return component;

}

bool JSComponentFile::InitModule()
{
    if (context_->GetEditorContext())
        return true;

    JSVM* vm = JSVM::GetJSVM(NULL);

    duk_context* ctx = vm->GetJSContext();

    duk_idx_t top = duk_get_top(ctx);

    if (!PushModule())
        return false;

    if (!duk_is_function(ctx, -1))
    {
        LOGERRORF("Component file does not export a function: %s", GetName().CString());
        duk_set_top(ctx, top);
        return false;
    }

    // detect a script class vs a simple "flat" javascript component
    // this means that if a script component class defines a constructor,
    // it must take 0 arguments (which makes sense as when it is new'd from
    // serialization, etc there will be no args to pass it

    if (duk_get_length(ctx, -1) == 0)
    {
        scriptClass_ = true;
    }

    duk_set_top(ctx, top);

    return true;
}


bool JSComponentFile::BeginLoad(Deserializer& source)
{
    if (!InitModule())
        return false;

    // TODO: cache these for player builds
    // FIXME: this won't work with obfusication or minimization

    unsigned dataSize = source.GetSize();
    if (!dataSize && !source.GetName().Empty())
    {
        LOGERROR("Zero sized component file in " + source.GetName());
        return false;
    }

    SharedArrayPtr<char> buffer(new char[dataSize + 1]);
    if (source.Read(buffer.Get(), dataSize) != dataSize)
        return false;
    buffer[dataSize] = '\0';

    String text = buffer.Get();
    text.Replace("\r", "");
    Vector<String> lines = text.Split('\n');

    String eval;
    bool valid = false;
    for (unsigned i = 0; i < lines.Size(); i++)
    {
        String line = lines[i];

        if (!eval.Length())
        {
            line = line.Trimmed();

            if (line.StartsWith("inspectorFields"))
            {
                eval = line;
                if (line.Contains("}"))
                {
                    valid = true;
                    break;
                }
            }
            else if (line.StartsWith("this.inspectorFields"))
            {
                eval = line.Substring(5);
                if (line.Contains("}"))
                {
                    valid = true;
                    break;
                }
            }
            else if (line.StartsWith("var inspectorFields"))
            {
                eval = line.Substring(4);
                if (line.Contains("}"))
                {
                    valid = true;
                    break;
                }
            }

        }
        else
        {
            eval += line;
        }

        if (line.Contains("}") && eval.Length())
        {
            valid = true;
            break;
        }

    }

    if (valid)
    {
        JSVM* vm = JSVM::GetJSVM(NULL);

        if (!vm)
            return true;

        duk_context* ctx = vm->GetJSContext();

        int top = duk_get_top(ctx);

        duk_push_string(ctx, eval.CString());
        duk_push_string(ctx, "eval");
        duk_compile(ctx, DUK_COMPILE_EVAL);
        if (duk_is_function(ctx, -1) && duk_pcall(ctx, 0) == DUK_EXEC_SUCCESS)
        {
            if (duk_is_object(ctx, -1))
            {
                duk_enum(ctx, -1, DUK_ENUM_OWN_PROPERTIES_ONLY);

                while (duk_next(ctx, -1, 1)) {

                    String name = duk_get_string(ctx, -2);

                    VariantType variantType = VAR_NONE;
                    Variant defaultValue;

                    if (duk_is_string(ctx, -1))
                    {
                        variantType = VAR_STRING;
                        defaultValue = duk_to_string(ctx, -1);
                    }
                    else if (duk_is_number(ctx, -1))
                    {
                        variantType = VAR_FLOAT;
                        defaultValue = (float) duk_to_number(ctx, -1);
                    }
                    else if (duk_is_array(ctx, -1))
                    {
                        if (duk_get_length(ctx, -1) > 0)
                        {
                            duk_get_prop_index(ctx, -1, 0);

                            // TODO: class types
                            variantType = (VariantType) ((int)duk_require_number(ctx, -1));

                            duk_pop(ctx);
                        }

                        if (duk_get_length(ctx, -1) > 1)
                        {
                            duk_get_prop_index(ctx, -1, 1);
                            // default value
                            js_to_variant(ctx, -1, defaultValue);

                            duk_pop(ctx);
                        }

                    }
                    else if (duk_is_boolean(ctx, -1))
                    {
                        variantType = VAR_BOOL;
                        defaultValue = duk_to_boolean(ctx, -1) ? true : false;
                    }


                    if (defaultValue.GetType() != VAR_NONE)
                    {
                        defaultFieldValues_[name] = defaultValue;

                    }

                    if (variantType != VAR_NONE)
                        fields_[name] = variantType;

                    duk_pop_2(ctx);  // pop key value
                }

                duk_pop(ctx);  // pop enum object

            }

        }

        duk_set_top(ctx, top);
    }

    SetMemoryUse(0);

    return true;
}

bool JSComponentFile::Save(Serializer& dest) const
{
    return true;
}



}