### override mail method
module ActionMailer
  class Base
    alias_method :origin_mail, :mail
    def mail(headers={}, &block)
      # order by latest uploaded, for get the latest updated 'subject'
      current_template = MailEngine::MailTemplate.where(:path => "#{controller_path}/#{action_name}", :locale => I18n.locale, :partial => false).order("updated_at desc").first
      headers[:subject] = current_template.subject if headers[:subject].blank? && current_template.present?
      headers[:message_id] = "#{controller_path}/#{action_name}"

      # Add sendgrid header before sending mail.
      # Why here but not add to default_params of action_mailer? because the receiver email [:to] only can get here.
      if self.sendgrid_config
        self.sendgrid_config.add_to headers[:to]
        origin_mail(headers.merge(self.sendgrid_config.to_hash), &block)
      else
        origin_mail(headers, &block)
      end
    end

  protected

    ###
    # REASON TO OVERRIDE THIS METHOD:
    # need to set layout and pass partial path to each format of mail template.
    #
    # Completed copied from lib/action_mailer/base.rb#665
    # but used revisioned ActionMailer::Collector
    def collect_responses_and_parts_order(headers) #:nodoc:
      responses, parts_order = [], nil

      if block_given?
        ### modified this like, it's very strange when use resolver.is_a?(MailEngine::MailTemplate::Resolver) it will give false sometime.
        if lookup_context.view_paths.detect {|resolver| resolver.class.to_s == "MailEngine::MailTemplate::Resolver" }
          collector = ActionMailer::Collector.new(lookup_context) { |mime| set_layout_and_partials(mime) }
        else
          collector = ActionMailer::Collector.new(lookup_context) { render(action_name) }
        end
        ######################################################################################

        yield(collector)
        parts_order = collector.responses.map { |r| r[:content_type] }
        responses  = collector.responses
      elsif headers[:body]
        responses << {
          :body => headers.delete(:body),
          :content_type => self.class.default[:content_type] || "text/plain"
        }
      else
        templates_path = headers.delete(:template_path) || self.class.mailer_name
        templates_name = headers.delete(:template_name) || action_name

        each_template(templates_path, templates_name) do |template|
          self.formats = template.formats

          responses << {
            :body => render(:template => template),
            :content_type => template.mime_type.to_s
          }
        end
      end

      [responses, parts_order]
    end


    # set layout and partials
    def set_layout_and_partials(format)
      # looking for system mail.
      template = MailEngine::MailTemplate.where(:path => "#{controller_path}/#{action_name}", :format => format, :locale => I18n.locale, :partial => false, :for_marketing => false).first
      # looking for marketing mail.
      template = MailEngine::MailTemplate.where(:path => action_name, :format => format, :locale => I18n.locale, :partial => false, :for_marketing => true).first if template.blank?

      # if found db template set the layout and partial for it.
      if template
        related_partial_paths = {}
        # set @footer or @header
        template.template_partials.each do |tmp|
          related_partial_paths["#{tmp.placeholder_name}_path".to_sym] = tmp.partial.path
        end

        # set layout
        render :template => "#{controller_path}/#{action_name}", :layout => "layouts/mail_engine/mail_template_layouts/#{template.layout}", :locals => related_partial_paths
      else
        # if not found db template should render file template
        render(action_name)
     end
    end
  end
end

###
# REASON TO OVERRIDE THIS METHOD:
# pass mime format to renderrer
#
# One concern if user provided block, there will be problem. like:
# format.text { xxxx; render 'xxx' }
#
module ActionMailer
  class Collector
    def custom(mime, options={})
      options.reverse_merge!(:content_type => mime.to_s)
      @context.freeze_formats([mime.to_sym])
      ### modified this line
      # change from:
      # options[:body] = block_given? ? yield : @default_render.call
      options[:body] = block_given? ? yield : @default_render.call(mime.to_sym)
      @responses << options
    end
  end
end