module AbAdmin
  module I18nTools
    class ModelTranslator

      IGNORE_COLUMNS = %w(id reset_password_sent_at remember_created_at current_sign_in_at confirmation_token
                      reset_password_token password_salt failed_attempts)

      def initialize(options: {})
        @options = options
        @locales = AbAdmin.translated_locales
        @models = AbAdmin.translate_models.map{|m| m.constantize }
        @models_i18n_hash = {}
      end

      def make_hash
        @locales.each do |locale|
          I18n.with_locale(locale) do
            @models_i18n_hash[locale] = {'activerecord' => {'attributes' => {}, 'models' => {}}}

            models_hash = @models.each_with_object({}) do |model, h|
              model_i18n = {
                  'zero' => model.model_name.human(count: 0),
                  'one' => model.model_name.human(count: 1),
                  'few' => (model.model_name.human(count: 2) rescue model.model_name.human(count: 1)),
                  'many' => (model.model_name.human(count: 9) rescue model.model_name.human(count: 1)),
                  'other' => (model.model_name.human(count: 9) rescue model.model_name.human(count: 1))
              }
              @models_i18n_hash[locale]['activerecord']['models'][model.model_name.i18n_key.to_s]= model_i18n
              attributes = model.columns.map(&:name)
              attributes.concat(model.translated_attribute_names.map(&:to_s)) if model.try!(:translates?)
              attributes.reject! { |el| IGNORE_COLUMNS.include?(el) }
              h[model.model_name.i18n_key.to_s] = attributes.each_with_object({}) do |attr, o|
                o[attr] = ha(model, attr, locale).presence || attr
                model.reflect_on_all_associations.map(&:name).map(&:to_s).reject { |a| a =~ /^translation/ }.each do |assoc|
                  o[assoc] = ha(model, assoc, locale)
                end
                if @options[:locale_attributes] && model.new.respond_to?("#{attr}_#{locale.to_s}".to_sym)
                  @locales.each do |locale_1|
                    o["#{attr}_#{locale_1.to_s}"] = "#{ha(model, attr, locale)} (#{locale_1.to_s})"
                  end
                end
              end.sort.to_h
            end
            @models_i18n_hash[locale]['activerecord']['attributes'] = models_hash
          end
        end
      end

      def ha(model, attr, locale)
        model.human_attribute_name(attr, locale: locale)
      end

      def write_yaml
        locale_dir = Rails.root.join('config/locales')
        @locales.each do |locale|
          Locator.save(locale_dir.join("#{locale}.attrs.yml"), {locale.to_s => @models_i18n_hash[locale]})
        end
      end

      def self.i18n_models!
        model_translator = new
        model_translator.make_hash
        model_translator.write_yaml
      end

    end
  end
end