-##################
-# Filename: dice.rb
-# Description: Rbot plugin. Rolls rpg style dice
-# Author: David Dorward (http://david.us-lot.org/ - you might find a more up to date version of this plugin there)
-# Version: 0.3.2
-# Date: Sat 6 Apr 2002
+#-- vim:sw=2:et
+#++
#
-# You can get rbot from: http://www.linuxbrit.co.uk/rbot/
+# :title: dice plugin for rbot
+#
+# Author:: David Dorward (http://david.us-lot.org/ - you might find a more up to date version of this plugin there)
+# Author:: Moritz Augsburger <moritz@augsburger.name>
+#
+# Description:: Rolls rpg style dice
+# Version:: 0.4
+# Date:: Mon 8 Feb 2008
#
# Changelog
# 0.1 - Initial release
# - Return results of each roll
# 0.3.1 - Minor documentation update
# 0.3.2 - Bug fix, could not subtract numbers (String can't be coerced into Fixnum)
+# 0.4 - Limit number of dices and number of sides per dice
#
-# TODO: Test! Test! Test!
-# Comment!
-# Fumble/Critical counter (1's and x's where x is sides on dice)
-####################################################
+# TODO:: Test! Test! Test!
+# Comment!
+# Fumble/Critical counter (1's and x's where x is sides on dice)
class DiceDisplay
- attr_reader :total, :view
- def initialize(view, total)
+ attr_reader :total, :view, :dice
+ def initialize(dice, view, total)
@total = total
+ @dice = dice
@view = view
end
+
+ def get_view()
+ return "["+ dice.to_s + ": " + total.to_s + " | " + view + "] "
+ end
end
class DicePlugin < Plugin
+ Config.register Config::IntegerValue.new('dice.max_dices',
+ :default => 100, :validate => Proc.new{|v| v > 0},
+ :desc => "Maximum number of dices to throw.")
+
+ Config.register Config::IntegerValue.new('dice.max_sides',
+ :default => 100, :validate => Proc.new{|v| v > 0},
+ :desc => "Maximum number of sides per dice.")
+
def help(plugin, topic="")
- "dice <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"
+ 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"
end
def rolldice(d)
dice = d.split(/d/)
+ repr = []
r = 0
- unless dice[0] =~ /^[0-9]+/
+ unless dice[0] =~ /^\d+/
dice[0] = 1
end
for i in 0...dice[0].to_i
- r = r + rand(dice[1].to_i) + 1
+ tmp = rand(dice[1].to_i) + 1
+ repr << tmp.to_s
+ r = r + tmp
end
- return r
+ return DiceDisplay.new(d, repr.join(", "), r)
end
def iddice(d)
+ dice = d
porm = d.slice!(0,1)
if d =~ /d/
- r = rolldice(d)
+ rolled = rolldice(d)
+ d = rolled.view
+ r = rolled.total
else
r = d
end
+
if porm == "-"
r = 0 - r.to_i
end
- viewer = DiceDisplay.new("[" + porm.to_s + d.to_s + "=" + r.to_s + "] ", r)
+
+ viewer = DiceDisplay.new(porm + dice, d.to_s, r)
return viewer
end
def privmsg(m)
- unless(m.params && m.params =~ /^[0-9]*d[0-9]+([+-]([0-9]+|[0-9]*d[0-9])+)*$/)
- m.nickreply "incorrect usage: " + help(m.plugin)
+ # If either not given parameters or given incorrect parameters, return with
+ # the help message
+ unless m.params && m.params =~ /^\d*d\d+(\s*[+-]\s*(\d+|\d*d\d)+)*$/
+ m.reply "incorrect usage: " + help(m.plugin)
+ return
+ end
+
+ # Extract the actual dice request from the message parameters, splitting it
+ # into dice and modifiers
+ a = m.params.gsub(/\s+/,'').scan(/^\d*d\d+|[+-]\d*d\d+|[+-]\d+/)
+ # check nr of total dices and sides per dice
+ nr = 0
+ a.each { |dice|
+ dc, ds = dice.split(/d/)
+ # check sides
+ if ds.to_i > @bot.config['dice.max_sides']
+ m.reply "sorry, don't have any dices with more than %u sides" % @bot.config['dice.max_sides'], :nick => true
+ return
+ end
+ # We use .max with 1 so that specs such as d6 count as 1 and not as 0
+ nr += [dc.to_i, 1].max
+ }
+ if nr > @bot.config['dice.max_dices']
+ m.reply "can't handle more than %u dices" % @bot.config['dice.max_dices'], :nick => true
return
end
- a = m.params.scan(/^[0-9]*d[0-9]+|[+-][0-9]*d[0-9]+|[+-][0-9]+/)
- r = rolldice(a[0])
- t = "[" + a[0].to_s + "=" + r.to_s + "] "
+
+ # Roll the dice with the extracted request
+ rolled = rolldice(a[0])
+ r = rolled.total
+ t = rolled.get_view()
+
+ # Deal with all the remaining parts of the given dice request
for i in 1...a.length
tmp = iddice(a[i])
r = r + tmp.total.to_i
- t = t + tmp.view.to_s
+ t = t + tmp.get_view
end
- m.nickreply r.to_s + " | " + t
+ t.chop!
+ m.reply(r.to_s + " || " + m.params + ": " + t, :nick => true)
end
end
plugin = DicePlugin.new
projects / user / henk / code / ruby / rbot.git / blobdiff