Chris@1464: desc 'Updates and checks locales against en.yml'
Chris@1464: task :locales do
Chris@1464:   %w(locales:update locales:check_interpolation).collect do |task|
Chris@1464:     Rake::Task[task].invoke
Chris@1464:   end
Chris@1464: end
Chris@1464: 
Chris@1464: namespace :locales do
Chris@1464:   desc 'Updates language files based on en.yml content (only works for new top level keys).'
Chris@1464:   task :update do
Chris@1464:     dir = ENV['DIR'] || './config/locales'
Chris@1464: 
Chris@1464:     en_strings = YAML.load(File.read(File.join(dir,'en.yml')))['en']
Chris@1464: 
Chris@1464:     files = Dir.glob(File.join(dir,'*.{yaml,yml}'))
Chris@1464:     files.sort.each do |file|
Chris@1464:       puts "Updating file #{file}"
Chris@1464:       file_strings = YAML.load(File.read(file))
Chris@1464:       file_strings = file_strings[file_strings.keys.first]
Chris@1464: 
Chris@1464:       missing_keys = en_strings.keys - file_strings.keys
Chris@1464:       next if missing_keys.empty?
Chris@1464: 
Chris@1464:       puts "==> Missing #{missing_keys.size} keys (#{missing_keys.join(', ')})"
Chris@1464:       lang = File.open(file, 'a')
Chris@1464: 
Chris@1464:       missing_keys.each do |key|
Chris@1464:         {key => en_strings[key]}.to_yaml.each_line do |line|
Chris@1464:           next if line =~ /^---/ || line.empty?
Chris@1464:           puts "  #{line}"
Chris@1464:           lang << "  #{line}"
Chris@1464:         end
Chris@1464:       end
Chris@1464: 
Chris@1464:       lang.close
Chris@1464:     end
Chris@1464:   end
Chris@1464: 
Chris@1464:   desc 'Checks interpolation arguments in locals against en.yml'
Chris@1464:   task :check_interpolation do
Chris@1464:     dir = ENV['DIR'] || './config/locales'
Chris@1464:     en_strings = YAML.load(File.read(File.join(dir,'en.yml')))['en']
Chris@1464:     files = Dir.glob(File.join(dir,'*.{yaml,yml}'))
Chris@1464:     files.sort.each do |file|
Chris@1464:       puts "parsing #{file}..."
Chris@1464:       file_strings = YAML.load_file(file)
Chris@1464:       unless file_strings.is_a?(Hash)
Chris@1464:         puts "#{file}: content is not a Hash (#{file_strings.class.name})"
Chris@1464:         next
Chris@1464:       end
Chris@1464:       unless file_strings.keys.size == 1
Chris@1464:         puts "#{file}: content has multiple keys (#{file_strings.keys.size})"
Chris@1464:         next
Chris@1464:       end
Chris@1464:       file_strings = file_strings[file_strings.keys.first]
Chris@1464: 
Chris@1464:       file_strings.each do |key, string|
Chris@1464:         next unless string.is_a?(String)
Chris@1464:         string.scan /%\{\w+\}/ do |match|
Chris@1464:           unless en_strings[key].nil? || en_strings[key].include?(match)
Chris@1464:             puts "#{file}: #{key} uses #{match} not found in en.yml"
Chris@1464:           end
Chris@1464:         end
Chris@1464:       end
Chris@1464:     end
Chris@1464:   end
Chris@1464: 
Chris@1464:   desc <<-END_DESC
Chris@1464: Removes a translation string from all locale file (only works for top-level childless non-multiline keys, probably doesn\'t work on windows).
Chris@1464: 
Chris@1464: This task does not work on Ruby 1.8.6.
Chris@1464: You need to use Ruby 1.8.7 or later.
Chris@1464: 
Chris@1464: Options:
Chris@1464:   key=key_1,key_2    Comma-separated list of keys to delete
Chris@1464:   skip=en,de         Comma-separated list of locale files to ignore (filename without extension)
Chris@1464: END_DESC
Chris@1464: 
Chris@1464:   task :remove_key do
Chris@1464:     dir = ENV['DIR'] || './config/locales'
Chris@1464:     files = Dir.glob(File.join(dir,'*.yml'))
Chris@1464:     skips = ENV['skip'] ? Regexp.union(ENV['skip'].split(',')) : nil
Chris@1464:     deletes = ENV['key'] ? Regexp.union(ENV['key'].split(',')) : nil
Chris@1464:     # Ignore multiline keys (begin with | or >) and keys with children (nothing meaningful after :)
Chris@1464:     delete_regex = /\A  #{deletes}: +[^\|>\s#].*\z/
Chris@1464: 
Chris@1464:     files.each do |path|
Chris@1464:       # Skip certain locales
Chris@1464:       (puts "Skipping #{path}"; next) if File.basename(path, ".yml") =~ skips
Chris@1464:       puts "Deleting selected keys from #{path}"
Chris@1464:       orig_content = File.open(path, 'r') {|file| file.read}
Chris@1464:       File.open(path, 'w') {|file| orig_content.each_line {|line| file.puts line unless line.chomp =~ delete_regex}}
Chris@1464:     end
Chris@1464:   end
Chris@1464: 
Chris@1464:   desc <<-END_DESC
Chris@1464: Adds a new top-level translation string to all locale file (only works for childless keys, probably doesn\'t work on windows, doesn't check for duplicates).
Chris@1464: 
Chris@1464: Options:
Chris@1464:   key="some_key=foo"
Chris@1464:   key1="another_key=bar"
Chris@1464:   key_fb="foo=bar"         Keys to add in the form key=value, every option of the form key[,\\d,_*] will be recognised
Chris@1464:   skip=en,de               Comma-separated list of locale files to ignore (filename without extension)
Chris@1464: END_DESC
Chris@1464: 
Chris@1464:   task :add_key do
Chris@1464:     dir = ENV['DIR'] || './config/locales'
Chris@1464:     files = Dir.glob(File.join(dir,'*.yml'))
Chris@1464:     skips = ENV['skip'] ? Regexp.union(ENV['skip'].split(',')) : nil
Chris@1464:     keys_regex = /\Akey(\d+|_.+)?\z/
Chris@1464:     adds = ENV.reject {|k,v| !(k =~ keys_regex)}.values.collect {|v| Array.new v.split("=",2)}
Chris@1464:     key_list = adds.collect {|v| v[0]}.join(", ")
Chris@1464: 
Chris@1464:     files.each do |path|
Chris@1464:       # Skip certain locales
Chris@1464:       (puts "Skipping #{path}"; next) if File.basename(path, ".yml") =~ skips
Chris@1464:       # TODO: Check for dupliate/existing keys
Chris@1464:       puts "Adding #{key_list} to #{path}"
Chris@1464:       File.open(path, 'a') do |file|
Chris@1464:         adds.each do |kv|
Chris@1464:           Hash[*kv].to_yaml.each_line do |line|
Chris@1464:             file.puts "  #{line}" unless (line =~ /^---/ || line.empty?)
Chris@1464:           end
Chris@1464:         end
Chris@1464:       end
Chris@1464:     end
Chris@1464:   end
Chris@1464: 
Chris@1464:   desc 'Duplicates a key. Exemple rake locales:dup key=foo new_key=bar'
Chris@1464:   task :dup do
Chris@1464:     dir = ENV['DIR'] || './config/locales'
Chris@1464:     files = Dir.glob(File.join(dir,'*.yml'))
Chris@1464:     skips = ENV['skip'] ? Regexp.union(ENV['skip'].split(',')) : nil
Chris@1464:     key = ENV['key']
Chris@1464:     new_key = ENV['new_key']
Chris@1464:     abort "Missing key argument" if key.blank?
Chris@1464:     abort "Missing new_key argument" if new_key.blank?
Chris@1464: 
Chris@1464:     files.each do |path|
Chris@1464:       # Skip certain locales
Chris@1464:       (puts "Skipping #{path}"; next) if File.basename(path, ".yml") =~ skips
Chris@1464:       puts "Adding #{new_key} to #{path}"
Chris@1464: 
Chris@1464:       strings = File.read(path)
Chris@1464:       unless strings =~ /^(  #{key}: .+)$/
Chris@1464:         puts "Key not found in #{path}"
Chris@1464:         next
Chris@1464:       end
Chris@1464:       line = $1
Chris@1464: 
Chris@1464:       File.open(path, 'a') do |file|
Chris@1464:         file.puts(line.sub(key, new_key))
Chris@1464:       end
Chris@1464:     end
Chris@1464:   end
Chris@1464: 
Chris@1464:   desc 'Check parsing yaml by psych library on Ruby 1.9.'
Chris@1464: 
Chris@1464:   # On Fedora 12 and 13, if libyaml-devel is available,
Chris@1464:   # in case of installing by rvm,
Chris@1464:   # Ruby 1.9 default yaml library is psych.
Chris@1464: 
Chris@1464:   task :check_parsing_by_psych do
Chris@1464:     begin
Chris@1464:       require 'psych'
Chris@1464:       parser = Psych::Parser.new
Chris@1464:       dir = ENV['DIR'] || './config/locales'
Chris@1464:       files = Dir.glob(File.join(dir,'*.yml'))
Chris@1464:       files.sort.each do |filename|
Chris@1464:         next if File.directory? filename
Chris@1464:         puts "parsing #{filename}..."
Chris@1464:         begin
Chris@1464:           parser.parse File.open(filename)
Chris@1464:         rescue Exception => e1
Chris@1464:           puts(e1.message)
Chris@1464:           puts("")
Chris@1464:         end
Chris@1464:       end
Chris@1464:     rescue Exception => e
Chris@1464:       puts(e.message)
Chris@1464:     end
Chris@1464:   end
Chris@1464: end