4 # :title: Quiz plugin for rbot
6 # Author:: Mark Kretschmann <markey@web.de>
7 # Author:: Jocke Andersson <ajocke@gmail.com>
8 # Author:: Giuseppe Bilotta <giuseppe.bilotta@gmail.com>
9 # Author:: Yaohan Chen <yaohan.chen@gmail.com>
11 # Copyright:: (C) 2006 Mark Kretschmann, Jocke Andersson, Giuseppe Bilotta
12 # Copyright:: (C) 2007 Giuseppe Bilotta, Yaohan Chen
16 # A trivia quiz game. Fast paced, featureful and fun.
18 # FIXME:: interesting fact: in the Quiz class, @registry.has_key? seems to be
19 # case insensitive. Although this is all right for us, this leads to
20 # rank vs registry mismatches. So we have to make the @rank_table
21 # comparisons case insensitive as well. For the moment, redefine
22 # everything to downcase before matching the nick.
24 # TODO:: define a class for the rank table. We might also need it for scoring
27 # TODO:: when Ruby 2.0 gets out, fix the FIXME 2.0 UTF-8 workarounds
29 # Class for storing question/answer pairs
30 define_structure :QuizBundle, :question, :answer
32 # Class for storing player stats
33 define_structure :PlayerStats, :score, :jokers, :jokers_time
34 # Why do we still need jokers_time? //Firetech
36 # Maximum number of jokers a player can gain
44 #######################################################################
46 # Abstract an answer to a quiz question, by providing self as a string
47 # and a core that can be answered as an alternative. It also provides
48 # a boolean that tells if the core is numeric or not
49 #######################################################################
56 if @string =~ /#(.+)#/
58 @string.gsub!('#', '')
60 raise ArgumentError, "empty string can't be a valid answer!" if @string.empty?
61 raise ArgumentError, "empty core can't be a valid answer!" if @core and @core.empty?
63 @numeric = (core.to_i.to_s == core) || (core.to_f.to_s == core)
76 str.downcase == core.downcase || str.downcase == @string.downcase
88 #######################################################################
90 # One Quiz instance per channel, contains channel specific data
91 #######################################################################
93 attr_accessor :registry, :registry_conf, :questions,
94 :question, :answers, :canonical_answer, :answer_array,
95 :first_try, :hint, :hintrange, :rank_table, :hinted, :has_errors,
98 def initialize( channel, registry )
100 @registry = registry.sub_registry( 'private' )
102 @registry = registry.sub_registry( channel.downcase )
105 @registry.each_key { |k|
106 unless @registry.has_key?(k)
108 error "Data for #{k} is NOT ACCESSIBLE! Database corrupt?"
112 debug @registry.to_a.map { |a| a.join(", ")}.join("\n")
115 @registry_conf = @registry.sub_registry( "config" )
117 # Per-channel list of sources. If empty, the default one (quiz/quiz.rbot)
119 @registry_conf["sources"] = [] unless @registry_conf.has_key?( "sources" )
121 # Per-channel copy of the global questions table. Acts like a shuffled queue
122 # from which questions are taken, until empty. Then we refill it with questions
123 # from the global table.
124 @registry_conf["questions"] = [] unless @registry_conf.has_key?( "questions" )
126 # Autoask defaults to true
127 @registry_conf["autoask"] = true unless @registry_conf.has_key?( "autoask" )
129 # Autoask delay defaults to 0 (instantly)
130 @registry_conf["autoask_delay"] = 0 unless @registry_conf.has_key?( "autoask_delay" )
132 @questions = @registry_conf["questions"]
135 @canonical_answer = nil
144 # True if the answers is entirely done by separators
147 # We keep this array of player stats for performance reasons. It's sorted by score
148 # and always synced with the registry player stats hash. This way we can do fast
149 # rank lookups, without extra sorting.
150 @rank_table = @registry.to_a.sort { |a,b| b[1].score<=>a[1].score }
155 #######################################################################
157 #######################################################################
158 class QuizPlugin < Plugin
159 Config.register Config::BooleanValue.new('quiz.dotted_nicks',
161 :desc => "When true, nicks in the top X scores will be camouflaged to prevent IRC hilighting")
163 Config.register Config::ArrayValue.new('quiz.sources',
164 :default => ['quiz.rbot'],
165 :desc => "List of files and URLs that will be used to retrieve quiz questions")
170 @questions = Array.new
173 @ask_mutex = Mutex.new
176 # Function that returns whether a char is a "separator", used for hints
183 # Fetches questions from the data sources, which can be either local files
184 # (in quiz/) or web pages.
187 # Read the winning messages file
188 @win_messages = Array.new
189 winfile = datafile 'win_messages'
190 if File.exists? winfile
191 IO.foreach(winfile) { |line| @win_messages << line.chomp }
193 warning( "win_messages file not found!" )
194 # Fill the array with a least one message or code accessing it would fail
195 @win_messages << "<who> guessed right! The answer was <answer>"
198 m.reply "Fetching questions ..."
200 # TODO Per-channel sources
203 @bot.config['quiz.sources'].each { |p|
204 if p =~ /^https?:\/\//
207 serverdata = @bot.httputil.get(p) # "http://amarok.kde.org/amarokwiki/index.php/Rbot_Quiz"
208 serverdata = serverdata.split( "QUIZ DATA START\n" )[1]
209 serverdata = serverdata.split( "\nQUIZ DATA END" )[0]
210 serverdata = serverdata.gsub( / /, " " ).gsub( /&/, "&" ).gsub( /"/, "\"" )
211 data << "\n\n" << serverdata
213 m.reply "Failed to download questions from #{p}, ignoring sources"
217 debug "Fetching from #{path}"
221 file = File.new( path, File::RDONLY )
222 data << "\n\n" << file.read
224 m.reply "Failed to read from local database file #{p}, skipping."
231 # Fuse together and remove comments, then split
232 entries = data.strip.gsub( /^#.*$/, "" ).split( /(?:^|\n+)Question: / )
236 # We'll need at least two lines of data
238 # Check if question isn't empty
240 while p[1].match( /^Answer: (.*)$/ ) == nil and p.size > 2
241 # Delete all lines between the question and the answer
244 p[1] = p[1].gsub( /Answer: /, "" ).strip
245 # If the answer was found
247 # Add the data to the array
248 b = QuizBundle.new( p[0], p[1] )
255 m.reply "done, #{@questions.length} questions loaded."
259 # Returns new Quiz instance for channel, or existing one
261 def create_quiz( channel )
262 unless @quizzes.has_key?( channel )
263 @quizzes[channel] = Quiz.new( channel, @registry )
266 if @quizzes[channel].has_errors
269 return @quizzes[channel]
274 def say_score( m, nick )
276 q = create_quiz( chan )
278 m.reply "Sorry, the quiz database for #{chan} seems to be corrupt"
282 if q.registry.has_key?( nick )
283 score = q.registry[nick].score
284 jokers = q.registry[nick].jokers
287 q.rank_table.each_index { |rank| break if nick.downcase == q.rank_table[rank][0].downcase }
290 m.reply "#{nick}'s score is: #{score} Rank: #{rank} Jokers: #{jokers}"
292 m.reply "#{nick} does not have a score yet. Lamer."
297 def help( plugin, topic="" )
299 "Quiz game aministration commands (requires authentication): 'quiz autoask <on/off>' => enable/disable autoask mode. 'quiz autoask delay <secs>' => delay next quiz by <secs> seconds when in autoask mode. 'quiz transfer <source> <dest> [score] [jokers]' => transfer [score] points and [jokers] jokers from <source> to <dest> (default is entire score and all jokers). 'quiz setscore <player> <score>' => set <player>'s score to <score>. 'quiz setjokers <player> <jokers>' => set <player>'s number of jokers to <jokers>. 'quiz deleteplayer <player>' => delete one player from the rank table (only works when score and jokers are set to 0). 'quiz cleanup' => remove players with no points and no jokers."
301 urls = @bot.config['quiz.sources'].select { |p| p =~ /^https?:\/\// }
302 "A multiplayer trivia quiz. 'quiz' => ask a question. 'quiz hint' => get a hint. 'quiz solve' => solve this question. 'quiz skip' => skip to next question. 'quiz joker' => draw a joker to win this round. 'quiz score [player]' => show score for [player] (default is yourself). 'quiz top5' => show top 5 players. 'quiz top <number>' => show top <number> players (max 50). 'quiz stats' => show some statistics. 'quiz fetch' => refetch questions from databases. 'quiz refresh' => refresh the question pool for this channel." + (urls.empty? ? "" : "\nYou can add new questions at #{urls.join(', ')}")
307 # Updates the per-channel rank table, which is kept for performance reasons.
308 # This table contains all players sorted by rank.
310 def calculate_ranks( m, q, nick )
311 if q.registry.has_key?( nick )
312 stats = q.registry[nick]
314 # Find player in table
317 q.rank_table.each_index do |i|
318 if nick.downcase == q.rank_table[i][0].downcase
324 # Remove player from old position
327 q.rank_table.delete_at( i )
332 # Insert player at new position
334 q.rank_table.each_index do |i|
335 if stats.score > q.rank_table[i][1].score
336 q.rank_table[i,0] = [[nick, stats]]
342 # If less than all other players' scores, append to table
344 i += 1 unless q.rank_table.empty?
345 q.rank_table << [nick, stats]
348 # Print congratulations/condolences if the player's rank has changed
351 m.reply "#{nick} ascends to rank #{i + 1}. Congratulations :)"
353 m.reply "#{nick} slides down to rank #{i + 1}. So Sorry! NOT. :p"
357 q.rank_table << [[nick, PlayerStats.new( 1 )]]
362 # Reimplemented from Plugin
366 return unless @quizzes.has_key?( chan )
369 return if q.question == nil
371 message = m.message.downcase.strip
373 nick = m.sourcenick.to_s
375 # Support multiple alternate answers and cores
376 answer = q.answers.find { |ans| ans.valid?(message) }
378 # List canonical answer which the hint was based on, to avoid confusion
379 # FIXME display this more friendly
380 answer.info = " (hints were for alternate answer #{q.canonical_answer.core})" if answer != q.canonical_answer and q.hinted
385 reply = "WHOPEEE! #{nick} got it on the first try! That's worth an extra point. Answer was: #{answer}"
386 elsif q.rank_table.length >= 1 and nick.downcase == q.rank_table[0][0].downcase
387 reply = "THE QUIZ CHAMPION defends his throne! Seems like #{nick} is invicible! Answer was: #{answer}"
388 elsif q.rank_table.length >= 2 and nick.downcase == q.rank_table[1][0].downcase
389 reply = "THE SECOND CHAMPION is on the way up! Hurry up #{nick}, you only need #{q.rank_table[0][1].score - q.rank_table[1][1].score - 1} points to beat the king! Answer was: #{answer}"
390 elsif q.rank_table.length >= 3 and nick.downcase == q.rank_table[2][0].downcase
391 reply = "THE THIRD CHAMPION strikes again! Give it all #{nick}, with #{q.rank_table[1][1].score - q.rank_table[2][1].score - 1} more points you'll reach the 2nd place! Answer was: #{answer}"
393 reply = @win_messages[rand( @win_messages.length )].dup
394 reply.gsub!( "<who>", nick )
395 reply.gsub!( "<answer>", answer )
401 if q.registry.has_key?(nick)
402 player = q.registry[nick]
404 player = PlayerStats.new( 0, 0, 0 )
407 player.score = player.score + points
409 # Reward player with a joker every X points
410 if player.score % 15 == 0 and player.jokers < Max_Jokers
412 m.reply "#{nick} gains a new joker. Rejoice :)"
415 q.registry[nick] = player
416 calculate_ranks( m, q, nick)
419 if q.registry_conf["autoask"]
420 delay = q.registry_conf["autoask_delay"]
422 m.reply "#{Bold}#{Color}03Next question in #{Bold}#{delay}#{Bold} seconds"
423 timer = @bot.timer.add_once(delay) {
424 @ask_mutex.synchronize do
425 @waiting.delete(chan)
429 @waiting[chan] = timer
435 # First try is used, and it wasn't the answer.
441 # Stretches an IRC nick with dots, simply to make the client not trigger a hilight,
442 # which is annoying for those not watching. Example: markey -> m.a.r.k.e.y
444 def unhilight_nick( nick )
445 return nick unless @bot.config['quiz.dotted_nicks']
446 return nick.split(//).join(".")
450 #######################################################################
452 #######################################################################
453 def cmd_quiz( m, params )
454 fetch_data( m ) if @questions.empty?
457 @ask_mutex.synchronize do
458 if @waiting.has_key?(chan)
459 m.reply "Next quiz question will be automatically asked soon, have patience"
464 q = create_quiz( chan )
466 m.reply "Sorry, the quiz database for #{chan} seems to be corrupt"
471 m.reply "#{Bold}#{Color}03Current question: #{Color}#{Bold}#{q.question}"
472 m.reply "Hint: #{q.hint}" if q.hinted
476 # Fill per-channel questions buffer
477 if q.questions.empty?
478 q.questions = @questions.sort_by { rand }
481 # pick a question and delete it (delete_at returns the deleted item)
482 picked = q.questions.delete_at( rand(q.questions.length) )
484 q.question = picked.question
485 q.answers = picked.answer.split(/\s+\|\|\s+/).map { |ans| QuizAnswer.new(ans) }
487 # Check if any core answer is numerical and tell the players so, if that's the case
488 # The rather obscure statement is needed because to_i and to_f returns 99(.0) for "99 red balloons", and 0 for "balloon"
490 # The "canonical answer" is also determined here, defined to be the first found numerical answer, or
492 numeric = q.answers.find { |ans| ans.numeric? }
494 q.question += "#{Color}07 (Numerical answer)#{Color}"
495 q.canonical_answer = numeric
497 q.canonical_answer = q.answers.first
505 q.canonical_answer.core.scan(/./u) { |ch|
514 # It's possible that an answer is entirely done by separators,
515 # in which case we'll hide everything
516 if q.answer_array == q.hint
524 # Generate array of unique random range
525 q.hintrange = (0..q.hint.length-1).sort_by{ rand }
527 m.reply "#{Bold}#{Color}03Question: #{Color}#{Bold}" + q.question
531 def cmd_solve( m, params )
534 return unless @quizzes.has_key?( chan )
537 m.reply "The correct answer was: #{q.canonical_answer}"
541 cmd_quiz( m, nil ) if q.registry_conf["autoask"]
545 def cmd_hint( m, params )
547 nick = m.sourcenick.to_s
549 return unless @quizzes.has_key?(chan)
553 m.reply "#{nick}: Get a question first!"
555 num_chars = case q.hintrange.length # Number of characters to reveal
568 index = q.hintrange.pop
569 # New hint char until the char isn't a "separator" (space etc.)
570 end while is_sep(q.answer_array[index]) and not q.all_seps
571 q.hint[index] = q.answer_array[index]
573 m.reply "Hint: #{q.hint}"
577 if q.hint == q.answer_array
578 m.reply "#{Bold}#{Color}04BUST!#{Color}#{Bold} This round is over. #{Color}04Minus one point for #{nick}#{Color}."
581 if q.registry.has_key?( nick )
582 stats = q.registry[nick]
584 stats = PlayerStats.new( 0, 0, 0 )
587 stats["score"] = stats.score - 1
588 q.registry[nick] = stats
590 calculate_ranks( m, q, nick)
593 cmd_quiz( m, nil ) if q.registry_conf["autoask"]
599 def cmd_skip( m, params )
601 return unless @quizzes.has_key?(chan)
605 cmd_quiz( m, params )
609 def cmd_joker( m, params )
611 nick = m.sourcenick.to_s
612 q = create_quiz(chan)
614 m.reply "Sorry, the quiz database for #{chan} seems to be corrupt"
619 m.reply "#{nick}: There is no open question."
623 if q.registry[nick].jokers > 0
624 player = q.registry[nick]
627 q.registry[nick] = player
629 calculate_ranks( m, q, nick )
631 if player.jokers != 1
636 m.reply "#{Bold}#{Color}12JOKER!#{Color}#{Bold} #{nick} draws a joker and wins this round. You have #{player.jokers} #{jokers} left."
637 m.reply "The answer was: #{q.canonical_answer}."
640 cmd_quiz( m, nil ) if q.registry_conf["autoask"]
642 m.reply "#{nick}: You don't have any jokers left ;("
647 def cmd_fetch( m, params )
652 def cmd_refresh( m, params )
653 q = create_quiz(m.channel)
656 cmd_quiz( m, params )
660 def cmd_top5( m, params )
662 q = create_quiz( chan )
664 m.reply "Sorry, the quiz database for #{chan} seems to be corrupt"
668 if q.rank_table.empty?
669 m.reply "There are no scores known yet!"
673 m.reply "* Top 5 Players for #{chan}:"
675 [5, q.rank_table.length].min.times do |i|
676 player = q.rank_table[i]
678 score = player[1].score
679 m.reply " #{i + 1}. #{unhilight_nick( nick )} (#{score})"
684 def cmd_top_number( m, params )
685 num = params[:number].to_i
686 return if num < 1 or num > 50
688 q = create_quiz( chan )
690 m.reply "Sorry, the quiz database for #{chan} seems to be corrupt"
694 if q.rank_table.empty?
695 m.reply "There are no scores known yet!"
700 m.reply "* Top #{num} Players for #{chan}:"
701 n = [ num, q.rank_table.length ].min
703 player = q.rank_table[i]
705 score = player[1].score
706 ar << "#{i + 1}. #{unhilight_nick( nick )} (#{score})"
708 m.reply ar.join(" | "), :split_at => /\s+\|\s+/
712 def cmd_stats( m, params )
713 fetch_data( m ) if @questions.empty?
715 m.reply "* Total Number of Questions:"
716 m.reply " #{@questions.length}"
720 def cmd_score( m, params )
721 nick = m.sourcenick.to_s
726 def cmd_score_player( m, params )
727 say_score( m, params[:player] )
731 def cmd_autoask( m, params )
733 q = create_quiz( chan )
735 m.reply "Sorry, the quiz database for #{chan} seems to be corrupt"
739 case params[:enable].downcase
741 q.registry_conf["autoask"] = true
742 m.reply "Enabled autoask mode."
743 cmd_quiz( m, nil ) if q.question == nil
745 q.registry_conf["autoask"] = false
746 m.reply "Disabled autoask mode."
748 m.reply "Invalid autoask parameter. Use 'on' or 'off'."
752 def cmd_autoask_delay( m, params )
754 q = create_quiz( chan )
756 m.reply "Sorry, the quiz database for #{chan} seems to be corrupt"
760 delay = params[:time].to_i
761 q.registry_conf["autoask_delay"] = delay
762 m.reply "Autoask delay now #{q.registry_conf['autoask_delay']} seconds"
766 def cmd_transfer( m, params )
768 q = create_quiz( chan )
770 m.reply "Sorry, the quiz database for #{chan} seems to be corrupt"
774 debug q.rank_table.inspect
776 source = params[:source]
778 transscore = params[:score].to_i
779 transjokers = params[:jokers].to_i
780 debug "Transferring #{transscore} points and #{transjokers} jokers from #{source} to #{dest}"
782 if q.registry.has_key?(source)
783 sourceplayer = q.registry[source]
784 score = sourceplayer.score
788 if score < transscore
789 m.reply "#{source} only has #{score} points!"
792 jokers = sourceplayer.jokers
796 if jokers < transjokers
797 m.reply "#{source} only has #{jokers} jokers!!"
800 if q.registry.has_key?(dest)
801 destplayer = q.registry[dest]
803 destplayer = PlayerStats.new(0,0,0)
806 if sourceplayer.object_id == destplayer.object_id
807 m.reply "Source and destination are the same, I'm not going to touch them"
811 sourceplayer.score -= transscore
812 destplayer.score += transscore
813 sourceplayer.jokers -= transjokers
814 destplayer.jokers += transjokers
816 q.registry[source] = sourceplayer
817 calculate_ranks(m, q, source)
819 q.registry[dest] = destplayer
820 calculate_ranks(m, q, dest)
822 m.reply "Transferred #{transscore} points and #{transjokers} jokers from #{source} to #{dest}"
824 m.reply "#{source} doesn't have any points!"
829 def cmd_del_player( m, params )
831 q = create_quiz( chan )
833 m.reply "Sorry, the quiz database for #{chan} seems to be corrupt"
837 debug q.rank_table.inspect
840 if q.registry.has_key?(nick)
841 player = q.registry[nick]
844 m.reply "Can't delete player #{nick} with score #{score}."
847 jokers = player.jokers
849 m.reply "Can't delete player #{nick} with #{jokers} jokers."
852 q.registry.delete(nick)
855 q.rank_table.each_index { |rank|
856 if nick.downcase == q.rank_table[rank][0].downcase
861 q.rank_table.delete_at(player_rank)
863 m.reply "Player #{nick} deleted."
865 m.reply "Player #{nick} isn't even in the database."
870 def cmd_set_score(m, params)
872 q = create_quiz( chan )
874 m.reply "Sorry, the quiz database for #{chan} seems to be corrupt"
877 debug q.rank_table.inspect
880 val = params[:score].to_i
881 if q.registry.has_key?(nick)
882 player = q.registry[nick]
885 player = PlayerStats.new( val, 0, 0)
887 q.registry[nick] = player
888 calculate_ranks(m, q, nick)
889 m.reply "Score for player #{nick} set to #{val}."
893 def cmd_set_jokers(m, params)
895 q = create_quiz( chan )
897 m.reply "Sorry, the quiz database for #{chan} seems to be corrupt"
900 debug q.rank_table.inspect
903 val = [params[:jokers].to_i, Max_Jokers].min
904 if q.registry.has_key?(nick)
905 player = q.registry[nick]
908 player = PlayerStats.new( 0, val, 0)
910 q.registry[nick] = player
911 m.reply "Jokers for player #{nick} set to #{val}."
915 def cmd_cleanup(m, params)
917 q = create_quiz( chan )
919 m.reply "Sorry, the quiz database for #{chan} seems to be corrupt"
924 q.registry.each { |nick, player|
925 null_players << nick if player.jokers == 0 and player.score == 0
927 debug "Cleaning up by removing #{null_players * ', '}"
928 null_players.each { |nick|
929 cmd_del_player(m, :nick => nick)
936 m.reply 'you must be on some channel to use this command'
939 if @quizzes.delete m.channel
940 @ask_mutex.synchronize do
941 t = @waiting.delete(m.channel)
942 @bot.timer.remove t if t
946 m.reply(_("there is no active quiz on #{m.channel}"))
952 plugin = QuizPlugin.new
953 plugin.default_auth( 'edit', false )
956 plugin.map 'quiz', :action => 'cmd_quiz'
957 plugin.map 'quiz solve', :action => 'cmd_solve'
958 plugin.map 'quiz hint', :action => 'cmd_hint'
959 plugin.map 'quiz skip', :action => 'cmd_skip'
960 plugin.map 'quiz joker', :action => 'cmd_joker'
961 plugin.map 'quiz score', :action => 'cmd_score'
962 plugin.map 'quiz score :player', :action => 'cmd_score_player'
963 plugin.map 'quiz fetch', :action => 'cmd_fetch'
964 plugin.map 'quiz refresh', :action => 'cmd_refresh'
965 plugin.map 'quiz top5', :action => 'cmd_top5'
966 plugin.map 'quiz top :number', :action => 'cmd_top_number'
967 plugin.map 'quiz stats', :action => 'cmd_stats'
968 plugin.map 'quiz stop', :action => :stop
971 plugin.map 'quiz autoask :enable', :action => 'cmd_autoask', :auth_path => 'edit'
972 plugin.map 'quiz autoask delay :time', :action => 'cmd_autoask_delay', :auth_path => 'edit', :requirements => {:time => /\d+/}
973 plugin.map 'quiz transfer :source :dest :score :jokers', :action => 'cmd_transfer', :auth_path => 'edit', :defaults => {:score => '-1', :jokers => '-1'}
974 plugin.map 'quiz deleteplayer :nick', :action => 'cmd_del_player', :auth_path => 'edit'
975 plugin.map 'quiz setscore :nick :score', :action => 'cmd_set_score', :auth_path => 'edit'
976 plugin.map 'quiz setjokers :nick :jokers', :action => 'cmd_set_jokers', :auth_path => 'edit'
977 plugin.map 'quiz cleanup', :action => 'cmd_cleanup', :auth_path => 'edit'