lib/govspeak.rb in govspeak-6.8.0 vs lib/govspeak.rb in govspeak-6.8.1

@@ -52,10 +52,11 @@
       options = options.dup.deep_symbolize_keys
       @source = source ? source.dup : ""
 
       @images = options.delete(:images) || []
       @allowed_elements = options.delete(:allowed_elements) || []
+      @allowed_image_hosts = options.delete(:allowed_image_hosts) || []
       @attachments = Array.wrap(options.delete(:attachments))
       @links = Array.wrap(options.delete(:links))
       @contacts = Array.wrap(options.delete(:contacts))
       @locale = options.fetch(:locale, "en")
       @options = { input: PARSER_CLASS_NAME,
@@ -67,10 +68,11 @@
     end
 
     def to_html
       @to_html ||= begin
         html = if @options[:sanitize]
-                 HtmlSanitizer.new(kramdown_doc.to_html).sanitize(allowed_elements: @allowed_elements)
+                 HtmlSanitizer.new(kramdown_doc.to_html, allowed_image_hosts: @allowed_image_hosts)
+                              .sanitize(allowed_elements: @allowed_elements)
                else
                  kramdown_doc.to_html
                end
 
         unless @footnote_definition_html.nil?