require_dependency 'oneboxer' class PostAnalyzer def initialize(raw, topic_id) @raw = raw @topic_id = topic_id @found_oneboxes = false end def found_oneboxes? @found_oneboxes end # What we use to cook posts def cook(*args) cooked = PrettyText.cook(*args) result = Oneboxer.apply(cooked, topic_id: @topic_id) do |url, _| @found_oneboxes = true Oneboxer.invalidate(url) if args.last[:invalidate_oneboxes] Oneboxer.cached_onebox(url) end cooked = result.to_html if result.changed? cooked end # How many images are present in the post def image_count return 0 unless @raw.present? cooked_stripped.css("img").reject do |t| if dom_class = t["class"] (Post.white_listed_image_classes & dom_class.split).count > 0 end end.count end # How many attachments are present in the post def attachment_count return 0 unless @raw.present? attachments = cooked_stripped.css("a.attachment[href^=\"#{Discourse.store.absolute_base_url}\"]") attachments += cooked_stripped.css("a.attachment[href^=\"#{Discourse.store.relative_base_url}\"]") if Discourse.store.internal? attachments.count end def raw_mentions return [] if @raw.blank? return @raw_mentions if @raw_mentions.present? raw_mentions = cooked_stripped.css('.mention, .mention-group').map do |e| if name = e.inner_text name = name[1..-1] name.downcase! if name name end end raw_mentions.compact! raw_mentions.uniq! @raw_mentions = raw_mentions end # from rack ... compat with ruby 2.2 def self.parse_uri_rfc2396(uri) @parser ||= defined?(URI::RFC2396_Parser) ? URI::RFC2396_Parser.new : URI @parser.parse(uri) end # Count how many hosts are linked in the post def linked_hosts return {} if raw_links.blank? return @linked_hosts if @linked_hosts.present? @linked_hosts = {} raw_links.each do |u| begin uri = self.class.parse_uri_rfc2396(u) host = uri.host @linked_hosts[host] ||= 1 unless host.nil? rescue URI::InvalidURIError # An invalid URI does not count as a raw link. next end end @linked_hosts end # Returns an array of all links in a post excluding mentions def raw_links return [] unless @raw.present? return @raw_links if @raw_links.present? @raw_links = [] cooked_stripped.css("a[href]").each do |l| # Don't include @mentions in the link count next if l['href'].blank? || link_is_a_mention?(l) @raw_links << l['href'].to_s end @raw_links end # How many links are present in the post def link_count raw_links.size end private def cooked_stripped @cooked_stripped ||= begin doc = Nokogiri::HTML.fragment(cook(@raw, topic_id: @topic_id)) doc.css("pre, code, aside.quote, .onebox, .elided").remove doc end end def link_is_a_mention?(l) html_class = l['class'] return false if html_class.blank? href = l['href'].to_s html_class.to_s['mention'] && href[/^\/u\//] || href[/^\/users\//] end end