# Very alpha stage, so beware of sudden reaction syntax changes
class ::Reaction
- attr_reader :trigger, :reply
- attr_reader :raw_trigger, :raw_reply
- attr_accessor :author, :date, :channel
+ attr_reader :trigger, :replies
+ attr_reader :raw_trigger, :raw_replies
+
+ class ::Reply
+ attr_reader :act, :reply, :pct, :range
+ attr_reader :author, :date, :channel
+ attr_writer :date
+
+ def pct=(val)
+ @pct = val
+ @reaction.make_ranges
+ end
+
+ def author=(name)
+ @author = name.to_s
+ end
+
+ def channel=(name)
+ @channel = name.to_s
+ end
+
+ def initialize(reaction, act, expr, pct, author, date, channel)
+ @reaction = reaction
+ @act = act
+ @reply = expr
+ self.pct = pct
+ self.author = author
+ @date = date
+ self.channel = channel
+ end
+
+ def to_s
+ [
+ "#{act} #{reply} (#{pct} chance)",
+ @range ? "(#{@range})" : "",
+ "(#{author}, #{channel}, #{date})"
+ ].join(" ")
+ end
+
+ def apply(subs={})
+ [act, reply % subs]
+ end
+ end
def trigger=(expr)
@raw_trigger = expr.dup
end
@trigger = [act]
if rex.sub!(%r@^([/!])(.*)\1$@, '\2')
- debug rex
@trigger << Regexp.new(rex)
else
@trigger << Regexp.new(/\b#{Regexp.escape(rex)}\b/u)
end
end
- def reply=(expr)
- @raw_reply = expr.dup
+ def add_reply(expr, *args)
+ @raw_replies << expr.dup
act = false
rex = expr.dup
if rex.sub!(/^act:/,'')
act = true
end
- @reply = [act ? :act : :reply]
- @reply << rex
+ @replies << Reply.new(self, act ? :act : :reply, rex, *args)
+ make_ranges
+ return @replies.last
+ end
+
+ def find_reply(expr)
+ @replies[@raw_replies.index(expr)] rescue nil
+ end
+
+ def make_ranges
+ totals = 0
+ pcts = @replies.map { |rep|
+ totals += rep.pct
+ rep.pct
+ }
+ pcts.map! { |p|
+ p/totals
+ } if totals > 1
+ debug "percentages: #{pcts.inspect}"
+
+ last = 0
+ @replies.each_with_index { |r, i|
+ p = pcts[i]
+ r.range = last..(last+p)
+ last+=p
+ }
+ debug "ranges: #{@replies.map { |r| r.range}.inspect}"
+ end
+
+ def pick_reply
+ pick = rand()
+ debug "#{pick} in #{@replies.map { |r| r.range}.inspect}"
+ @replies.each { |r|
+ return r if r.range and r.range === pick
+ }
+ return nil
end
def ===(message)
return message.message.match(@trigger.last)
end
- def initialize(trig, msg , auth, dt, chan)
+ def initialize(trig)
self.trigger=trig
- self.reply=msg
- self.author=auth.to_s
- self.date=dt
- self.channel=chan.to_s
+ @raw_replies = []
+ @replies = []
end
def to_s
- "trigger #{raw_trigger} (#{author}, #{channel}, #{date})"
+ raw_trigger
end
end
class ReactionPlugin < Plugin
- ADD_SYNTAX = 'react to *trigger with *reply'
+ ADD_SYNTAX = 'react to *trigger with *reply at :chance chance'
def add_syntax
return ADD_SYNTAX
def help(plugin, topic="")
if plugin.to_sym == :react
- return "react to <trigger> with <reply> => create a new reaction to expression <trigger> to which the bot will reply <reply>, seek help for reaction trigger and reaction reply for more details"
+ return "react to <trigger> with <reply> [at <chance> chance] => " +
+ "create a new reaction to expression <trigger> to which the bot will reply <reply>, optionally at chance <chance>, " +
+ "seek help for reaction trigger, reaction reply and reaction chance for more details"
end
case (topic.to_sym rescue nil)
+ when :add
+ help(:react)
when :remove, :delete, :rm, :del
"reaction #{topic} <trigger> => removes the reaction to expression <trigger>"
+ when :chance, :chances
+ "reaction chances are expressed either in terms of percentage (like 30%) or in terms of floating point numbers (like 0.3), and are clipped to be " +
+ "between 0 and 1 (i.e. 0% and 100%). A reaction can have multiple replies, each with a different chance; if the total of the chances is less than one, " +
+ "there is a chance that the trigger will not actually cause a reply. Otherwise, the chances express the relative frequency of the replies."
when :trigger, :triggers
"reaction triggers can have one of the format: single_word 'multiple words' \"multiple words \" /regular_expression/ !regular_expression!. " +
"If prefixed by 'act:' (e.g. act:/(order|command)s/) the bot will only respond if a CTCP ACTION matches the trigger"
"stuff (everything that follows the trigger), match (the actual matched text)"
when :list
"reaction list [n]: lists the n-the page of programmed reactions (30 reactions are listed per page)"
+ when :show
+ "reaction show <trigger>: list the programmed replies to trigger <trigger>"
else
- "reaction topics: add, remove, delete, rm, del, triggers, replies, list"
+ "reaction topics: add, remove, delete, rm, del, triggers, replies, chance, list, show"
end
end
:stuff => stuff
}
subs = @subs.dup.merge extra
- args = [wanted.reply.first]
- args << wanted.reply.last % extra
+ reply = wanted.pick_reply
+ debug "picked #{reply}"
+ return unless reply
+ args = reply.apply(subs)
m.__send__(*args)
end
def handle_add(m, params)
trigger = params[:trigger].to_s
reply = params[:reply].to_s
- if find_reaction(trigger)
- m.reply "there's already a reaction to #{trigger}"
- return
+
+ pct = params[:chance] || "1"
+ if pct.sub!(/%$/,'')
+ pct = (pct.to_f/100).clip(0,1)
+ else
+ pct = pct.to_f.clip(0,1)
end
- reaction = Reaction.new(trigger, reply, m.sourcenick, Time.now, m.channel)
- @reactions << reaction
- m.reply "added reaction to #{reaction.trigger.last} with #{reaction.reply.last}"
+ reaction = find_reaction(trigger)
+ if not reaction
+ reaction = Reaction.new(trigger)
+ @reactions << reaction
+ m.reply "Ok, I'll start reacting to #{reaction.raw_trigger}"
+ end
+ found = reaction.find_reply(reply)
+ if found
+ found.pct = pct
+ found.author = m.sourcenick
+ found.date = Time.now
+ found.channel = m.channel
+ else
+ found = reaction.add_reply(reply, pct, m.sourcenick, Time.now, m.channel)
+ end
+ m.reply "I'll react to #{reaction.raw_trigger} with #{reaction.raw_replies.last} (#{(reaction.replies.last.pct * 100).to_i}%)"
end
def handle_rm(m, params)
found = find_reaction(trigger)
if found
@reactions.delete(found)
- m.reply "removed reaction to #{found.trigger.last} with #{found.reply.last}"
+ m.reply "I won't react to #{found.raw_trigger} anymore"
else
m.reply "no reaction programmed for #{trigger}"
end
m.reply "Programmed reactions (page #{page}/#{pages}): #{str}"
end
+ def handle_show(m, params)
+ if @reactions.empty?
+ m.reply "no reactions programmed"
+ return
+ end
+
+ trigger = params[:trigger].to_s
+
+ found = find_reaction(trigger)
+
+ unless found
+ m.reply "I'm not reacting to #{trigger}"
+ return
+ end
+
+ m.reply found.replies.join(", ")
+ end
+
end
plugin = ReactionPlugin.new
plugin.map 'reaction list [:page]', :action => 'handle_list',
:requirements => { :page => /^\d+$/ }
+plugin.map 'reaction show *trigger', :action => 'handle_show'
+
plugin.map 'reaction del[ete] *trigger', :action => 'handle_rm'
plugin.map 'reaction delete *trigger', :action => 'handle_rm'
plugin.map 'reaction remove *trigger', :action => 'handle_rm'
projects / user / henk / code / ruby / rbot.git / blobdiff