#include "security.hpp"

#include "../mwworld/class.hpp"
#include "../mwworld/player.hpp"

#include "../mwbase/world.hpp"
#include "../mwbase/environment.hpp"
#include "../mwbase/windowmanager.hpp"

#include "npcstats.hpp"
#include "creaturestats.hpp"

namespace MWMechanics
{

    void Security::pickLock(const MWWorld::Ptr &actor, const MWWorld::Ptr &lock, const MWWorld::Ptr &lockpick)
    {
        if (lock.getCellRef().mLockLevel <= 0)
            return;

        int lockStrength = lock.getCellRef().mLockLevel;

        float pickQuality = lockpick.get<ESM::Lockpick>()->mBase->mData.mQuality;

        CreatureStats& creatureStats = MWWorld::Class::get(actor).getCreatureStats(actor);
        NpcStats& npcStats = MWWorld::Class::get(actor).getNpcStats(actor);
        float pcAgility = creatureStats.getAttribute(ESM::Attribute::Agility).getModified();
        float pcLuck = creatureStats.getAttribute(ESM::Attribute::Luck).getModified();
        float securitySkill = npcStats.getSkill(ESM::Skill::Security).getModified();
        float fatigueTerm = creatureStats.getFatigueTerm();

        float fPickLockMult = MWBase::Environment::get().getWorld()->getStore().get<ESM::GameSetting>().find("fPickLockMult")->getFloat();

        float x = 0.2 * pcAgility + 0.1 * pcLuck + securitySkill;
        x *= pickQuality * fatigueTerm;
        x += fPickLockMult * lockStrength;

        bool isPlayer = actor == MWBase::Environment::get().getWorld()->getPlayer().getPlayer();

        if (x <= 0)
        {
            if (isPlayer)
                MWBase::Environment::get().getWindowManager()->messageBox("#{sLockImpossible}");
        }
        else
        {
            int roll = static_cast<float> (std::rand()) / RAND_MAX * 100;
            if (roll <= x)
            {
                MWWorld::Class::get(lock).unlock(lock);
                if (isPlayer)
                    MWBase::Environment::get().getWindowManager()->messageBox("#{sLockSuccess}");
                MWWorld::Class::get(actor).skillUsageSucceeded(actor, ESM::Skill::Security, 1);
            }
            else if (isPlayer)
                MWBase::Environment::get().getWindowManager()->messageBox("#{sLockFail}");
        }

        if (lockpick.getCellRef().mCharge == -1)
            lockpick.getCellRef().mCharge = lockpick.get<ESM::Lockpick>()->mBase->mData.mUses;
        --lockpick.getCellRef().mCharge;
        if (!lockpick.getCellRef().mCharge)
            lockpick.getRefData().setCount(0);
    }

    void Security::probeTrap(const MWWorld::Ptr &actor, const MWWorld::Ptr &trap, const MWWorld::Ptr &probe)
    {
        if (trap.getCellRef().mTrap  == "")
            return;

        CreatureStats& creatureStats = MWWorld::Class::get(actor).getCreatureStats(actor);
        NpcStats& npcStats = MWWorld::Class::get(actor).getNpcStats(actor);

        float probeQuality = probe.get<ESM::Probe>()->mBase->mData.mQuality;

        const ESM::Spell* trapSpell = MWBase::Environment::get().getWorld()->getStore().get<ESM::Spell>().find(trap.getCellRef().mTrap);
        float trapSpellPoints = trapSpell->mData.mCost;

        float pcAgility = creatureStats.getAttribute(ESM::Attribute::Agility).getModified();
        float pcLuck = creatureStats.getAttribute(ESM::Attribute::Luck).getModified();
        float securitySkill = npcStats.getSkill(ESM::Skill::Security).getModified();
        float fatigueTerm = creatureStats.getFatigueTerm();

        float fTrapCostMult = MWBase::Environment::get().getWorld()->getStore().get<ESM::GameSetting>().find("fTrapCostMult")->getFloat();

        float x = 0.2 * pcAgility + 0.1 * pcLuck + securitySkill;
        x += fTrapCostMult * trapSpellPoints;
        x *= probeQuality * fatigueTerm;

        bool isPlayer = actor == MWBase::Environment::get().getWorld()->getPlayer().getPlayer();

        if (x <= 0)
        {
            if (isPlayer)
                MWBase::Environment::get().getWindowManager()->messageBox("#{sTrapImpossible}");
        }
        else
        {
            int roll = static_cast<float> (std::rand()) / RAND_MAX * 100;
            if (roll <= x)
            {
                trap.getCellRef().mTrap = "";
                if (isPlayer)
                    MWBase::Environment::get().getWindowManager()->messageBox("#{sTrapSuccess}");
                MWWorld::Class::get(actor).skillUsageSucceeded(actor, ESM::Skill::Security, 0);
            }
            else if (isPlayer)
                MWBase::Environment::get().getWindowManager()->messageBox("#{sTrapFail}");
        }

        if (probe.getCellRef().mCharge == -1)
            probe.getCellRef().mCharge = probe.get<ESM::Probe>()->mBase->mData.mUses;
        --probe.getCellRef().mCharge;
        if (!probe.getCellRef().mCharge)
            probe.getRefData().setCount(0);
    }

}