#!/usr/bin/ruby

class DepthCharge

  def run_game
    output_title()
    while true
      printf("----------\n")
      print_instructions()
      setup_game()
      printf("\n")
      game_loop()
      break if ! get_input_another_game()
    end

    printf("OK.  HOPE YOU ENJOYED YOURSELF.\n")
  end

  def output_title
    printf("--- DEPTH CHARGE ---\n")
    printf("CREATIVE COMPUTING  MORRISTOWN, NEW JERSEY\n")
    printf("\n")
  end

  def get_input_y_or_n(message)
    while true
      print(message)

      value = gets.chomp

      if (value == 'Y' || value == 'y')
        return true
      elsif value == 'N' || value == 'n'
        return false
      end

      printf("PLEASE ENTER Y/y OR N/n...\n\n")
    end
  end

  def get_input_positive_integer(message)

    while true
      print(message)
      value = gets.chomp
      if (value == 'd')
        debug_game()
        next
      end

      the_input = Integer(value) rescue nil

      if the_input == nil || the_input < 1
        printf("PLEASE ENTER A POSITIVE NUMBER\n\n")
        next

      end

      return the_input
    end
  end

  def print_instructions
    printf( <<~INSTRUCTIONS
YOU ARE THE CAPTAIN OF THE DESTROYER USS COMPUTER
AN ENEMY SUB HAS BEEN CAUSING YOU TROUBLE.  YOUR
MISSION IS TO DESTROY IT.

SPECIFY DEPTH CHARGE EXPLOSION POINT WITH A
TRIO OF NUMBERS -- THE FIRST TWO ARE THE
SURFACE COORDINATES (X, Y):
     WEST < X < EAST
    SOUTH < Y < NORTH

THE THIRD IS THE DEPTH (Z):
  SHALLOW < Z < DEEP

GOOD LUCK !

    INSTRUCTIONS
    )
  end

  def debug_game
    printf("@enemy_x: %d\n", @enemy_x)
    printf("@enemy_y: %d\n", @enemy_y)
    printf("@enemy_z: %d\n", @enemy_z)
    printf("@num_tries: %d\n", @num_tries)
    printf("@trial: %d\n", @trial)
    printf("\n")
  end

  def setup_game
    @search_area_dimension = get_input_positive_integer("DIMENSION OF SEARCH AREA: ")

    @num_tries = Integer(
      Math.log(@search_area_dimension)/Math.log(2) + 1
    )
    setup_enemy()
  end

  def setup_enemy
    @enemy_x = rand(1..@search_area_dimension)
    @enemy_y = rand(1..@search_area_dimension)
    @enemy_z = rand(1..@search_area_dimension)
  end

  def game_loop
    for @trial in 1..@num_tries do
      output_game_status()

      @shot_x = get_input_positive_integer("X: ")
      @shot_y = get_input_positive_integer("Y: ")
      @shot_z = get_input_positive_integer("Z: ")

      if (
        (@enemy_x - @shot_x).abs \
        + (@enemy_y - @shot_y).abs \
        + (@enemy_z - @shot_z).abs \
        == 0
      )
        you_win()
        return
      else
        missed_shot()
      end
    end

    printf("\n")

    you_lose()

  end

  def output_game_status
    printf("YOU HAVE %d SHOTS REMAINING.\n", @num_tries - @trial + 1)
    printf("TRIAL \#%d\n", @trial)
  end
  def you_win
    printf("\nB O O M ! ! YOU FOUND IT IN %d TRIES!\n\n", @trial )
  end
  def missed_shot
    missed_directions = []

    if @shot_x > @enemy_x
      missed_directions.push('TOO FAR EAST')
    elsif @shot_x < @enemy_x
      missed_directions.push('TOO FAR WEST')
    end

    if @shot_y > @enemy_y
      missed_directions.push('TOO FAR NORTH')
    elsif @shot_y < @enemy_y
      missed_directions.push('TOO FAR SOUTH')
    end

    if @shot_z > @enemy_z
      missed_directions.push('TOO DEEP')
    elsif @shot_z < @enemy_z
      missed_directions.push('TOO SHALLOW')
    end

    printf("SONAR REPORTS SHOT WAS: \n")
    printf("%s\n", "\t" + missed_directions.join("\n\t"))
  end

  def you_lose
    printf("YOU HAVE BEEN TORPEDOED!  ABANDON SHIP!\n")
    printf("THE SUBMARINE WAS AT %d %d %d\n", @enemy_x, @enemy_y, @enemy_z)

  end

  def get_input_another_game
    return get_input_y_or_n("ANOTHER GAME (Y OR N): ")
  end
end

game = DepthCharge.new
game.run_game()