2015-10-24 18:46:33 +05:30
|
|
|
require 'uri'
|
|
|
|
|
2015-12-23 02:04:40 +05:30
|
|
|
module Banzai
|
|
|
|
module Filter
|
2015-10-24 18:46:33 +05:30
|
|
|
# HTML filter that "fixes" relative upload links to files.
|
|
|
|
# Context options:
|
|
|
|
# :project (required) - Current project
|
|
|
|
#
|
|
|
|
class UploadLinkFilter < HTML::Pipeline::Filter
|
|
|
|
def call
|
2016-06-02 11:05:42 +05:30
|
|
|
return doc unless project
|
|
|
|
|
2015-10-24 18:46:33 +05:30
|
|
|
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)
|
2016-06-02 11:05:42 +05:30
|
|
|
File.join(Gitlab.config.gitlab.url, project.path_with_namespace, uri)
|
|
|
|
end
|
|
|
|
|
|
|
|
def project
|
|
|
|
context[:project]
|
2015-10-24 18:46:33 +05:30
|
|
|
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
|