rbot-plugins/coin.rb at master · gfax/rbot-plugins · GitHub

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
# :title: Coin Toss
#
# Author:: jazzyfox <[email protected]>
# Author:: Lite <[email protected]>
# URL:: https://github.com/gfax/rbot-plugins.git
# Version:: 2013-02-09
#


class CoinPlugin < Plugin

  def help(plugin, topic='')
    p = @bot.config['core.address_prefix'].first
    return "#{p}#{plugin} => flip a coin, " +
           "#{p}#{plugin} stats => show your coin flipping stats"
  end

  def initialize
    @coin = ['heads', 'tails']
    super
  end

  def flip_coin(m, params)
    num = params[:number].to_i
    m.reply "I can only flip positive numbers of coins." unless num > 0
    m.reply "Sorry, I've only got 20 coins in my pocket." unless num <= 20
    return unless num > 0 && num <= 20
    playerdata = get_data(m.sourcenick) || Array.new(2,0)
    result = []
    1.upto(num) do |i|
      flip = rand(2)
      playerdata[flip] += 1
      result << @coin[flip]
    end
    m.reply result.join(', ')
    @registry[m.sourcenick.downcase] = playerdata
  end

  def coin_stats(m, params)
    data = params[:nick] ? get_data(params[:nick]) : get_data(m.sourcenick)
    if data
      total = data.inject {|sum, i| sum + i}
      heads, tails = data
      hp, tp = data.map {|i| sprintf("%.2f", (i.to_f / total) * 100)}
      m.reply "#{heads} heads (#{hp}%) and #{tails} tails (#{tp}%)"
    elsif params[:nick]
      m.reply "Sorry, #{params[:nick]} hasn't flipped any coins."
    else
      m.reply "Sorry, you haven't flipped any coins."
    end
  end

  private

  def get_data(name)
    @registry[name.downcase] if @registry.has_key?(name.downcase)
  end

end

plugin = CoinPlugin.new

[ 'coin', 'cointoss' ].each do |scope|
  plugin.map "#{scope} stats [:nick]",
    :action => 'coin_stats'
  plugin.map "#{scope} :number",
    :action => 'flip_coin', :defaults => { :number => 1 }
end