data/rbot/plugins/karma.rb

   1 class KarmaPlugin < Plugin
   2   def initialize
   3     super
   4
   5     # this plugin only wants to store ints!
   6     class << @registry
   7       def store(val)
   8         val.to_i
   9       end
  10       def restore(val)
  11         val.to_i
  12       end
  13     end
  14     @registry.set_default(0)
  15
  16     # import if old file format found
  17     if(File.exist?("#{@bot.botclass}/karma.rbot"))
  18       log "importing old karma data"
  19       IO.foreach("#{@bot.botclass}/karma.rbot") do |line|
  20         if(line =~ /^(\S+)<=>([\d-]+)$/)
  21           item = $1
  22           karma = $2.to_i
  23           @registry[item] = karma
  24         end
  25       end
  26       File.delete("#{@bot.botclass}/karma.rbot")
  27     end
  28   end
  29
  30   def stats(m, params)
  31     if (@registry.length)
  32       max = @registry.values.max
  33       min = @registry.values.min
  34       best = @registry.to_hash.index(max)
  35       worst = @registry.to_hash.index(min)
  36       m.reply "#{@registry.length} items. Best: #{best} (#{max}); Worst: #{worst} (#{min})"
  37     end
  38   end
  39
  40   def karma(m, params)
  41     thing = params[:key]
  42     thing = m.sourcenick unless thing
  43     thing = thing.to_s
  44     karma = @registry[thing]
  45     if(karma != 0)
  46       m.reply "karma for #{thing}: #{@registry[thing]}"
  47     else
  48       m.reply "#{thing} has neutral karma"
  49     end
  50   end
  51
  52   def setkarma(m, params)
  53     thing = (params[:key] || m.sourcenick).to_s
  54     @registry[thing] = params[:val].to_i
  55     karma(m, params)
  56   end
  57
  58   def help(plugin, topic="")
  59     "karma module: Listens to everyone's chat. <thing>++/<thing>-- => increase/decrease karma for <thing>, karma for <thing>? => show karma for <thing>, karmastats => show stats. Karma is a community rating system - only in-channel messages can affect karma and you cannot adjust your own."
  60   end
  61
  62   def listen(m)
  63     return unless m.kind_of?(PrivMessage) && m.public?
  64     # in channel message, the kind we are interested in
  65     if(m.message =~ /(\+\+|--)/)
  66       string = m.message.sub(/\W(--|\+\+)(\(.*?\)|[^(++)(\-\-)\s]+)/, "\2\1")
  67       seen = Hash.new
  68       while(string.sub!(/(\(.*?\)|[^(++)(\-\-)\s]+)(\+\+|--)/, ""))
  69         key = $1
  70         change = $2
  71         next if seen[key]
  72         seen[key] = true
  73
  74         key.sub!(/^\((.*)\)$/, "\1")
  75         key.gsub!(/\s+/, " ")
  76         next unless(key.length > 0)
  77         next if(key.downcase == m.sourcenick.downcase)
  78         if(change == "++")
  79           @registry[key] += 1
  80           if key =~ /^#{@bot.nick}$/i
  81             @bot.say m.replyto, @bot.lang.get("thanks")
  82           end
  83         elsif(change == "--")
  84           @registry[key] -= 1
  85         end
  86       end
  87     end
  88   end
  89 end
  90
  91 plugin = KarmaPlugin.new
  92
  93 plugin.default_auth( 'edit', false )
  94
  95 plugin.map 'karmastats', :action => 'stats'
  96 plugin.map 'karma :key', :defaults => {:key => false}
  97 plugin.map 'setkarma :key :val', :defaults => {:key => false}, :requirements => {:val => /^-?\d+$/}, :auth_path => 'edit::set!'
  98 plugin.map 'karma for :key'