-require 'uri/common'
-require 'cgi'
-
-# the 4q.cc "id => full name" mapping\r
-FACTMAP = { "mrt" => "Mr\. T",\r
- "vin" => "Vin Diesel",\r
- "chuck" => "Chuck Norris" }\r
+require 'yaml'
+require 'zlib'
MIN_RATING = 6.0
-
-PISSED_EXPRESSIONS = [
- "fuck this, i'm going to go get toed up.",
- "screw this, i'm going to get hammered.",
- "forget this, i'm going to iron some shirts.",
- "disregard this, i'm going out to kill me some prostitutes.",
-]
-
-# exceptions
-class HTTPError < Exception; end
-class ParseError < Exception; end
+MIN_VOTES = 25
# the plugin
class ChuckNorrisPlugin < Plugin
- def help(plugin, topic="")
- "fact [person] => \"fact\" shows a random Chuck Norris, Vin Diesel, or Mr. T fact. \"fact [person]\" shows a fact about someone in the channel. || chucknorris, chuck, norris => random Chuck Norris fact || vindiesel, vin, diesel => random Vin Diesel fact || mrt => I pity the foo who can't figure this one out."
+ # Loadez les factes
+ def initialize
+ if path = find_facts_file('chucknorris.yml.gz')
+ fyml = Zlib::GzipReader.open(path)
+ elsif path = find_facts_file('chucknorris.yml')
+ fyml = open(path)
+ else
+ raise "Error: Couldn't find chucknorris.yml[.gz]"
+ end
+
+ debug "+ [chucknorris] Loading #{path}..."
+
+ @@facts = YAML.load(fyml).map{|fact,(score,votes)| votes >= MIN_VOTES ? [score,fact] : nil}.compact
+ debug "+ [chucknorris] #{@@facts.length} Chuck Norris facts loaded..."
+ debug " Random fact: #{@@facts[rand(@@facts.size)].inspect}"
+
+ super
end
- def getfact(who)
- raise "Unknown name: #{who}" unless FACTMAP.keys.include? who
- # get the fact
- factdata = @bot.httputil.get(URI.parse("http://www.4q.cc/index.php?pid=fact&person=#{who}"))
- unless factdata
- raise HTTPError
- end
-
- longwho = FACTMAP[who]
-
- # regexes
- fact_matcher = %r{<h1> And now a random fact about #{longwho}...</h1>(.+?)<hr />}
- rating_matcher = %r{Current Rating: <b>(\d+\.\d+)</b>}
-
- # parse the fact
- if factdata =~ fact_matcher
- fact = CGI::unescapeHTML($1)
- if factdata =~ rating_matcher
- rating = $1.to_f
- puts "fact=[#{fact}], rating=[#{rating}]"
- return [fact, rating]
- end
- end
-
- raise ParseError
-
+ def name
+ "chucknorris"
end
- def fact(m, params)
- who = params[:who]
- max_tries = (params[:tries] or "10").to_i
-
- valid_people = FACTMAP.keys + ["random"]
-
- # if the person wants a fact about themselves, then it'll substitute the name.
- if valid_people.include? who
- substitute_name = nil
+ # Just a little helper for the initialize method...
+ def find_facts_file(name)
+ full_path = File.join Config::datadir, "plugins", name
+ found_files = Dir[full_path]
+ if found_files.empty?
+ nil
else
- substitute_name = who
- who = 'random'
- end\r
-
- # pick a random person\r
- if who == 'random'
- if substitute_name
- # take out the Mr. T facts if you're inserting someone's name
- # beacuse tons of them suck, and most of them revolve around
- # "pitying" someone or something.
- people = FACTMAP.keys - ["mrt"]
- who = people[rand(people.length)]
- else
- who = FACTMAP.keys[rand(FACTMAP.length)]
- end\r
- end\r
-
- # get the long name\r
- longwho = FACTMAP[who]\r
- unless longwho\r
- m.reply "Who the crap is #{who}?!?!"\r
- return\r
- end\r
- \r
- # get the fact
-
- m.reply "alright, let's see if I can find a good one..."
-
- tries = 0
- results = []
- loop do
-
- begin
-
- puts "[chucknorris] Try number #{tries}/#{max_tries}..."
-
- tries += 1
- fact, rating = getfact(who)
-
- if rating >= MIN_RATING
- fact.gsub!(longwho, substitute_name) if substitute_name
- m.reply "#{results.join(', ') + "... "}hrm, this one's not bad:"
- m.reply "#{fact} [rating: #{rating}]"
- return
- else
- results << "lame"
- end
-
- if tries > max_tries
- m.reply "#{results.join(', ')}... these all suck. #{PISSED_EXPRESSIONS[rand(PISSED_EXPRESSIONS.length)]}"
- return
- end
-
- rescue HTTPError
- #m.reply "This #{longwho} fact punched my teeth in. (HTTP error)"
- results << "DOH!"
- tries += 1
- rescue ParseError
- #m.reply "This #{longwho} fact made my brain explode. (Parse error)"
- results << "wtf?"
- tries += 1
- end
-
+ found_files[0]
end
-
end
+ # HELP!
+ def help(plugin, topic="chuck")
+ "chuck|norris|chucknorris [min_rating] => show a random Chuck Norris fact (optional minimum rating from 1-10, default=6.0)."
+ #\"fact [person]\" shows a fact about someone in the channel.
+ end
-end
-
-plugin = ChuckNorrisPlugin.new\r
+ # The meat.
+ def fact(m, params)
+ min = params[:minrating].to_f
+ debug "+ Getting Chuck Norris fact (rating > #{min})..."
+
+ viable_facts = @@facts.select {|rating, fact| rating >= min}
+ if viable_facts.empty?
+ debug " - no facts found with rating >= #{min}"
+ m.reply "Are you nuts?!? There are no facts better than #{min}!!!"
+ return
+ end
-plugin.map 'fact :who :tries', :action => 'fact',
- :defaults => {:who => 'random', :tries=>10}
+ rating, fact = viable_facts[rand(viable_facts.length)]
+ m.reply "#{fact} [score=#{rating}]"
+ end
-plugin.map 'chucknorris :who', :action => 'fact', :defaults => {:who => "chuck"}
-plugin.map 'chuck :who', :action => 'fact', :defaults => {:who => "chuck"}
-plugin.map 'norris :who', :action => 'fact', :defaults => {:who => "chuck"}
+end
-plugin.map 'vindiesel :who', :action => 'fact', :defaults => {:who => "vin"}
-plugin.map 'diesel :who', :action => 'fact', :defaults => {:who => "vin"}
-plugin.map 'vin :who', :action => 'fact', :defaults => {:who => "vin"}
+plugin = ChuckNorrisPlugin.new
-plugin.map 'mrt :who', :action => 'fact', :defaults => {:who => "mrt"}
+# plugin.map 'fact :minrating', :action => 'fact', :defaults => {:minrating=>MIN_RATING}
+plugin.map 'chucknorris :minrating', :action => 'fact', :defaults => {:minrating=>MIN_RATING}
+plugin.map 'chuck :minrating', :action => 'fact', :defaults => {:minrating=>MIN_RATING}
+plugin.map 'norris :minrating', :action => 'fact', :defaults => {:minrating=>MIN_RATING}
projects / user / henk / code / ruby / rbot.git / blobdiff