uno plugin: clean up game management permissions
[rbot] / data / rbot / plugins / games / uno.rb
1 #-- vim:sw=2:et
2 #++
3 #
4 # :title: Uno Game Plugin for rbot
5 #
6 # Author:: Giuseppe "Oblomov" Bilotta <giuseppe.bilotta@gmail.com>
7 #
8 # Copyright:: (C) 2008 Giuseppe Bilotta
9 #
10 # License:: GPL v2
11 #
12 # Uno Game: get rid of the cards you have
13 #
14 # TODO documentation
15 # TODO allow full form card names for play
16 # TODO allow choice of rules re stacking + and playing Reverse with them
17
18 class UnoGame
19   COLORS = %w{Red Green Blue Yellow}
20   SPECIALS = %w{+2 Reverse Skip}
21   NUMERICS = (0..9).to_a
22   VALUES = NUMERICS + SPECIALS
23
24   def UnoGame.color_map(clr)
25     case clr
26     when 'Red'
27       :red
28     when 'Blue'
29       :royal_blue
30     when 'Green'
31       :limegreen
32     when 'Yellow'
33       :yellow
34     end
35   end
36
37   def UnoGame.irc_color_bg(clr)
38     Irc.color([:white,:black][COLORS.index(clr)%2],UnoGame.color_map(clr))
39   end
40
41   def UnoGame.irc_color_fg(clr)
42     Irc.color(UnoGame.color_map(clr))
43   end
44
45   def UnoGame.colorify(str, fg=false)
46     ret = Bold.dup
47     str.length.times do |i|
48       ret << (fg ?
49               UnoGame.irc_color_fg(COLORS[i%4]) :
50               UnoGame.irc_color_bg(COLORS[i%4]) ) +str[i,1]
51     end
52     ret << NormalText
53   end
54
55   UNO = UnoGame.colorify('UNO!', true)
56
57   # Colored play cards
58   class Card
59     attr_reader :color
60     attr_reader :value
61     attr_reader :shortform
62     attr_reader :to_s
63     attr_reader :score
64
65     def initialize(color, value)
66       raise unless COLORS.include? color
67       @color = color.dup
68       raise unless VALUES.include? value
69       if NUMERICS.include? value
70         @value = value
71         @score = value
72       else
73         @value = value.dup
74         @score = 20
75       end
76       if @value == '+2'
77         @shortform = (@color[0,1]+@value).downcase
78       else
79         @shortform = (@color[0,1]+@value.to_s[0,1]).downcase
80       end
81       @to_s = UnoGame.irc_color_bg(@color) +
82         Bold + ['', @color, @value, ''].join(' ') + NormalText
83     end
84
85     def picker
86       return 0 unless @value.to_s[0,1] == '+'
87       return @value[1,1].to_i
88     end
89
90     def special?
91       SPECIALS.include?(@value)
92     end
93
94     def <=>(other)
95       cc = self.color <=> other.color
96       if cc == 0
97         return self.value.to_s <=> other.value.to_s
98       else
99         return cc
100       end
101     end
102     include Comparable
103   end
104
105   # Wild, Wild +4 cards
106   class Wild < Card
107     def initialize(value=nil)
108       @color = 'Wild'
109       raise if value and not value == '+4'
110       if value
111         @value = value.dup 
112         @shortform = 'w'+value
113       else
114         @value = nil
115         @shortform = 'w'
116       end
117       @score = 50
118       @to_s = UnoGame.colorify(['', @color, @value, ''].compact.join(' '))
119     end
120     def special?
121       @value
122     end
123   end
124
125   class Player
126     attr_accessor :cards
127     attr_accessor :user
128     def initialize(user)
129       @user = user
130       @cards = []
131     end
132     def has_card?(short)
133       has = []
134       @cards.each { |c|
135         has << c if c.shortform == short
136       }
137       if has.empty?
138         return false
139       else
140         return has
141       end
142     end
143     def to_s
144       Bold + @user.to_s + Bold
145     end
146   end
147
148   # cards in stock
149   attr_reader :stock
150   # current discard
151   attr_reader :discard
152   # previous discard, in case of challenge
153   attr_reader :last_discard
154   # channel the game is played in
155   attr_reader :channel
156   # list of players
157   attr :players
158   # true if the player picked a card (and can thus pass turn)
159   attr_reader :player_has_picked
160   # number of cards to be picked if the player can't play an appropriate card
161   attr_reader :picker
162
163   def initialize(plugin, channel)
164     @channel = channel
165     @plugin = plugin
166     @bot = plugin.bot
167     @players = []
168     @dropouts = []
169     @discard = nil
170     @last_discard = nil
171     @value = nil
172     @color = nil
173     make_base_stock
174     @stock = []
175     make_stock
176     @start_time = nil
177     @join_timer = nil
178     @picker = 0
179     @last_picker = 0
180     @must_play = nil
181   end
182
183   def get_player(user)
184     case user
185     when User
186       @players.each do |p|
187         return p if p.user == user
188       end
189     when String
190       @players.each do |p|
191         return p if p.user.irc_downcase == user.irc_downcase(channel.casemap)
192       end
193     else
194       get_player(user.to_s)
195     end
196     return nil
197   end
198
199   def announce(msg, opts={})
200     @bot.say channel, msg, opts
201   end
202
203   def notify(player, msg, opts={})
204     @bot.notice player.user, msg, opts
205   end
206
207   def make_base_stock
208     @base_stock = COLORS.inject([]) do |list, clr|
209       VALUES.each do |n|
210         list << Card.new(clr, n)
211         list << Card.new(clr, n) unless n == 0
212       end
213       list
214     end
215     4.times do
216       @base_stock << Wild.new
217       @base_stock << Wild.new('+4')
218     end
219   end
220
221   def make_stock
222     @stock.replace @base_stock
223     # remove the cards in the players hand
224     @players.each { |p| p.cards.each { |c| @stock.delete_one c } }
225     # remove current top discarded card if present
226     if @discard
227       @stock.delete_one(discard)
228     end
229     @stock.shuffle!
230   end
231
232   def start_game
233     debug "Starting game"
234     @players.shuffle!
235     show_order
236     announce _("%{p} deals the first card from the stock") % {
237       :p => @players.first
238     }
239     card = @stock.shift
240     @picker = 0
241     @special = false
242     while Wild === card do
243       @stock.insert(rand(@stock.length), card)
244       card = @stock.shift
245     end
246     set_discard(card)
247     show_discard
248     if @special
249       do_special
250     end
251     next_turn
252     @start_time = Time.now
253   end
254
255   def elapsed_time
256     if @start_time
257       Utils.secs_to_string(Time.now-@start_time)
258     else
259       _("no time")
260     end
261   end
262
263   def reverse_turn
264     # if there are two players, the Reverse acts like a Skip, unless
265     # there's a @picker running, in which case the Reverse should bounce the
266     # pick on the other player
267     if @players.length > 2
268       @players.reverse!
269       # put the current player back in its place
270       @players.unshift @players.pop
271       announce _("Playing order was reversed!")
272     elsif @picker > 0
273       announce _("%{cp} bounces the pick to %{np}") % {
274         :cp => @players.first,
275         :np => @players.last
276       }
277     else
278       skip_turn
279     end
280   end
281
282   def skip_turn
283     @players << @players.shift
284     announce _("%{p} skips a turn!") % {
285       # this is first and not last because the actual
286       # turn change will be done by the following next_turn
287       :p => @players.first
288     }
289   end
290
291   def do_special
292     case @discard.value
293     when 'Reverse'
294       reverse_turn
295       @special = false
296     when 'Skip'
297       skip_turn
298       @special = false
299     end
300   end
301
302   def set_discard(card)
303     @discard = card
304     @value = card.value.dup rescue card.value
305     if Wild === card
306       @color = nil
307     else
308       @color = card.color.dup
309     end
310     if card.picker > 0
311       @picker += card.picker
312       @last_picker = @discard.picker
313     end
314     if card.special?
315       @special = true
316     else
317       @special = false
318     end
319     @must_play = nil
320   end
321
322   def next_turn(opts={})
323     @players << @players.shift
324     @player_has_picked = false
325     show_turn
326   end
327
328   def can_play(card)
329     # if play is forced, check against the only allowed cards
330     return false if @must_play and not @must_play.include?(card)
331
332     # When a +something is online, you can only play a +something of same or
333     # higher something, or a Reverse of the correct color, or a Reverse on
334     # a Reverse
335     # TODO make optional
336     if @picker > 0
337       return true if card.picker >= @last_picker
338       return true if card.value == 'Reverse' and (card.color == @color or @discard.value == card.value)
339       return false
340     else
341       # You can always play a Wild
342       return true if Wild === card
343       # On a Wild, you must match the color
344       if Wild === @discard
345         return card.color == @color
346       else
347         # Otherwise, you can match either the value or the color
348         return (card.value == @value) || (card.color == @color)
349       end
350     end
351   end
352
353   def play_card(source, cards)
354     debug "Playing card #{cards}"
355     p = get_player(source)
356     shorts = cards.gsub(/\s+/,'').match(/^(?:([rbgy]\+?\d){1,2}|([rbgy][rs])|(w(?:\+4)?)([rbgy])?)$/).to_a
357     debug shorts.inspect
358     if shorts.empty?
359       announce _("what cards were that again?")
360       return
361     end
362     full = shorts[0]
363     short = shorts[1] || shorts[2] || shorts[3]
364     jolly = shorts[3]
365     jcolor = shorts[4]
366     if jolly
367       toplay = 1
368     else
369       toplay = (full == short) ? 1 : 2
370     end
371     debug [full, short, jolly, jcolor, toplay].inspect
372     # r7r7 -> r7r7, r7, nil, nil
373     # r7 -> r7, r7, nil, nil
374     # w -> w, nil, w, nil
375     # wg -> wg, nil, w, g
376     if cards = p.has_card?(short)
377       debug cards
378       unless can_play(cards.first)
379         announce _("you can't play that card")
380         return
381       end
382       if cards.length >= toplay
383         # if the played card is a W+4 not played during a stacking +x
384         # TODO if A plays an illegal W+4, B plays a W+4, should the next
385         # player be able to challenge A? For the time being we say no,
386         # but I think he should, and in case A's move was illegal
387         # game would have to go back, A would get the penalty and replay,
388         # while if it was legal the challenger would get 50% more cards,
389         # i.e. 12 cards (or more if the stacked +4 were more). This would
390         # only be possible if the first W+4 was illegal, so it wouldn't
391         # apply for a W+4 played on a +2 anyway.
392         #
393         if @picker == 0 and Wild === cards.first and cards.first.value 
394           # save the previous discard in case of challenge
395           @last_discard = @discard.dup
396           # save the color too, in case it was a Wild
397           @last_color = @color.dup
398         else
399           # mark the move as not challengeable
400           @last_discard = nil
401           @last_color = nil
402         end
403         set_discard(p.cards.delete_one(cards.shift))
404         if toplay > 1
405           set_discard(p.cards.delete_one(cards.shift))
406           announce _("%{p} plays %{card} twice!") % {
407             :p => p,
408             :card => @discard
409           }
410         else
411           announce _("%{p} plays %{card}") % { :p => p, :card => @discard }
412         end
413         if p.cards.length == 1
414           announce _("%{p} has %{uno}!") % {
415             :p => p, :uno => UNO
416           }
417         elsif p.cards.length == 0
418           end_game
419           return
420         end
421         show_picker
422         if @color
423           if @special
424             do_special
425           end
426           next_turn
427         elsif jcolor
428           choose_color(p.user, jcolor)
429         else
430           announce _("%{p}, choose a color with: co r|b|g|y") % { :p => p }
431         end
432       else
433         announce _("you don't have two cards of that kind")
434       end
435     else
436       announce _("you don't have that card")
437     end
438   end
439
440   def challenge
441     return unless @last_discard
442     # current player
443     cp = @players.first
444     # previous player
445     lp = @players.last
446     announce _("%{cp} challenges %{lp}'s %{card}!") % {
447       :cp => cp, :lp => lp, :card => @discard
448     }
449     # show the cards of the previous player to the current player
450     notify cp, _("%{p} has %{cards}") % {
451       :p => lp, :cards => lp.cards.join(' ')
452     }
453     # check if the previous player had a non-special card of the correct color
454     legal = true
455     lp.cards.each do |c|
456       if c.color == @last_color and not c.special?
457         legal = false
458       end
459     end
460     if legal
461       @picker += 2
462       announce _("%{lp}'s move was legal, %{cp} must pick %{b}%{n}%{b} cards!") % {
463         :cp => cp, :lp => lp, :b => Bold, :n => @picker
464       }
465       @last_color = nil
466       @last_discard = nil
467       deal(cp, @picker)
468       @picker = 0
469       next_turn
470     else
471       announce _("%{lp}'s move was %{b}not%{b} legal, %{lp} must pick %{b}%{n}%{b} cards and play again!") % {
472         :cp => cp, :lp => lp, :b => Bold, :n => @picker
473       }
474       lp.cards << @discard # put the W+4 back in place
475
476       # reset the discard
477       @color = @last_color.dup
478       @discard = @last_discard.dup
479       @special = false
480       @value = @discard.value.dup rescue @discard.value
481       @last_color = nil
482       @last_discard = nil
483
484       # force the player to play the current cards
485       @must_play = lp.cards.dup
486
487       # give him the penalty cards
488       deal(lp, @picker)
489       @picker = 0
490
491       # and restore the turn
492       @players.unshift @players.pop
493     end
494   end
495
496   def pass(user)
497     p = get_player(user)
498     if @picker > 0
499       announce _("%{p} passes turn, and has to pick %{b}%{n}%{b} cards!") % {
500         :p => p, :b => Bold, :n => @picker
501       }
502       deal(p, @picker)
503       @picker = 0
504     else
505       if @player_has_picked
506         announce _("%{p} passes turn") % { :p => p }
507       else
508         announce _("you need to pick a card first")
509         return
510       end
511     end
512     next_turn
513   end
514
515   def choose_color(user, color)
516     case color
517     when 'r'
518       @color = 'Red'
519     when 'b'
520       @color = 'Blue'
521     when 'g'
522       @color = 'Green'
523     when 'y'
524       @color = 'Yellow'
525     else
526       announce _('what color is that?')
527       return
528     end
529     announce _('color is now %{c}') % {
530       :c => UnoGame.irc_color_bg(@color)+" #{@color} "
531     }
532     next_turn
533   end
534
535   def show_time
536     if @start_time
537       announce _("This %{uno} game has been going on for %{time}") % {
538         :uno => UNO,
539         :time => elapsed_time
540       }
541     else
542       announce _("The game hasn't started yet")
543     end
544   end
545
546   def show_order
547     announce _("%{uno} playing turn: %{players}") % {
548       :uno => UNO, :players => players.join(' ')
549     }
550   end
551
552   def show_turn(opts={})
553     cards = true
554     cards = opts[:cards] if opts.key?(:cards)
555     player = @players.first
556     announce _("it's %{player}'s turn") % { :player => player }
557     show_user_cards(player) if cards
558   end
559
560   def has_turn?(source)
561     @players.first.user == source
562   end
563
564   def show_picker
565     if @picker > 0
566       announce _("next player must respond correctly or pick %{b}%{n}%{b} cards") % {
567         :b => Bold, :n => @picker
568       }
569     end
570   end
571
572   def show_discard
573     announce _("Current discard: %{card} %{c}") % { :card => @discard,
574       :c => (Wild === @discard) ? UnoGame.irc_color_bg(@color) + " #{@color} " : nil
575     }
576     show_picker
577   end
578
579   def show_user_cards(player)
580     p = Player === player ? player : get_player(player)
581     notify p, _('Your cards: %{cards}') % {
582       :cards => p.cards.join(' ')
583     }
584   end
585
586   def show_all_cards(u=nil)
587     announce(@players.inject([]) { |list, p|
588       list << [p, p.cards.length].join(': ')
589     }.join(', '))
590     if u
591       show_user_cards(u)
592     end
593   end
594
595   def pick_card(user)
596     p = get_player(user)
597     announce _("%{player} picks a card") % { :player => p }
598     deal(p, 1)
599     @player_has_picked = true
600   end
601
602   def deal(player, num=1)
603     picked = []
604     num.times do
605       picked << @stock.delete_one
606       if @stock.length == 0
607         announce _("Shuffling discarded cards")
608         make_stock
609         if @stock.length == 0
610           announce _("No more cards!")
611           end_game # FIXME nope!
612         end
613       end
614     end
615     picked.sort!
616     notify player, _("You picked %{picked}") % { :picked => picked.join(' ') }
617     player.cards += picked
618     player.cards.sort!
619   end
620
621   def add_player(user)
622     if p = get_player(user)
623       announce _("you're already in the game, %{p}") % {
624         :p => p
625       }
626       return
627     end
628     @dropouts.each do |dp|
629       if dp.user == user
630         announce _("you dropped from the game, %{p}, you can't get back in") % {
631           :p => dp
632         }
633         return
634       end
635     end
636     cards = 7
637     if @start_time
638       cards = @players.inject(0) do |s, pl|
639         s +=pl.cards.length
640       end/@players.length
641     end
642     p = Player.new(user)
643     @players << p
644     announce _("%{p} joins this game of %{uno}") % {
645       :p => p, :uno => UNO
646     }
647     deal(p, cards)
648     return if @start_time
649     if @join_timer
650       @bot.timer.reschedule(@join_timer, 10)
651     elsif @players.length > 1
652       announce _("game will start in 20 seconds")
653       @join_timer = @bot.timer.add_once(20) {
654         start_game
655       }
656     end
657   end
658
659   def drop_player(nick)
660     # A nick is passed because the original player might have left
661     # the channel or IRC
662     unless p = get_player(nick)
663       announce _("%{p} isn't playing %{uno}") % {
664         :p => p, :uno => UNO
665       }
666       return
667     end
668     announce _("%{p} gives up this game of %{uno}") % {
669       :p => p, :uno => UNO
670     }
671     case @players.length
672     when 2
673       if p == @players.first
674         next_turn
675       end
676       end_game
677       return
678     when 1
679       end_game(true)
680       return
681     end
682     debug @stock.length
683     while p.cards.length > 0
684       @stock.insert(rand(@stock.length), p.cards.shift)
685     end
686     debug @stock.length
687     @dropouts << @players.delete_one(p)
688   end
689
690   def replace_player(old, new)
691     # The new user
692     user = channel.get_user(new)
693     if p = get_player(user)
694       announce _("%{p} is already playing %{uno} here") % {
695         :p => p, :uno => UNO
696       }
697       return
698     end
699     # We scan the player list of the player with the old nick, instead
700     # of using get_player, in case of IRC drops etc
701     @players.each do |p|
702       if p.user.nick == old
703         p.user = user
704         announce _("%{p} takes %{b}%{old}%{b}'s place at %{uno}") % {
705           :p => p, :b => Bold, :old => old, :uno => UNO
706         }
707         return
708       end
709     end
710     announce _("%{b}%{old}%{b} isn't playing %{uno} here") % {
711       :uno => UNO, :b => Bold, :old => old
712     }
713   end
714
715   def end_game(halted = false)
716     runtime = @start_time ? Time.now -  @start_time : 0
717     if halted
718       if @start_time
719         announce _("%{uno} game halted after %{time}") % {
720           :time => elapsed_time,
721           :uno => UNO
722         }
723       else
724         announce _("%{uno} game halted before it could start") % {
725           :uno => UNO
726         }
727       end
728     else
729       announce _("%{uno} game finished after %{time}! The winner is %{p}") % {
730         :time => elapsed_time,
731         :uno => UNO, :p => @players.first
732       }
733     end
734     if @picker > 0 and not halted
735       p = @players[1]
736       announce _("%{p} has to pick %{b}%{n}%{b} cards!") % {
737         :p => p, :n => @picker, :b => Bold
738       }
739       deal(p, @picker)
740       @picker = 0
741     end
742     score = @players.inject(0) do |sum, p|
743       if p.cards.length > 0
744         announce _("%{p} still had %{cards}") % {
745           :p => p, :cards => p.cards.join(' ')
746         }
747         sum += p.cards.inject(0) do |cs, c|
748           cs += c.score
749         end
750       end
751       sum
752     end
753
754     closure = { :dropouts => @dropouts, :players => @players, :runtime => runtime }
755     if not halted
756       announce _("%{p} wins with %{b}%{score}%{b} points!") % {
757         :p => @players.first, :score => score, :b => Bold
758       }
759       closure.merge!(:winner => @players.first, :score => score,
760         :opponents => @players.length - 1)
761     end
762
763     @plugin.do_end_game(@channel, closure)
764   end
765
766 end
767
768 # A won game: store score and number of opponents, so we can calculate
769 # an average score per opponent (requested by Squiddhartha)
770 define_structure :UnoGameWon, :score, :opponents
771 # For each player we store the number of games played, the number of
772 # games forfeited, and an UnoGameWon for each won game
773 define_structure :UnoPlayerStats, :played, :forfeits, :won
774
775 class UnoPlugin < Plugin
776   attr :games
777   def initialize
778     super
779     @games = {}
780   end
781
782   def help(plugin, topic="")
783     case topic
784     when 'commands'
785       [
786       _("'jo' to join in"),
787       _("'pl <card>' to play <card>: e.g. 'pl g7' to play Green 7, or 'pl rr' to play Red Reverse, or 'pl y2y2' to play both Yellow 2 cards"),
788       _("'pe' to pick a card"),
789       _("'pa' to pass your turn"),
790       _("'co <color>' to pick a color after playing a Wild: e.g. 'co g' to select Green (or 'pl w+4 g' to select the color when playing the Wild)"),
791       _("'ca' to show current cards"),
792       _("'cd' to show the current discard"),
793       _("'ch' to challenge a Wild +4"),
794       _("'od' to show the playing order"),
795       _("'ti' to show play time"),
796       _("'tu' to show whose turn it is")
797     ].join(" ; ")
798     when 'challenge'
799       _("A Wild +4 can only be played legally if you don't have normal (not special) cards of the current color. ") +
800       _("The next player can challenge a W+4 by using the 'ch' command. ") +
801       _("If the W+4 play was illegal, the player who played it must pick the W+4, pick 4 cards from the stock, and play a legal card. ") +
802       _("If the W+4 play was legal, the challenger must pick 6 cards instead of 4.")
803     when 'rules'
804       _("play all your cards, one at a time, by matching either the color or the value of the currently discarded card. ") +
805       _("cards with special effects: Skip (next player skips a turn), Reverse (reverses the playing order), +2 (next player has to take 2 cards). ") +
806       _("Wilds can be played on any card, and you must specify the color for the next card. ") +
807       _("Wild +4 also forces the next player to take 4 cards, but it can only be played if you can't play a color card. ") +
808       _("you can play another +2 or +4 card on a +2 card, and a +4 on a +4, forcing the first player who can't play one to pick the cumulative sum of all cards. ") +
809       _("you can also play a Reverse on a +2 or +4, bouncing the effect back to the previous player (that now comes next). ")
810     else
811       (_("%{uno} game. !uno to start a game. see help uno rules for the rules. commands: %{cmds}") % {
812         :uno => UnoGame::UNO,
813         :cmds => help(plugin, 'commands')
814       })
815     end
816   end
817
818   def message(m)
819     return unless @games.key?(m.channel)
820     g = @games[m.channel]
821     case m.plugin.intern
822     when :jo # join game
823       return if m.params
824       g.add_player(m.source)
825     when :pe # pick card
826       return if m.params
827       if g.has_turn?(m.source)
828         if g.player_has_picked
829           m.reply _("you already picked a card")
830         elsif g.picker > 0
831           g.pass(m.source)
832         else
833           g.pick_card(m.source)
834         end
835       else
836         m.reply _("It's not your turn")
837       end
838     when :pa # pass turn
839       return if m.params
840       if g.has_turn?(m.source)
841         g.pass(m.source)
842       else
843         m.reply _("It's not your turn")
844       end
845     when :pl # play card
846       if g.has_turn?(m.source)
847         g.play_card(m.source, m.params.downcase)
848       else
849         m.reply _("It's not your turn")
850       end
851     when :co # pick color
852       if g.has_turn?(m.source)
853         g.choose_color(m.source, m.params.downcase)
854       else
855         m.reply _("It's not your turn")
856       end
857     when :ca # show current cards
858       return if m.params
859       g.show_all_cards(m.source)
860     when :cd # show current discard
861       return if m.params
862       g.show_discard
863     when :ch
864       if g.has_turn?(m.source)
865         if g.last_discard
866           g.challenge
867         else
868           m.reply _("previous move cannot be challenged")
869         end
870       else
871         m.reply _("It's not your turn")
872       end
873     when :od # show playing order
874       return if m.params
875       g.show_order
876     when :ti # show play time
877       return if m.params
878       g.show_time
879     when :tu # show whose turn is it
880       return if m.params
881       if g.has_turn?(m.source)
882         m.nickreply _("it's your turn, sleepyhead")
883       else
884         g.show_turn(:cards => false)
885       end
886     end
887   end
888
889   def create_game(m, p)
890     if @games.key?(m.channel)
891       m.reply _("There is already an %{uno} game running here, say 'jo' to join in") % { :uno => UnoGame::UNO }
892       return
893     end
894     @games[m.channel] = UnoGame.new(self, m.channel)
895     m.reply _("Ok, created %{uno} game on %{channel}, say 'jo' to join in") % {
896       :uno => UnoGame::UNO,
897       :channel => m.channel
898     }
899   end
900
901   def end_game(m, p)
902     unless @games.key?(m.channel)
903       m.reply _("There is no %{uno} game running here") % { :uno => UnoGame::UNO }
904       return
905     end
906     @games[m.channel].end_game(true)
907   end
908
909   def chan_reg(channel)
910     @registry.sub_registry(channel.downcase)
911   end
912
913   def chan_stats(channel)
914     stats = chan_reg(channel).sub_registry('stats')
915     class << stats
916       def store(val)
917         val.to_i
918       end
919       def restore(val)
920         val.to_i
921       end
922     end
923     stats.set_default(0)
924     return stats
925   end
926
927   def chan_pstats(channel)
928     pstats = chan_reg(channel).sub_registry('players')
929     pstats.set_default(UnoPlayerStats.new(0,0,[]))
930     return pstats
931   end
932
933   def do_end_game(channel, closure)
934     reg = chan_reg(channel)
935     stats = chan_stats(channel)
936     stats['played'] += 1
937     stats['played_runtime'] += closure[:runtime]
938     if closure[:winner]
939       stats['finished'] += 1
940       stats['finished_runtime'] += closure[:runtime]
941
942       pstats = chan_pstats(channel)
943
944       closure[:players].each do |pl|
945         k = pl.user.downcase
946         pls = pstats[k]
947         pls.played += 1
948         pstats[k] = pls
949       end
950
951       closure[:dropouts].each do |pl|
952         k = pl.user.downcase
953         pls = pstats[k]
954         pls.played += 1
955         pls.forfeits += 1
956         pstats[k] = pls
957       end
958
959       winner = closure[:winner]
960       won = UnoGameWon.new(closure[:score], closure[:opponents])
961       k = winner.user.downcase
962       pls = pstats[k] # already marked played +1 above
963       pls.won << won
964       pstats[k] = pls
965     end
966
967     @games.delete(channel)
968   end
969
970   def do_chanstats(m, p)
971     stats = chan_stats(m.channel)
972     np = stats['played']
973     nf = stats['finished']
974     if np > 0
975       str = _("%{nf} %{uno} games completed over %{np} games played. ") % {
976         :np => np, :uno => UnoGame::UNO, :nf => nf
977       }
978       cgt = stats['finished_runtime']
979       tgt = stats['played_runtime']
980       str << _("%{cgt} game time for completed games") % {
981         :cgt => Utils.secs_to_string(cgt)
982       }
983       if np > nf
984         str << _(" on %{tgt} total game time. ") % {
985           :tgt => Utils.secs_to_string(tgt)
986         }
987       else
988         str << ". "
989       end
990       str << _("%{avg} average game time for completed games") % {
991         :avg => Utils.secs_to_string(cgt/nf)
992       }
993       str << _(", %{tavg} for all games") % {
994         :tavg => Utils.secs_to_string(tgt/np)
995       } if np > nf
996       m.reply str
997     else
998       m.reply _("nobody has played %{uno} on %{chan} yet") % {
999         :uno => UnoGame::UNO, :chan => m.channel
1000       }
1001     end
1002   end
1003
1004   def do_pstats(m, p)
1005     dnick = p[:nick] || m.source # display-nick, don't later case
1006     nick = dnick.downcase
1007     ps = chan_pstats(m.channel)[nick]
1008     if ps.played == 0
1009       m.reply _("%{nick} never played %{uno} here") % {
1010         :uno => UnoGame::UNO, :nick => dnick
1011       }
1012       return
1013     end
1014     np = ps.played
1015     nf = ps.forfeits
1016     nw = ps.won.length
1017     score = ps.won.inject(0) { |sum, w| sum += w.score }
1018     str = _("%{nick} played %{np} %{uno} games here, ") % {
1019       :nick => dnick, :np => np, :uno => UnoGame::UNO
1020     }
1021     str << _("forfeited %{nf} games, ") % { :nf => nf } if nf > 0
1022     str << _("won %{nw} games") % { :nw => nw}
1023     if nw > 0
1024       str << _(" with %{score} total points") % { :score => score }
1025       avg = ps.won.inject(0) { |sum, w| sum += w.score/w.opponents }/nw
1026       str << _(" and an average of %{avg} points per opponent") % { :avg => avg }
1027     end
1028     m.reply str
1029   end
1030
1031   def replace_player(m, p)
1032     unless @games.key?(m.channel)
1033       m.reply _("There is no %{uno} game running here") % { :uno => UnoGame::UNO }
1034       return
1035     end
1036     @games[m.channel].replace_player(p[:old], p[:new])
1037   end
1038
1039   def drop_player(m, p)
1040     unless @games.key?(m.channel)
1041       m.reply _("There is no %{uno} game running here") % { :uno => UnoGame::UNO }
1042       return
1043     end
1044     @games[m.channel].drop_player(p[:nick] || m.source.nick)
1045   end
1046
1047   def print_stock(m, p)
1048     unless @games.key?(m.channel)
1049       m.reply _("There is no %{uno} game running here") % { :uno => UnoGame::UNO }
1050       return
1051     end
1052     stock = @games[m.channel].stock
1053     m.reply(_("%{num} cards in stock: %{stock}") % {
1054       :num => stock.length,
1055       :stock => stock.join(' ')
1056     }, :split_at => /#{NormalText}\s*/)
1057   end
1058
1059   def do_top(m, p)
1060     pstats = chan_pstats(m.channel)
1061     scores = []
1062     wins = []
1063     pstats.each do |k, v|
1064       wins << [v.won.length, k]
1065       scores << [v.won.inject(0) { |s, w| s+=w.score }, k]
1066     end
1067
1068     if n = p[:scorenum]
1069       msg = _("%{uno} %{num} highest scores: ") % {
1070         :uno => UnoGame::UNO, :num => p[:scorenum]
1071       }
1072       scores.sort! { |a1, a2| -(a1.first <=> a2.first) }
1073       scores = scores[0, n.to_i].compact
1074       if scores.length <= 5
1075         i = 0
1076         list = "\n" + scores.map { |a|
1077           i+=1
1078           _("%{i}. %{b}%{nick}%{b} with %{b}%{score}%{b} points") % {
1079             :i => i, :b => Bold, :nick => a.last, :score => a.first
1080           }
1081         }.join("\n")
1082       else
1083         list = scores.map { |a|
1084           i+=1
1085           _("%{i}. %{nick} ( %{score} )") % {
1086             :i => i, :nick => a.last, :score => a.first
1087           }
1088         }.join(" | ")
1089       end
1090     elsif n = p[:winnum]
1091       msg = _("%{uno} %{num} most wins: ") % {
1092         :uno => UnoGame::UNO, :num => p[:winnum]
1093       }
1094       wins.sort! { |a1, a2| -(a1.first <=> a2.first) }
1095       wins = wins[0, n.to_i].compact
1096       if wins.length <= 5
1097         i = 0
1098         list = "\n" + wins.map { |a|
1099           i+=1
1100           _("%{i}. %{b}%{nick}%{b} with %{b}%{score}%{b} wins") % {
1101             :i => i, :b => Bold, :nick => a.last, :score => a.first
1102           }
1103         }.join("\n")
1104       else
1105         list = wins.map { |a|
1106           i+=1
1107           _("%{i}. %{nick} ( %{score} )") % {
1108             :i => i, :nick => a.last, :score => a.first
1109           }
1110         }.join(" | ")
1111       end
1112     else
1113       msg = _("uh, what kind of score list did you want, again?")
1114       list = _(" I can only show the top scores (with top) and the most wins (with topwin)")
1115     end
1116     m.reply msg + list, :max_lines => (msg+list).count("\n")+1
1117   end
1118 end
1119
1120 pg = UnoPlugin.new
1121
1122 pg.map 'uno', :private => false, :action => :create_game
1123 pg.map 'uno end', :private => false, :action => :end_game, :auth_path => 'manage'
1124 pg.map 'uno drop', :private => false, :action => :drop_player, :auth_path => 'manage::drop::self!'
1125 pg.map 'uno giveup', :private => false, :action => :drop_player, :auth_path => 'manage::drop::self!'
1126 pg.map 'uno drop :nick', :private => false, :action => :drop_player, :auth_path => 'manage::drop::other!'
1127 pg.map 'uno replace :old [with] :new', :private => false, :action => :replace_player, :auth_path => 'manage'
1128 pg.map 'uno stock', :private => false, :action => :print_stock
1129 pg.map 'uno chanstats', :private => false, :action => :do_chanstats
1130 pg.map 'uno stats [:nick]', :private => false, :action => :do_pstats
1131 pg.map 'uno top :scorenum', :private => false, :action => :do_top, :defaults => { :scorenum => 5 }
1132 pg.map 'uno topwin :winnum', :private => false, :action => :do_top, :defaults => { :winnum => 5 }
1133
1134 pg.default_auth('stock', false)
1135 pg.default_auth('manage', false)
1136 pg.default_auth('manage::drop::self', true)