rails--rails/actionmailer/lib/action_mailer/inline_preview_interceptor.rb

58 lines
1.4 KiB
Ruby

# frozen_string_literal: true
require "base64"
module ActionMailer
# Implements a mailer preview interceptor that converts image tag src attributes
# that use inline cid: style URLs to data: style URLs so that they are visible
# when previewing an HTML email in a web browser.
#
# This interceptor is enabled by default. To disable it, delete it from the
# <tt>ActionMailer::Base.preview_interceptors</tt> array:
#
# ActionMailer::Base.preview_interceptors.delete(ActionMailer::InlinePreviewInterceptor)
#
class InlinePreviewInterceptor
PATTERN = /src=(?:"cid:[^"]+"|'cid:[^']+')/i
include Base64
def self.previewing_email(message) #:nodoc:
new(message).transform!
end
def initialize(message) #:nodoc:
@message = message
end
def transform! #:nodoc:
return message if html_part.blank?
html_part.body = html_part.decoded.gsub(PATTERN) do |match|
if part = find_part(match[9..-2])
%[src="#{data_url(part)}"]
else
match
end
end
message
end
private
attr_reader :message
def html_part
@html_part ||= message.html_part
end
def data_url(part)
"data:#{part.mime_type};base64,#{strict_encode64(part.body.raw_source)}"
end
def find_part(cid)
message.all_parts.find { |p| p.attachment? && p.cid == cid }
end
end
end