OSDN Git Service

Added test cases.
[shogi-server/shogi-server.git] / shogi_server / command.rb
index 6334c28..f604ea9 100644 (file)
@@ -1,3 +1,22 @@
+## $Id$
+
+## Copyright (C) 2004 NABEYA Kenichi (aka nanami@2ch)
+## Copyright (C) 2007-2008 Daigo Moriwaki (daigo at debian dot org)
+##
+## This program is free software; you can redistribute it and/or modify
+## it under the terms of the GNU General Public License as published by
+## the Free Software Foundation; either version 2 of the License, or
+## (at your option) any later version.
+##
+## This program is distributed in the hope that it will be useful,
+## but WITHOUT ANY WARRANTY; without even the implied warranty of
+## MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+## GNU General Public License for more details.
+##
+## You should have received a copy of the GNU General Public License
+## along with this program; if not, write to the Free Software
+## Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
+
 require 'kconv'
 require 'shogi_server'
 
@@ -6,7 +25,7 @@ module ShogiServer
   class Command
     # Factory method
     #
-    def Command.factory(str, player)
+    def Command.factory(str, player, time=Time.now)
       cmd = nil
       case str 
       when "" 
@@ -30,6 +49,12 @@ module ShogiServer
       when /^%%MONITOROFF\s+(\S+)/
         game_id = $1
         cmd = MonitorOffCommand.new(str, player, $league.games[game_id])
+      when /^%%MONITOR2ON\s+(\S+)/
+        game_id = $1
+        cmd = Monitor2OnCommand.new(str, player, $league.games[game_id])
+      when /^%%MONITOR2OFF\s+(\S+)/
+        game_id = $1
+        cmd = Monitor2OffCommand.new(str, player, $league.games[game_id])
       when /^%%HELP/
         cmd = HelpCommand.new(str, player)
       when /^%%RATING/
@@ -55,18 +80,51 @@ module ShogiServer
         cmd = LogoutCommand.new(str, player)
       when /^CHALLENGE/
         cmd = ChallengeCommand.new(str, player)
+      when /^%%SETBUOY\s+(\S+)\s+(\S+)(.*)/
+        game_name = $1
+        moves     = $2
+        count = 1 # default
+        if $3 && /^\s+(\d*)/ =~ $3
+          count = $1.to_i
+        end
+        cmd = SetBuoyCommand.new(str, player, game_name, moves, count)
+      when /^%%DELETEBUOY\s+(\S+)/
+        game_name = $1
+        cmd = DeleteBuoyCommand.new(str, player, game_name)
+      when /^%%GETBUOYCOUNT\s+(\S+)/
+        game_name = $1
+        cmd = GetBuoyCountCommand.new(str, player, game_name)
       when /^\s*$/
         cmd = SpaceCommand.new(str, player)
+      when /^%%%[^%]/
+        # TODO: just ignore commands specific to 81Dojo.
+        # Need to discuss with 81Dojo people.
+        cmd = VoidCommand.new(str, player)
       else
         cmd = ErrorCommand.new(str, player)
       end
 
+      cmd.time = time
       return cmd
     end
 
     def initialize(str, player)
       @str    = str
       @player = player
+      @time   = Time.now # this should be replaced later with a real time
+    end
+    attr_accessor :time
+  end
+
+  # Dummy command which does nothing.
+  #
+  class VoidCommand < Command
+    def initialize(str, player)
+      super
+    end
+
+    def call
+      return :continue
     end
   end
 
@@ -97,12 +155,17 @@ module ShogiServer
       if (@player.status == "game")
         array_str = @str.split(",")
         move = array_str.shift
+        if @player.game.last_move &&
+           @player.game.last_move.split(",").first == move
+          log_warning("Received two sequencial identical moves [#{move}] from #{@player.name}. The last one was ignored.")
+          return :continue
+        end
         additional = array_str.shift
         comment = nil
         if /^'(.*)/ =~ additional
           comment = array_str.unshift("'*#{$1.toeuc}")
         end
-        s = @player.game.handle_one_move(move, @player)
+        s = @player.game.handle_one_move(move, @player, @time)
         @player.game.log_game(Kconv.toeuc(comment.first)) if (comment && comment.first && !s)
         return :return if (s && @player.protocol == LoginCSA::PROTOCOL)
       end
@@ -132,7 +195,7 @@ module ShogiServer
     def in_game_status
       rc = :continue
 
-      s = @player.game.handle_one_move(@str, @player)
+      s = @player.game.handle_one_move(@str, @player, @time)
       rc = :return if (s && @player.protocol == LoginCSA::PROTOCOL)
 
       return rc
@@ -142,7 +205,7 @@ module ShogiServer
       rc = :continue
 
       if @player.game.prepared_expire?
-        log_warning("#{@player.status} lasted too long. This play has been expired.")
+        log_warning("#{@player.status} lasted too long. This play has been expired: %s" % [@player.game.game_id])
         @player.game.reject("the Server (timed out)")
         rc = :return if (@player.protocol == LoginCSA::PROTOCOL)
       end
@@ -230,6 +293,53 @@ module ShogiServer
     end
   end
 
+  class MonitorHandler
+    def initialize(player)
+      @player = player
+      @type = nil
+      @header = nil
+    end
+    attr_reader :player, :type, :header
+
+    def ==(rhs)
+      return rhs != nil &&
+             rhs.is_a?(MonitorHandler) &&
+             @player == rhs.player &&
+             @type   == rhs.type
+    end
+
+    def write_safe(game_id, str)
+      str.chomp.split("\n").each do |line|
+        @player.write_safe("##[%s][%s] %s\n" % [@header, game_id, line.chomp])
+      end
+      @player.write_safe("##[%s][%s] %s\n" % [@header, game_id, "+OK"])
+    end
+  end
+
+  class MonitorHandler1 < MonitorHandler
+    def initialize(player)
+      super
+      @type = 1
+      @header = "MONITOR"
+    end
+
+    def write_one_move(game_id, game)
+      write_safe(game_id, game.show.chomp)
+    end
+  end
+
+  class MonitorHandler2 < MonitorHandler
+    def initialize(player)
+      super
+      @type = 2
+      @header = "MONITOR2"
+    end
+
+    def write_one_move(game_id, game)
+      write_safe(game_id, game.last_move.gsub(",", "\n"))
+    end
+  end
+
   # Command of MONITORON
   #
   class MonitorOnCommand < BaseCommandForGame
@@ -239,9 +349,9 @@ module ShogiServer
 
     def call
       if (@game)
-        @game.monitoron(@player)
-        @player.write_safe(@game.show.gsub(/^/, "##[MONITOR][#{@game_id}] "))
-        @player.write_safe("##[MONITOR][#{@game_id}] +OK\n")
+        monitor_handler = MonitorHandler1.new(@player)
+        @game.monitoron(monitor_handler)
+        monitor_handler.write_safe(@game_id, @game.show)
       end
       return :continue
     end
@@ -256,7 +366,38 @@ module ShogiServer
 
     def call
       if (@game)
-        @game.monitoroff(@player)
+        @game.monitoroff(MonitorHandler1.new(@player))
+      end
+      return :continue
+    end
+  end
+
+  # Command of MONITOR2ON
+  #
+  class Monitor2OnCommand < BaseCommandForGame
+    def initialize(str, player, game)
+      super
+    end
+
+    def call
+      if (@game)
+        monitor_handler = MonitorHandler2.new(@player)
+        @game.monitoron(monitor_handler)
+        lines = IO::readlines(@game.logfile).join("")
+        monitor_handler.write_safe(@game_id, lines)
+      end
+      return :continue
+    end
+  end
+
+  class Monitor2OffCommand < MonitorOffCommand
+    def initialize(str, player, game)
+      super
+    end
+
+    def call
+      if (@game)
+        @game.monitoroff(MonitorHandler2.new(@player))
       end
       return :continue
     end
@@ -349,6 +490,14 @@ module ShogiServer
       end
 
       rival = nil
+      if (Buoy.game_name?(@game_name))
+        if (@my_sente_str != "*")
+          @player.write_safe(sprintf("##[ERROR] You are not allowed to specify TEBAN %s for the game %s\n", @my_sente_str, @game_name))
+          return :continue
+        end
+        @player.sente = nil
+      end # really end
+
       if (League::Floodgate.game_name?(@game_name))
         if (@my_sente_str != "*")
           @player.write_safe(sprintf("##[ERROR] You are not allowed to specify TEBAN %s for the game %s\n", @my_sente_str, @game_name))
@@ -356,14 +505,13 @@ module ShogiServer
         end
         @player.sente = nil
       else
-        if (@my_sente_str == "*")
+        if (@my_sente_str == "*") && !Login.handicapped_game_name?(@game_name)
           rival = $league.get_player("game_waiting", @game_name, nil, @player) # no preference
         elsif (@my_sente_str == "+")
           rival = $league.get_player("game_waiting", @game_name, false, @player) # rival must be gote
         elsif (@my_sente_str == "-")
           rival = $league.get_player("game_waiting", @game_name, true, @player) # rival must be sente
         else
-          ## never reached
           @player.write_safe(sprintf("##[ERROR] bad game option\n"))
           return :continue
         end
@@ -371,6 +519,7 @@ module ShogiServer
 
       if (rival)
         @player.game_name = @game_name
+        
         if ((@my_sente_str == "*") && (rival.sente == nil))
           if (rand(2) == 0)
             @player.sente = true
@@ -392,7 +541,36 @@ module ShogiServer
         else
           ## never reached
         end
-        Game::new(@player.game_name, @player, rival)
+        if (Buoy.game_name?(@game_name))
+          buoy = Buoy.new # TODO config
+          if buoy.is_new_game?(@game_name)
+            # The buoy game is not ready yet.
+            # When the game is set, it will be started.
+            @player.status = "game_waiting"
+          else
+            buoy_game = buoy.get_game(@game_name)
+            if buoy_game.instance_of? NilBuoyGame
+              # error. never reach
+            end
+
+            moves_array = Board::split_moves(buoy_game.moves)
+            board = Board.new
+            begin
+              board.set_from_moves(moves_array)
+            rescue => err
+              # it will never happen since moves have already been checked
+              log_error "Failed to set up a buoy game: #{moves}"
+              return :continue
+            end
+            buoy.decrement_count(buoy_game)
+            Game::new(@player.game_name, @player, rival, board)
+          end
+        else
+          klass = Login.handicapped_game_name?(@game_name) || Board
+          board = klass.new
+          board.initial
+          Game::new(@player.game_name, @player, rival, board)
+        end
       else # rival not found
         if (@command_name == "GAME")
           @player.status = "game_waiting"
@@ -531,12 +709,120 @@ module ShogiServer
     end
 
     def call
-      msg = "##[ERROR] unknown command %s\n" % [@str]
+      msg = "##[ERROR] unknown command %s\n" % [@str.chomp]
       @player.write_safe(msg)
       log_error(msg)
       return :continue
     end
   end
 
+  #
+  #
+  class SetBuoyCommand < Command
+
+    def initialize(str, player, game_name, moves, count)
+      super(str, player)
+      @game_name = game_name
+      @moves     = moves
+      @count     = count
+    end
+
+    def call
+      unless (Buoy.game_name?(@game_name))
+        @player.write_safe(sprintf("##[ERROR] wrong buoy game name: %s\n", @game_name))
+        log_error "Received a wrong buoy game name: %s from %s." % [@game_name, @player.name]
+        return :continue
+      end
+      buoy = Buoy.new
+      unless buoy.is_new_game?(@game_name)
+        @player.write_safe(sprintf("##[ERROR] duplicated buoy game name: %s\n", @game_name))
+        log_error "Received duplicated buoy game name: %s from %s." % [@game_name, @player.name]
+        return :continue
+      end
+      if @count < 1
+        @player.write_safe(sprintf("##[ERROR] invalid count: %s\n", @count))
+        log_error "Received an invalid count for a buoy game: %s, %s from %s." % [@count, @game_name, @player.name]
+        return :continue
+      end
+
+      # check moves
+      moves_array = Board::split_moves(@moves)
+      board = Board.new
+      begin
+        board.set_from_moves(moves_array)
+      rescue
+        raise WrongMoves
+      end
+
+      buoy_game = BuoyGame.new(@game_name, @moves, @player.name, @count)
+      buoy.add_game(buoy_game)
+      @player.write_safe(sprintf("##[SETBUOY] +OK\n"))
+      log_info("A buoy game was created: %s by %s" % [@game_name, @player.name])
+
+      # if two players, who are not @player, are waiting for a new game, start it
+      p1 = $league.get_player("game_waiting", @game_name, true, @player)
+      return :continue unless p1
+      p2 = $league.get_player("game_waiting", @game_name, false, @player)
+      return :continue unless p2
+
+      buoy.decrement_count(buoy_game)
+      game = Game::new(@game_name, p1, p2, board)
+      return :continue
+    rescue WrongMoves => e
+      @player.write_safe(sprintf("##[ERROR] wrong moves: %s\n", @moves))
+      log_error "Received wrong moves: %s from %s. [%s]" % [@moves, @player.name, e.message]
+      return :continue
+    end
+  end
+
+  #
+  #
+  class DeleteBuoyCommand < Command
+    def initialize(str, player, game_name)
+      super(str, player)
+      @game_name = game_name
+    end
+
+    def call
+      buoy = Buoy.new
+      buoy_game = buoy.get_game(@game_name)
+      if buoy_game.instance_of?(NilBuoyGame)
+        @player.write_safe(sprintf("##[ERROR] buoy game not found: %s\n", @game_name))
+        log_error "Game name not found: %s by %s" % [@game_name, @player.name]
+        return :continue
+      end
+
+      if buoy_game.owner != @player.name
+        @player.write_safe(sprintf("##[ERROR] you are not allowed to delete a buoy game that you did not set: %s\n", @game_name))
+        log_error "%s are not allowed to delete a game: %s" % [@player.name, @game_name]
+        return :continue
+      end
+
+      buoy.delete_game(buoy_game)
+      @player.write_safe(sprintf("##[DELETEBUOY] +OK\n"))
+      log_info("A buoy game was deleted: %s" % [@game_name])
+      return :continue
+    end
+  end
+
+  #
+  #
+  class GetBuoyCountCommand < Command
+    def initialize(str, player, game_name)
+      super(str, player)
+      @game_name = game_name
+    end
+
+    def call
+      buoy = Buoy.new
+      buoy_game = buoy.get_game(@game_name)
+      if buoy_game.instance_of?(NilBuoyGame)
+        @player.write_safe("##[GETBUOYCOUNT] -1\n")
+      else
+        @player.write_safe("##[GETBUOYCOUNT] %s\n" % [buoy_game.count])
+      end
+      @player.write_safe("##[GETBUOYCOUNT] +OK\n")
+    end
+  end
 
 end # module ShogiServer