module AwesomeMailer
  class Renderer
    attr_accessor :document

    def initialize(document_string)
      self.document = Nokogiri::HTML.parse(document_string)
      stylesheets = document.search('link[@rel=stylesheet]')
      stylesheets.each do |stylesheet|
        # Must be intended for digital screens!
        load_stylesheet(stylesheet) if stylesheet['media'] =~ /^(all|handheld|screen)$/
      end
      apply_css!(document)
    end

    def to_html
      document.to_html
    end

    private
    def append_styles_to_body!(document, selector, declarations)
      declarations = "#{declarations.join('; ')};"
      document.search(selector).each do |element|
        element['style'] = [element['style'], declarations].compact.join(';')
      end
    end

    def append_styles_to_head!(document, selector, declarations, media = nil)
      selector_and_declarations = "#{selector} { #{declarations.join('; ')}; }"
      if media != :all
        selector_and_declarations = "@media #{media} { #{selector_and_declarations} }"
      end
      header_stylesheet.inner_html += "#{selector_and_declarations}\n"
    end

    def apply_css!(document)
      css_parser.rules_by_media_query.each do |media_query, rules|
        rules.each do |rule|
          rule.each_selector do |selector, declarations, specificity|
            # Rewrite relative URLs to match their parent CSS's URL path
            rewrite_relative_urls(declarations) if host
            declarations = declarations.split(';').map(&:strip)
            if media_query != :all || selector =~ /(^@|:|moz)/
              # Include difficult styles in the head
              append_styles_to_head!(document, selector, declarations, media_query)
            else
              # Strip vendor prefix styles and append them to the header
              vendor_specific_declarations = declarations.select {|declaration| declaration =~ /^-/ }
              if vendor_specific_declarations.any?
                declarations -= vendor_specific_declarations
                append_styles_to_head!(document, selector, vendor_specific_declarations, media_query)
              end

              # Include regular styles inline
               append_styles_to_body!(document, selector, declarations)
            end
          end
        end
      end
    end

    def asset_pipeline_path
      return false unless sprockets?
      /^#{Regexp.escape(Rails.configuration.assets[:prefix])}\//
    end

    def css_parser
      @css_parser ||= CssParser::Parser.new
    end

    def head
      @head ||= document.at('head') || Nokogiri::XML::Node.new('head', document.root).tap do |head|
        document.root.children.first.add_previous_sibling(head)
      end
    end

    def header_stylesheet
      @header_stylesheet ||= head.at('style[@type="text/css"]') || Nokogiri::XML::Node.new('style', head).tap do |style|
        style['type'] = 'text/css'
        style.inner_html = "\n"
        head.add_child(style)
      end
    end

    def host
      if host = AwesomeMailer::Base.default_url_options[:host]
        Addressable::URI.heuristic_parse(host, scheme: 'http')
      end
    end

    def load_stylesheet(stylesheet)
      stylesheet_path = stylesheet['href'].split('?').shift
      stylesheet_path.gsub!(/^#{Regexp.escape(host)}/, '') if host
      case stylesheet_path
      when asset_pipeline_path
        if asset = read_asset_pipeline_asset(stylesheet_path)
          css_parser.add_block!(asset.to_s)
        end
      when /^\//
        local_path = rails? && Rails.root.join('public', stylesheet_path.gsub(/^\//, '')).to_s
        css_parser.load_file!(local_path) if local_path && File.file?(local_path)
      else
        dirname = File.dirname(stylesheet['href'])
        css_parser.load_uri!(stylesheet['href'], base_uri: dirname)
      end
      stylesheet.remove
    end

    def rails?
      defined? Rails
    end

    def read_asset_pipeline_asset(path)
      path = path.gsub(asset_pipeline_path, '')
      Rails.application.assets[path]
    end

    def rewrite_relative_urls(css_declarations)
      css_declarations.scan(/(url\s*\(?["']+(.[^'"]*)["']\))/i).each do |url_command, item|
        next if item =~ /^http(s){0,1}:\/\//
        item_url = host.dup
        item_url.path = File.join(item_url.path, item)
        new_url_command = url_command.gsub(item, item_url.to_s)
        css_declarations[url_command] = new_url_command
      end
    end

    def sprockets?
      rails? && Rails.application.respond_to?(:assets)
    end
  end
end