class Trainer
  attr_accessor :safari_record
end

class Game_Temp
  attr_accessor :last_safari_stats
end

class SafariState
  attr_accessor :ballcount
  attr_accessor :decision
  attr_accessor :steps

  def initialize
    @start      = nil
    @ballcount  = 0
    @inProgress = false
    @steps      = 0
    @decision   = 0
    @caught_pokemon = []
  end

  def pbReceptionMap
    return @inProgress ? @start[0] : 0
  end

  def inProgress?
    return @inProgress
  end

  def recordLastSafari
    $game_temp.last_safari_stats = [@caught_pokemon, @steps]
  end

  def pbGoToStart
    start_map = Settings::KANTO ? 472 : MAP_SAFARI_ENTRANCE
    start_x = Settings::KANTO ? 28 : 8
    start_y = Settings::KANTO ? 11 : 7
    start_direction = Settings::KANTO ? DIRECTION_DOWN : DIRECTION_LEFT


    if $scene.is_a?(Scene_Map)
      pbFadeOutIn {
        $game_temp.player_transferring   = true
        $game_temp.transition_processing = true
        $game_temp.player_new_map_id    = start_map #@start[0]
        $game_temp.player_new_x         = start_x#@start[1]
        $game_temp.player_new_y         = start_y#@start[2]
        $game_temp.player_new_direction = start_direction
        $scene.transfer_player
      }
    end
    pbSafariState.pbEnd
  end

  def pbStart(ballcount)
    @start      = [$game_map.map_id,$game_player.x,$game_player.y,$game_player.direction]
    @ballcount  = ballcount
    @inProgress = true
    @steps      =  Settings::SAFARI_STEPS
  end

  def pbEnd
    recordLastSafari
    $PokemonGlobal.encounter_version = 0
    @start      = nil
    @ballcount  = 0
    @inProgress = false
    @steps      = 0
    @decision   = 0
    $game_map.need_refresh = true
    clear_all_images
    pbSafariState.caught_pokemon=nil
    disposeBallDisplay
  end
end

BALL_ICON_SIZE = 28
BALL_MARGIN    = 8
def updateBallDisplay
  return if !@inProgress
  createBallDisplay if !@ball_sprite || @ball_sprite.disposed?
  bmp = @ball_sprite.bitmap
  bmp.clear

  icon = RPG::Cache.picture("Safari/safariball_ui")
  bmp.blt(0, 0, icon, Rect.new(0, 0, icon.width, icon.height))
  baseColor   = pbColor(:LIGHT_TEXT_MAIN_COLOR)
  shadowColor = pbColor(:LIGHT_TEXT_SHADOW_COLOR)
  pbSetSystemFont(bmp)
  text_x = BALL_ICON_SIZE + 4
  pbDrawShadowText(bmp, text_x, 0, -1, -1, "x #{@ballcount}", baseColor, shadowColor)
end

def createBallDisplay
  disposeBallDisplay
  @ball_viewport = Viewport.new(0, 0, Graphics.width, Graphics.height)
  @ball_viewport.z = 99999
  @ball_sprite = Sprite.new(@ball_viewport)
  @ball_sprite.bitmap = Bitmap.new(120, BALL_ICON_SIZE)
  @ball_sprite.x = BALL_MARGIN
  @ball_sprite.y = BALL_MARGIN
end

def disposeBallDisplay
  if @ball_sprite
    @ball_sprite.bitmap.dispose if @ball_sprite.bitmap && !@ball_sprite.bitmap.disposed?
    @ball_sprite.dispose if !@ball_sprite.disposed?
    @ball_sprite = nil
  end
  if @ball_viewport
    @ball_viewport.dispose if !@ball_viewport.disposed?
    @ball_viewport = nil
  end
end
def has_caught_some_pokemon?
  return false unless $game_temp.last_safari_stats
  return false unless $game_temp.last_safari_stats.size >= 2
  return false unless $game_temp.last_safari_stats[0]
  return false unless $game_temp.last_safari_stats[1]
  return true
end
def tellLatestSafariStats(talking_event_id)
  pbCallBub(2, talking_event_id)
  pbMessage(_INTL("Welcome back! How was your Safari adventure?"))

  return unless has_caught_some_pokemon?

  nb_caught = $game_temp.last_safari_stats[0].size
  nb_steps = $game_temp.last_safari_stats[1].abs

  pbCallBub(2, talking_event_id)
  pbMessage(_INTL("You took \\C[1]{1}\\C[0] steps and caught \\C[1]{2}\\C[0] Pokémon!",nb_steps,nb_caught))
  $Trainer.safari_record = 0 unless $Trainer.safari_record
  if nb_caught > $Trainer.safari_record
    pbCallBub(2, talking_event_id)
    pbMessage(_INTL("Wow! That's a new personal record. Congratulations!"))
    $Trainer.safari_record = nb_caught
  else
    pbCallBub(2, talking_event_id)
    pbMessage(_INTL("I hope you had a lot of fun!"))
  end
end

Events.onMapChange += proc { |_sender,*args|
  pbSafariState.pbEnd if !pbInSafari?
}

def pbInSafari?
  if pbSafariState.inProgress?
    # Reception map is handled separately from safari map since the reception
    # map can be outdoors, with its own grassy patches.
    reception = pbSafariState.pbReceptionMap
    return true if $game_map.map_id == reception
    map_metadata = GameData::MapMetadata.try_get($game_map.map_id)
    return true if map_metadata && map_metadata.safari_map || is_hardcoded_safari_map($game_map.map_id)
  end
  return false
end

def is_hardcoded_safari_map(map_id)
  safari_maps = [445,74,484,75,485,82,486,107,487,715,718,488,717] if Settings::KANTO
  safari_maps = HOENN_SAFARI_MAPS if Settings::HOENN
  return safari_maps.include?(map_id)
end


def pbSafariState
  $PokemonGlobal.safariState = SafariState.new if !$PokemonGlobal.safariState
  return $PokemonGlobal.safariState
end

Events.onStepTakenTransferPossible += proc { |_sender,e|
  handled = e[0]
  next if handled[0]
  if pbInSafari? && pbSafariState.decision==0
    pbSafariState.steps -= 1
    if pbSafariState.steps==0
      pbMessage(_INTL("PA:  Ding-dong!\1"))
      pbMessage(_INTL("PA:  Your safari game is over!"))
      pbSafariState.decision = 1
      pbSafariState.pbGoToStart
      handled[0] = true
    end
  end
}

Events.onWildBattleOverride += proc { |_sender,e|
  species = e[0]
  level   = e[1]
  handled = e[2]
  next if handled[0]!=nil
  next if !pbInSafari?
  handled[0] = pbSafariBattle(species,level)
}

def pbSafariBattle(species,level)
  # Generate a wild Pokémon based on the species and level
  pkmn = pbGenerateWildPokemon(species,level)
  foeParty = [pkmn]
  # Calculate who the trainer is
  playerTrainer = $Trainer
  # Create the battle scene (the visual side of it)
  scene = pbNewBattleScene
  # Create the battle class (the mechanics side of it)
  battle = PokeBattle_SafariZone.new(scene,playerTrainer,foeParty)
  battle.ballCount = pbSafariState.ballcount
  pbPrepareBattle(battle)
  # Perform the battle itself
  decision = 0
  pbBattleAnimation(pbGetWildBattleBGM(foeParty),0,foeParty) {
    pbSceneStandby {
      decision = battle.pbStartBattle
    }
  }
  Input.update
  # Update Safari game data based on result of battle
  pbSafariState.ballcount = battle.ballCount
  if pbSafariState.ballcount<=0
    if decision!=2   # Last Safari Ball was used to catch the wild Pokémon
      pbMessage(_INTL("Announcer: You're out of Safari Balls! Game over!"))
    end
    pbSafariState.decision = 1
    pbSafariState.pbGoToStart
  end
  # Save the result of the battle in Game Variable 1
  #    0 - Undecided or aborted
  #    2 - Player ran out of Safari Balls
  #    3 - Player or wild Pokémon ran from battle, or player forfeited the match
  #    4 - Wild Pokémon was caught
  pbSet(1,decision)
  # Used by the Poké Radar to update/break the chain
  Events.onWildBattleEnd.trigger(nil,species,level,decision)
  # Return the outcome of the battle
  return decision
end