class GettextSimple

# File lib/gettext_simple.rb, line 4
def initialize(args = {})
  @args = {
    :encoding => "utf-8",
    :i18n => false,
    :default_locale => "en"
  }.merge(args)
  @locales = {}
  @debug = @args[:debug]
  
  # To avoid doing a hash lookup on every translate.
  @i18n = @args[:i18n]
  
  if @i18n
    @default_locale = I18n.default_locale
  else
    @default_locale = @args[:default_locale]
  end
  
  @locale = @default_locale
end

# File lib/gettext_simple.rb, line 37
def load_dir(dir)
  check_folders = ["LC_MESSAGES", "LC_ALL"]
  
  Dir.foreach(dir) do |file|
    debug "New locale folder found: '#{file}'." if @debug
    
    fn = "#{dir}/#{file}"
    if !File.directory?(fn) || !file.match(/^[a-z]{2}/)
      debug "Skipping: '#{file}'." if @debug
      next
    end
    
    @locales[file] = {} unless @locales[file]
    
    check_folders.each do |fname|
      fpath = "#{dir}/#{file}/#{fname}"
      next if !File.exists?(fpath) || !File.directory?(fpath)
      debug "Found subfolder in '#{file}': '#{fname}'." if @debug
      
      Dir.foreach(fpath) do |pofile|
        if pofile.match(/\.po$/)
          debug "Starting to parse '#{pofile}'." if @debug
          pofn = "#{dir}/#{file}/#{fname}/#{pofile}"
          scan_pofile(file, pofn)
        end
      end
    end
  end
end

# File lib/gettext_simple.rb, line 92
def locale_exists?(locale)
  @locales.key?(locale)
end

# File lib/gettext_simple.rb, line 30
def locale_for_thread=(newlocale)
  Thread.current[:gettext_simple_locale] = newlocale
end

# File lib/gettext_simple.rb, line 109
def locales
  return @locales.keys
end

# File lib/gettext_simple.rb, line 25
def register_kernel_methods
  require "#{File.dirname(__FILE__)}/../include/kernel_methods"
  $gettext_simple_kernel_instance = self
end

# File lib/gettext_simple.rb, line 96
def translate(str, replaces = nil)
  if @i18n
    locale = I18n.locale.to_s
  elsif locale = Thread.current[:gettext_simple_locale]
    # Locale already set through condition.
  else
    locale = @default_locale
  end
  
  translate_with_locale(locale, str, replaces)
end

# File lib/gettext_simple.rb, line 71
def translate_with_locale(locale, str, replaces = nil)
  str = str.to_s
  raise ArgumentError, "Locale was not found: '#{locale}' in '#{@locales.keys.join(", ")}'." unless @locales.key?(locale)
  
  if !@locales[locale].key?(str)
    translated_str = str.to_s
  else
    translated_str = @locales[locale][str]
  end
  
  if replaces
    replaces.each do |key, val|
      translated_str = translated_str.gsub("%{#{key}}", val.to_s)
    end
  end
  
  debug "Translation with locale '#{locale}' and replaces '#{replaces}': '#{str}' to '#{translated_str}'." if @debug
  
  return translated_str
end

# File lib/gettext_simple.rb, line 173
def add_translation(locale, key, val)
  raise ArgumentError, "No such language: '#{locale}'." unless @locales.key?(locale)
  
  if !key.to_s.empty? && !val.to_s.empty?
    debug "Found translation for locale '#{locale}' '#{key}' which is: '#{val}'." if @debug
    @locales[locale][key] = val
  end
end

# File lib/gettext_simple.rb, line 115
def debug(str)
  $stderr.puts str if @debug
end

# File lib/gettext_simple.rb, line 119
def scan_pofile(locale, filepath)
  current_id = nil
  current_translation = nil
  
  reading_id = false
  reading_translation = false
  
  fuzzy = false
  
  debug "Opening file for parsing: '#{filepath}' in locale '#{locale}'." if @debug
  File.open(filepath, "r", :encoding => @args[:encoding]) do |fp|
    fp.each_line do |line|
      if match = line.match(/^(msgid|msgstr)\s+"(.*)"$/)
        if match[1] == "msgid"
          current_id = match[2]
          reading_id = true
        elsif match[1] == "msgstr"
          current_translation = match[2]
          reading_id = false
          reading_translation = true
        else
          raise "Unknown translation parameter: '#{match[1]}'."
        end
      elsif match = line.match(/^"(.*)"$/)
        if reading_id
          current_id << match[1]
        elsif reading_translation
          current_translation << match[1]
        else
          raise "Text given but don't know what to do with it."
        end
      elsif line.match(/^(\r|)\n$/)
        if reading_id
          reading_id = false
        elsif reading_translation
          add_translation(locale, current_id, current_translation) unless fuzzy
          reading_translation = false
          current_id = nil
          current_translation = nil
          fuzzy = false
        end
      elsif line.match(/^#, fuzzy$/)
        fuzzy = true
      elsif line.start_with?("#")
        # Line is a comment - ignore.
      else
        raise "Couldn't understand line: '#{line}'."
      end
    end
    
    add_translation(locale, current_id, current_translation) if reading_translation && !fuzzy
  end
end