forked from threedaymonk/battleship
-
Notifications
You must be signed in to change notification settings - Fork 0
/
play.rb
90 lines (69 loc) · 1.75 KB
/
play.rb
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
$:.unshift File.expand_path("../../lib", __FILE__)
require "battleship/game"
require "battleship/console_renderer"
require "battleship/util"
require "stringio"
require "digest/sha1"
require "forwardable"
require "drb"
DELAY = 0.2
PORT = 4432
class PlayerClient
extend Forwardable
def initialize(secret, object)
@secret = secret
@object = object
end
def method_missing(m, *args)
args.unshift(@secret)
@object.__send__(m, *args)
end
def kill
@object.stop(@secret)
end
end
begin
DRb.start_service
player_server = File.expand_path("../player_server.rb", __FILE__)
players = []
2.times.each do |i|
path = ARGV[i]
port = PORT + i
secret = Digest::SHA1.hexdigest("#{Time.now}#{rand}#{i}")
system %{ruby #{player_server} "#{path}" #{port} #{secret} &}
Battleship::Util.wait_for_socket("localhost", port)
players << PlayerClient.new(secret, DRbObject.new(nil, "druby://localhost:#{port}"))
end
winners = []
3.times do |i|
stderr = ""
$stderr = StringIO.new(stderr)
game = Battleship::Game.new(10, [2, 3, 3, 4, 5], *players)
renderer = Battleship::DeluxeConsoleRenderer.new
$stdout << renderer.render(game)
$stdout << stderr
until game.winner
t0 = Time.now
game.tick
time_taken = Time.now - t0
$stdout << renderer.render(game)
$stdout << stderr
sleep [DELAY - time_taken, 0].max
end
puts "", "#{game.winner.name} won round #{i+1}!"
winners << game.winner.name
sleep 3
break if i == 1 && winners[0] == winners[1]
players.reverse!
end
puts
winners.each_with_index do |name, i|
puts "Round #{i+1}. #{name}"
end
players.each &:kill
rescue Exception => e
$stderr = STDERR
raise e
ensure
players.each &:kill
end