dungeon.rb

class Dungeon

  attr_reader :height, :width, :dungeon, :current, :stack

  def initialize(args = {})
    @height = args.fetch(:height, 1)
    @width = args.fetch(:width, 1)
    @dungeon = []
    @current
    @num_of_cells = @height * @width
    @visited_cells = []
    @stack = []
  end

  def create_dungeon
    (1..@height).each do |row|
      new_row = []
      (1..@width).each do |col|
        new_row << Cell.new(x: col - 1, y: row - 1)
      end
      @dungeon << new_row
    end
  end

  def check_neighbors(cell)
    neighbors = []
    top = nil
    right = nil
    bottom = nil
    left = nil

    if cell.y > 0
      top = @dungeon[cell.y - 1][cell.x]
    end
    if cell.x > 0
      left = @dungeon[cell.y][cell.x - 1]
    end
    if cell.x < @width - 1
      right = @dungeon[cell.y][cell.x + 1]
    end
    if cell.y < @height - 1
      bottom = @dungeon[cell.y + 1][cell.x]
    end

    if top
      if !top.visited
        neighbors << top
      end
    end
    if right
      if !right.visited
        neighbors << right
      end
    end
    if bottom
      if !bottom.visited
        neighbors << bottom
      end
    end
    if left
      if !left.visited
        neighbors << left
      end
    end

    if neighbors.length > 0
      return neighbors.sample
    else
      return nil
    end
  end

  def create_maze
    # start in top left corner
    @current = @dungeon[0][0]
    # p "y:#{current.y}, x:#{@current.x}"
    @current.visited = true
    @visited_cells << @current

    until @visited_cells.length == @num_of_cells
      # binding.pry
      next_cell = check_neighbors(@current)

      if next_cell
        next_cell.visited = true
        @stack << @current
        remove_walls(@current, next_cell)
        @current = next_cell
        # p "y:#{current.y}, x:#{@current.x}"
        @visited_cells << @current
      else
        @current = @stack.pop
      end
    end
  end

  def remove_walls(current_cell, next_cell)
    dx = current_cell.x - next_cell.x
    if dx == 1
      current_cell.walls[3] = false
      next_cell.walls[1] = false
    elsif dx == -1
      current_cell.walls[1] = false
      next_cell.walls[3] = false
    end

    dy = current_cell.y - next_cell.y
    if dy == 1
      current_cell.walls[0] = false
      next_cell.walls[2] = false
    elsif dy == -1
      current_cell.walls[2] = false
      next_cell.walls[0] = false
    end
  end

  def show_dungeon
    @dungeon.each do |row|
      row.each do |cell|
        #
      end
    end
  end

end