module AttrTranslate

# File lib/attr_translate/attr_translate.rb, line 8
def attr_translate(*attrs)
  class_attribute :translated_attribute_names, :permitted_translated_attributes

  self.translated_attribute_names = attrs
  self.permitted_translated_attributes = [
    *self.ancestors
      .select { |klass| klass.respond_to?(:permitted_translated_attributes) }
      .map(&:permitted_translated_attributes),
    *attrs.product(I18n.available_locales)
      .map { |attribute, locale| :"#{attribute}_#{locale}" }
  ].flatten.compact

  attrs.each do |attr_name|
    define_method attr_name do |**params|
      read_json_translation(attr_name, params)
    end

    define_method "#{attr_name}=" do |value|
      write_json_translation(attr_name, value)
    end

    define_singleton_method "with_#{attr_name}_translation" do |value, locale = I18n.locale|
      quoted_translation_store = connection.quote_column_name("#{attr_name}_translations")
      translation_hash = { "#{locale}" => value }
      where("#{quoted_translation_store} @> :translation::jsonb", translation: translation_hash.to_json)
    end
  end

  send(:prepend, TranslationAttribute)
end

# File lib/attr_translate/attr_translate.rb, line 46
def disable_fallback
  toggle_fallback(false)
end

# File lib/attr_translate/attr_translate.rb, line 50
def enable_fallback
  toggle_fallback(true)
end

# File lib/attr_translate/attr_translate.rb, line 40
def translates?
  true
end

# File lib/attr_translate/attr_translate.rb, line 58
def json_translate_fallback_locales(locale)
  return locale if enabled_fallback == false || !::I18n.respond_to?(:fallbacks)
  ::I18n.fallbacks[locale]
end

# File lib/attr_translate/attr_translate.rb, line 94
def method_missing_with_translates(method_name, *args)
  translated_attr_name, locale, assigning = parse_translated_attribute_accessor(method_name)

  return method_missing_without_translates(method_name, *args) unless translated_attr_name

  if assigning
    write_json_translation(translated_attr_name, args.first, locale)
  else
    read_json_translation(translated_attr_name, locale)
  end
end

# File lib/attr_translate/attr_translate.rb, line 120
def parse_translated_attribute_accessor(method_name)
  return unless /\A(?<attribute>[a-z0-9_]+)_(?<locale>[a-z]{2})(?<assignment>=?)\z/ =~ method_name

  translated_attr_name = attribute.to_sym
  return unless translated_attribute_names.include?(translated_attr_name)

  locale = locale.to_sym
  assigning = assignment.present?

  [translated_attr_name, locale, assigning]
end

# File lib/attr_translate/attr_translate.rb, line 63
def read_json_translation(attr_name, locale = I18n.locale, **params)
  translations = public_send("#{attr_name}_translations") || {}

  available = Array(json_translate_fallback_locales(locale)).detect do |available_locale|
    translations[available_locale.to_s].present?
  end

  translation = translations[available.to_s]
  # Rescue from MissingInterpolationArgument
  # so the default behaviour doesn't change.
  begin
    ::I18n.interpolate(translation, params) if translation
  rescue ::I18n::MissingInterpolationArgument
    translation
  end
end

# File lib/attr_translate/attr_translate.rb, line 89
def respond_to_with_translates?(symbol, include_all = false)
  return true if parse_translated_attribute_accessor(symbol)
  respond_to_without_translates?(symbol, include_all)
end

# File lib/attr_translate/attr_translate.rb, line 132
def toggle_fallback(enabled)
  if block_given?
    old_value = @enabled_fallback
    begin
      @enabled_fallback = enabled
      yield
    ensure
      @enabled_fallback = old_value
    end
  else
    @enabled_fallback = enabled
  end
end

# File lib/attr_translate/attr_translate.rb, line 80
def write_json_translation(attr_name, value, locale = I18n.locale)
  translation_store = "#{attr_name}_translations"
  translations = public_send(translation_store) || {}
  public_send("#{translation_store}_will_change!") unless translations[locale.to_s] == value
  translations[locale.to_s] = value
  public_send("#{translation_store}=", translations)
  value
end