discourse/lib/cooked_post_processor.rb

# 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'
require_dependency 'image_optimizer'

class CookedPostProcessor

  def initialize(post, opts={})
    @dirty = false
    @opts = opts
    @post = post
    @doc = Nokogiri::HTML::fragment(post.cooked)
    @size_cache = {}
  end

  def dirty?
    @dirty
  end

  # Bake onebox content into the post
  def post_process_oneboxes
    args = {post_id: @post.id}
    args[:invalidate_oneboxes] = true if @opts[:invalidate_oneboxes]

    result = Oneboxer.apply(@doc) do |url, element|
      Oneboxer.onebox(url, args)
    end
    @dirty ||= result.changed?
  end

  # First let's consider the images
  def post_process_images
    images = @doc.search("img")
    return unless images.present?

    images.each do |img|
      src = img['src']
      src = Discourse.base_url_no_prefix + src if src[0] == "/"

      if src.present?

        if img['width'].blank? || img['height'].blank?
          w, h = get_size_from_image_sizes(src, @opts[:image_sizes]) || image_dimensions(src)

          if w && h
            img['width'] = w.to_s
            img['height'] = h.to_s
            @dirty = true
          end
        end

        if src != img['src']
          img['src'] = src
          @dirty = true
        end

        convert_to_link!(img)
        img['src'] = optimize_image(img)

      end
    end

    # 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

  end

  def optimize_image(img)
    src = img["src"]

    # supports only local uploads
    return src if SiteSetting.enable_imgur? || SiteSetting.enable_s3_uploads?

    width, height = img["width"].to_i, img["height"].to_i

    ImageOptimizer.new(src).optimized_image_url(width, height)
  end

  def convert_to_link!(img)
    src = img["src"]
    width, height = img["width"].to_i, 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

    # not a hyperlink so we can apply
    a = Nokogiri::XML::Node.new "a", @doc
    img.add_next_sibling(a)
    a["href"] = src
    a["class"] = "lightbox"
    a.add_child(img)
    @dirty = true

  end

  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

  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

  def get_size(url)
    return nil unless SiteSetting.crawl_images? || url.start_with?(Discourse.base_url)
    @size_cache[url] ||= FastImage.size(url)
  end

  def get_image_uri(url)
    uri = URI.parse(url)
    if %w(http https).include?(uri.scheme)
      uri
    else
      nil
    end
  end

  # Retrieve the image dimensions for a url
  def image_dimensions(url)
    uri = get_image_uri(url)
    return nil unless uri
    w, h = get_size(url)
    ImageSizer.resize(w, h) if w && h
  end

end