lib/rbot/language.rb

   1 #-- vim:sw=2:et
   2 #++
   3 #
   4 # :title: Language module for rbot
   5 #
   6 # This module takes care of language handling for rbot:
   7 # setting the core.language value, loading the appropriate
   8 # .lang file etc.
   9
  10 module Irc
  11 class Bot
  12   class Language
  13
  14     # This constant hash holds the mapping
  15     # from long language names to the usual POSIX
  16     # locale specifications
  17     Lang2Locale = {
  18       :english  => 'en',
  19       :british_english  => 'en_GB',
  20       :american_english  => 'en_US',
  21       :italian  => 'it',
  22       :french   => 'fr',
  23       :german   => 'de',
  24       :dutch    => 'nl',
  25       :japanese => 'ja',
  26       :russian  => 'ru',
  27       :traditional_chinese => 'zh_TW',
  28       :simplified_chinese => 'zh_CN'
  29     }
  30
  31     # Return the shortest language for the current
  32     # GetText locale
  33     def Language.from_locale
  34       return 'english' unless defined?(GetText)
  35       lang = locale.language
  36       if locale.country
  37         str = lang + "_#{locale.country}"
  38         if Lang2Locale.value?(str)
  39           # Get the shortest key in Lang2Locale which maps to the given lang_country
  40           lang_str = Lang2Locale.select { |k, v| v == str }.transpose.first.map { |v| v.to_s }.sort { |a, b| a.length <=> b.length }.first
  41           if File.exist?(File.join(Config::datadir, "languages/#{lang_str}.lang"))
  42             return lang_str
  43           end
  44         end
  45       end
  46       # lang_country didn't work, let's try lan
  47       if Lang2Locale.value?(lang)
  48         # Get the shortest key in Lang2Locale which maps to the given lang
  49         lang_str = Lang2Locale.select { |k, v| v == lang }.transpose.first.map { |v| v.to_s }.sort { |a, b| a.length <=> b.length }.first
  50         if File.exist?(File.join(Config::datadir, "/languages/#{lang_str}.lang"))
  51           return lang_str
  52         end
  53       end
  54       # all else fail, return 'english'
  55       return 'english'
  56     end
  57
  58     Config.register Config::EnumValue.new('core.language',
  59       :default => Irc::Bot::Language.from_locale, :wizard => true,
  60       :values => Proc.new{|bot|
  61             Dir.new(Config::datadir + "/languages").collect {|f|
  62               f =~ /\.lang$/ ? f.gsub(/\.lang$/, "") : nil
  63             }.compact
  64           },
  65       :on_change => Proc.new {|bot, v| bot.lang.set_language v},
  66       :desc => "Which language file the bot should use")
  67
  68     def initialize(bot, language)
  69       @bot = bot
  70       set_language language
  71     end
  72     attr_reader :language
  73
  74     def set_language(language)
  75       lang_str = language.to_s.downcase.gsub(/\s+/,'_')
  76       lang_sym = lang_str.intern
  77       if defined?(GetText) and Lang2Locale.key?(lang_sym)
  78         setlocale(Lang2Locale[lang_sym])
  79         debug "locale set to #{locale}"
  80         rbot_gettext_debug
  81       else
  82         warning "Unable to set locale, unknown language #{language} (#{lang_str})"
  83       end
  84
  85       file = Config::datadir + "/languages/#{lang_str}.lang"
  86       unless(FileTest.exist?(file))
  87         raise "no such language: #{lang_str} (no such file #{file})"
  88       end
  89       @language = lang_str
  90       @file = file
  91       scan
  92       return if @bot.plugins.nil?
  93       @bot.plugins.core_modules.each { |p|
  94         if p.respond_to?('set_language')
  95           p.set_language(@language)
  96         end
  97       }
  98       @bot.plugins.plugins.each { |p|
  99         if p.respond_to?('set_language')
 100           p.set_language(@language)
 101         end
 102       }
 103     end
 104
 105     def scan
 106       @strings = Hash.new
 107       current_key = nil
 108       IO.foreach(@file) {|l|
 109         next if l =~ /^$/
 110         next if l =~ /^\s*#/
 111         if(l =~ /^(\S+):$/)
 112           @strings[$1] = Array.new
 113           current_key = $1
 114         elsif(l =~ /^\s*(.*)$/)
 115           @strings[current_key] << $1
 116         end
 117       }
 118     end
 119
 120     def rescan
 121       scan
 122     end
 123
 124     def get(key)
 125       if(@strings.has_key?(key))
 126         return @strings[key][rand(@strings[key].length)]
 127       else
 128         raise "undefined language key"
 129       end
 130     end
 131
 132     def save
 133       File.open(@file, "w") {|file|
 134         @strings.each {|key,val|
 135           file.puts "#{key}:"
 136           val.each_value {|v|
 137             file.puts "   #{v}"
 138           }
 139         }
 140       }
 141     end
 142   end
 143
 144 end
 145 end