discourse/lib/feed_element_installer.rb

# frozen_string_literal: true

require "rexml/document"
require "rss"

class FeedElementInstaller
  private_class_method :new

  def self.install(element_name, feed)
    # RSS Specification at http://cyber.harvard.edu/rss/rss.html#extendingRss
    # > A RSS feed may contain [non-standard elements], only if those elements are *defined in a namespace*

    new(element_name, feed).install if element_name.include?(":")
  end

  attr_reader :feed, :original_name, :element_namespace, :element_name, :element_accessor

  def initialize(element_name, feed)
    @feed = feed
    @original_name = element_name
    @element_namespace, @element_name = *element_name.split(":")
    @element_accessor = "#{@element_namespace}_#{@element_name}"
  end

  def element_uri
    @element_uri ||=
      REXML::Document.new(feed).root&.attributes&.namespaces&.fetch(@element_namespace, "") || ""
  end

  def install
    install_in_rss unless installed_in_rss?
    install_in_atom unless installed_in_atom?
  end

  private

  def install_in_rss
    RSS::Rss::Channel::Item.install_text_element(
      element_name,
      element_uri,
      "?",
      element_accessor,
      nil,
      original_name,
    )
    RSS::BaseListener.install_get_text_element(element_uri, element_name, element_accessor)
  end

  def install_in_atom
    RSS::Atom::Entry.install_text_element(
      element_name,
      element_uri,
      "?",
      element_accessor,
      nil,
      original_name,
    )
    RSS::Atom::Feed::Entry.install_text_element(
      element_name,
      element_uri,
      "?",
      element_accessor,
      nil,
      original_name,
    )
    RSS::BaseListener.install_get_text_element(element_uri, element_name, element_accessor)
  end

  def installed_in_rss?
    RSS::Rss::Channel::Item.method_defined?(element_accessor)
  end

  def installed_in_atom?
    RSS::Atom::Entry.method_defined?(element_accessor) ||
      RSS::Atom::Feed::Entry.method_defined?(element_accessor)
  end
end