################################################################################ # montybot provides an interface to MontyLingua. # # Put this file in the MontyLingua directory, with the following two files: # subbot.org/montylingua/montybot-api.yaml # subbot.org/montylingua/MontyLingua2.py # # > subject: "John ate an apple." # John # # > what is the object of: "John loves Mary." # Mary # ################################################################################ require 'yaml' require 'montylingua' class MyAgent def initialize(args=Array.new) @workingDir = File.expand_path(File.dirname(__FILE__)) @filename = @workingDir + "/montybot-api.yaml" unless load_message = load_synonyms @patterns = [ [/^(.*?) is a synonym for (.*)/, :add_synonym], [/^save syn/, :save_synonyms], [/^load syn/, :load_synonyms], [/^add pattern (.*), (.*)/, :add_pattern], [/^>(.*)/, :defining_method], [/^restart/, :restart], [/^begin log (.*)/, :begin_log], [/^end log (.*)/, :end_log] ] save_synonyms # creates patterns.yaml on first use. else puts load_message end @method_def = '' @monty = MontyLingua.new end def def_method method_def file = IO.read(__FILE__) return "Doesn't have 'end # class'" unless file =~ /(.*)end # class(.*)if \$0 == __FILE__(.*)/m grp1 = $1.to_s grp2 = $2.to_s grp3 = $3.to_s new_filename = get_new_filename File.open(new_filename, "w+") { |f| f.puts grp1 + method_def + "\n\nend # class\n\nif $0 == __FILE__" + grp3 } @patterns.push( [@new_regexp, @new_symbol] ) save_synonyms MyAgent.class_eval(method_def) end # If this file has a '.' + a number before the '.rb' extension # (i.e. foo.0.rb), increment the number (i.e. return foo.1.rb). # Otherwise return the name of this file unchanged. def get_new_filename old_filename = __FILE__ if old_filename =~ /(.*)\.(\d+)\.rb$/ incr = $2.to_i + 1 return new_filename = $1.to_s + "." + incr.to_s + ".rb" end return old_filename # Dangerous! end # Match input against each regexp pattern in @patterns; # if there's a match, call the method associated with the pattern. def getResponse input r = "Default response." puts "[montybot] input==#{input}" @patterns.each { |pattern| if input =~ pattern[0] puts "[montybot] responding at: #{Time.now}" return self.send(pattern[1], pattern[0], input) rescue return r end } return r end def load_synonyms(pattern=nil, input=nil) if File.exist?(@filename) @patterns = YAML.load_file(@filename) return "I have loaded #{@filename}." end return false end def save_synonyms(pattern=nil, input=nil) r = "Eep! error saving synonyms! " begin File.open(@filename, 'w') do |out| YAML.dump(@patterns, out) end return "Okay I have saved the synonyms to disk." rescue => error return r << error end end def add_synonym(pattern, input) return "Problem matching pattern." unless input =~ pattern #grp1 = %r{^#{$1}} grp1 = %r{(?i-mx:^#{$1})} #grp2 = %r{^#{$2}} grp2 = %r{(?i-mx:^#{$2})} method ='' @patterns.each { |p| #puts "p[0]==#{p[0]}; p[0].class==#{p[0].class}; grp2==#{grp2}; grp2.class==#{grp2.class}" if p[0].to_s == grp2.to_s method = p[1] break end } return unless method.to_s != nil @patterns.push( [grp1, method] ) save_synonyms return "Okay, #{grp1} has been added, and will now call #{method}." end def add_pattern(pattern=nil, input=nil) if input =~ pattern @new_regexp = $1 @new_symbol = $2 @new_regexp.gsub!(/\//, '') new_re = %r{#{@new_regexp}} # Make the regexp case insensitive. new_re_s = new_re.to_s if new_re_s =~ /(-mix)/ new_re_s = $~.pre_match << "i-mx" << $~.post_match end @new_regexp = %r{#{new_re_s}} #puts "@new_regexp==#{@new_regexp}" method_def = " def #{@new_symbol}(pattern=nil, input=nil)\n return \"#\{input} doesn't match #\{pattern}\" unless input =~ pattern\n return 'Default response from " + @new_symbol + "'\n end # method_def" return def_method(method_def) else return "#{input} doesn't match #{pattern}." end end def defining_method(pattern=nil, input=nil) if input =~ pattern @method_def << "\n " << $1.to_s if @method_def =~ /end # method_def/ return def_method(@method_def) end return '' else return "#{input} doesn't match #{pattern}." end end def method_missing input, *rest return getResponse(input.to_s) end # Strips delimiters. def process str str.strip! str.sub!(/^\[/, '') str.sub!(/\]$/, '') str.sub!(/^(?:"|')/, '') str.sub!(/(?:"|')$/, '') str.sub!(/\.$/, '') str.sub!(/;$/, '') str.sub!(/,$/, '') str.sub!(/!$/, '') str.sub!(/\?$/, '') return str end def restart(pattern=nil,input=nil) exec "ruby #{__FILE__}" end # method_def def begin_log(pattern=nil, input=nil) return "#{input} doesn't match #{pattern}" unless input =~ pattern file = process $1 file = "log.txt" unless file != nil and file != '' $logger = File.open(file, "w") t = Time.now; y = t.year; m = t.month; d = t.day; h= t.hour; min = t.min; s = t.sec mins = sprintf("%02d", min) secs = sprintf("%02d", s) time = "#{y}/#{m}/#{d} #{h}:#{mins}:#{secs}" return "Okay I have started logging to #{file} at #{time}" end # method_def def end_log(pattern=nil, input=nil) return "#{input} doesn't match #{pattern}" unless input =~ pattern $logger.close $logger == nil t = Time.now; y = t.year; m = t.month; d = t.day; h= t.hour; min = t.min; s = t.sec mins = sprintf("%02d", min) secs = sprintf("%02d", s) time = "#{y}/#{m}/#{d} #{h}:#{mins}:#{secs}" return "Okay I have stopped logging at #{time}." end # method_def def getSubject(pattern=nil, input=nil) return "#{input} doesn't match #{pattern}" unless input =~ pattern grp1 = process $1 return @monty.getSubject(grp1) end # method_def def getObject(pattern=nil, input=nil) return "#{input} doesn't match #{pattern}" unless input =~ pattern grp1 = process $1 return @monty.getObject(grp1) end # method_def def getVerb(pattern=nil, input=nil) return "#{input} doesn't match #{pattern}" unless input =~ pattern grp1 = process $1 return @monty.getVerb(grp1) end # method_def def help(pattern=nil, input=nil) return ["I provide Montylingua parsing services.", 4] end end # class if $0 == __FILE__ bot = MyAgent.new print "\n> "; $stdout.flush quitwords = [":q", "quit", "exit", "bye"] while (line = gets) !~ /^#{quitwords.join('|')}/i and line !~ /^$/ begin; $logger.puts("> #{line}"); rescue; end response = bot.send(line.to_s).to_s + "\n\n" print response begin; $logger.puts(response); rescue; end print "> "; $stdout.flush end puts "Bye" end # of file