module TranslationCenter

  # includes models from the models folder
  # this method is needed to make sure that acts_as_translator is loaded
  # in the translator class
  def self.prepare_translator
    # if translator is not defined then load the models
    if TranslationCenter::Translation.translator.blank?
      Dir.glob(File.expand_path("app/models/*.rb", Rails.root)).each do |model_file|
        require model_file
      end
    end
    translator = TranslationCenter::Translation.translator.where(TranslationCenter::CONFIG['identifier_type'] => TranslationCenter::CONFIG['yaml_translator_identifier']).first
    # if translator doesn't exist then create him
    if translator.blank?
      translator = TranslationCenter::Translation.translator.new(TranslationCenter::CONFIG['identifier_type'] => TranslationCenter::CONFIG['yaml_translator_identifier'])
      begin
        translator.save(validate: false)
      rescue
        translator = nil
      end
    end
    translator
  end

  def self.included(base)
    base.class_eval do
      alias_method_chain :translate, :adding if(TranslationCenter::CONFIG['enabled'])
    end
  end

  # wraps a span if inspector option is set to all
  def wrap_span(translation, translation_key)
    # put the inspector class if inspector is all and the key doesn't belongs to translation_center
    if TranslationCenter::CONFIG['inspector'] == 'all' && translation_key.name.to_s.split('.').first != 'translation_center'
      "<span class='tc-inspector-key' data-type='#{translation_key.status(I18n.locale)}' data-id='#{translation_key.id}'> #{translation} </span>".html_safe
    else
      translation
    end
  end

  # make sure the complete key is build using the options such as scope and count
  def prepare_key(key, options)
    complete_key  = key

    # if a scope is passed in options then build the full key
    complete_key = options[:scope].present? ? "#{options[:scope].to_s}.#{complete_key}" : complete_key

    # add the correct count suffix
    if options[:count].present? && options[:count] == 1
      complete_key = "#{complete_key}.one"
    elsif options[:count].present? && options[:count] != 1
      complete_key = "#{complete_key}.other"
    end
    complete_key
  end

  def translate_with_adding(locale, key, options = {})
    # handle calling translation with a blank key
    # or translation center tables don't exist
    return translate_without_adding(locale, key, options) if key.blank? || !ActiveRecord::Base.connection.table_exists?('translation_center_translation_keys')

    complete_key = prepare_key(key, options) # prepare complete key

    # add the new key or update it
    translation_key = TranslationCenter::TranslationKey.find_or_create_by_name(complete_key)
    #  UNCOMMENT THIS LATER TO SET LAST ACCESSED AT
    # translation_key.update_attribute(:last_accessed, Time.now)

    # save the default value (Which is the titleized key name as the translation) if the option is enabled and no translation exists for that key in the db
    translation_key.create_default_translation if TranslationCenter::CONFIG['save_default_translation'] && translation_key.translations.in(:en).empty?

    # if i18n_source is set to db and not overriden by options then fetch from db
    if TranslationCenter::CONFIG['i18n_source']  == 'db' && options.delete(:yaml).blank?
      val = translation_key.accepted_translation_in(locale).try(:value) || options[:default]
      # replace variables in a translation with passed values
      options.each_pair{ |key, value| val.gsub!("%{#{key.to_s}}", value.to_s) } if val.is_a?(String)
      throw(:exception, I18n::MissingTranslation.new(locale, complete_key, options)) unless val
      wrap_span(val, translation_key)
    else
      # just return the normal I18n translation
      wrap_span(translate_without_adding(locale, key, options), translation_key)
    end
  end


  # load tha translation config
  if FileTest.exists?("config/translation_center.yml")
    TranslationCenter::CONFIG = YAML.load_file("config/translation_center.yml")[Rails.env]
    # identifier is by default email
    TranslationCenter::CONFIG['identifier_type'] ||= 'email'
  else
    puts "WARNING: translation_center will be using default options if config/translation_center.yml doesn't exists"
    TranslationCenter::CONFIG = {'enabled' => false, 'inspector' => 'missing', 'lang' => {'en' => {'name' => 'English', 'direction' => 'ltr'}}, 'yaml_translator_identifier' => 'coder@tc.com', 'i18n_source' => 'yaml', 'yaml2db_translations_accepted' => true,
                                'accept_admin_translations' => true,  'save_default_translation' => true, 'identifier_type' => 'email' }
  end
  I18n.available_locales = TranslationCenter::CONFIG['lang'].keys

end

# override html_message to add a class to the returned span
module I18n
  class MissingTranslation
    module Base
      # added another class to be used
      def html_message
        category = keys.first
        key = keys.last.to_s.gsub('_', ' ').gsub(/\b('?[a-z])/) { $1.capitalize }
        translation_key = keys
        # remove locale
        translation_key.shift

        translation_key = TranslationCenter::TranslationKey.find_by_name(translation_key.join('.'))
        # don't put the inspector class if inspector is off or the key belongs to translation_center
        if TranslationCenter::CONFIG['inspector'] == 'off' || category == 'translation_center'
          %(<span class="translation_missing" title="translation missing: #{keys.join('.')}">#{key}</span>)
        else
          %(<span class="translation_missing tc-inspector-key" data-type="#{translation_key.status(I18n.locale)}" data-id="#{translation_key.id}" title="translation missing: #{keys.join('.')}">#{key}</span>)
        end
      end

    end
  end
end

I18n::Backend::Base.send :include, TranslationCenter