Weapon Upgrade Item Requirement addon

=begin

  Addon to Selchar's Weapon Level Up + Weapon Upgrade Scene
  Allows setting up item requirement for weapon levelling

  by Ekhart

=end

module TH_Instance
  module Scene_EquipUpgrade

    # Upgrade Cost Array
    # UpgradeCost = { level => [item_ID, item_Quantity]}
    UpgradeCost = {
      1 => [90,1],
      2 => [90,1],
      3 => [90,2],
      4 => [90,2],
      5 => [90,3],
      6 => [91,1],
      7 => [91,1],
      8 => [91,2],
      9 => [91,3],
      10 => [92,1]
    }
  end

end

class Window_EquipUpgradeSelect < Window_Selectable

  def enable?(item)
    if item.is_a?(RPG::EquipItem) && item.can_level && item.level < item.max_level &&
      $game_party.gold >=  item.level_upgrade_price && has_material?(item.level)
      return true
    else
      return false
    end
  end

  def has_material?(item_level)
    # Check for Upgrade Material and Quantity
    material = TH_Instance::Scene_EquipUpgrade::UpgradeCost[item_level+1][0]
    material_quantity = TH_Instance::Scene_EquipUpgrade::UpgradeCost[item_level+1][1]
    return $game_party.items.include?($data_items[material]) &&
      $game_party.item_number($data_items[material]) >= material_quantity
  end
end

class RPG::EquipItem

  #Call this to increase an equip's level
  def level_up
    return unless can_level
    return if @level == @max_level
    @level += 1
    # Removes QTY of Upgrade Material, then levels up the weapon
    material = TH_Instance::Scene_EquipUpgrade::UpgradeCost[@level][0]
    amount = TH_Instance::Scene_EquipUpgrade::UpgradeCost[@level][1]
    $game_party.lose_item($data_items[material], amount)
    level_update
  end
end