# Wrapper for MegaHAL, adding user commands to modify the score and volume, # and provide help. # This class can function both as a command-line program and # as an IRC bot when it is loaded by the RedTurnip IRC bot framework. begin require 'megahal' # for when we start from the command line rescue LoadError require 'mod/bot/megahal' # for when we start from RedTurnip end class MegaHALBot def initialize print "Hello\n" @agent = MegaHAL.new @THRESHHOLD = 1 @SCORE_MOD = 0 @MAX_RESPONSE = 10000 @MY_NAME = "(?i)(?:megahal|megahalbot|megahalagent)" @rules = [] @SAFE_LEVEL = "low" @Low_safe_level_words = ["low", "default", "0", "off"] @High_safe_level_words = ["high", "on", "1"] end def getResponse(input) input = input.to_s #print 'INPUT==' + input + "\n" # Set threshhold. if input =~ /^what is your name?/ name = @MY_NAME name.gsub!(/\(\?i\)\(\?\:/, '') name.gsub!(/\|/, " or ") name.gsub!(/\)/, '') return name end if input =~ /^#{@MY_NAME}[:,!]* (quiet|be quiet|stfu|shut.*up)/ @THRESHHOLD = @THRESHHOLD + 1 @SCORE_MOD -= 1 return "My threshhold is now " << @THRESHHOLD.to_s << ", and my score_mod is #{@SCORE_MOD}." end if input =~ /^#{@MY_NAME}[:,!]* (speak.*up|loud|louder|i can't hear you)/ @THRESHHOLD = @THRESHHOLD - 1 @SCORE_MOD += 1 return "My threshhold is now " << @THRESHHOLD.to_s << ", and my score_mod is #{@SCORE_MOD}." end # Return threshhold if input =~ /^#{@MY_NAME}[:,!]* what is your (threshhold|threshold|volume)/ or input =~ /^#{@MY_NAME}[:,!]* (?:threshold|threshhold|volume)/ return "My threshhold is currently " << @THRESHHOLD.to_s end # Provide help message. if input =~ /^#{@MY_NAME}[:,!]* help/ return "I am a conversational simulator, by Jason Hutchens." end # trap restart case input when /^#{@MY_NAME}[:,!]* restart/, /^restart/ return "That command is disabled." # for safety end # add new if-then rules to post-process responses case input when /^#{@MY_NAME}[:,!]* (if .*? =~ .*? then .*)$/ return add_if_rule($1) end # show the rules case input when /^#{@MY_NAME}[:,!]*.* show.* rules/ return show_rules[0] end # delete a rule case input when /^#{@MY_NAME}[:,!]*.* delete.* rule.* (\d+)/ return delete_rule($1.to_i) end # set safe level case input when /^#{@MY_NAME}[:,! ]*(?:set )?.*safe(?:ty)? level (?:to )?(.*)/ return set_safe_level($1) end # get safe level case input when /^#{@MY_NAME}[:,! ]*what is.*?safe(?:ty)? level/ return get_safe_level end # Get the agent's response. if input =~ /^#{@MY_NAME}[:,! ]*(.*)/ then input = $1 end r, s = @agent.getResponse(input) if s == nil then s = rand(3) + 1 end if s > 3 then s -= 1 end r,s = apply_if_rules(r,s,input) s += @SCORE_MOD if r.to_s.size > @MAX_RESPONSE r = r[0..@MAX_RESPONSE] end puts "\n[megahalbot] score==" << s.to_s << "; threshhold==" << @THRESHHOLD.to_s if s > @THRESHHOLD if $0 == __FILE__ then return postprocess(r) end return [postprocess(r), s] else print 'r==' + postprocess(r) end return "" end def postprocess input r = input return r end # Respond to natural language method calls. def method_missing input return getResponse(input.to_s) end # add an if-then rule def add_if_rule(rule = '') if rule == '' then return "No rule found." end if @SAFE_LEVEL =~ /#{@Low_safe_level_words.join('|')}$/ @rules.push(rule) return "Okay I have added #{rule}." end return "I can't add any if rules because my safe level is #{@SAFE_LEVEL}." end # apply if-then rules def apply_if_rules(response, score, input) @rules.each { |rule| puts "rule==#{rule}; response==#{response}; score==#{score}" begin eval(rule) rescue; puts "Can't apply rule: #{rule}"; next; end } #puts "returning: response==#{response}; score==#{score}" return [response, score] end # return the rules def show_rules return [@rules.join("\n"), 4] end # delete a rule by its index def delete_rule(i) if i < 1 or i > @rules.size return "The rules go from 1 to #{@rules.size}." end rule = @rules[i-1] @rules.delete_at(i-1) return "Okay, \"#{rule}\" deleted." end def set_safe_level(level) case level when /\b#{@Low_safe_level_words.join('|')}\b/ @SAFE_LEVEL = "off" return "Okay I have turned off the safe level." when /\b#{@High_safe_level_words.join('|')}\b/ @SAFE_LEVEL = "on" return "Okay I have turned the safe level on." end return "I don't understand #{level}." end def get_safe_level return "My safe level is set to #{@SAFE_LEVEL}." end # method_def end if $0 == __FILE__ b = MegaHALBot.new print("\n> "); $stdout.flush quitwords = [":q", "quit", "exit", "bye"] while (line = gets) !~ /^#{quitwords.join('|')}$/ and line !~ /^$/ print "\n" + b.send(line) + "\n" print("\n> "); $stdout.flush end print "Bye!" end