Loofah custom scrubber for youtube and vimeo iframes

sanitize.rb

class CustomScrubber < Loofah::Scrubber
  ALLOWED_IFRAME_ATTRS = %w[allowfullscreen frameborder height src width].freeze
  ALLOWED_VIDEO_REGEX = %r{\A(?:https?:)?//(?:www\.)?youtube|vimeo(?:-nocookie)?\.com/}

  def scrub(node)
    if node.name == 'iframe' && node['src'] =~ ALLOWED_VIDEO_REGEX
      node.attribute_nodes.each { |a| a.remove unless ALLOWED_IFRAME_ATTRS.include?(a.name) }
      return CONTINUE
    end
    return CONTINUE if html5lib_sanitize(node) == CONTINUE
    node.before node.children
    node.remove
  end
end

# usage 
Loofah.fragment(resource).scrub!(CustomScrubber.new)