module Rich
  module I18n
    module Core

      class EnrichedString
        attr_reader :meta_data, :merged_strings

        def initialize(string = "", meta = nil, merged = nil)
          @string         = string.dup
          @meta_data      = HashWithIndifferentAccess.new meta.try(:dup) || (string.meta_data.dup if string.enriched?) || {}
          @merged_strings = merged
        end

        def enriched?
          true
        end

        def html_safe?
          true
        end

        def concat(other)
         (@merged_strings ||= [EnrichedString.new @string, @meta_data]) << other
          @meta_data = nil
          @string.concat other
          self
        end
        alias_method :<<, :concat

        def +(other)
          dup.concat(other)
        end

        def to_str
          @string
        end

        def to_s
          (@merged_strings || [self]).collect(&:to_tag).join("").html_safe
        end

        def method_missing(method, *args)
          if @string.respond_to? method
            @string.send method
          else
            super
          end
        end

      protected

        def to_tag
          (!@meta_data.empty? && Engine.can_enrich_output?) ? to_enriched_tag : to_html
        end

        def to_enriched_tag
          tag   = :i18n
          attrs = []

          keys  = [:key, :value, :locale, :derivative_key]
          data  = @meta_data.reject{|k, v| !keys.include?(k.to_sym)}

          data[:editable_input_type] = editable_input_type

          attrs << @meta_data[:html].collect{|k, v|      "#{k}='#{::ERB::Util.html_escape v}'"}.join(" ") if @meta_data[:html]
          attrs << data             .collect{|k, v| "data-#{k}='#{::ERB::Util.html_escape v}'"}.join(" ")
          attrs << "data-i18n_translation='#{::ERB::Util.html_escape to_str}'"
          attrs << "data-i18n_tag='#{@meta_data[:tag]}'" if @meta_data[:tag]

          "<#{tag} #{attrs.join(" ")}></#{tag}>"
        end

        def to_html
          tag = (@meta_data[:tag] if @meta_data) || (:div if %w(text html).include?(editable_input_type))
          tag ? "<#{tag}>#{to_str}</#{tag}>" : to_str
        end

      private

        def editable_input_type
          @meta_data[:as] if @meta_data && %w(string text html).include?(@meta_data[:as].try(:to_s).try(:downcase))
        end

      end

    end
  end
end

EnrichedString = Rich::I18n::Core::EnrichedString