#include <Atomic/Core/ProcessUtils.h>
#include <Atomic/IO/Log.h>
#include <Atomic/IO/File.h>
#include <Atomic/Resource/JSONFile.h>

#include "JSBModule.h"
#include "JSBClass.h"
#include "JSBFunction.h"

namespace ToolCore
{

JSBFunctionSignature::JSBFunctionSignature(const String &name, const Vector<String>& sig) : parsed_(false)
{
    name_ = name;
    sig_ = sig;
}

void JSBFunctionSignature::Parse()
{
    if (parsed_)
        return;

    parsed_ = true;

    for (unsigned i = 0; i < sig_.Size(); i++)
    {
        JSBType* type = JSBType::Parse(sig_.At(i));
        if (!type)
        {
            ErrorExit(ToString("Unable to parse override type: %s", sig_.At(i).CString()));
        }
        types_.Push(type);
    }

}

bool JSBFunctionSignature::Match(JSBFunction* function)
{

    if (name_ != function->GetName())
        return false;

    Vector<JSBFunctionType*>& parameters = function->GetParameters();

    if (types_.Size() != parameters.Size())
        return false;

    for (unsigned x = 0; x < parameters.Size(); x++)
    {
        JSBType* ot = types_[x];
        JSBType* pt = parameters[x]->type_;

        // should add an == operator
        if ((ot->asPrimitiveType() == NULL) != (pt->asPrimitiveType() == NULL) ||
                (ot->asClassType() == NULL) != (pt->asClassType() == NULL) )
        {
            return false;
        }

        if (ot->asPrimitiveType())
        {
            JSBPrimitiveType* pot = ot->asPrimitiveType();
            JSBPrimitiveType* ppt = pt->asPrimitiveType();

            if (pot->kind_ != ppt->kind_)
            {
                return false;
            }
        }
        if (ot->asClassType())
        {
            JSBClassType* cot = ot->asClassType();
            JSBClassType* cpt = pt->asClassType();

            if (cot->class_ != cpt->class_)
            {
                return false;
            }
        }
        if (ot->asStringType())
        {
            if (!pt->asStringType())
            {
                return false;
            }
        }
        if (ot->asStringHashType())
        {
            if (!pt->asStringHashType())
            {
                return false;
            }
        }


    }

    return true;

}

JSBClass::JSBClass(Context* context, JSBModule *module, const String& name, const String& nativeName) : Object(context),
    module_(module), name_(name), nativeName_(nativeName),
    isAbstract_(false), isObject_(false),
    numberArrayElements_(0), arrayElementType_("float"),
    hasProperties_(false)
{

    if (nativeName == "Object")
        SetObject(true);

    // detect array type, TODO: this could go in module json

    if (name_ == "Color")
        numberArrayElements_ = 4;
    else if (name_ == "Vector2")
        numberArrayElements_ = 2;
    else if (name_ == "Vector3")
        numberArrayElements_ = 3;
    else if (name_ == "Vector4")
        numberArrayElements_ = 4;
    else if (name_ == "Quaternion")
        numberArrayElements_ = 4;
    else if (name_ == "BoundingBox")
        numberArrayElements_ = 6;
    else if (name_ == "Rect")
        numberArrayElements_ = 4;
    else if (name_ == "IntRect")
    {
        numberArrayElements_ = 4;
        arrayElementType_ = "int";
    }
    else if (name_ == "IntVector2")
    {
        numberArrayElements_ = 2;
        arrayElementType_ = "int";
    }

}

JSBClass::~JSBClass()
{

}

void JSBClass::AddFunction(JSBFunction* function)
{
    functions_.Push(function);
}

JSBClass* JSBClass::GetBaseClass()
{
    if (!baseClasses_.Size())
        return 0;

    return baseClasses_[0];

}

JSBFunction* JSBClass::GetConstructor()
{
    {
        for (unsigned i = 0; i < functions_.Size(); i++)
            if (functions_[i]->IsConstructor() && !functions_[i]->Skip())
                return functions_[i];
    }

    return NULL;

}

void JSBClass::SetSkipFunction(const String& name, bool skip)
{
    for (unsigned i = 0; i < functions_.Size(); i++)
    {
        if (functions_[i]->IsConstructor())
            continue;

        if (functions_[i]->GetName() == name)
            functions_[i]->SetSkip(skip);
    }

}
void JSBClass::SetBaseClass(JSBClass *baseClass)
{
    // we can't hook up chain here, as base class may not have been visited yet

    assert(!baseClasses_.Size());
    baseClasses_.Push(baseClass);
}

void JSBClass::RecursiveAddBaseClass(PODVector<JSBClass*>& baseClasses)
{
    for (unsigned i = 0; i < baseClasses.Size(); i++)
    {
        JSBClass* base = baseClasses.At(i);

        if (base->IsObject())
            SetObject(true);

        if (!baseClasses_.Contains(base))
            baseClasses_.Push(base);

        RecursiveAddBaseClass(base->baseClasses_);
    }
}

void JSBClass::Preprocess()
{
    RecursiveAddBaseClass(baseClasses_);
}

void JSBClass::Process()
{
    for (unsigned j = 0; j < overrides_.Size(); j++)
    {
        overrides_.At(j)->Parse();
    }

    for (unsigned j = 0; j < excludes_.Size(); j++)
    {
        excludes_.At(j)->Parse();
    }


    // detect skips

    for (unsigned j = 0; j < functions_.Size(); j++)
    {

        for (unsigned k = 0; k < excludes_.Size(); k++)
        {
            if (excludes_[k]->Match(functions_[j]))
            {
                functions_[j]->SetSkip(true);
                break;
            }

        }
    }

    // detect overridden functions, only works for in class atm (not baseclasses)
    for (unsigned j = 0; j < functions_.Size(); j++)
    {
        JSBFunction* function = functions_[j];

        if (IsNumberArray())
        {
            function->SetSkip(true);
            continue;
        }

        // skip function if only one parameter of type Context, if not Constuctor
        if (!function->IsConstructor())
        {
            Vector<JSBFunctionType*>& parameters = function->GetParameters();

            if (parameters.Size() == 1 && parameters.At(0)->type_->asClassType())
            {
                if (parameters.At(0)->type_->asClassType()->class_->GetName() == "Context")
                {
                    function->SetSkip(true);
                    continue;
                }
            }
        }


        if (function->IsOverride())
            continue;

        for (unsigned k = 0; k < functions_.Size(); k++)
        {
            if (j == k)
                continue;

            JSBFunction* function2 = functions_[k];

            if (function->GetName() == function2->GetName())
            {
                function->SetOverride();
                function2->SetOverride();
                // initially set all overridden functions to skip
                function->SetSkip(true);
                function2->SetSkip(true);
                break;
            }
        }

    }

    // mark overrides
    for (unsigned j = 0; j < functions_.Size(); j++)
    {
        JSBFunction* function = functions_[j];

        if (function->IsOverride())
        {
            for (unsigned k = 0; k < overrides_.Size(); k++)
            {                
                JSBFunctionSignature* override =  overrides_[k];

                if (!override->Match(function))
                    continue;

                function->SetSkip(false);

                break;

            }

        }

    }

}

void JSBClass::PostProcess()
{
    for (unsigned j = 0; j < functions_.Size(); j++)
    {
        JSBFunction* function = functions_[j];
        function->Process();
    }
}

void JSBClass::AddPropertyFunction(JSBFunction* function)
{
    hasProperties_ = true;

    if (!function->GetPropertyName().Length())
    {
        ErrorExit("Function has no property name");
    }

    if (!function->IsGetter() && !function->IsSetter())
    {
        ErrorExit("Function is not a getter or setter");
    }

    JSBProperty* prop = NULL;

    if (properties_.Contains(function->GetPropertyName()))
    {
        prop = properties_[function->GetPropertyName()];
    }

    if (!prop)
    {
        prop = new JSBProperty();
        prop->name_ = function->GetPropertyName();
        properties_[function->GetPropertyName()] = prop;
    }

    if (prop->getter_ && function->IsGetter())
    {
        ErrorExit("getter already registered");
    }

    if (prop->setter_ && function->IsSetter())
    {
        ErrorExit("setter already registered");
    }

    if (function->IsSetter())
        prop->setter_ = function;
    else
        prop->getter_ = function;

}

void JSBClass::Dump()
{
    if (name_ != nativeName_)
    {
        LOGINFOF("Class: %s (%s)", name_.CString(), nativeName_.CString());
    }
    else
    {
        LOGINFOF("Class: %s", name_.CString());
    }

    LOGINFOF("   IsObject: %s", IsObject() ? "true" : "false");

    if (baseClasses_.Size())
    {
        LOGINFOF("   Bases:");
        for (unsigned i = 0; i < baseClasses_.Size(); i++)
        {

            LOGINFOF("      %s", baseClasses_.At(i)->GetName().CString());
        }
    }

    if (functions_.Size())
    {
        for (unsigned i = 0; i < functions_.Size(); i++)
        {
            functions_.At(i)->Dump();
        }
    }

}


}