infinitefusion-e18/Data/Scripts/050_Outfits/UI/clothesShop/HatsMartAdapter.rb

class HatsMartAdapter < OutfitsMartAdapter
  attr_accessor :worn_clothes
  attr_accessor :worn_clothes2

  DEFAULT_NAME = "[unknown]"
  DEFAULT_DESCRIPTION = "A headgear that trainers can wear."

  def initialize(stock = nil, isShop = nil, isSecondaryHat = false, prices_override = {})
    super(stock, isShop, isSecondaryHat, prices_override)
    @worn_clothes = $Trainer.hat
    @worn_clothes2 = $Trainer.hat2
    @second_hat_visible = true
  end

  # Used in shops only
  def toggleSecondHat()
    @second_hat_visible = !@second_hat_visible
    $Trainer.hat2 = @second_hat_visible ? @worn_clothes2 : nil
  end

  def toggleEvent(item)
    if isShop?
      toggleSecondHat
    else
      $Trainer.set_hat(nil, @is_secondary_hat)
      @worn_clothes = nil
    end
  end

  def set_secondary_hat(value)
    @is_secondary_hat = value
  end

  def is_wearing_clothes(outfit_id)
    return outfit_id == @worn_clothes || outfit_id == @worn_clothes2
  end

  def toggleText()
    return
    # return if @isShop
    # toggleKey = "D"#getMappedKeyFor(Input::SPECIAL)
    # return "Remove hat: #{toggleKey}"
  end

  def switchVersion(item, delta = 1)
    pbSEPlay("GUI storage put down", 80, 100)
    return toggleSecondHat if isShop?
    @is_secondary_hat = !@is_secondary_hat
  end

  def getName(item)
    return item.id
  end

  def getDescription(item)
    return DEFAULT_DESCRIPTION if !item.description
    return item.description
  end

  def getItemIcon(item)
    return Settings::BACK_ITEM_ICON_PATH if !item
    return getOverworldHatFilename(item.id)
  end

  def updateTrainerPreview(item, previewWindow)
    if item.is_a?(Outfit)
      hat1 = @is_secondary_hat ? get_hat_by_id($Trainer.hat) : item
      hat2 = @is_secondary_hat ? item : get_hat_by_id($Trainer.hat2)

      previewWindow.set_hat(hat1.id, false) if hat1
      previewWindow.set_hat(hat2.id, true) if hat2
      previewWindow.set_hat(nil, true) if !@second_hat_visible # for toggling in shops

      hat1_color = 0
      hat2_color = 0
      hat1_color = $Trainer.dyed_hats[hat1.id] if hat1 && $Trainer.dyed_hats.include?(hat1.id)
      hat2_color = $Trainer.dyed_hats[hat2.id] if hat2 && $Trainer.dyed_hats.include?(hat2.id)
      previewWindow.hat_color = hat1_color
      previewWindow.hat2_color = hat2_color

      $Trainer.hat = hat1&.id
      $Trainer.hat2 = hat2&.id
      $Trainer.hat_color = hat1_color
      $Trainer.hat2_color = hat2_color

    else
      $Trainer.set_hat(nil, @is_secondary_hat)
      previewWindow.set_hat(nil, @is_secondary_hat)
    end

    pbRefreshSceneMap
    previewWindow.updatePreview()
  end

  def get_dye_color(item_id)
    return if !item_id
    return 0 if isShop?
    $Trainer.dyed_hats = {} if !$Trainer.dyed_hats
    if $Trainer.dyed_hats.include?(item_id)
      return $Trainer.dyed_hats[item_id]
    end
    return 0
  end

  def set_dye_color(item, previewWindow, is_secondary_hat = false)
    return if !item
    if !isShop?

    else
      $Trainer.set_hat_color(0, is_secondary_hat)
      previewWindow.hat_color = 0
    end
  end

  # def set_dye_color(item,previewWindow,is_secondary_hat=false)
  #   return if !item
  #   if !isShop?
  #     $Trainer.dyed_hats= {} if !$Trainer.dyed_hats
  #
  #     echoln item.id
  #     echoln $Trainer.dyed_hats.include?(item.id)
  #     echoln $Trainer.dyed_hats[item.id]
  #
  #     if $Trainer.dyed_hats.include?(item.id)
  #       dye_color = $Trainer.dyed_hats[item.id]
  #       $Trainer.set_hat_color(dye_color,is_secondary_hat)
  #       previewWindow.hat_color = dye_color
  #     else
  #       $Trainer.set_hat_color(0,is_secondary_hat)
  #       previewWindow.hat_color=0
  #     end
  #     #echoln $Trainer.dyed_hats
  #   else
  #     $Trainer.set_hat_color(0,is_secondary_hat)
  #     previewWindow.hat_color=0
  #   end
  # end

  def addItem(item)
    return unless item.is_a?(Outfit)
    changed_clothes = obtainHat(item.id, @is_secondary_hat)
    if changed_clothes
      @worn_clothes = item.id
    end
  end

  def get_current_clothes()
    return $Trainer.hat(@is_secondary_hat)
  end

  def player_changed_clothes?()
    echoln("Trainer hat: #{$Trainer.hat}, Worn hat: #{@worn_clothes}")
    echoln("Trainer hat2: #{$Trainer.hat2}, Worn hat2: #{@worn_clothes2}")
    $Trainer.hat != @worn_clothes || $Trainer.hat2 != @worn_clothes2
  end

  def putOnSelectedOutfit()

    putOnHat($Trainer.hat, true, false) if $Trainer.hat
    putOnHat($Trainer.hat2, true, true) if $Trainer.hat2

    @worn_clothes = $Trainer.hat
    @worn_clothes2 = $Trainer.hat2

    playOutfitChangeAnimation()
    pbMessage(_INTL("You put on the hat(s)!\\wtnp[30]"))
  end

  def putOnOutfit(item)
    return unless item.is_a?(Outfit)
    putOnHat(item.id, false, @is_secondary_hat)
    @worn_clothes = item.id
  end

  def reset_player_clothes()
    $Trainer.set_hat(@worn_clothes, false)
    $Trainer.set_hat(@worn_clothes2, true)

    $Trainer.set_hat_color($Trainer.dyed_hats[@worn_clothes], false) if $Trainer.dyed_hats && $Trainer.dyed_hats[@worn_clothes]
    $Trainer.set_hat_color($Trainer.dyed_hats[@worn_clothes2], true) if $Trainer.dyed_hats && $Trainer.dyed_hats[@worn_clothes2]
  end

  def get_unlocked_items_list()
    return $Trainer.unlocked_hats
  end

  def getSpecialItemCaption(specialType)
    case specialType
    when :REMOVE_HAT
      return _INTL("Remove hat")
    end
    return nil
  end

  def getSpecialItemBaseColor(specialType)
    case specialType
    when :REMOVE_HAT
      return MessageConfig::BLUE_TEXT_MAIN_COLOR
    end
    return nil
  end

  def getSpecialItemShadowColor(specialType)
    case specialType
    when :REMOVE_HAT
      return MessageConfig::BLUE_TEXT_SHADOW_COLOR
    end
    return nil
  end
  def isBald?
    return true unless $Trainer.hair
    bald_ids = [HAIR_BALD, HAIR_BALD_POLISHED]
    hair_id = getSimplifiedHairIdFromFullID($Trainer.hair)
    return bald_ids.include?(hair_id)
  end
  def getSpecialItemDescription(specialType)
    hair_situation = isBald?() ? "bald head" : "fabulous hair"
    return _INTL("Go without a hat and show off your {1}!",hair_situation)
  end

  def doSpecialItemAction(specialType, item = nil)
    toggleEvent(item)
  end
end