48 lines
1.1 KiB
Ruby
48 lines
1.1 KiB
Ruby
require 'banzai'
|
|
require 'html/pipeline/filter'
|
|
require 'uri'
|
|
|
|
module Banzai
|
|
module Filter
|
|
# HTML filter that "fixes" relative upload links to files.
|
|
# Context options:
|
|
# :project (required) - Current project
|
|
#
|
|
class UploadLinkFilter < HTML::Pipeline::Filter
|
|
def call
|
|
doc.search('a').each do |el|
|
|
process_link_attr el.attribute('href')
|
|
end
|
|
|
|
doc.search('img').each do |el|
|
|
process_link_attr el.attribute('src')
|
|
end
|
|
|
|
doc
|
|
end
|
|
|
|
protected
|
|
|
|
def process_link_attr(html_attr)
|
|
return if html_attr.blank?
|
|
|
|
uri = html_attr.value
|
|
if uri.starts_with?("/uploads/")
|
|
html_attr.value = build_url(uri).to_s
|
|
end
|
|
end
|
|
|
|
def build_url(uri)
|
|
File.join(Gitlab.config.gitlab.url, context[:project].path_with_namespace, uri)
|
|
end
|
|
|
|
# Ensure that a :project key exists in context
|
|
#
|
|
# Note that while the key might exist, its value could be nil!
|
|
def validate
|
|
needs :project
|
|
end
|
|
end
|
|
end
|
|
end
|