remind plugin: fix for repeated reminders failing after [717].
[rbot] / data / rbot / plugins / chucknorris.rb
1 require 'yaml'
2 require 'zlib'
3
4 MIN_RATING = 6.0
5 MIN_VOTES = 25
6
7 # the plugin
8 class ChuckNorrisPlugin < Plugin
9
10   # Loadez les factes
11   def initialize
12     if path = find_facts_file('chucknorris.yml.gz')
13       fyml = Zlib::GzipReader.open(path)
14     elsif path = find_facts_File('chucknorris.yml')
15       fyml = open(path)
16     else
17       raise "Error: Couldn't find chucknorris.yml[.gz]"
18     end
19     
20     debug "+ [chucknorris] Loading #{path}..."
21     
22     @@facts = YAML.load(fyml).map{|fact,(score,votes)| votes >= MIN_VOTES ? [score,fact] : nil}.compact
23     debug "+ [chucknorris] #{@@facts.length} Chuck Norris facts loaded..."
24     debug "  Random fact: #{@@facts[rand(@@facts.size)].inspect}"
25     
26     super
27   end
28
29   def name
30     "chucknorris"
31   end
32   
33   # Just a little helper for the initialize method...
34   def find_facts_file(name)
35     full_path = File.join Config::datadir, "plugins", name
36     found_files = Dir[full_path]
37     if found_files.empty?
38       nil
39     else
40       found_files[0]
41     end
42   end
43   
44   # HELP!
45   def help(plugin, topic="chuck")
46     "fact|chuck|norris|chucknorris [min_rating] => \"fact\" shows a random Chuck Norris fact (optional minimum rating from 1-10, default=6.0)."
47     #\"fact [person]\" shows a fact about someone in the channel. 
48   end
49
50   # The meat.
51   def fact(m, params)
52     min = params[:minrating].to_f
53     debug "+ Getting Chuck Norris fact (rating > #{min})..."
54
55     viable_facts = @@facts.select {|rating, fact| rating >= min}
56     if viable_facts.empty?
57       debug "  - no facts found with rating >= #{min}"
58       m.reply "Are you nuts?!? There are no facts better than #{min}!!!"
59       return
60     end
61
62     rating, fact = viable_facts[rand(viable_facts.length)]
63     m.reply "#{fact} [score=#{rating}]"
64   end
65
66 end
67
68 plugin = ChuckNorrisPlugin.new
69
70 plugin.map 'fact :minrating', :action => 'fact', :defaults => {:minrating=>MIN_RATING}
71 plugin.map 'chucknorris :minrating', :action => 'fact', :defaults => {:minrating=>MIN_RATING}
72 plugin.map 'chuck :minrating', :action => 'fact', :defaults => {:minrating=>MIN_RATING}
73 plugin.map 'norris :minrating', :action => 'fact', :defaults => {:minrating=>MIN_RATING}
74