data/rbot/plugins/dice.rb

   1 ##################
   2 # Filename: dice.rb
   3 # Description: Rbot plugin. Rolls rpg style dice
   4 # Author: David Dorward (http://david.us-lot.org/ - you might find a more up to date version of this plugin there)
   5 # Version: 0.3.2
   6 # Date: Sat 6 Apr 2002
   7 #
   8 # You can get rbot from: http://www.linuxbrit.co.uk/rbot/
   9 #
  10 # Changelog
  11 # 0.1 - Initial release
  12 # 0.1.1 - bug fix, only 1 digit for number of dice sides on first roll
  13 # 0.3.0 - Spelling correction on changelog 0.1.1
  14 #       - Return results of each roll
  15 # 0.3.1 - Minor documentation update
  16 # 0.3.2 - Bug fix, could not subtract numbers (String can't be coerced into Fixnum)
  17 #
  18 # TODO: Test! Test! Test!
  19 #       Comment!
  20 #       Fumble/Critical counter (1's and x's where x is sides on dice)
  21 ####################################################
  22
  23 class DiceDisplay
  24   attr_reader :total, :view, :dice
  25   def initialize(dice, view, total)
  26     @total = total
  27     @dice = dice
  28     @view = view
  29   end
  30
  31   def get_view()
  32     return "["+ dice.to_s + ": " + total.to_s + " | " + view + "] "
  33   end
  34 end
  35
  36 class DicePlugin < Plugin
  37   Config.register Config::IntegerValue.new('dice.max_dices',
  38       :default => 100, :validate => Proc.new{|v| v > 0},
  39       :desc => "Maximum number of dices to throw.")
  40
  41   Config.register Config::IntegerValue.new('dice.max_sides',
  42       :default => 100, :validate => Proc.new{|v| v > 0},
  43       :desc => "Maximum number of sides per dice.")
  44
  45   def help(plugin, topic="")
  46     plugin + " <string> (where <string> is something like: d6 or 2d6 or 2d6+4 or 2d6+1d20 or 2d6+1d5+4d7-3d4-6) => Rolls that set of virtual dice"
  47   end
  48
  49   def rolldice(d)
  50     dice = d.split(/d/)
  51     repr = []
  52     r = 0
  53     unless dice[0] =~ /^[0-9]+/
  54       dice[0] = 1
  55     end
  56     for i in 0...dice[0].to_i
  57       tmp = rand(dice[1].to_i) + 1
  58       repr << tmp.to_s
  59       r = r + tmp
  60     end
  61     return DiceDisplay.new(d, repr.join(", "), r)
  62   end
  63
  64   def iddice(d)
  65     dice = d
  66     porm = d.slice!(0,1)
  67     if d =~ /d/
  68       rolled = rolldice(d)
  69       d = rolled.view
  70       r = rolled.total
  71     else
  72       r = d
  73     end
  74
  75     if porm == "-"
  76       r = 0 - r.to_i
  77     end
  78
  79     viewer = DiceDisplay.new(porm + dice, d.to_s, r)
  80     return viewer
  81   end
  82
  83   def privmsg(m)
  84     # If either not given parameters or given incorrect parameters, return with
  85     # the help message
  86     unless(m.params && m.params =~ /^[0-9]*d[0-9]+(\s*[+-]\s*([0-9]+|[0-9]*d[0-9])+)*$/)
  87       m.nickreply "incorrect usage: " + help(m.plugin)
  88       return
  89     end
  90
  91     # Extract the actual dice request from the message parameters, splitting it
  92     # into dice and modifiers
  93     a = m.params.gsub(/\s+/,'').scan(/^[0-9]*d[0-9]+|[+-][0-9]*d[0-9]+|[+-][0-9]+/)
  94     # check nr of total dices and sides per dice
  95     nr = 0
  96     a.each { |dice|
  97       dc, ds = dice.split(/d/)
  98       # check sides
  99       if ds.to_i > @bot.config['dice.max_sides']
 100        m.reply "sorry, don't have any dices with more than %u sides" % @bot.config['dice.max_sides']
 101        return
 102       end
 103       # We use .max with 1 so that specs such as d6 count as 1 and not as 0
 104       nr += [dc.to_i, 1].max
 105     }
 106     if nr > @bot.config['dice.max_dices']
 107       m.reply "can't handle more than %u dices" % @bot.config['dice.max_dices']
 108       return
 109     end
 110
 111     # Roll the dice with the extracted request
 112     rolled = rolldice(a[0])
 113     r = rolled.total
 114     t = rolled.get_view()
 115
 116     # Deal with all the remaining parts of the given dice request
 117     for i in 1...a.length
 118       tmp = iddice(a[i])
 119       r = r + tmp.total.to_i
 120       t = t + tmp.get_view
 121     end
 122     t.chop!
 123     m.nickreply r.to_s + " || " + m.params + ": " + t
 124   end
 125 end
 126 plugin = DicePlugin.new
 127 plugin.register("dice")
 128 plugin.register("roll")
 129 ##############################################
 130 #fin