require "active_generator/base"
require "generators/locale/base"

module Locale
  class TaskGenerator < Rails::Generators::NamedBase
    include ActiveGenerator::Base
    include Locale::Base

    source_root File.expand_path('../templates', __FILE__)
    argument :language, type: :string, required: false, default: "en", desc: "The locale alpha code"

    def copy_task_file
      assert_valid_language!
      template("task.yml", "config/locales/#{language}/tasks/#{file_name}.yml")
    end

  end
end

if ActiveGenerator.configuration.autoload_task_generator_locale
  begin
    require "rails/generators/rails/task/task_generator"

    Rails::Generators::TaskGenerator.class_eval do
      def generate_task_locale_file
        invoke("locale:task", [name])
      end
    end
  rescue LoadError => e
    # Skip over missing files
  end
end