discourse/lib/cooked_post_processor.rb

154 lines
3.3 KiB
Ruby
Raw Normal View History

2013-02-05 14:16:51 -05:00
# Post processing that we can do after a post has already been cooked. For
# example, inserting the onebox content, or image sizes.
require_dependency 'oneboxer'
class CookedPostProcessor
2013-02-20 20:07:36 -05:00
require 'open-uri'
2013-02-19 01:57:14 -05:00
2013-02-05 14:16:51 -05:00
def initialize(post, opts={})
@dirty = false
@opts = opts
@post = post
@doc = Nokogiri::HTML::fragment(post.cooked)
2013-02-19 01:57:14 -05:00
@size_cache = {}
2013-02-05 14:16:51 -05:00
end
def dirty?
@dirty
end
# Bake onebox content into the post
2013-02-25 11:42:20 -05:00
def post_process_oneboxes
2013-02-05 14:16:51 -05:00
args = {post_id: @post.id}
args[:invalidate_oneboxes] = true if @opts[:invalidate_oneboxes]
result = Oneboxer.apply(@doc) do |url, element|
Oneboxer.onebox(url, args)
2013-02-25 11:42:20 -05:00
end
@dirty ||= result.changed?
2013-02-05 14:16:51 -05:00
end
2013-02-25 11:42:20 -05:00
# First let's consider the images
def post_process_images
2013-02-05 14:16:51 -05:00
images = @doc.search("img")
2013-02-25 11:42:20 -05:00
return unless images.present?
2013-02-05 14:16:51 -05:00
2013-02-19 01:57:14 -05:00
# Extract the first image from the first post and use it as the 'topic image'
if @post.post_number == 1
img = images.first
@post.topic.update_column :image_url, img['src'] if img['src'].present?
end
images.each do |img|
src = img['src']
src = Discourse.base_url_no_prefix + src if src[0] == "/"
2013-02-05 14:16:51 -05:00
2013-02-19 01:57:14 -05:00
if src.present? && (img['width'].blank? || img['height'].blank?)
2013-02-05 14:16:51 -05:00
2013-02-25 11:42:20 -05:00
w,h =
get_size_from_image_sizes(src, @opts[:image_sizes]) ||
2013-02-19 01:57:14 -05:00
image_dimensions(src)
if w && h
img['width'] = w.to_s
img['height'] = h.to_s
@dirty = true
end
end
2013-02-25 11:42:20 -05:00
if src.present?
2013-02-19 01:57:14 -05:00
if src != img['src']
img['src'] = src
@dirty = true
2013-02-05 14:16:51 -05:00
end
2013-02-25 11:42:20 -05:00
convert_to_link!(img)
2013-02-19 01:57:14 -05:00
img.set_attribute('src', optimize_image(src))
2013-02-05 14:16:51 -05:00
end
2013-02-19 01:57:14 -05:00
end
end
def optimize_image(src)
2013-02-20 20:07:36 -05:00
# uri = get_image_uri(src)
# uri.open(read_timeout: 20) do |f|
2013-02-25 11:42:20 -05:00
#
2013-02-20 20:07:36 -05:00
# end
2013-02-19 01:57:14 -05:00
src
end
def convert_to_link!(img)
src = img["src"]
width = img["width"].to_i
height = img["height"].to_i
return unless src.present? && width > SiteSetting.auto_link_images_wider_than
original_width, original_height = get_size(src)
return unless original_width.to_i > width && original_height.to_i > height
parent = img.parent
while parent
return if parent.name == "a"
break unless parent.respond_to? :parent
parent = parent.parent
end
2013-02-25 11:42:20 -05:00
# not a hyperlink so we can apply
2013-02-19 01:57:14 -05:00
a = Nokogiri::XML::Node.new "a", @doc
img.add_next_sibling(a)
a["href"] = src
a["class"] = "lightbox"
a.add_child(img)
@dirty = true
2013-02-05 14:16:51 -05:00
end
2013-02-19 01:57:14 -05:00
def get_size_from_image_sizes(src, image_sizes)
if image_sizes.present?
if dim = image_sizes[src]
ImageSizer.resize(dim['width'], dim['height'])
end
end
end
2013-02-05 14:16:51 -05:00
def post_process
return unless @doc.present?
post_process_images
post_process_oneboxes
end
def html
@doc.try(:to_html)
end
def doc
@doc
end
2013-02-19 01:57:14 -05:00
def get_size(url)
return nil unless SiteSetting.crawl_images? || url.start_with?(Discourse.base_url)
@size_cache[url] ||= FastImage.size(url)
end
2013-02-20 20:07:36 -05:00
def get_image_uri(url)
uri = URI.parse(url)
if %w(http https).include?(uri.scheme)
uri
else
nil
end
end
2013-02-05 14:16:51 -05:00
# Retrieve the image dimensions for a url
2013-02-19 01:57:14 -05:00
def image_dimensions(url)
2013-02-25 11:42:20 -05:00
uri = get_image_uri(url)
2013-02-20 20:07:36 -05:00
return nil unless uri
2013-02-19 01:57:14 -05:00
w, h = get_size(url)
ImageSizer.resize(w, h) if w && h
2013-02-05 14:16:51 -05:00
end
end