discourse/lib/onebox/normalizer.rb

# frozen_string_literal: true

module Onebox
  class Normalizer
    attr_reader :data

    def get(attr, length = nil, sanitize = true)
      return nil if Onebox::Helpers::blank?(data)

      value = data[attr]

      return nil if Onebox::Helpers::blank?(value)

      value = html_entities.decode(value)
      value = Sanitize.fragment(value) if sanitize
      value.strip!
      value = Onebox::Helpers.truncate(value, length) unless length.nil?

      value
    end

    def method_missing(attr, *args, &block)
      value = get(attr, *args)

      return nil if Onebox::Helpers::blank?(value)

      method_name = attr.to_s
      if method_name.end_with?(*integer_suffixes)
        value.to_i
      elsif method_name.end_with?(*url_suffixes)
        result = Onebox::Helpers.normalize_url_for_output(value)
        result unless Onebox::Helpers::blank?(result)
      else
        value
      end
    end

    private

    def integer_suffixes
      ['width', 'height']
    end

    def url_suffixes
      ['url', 'image', 'video']
    end

    def html_entities
      @html_entities ||= HTMLEntities.new
    end
  end
end