chess_validator 0.2.1 → 0.2.6

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 1af34872a259842b2169becd1ee7570ceac96f7fc626e4c4306e7544fca7e25a
-  data.tar.gz: 821da801fa31123bd8ec1a6d151bb14364da5ad60d833b11b129ef205ec51411
+  metadata.gz: 02b8644ab9c87a26ef76b6779c8020ba4efc1a9d26a3be24bae05a1d80317563
+  data.tar.gz: ad114955aa50aacc935d43679c79b1d4396f8d4fcb882be752f093264351b7d7
 SHA512:
-  metadata.gz: dae4d729c77310493648f7c3ec78ee954b3f04055b94a9a7410a74a3028d1b86918a6161e148bdaa27eaf75249a3a80e9ee9e2c54ed69cfcae8250740a62ba54
-  data.tar.gz: d0dcfb4661d6c5f8b231207ad0cd96c7634ed84898c3d76331c10f64aad54f345d4ef10538e3f27fb428eadf686958f7ccd69ea48fbb18bd1ceb0d40a4bf2b05
+  metadata.gz: 6057987c04b5ed6bf8c3ba9ab09174930e8581fc41f4eee2da7d3c1913e072bc442aa8f5ff5594ce5787a4e7bb5787c5b5095a82ea3162c6d4322e657d001fa9
+  data.tar.gz: d3d9a0155b1615ca6f1bbb48a1746f07ad14c47f340a59ffac742c48e99d1c46b74029caa201668a5aace4a327d0653727c487e4c2400bc6bf75cb425ea06835

data/lib/chess_validator.rb

@@ -1,2 +1 @@
-require 'move_logic'
-require 'game_logic'
+require 'engine'

data/lib/engine.rb

@@ -0,0 +1,37 @@
+require 'move_logic'
+require 'game_logic'
+
+module ChessValidator
+  class Engine
+    class << self
+      def find_next_moves(fen_notation)
+        fen = PGN::FEN.new(fen_notation)
+        MoveLogic.next_moves(fen)
+      end
+
+      def find_next_moves_from_moves(moves)
+        fen = PGN::Game.new(moves).positions.last.to_fen
+        MoveLogic.next_moves(fen)
+      end
+
+      def make_random_move(fen_notation, pieces_with_moves)
+        piece_to_move = pieces_with_moves.sample
+        move = piece_to_move.valid_moves.sample
+        MoveLogic.make_move(piece_to_move, move, fen_notation)
+      end
+
+      def move(piece, move, fen_notation)
+        MoveLogic.make_move(piece, move, fen_notation)
+      end
+
+      def pieces(fen_notation)
+        fen = PGN::FEN.new(fen_notation)
+        BoardLogic.build_board(fen).values
+      end
+
+      def result(fen_notation)
+        GameLogic.find_game_result(fen_notation)
+      end
+    end
+  end
+end

data/lib/move_logic.rb

@@ -4,16 +4,6 @@ require 'pgn'
 module ChessValidator
   class MoveLogic
     class << self
-      def find_next_moves(fen_notation)
-        fen = PGN::FEN.new(fen_notation)
-        next_moves(fen)
-      end
-
-      def find_next_moves_from_moves(moves)
-        fen = PGN::Game.new(moves).positions.last.to_fen
-        next_moves(fen)
-      end
-
       def next_moves(fen)
         board = BoardLogic.build_board(fen)
         pieces = []
@@ -29,7 +19,20 @@ module ChessValidator
 
       def load_valid_moves(board, piece, fen)
         moves_for_piece(piece).each do |move|
-          piece.valid_moves << move if valid_move?(piece, board, move, fen)
+          if valid_move?(piece, board, move, fen)
+            piece.valid_moves << move
+            target = find_target(board, piece, move)
+            piece.targets << target if target
+          end
+        end
+      end
+
+      def find_target(board, piece, move)
+        if board[INDEX_KEY[move]]
+          board[INDEX_KEY[move]]
+        elsif piece.piece_type.downcase == 'p' && piece.position[0] != move[0]
+          en_passant_position = piece.color == 'w' ? move[0] + '5' : move[0] + '4'
+          board[INDEX_KEY[en_passant_position]]
         end
       end
 
@@ -51,7 +54,8 @@ module ChessValidator
       def with_next_move(piece, board, move)
         index = INDEX_KEY[move]
         new_board = board.clone
-        new_piece = Piece.new(piece.piece_type, index)
+        piece_type = resolve_piece_type(piece.piece_type, move)
+        new_piece = Piece.new(piece_type, index)
         new_board.delete(piece.square_index)
         new_board[index] = new_piece
         new_board = handle_castle(new_board, move) if castled?(piece, move)
@@ -59,6 +63,14 @@ module ChessValidator
         new_board
       end
 
+      def resolve_piece_type(piece_type, move)
+        if piece_type.downcase == 'p' && ['1', '8'].include?(move[1])
+          move[1] == '8' ? 'Q' : 'q'
+        else
+          piece_type
+        end
+      end
+
       def handle_en_passant(board, pawn_color, move)
         if pawn_color == 'w'
           index = INDEX_KEY[move[0] + '5']

data/lib/piece.rb

@@ -1,16 +1,16 @@
-require_relative './constants/square_key'
+require_relative './constants/move_key'
 
 module ChessValidator
   class Piece
-    attr_accessor :position, :piece_type, :color, :square_index, :valid_moves
+    attr_accessor :position, :piece_type, :color, :square_index, :valid_moves, :targets
 
     def initialize(char, square_index)
       @piece_type = char
       @square_index = square_index
       @color = get_color(char)
       @position = get_position(square_index)
-      @enemy_targets = []
       @valid_moves = []
+      @targets = []
     end
 
     def get_position(square_index)

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: chess_validator
 version: !ruby/object:Gem::Version
-  version: 0.2.1
+  version: 0.2.6
 platform: ruby
 authors:
 - Charles Ellison
@@ -18,7 +18,8 @@ extra_rdoc_files: []
 files:
 - lib/board_logic.rb
 - lib/chess_validator.rb
-- lib/constants/square_key.rb
+- lib/constants/move_key.rb
+- lib/engine.rb
 - lib/game_logic.rb
 - lib/move_logic.rb
 - lib/piece.rb