stack_wars 0.1.0

data/lib/stack_wars.rb

@@ -0,0 +1,10 @@
+require "json"
+
+require_relative "stack_wars/version"
+require_relative "stack_wars/errors"
+require_relative "stack_wars/text_display"
+require_relative "stack_wars/battlefield"
+require_relative "stack_wars/player"
+require_relative "stack_wars/game"
+require_relative "stack_wars/territory"
+require_relative "stack_wars/text_client"

data/lib/stack_wars/battlefield.rb

@@ -0,0 +1,65 @@
+module StackWars
+  class Battlefield
+    SIZE           = 9
+    WHITE_BASELINE = 0
+    BLACK_BASELINE = SIZE - 1
+
+    def self.from_json(json_file)
+      territory_data = JSON.parse(File.read(json_file))
+
+      territories = territory_data.map.with_index do |row, y|
+        row.map.with_index do |t, x|
+          occupant, strength = t
+          if occupant == "unclaimed"
+            Territory.new([x,y])
+          else
+            Territory.new([x,y], :occupant      => occupant, 
+                                 :army_strength => strength)
+          end
+        end
+      end
+      
+      new(territories)
+    end
+
+    def initialize(territories=nil)
+      if territories
+        @territories = territories
+      else
+        @territories = 
+          SIZE.times.map { |y| SIZE.times.map { |x| Territory.new([x,y]) } }   
+      end
+    end
+
+    def[](x,y)
+      unless [x,y].all? { |c| (WHITE_BASELINE..BLACK_BASELINE).include?(c) }
+        raise Errors::OutOfBounds 
+      end
+
+      @territories[y][x]
+    end
+
+    def adjacent?(pos1, pos2)
+      (pos1.row == pos2.row && (pos1.column - pos2.column).abs == 1) ||
+      (pos2.column == pos2.column && (pos1.row - pos2.row).abs == 1)
+    end
+
+    def deployed_armies(player)
+      # FIXME: Inefficient, can probably store integer values on write
+      @territories.flatten.reduce(0) do |s, t| 
+         s + (t.occupant == player.color ? t.army_strength : 0)
+      end
+    end
+
+    # loses instance variables, but better than hitting to_s() by default
+    alias_method :inspect, :to_s
+
+    def to_s
+      TextDisplay.new(to_a).to_s
+    end
+
+    def to_a
+      Marshal.load(Marshal.dump(@territories))
+    end
+  end
+end

data/lib/stack_wars/errors.rb

@@ -0,0 +1,7 @@
+module StackWars
+  module Errors
+    IllegalMove = Class.new(StandardError)
+    OutOfBounds = Class.new(StandardError)
+    ParseError  = Class.new(StandardError)
+  end
+end

data/lib/stack_wars/game.rb

@@ -0,0 +1,85 @@
+module StackWars
+  class Game
+    def initialize(battlefield)
+      @players         = [Player.new("black"), Player.new("white")].cycle
+      @battlefield     = battlefield
+      start_new_turn 
+    end
+
+    attr_reader :active_player
+
+    def opponent
+      @players.peek
+    end
+
+    def play(pos1, pos2=nil)
+      if pos2.nil?
+        territory = territory_at(pos1)     
+        
+        territory.fortify(active_player)
+      else
+        from = territory_at(pos1)
+        to   = territory_at(pos2)
+
+        raise Errors::IllegalMove unless battlefield.adjacent?(from, to)
+        raise Errors::IllegalMove unless from.occupied_by?(active_player)
+
+        if to.occupied_by?(opponent)
+          attack(from, to)
+        else
+          move_army(from, to) 
+        end
+      end
+      
+      game_over || start_new_turn
+    end
+
+    def game_over
+      ap_deployed_armies  = battlefield.deployed_armies(active_player)
+      opp_deployed_armies = battlefield.deployed_armies(opponent)
+
+      if active_player.reserves + ap_deployed_armies == 0 ||
+         opponent.reserves      + opp_deployed_armies == 0
+        case
+        when active_player.successful_invasions > opponent.successful_invasions     
+          throw :game_over, "#{active_player.color} won!"
+        when active_player.successful_invasions < opponent.successful_invasions
+          throw :game_over, "#{opponent.color} won!"
+        else
+          throw :game_over, "its a draw"
+        end
+      end
+    end
+
+    def start_new_turn
+      @active_player = @players.next
+    end
+
+    private
+
+    attr_reader :battlefield
+
+    def move_army(from, to)
+      from.remove_one_army
+
+      if to.baseline_for?(opponent)
+        active_player.invade_enemy_territory
+      else
+        to.add_one_army(active_player)
+      end
+    end
+
+    def attack(from, to)
+      raise Errors::IllegalMove if from.army_strength != 1
+
+      to.remove_one_army
+      move_army(from, to) unless to.occupied?
+    end
+
+    def territory_at(position)
+      @battlefield[*position]
+    rescue Errors::OutOfBounds
+      raise Errors::IllegalMove
+    end
+  end
+end

data/lib/stack_wars/player.rb

@@ -0,0 +1,20 @@
+module StackWars
+  class Player
+    def initialize(color)
+      @color                = color
+      @reserves             = 3*Battlefield::SIZE
+      @successful_invasions = 0
+    end
+
+    attr_reader :color, :reserves, :successful_invasions
+
+    def deploy_army
+      raise Errors::IllegalMove unless @reserves > 0
+      @reserves -= 1
+    end
+
+    def invade_enemy_territory
+      @successful_invasions += 1
+    end
+  end
+end

data/lib/stack_wars/territory.rb

@@ -0,0 +1,75 @@
+module StackWars
+  class Territory
+    def initialize(position, params={})
+      @position = position
+      @occupant = params.fetch(:occupant, nil)
+
+      if occupied?
+        @army_strength = params.fetch(:army_strength)
+      else
+        @army_strength = 0
+      end
+    end
+
+    attr_reader :occupant, :army_strength
+
+    def controlled_by?(player)
+      baseline_for?(player) || occupied_by?(player) 
+    end
+
+    def add_one_army(player)
+      @army_strength += 1
+
+      @occupant ||= player.color
+    end
+
+    def remove_one_army
+      raise Errors::IllegalMove unless @army_strength > 0
+
+      @army_strength -= 1
+      @occupant = nil if @army_strength == 0
+    end
+
+    def fortify(player)
+      if controlled_by?(player)
+        player.deploy_army
+        add_one_army(player)
+
+        @occupant ||= player.color
+      else
+        raise Errors::IllegalMove
+      end
+    end
+
+    def occupied?
+      !!@occupant
+    end
+
+    def unoccupied?
+      !occupied?
+    end
+
+    def occupied_by?(player)
+      @occupant == player.color
+    end
+
+    def baseline_for?(player)
+      case player.color
+      when "white"
+        row == Battlefield::WHITE_BASELINE
+      when "black"
+        row == Battlefield::BLACK_BASELINE
+      else
+        raise ArgumentError, "#{color} is not a valid color"
+      end
+    end
+
+    def row
+      @position[1]
+    end
+
+    def column
+      @position[0]
+    end
+  end
+end

data/lib/stack_wars/text_client.rb

@@ -0,0 +1,25 @@
+module StackWars
+  class TextClient
+    def initialize(game)
+      @game = game
+    end
+
+    def play(string)
+      md = string.match(/^ *(?<pos1>\d+ +\d+) *(?<pos2>\d+ +\d+)? *$/)
+
+      raise Errors::ParseError unless md
+
+      if md[:pos2]
+        @game.play(point(md[:pos1]), point(md[:pos2]))
+      else
+        @game.play(point(md[:pos1]))
+      end
+    end
+
+    private
+
+    def point(string)
+      string.scan(/\d+/).map(&:to_i)
+    end
+  end
+end

data/lib/stack_wars/text_display.rb

@@ -0,0 +1,34 @@
+module StackWars
+  class TextDisplay
+    COLOR_SYM = { "black" => "B", "white" => "W" }
+    HEADER    = "#{' '*7}#{(0..8).to_a.join(' '*6)}"
+    SEPARATOR = "#{' '*6} #{9.times.map { '|' }.join(' '*6)}"
+
+    BLACK_BORDER  = "#{' '*5}#{COLOR_SYM['black']*61}"
+    WHITE_BORDER  = "#{' '*5}#{COLOR_SYM['white']*61}"
+
+    def initialize(battlefield)
+      @battlefield = battlefield
+    end
+
+    # loses instance variables, but better than hitting to_s() by default
+    alias_method :inspect, :to_s
+
+    def to_s
+      battlefield_text = @battlefield.map.with_index do |row, row_index|
+        r_text = row.map { |territory| format_territory(territory) }.join("--")
+        "#{row_index.to_s.rjust(3)}  #{r_text}\n"
+      end.join("#{SEPARATOR}\n")
+
+      [HEADER, WHITE_BORDER, battlefield_text.chomp, BLACK_BORDER].join("\n")
+    end
+
+    def format_territory(territory)
+      if territory.occupied?
+        "(#{COLOR_SYM[territory.occupant]}#{territory.army_strength.to_s.rjust(2)})"
+      else
+        "(___)"
+      end
+    end
+  end
+end

data/lib/stack_wars/version.rb

@@ -0,0 +1,3 @@
+module StackWars
+  VERSION = "0.1.0"
+end

data/test/fixtures/active_battlefield.json

@@ -0,0 +1,9 @@
+[[["unclaimed", 0], ["unclaimed", 0], ["unclaimed", 0], ["unclaimed", 0], ["unclaimed", 0], ["unclaimed", 0], ["unclaimed", 0], ["unclaimed", 0], ["unclaimed", 0]],
+ [["unclaimed", 0], ["unclaimed", 0], ["unclaimed", 0], ["unclaimed", 0], ["unclaimed", 0], ["unclaimed", 0], ["unclaimed", 0], ["unclaimed", 0], ["unclaimed", 0]],
+ [["unclaimed", 0], ["unclaimed", 0], ["unclaimed", 0], ["unclaimed", 0], ["unclaimed", 0], ["unclaimed", 0], ["unclaimed", 0], ["unclaimed", 0], ["unclaimed", 0]],
+ [["unclaimed", 0], ["unclaimed", 0], ["unclaimed", 0], ["unclaimed", 0], ["unclaimed", 0], ["unclaimed", 0], ["unclaimed", 0], ["unclaimed", 0], ["unclaimed", 0]],
+ [["unclaimed", 0], ["unclaimed", 0], ["unclaimed", 0], ["unclaimed", 0], ["unclaimed", 0], ["unclaimed", 0], ["unclaimed", 0], ["unclaimed", 0], ["unclaimed", 0]],
+ [["unclaimed", 0], ["unclaimed", 0], ["unclaimed", 0], ["unclaimed", 0], ["unclaimed", 0], ["unclaimed", 0], ["unclaimed", 0], ["unclaimed", 0], ["unclaimed", 0]],
+ [["black", 2], ["unclaimed", 0], ["unclaimed", 0], ["unclaimed", 0], ["unclaimed", 0], ["unclaimed", 0], ["unclaimed", 0], ["unclaimed", 0], ["unclaimed", 0]],
+ [["unclaimed", 0], ["white", 2], ["unclaimed", 0], ["unclaimed", 0], ["unclaimed", 0], ["unclaimed", 0], ["unclaimed", 0], ["unclaimed", 0], ["unclaimed", 0]],
+ [["unclaimed", 0], ["unclaimed", 0], ["unclaimed", 0], ["unclaimed", 0], ["unclaimed", 0], ["unclaimed", 0], ["unclaimed", 0], ["unclaimed", 0], ["unclaimed", 0]]]

data/test/fixtures/empty_battlefield.json

@@ -0,0 +1,9 @@
+[[["unclaimed", 0], ["unclaimed", 0], ["unclaimed", 0], ["unclaimed", 0], ["unclaimed", 0], ["unclaimed", 0], ["unclaimed", 0], ["unclaimed", 0], ["unclaimed", 0]],
+ [["unclaimed", 0], ["unclaimed", 0], ["unclaimed", 0], ["unclaimed", 0], ["unclaimed", 0], ["unclaimed", 0], ["unclaimed", 0], ["unclaimed", 0], ["unclaimed", 0]],
+ [["unclaimed", 0], ["unclaimed", 0], ["unclaimed", 0], ["unclaimed", 0], ["unclaimed", 0], ["unclaimed", 0], ["unclaimed", 0], ["unclaimed", 0], ["unclaimed", 0]],
+ [["unclaimed", 0], ["unclaimed", 0], ["unclaimed", 0], ["unclaimed", 0], ["unclaimed", 0], ["unclaimed", 0], ["unclaimed", 0], ["unclaimed", 0], ["unclaimed", 0]],
+ [["unclaimed", 0], ["unclaimed", 0], ["unclaimed", 0], ["unclaimed", 0], ["unclaimed", 0], ["unclaimed", 0], ["unclaimed", 0], ["unclaimed", 0], ["unclaimed", 0]],
+ [["unclaimed", 0], ["unclaimed", 0], ["unclaimed", 0], ["unclaimed", 0], ["unclaimed", 0], ["unclaimed", 0], ["unclaimed", 0], ["unclaimed", 0], ["unclaimed", 0]],
+ [["unclaimed", 0], ["unclaimed", 0], ["unclaimed", 0], ["unclaimed", 0], ["unclaimed", 0], ["unclaimed", 0], ["unclaimed", 0], ["unclaimed", 0], ["unclaimed", 0]],
+ [["unclaimed", 0], ["unclaimed", 0], ["unclaimed", 0], ["unclaimed", 0], ["unclaimed", 0], ["unclaimed", 0], ["unclaimed", 0], ["unclaimed", 0], ["unclaimed", 0]],
+ [["unclaimed", 0], ["unclaimed", 0], ["unclaimed", 0], ["unclaimed", 0], ["unclaimed", 0], ["unclaimed", 0], ["unclaimed", 0], ["unclaimed", 0], ["unclaimed", 0]]]

data/test/fixtures/moves.json

@@ -0,0 +1,13 @@
+[[[3,8]],[[6,0]],[[6,8]],[[6,0],[6,1]],[[6,8],[6,7]],[[6,1],[6,2]],[[6,7],[6,6]],[[6,2],[6,3]],[[6,6],[6,5]],[[6,3]],[[6,5]],[[6,3],[6,4]],[[6,5],[5,5]],[[6,4],[7,4]],[[6,5],[6,6]],[[7,4],[7,5]
+],[[5,5]],[[7,5]],[[5,5],[5,6]],[[7,5],[7,6]],[[7,8]],[[6,3],[7,3]],[[8,8]],[[7,3],[7,4]],[[8,8]],[[7,4],[8,4]],[[8,8],[8,7]],[[8,4],[8,5]],[[5,5],[5,4]],[[8,5]],[[7,8],[7,7]],[[8,5],[8,6]],[[7
+,7],[7,6]],[[8,6],[7,6]],[[5,4],[5,3]],[[8,5]],[[5,3],[5,2]],[[8,5],[8,6]],[[5,2]],[[7,6],[6,6]],[[5,2],[5,1]],[[6,6],[5,6]],[[4,8]],[[7,5]],[[5,8]],[[7,5],[7,6]],[[5,2]],[[7,6]],[[8,7],[7,7]],
+[[7,6],[6,6]],[[7,8]],[[6,6]],[[6,8]],[[5,6]],[[5,8],[5,7]],[[5,6],[4,6]],[[5,8]],[[4,6]],[[5,1],[5,0]],[[4,6],[3,6]],[[5,2],[5,1]],[[3,6]],[[5,2]],[[3,6],[3,7]],[[5,1],[5,0]],[[3,7],[3,8]],[[3
+,8]],[[3,6]],[[5,8],[5,7]],[[3,6],[3,7]],[[5,7],[5,8]],[[3,7],[3,8]],[[3,8]],[[5,0]],[[6,8],[6,7]],[[7,6],[7,7]],[[7,8],[7,7]],[[5,6],[5,7]],[[5,8],[5,7]],[[6,6],[5,6]],[[5,7],[5,6]],[[6,6],[5,
+6]],[[5,2],[4,2]],[[4,6]],[[4,8],[4,7]],[[4,6],[3,6]],[[4,7],[4,6]],[[5,6],[4,6]],[[3,8],[3,7]],[[3,6],[2,6]],[[4,2],[4,1]],[[3,6],[3,7]],[[3,8]],[[2,6],[2,7]],[[3,8],[3,7]],[[2,7],[3,7]],[[4,8
+]],[[4,6]],[[6,7],[5,7]],[[4,6],[5,6]],[[5,8]],[[3,7]],[[3,8]],[[3,7],[3,6]],[[5,2]],[[8,6]],[[5,2],[4,2]],[[8,6],[7,6]],[[7,8]],[[3,6]],[[3,8],[3,7]],[[3,6],[2,6]],[[3,7],[3,6]],[[2,6],[3,6]],
+[[3,8]],[[7,5]],[[4,2]],[[7,5],[6,5]],[[4,1],[4,0]],[[4,0]],[[4,2],[4,1]],[[4,0],[4,1]],[[4,2],[4,1]],[[6,5],[6,6]],[[6,8]],[[6,6]],[[6,8],[6,7]],[[6,6],[6,5]],[[5,2],[5,1]],[[5,0],[5,1]],[[4,1
+],[5,1]],[[3,6]],[[4,8],[4,7]],[[4,6],[4,7]],[[5,7],[4,7]],[[3,6],[2,6]],[[2,8]],[[2,6]],[[2,8],[2,7]],[[2,6],[1,6]],[[2,7],[2,6]],[[1,6],[2,6]],[[3,8],[3,7]],[[3,6],[2,6]],[[3,7],[2,7]],[[2,6]
+,[1,6]],[[2,7],[2,6]],[[1,6],[2,6]],[[4,7],[3,7]],[[2,6]],[[3,7],[2,7]],[[2,6],[1,6]],[[2,7],[2,6]],[[1,6],[2,6]],[[5,8],[5,7]],[[2,6],[2,7]],[[5,7],[4,7]],[[2,7]],[[4,7],[3,7]],[[2,7],[1,7]],[
+[3,7],[2,7]],[[1,7],[2,7]],[[7,8],[6,8]],[[2,7]],[[6,8],[5,8]],[[2,7]],[[5,8],[4,8]],[[2,7],[2,8]],[[4,8],[3,8]],[[2,7],[2,8]],[[5,1],[5,0]],[[7,6],[7,7]],[[6,7],[7,7]],[[2,7],[2,8]],[[3,8],[3,
+7]],[[5,6],[5,7]],[[3,7],[3,6]],[[5,7],[5,8]],[[3,6],[4,6]],[[7,5],[7,6]],[[7,7],[7,8]],[[6,6],[6,7]],[[4,6],[5,6]],[[6,5],[6,6]],[[5,6],[6,6]],[[7,6],[6,6]],[[7,8],[7,7]],[[6,7],[7,7]],[[8,8],
+[8,7]],[[8,6],[8,7]]] 

data/test/integration/full_game_test.rb

@@ -0,0 +1,32 @@
+require_relative "../test_helper"
+
+describe "A full game of stack wars" do
+  let(:moves) do     
+    moves_file = "#{File.dirname(__FILE__)}/../fixtures/moves.json"
+    JSON.load(File.read(moves_file))
+  end
+
+  let(:battlefield) { StackWars::Battlefield.new }
+
+  let(:game) { StackWars::Game.new(battlefield) }
+
+  it "must end as expected" do
+    message = catch(:game_over) do
+      moves.each { |move| game.play(*move) } 
+    end
+
+    message.must_equal "white won!"
+
+    white = game.active_player
+    black = game.opponent
+
+    battlefield.deployed_armies(white).must_equal(4)
+    battlefield.deployed_armies(black).must_equal(0)
+
+    white.reserves.must_equal(0)
+    black.reserves.must_equal(0)
+
+    white.successful_invasions.must_equal(6)
+    black.successful_invasions.must_equal(4)
+  end
+end

data/test/suite.rb

@@ -0,0 +1,2 @@
+require_relative "units/territory_test"
+require_relative "integration/full_game_test"

data/test/test_helper.rb

@@ -0,0 +1,11 @@
+gem "minitest", "~> 2.6.1"
+
+require "minitest/autorun"
+
+require_relative "../lib/stack_wars"
+
+module MiniTest
+  class Spec
+    alias insist lambda
+  end
+end

data/test/units/territory_test.rb

@@ -0,0 +1,107 @@
+require_relative "../test_helper"
+
+describe "StackWars::Territory" do
+  let(:black_player) { StackWars::Player.new("black") }
+  let(:white_player) { StackWars::Player.new("white") }
+
+  let(:white_baseline) { StackWars::Battlefield::WHITE_BASELINE }
+  let(:black_baseline) { StackWars::Battlefield::BLACK_BASELINE }
+
+  let(:columns)        { (0...StackWars::Battlefield::SIZE) }
+  let(:interior_rows)  { (white_baseline+1..black_baseline-1) }
+
+  def new_territory(*args)
+    StackWars::Territory.new(*args)
+  end
+
+  it "must know when it is part of a baseline" do
+    columns.each do |x|
+      black_baseline_territory = new_territory([x, black_baseline])
+      black_baseline_territory.must_be :baseline_for?, black_player
+      black_baseline_territory.wont_be :baseline_for?, white_player
+
+      white_baseline_territory = new_territory([x, white_baseline])
+      white_baseline_territory.must_be :baseline_for?, white_player
+      white_baseline_territory.wont_be :baseline_for?, black_player
+    end
+  end
+
+  it "must know when it not part of a baseline" do
+    columns.each do |x|
+      interior_rows.each do |y|
+        territory = new_territory([x,y])
+        territory.wont_be :baseline_for?, black_player
+        territory.wont_be :baseline_for?, white_player
+      end
+    end
+  end
+
+  it "treats each players baseline as controlled" do
+    columns.each do |x|
+      black_baseline_territory = new_territory([x, black_baseline])
+      black_baseline_territory.must_be :controlled_by?, black_player
+      black_baseline_territory.wont_be :controlled_by?, white_player
+
+      white_baseline_territory = new_territory([x, white_baseline])
+      white_baseline_territory.must_be :controlled_by?, white_player
+      white_baseline_territory.wont_be :controlled_by?, black_player
+    end
+  end
+
+  it "treats any occupied territory as controlled by the occupying player" do
+    black_controlled_territory = 
+      new_territory([3,3], :occupant => black_player.color, :army_strength => 1)
+
+    white_controlled_territory = 
+      new_territory([4,4], :occupant => white_player.color, :army_strength => 1)
+
+    black_controlled_territory.must_be :controlled_by?, black_player
+    black_controlled_territory.wont_be :controlled_by?, white_player
+
+    white_controlled_territory.must_be :controlled_by?, white_player
+    white_controlled_territory.wont_be :controlled_by?, black_player
+  end
+
+  it "allows the player to fortify any position on their own baseline" do
+    columns.each do |x|
+      black_territory = new_territory([x, black_baseline])
+      black_territory.fortify(black_player)
+        
+      black_territory.army_strength.must_equal(1)
+      black_territory.must_be :occupied_by?, black_player
+
+      black_territory.fortify(black_player)
+      black_territory.army_strength.must_equal(2)
+
+      white_territory = new_territory([x, white_baseline])
+      white_territory.fortify(white_player)
+        
+      white_territory.army_strength.must_equal(1)
+      white_territory.must_be :occupied_by?, white_player
+
+      white_territory.fortify(white_player)
+      white_territory.army_strength.must_equal(2)
+    end
+  end
+
+  it "allows the player to fortify their own occupied territories" do
+    [black_player, white_player].each do |player|
+       territory = 
+         new_territory([3,3], :occupant =>player.color, :army_strength => 1)
+       territory.fortify(player)
+       territory.must_be :occupied_by?, player
+       territory.army_strength.must_equal(2)
+    end
+  end
+
+  it "does not allow fortifying unoccupied internal territories" do
+    [black_player, white_player].each do |player|
+      territory = new_territory([2,3])
+      territory.wont_be :baseline_for?, player
+      territory.wont_be :occupied?
+
+      insist { territory.fortify(player) }.
+        must_raise(StackWars::Errors::IllegalMove)
+    end
+  end
+end