AndyObtiva/glimmer-dsl-swt

# Copyright (c) 2007-2024 Andy Maleh
#
# Permission is hereby granted, free of charge, to any person obtaining
# a copy of this software and associated documentation files (the
# "Software"), to deal in the Software without restriction, including
# without limitation the rights to use, copy, modify, merge, publish,
# distribute, sublicense, and/or sell copies of the Software, and to
# permit persons to whom the Software is furnished to do so, subject to
# the following conditions:
#
# The above copyright notice and this permission notice shall be
# included in all copies or substantial portions of the Software.
#
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
# EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
# MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
# NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
# LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
# OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
# WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.

require_relative 'cell'

class Battleship
  module Model
    class Ship
      ORIENTATIONS = [:horizontal, :vertical]
      
      attr_reader :ship_collection, :name, :length
      attr_accessor :orientation, :top_left_cell, :sunk
      alias sunk? sunk
            
      def initialize(ship_collection, name, length)
        @ship_collection = ship_collection
        @name = name
        @length = length
        @orientation = :horizontal
      end
      
      def to_s
        "#{name.titlecase} (#{length} cells)"
      end
      
      def top_left_cell=(a_cell)
        if a_cell
          raise "Top left cell #{a_cell} already occupied by ship #{a_cell.ship.name}" if a_cell.ship && a_cell.ship != self
          raise "Top left cell #{a_cell} cannot fit ship #{name}" if a_cell.column_index + length > Grid::WIDTH
          grid = ship_collection.game.grids[ship_collection.player]
          1.upto(length - 1).each do |index|
            if orientation == :horizontal
              other_cell = grid.cell_rows[a_cell.row_index][a_cell.column_index + index]
            else
              other_cell = grid.cell_rows[a_cell.row_index + index][a_cell.column_index]
            end
            raise "Cell #{other_cell} already occupied by ship #{other_cell.ship.name}" if other_cell.ship && other_cell.ship != self
          end
        end
        @top_left_cell = a_cell
      end
      
      def orientation=(value)
        if top_left_cell
          if value == :horizontal
            if top_left_cell.column_index + length > Grid::WIDTH
              raise "Top left cell #{top_left_cell} cannot fit ship #{name}"
            end
          else
            if top_left_cell.row_index + length > Grid::HEIGHT
              raise "Top left cell #{top_left_cell} cannot fit ship #{name}"
            end
          end
          if cells(value)[1..-1].map(&:ship).reject {|s| s == self}.any?
            raise 'Orientation #{value} cells occupied by another ship'
          end
          if value != @orientation
            cells.each(&:reset!)
            @orientation = value
            cells.each_with_index do |cell, index|
              cell.ship = self
              cell.ship_index = index
            end
          end
        end
      end
        
      def cells(for_orientation = nil)
        for_orientation ||= orientation
        if top_left_cell
          length.times.map do |index|
            if for_orientation == :horizontal
              top_left_cell.grid.cell_rows[top_left_cell.row_index][top_left_cell.column_index + index]
            else
              top_left_cell.grid.cell_rows[top_left_cell.row_index + index][top_left_cell.column_index]
            end
          end
        end
      end
      
      def all_cells_hit?
        cells&.map(&:hit)&.all? {|h| h == true}
      end
      
      def reset!
        self.sunk = nil
        self.top_left_cell = nil
        self.orientation = :horizontal
      end
      
      def toggle_orientation!
        self.orientation = ORIENTATIONS[(ORIENTATIONS.index(orientation) + 1) % ORIENTATIONS.length]
      end
    end
  end
end