# Wrapper for ISRAgent, adding user commands to modify the score, etc. 
# 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 'isragent' # for when we start from the command line
rescue LoadError
   require 'mod/bot/isragent' # for when we start from RedTurnip 
end

class ISRBot 

   def initialize
      print "Hello\n"
      @agent = MyAgent.new
      @THRESHHOLD = 1
      @SCORE_MOD = 0
      @MAX_RESPONSE = 10000
      @MY_NAME = "(?i)(?:isragentbot|isrbot|isragent|srbot|sragent|isr|sr)"
      @rules = []
      @rules.push("if response =~ /(?:I can't|I don't)/; response = \"That never happens in Soviet Russia\"; score = 1; end")
      @rules.push("if response =~ / {2,}s YOU!/; response = \"That doesn\'t happen in Soviet Russia.\"; score = 1; end")
      @rules.push("if response =~ /^  s YOU/; response = \"That doesn\'t happen in Soviet Russia.\"; score = 1; end")
      @rules.push("if response =~ /default response/i; response = \"That doesn\'t happen in Soviet Russia.\"; score = 1; end")
      @rules.push("if response =~ /ss YOU!/ then response.gsub!(/ss YOU!/, 's YOU!') end")
      @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"

      if input =~ /^what is your name?/
        my_name = @MY_NAME.to_s
        my_name.gsub!(/\(\?i\)/, '')
        my_name.gsub!(/\(\?\:\)/, '')
        my_name.gsub!(/\|/, ' or ')
        my_name.gsub!(/\)/, '') 
        return my_name
      end
      # Set threshhold.
      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 make In Soviet Russia jokes. !sr <sentence>"
      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}[:,! ]*.*turn (.*) (?:the )safe(?:ty)? level/,
             /^#{@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 = 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[isrbot] score==" << s.to_s << "; threshhold==" << @THRESHHOLD.to_s
      if s > @THRESHHOLD 
        if $0 == __FILE__ then return r end
        return [r, s]
      else
         print 'r==' + r
      end

      return ""
   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 SyntaxError; 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

   # Respond to natural language method calls.
   def method_missing input
     return getResponse(input.to_s)
   end

end


class PostProc
   private_class_method :new
   @@postproc = nil
   @@count = 0
   
   def PostProc.create
      @@postproc = new unless @@postproc
      @@postproc
   end

   def process(input)
      response = input.to_s 
      self.methods.each { |method|
         if method =~ /^newMethod/
            begin
               response = self.send(method, input)
            rescue NoMethodError, SyntaxError => boom
               return "Eeep! Error!: " + boom
            end
         end
      }
      return response
   end

   def getNewName
      @@count = @@count + 1
      return "newMethod" << @@count.to_s
   end

   def create_method(name, &block)
      self.class.send(:define_method, name, &block)
   end

end

if $0 == __FILE__
  b = ISRBot.new
  print("\n> "); $stdout.flush
  quitwords = [":q", "quit", "exit", "bye"]
  while (line = gets) !~ /^#{quitwords.join('|')}$/ and line !~ /^$/
    print "\n" + b.send(line).to_s + "\n"
    print("\n> "); $stdout.flush
  end
  print "Bye!"
end