//
// Copyright (c) 2014-2016 THUNDERBEAST GAMES LLC
//
// 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.
//

#pragma once

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

using namespace Atomic;


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

#include "JSBNameVisitor.h"


namespace ToolCore
{

class JSBHeader;
class JSBPrimitiveType;


class JSBHeaderVisitor : public SymbolVisitor
{
    JSBHeader* header_;
    JSBModule* module_;
    TranslationUnit* unit_;

    Namespace* globalNamespace_;

public:

    JSBHeaderVisitor(JSBHeader* header, TranslationUnit *unit, Namespace* globalNamespace) :
        header_(header),
        unit_(unit),
        globalNamespace_(globalNamespace)
    {
        module_ = header_->GetModule();

        accept(globalNamespace_);

    }

    String getNameString(const Name* name)
    {
        JSBNameVisitor nvisitor;
        return nvisitor(name);
    }

    JSBType* processTypeConversion(Type* type, FullySpecifiedType fst)
    {
        JSBType* jtype = NULL;

        if (type->isIntegerType())
        {
            IntegerType* itype = type->asIntegerType();
            jtype = new JSBPrimitiveType(itype->kind());
        }
        else if (type->isFloatType())
        {
            jtype = new JSBPrimitiveType(JSBPrimitiveType::Float);
        }
        else if (type->isNamedType())
        {
            NamedType* ntype = type->asNamedType();
            String classname = getNameString(ntype->name());

            if (classname.StartsWith("Atomic::"))
                classname.Replace("Atomic::", "");

            if (classname == "VariantVector")
            {
                JSBClass* jclass = JSBPackage::GetClassAllPackages("ScriptVariant");
                assert(jclass);

                jtype = new JSBVectorType(new JSBClassType(jclass), false, true);

            }
            else if (classname == "Vector" || classname == "PODVector")
            {
                PODVector<TemplateType> types;
                unwrapTemplateType(fst, types);

                if (types.Size() == 2)
                {
                    JSBType* vtype = processTypeConversion((Type*) types[1].type_, types[1].fstype_);

                    if (vtype)
                    {
                        jtype = new JSBVectorType(vtype, classname == "PODVector");
                    }

                }
                else if (types.Size() == 3 && ( getNameString(types[1].name_) == "SharedPtr" || getNameString(types[1].name_) == "WeakPtr"))
                {
                    JSBType* vtype = processTypeConversion((Type*) types[2].type_, types[2].fstype_);

                    if (vtype)
                    {
                        JSBVectorType* jvtype = new JSBVectorType(vtype, classname == "PODVector");

                        if (getNameString(types[1].name_) == "SharedPtr")
                        {
                            jvtype->vectorTypeIsSharedPtr_ = true;
                        }
                        else if (getNameString(types[1].name_) == "WeakPtr")
                        {
                            jvtype->vectorTypeIsWeakPtr_ = true;
                        }

                        jtype = jvtype;
                    }

                }

            }
            else if (classname == "String")
            {
                jtype = new JSBStringType();
            }
            else if (classname == "StringHash")
            {
                jtype = new JSBStringHashType();
            }
            else if (classname == "JS_HEAP_PTR")
            {
                jtype = new JSBHeapPtrType();
            }
            else
            {
                JSBClass* jclass = JSBPackage::GetClassAllPackages(classname, true);

                if (jclass)
                    jtype = new JSBClassType(jclass);
                else
                {
                    // this might be an enum
                    JSBEnum* jenum = JSBPackage::GetEnumAllPackages(classname);
                    if (jenum)
                        jtype = new JSBEnumType(jenum);

                }
            }
        }
        else if (type->asUndefinedType())
        {
            UndefinedType* utype = type->asUndefinedType();
            //ErrorExit("Undefined type");
        }

        return jtype;

    }

    struct TemplateType
    {
        FullySpecifiedType fstype_;
        bool isPointer_;
        bool isReference_;
        const Name* name_;
        const Type* type_;

        static void Init(TemplateType& ttype)
        {
            ttype.isPointer_ = false;
            ttype.isReference_ = false;
            ttype.name_ = 0;
            ttype.type_ = 0;
        }
    };

    bool unwrapTemplateType(const FullySpecifiedType& fstype, PODVector<TemplateType>& types)
    {
        TemplateType ttype;

        TemplateType::Init(ttype);

        ttype.fstype_ = fstype;
        ttype.type_ = fstype.type();

        if (ttype.type_->isPointerType())
        {
            ttype.isPointer_=true;
            FullySpecifiedType pfst = ttype.type_->asPointerType()->elementType();
            ttype.type_ = pfst.type();
        }

        if (ttype.type_->isReferenceType())
        {
            ttype.isReference_=true;
            FullySpecifiedType pfst = ttype.type_->asReferenceType()->elementType();
            ttype.type_ = pfst.type();
        }

        const NamedType* ntype = ttype.type_->asNamedType();

        if (!ntype)
            return false;

        if (ntype->name()->asTemplateNameId())
        {
            const TemplateNameId* tnid = ntype->name()->asTemplateNameId();

            ttype.name_ = tnid->identifier()->asNameId();

            types.Push(ttype);

            unwrapTemplateType(tnid->templateArgumentAt(0), types);

            return true;
        }

        ttype.name_ = ntype->name();

        types.Push(ttype);

        return false;
    }

    JSBFunctionType* processFunctionType(FullySpecifiedType fst, bool retType = false)
    {
        JSBType* jtype = NULL;
        Type* type = fst.type();

        bool isPointer = false;
        bool isSharedPtr = false;
        bool isReference = false;
        bool isTemplate = false;

        bool isConst = false;

        if (type->isPointerType())
        {
            isPointer=true;
            FullySpecifiedType pfst = type->asPointerType()->elementType();
            type = pfst.type();
        }
        if (type->isReferenceType())
        {
            isReference=true;
            FullySpecifiedType pfst = type->asReferenceType()->elementType();
            type = pfst.type();
            isConst = pfst.isConst();
        }
        if (!isPointer && retType)
        {
            if (type->isNamedType())
            {
                NamedType* ntype = type->asNamedType();

                if (ntype->name()->asTemplateNameId())
                {
                    PODVector<TemplateType> types;
                    unwrapTemplateType(fst, types);

                    String classname = getNameString(types[0].name_);

                    // SharedPtr
                    if ( classname == "SharedPtr" && types.Size() == 2 )
                    {
                        type = (Type*) types[1].type_;
                        isTemplate = true;
                        isSharedPtr = true;
                    }
                }
            }
        }

        if (fst.isUnsigned())
        {
            if (type->isUndefinedType())
            {
                // this happens when just using "unsigned" in code
                jtype = new JSBPrimitiveType(JSBPrimitiveType::Int, true);
            }
        }


        if (!jtype)
        {
            jtype = processTypeConversion(type, fst);

            // explicit script string -> StringHash required
            if (jtype && jtype->asStringHashType())
                isReference = false;

            if (fst.isUnsigned() && jtype->asPrimitiveType())
                jtype->asPrimitiveType()->isUnsigned_ = true;

        }

        if (!jtype)
            return NULL;

        bool skip = false;

        // no pointers to prim atm
        if (isPointer || isReference)
        {
            if (jtype->asPrimitiveType())
                skip = true;
            else if (!retType && !isConst && (jtype->asStringType() || jtype->asStringHashType()))
            {
                skip = true;
            }

            if (skip)
                return NULL;
        }

        // We don't support pointers to Vector2/Vector3/etc which are generally out values 
        if (isPointer && jtype->asClassType() && jtype->asClassType()->class_->IsNumberArray())
            return NULL;

        JSBFunctionType* ftype = new JSBFunctionType(jtype);
        ftype->isPointer_ = isPointer;
        ftype->isSharedPtr_ = isSharedPtr;
        ftype->isReference_ = isReference;
        ftype->isTemplate_ = isTemplate;
        ftype->isConst_ = isConst;

        return ftype;

    }

    JSBFunctionType* processFunctionArgType(Argument* arg)
    {

        JSBFunctionType* jtype = processFunctionType(arg->type());

        if (!jtype)
            return NULL;

        jtype->name_ = getNameString(arg->name());

        return jtype;

    }


    JSBFunctionType* processFunctionReturnType(Function* function)
    {
        if (!function->hasReturnType())
        {
            return NULL;
        }

        JSBFunctionType* jtype = processFunctionType(function->returnType(), true);

        if (!jtype)
            return NULL;

        return jtype;

    }

    String parseDocString(int commentLine) const
    {
        const char* comment = NULL;
        for (unsigned i = 0; i < unit_->commentCount(); i++)
        {
            const Token &tcomment = unit_->commentAt(i);
            unsigned line;
            unit_->getPosition(tcomment.utf16charsEnd(), &line);

            if (line ==  commentLine)
            {
                comment = unit_->firstSourceChar() + tcomment.byteOffset;
                break;
            }
        }

        String docString;

        if (comment && strlen(comment) > 3)
        {
            if (comment[0] == '/' && comment[1] == '/' && comment[2] == '/')
            {
                int index = 3;
                while(comment[index] && comment[index] != '\n' && comment[index] != '\r')
                {
                    docString += comment[index++];
                }

            }

            if (comment[0] == '/' && comment[1] == '*' && comment[2] == '*')
            {
                int index = 3;
                bool foundStar = false;
                while(comment[index])
                {
                    // did we find a star in the last loop?
                    if (foundStar)
                    {
                        // We have a an end of block indicator, let's break
                        if (comment[index] == '/' && foundStar)
                            break;

                        // This is just a star in the comment, not an end of comment indicator.  Let's keep it
                        docString += '*';
                    }

                    foundStar = comment[index] == '*';

                    if (!foundStar)
                        docString += comment[index];

                    index++;
                }
            }

        }

        return docString;


    }

    JSBFunction* processFunction(JSBClass* klass, Function* function)
    {
        JSBFunction* jfunction = new JSBFunction(klass);

        // don't ... atm
        if (function->isVariadic())
            return NULL;

        String name = getNameString(function->name());

        jfunction->SetName(name);

        // don't support operators atm
        if (name.StartsWith("operator "))
            return NULL;

        if (name == klass->GetNativeName())
            jfunction->SetConstructor();

        if (name.StartsWith("~"))
            jfunction->SetDestructor();

        if (function->isVirtual())
            jfunction->SetVirtual(true);

        if (function->isStatic())
            jfunction->SetStatic(true);

        // see if we support return type
        if (function->hasReturnType() && !function->returnType().type()->isVoidType())
        {
            JSBFunctionType* returnType = processFunctionReturnType(function);

            if (!returnType)
                return NULL;

            jfunction->SetReturnType(returnType);
        }

        if (function->hasArguments())
        {
            for (unsigned i = 0; i < function->argumentCount(); i++)
            {
                Symbol* symbol = function->argumentAt(i);

                if (symbol->isArgument())
                {
                    Argument* arg = symbol->asArgument();

                    JSBFunctionType* ftype = processFunctionArgType(arg);

                    // TODO: expose interfaces to TS
                    if (ftype && ftype->type_->asClassType() && ftype->type_->asClassType()->class_->IsInterface())
                    {
                        jfunction->SetSkipLanguage(BINDINGLANGUAGE_JAVASCRIPT);
                    }

                    if (!ftype)
                    {
                        // if we don't have an initializer, the function cannot be bound
                        // as unscriptable type
                        if (!arg->hasInitializer())
                            return NULL;

                        // otherwise, break and the optional args will be ignored
                        break;
                    }

                    if (arg->hasInitializer())
                    {
                        ftype->initializer_ = arg->initializer()->chars();

                        if (arg->initializer()->_quotedString)
                            ftype->initializer_ = "\"" + ftype->initializer_ + "\"";

                    }

                    jfunction->AddParameter(ftype);
                }
                else
                {
                    return NULL;
                }

            }
        }

        jfunction->sourceLocation_ = function->sourceLocation();
        jfunction->fileName_ = function->fileName();
        jfunction->sourceLine_ = function->line();
        jfunction->sourceColumn_ = function->column();
        //const Token &token = unit_->tokenAt(function->sourceLocation());
        //const char* source = unit_->firstSourceChar() + token.byteOffset;

        String docString = parseDocString(function->line() - 1);

        if (docString.Length())
            jfunction->SetDocString(docString);

        return jfunction;

    }

    virtual bool visit(Namespace *nspace)
    {
        String name = getNameString(nspace->name());

        // LOGINFOF("Namespace: %s", name.CString());

        return true;

    }

    // reject template types
    virtual bool visit(Template *t)
    {
        return false;
    }

    // enums visited in preprocessor visitor
    virtual bool visit(Enum *penum)
    {
        return false;
    }

    // global var decl or function
    virtual bool visit(Declaration* decl)
    {
        if (decl->isTypedef())
            return true;

        FullySpecifiedType dtype = decl->type();

        Type* type = dtype.type();

        if (type->isNamedType())
        {
            String classname = getNameString(type->asNamedType()->name());
            if ( dtype.isConst() && classname == "String" )  // find a const string
            {
                const StringLiteral* sinit = decl->getInitializer();
                if (sinit)
                {
                    if (sinit->chars())
                    {
                        String svalue = sinit->chars();
                        module_->RegisterConstant(getNameString(decl->name()).CString(), svalue, JSBPrimitiveType::Char );
                        return true;  // tag this constant as a Char type, because there is no String Primitive type
                   }
                }
            }
        }

        if (type->isPointerType() || type->isReferenceType())
            return true;

        if (type->asEnumType())
            return true;

        bool _unsigned = false;

        if (dtype.isUnsigned())
            _unsigned = true;


        if (!type->asFloatType() && !type->asIntegerType() && !_unsigned)
        {
            return true;
        }

        String value;

        const StringLiteral* init = decl->getInitializer();
        if (init)
        {
            if (init->chars())
                value = init->chars();
        }

        if (dtype.isConst())
        {
            if (type->isIntegerType() || _unsigned)
            {
                module_->RegisterConstant(getNameString(decl->name()).CString(), value, JSBPrimitiveType::Int, _unsigned);
            }
            else
            {
                module_->RegisterConstant(getNameString(decl->name()).CString(), value, JSBPrimitiveType::Float);
            }
        }

        return true;

    }

    virtual bool visit(Class *klass)
    {
        String name = getNameString(klass->name());

        JSBClass* jclass = module_->GetClass(name, true);

        if (!jclass)
        {
            return false;
        }

        jclass->SetHeader(header_);

        String docString = parseDocString(klass->line() - 1);
        if (docString.Length())
            jclass->SetDocString(docString);

        for (unsigned i = 0; i < klass->baseClassCount(); i++)
        {
            BaseClass* baseclass = klass->baseClassAt(i);
            String baseclassname = getNameString(baseclass->name());

            JSBClass* base = JSBPackage::GetClassAllPackages(baseclassname);

            if (!base)
            {
                JSBClass* interface = JSBPackage::GetClassAllPackages(baseclassname, true);

                if (interface)
                {
                    jclass->AddInterface(interface);
                }
                else
                {
                    if (!i)
                    {
                        ATOMIC_LOGINFOF("Warning: %s first baseclass %s not in bindings", name.CString(), baseclassname.CString());
                    }
                }
            }
            else
            {
                jclass->SetBaseClass(base);
            }
        }

        for (unsigned i = 0; i < klass->memberCount(); i++)
        {
            Symbol* symbol = klass->memberAt(i);

            Declaration* decl = symbol->asDeclaration();

            // if the function describes the body in the header
            Function* function = symbol->asFunction();

            // otherwise it could be a decl
            if (!function && decl)
                function = decl->type()->asFunctionType();

            if (function)
            {
                if (function->isPureVirtual())
                    jclass->SetAbstract();

                // only want public functions
                if (!symbol->isPublic())
                    continue;

                JSBFunction* jfunction = processFunction(jclass, function);
                if (jfunction)
                    jclass->AddFunction(jfunction);
            }

        }
        // return false so we don't traverse the class members
        return false;
    }

};

}