diff options
Diffstat (limited to 'data')
| -rw-r--r-- | data/rbot/plugins/salut.rb | 174 | ||||
| -rw-r--r-- | data/rbot/templates/salut/salut-english | 33 | ||||
| -rw-r--r-- | data/rbot/templates/salut/salut-italian | 19 |
3 files changed, 226 insertions, 0 deletions
diff --git a/data/rbot/plugins/salut.rb b/data/rbot/plugins/salut.rb new file mode 100644 index 00000000..1a301b43 --- /dev/null +++ b/data/rbot/plugins/salut.rb @@ -0,0 +1,174 @@ +# vim: set sw=2 et:
+# Salutations plugin: respond to salutations
+
+unless Array.respond_to?(:pick_one)
+ debug "Defining the pick_one method for Array"
+ class ::Array
+ def pick_one
+ return nil if self.empty?
+ self[rand(self.length)]
+ end
+ end
+end
+
+
+class SalutPlugin < Plugin
+ BotConfig.register BotConfigBooleanValue.new('salut.all_languages',
+ :default => true,
+ :desc => "Check for a salutation in all languages and not just in the one defined by core.language",
+ :on_change => Proc.new {|bot, v| bot.plugins['salut'].reload}
+ )
+
+ def initialize
+ @salutations = Hash.new
+ @match = nil
+ @main_lang_str = nil
+ @main_lang = nil
+ @all_langs = true
+ super
+ reload
+ end
+
+ def set_language(what)
+ reload
+ end
+
+ def create_match
+ @match = Hash.new
+ ar_in = Array.new
+ ar_out = Array.new
+ ar_both = Array.new
+ @salutations.each { |lang, hash|
+ hash.each { |situation, array|
+ case situation.to_s
+ when /in$/
+ ar_in += array
+ when /out$/
+ ar_out += array
+ else
+ ar_both += array
+ end
+ }
+ }
+ @match[:in] = Regexp.new("\\b(?:" + ar_in.uniq.map { |txt|
+ Regexp.escape(txt)
+ }.join('|') + ")\\b", Regexp::IGNORECASE) unless ar_in.empty?
+ @match[:out] = Regexp.new("\\b(?:" + ar_out.uniq.map { |txt|
+ Regexp.escape(txt)
+ }.join('|') + ")\\b", Regexp::IGNORECASE) unless ar_out.empty?
+ @match[:both] = Regexp.new("\\b(?:" + ar_both.uniq.map { |txt|
+ Regexp.escape(txt)
+ }.join('|') + ")\\b", Regexp::IGNORECASE) unless ar_both.empty?
+ debug "Matches: #{@match.inspect}"
+ end
+
+ def listen(m)
+ salut = nil
+ [:both, :in, :out].each { |k|
+ next unless @match[k]
+ debug "Checking salutations #{k} (#{@match[k].inspect})"
+ if m.message =~ /^#{@match[k]}/
+ salut = k
+ break
+ end
+ }
+ return unless salut
+ h = Time.new.hour
+ case h
+ when 4...12
+ salut_reply(m, salut, :morning)
+ when 12...18
+ salut_reply(m, salut, :afternoon)
+ else
+ salut_reply(m, salut, :evening)
+ end
+ end
+
+ def salut_reply(m, k, time)
+ debug "Replying to #{k} in the #{time}"
+ case k
+ when :both
+ sfx = ""
+ else
+ sfx = "-#{k}"
+ end
+ debug "Building array ..."
+ rep_ar = Array.new
+ rep_ar += @salutations[@main_lang].fetch("#{time}#{sfx}".to_sym, [])
+ rep_ar += @salutations[@main_lang].fetch("#{time}".to_sym, []) unless sfx.empty?
+ rep_ar += @salutations[@main_lang].fetch("generic#{sfx}".to_sym, [])
+ rep_ar += @salutations[@main_lang].fetch("generic".to_sym, []) unless sfx.empty?
+ debug "Choosing reply in #{rep_ar.inspect} ..."
+ if rep_ar.empty?
+ if m.public? and (m.address? or m =~ /#{Regexp.escape(@bot.nick)}/)
+ choice = @bot.lang.get("hello_X") % m.sourcenick
+ else
+ choice = @bot.lang.get("hello") % m.sourcenick
+ end
+ else
+ choice = rep_ar.pick_one
+ if m.public? and (m.address? or m.message =~ /#{Regexp.escape(@bot.nick)}/)
+ choice += "#{[',',''].pick_one} #{m.sourcenick}"
+ choice += [" :)", " :D", "!", "", "", ""].pick_one
+ end
+ end
+ debug "Replying #{choice}"
+ m.plainreply choice
+ end
+
+ def reload
+ save
+ @main_lang_str = @bot.config['core.language']
+ @main_lang = @main_lang_str.to_sym
+ @all_langs = @bot.config['salut.all_languages']
+ if @all_langs
+ # Get all available languages
+ langs = Dir.new("#{@bot.botclass}/salut").collect {|f|
+ f =~ /salut-([^.]+)/ ? $1 : nil
+ }.compact
+ langs.each { |lang|
+ @salutations[lang.to_sym] = load_lang(lang)
+ }
+ else
+ @salutations.clear
+ @salutations[@main_lang] = load_lang(@main_lang_str)
+ end
+ create_match
+ end
+
+ def load_lang(lang)
+ file = "#{@bot.botclass}/salut/salut-#{lang}"
+ if File.exist?(file)
+ begin
+ salutations = Hash.new
+ content = YAML::load_file(file)
+ content.each { |key, val|
+ salutations[key.to_sym] = val
+ }
+ return salutations
+ rescue
+ error "failed to read salutations in #{lang}: #{$!}"
+ end
+ end
+ return nil
+ end
+
+ def save
+ return if @salutations.empty?
+ @salutations.each { |lang, val|
+ l = lang.to_s
+ save_lang(lang, val)
+ }
+ end
+
+ def save_lang(lang, val)
+ file = "#{@bot.botclass}/salut/salut-#{lang}"
+ Utils.safe_save(file) { |file|
+ file.puts val.to_yaml
+ }
+ end
+
+end
+
+plugin = SalutPlugin.new
+
diff --git a/data/rbot/templates/salut/salut-english b/data/rbot/templates/salut/salut-english new file mode 100644 index 00000000..f1c1dc43 --- /dev/null +++ b/data/rbot/templates/salut/salut-english @@ -0,0 +1,33 @@ +--- +:"generic-in": +- hello +- howdy +- hola +- salut +- sup +- re +- hey +- hi +- yo +:"morning-in": +- good morning +- goodmorning +- morning +- "'morgning" +:"afternoon-in": +- good afternoon +- goodafternoon +- "'afternoon" +:"generic-out": +- bye +- cya +- see you +- see ya +- g'bye +- goodbye +- later +:"evening-in": +- good evening +- goodevening +- "'evening" +- evening diff --git a/data/rbot/templates/salut/salut-italian b/data/rbot/templates/salut/salut-italian new file mode 100644 index 00000000..68632f03 --- /dev/null +++ b/data/rbot/templates/salut/salut-italian @@ -0,0 +1,19 @@ +--- +:"generic-in": +- salve +- ehila' +- ola' +- yo +:"morning-in": +- buongiorno +:"afternoon-in": +- buonasera +:"generic-out": +- a dopo +- ci vediamo +- adios +- se vedemio +:"evening-in": +- buonasera +:"generic": +- ciao |