Arcemu Vehicle PowerUps v3 - Code by Nirual

#include "Setup.h"

/*
 *  VehiclePower-Ups v3 - Code by Nirual
 *      http://tinyurl.com/cklfzag
 */

struct VehiclePowerUp
{
    uint32 creature_entry;
    uint32 vehicle_time;
    uint32 respawn_time;
};

std::map<uint32, VehiclePowerUp*> m_VehicleStorage;

class PowerUpVehicle : public CreatureAIScript
{
public:
    ADD_CREATURE_FACTORY_FUNCTION(PowerUpVehicle);
    PowerUpVehicle(Creature* pCreature) : CreatureAIScript(pCreature) {}

    void OnLastPassengerLeft( Unit *passenger )
    {
        _unit->Despawn(100, 0);
    }
};

class VehicleActivator : public GameObjectAIScript
{
public:
    VehicleActivator(GameObject*  goinstance) : GameObjectAIScript(goinstance) {}
    static GameObjectAIScript* Create(GameObject* GO) { return new VehicleActivator(GO); }

    void OnCreate()
    {
        VehiclePowerUp* power_up = m_VehicleStorage[_gameobject->GetEntry()];

        if(power_up == NULL)
        {
            sLog.Error("[VehiclePowerUp]", "Gameobject with entry %u should not have that script");
            return;
        }

        CreatureEntry = power_up->creature_entry;
        vehicle_time = power_up->vehicle_time;
        respawn_time = power_up->respawn_time;


        RegisterAIUpdateEvent(100);
    }

    void AIUpdate()
    {
        float d2 = 0;
        float dist = 0;
        Player* ret = NULL;

        for(set<Object*>::iterator itr = _gameobject->GetInRangePlayerSetBegin(); itr != _gameobject->GetInRangePlayerSetEnd(); ++itr)
        {
            d2 = (TO< Player* >(*itr))->GetDistanceSq(_gameobject);
            if(!ret || d2 < dist)
            {
                dist = d2;
                ret = TO< Player* >(*itr);
            }
        }

        if(ret == NULL)
            return;

        if(ret->GetCurrentVehicle() != NULL)
            return;

        if(_gameobject->GetDistance2dSq(ret) > 1.5)
            return;

        _gameobject->Despawn(1, respawn_time);


        Creature* pVehicle = ret->GetMapMgr()->CreateCreature(CreatureEntry);

        pVehicle->Load(CreatureProtoStorage.LookupEntry(CreatureEntry), ret->GetPositionX(), ret->GetPositionY(), ret->GetPositionZ(), ret->GetOrientation());
        pVehicle->m_loadedFromDB = true;
        pVehicle->SetFaction(ret->GetFaction());
        pVehicle->Phase(PHASE_SET, ret->GetPhase());
        pVehicle->m_noRespawn = true;
        pVehicle->PushToWorld(_gameobject->GetMapMgr());

        ret->EnterVehicle(pVehicle->GetGUID(), 200);

        pVehicle->Despawn(vehicle_time, 0);

    }


private:
    uint32 CreatureEntry, respawn_time, vehicle_time;

};

void SetupPowerUps(ScriptMgr* mgr)
{
    QueryResult * query = WorldDatabase.Query("SELECT * FROM vehicle_powerup;");
    if(query == NULL)
        return;

    uint32 gameobject_entry, creature_entry;
    uint32 i = 0;
    do
    {
        gameobject_entry = query->Fetch()[0].GetUInt32();
        creature_entry = query->Fetch()[1].GetUInt32();

        CreatureProto* pCreatureProto = CreatureProtoStorage.LookupEntry(creature_entry);
        CreatureInfo* pCreatureInfo = CreatureNameStorage.LookupEntry(creature_entry);

        if(GameObjectNameStorage.LookupEntry(gameobject_entry) == NULL)
            sLog.Error("[VehiclePowerUp]", "Gameobject with entry %u not found.", gameobject_entry );
        else if(pCreatureProto == NULL || pCreatureInfo == NULL)
            sLog.Error("[VehiclePowerUp]", "Creature with entry %u not found (gameobject id: %u).", creature_entry, gameobject_entry );
        else if(pCreatureProto->vehicleid == 0)
            sLog.Error("[VehiclePowerUp]", "Creature with entry %u is not a vehicle.", creature_entry );
        else
        {
            VehiclePowerUp* power_up = new VehiclePowerUp;
            power_up->creature_entry = creature_entry;
            power_up->vehicle_time = query->Fetch()[2].GetUInt32();
            power_up->respawn_time = query->Fetch()[3].GetUInt32();
            m_VehicleStorage[gameobject_entry] = power_up;


            mgr->register_creature_script(creature_entry, &PowerUpVehicle::Create);
            mgr->register_gameobject_script(gameobject_entry, &VehicleActivator::Create);
            i++;
        }
    } while (query->NextRow());

    delete query;
    printf("[VehiclePowerUp] Loaded %u gameobject(s). \n", i);
}