terminal_chess 0.1.2 → 0.2.0

data/lib/local_chess_client.rb

@@ -0,0 +1,45 @@
+#!/usr/bin/env ruby
+
+$LOAD_PATH << __FILE__ # '.'
+
+require_relative "terminal_chess/version"
+require_relative "printer.rb"
+require_relative "move.rb"
+require_relative "board.rb"
+
+class LocalChessClient
+  def initialize
+    @board = Board.new
+    @turn_by_turn_playback = []
+    @board.display_board
+
+    start
+  end
+
+  def start
+    # Gameplay
+    loop do
+      if @board.checkmate
+        puts "\nTurn by Turn Playback : #{@turn_by_turn_playback}\n"
+        exit
+      end
+
+      print "\nPiece to Move [#{@board.player_turn.capitalize}]: "
+      from = gets.chomp.upcase
+
+      begin
+        print "Valid destinations: #{@board.valid_destinations(from).join(", ")}"
+
+        print "\nLocation: "
+        to = gets.chomp.upcase
+        @board.move(from, to)
+
+      rescue Exception => e
+        puts "Invalid selection #{e if !ENV["DEV"].nil?}"
+      else
+        @turn_by_turn_playback << [from, to]
+      end
+    end
+  end
+end
+

data/lib/move.rb

@@ -1,61 +1,63 @@
 #!/usr/bin/env ruby
 
-module MOVE
+module Move
 
   def constants(piece_mapping, color, piece)
     @@pieces = piece_mapping
-    @@color = color
-    @@enemy_color = (["black", "red"] - ["#{color.downcase}"]).first
-    @@type = piece
+    @@color  = color
+    @@type   = piece
+    @@enemy_color = ([:black, :red] - [color]).first
   end
 
 
   # Calls methods below to return a list of positions which are valid moves
   # for piece at index p1, given the current board layout as defined in manifest
   def possible_moves(p1, manifest, castling = false)
+    return [] if manifest[p1][:type].nil?
+
     allowed   = []
-    type      = manifest[p1]["type"]
-    my_color  = manifest[p1]["color"]
-    constants(manifest, my_color, type)
+    type      = manifest[p1][:type]
+    my_color  = manifest[p1][:color]
 
-    unless unoccupied?(p1)
+    constants(manifest, my_color, type)
 
-      if type == "king"
-        allowed += [move_lateral(p1, 1)].flatten
-        allowed += [move_diagonal(p1, 1)].flatten
-        allowed += [castle(p1)].flatten if castling
+    return [] if unoccupied?(p1)
 
-      elsif type == "queen"
-        allowed += [move_lateral(p1)].flatten
-        allowed += [move_diagonal(p1)].flatten
+    if type == :king
+      allowed += [move_lateral(p1, 1)].flatten
+      allowed += [move_diagonal(p1, 1)].flatten
+      allowed += [castle(p1)].flatten if castling
 
-      elsif type == "rook"
-        allowed += [move_lateral(p1)].flatten
+    elsif type == :queen
+      allowed += [move_lateral(p1)].flatten
+      allowed += [move_diagonal(p1)].flatten
 
-      elsif type == "bishop"
-        allowed += [move_diagonal(p1)].flatten
+    elsif type == :rook
+      allowed += [move_lateral(p1)].flatten
 
-      elsif type == "pawn"
-        allowed += [pawn(p1)].flatten
+    elsif type == :bishop
+      allowed += [move_diagonal(p1)].flatten
 
-      elsif type == "knight"
-        allowed += [knight(p1)].flatten
-      end
+    elsif type == :pawn
+      allowed += [move_pawn(p1)].flatten
 
+    elsif type == :knight
+      allowed += [move_knight(p1)].flatten
     end
-    return allowed
+
+    allowed
   end
 
 
   # Returns all valid positions a pawn at index p1 can move to
-  def pawn(p1)
+  def move_pawn(p1)
     row = get_row_from_index(p1)
     col = get_col_from_index(p1)
     valid = []
 
     # Piece color defines direction of travel. Enemy presence defines
     # the validity of diagonal movements
-    if @@color == "red"
+    if @@color == :red
       if unoccupied?(p1 - 8)
         valid << (p1 - 8)
       end
@@ -66,10 +68,12 @@ module MOVE
         valid << (p1 - 9)
       end
       # Only if the pieces is unmoved, can it move forward two rows
-      if !@@pieces[p1]["moved"] && unoccupied?(p1 - 16) && unoccupied?(p1 - 8)
+      if !@@pieces[p1][:moved] && unoccupied?(p1 - 8) && unoccupied?(p1 - 16)
         valid << (p1 - 16)
       end
-    elsif @@color == "black"
+
+    elsif @@color == :black
+
       if unoccupied?(p1 + 8)
         valid << (p1 + 8)
       end
@@ -79,21 +83,22 @@ module MOVE
       if piece_color(p1 + 9) == @@enemy_color && col < 8
         valid << (p1 + 9)
       end
-      if !@@pieces[p1]["moved"] && unoccupied?(p1 + 16) && unoccupied?(p1 + 8)
+      if !@@pieces[p1][:moved] && unoccupied?(p1 + 8) && unoccupied?(p1 + 16)
         valid << (p1 + 16)
       end
     end
 
-    return valid
+    valid
   end
 
 
   # Returns valid positions a knight at index p1 can move to
-  def knight(p1)
+  def move_knight(p1)
     row = get_row_from_index(p1)
     col = get_col_from_index(p1)
+
     valid = []
-    valid_no_ff = []
+    valid_moves_no_friendly_fire = []
 
     # Valid knight moves based on its board position
     if row < 7 && col < 8
@@ -125,11 +130,11 @@ module MOVE
     # This iterator filters for friendly fire, and removes indexes pointing to same color pices
     valid.each do |pos|
       unless piece_color(pos) == @@color
-        valid_no_ff << pos
+        valid_moves_no_friendly_fire << pos
       end
     end
 
-    return valid_no_ff
+    valid_moves_no_friendly_fire
   end
 
 
@@ -140,66 +145,70 @@ module MOVE
   def move_lateral(index, limit = 8)
     row = get_row_from_index(index)
     col = get_col_from_index(index)
+
     left, right = [col-1, limit].min, [8-col, limit].min
     up, down = [row-1, limit].min, [8-row, limit].min
+
     valid = []
 
     # Move down N places until board limit, piece in the way, or specified limit
     down.times do |i|
-        next_pos = index + (i+1)*8
-        # Valid move if position is unoccupied
-        if unoccupied?(next_pos)
+      next_pos = index + (i+1)*8
+      # Valid move if position is unoccupied
+      if unoccupied?(next_pos)
+        valid << next_pos
+      else
+        # Valid move is piece is an enemy, but then no subsequent tiles are attackable
+        # if the piece is not an enemy, it's not added as a valid move, and no subsequent tiles are attackable
+        # This function doesn't filter out the king from a valid enemy, but the Board class will drop King indexes
+        if piece_color(next_pos) == @@enemy_color
           valid << next_pos
-        else
-          # Valid move is piece is an enemy, but then no subsequent tiles are attackable
-          # if the piece is not an enemy, it's not added as a valid move, and no subsequent tiles are attackable
-          # This function doesn't filter out the king from a valid enemy, but the Board class will drop King indexes
-          if piece_color(next_pos) == @@enemy_color
-            valid << next_pos
-          end
-          break
         end
+
+        break
+      end
     end
 
     # Move up N places until board limit, piece in the way, or specified limit
     up.times do |i|
-        next_pos = index - (i+1)*8
-        if unoccupied?(next_pos)
+      next_pos = index - (i+1)*8
+      if unoccupied?(next_pos)
+        valid << next_pos
+      else
+        if piece_color(next_pos) == @@enemy_color
           valid << next_pos
-        else
-          if piece_color(next_pos) == @@enemy_color
-            valid << next_pos
-          end
-          break
         end
+        break
+      end
     end
 
     # Move right N places until board limit, piece in the way, or specified limit
     right.times do |i|
-        next_pos = index + (i+1)
-        if unoccupied?(next_pos)
+      next_pos = index + (i+1)
+      if unoccupied?(next_pos)
+        valid << next_pos
+      else
+        if piece_color(next_pos) == @@enemy_color
           valid << next_pos
-        else
-          if piece_color(next_pos) == @@enemy_color
-            valid << next_pos
-          end
-          break
         end
+        break
+      end
     end
 
     # Move left N places until board limit, piece in the way, or specified limit
     left.times do |i|
-        next_pos = index - (i+1)
-        if unoccupied?(next_pos)
+      next_pos = index - (i+1)
+      if unoccupied?(next_pos)
+        valid << next_pos
+      else
+        if piece_color(next_pos) == @@enemy_color
           valid << next_pos
-        else
-          if piece_color(next_pos) == @@enemy_color
-            valid << next_pos
-          end
-          break
         end
+        break
+      end
     end
-    return valid
+
+    valid
   end
 
 
@@ -271,7 +280,7 @@ module MOVE
       end
     end
 
-    return valid
+    valid
   end
 
   # Castle: king cannot move into check, or through check
@@ -279,70 +288,51 @@ module MOVE
     valid = []
     dangerous_tiles = attack_vectors
 
-    # Valid positions for a King to be in order to castle
-    if index == 5 || index == 61
-      # Empty space between a King and a Rook
-      if unoccupied?(index - 1) && unoccupied?(index - 2) && unoccupied?(index - 3) && @@pieces[index - 4]["moved"] == false
-        # Ensure king does not move through check or into check
-        if !dangerous_tiles.include?(index - 1) && !dangerous_tiles.include?(index - 2)
-          # King's castle position
-          valid << index - 2
-        end
-      elsif unoccupied?(index + 1) && unoccupied?(index + 2) && @@pieces[index + 3]["moved"] == false
-        if !dangerous_tiles.include?(index + 1) && !dangerous_tiles.include?(index + 2)
-          valid << index + 2
-        end
-      end
+    # King may never have moved
+    return valid unless [5, 61].include?(index) && @@pieces[index][:moved] == false
+
+    # Ensure empty space between a King and a Rook
+    if unoccupied?(index - 1) && unoccupied?(index - 2) && unoccupied?(index - 3) && @@pieces[index - 4][:moved] == false
+      # Ensure king does not move through check or into check, and then add its castle position
+      valid << index - 2 if !dangerous_tiles.include?(index - 1) && !dangerous_tiles.include?(index - 2)
+    end
+
+    if unoccupied?(index + 1) && unoccupied?(index + 2) && @@pieces[index + 3][:moved] == false
+      valid << index + 2 if !dangerous_tiles.include?(index + 1) && !dangerous_tiles.include?(index + 2)
     end
-    return valid
+
+    valid
   end
 
 
   # Check if board tile currently has a piece
   def unoccupied?(index)
-    if @@pieces[index]["color"].nil?
-      return true
-    else
-      return false
-    end
+    @@pieces[index][:color] == nil ? true : false
   end
 
-
   # Return true if the piece has moved before
   def moved?(index)
-    if @@pieces[index]["moved"]
-      return true
-    else
-      return false
-    end
+    @@pieces[index][:moved] ? true : false
   end
 
-
   # Return piece color ("red" or "black") from index (1 - 64)
   def piece_color(index)
-    return @@pieces[index]["color"]
+    @@pieces[index][:color]
   end
 
-
   # Method used when moving, to verify the piece at index (1 - 64) is not of type "king"
   def not_king(index)
-    return @@piece_locations[index]["type"] == "king"
+    @@piece_locations[index][:type] == :king
   end
 
-
   # Obtain chess board row number (1 + 8) from an index (1 - 64)
   def get_row_from_index(index)
-    return (index - 1)/8 + 1
+    (index - 1)/8 + 1
   end
 
-
   # Obtain chess board column number (1 - 8) from an index (1 - 64)
   def get_col_from_index(index)
-    if index % 8 == 0
-      return 8
-    else
-      return index % 8
-    end
+    index % 8 == 0 ? 8 : index % 8
   end
 
 end

data/lib/network_chess_client.rb

@@ -0,0 +1,104 @@
+
+$LOAD_PATH << __FILE__ # '.'
+
+require 'eventmachine'
+require 'faye/websocket'
+
+require_relative "terminal_chess/version"
+require_relative "terminal_chess/messages"
+require_relative "printer.rb"
+require_relative "move.rb"
+require_relative "board.rb"
+
+
+class NetworkChessClient
+  def initialize(ngrok)
+    @board = Board.new
+    @turn_by_turn_playback = []
+    @game_started = false
+    @player_num   = nil
+    @player_turn  = false
+    @messages     = []
+    @socket_url   = "ws://#{ngrok}.ngrok.io"
+    start_client
+  end
+
+  private
+
+  def start_client
+    Thread.new {
+      EM.run do
+        ws = Faye::WebSocket::Client.new(@socket_url)
+
+        ws.on :open do
+          p [:open]
+        end
+
+        ws.on :message do |msg|
+          p [:message, msg.data]
+
+          if msg.data.match "SETUP: You are player [1|2]"
+            @player_num   = msg.data.split(' ').last.strip.to_i
+            @player_turn  = true if @player_num == 1
+            @game_started = true
+
+            p [:local, "Awaiting opponent move"] if @player_num == 2
+          end
+
+          if msg.data.match "MOVE: [a-zA-Z][0-9], [a-zA-Z][0-9]"
+            from, to = msg.data.match("MOVE: \([a-zA-Z][0-9]\), \([a-zA-Z][0-9]\)").captures
+
+            @board.move(from, to)
+            @player_turn = true
+          end
+        end
+
+        ws.on :close do |e|
+          p [:closed, e.code, e.reason]
+          ws = nil
+          EventMachine::stop_event_loop
+        end
+
+        EventMachine::PeriodicTimer.new(1) do
+          next unless @player_turn && @game_started
+          
+          piece_moved = local_move
+          if piece_moved
+            @player_turn = false
+            ws.send "MOVE: #{@turn_by_turn_playback.last[0]}, #{@turn_by_turn_playback.last[1]}"
+            puts "Awaiting remote player move"
+          end
+        end
+      end
+    }.join 
+  end
+  
+  
+  def local_move
+    if @board.checkmate
+      puts "\nTurn by Turn Playback : #{@turn_by_turn_playback}\n"
+      exit
+    end
+
+    print "\nPiece to Move [#{@board.player_turn.capitalize}]: "
+    from = gets.chomp.upcase
+
+    begin
+      print "Valid destinations: #{@board.valid_destinations(from).join(", ")}"
+
+      print "\nLocation: "
+      to = gets.chomp.upcase
+      moved = @board.move(from, to)
+      
+      return local_move() unless moved == Messages.piece_moved
+
+    rescue Exception => e
+      puts "Invalid selection #{e if !ENV["DEV"].nil?}"
+    else
+      @turn_by_turn_playback << [from, to]
+      true
+    end
+  end
+
+end
+