module EscapeUtils
  module HtmlSafety
    if "".respond_to? :html_safe?
      def self.escape_once(s)
        s = s.to_s
        if s.html_safe?
          s.html_safe
        else
          yield(s).html_safe
        end
      end

      def _escape_html(s)
        if s.html_safe?
          s.to_s.html_safe
        else
          EscapeUtils.escape_html(s.to_s).html_safe
        end
      end
    else
      def self.escape_once(s)
        yield s.to_s
      end

      def _escape_html(s)
        EscapeUtils.escape_html(s.to_s)
      end
    end
  end
end