#include "ItemMgr.h"

#include "Inventory/ItemContainer.h"
#include "Inventory/Item.h"
#include <Network/CommonActorControl.h>

#include <Exd/ExdDataGenerated.h>
#include <Logging/Logger.h>
#include <Database/DatabaseDef.h>
#include <Service.h>

bool Sapphire::World::Manager::ItemMgr::isArmory( uint16_t containerId )
{
  return
    containerId == Common::ArmoryBody ||
    containerId == Common::ArmoryEar ||
    containerId == Common::ArmoryFeet ||
    containerId == Common::ArmoryHand ||
    containerId == Common::ArmoryHead ||
    containerId == Common::ArmoryLegs ||
    containerId == Common::ArmoryMain ||
    containerId == Common::ArmoryOff ||
    containerId == Common::ArmoryRing ||
    containerId == Common::ArmoryWaist ||
    containerId == Common::ArmoryWrist ||
    containerId == Common::ArmorySoulCrystal;
}


uint16_t Sapphire::World::Manager::ItemMgr::getCharaEquipSlotCategoryToArmoryId( Common::EquipSlotCategory slot )
{

  switch( slot )
  {
    case Common::EquipSlotCategory::CharaHead:
      return Common::ArmoryHead;

    case Common::EquipSlotCategory::CharaBody:
    //case Common::EquipSlotCategory::BodyDisallowHead:
    //case Common::EquipSlotCategory::BodyDisallowHandsLegsFeet:
    //case Common::EquipSlotCategory::BodyDisallowAll:
    //case Common::EquipSlotCategory::BodyDisallowHands:
    //case Common::EquipSlotCategory::BodyDisallowLegsFeet:
      return Common::ArmoryBody;

    case Common::EquipSlotCategory::CharaEars:
      return Common::ArmoryEar;

    case Common::EquipSlotCategory::CharaFeet:
      return Common::ArmoryFeet;

    case Common::EquipSlotCategory::CharaHands:
      return Common::ArmoryHand;

    case Common::EquipSlotCategory::CharaLegs:
    //case Common::EquipSlotCategory::LegsDisallowFeet:
      return Common::ArmoryLegs;

    case Common::EquipSlotCategory::CharaMainHand:
    //case Common::EquipSlotCategory::MainTwoHandedWeapon:
    //case Common::EquipSlotCategory::MainOrOffHand:
      return Common::ArmoryMain;

    case Common::EquipSlotCategory::CharaOffHand:
      return Common::ArmoryOff;

    case Common::EquipSlotCategory::CharaRing:
      return Common::ArmoryRing;

    case Common::EquipSlotCategory::CharaWaist:
      return Common::ArmoryWaist;

    case Common::EquipSlotCategory::CharaWrist:
      return Common::ArmoryWrist;

    case Common::EquipSlotCategory::CharaNeck:
      return Common::ArmoryNeck;

    case Common::EquipSlotCategory::CharaSoulCrystal:
      return Common::ArmorySoulCrystal;

    default:
      return 0;
  }
}


bool Sapphire::World::Manager::ItemMgr::isEquipment( uint16_t containerId )
{
  return containerId == Common::GearSet0;
}

bool Sapphire::World::Manager::ItemMgr::isOneHandedWeapon( Common::ItemUICategory weaponCategory )
{
  switch( weaponCategory )
  {
    case Common::ItemUICategory::AlchemistsPrimaryTool:
    case Common::ItemUICategory::ArmorersPrimaryTool:
    case Common::ItemUICategory::BotanistsPrimaryTool:
    case Common::ItemUICategory::CulinariansPrimaryTool:
    case Common::ItemUICategory::OnehandedConjurersArm:
    case Common::ItemUICategory::CarpentersPrimaryTool:
    case Common::ItemUICategory::FishersPrimaryTool:
    case Common::ItemUICategory::GladiatorsArm:
    case Common::ItemUICategory::GoldsmithsPrimaryTool:
    case Common::ItemUICategory::LeatherworkersPrimaryTool:
    case Common::ItemUICategory::MinersPrimaryTool:
    case Common::ItemUICategory::OnehandedThaumaturgesArm:
    case Common::ItemUICategory::WeaversPrimaryTool:
    case Common::ItemUICategory::BlacksmithsPrimaryTool:
      return true;
    default:
      return false;
  }
}

Sapphire::ItemPtr Sapphire::World::Manager::ItemMgr::loadItem( uint64_t uId )
{
  auto& exdData = Common::Service< Data::ExdDataGenerated >::ref();
  auto& db = Common::Service< Db::DbWorkerPool< Db::ZoneDbConnection > >::ref();

  //  1 catalogId, 2 stack, 3 reservedFlag, 4 signatureId, 5 flags, 6 durability, 7 refine, 8 materia_0, 9 materia_1,
  //  10 materia_2, 11 materia_3, 12 materia_4, 13 stain, 14 pattern, 15 buffer_0, 16 buffer_1, 17 buffer_2,
  //  18 buffer_3, 19 buffer_4
  auto query = db.getPreparedStatement( Db::CHARA_ITEMGLOBAL_SELECT );
  query->setUInt64( 1, uId );

  auto itemRes = db.query( query );
  if( !itemRes->next() )
    return nullptr;

  try
  {
    auto itemInfo = exdData.get< Sapphire::Data::Item >( itemRes->getUInt( 1 ) );
    bool isHq = itemRes->getUInt( 3 ) == 1;

    ItemPtr pItem = make_Item( uId,
                               itemRes->getUInt( 1 ),
                               isHq );

    pItem->setStackSize( itemRes->getUInt( 2 ) );
    pItem->setStain( itemRes->getUInt16( 13 ) );
    pItem->setDurability( itemRes->getInt16( 6 ) );

    return pItem;
  }
  catch( ... )
  {
    return nullptr;
  }
}


Sapphire::Common::ContainerType Sapphire::World::Manager::ItemMgr::getContainerType( uint32_t containerId )
{
  if( containerId < 5 )
  {
    return Common::Bag;
  }
  else if( containerId < 2000 )
  {
    return Common::GearSet;
  }
  else if( containerId < 3200 )
  {
    return Common::CurrencyCrystal;
  }
  else if( containerId < 3600 )
  {
    return Common::Armory;
  }
  else
  {
    return Common::Unknown;
  }
}

uint32_t Sapphire::World::Manager::ItemMgr::getNextUId()
{
  uint32_t charId = 0;

  auto& db = Common::Service< Db::DbWorkerPool< Db::ZoneDbConnection > >::ref();
  auto pQR = db.query( "SELECT MAX(ItemId) FROM charaglobalitem" );

  if( !pQR->next() )
    return 0x00500001;

  charId = pQR->getUInt( 1 ) + 1;
  if( charId < 0x00500001 )
    return 0x00500001;

  return charId;
}