/*
* All or portions of this file Copyright (c) Amazon.com, Inc. or its affiliates or
* its licensors.
*
* For complete copyright and license terms please see the LICENSE at the root of this
* distribution (the "License"). All use of this software is governed by the License,
* or, if provided, by the license below or the license accompanying this file. Do not
* remove or modify any license notices. This file is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
*
*/
// Original file Copyright Crytek GMBH or its affiliates, used under license.

#include "CryLegacy_precompiled.h"
#include "ScriptBind_Inventory.h"
#include "Inventory.h"
#include "IGameObject.h"


//------------------------------------------------------------------------
CScriptBind_Inventory::CScriptBind_Inventory(ISystem* pSystem, IGameFramework* pGameFramework)
    : m_pSystem(pSystem)
    , m_pEntitySystem(gEnv->pEntitySystem)
    , m_pGameFramework(pGameFramework)
{
    Init(m_pSystem->GetIScriptSystem(), m_pSystem, 1);

    RegisterMethods();
    RegisterGlobals();
}

//------------------------------------------------------------------------
CScriptBind_Inventory::~CScriptBind_Inventory()
{
}

//------------------------------------------------------------------------
void CScriptBind_Inventory::AttachTo(CInventory* pInventory)
{
    IScriptTable* pScriptTable = pInventory->GetEntity()->GetScriptTable();

    if (pScriptTable)
    {
        SmartScriptTable thisTable(m_pSS);

        thisTable->SetValue("__this", ScriptHandle(pInventory));
        thisTable->Delegate(GetMethodsTable());

        pScriptTable->SetValue("inventory", thisTable);
    }
}

//------------------------------------------------------------------------
void CScriptBind_Inventory::DetachFrom(CInventory* pInventory)
{
    IScriptTable* pScriptTable = pInventory->GetEntity()->GetScriptTable();

    if (pScriptTable)
    {
        pScriptTable->SetToNull("inventory");
    }
}

//------------------------------------------------------------------------
void CScriptBind_Inventory::RegisterGlobals()
{
}

//------------------------------------------------------------------------
void CScriptBind_Inventory::RegisterMethods()
{
#undef SCRIPT_REG_CLASSNAME
#define SCRIPT_REG_CLASSNAME &CScriptBind_Inventory::

    SCRIPT_REG_TEMPLFUNC(Destroy, "");
    SCRIPT_REG_TEMPLFUNC(Clear, "");

    SCRIPT_REG_TEMPLFUNC(Dump, "");

    SCRIPT_REG_TEMPLFUNC(GetItemByClass, "className");
    SCRIPT_REG_TEMPLFUNC(GetGrenadeWeaponByClass, "className");
    SCRIPT_REG_TEMPLFUNC(HasAccessory, "accessoryName");

    SCRIPT_REG_TEMPLFUNC(GetCurrentItemId, "");
    SCRIPT_REG_TEMPLFUNC(GetCurrentItem, "");
    SCRIPT_REG_TEMPLFUNC(GetAmmoCapacity, "ammoName");
    SCRIPT_REG_TEMPLFUNC(GetAmmoCount, "ammoName");
    SCRIPT_REG_TEMPLFUNC(SetAmmoCount, "ammoName, count");
}

//------------------------------------------------------------------------
CInventory* CScriptBind_Inventory::GetInventory(IFunctionHandler* pH)
{
    return (CInventory*)pH->GetThis();
}

//------------------------------------------------------------------------
int CScriptBind_Inventory::Destroy(IFunctionHandler* pH)
{
    CInventory* pInventory = GetInventory(pH);
    CRY_ASSERT(pInventory);

    pInventory->Destroy();
    return pH->EndFunction();
}

//------------------------------------------------------------------------
int CScriptBind_Inventory::Clear(IFunctionHandler* pH)
{
    CInventory* pInventory = GetInventory(pH);
    CRY_ASSERT(pInventory);

    pInventory->Clear();
    return pH->EndFunction();
}

//------------------------------------------------------------------------
int CScriptBind_Inventory::Dump(IFunctionHandler* pH)
{
#if DEBUG_INVENTORY_ENABLED
    CInventory* pInventory = GetInventory(pH);
    CRY_ASSERT(pInventory);

    pInventory->Dump();
#endif

    return pH->EndFunction();
}

//------------------------------------------------------------------------
int CScriptBind_Inventory::GetItemByClass(IFunctionHandler* pH, const char* className)
{
    CInventory* pInventory = GetInventory(pH);
    CRY_ASSERT(pInventory);

    IEntityClass* pClass = gEnv->pEntitySystem->GetClassRegistry()->FindClass(className);
    EntityId itemId = pInventory->GetItemByClass(pClass);
    if (itemId)
    {
        return pH->EndFunction(ScriptHandle(itemId));
    }

    return pH->EndFunction();
}

//----------------------------------------------------------------------
int CScriptBind_Inventory::GetGrenadeWeaponByClass(IFunctionHandler* pH, const char* className)
{
    CInventory* pInventory = GetInventory(pH);
    CRY_ASSERT(pInventory);

    IEntityClass* pClass = gEnv->pEntitySystem->GetClassRegistry()->FindClass(className);
    EntityId itemId = pInventory->GetItemByClass(pClass, NULL);
    if (itemId)
    {
        return pH->EndFunction(ScriptHandle(itemId));
    }

    return pH->EndFunction();
}

//-------------------------------------------------------------------
int CScriptBind_Inventory::HasAccessory(IFunctionHandler* pH, const char* accessoryName)
{
    CInventory* pInventory = GetInventory(pH);
    CRY_ASSERT(pInventory);

    IEntityClass* pClass = gEnv->pEntitySystem->GetClassRegistry()->FindClass(accessoryName);
    if (pClass)
    {
        return pH->EndFunction(pInventory->HasAccessory(pClass));
    }

    return pH->EndFunction();
}

//------------------------------------------------------------------------
int CScriptBind_Inventory::GetCurrentItemId(IFunctionHandler* pH)
{
    CInventory* pInventory = GetInventory(pH);
    CRY_ASSERT(pInventory);

    EntityId currentItemId = pInventory->GetCurrentItem();
    if (currentItemId)
    {
        return pH->EndFunction(ScriptHandle(currentItemId));
    }
    return pH->EndFunction();
}

//------------------------------------------------------------------------
int CScriptBind_Inventory::GetCurrentItem(IFunctionHandler* pH)
{
    CInventory* pInventory = GetInventory(pH);
    CRY_ASSERT(pInventory);

    EntityId currentItemId = pInventory->GetCurrentItem();
    if (currentItemId)
    {
        IEntity* pEntity = m_pEntitySystem->GetEntity(currentItemId);
        if (pEntity)
        {
            return pH->EndFunction(pEntity->GetScriptTable());
        }
    }
    return pH->EndFunction();
}

//------------------------------------------------------------------------
int CScriptBind_Inventory::GetAmmoCapacity(IFunctionHandler* pH, const char* ammoName)
{
    CInventory* pInventory = GetInventory(pH);
    CRY_ASSERT(pInventory);

    IEntityClass* pClass = gEnv->pEntitySystem->GetClassRegistry()->FindClass(ammoName);
    CRY_ASSERT(pClass);
    return pH->EndFunction(pInventory->GetAmmoCapacity(pClass));
}

//------------------------------------------------------------------------
int CScriptBind_Inventory::GetAmmoCount(IFunctionHandler* pH, const char* ammoName)
{
    CInventory* pInventory = GetInventory(pH);
    CRY_ASSERT(pInventory);

    IEntityClass* pClass = gEnv->pEntitySystem->GetClassRegistry()->FindClass(ammoName);
    CRY_ASSERT(pClass);
    return pH->EndFunction(pInventory->GetAmmoCount(pClass));
}

//------------------------------------------------------------------------
int CScriptBind_Inventory::SetAmmoCount(IFunctionHandler* pH, const char* ammoName, int count)
{
    CInventory* pInventory = GetInventory(pH);
    CRY_ASSERT(pInventory);

    IEntityClass* pClass = gEnv->pEntitySystem->GetClassRegistry()->FindClass(ammoName);
    CRY_ASSERT(pClass);
    if (pClass)
    {
        pInventory->SetAmmoCount(pClass, count);
        if (gEnv->bServer)
        {
            pInventory->GetActor()->GetGameObject()->InvokeRMI(CInventory::Cl_SetAmmoCount(),
                TRMIInventory_Ammo(ammoName, count),
                eRMI_ToRemoteClients);
        }
    }
    else
    {
        CryWarning(VALIDATOR_MODULE_GAME, VALIDATOR_ERROR, "Ammo class %s not found!", ammoName);
    }

    return pH->EndFunction();
}