namespace :gettext do

  desc 'Rewrite .po files into a consistent msgmerge format'
  task :clean do
    load_gettext

    Dir.glob("locale/*/app.po") do |po_file|
      GetText::msgmerge(po_file, po_file, 'alaveteli', :msgmerge => [:sort_output, :no_location, :no_wrap])
    end
  end

  desc "Update pot/po files for a theme."
  task :find_theme => :environment do
    theme = find_theme(ENV['THEME'])
    load_gettext
    msgmerge = Rails.application.config.gettext_i18n_rails.msgmerge
    msgmerge ||= %w[--sort-output --no-location --no-wrap]
    GetText.update_pofiles_org(
      text_domain,
      theme_files_to_translate(theme),
      "version 0.0.1",
      :po_root => theme_locale_path(theme),
      :msgmerge => msgmerge
     )
   end

    desc 'Rewrite theme .po files into a consistent msgmerge format'
    task :clean_theme do
        theme = find_theme(ENV['THEME'])
        load_gettext

        Dir.glob("#{ theme_locale_path(theme) }/*/app.po") do |po_file|
            GetText::msgmerge(po_file, po_file, 'alaveteli',
                              :msgmerge => [:sort_output, :no_location, :no_wrap])
        end
   end

   desc 'Update locale files with slightly changed English msgids using a csv file of old to new strings'
   task :update_msgids_from_csv do
       mapping_file = find_mapping_file(ENV['MAPPING_FILE'])
       mappings = {}
       CSV.parse(clean_csv_mapping_file(mapping_file)) do |csv_line|
           from,to = csv_line
           mappings[from] = to
       end
       Dir.glob("locale/**/app.po").each do |po_file|
           lang_mappings = mappings.clone
           lines = []
           File.read(po_file).each_line do |line|
               /^msgid "(.*)"/ =~ line
               if $1 && mappings[$1]
                    lines << "msgid \"#{lang_mappings.delete($1)}\""
                 else
                    lines << line
                end
           end
           puts "Mappings unused in #{po_file}: #{lang_mappings.keys}" unless lang_mappings.empty?
           File.open(po_file, "w") { |f| f.puts(lines) }
       end
   end

   # Use a quote for quote-escaping as CSV errors on the \" with "Missing or stray quote"
   def clean_csv_mapping_file(file)
       data = ''
       File.foreach(file) do |line|
         data += line.gsub('\"', '""')
       end
       data
   end

   def find_theme(theme)
       unless theme
           puts "Usage: Specify an Alaveteli-theme with THEME=[theme directory name]"
           exit(1)
       end
       theme
   end

   def find_mapping_file(file)
       unless file
           puts "Usage: Specify a csv file mapping old to new strings with MAPPING_FILE=[file name]"
           exit(1)
       end
       unless File.exists?(file)
           puts "Error: MAPPING_FILE #{file} not found"
           exit(1)
       end
       file
   end

   def theme_files_to_translate(theme)
       Dir.glob("{lib/themes/#{theme}/lib}/**/*.{rb,erb}")
   end

   def theme_locale_path(theme)
     Rails.root.join "lib", "themes", theme, "locale-theme"
   end

end